steam 0.0.5 → 0.0.6

data/lib/steam/browser/html_unit.rb

@@ -10,9 +10,10 @@ module Steam
     class HtmlUnit
       autoload :Actions,     'steam/browser/html_unit/actions'
       autoload :Client,      'steam/browser/html_unit/client'
+      autoload :Connection,  'steam/browser/html_unit/connection'
       autoload :Drb,         'steam/browser/html_unit/drb'
       autoload :Forker,      'steam/browser/html_unit/forker'
-      autoload :Connection,  'steam/browser/html_unit/connection'
+      autoload :Handler,     'steam/browser/html_unit/handler'
       autoload :Matchers,    'steam/browser/html_unit/matchers'
       autoload :Page,        'steam/browser/html_unit/page'
       autoload :WebResponse, 'steam/browser/html_unit/web_response'
@@ -71,6 +72,26 @@ module Steam
         Locator.within(response.body, *args, &block)
       end
 
+      def set_handler(type, &block) # TODO use delegate
+        @client.set_handler(type, &block)
+      end
+
+      def remove_handler(type)
+        @client.remove_handler(type)
+      end
+
+      def get_cookie(name)
+        @client.get_cookie(name)
+      end
+
+      def add_cookie(name, value)
+        @client.add_cookie(name, value)
+      end
+
+      def clear_cookies
+        @client.clear_cookies
+      end
+
       protected
 
         def respond_to

data/lib/steam/browser/html_unit/client.rb

@@ -13,6 +13,8 @@ module Steam
         Java.import 'com.gargoylesoftware.htmlunit.BrowserVersion'
         Java.import 'com.gargoylesoftware.htmlunit.NicelyResynchronizingAjaxController'
         Java.import 'com.gargoylesoftware.htmlunit.FailingHttpStatusCodeException'
+        Java.import 'org.apache.commons.httpclient.Cookie', 'Java::Com::Gargoylesoftware::Htmlunit::Cookie'
+        # Java.import 'com.gargoylesoftware.htmlunit.util.Cookie'
 
         include Java::Com::Gargoylesoftware::Htmlunit
 
@@ -20,10 +22,11 @@ module Steam
           def notify(message, origin); end
         end
 
-        attr_reader :connection
+        attr_reader :connection, :handlers
 
         def initialize(connection = nil, options = {})
           @connection = connection
+          @handlers   = {}
           options = Steam.config[:html_unit].merge(options)
 
           @java = WebClient.new(BrowserVersion.send(options[:browser_version]))
@@ -53,11 +56,11 @@ module Steam
         def get(request)
           perform(self.request_settings(request))
         end
-        
+
         def perform(request_settings)
           @java._invoke('getPage', 'Lcom.gargoylesoftware.htmlunit.WebRequestSettings;', request_settings)
         end
-        
+
         def request_settings(request)
           url      = Java::Net::Url.new(request.url)
           settings = WebRequestSettings.new(url)
@@ -70,7 +73,30 @@ module Steam
           yield if block_given?
         end
 
-        # FIXME setLevel throws "Fail: unknown method name `setLevel'". weird.
+        def set_handler(type, &block)
+          @java.send(:"set#{type.to_s.camelize}Handler", Handler.create(type, block))
+        end
+
+        def remove_handler(type)
+          @java.send(:"set#{type.to_s.camelize}Handler", nil)
+        end
+
+        def get_cookie(name)
+          cookie = @java.getCookieManager.getCookie(name)
+          cookie ? cookie.getValue : nil
+        end
+
+        def add_cookie(name, value) # TODO what about domain, path, expires etc?
+          cookie = Cookie.new
+          cookie.setName(name)
+          cookie.setValue(value)
+          @java.getCookieManager.addCookie(cookie)
+        end
+
+        def clear_cookies
+          @java.getCookieManager.clearCookies
+        end
+
         def log_level=(level)
           [ 'com.gargoylesoftware.htmlunit',
             'com.gargoylesoftware.htmlunit.html',

data/lib/steam/browser/html_unit/handler.rb

@@ -0,0 +1,36 @@
+module Steam
+  module Browser
+    class HtmlUnit
+      module Handler
+        class << self
+          def create(type, proc)
+            type    = type.to_s.camelize
+            handler = self.const_get(type).new(proc)
+            Rjb::bind(handler, "com.gargoylesoftware.htmlunit.#{type}Handler")
+          end
+        end
+        
+        class Base
+          attr_reader :proc
+
+          def initialize(proc)
+            @proc = proc
+          end
+        end
+
+        class Alert < Base
+          def handleAlert(page, message)
+            proc.call(page, message)
+          end
+        end
+
+        class Confirm < Base
+          def handleConfirm(page, message)
+            result = proc.call(page, message)
+            result.nil? || !!result
+          end
+        end
+      end
+    end
+  end
+end

data/lib/steam/version.rb

@@ -1,3 +1,3 @@
 module Steam
-  VERSION = "0.0.5"
+  VERSION = "0.0.6"
 end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 5
-  version: 0.0.5
+  - 6
+  version: 0.0.6
 platform: ruby
 authors: 
 - Sven Fuchs
@@ -65,6 +65,7 @@ files:
 - lib/steam/browser/html_unit/client.rb
 - lib/steam/browser/html_unit/connection.rb
 - lib/steam/browser/html_unit/drb.rb
+- lib/steam/browser/html_unit/handler.rb
 - lib/steam/browser/html_unit/page.rb
 - lib/steam/browser/html_unit/web_response.rb
 - lib/steam/browser/html_unit.rb