improved error handling

1 files changed, 155 insertions, 0 deletions

diff --git a/lib/rest_client_wrapper.rb b/lib/rest_client_wrapper.rb
new file mode 100644
index 0000000..1532a79
--- /dev/null
+++ b/lib/rest_client_wrapper.rb
@@ -0,0 +1,155 @@
+module OpenTox
+
+  #PENDING: implement ot error api, move to own file
+  class Error
+    
+    attr_accessor :code, :body, :uri, :payload, :headers
+    
+    def initialize(code, body, uri, payload, headers)
+      self.code = code
+      self.body = body
+      self.uri = uri
+      self.payload = payload
+      self.headers = headers
+    end
+    
+    def self.parse(error_array_string)
+      begin
+        err = YAML.load(error_array_string)
+        if err.is_a?(Array) and err.size>0 and err[0].is_a?(Error)
+          return err
+        else
+          return nil
+        end
+      rescue
+        return nil
+      end
+    end
+    
+  end
+
+  module RestClientWrapper
+    
+    # PENDING: remove as soon as redirect tasks are remove from partner webservices
+    def self.redirect_task( uri )
+     raise "no redirect task uri: "+uri.to_s unless uri.to_s =~ /194.141.0.136|ambit.*task|tu-muenchen.*task/
+     
+     while (uri.to_s =~ /194.141.0.136|ambit.*task|tu-muenchen.*task/) 
+       #HACK handle redirect
+       LOGGER.debug "REDIRECT TASK: "+uri.to_s
+       redirect = ""
+       while (redirect.size == 0)
+         IO.popen("bin/redirect.sh "+uri.to_s) do |f| 
+           while line = f.gets
+             redirect += line.chomp
+           end
+         end
+         raise "TASK ERROR" if $?!=0
+         sleep 0.3
+       end
+       uri = redirect
+       LOGGER.debug "REDIRECT TO: "+uri.to_s
+     end
+     return uri
+    end
+     
+    def self.get(uri, headers=nil, wait_for_task=false, curl_hack=false)
+      execute( "get", uri, nil, headers, wait_for_task, curl_hack )
+    end
+    
+    def self.post(uri, payload=nil, headers=nil, wait_for_task=false, curl_hack=false)
+      execute( "post", uri, payload, headers, wait_for_task, curl_hack )
+    end
+
+    def self.delete(uri, headers=nil, wait_for_task=false, curl_hack=false)
+      execute( "delete", uri, nil, headers, wait_for_task, curl_hack )
+    end
+
+    private
+    def self.execute( rest_call, uri, payload, headers, wait_for_task=false, curl_hack=false )
+      
+      do_halt 400,"uri is null",uri,payload,headers unless uri
+      begin
+        
+        payload.each{ |k,v| payload.delete(k) if v==nil } if payload #remove keys with empty values, as this can cause problems
+        unless curl_hack
+          if payload
+            result = RestClient.send(rest_call, uri, payload, headers)
+          else  
+            result = RestClient.send(rest_call, uri, headers)
+          end
+        else
+         result = ""
+         cmd = " curl -v -X "+rest_call.upcase+" "+payload.collect{|k,v| "-d "+k.to_s+"='"+v.to_s+"' "}.join("")+" "+uri.to_s
+         LOGGER.debug "CURL HACK: "+cmd
+         IO.popen(cmd+" 2> /dev/null") do |f| 
+           while line = f.gets
+             result += line
+           end
+         end
+         result.chomp!("\n")
+         LOGGER.debug "CURL -> "+result.to_s
+         raise "curl call failed "+result if $?!=0
+         #raise "STOP "+result
+      end
+       
+      if wait_for_task
+        if result.to_s =~ /ambit.*task|tu-muenchen.*task/
+          result = redirect_task(result)
+        elsif Utils.task_uri?(result)
+          task = OpenTox::Task.find(result)
+          task.wait_for_completion
+          raise task.description if task.failed?
+          result = task.resource
+        end
+      end
+      return result
+        
+      rescue RestClient::RequestFailed => ex
+        do_halt ex.http_code,ex.http_body,uri,payload,headers
+      rescue RestClient::RequestTimeout => ex
+        do_halt 408,ex.message,uri,payload,headers
+      rescue => ex
+        begin
+          code = ex.http_code
+          msg = ex.http_body
+        rescue
+          code = 500
+          msg = ex.to_s
+        end
+        do_halt code,msg,uri,payload,headers
+      end
+    end
+    
+    def self.do_halt( code, body, uri, payload, headers )
+      
+      #build error
+      causing_errors = Error.parse(body)
+      if causing_errors
+        error = causing_errors + [Error.new(code, "subsequent error", uri, payload, headers)]
+      else
+        error = [Error.new(code, body, uri, payload, headers)]
+      end
+
+#     debug utility: write error to file       
+#      error_dir = "/tmp/ot_errors"
+#      FileUtils.mkdir(error_dir) unless File.exist?(error_dir)
+#      raise "could not create error dir" unless File.exist?(error_dir) and File.directory?(error_dir)
+#      file_name = "error"
+#      time=Time.now.strftime("%m.%d.%Y-%H:%M:%S")
+#      count = 1
+#      count+=1 while File.exist?(File.join(error_dir,file_name+"_"+time+"_"+count.to_s))
+#      File.new(File.join(error_dir,file_name+"_"+time+"_"+count.to_s),"w").puts(body)
+      
+      # return error (by halting, halts should be logged)
+      # PENDING always return yaml for now
+      if defined?(halt)
+        halt(502,error.to_yaml)
+      elsif defined?($sinatra)
+        $sinatra.halt(502,error.to_yaml)
+      else
+        raise error.to_yaml
+      end
+    end
+  end
+end