rudra 1.1.2 → 1.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f6485433a80329770ea6ce43647f1d10176c20fa1eefa825b51e222615e0b4fe
-  data.tar.gz: 980b66578597240fab1f2b0c68ee9fcc68be47b7466a8b5d99decf3dec6787c4
+  metadata.gz: d1094036482a36f5196cd490c2594b66aabf2eebb1aa6c5adfb2b49c426d03e3
+  data.tar.gz: 4fd39aae44a3cf44d263014147cba3eb50a3c0622445fb0452dfda3f438120a1
 SHA512:
-  metadata.gz: f12e6ea46bb610708c6fa9cf13cd897bbd2e19ea645d7f5e748b1daccac7efdf00470b8333f16907401c9e63968f975f2c526188513862197db8771669e7c785
-  data.tar.gz: 7584799d278830b5912e6eb57e6582c3b2b7bb93cafff6c2bbc331137b06181674f5152e3cf179696ff3d1c6b66d0a91546fbd5800e2329662e5bb95357a8f40
+  metadata.gz: 8c1fca4fe84b87a53c2681df286f0862fa88fec448af16a4a2752c34dca6c9c0e3686b4a4f4db05fd222d539a08586e6750a524801033d127fda974345f79025
+  data.tar.gz: 813860ba0fa303826bcc1ea916592b73b0b42130c2ecb92082562b576be71a26a445cc3128964cc6b2342d1837d37b0b3855867bf66259b5c772cbe8d1702099

data/lib/rudra.rb

@@ -1,5 +1,9 @@
 require 'selenium-webdriver'
 require 'webdrivers'
+require 'zip'
+require 'base64'
+require 'json'
+require 'stringio'
 
 # Selenium IDE-like WebDriver based upon Ruby binding
 # @author Aaron Chen
@@ -15,6 +19,8 @@ require 'webdrivers'
 # @attr_reader [Integer] timeout The driver timeout
 # @attr_reader [Boolean] verbose Turn on/off Verbose mode
 # @attr_reader [Boolean] silent Turn off Turn on/off descriptions
+# @attr_reader [String] chrome_auth_username Chrome Basic Auth Extension - username
+# @attr_reader [String] chrome_auth_password Chrome Basic Auth Extension - password
 class Rudra
   # Supported Browsers
   BROWSERS = %i[chrome firefox ie safari].freeze
@@ -30,11 +36,13 @@ class Rudra
     browser driver install_dir locale
     headless window_size screen_dir
     log_prefix timeout verbose silent
+    chrome_auth_username chrome_auth_password
   ].freeze
 
   attr_reader :browser, :driver, :install_dir, :locale,
               :headless, :window_size, :screen_dir,
-              :log_prefix, :timeout, :verbose, :silent
+              :log_prefix, :timeout, :verbose, :silent,
+              :chrome_auth_username, :chrome_auth_password
 
   # Initialize an instance of Rudra
   # @param [Hash] options the options to initialize Rudra
@@ -50,6 +58,8 @@ class Rudra
   # @option options [Integer] :timeout (30) implicit_wait timeout
   # @option options [Boolean] :verbose (false) Turn on/off verbose mode
   # @option options [Boolean] :silent (false) Turn on/off descriptions
+  # @option options [String] :chrome_auth_username ('') username for Chrome Basic Auth extension
+  # @option options [String] :chrome_auth_password ('') password for Chrome Basic Auth extension
   def initialize(options = {})
     self.browser = options.fetch(:browser, :chrome)
     self.install_dir = options.fetch(:install_dir, './webdrivers/')
@@ -60,6 +70,8 @@ class Rudra
     self.log_prefix = options.fetch(:log_prefix, ' - ')
     self.verbose = options.fetch(:verbose, false)
     self.silent = options.fetch(:silent, false)
+    self.chrome_auth_username = options.fetch(:chrome_auth_username, '')
+    self.chrome_auth_password = options.fetch(:chrome_auth_password, '')
     self.main_label = caller_locations(2, 1).first.label
 
     initialize_driver
@@ -149,6 +161,26 @@ class Rudra
     driver.manage.delete_cookie(name)
   end
 
