Add rake tasks for update, remove, and version.

.rubocop.yml

@@ -22,6 +22,7 @@ Metrics/MethodLength:
 Metrics/BlockLength:
   Exclude:
     - 'spec/**/*'
+    - 'lib/webdrivers/tasks/*.rake'
 
 Metrics/ClassLength:
   Max: 116

lib/webdrivers.rb

@@ -4,3 +4,4 @@
 require 'webdrivers/geckodriver'
 require 'webdrivers/iedriver'
 require 'webdrivers/mswebdriver'
+require 'webdrivers/railtie' if defined?(Rails)

lib/webdrivers/Rakefile

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+require 'webdrivers'
+
+path = File.expand_path(__dir__)
+Dir.glob("#{path}/tasks/*.rake").each { |f| import f }

lib/webdrivers/common.rb

@@ -18,13 +18,14 @@ class NetworkError < StandardError
   end
 
   class << self
-    attr_accessor :proxy_addr, :proxy_port, :proxy_user, :proxy_pass, :install_dir
+    attr_accessor :proxy_addr, :proxy_port, :proxy_user, :proxy_pass
+    attr_writer :install_dir
 
     #
     # Returns the amount of time (Seconds) the gem waits between two update checks.
     #
     def cache_time
-      @cache_time || 0
+      @cache_time || ENV['WD_CACHE_TIME'].to_i || 0
     end
 
     #
@@ -36,6 +37,13 @@ def cache_time=(value)
       @cache_time = value
     end
 
+    #
+    # Returns the install (download) directory path for the drivers.
+    #
+    def install_dir
+      @install_dir || ENV['WD_INSTALL_DIR'] || File.expand_path(File.join(ENV['HOME'], '.webdrivers'))
+    end
+
     def logger
       @logger ||= Webdrivers::Logger.new
     end

lib/webdrivers/railtie.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require 'webdrivers'
+require 'rails'
+
+module Webdrivers
+  class Railtie < Rails::Railtie
+    railtie_name :webdrivers
+
+    rake_tasks do
+      path = File.expand_path(__dir__)
+      Dir.glob("#{path}/tasks/*.rake").each { |f| load f }
+    end
+  end
+end

lib/webdrivers/system.rb

@@ -26,7 +26,7 @@ def delete(file)
       end
 
       def install_dir
-        Webdrivers.install_dir || File.expand_path(File.join(ENV['HOME'], '.webdrivers'))
+        Webdrivers.install_dir
       end
 
       def cache_version(file_name, version)

lib/webdrivers/tasks/chromedriver.rake

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+namespace :webdrivers do
+  require 'webdrivers/chromedriver'
+
+  namespace :chromedriver do
+    Webdrivers.logger.level = :info
+
+    desc 'Print current chromedriver version'
+    task :version do
+      gem_ver = Webdrivers::Chromedriver.current_version
+      if gem_ver
+        Webdrivers.logger.info 'chromedriver ' + gem_ver.version
+      else
+        Webdrivers.logger.warn 'No existing chromedriver found.'
+      end
+    end
+
+    desc 'Remove and download updated chromedriver if necessary'
+    task :update, [:version] do |_, args|
+      args.with_defaults(version: 0)
+      Webdrivers.cache_time                     = ENV.fetch('WD_CACHE_TIME', 86_400)
+      Webdrivers.install_dir                    = ENV.fetch('WD_INSTALL_DIR', nil)
+      Webdrivers::Chromedriver.required_version = args.version
+      Webdrivers::Chromedriver.update
+      Webdrivers.logger.info "Updated to chromedriver #{Webdrivers::Chromedriver.current_version}"
+    end
+
+    desc 'Force remove chromedriver'
+    task :remove do
+      unless File.exist? Webdrivers::Chromedriver.driver_path
+        Webdrivers.logger.info 'No existing chromedriver to remove.'
+        next # Return early
+      end
+
+      cur_version = Webdrivers::Chromedriver.current_version
+      Webdrivers::Chromedriver.remove
+
+      if File.exist? Webdrivers::Chromedriver.driver_path # Failed for some reason
+        Webdrivers.logger.error 'Failed to remove chromedriver. Please try removing manually.'
+      else
+        Webdrivers.logger.info "Removed chromedriver #{cur_version}."
+      end
+    end
+  end
+end

