lib/pwn/www/coinbase_pro.rb in pwn-0.4.701 vs lib/pwn/www/coinbase_pro.rb in pwn-0.4.702

@@ -13,11 +13,12 @@
       # )
 
       public_class_method def self.open(opts = {})
         browser_obj = PWN::Plugins::TransparentBrowser.open(opts)
 
-        browser_obj.goto('https://pro.coinbase.com')
+        browser = browser_obj[:browser]
+        browser.goto('https://pro.coinbase.com')
 
         browser_obj
       rescue StandardError => e
         raise e
       end
@@ -33,28 +34,30 @@
       public_class_method def self.login(opts = {})
         browser_obj = opts[:browser_obj]
         username = opts[:username].to_s.scrub.strip.chomp
         password = opts[:password]
 
+        browser = browser_obj[:browser]
+
         if password.nil?
           password = PWN::Plugins::AuthenticationHelper.mask_password
         else
           password = opts[:password].to_s.scrub.strip.chomp
         end
         mfa = opts[:mfa]
 
-        browser_obj.goto('https://pro.coinbase.com')
+        browser.goto('https://pro.coinbase.com')
 
-        browser_obj.span(text: 'Sign in').wait_until(&:present?).click
-        browser_obj.text_field(name: 'email').wait_until(&:present?).set(username)
-        browser_obj.text_field(name: 'password').wait_until(&:present?).set(password)
-        browser_obj.button(text: 'Sign In').click!
+        browser.span(text: 'Sign in').wait_until(&:present?).click
+        browser.text_field(name: 'email').wait_until(&:present?).set(username)
+        browser.text_field(name: 'password').wait_until(&:present?).set(password)
+        browser.button(text: 'Sign In').click!
 
         if mfa
-          until browser_obj.url.include?('https://pro.coinbase.com')
-            browser_obj.text_field(name: 'token').wait_until(&:present?).set(PWN::Plugins::AuthenticationHelper.mfa(prompt: 'enter mfa token'))
-            browser_obj.button(text: 'Verify').click!
+          until browser.url.include?('https://pro.coinbase.com')
+            browser.text_field(name: 'token').wait_until(&:present?).set(PWN::Plugins::AuthenticationHelper.mfa(prompt: 'enter mfa token'))
+            browser.button(text: 'Verify').click!
             sleep 3
           end
           print "\n"
         end
 
@@ -68,12 +71,14 @@
       #   browser_obj: 'required - browser_obj returned from #open method'
       # )
 
       public_class_method def self.logout(opts = {})
         browser_obj = opts[:browser_obj]
-        browser_obj.goto('https://pro.coinbase.com/signout')
 
+        browser = browser_obj[:browser]
+        browser.goto('https://pro.coinbase.com/signout')
+
         browser_obj
       rescue StandardError => e
         raise e
       end
 
@@ -105,9 +110,10 @@
         puts "USAGE:
           browser_obj = #{self}.open(
             browser_type: 'optional - :firefox|:chrome|:ie|:headless (Defaults to :firefox)',
             proxy: 'optional - scheme://proxy_host:port || tor'
           )
+          browser = browser_obj[:browser]
           puts browser_obj.public_methods
 
           browser_obj = #{self}.login(
             browser_obj: 'required - browser_obj returned from #open method',
             username: 'required - username',