expose reponse header and raw body

Author: bishosilwal

lib/imagekitio/request.rb

@@ -42,17 +42,19 @@ def request(method, url, headers = create_headers, payload = nil)
           http.use_ssl = (uri.scheme == 'https')
           req = Net::HTTP::Post::Multipart.new uri.path, payload, headers
           resp = http.request(req)
-          if resp.code.to_i == 400
-            raise RestClient::ExceptionWithResponse, OpenStruct.new(code: 400, body: resp.body)
-          end
-          if resp.code.to_i == 403
-            raise RestClient::ExceptionWithResponse, OpenStruct.new(code: 403, body: resp.body)
+          response[:headers] = resp.to_hash
+          response[:raw_body] = resp.body
+          resp_c = resp.code.to_i
+          if [400, 403].include?(resp_c)
+            raise RestClient::ExceptionWithResponse, OpenStruct.new({ body: resp.body, code: resp_c })
           end
         else
           resp = RestClient::Request.new(method: method,
                                          url: url,
                                          headers: headers,
                                          payload: payload).execute
+          response[:headers] = resp.raw_headers
+          response[:raw_body] = resp.body
         end
         if (resp.code.to_i >= 200) && (resp.code.to_i < 204)
           content_type = resp.respond_to?(:headers) ? resp.headers[:content_type] : resp.content_type

test/imagekit/request_test.rb

@@ -85,5 +85,23 @@
       expect(response).to have_key(:error)
       expect(response[:error]).to eq({"message"=>"Your account cannot be authenticated.", "help"=>"For support kindly contact us at [email protected] ."})
     end
+
+    it 'test_response_headers' do
+      stub_request(:post, 'https://www.exampleservererror/upload').to_return(status: 200, body: "{\"message\":\"Success.\"}", headers: {"strict-transport-security"=>["max-age=15552000"],
+                                                                                                                                        "x-ik-requestid"=>["39e19bd4-c9c3-4025-453e-a13e6a412aa0"],
+                                                                                                                                        "content-type"=>["application/json; charset=utf-8"],
+                                                                                                                                        "x-request-id"=>["39e19bd4-c9c3-4025-a043-a13e6a412aa0"]})
+      response = @request_obj.request(:post, 'https://www.exampleservererror/upload', nil, { multipart: true })
+      expect(response[:headers]).to eq({"strict-transport-security"=>["max-age=15552000"],
+                                        "x-ik-requestid"=>["39e19bd4-c9c3-4025-453e-a13e6a412aa0"],
+                                        "content-type"=>["application/json; charset=utf-8"],
+                                        "x-request-id"=>["39e19bd4-c9c3-4025-a043-a13e6a412aa0"]})
+    end
+
+    it 'test_response_raw_body' do
+      stub_request(:post, 'https://www.exampleservererror/upload').to_return(status: 200, body: "{\"message\":\"Success.\"}", headers: {"strict-transport-security"=>["max-age=15552000"], "content-type"=>["application/json; charset=utf-8"]})
+      response = @request_obj.request(:post, 'https://www.exampleservererror/upload', nil, { multipart: false })
+      expect(response[:raw_body]).to eq("{\"message\":\"Success.\"}")
+    end
   end
 end