lib/webdrivers/tasks/geckodriver.rake

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+namespace :webdrivers do
+  require 'webdrivers/geckodriver'
+
+  namespace :geckodriver do
+    Webdrivers.logger.level = :info
+
+    desc 'Print current geckodriver version'
+    task :version do
+      gem_ver = Webdrivers::Geckodriver.current_version
+      if gem_ver
+        Webdrivers.logger.info 'geckodriver ' + gem_ver.version
+      else
+        Webdrivers.logger.warn 'No existing geckodriver found.'
+      end
+    end
+
+    desc 'Remove and download updated geckodriver if necessary'
+    task :update, [:version] do |_, args|
+      args.with_defaults(version: 0)
+      Webdrivers.cache_time                    = ENV.fetch('WD_CACHE_TIME', 86_400)
+      Webdrivers.install_dir                   = ENV.fetch('WD_INSTALL_DIR', nil)
+      Webdrivers::Geckodriver.required_version = args.version
+      Webdrivers::Geckodriver.update
+      Webdrivers.logger.info "Updated to geckodriver #{Webdrivers::Geckodriver.current_version}"
+    end
+
+    desc 'Force remove geckodriver'
+    task :remove do
+      unless File.exist? Webdrivers::Geckodriver.driver_path
+        Webdrivers.logger.info 'No existing geckodriver to remove.'
+        next # Return early
+      end
+
+      cur_version = Webdrivers::Geckodriver.current_version
+      Webdrivers::Geckodriver.remove
+
+      if File.exist? Webdrivers::Geckodriver.driver_path # Failed for some reason
+        Webdrivers.logger.error 'Failed to remove geckodriver. Please try removing manually.'
+      else
+        Webdrivers.logger.info "Removed geckodriver #{cur_version}."
+      end
+    end
+  end
+end

lib/webdrivers/tasks/iedriver.rake

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+namespace :webdrivers do
+  require 'webdrivers/iedriver'
+
+  namespace :iedriver do
+    Webdrivers.logger.level = :info
+
+    desc 'Print current IEDriverServer version'
+    task :version do
+      gem_ver = Webdrivers::IEdriver.current_version
+      if gem_ver
+        Webdrivers.logger.info 'IEDriverServer ' + gem_ver.version
+      else
+        Webdrivers.logger.warn 'No existing IEDriverServer found.'
+      end
+    end
+
+    desc 'Remove and download updated IEDriverServer if necessary'
+    task :update, [:version] do |_, args|
+      args.with_defaults(version: 0)
+      Webdrivers.cache_time                 = ENV.fetch('WD_CACHE_TIME', 86_400)
+      Webdrivers.install_dir                = ENV.fetch('WD_INSTALL_DIR', nil)
+      Webdrivers::IEdriver.required_version = args.version
+      Webdrivers::IEdriver.update
+      Webdrivers.logger.info "Updated to IEDriverServer #{Webdrivers::IEdriver.current_version}"
+    end
+
+    desc 'Force remove IEDriverServer'
+    task :remove do
+      unless File.exist? Webdrivers::IEdriver.driver_path
+        Webdrivers.logger.info 'No existing IEDriverServer to remove.'
+        next # Return early
+      end
+
+      cur_version = Webdrivers::IEdriver.current_version
+      Webdrivers::IEdriver.remove
+
+      if File.exist? Webdrivers::IEdriver.driver_path # Failed for some reason
+        Webdrivers.logger.error 'Failed to remove IEDriverServer. Please try removing manually.'
+      else
+        Webdrivers.logger.info "Removed IEDriverServer #{cur_version}."
+      end
+    end
+  end
+end

spec/webdrivers/chromedriver_spec.rb

@@ -116,6 +116,31 @@
 
       expect(Webdrivers::Network).to have_received(:get).twice
     end
+
+    context 'when required version is 0' do
+      it 'downloads the latest version' do
+        allow(chromedriver).to receive(:latest_version).and_return(Gem::Version.new('72.0.3626.7'))
+        chromedriver.required_version = 0
+        chromedriver.update
+        expect(chromedriver.current_version.version).to eq('72.0.3626.7')
+      end
+    end
+
+    context 'when required version is nil' do
+      it 'downloads the latest version' do
+        allow(chromedriver).to receive(:latest_version).and_return(Gem::Version.new('72.0.3626.7'))
+        chromedriver.required_version = nil
+        chromedriver.update
+        expect(chromedriver.current_version.version).to eq('72.0.3626.7')
+      end
+    end
+
+    context 'when ENV variable WD_CACHE_TIME is set' do
+      it 'uses cache time value from ENV variable' do
+        allow(ENV).to receive(:[]).with('WD_CACHE_TIME').and_return('999')
+        expect(Webdrivers.cache_time).to be(999)
+      end
+    end
   end
 
   describe '#current_version' do
@@ -234,19 +259,26 @@
 
   describe '#install_dir' do
     it 'uses ~/.webdrivers as default value' do
-      expect(Webdrivers::System.install_dir).to include('.webdriver')
+      expect(Webdrivers.install_dir).to include('.webdriver')
     end
 
     it 'uses provided value' do
       begin
-        install_dir = File.expand_path(File.join(ENV['HOME'], '.webdrivers2'))
+        install_dir            = File.expand_path(File.join(ENV['HOME'], '.webdrivers2'))
         Webdrivers.install_dir = install_dir
 
-        expect(Webdrivers::System.install_dir).to eq install_dir
+        expect(Webdrivers.install_dir).to eq install_dir
       ensure
         Webdrivers.install_dir = nil
       end
     end
+
+    context 'when ENV variable WD_INSTALL_DIR is set' do
+      it 'uses path from the ENV variable' do
+        allow(ENV).to receive(:[]).with('WD_INSTALL_DIR').and_return('custom_dir')
+        expect(Webdrivers.install_dir).to be('custom_dir')
+      end
+    end
   end
 
   describe '#driver_path' do