+  # Check if an element is found
+  # @param [String, Selenium::WebDriver::Element] locator the locator to
+  #   identify the element or Selenium::WebDriver::Element
+  # @param [Integer] seconds seconds before timed out
+  def element_found?(locator, seconds = 1)
+    how, what = parse_locator(locator)
+
+    implicit_wait(seconds)
+
+    begin
+      wait_for(seconds) { driver.find_element(how, what).displayed? }
+    rescue Selenium::WebDriver::Error::TimeoutError
+      false
+    rescue Net::ReadTimeout
+      false
+    ensure
+      implicit_wait(timeout)
+    end
+  end
+
   # Execute the given JavaScript
   # @param [String] script JavaScript source to execute
   # @param [Selenium::WebDriver::Element, Integer, Float, Boolean, NilClass,
@@ -380,8 +412,12 @@ class Rudra
 
     begin
       wait_for(seconds) do
-        elements = driver.find_elements(how, what)
-        elements.empty? || elements.map(&:displayed?).none?
+        begin
+          elements = driver.find_elements(how, what)
+          elements.empty? || elements.map(&:displayed?).none?
+        rescue Selenium::WebDriver::Error::StaleElementReferenceError
+          false
+        end
       end
     rescue Selenium::WebDriver::Error::TimeoutError
       true
@@ -477,7 +513,8 @@ class Rudra
   def click(locator)
     wait_for do
       begin
-        find_element(locator).click.nil?
+        element = find_element(locator)
+        element.enabled? && element.click.nil?
       rescue Selenium::WebDriver::Error::ElementClickInterceptedError
         false
       end
@@ -1176,7 +1213,7 @@ class Rudra
   private
 
   attr_accessor :main_label
-  attr_writer :silent, :window_size
+  attr_writer :silent, :window_size, :chrome_auth_username, :chrome_auth_password
 
   def browser=(brw)
     unless BROWSERS.include?(brw)
@@ -1247,6 +1284,10 @@ class Rudra
       options.add_argument('--headless')
       options.add_argument("--window-size=#{window_size}")
     end
+    if chrome_auth_username && chrome_auth_password
+      encoded = chrome_basic_auth_extension(chrome_auth_username, chrome_auth_password)
+      options.add_encoded_extension(encoded)
+    end
     options.add_option(
       'excludeSwitches',
       %w[enable-automation enable-logging]
@@ -1319,4 +1360,48 @@ class Rudra
     id = Array.new(length) { charset.sample }.join
     "rudra_#{id}"
   end
+
+  def chrome_basic_auth_extension(username, password)
+    manifest = {
+      "manifest_version": 2,
+      "name": 'Rudra Basic Auth Extension',
+      "version": '1.0.0',
+      "permissions": ['*://*/*', 'webRequest', 'webRequestBlocking'],
+      "background": {
+        "scripts": ['background.js']
+      }
+    }
+
+    background = <<~JSCRIPT
+      var username = '#{username}';
+      var password = '#{password}';
+
+      chrome.webRequest.onAuthRequired.addListener(
+        function handler(details) {
+          if (username == null) {
+            return { cancel: true };
+          }
+
+          var authCredentials = { username: username, password: username };
+          username = password = null;
+
+          return { authCredentials: authCredentials };
+        },
+        { urls: ['<all_urls>'] },
+        ['blocking']
+      );
+    JSCRIPT
+
+    stringio = Zip::OutputStream.write_buffer do |zos|
+      zos.put_next_entry('manifest.json')
+      zos.write manifest.to_json
+      zos.put_next_entry('background.js')
+      zos.write background
+    end
+    # File.open('basic_auth.crx', 'wb') do |f|
+    #   f << stringio.string
+    # end
+
+    Base64.strict_encode64(stringio.string)
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rudra
 version: !ruby/object:Gem::Version
-  version: 1.1.2
+  version: 1.1.3
 platform: ruby
 authors:
 - Aaron Chen
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-06-12 00:00:00.000000000 Z
+date: 2020-06-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yard
@@ -44,6 +44,26 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.9'
+- !ruby/object:Gem::Dependency
+  name: rubyzip
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.3.0
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.3.0
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.3'
 - !ruby/object:Gem::Dependency
   name: selenium-webdriver
   requirement: !ruby/object:Gem::Requirement