phantom_client 1.0.0 → 1.0.1

data/lib/phantom_client/phantom_client.rb

@@ -2,6 +2,10 @@ require 'net/http'
 require 'uri'
 
 module PhantomJSProxy
+  class DummyResponse
+    attr_accessor :body
+    attr_accessor :code
+  end
 	class PhantomJSClient
 	
 		attr_accessor :proxy_addr
@@ -14,6 +18,10 @@ module PhantomJSProxy
 			@connection = con
 		end
 		
+		def get_body(addr, options=nil)
+		  get(addr, options).body
+		end
+		
 		def get(addr, options=nil)			
 			url = URI.parse(addr)
 			req = Net::HTTP::Get.new(url.path)
@@ -28,7 +36,7 @@ module PhantomJSProxy
 			end
 			
 			if options && options['withIframes']
-				req['HTTP_GET_PAGE_WITH_IFRAMES'] = options['withIframes']
+				req['Get-Page-With-IFrames'] = options['withIframes']
 				puts "Do fetch iframes"
 			end
 			#::Proxy(@proxy_addr, @proxy_port)
@@ -54,7 +62,10 @@ module PhantomJSProxy
           @connection.do_request(element, url, req)
         rescue
           if count == 0
-            return "Could not connect to proxy"
+            resp = DummyResponse.new()
+            resp.code = 500
+            resp.body ="Could not connect to proxy"
+            return resp
           end
           do_get(url, req, count-1)
         end

data/lib/phantom_client/phantom_client_connection.rb

@@ -7,7 +7,7 @@ module PhantomJSProxy
 				res = Net::HTTP::Proxy(proxy[:addr], proxy[:port]).start(url.host, url.port) {|http|
 					http.request(req)
 				}
-				return res.body
+				return res
 		end
 	end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: phantom_client
 version: !ruby/object:Gem::Version 
-  hash: 23
+  hash: 21
   prerelease: 
   segments: 
   - 1
   - 0
-  - 0
-  version: 1.0.0
+  - 1
+  version: 1.0.1
 platform: ruby
 authors: 
 - Daniel Sudmann
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-03-23 00:00:00 +01:00
+date: 2012-04-19 00:00:00 +02:00
 default_executable: 
 dependencies: []