itamae-plugin-recipe-selenium 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA256:
-  metadata.gz: '091154a228fefdaf126d978d1a3d1af7f85c7a4e1ee6c9f9127320baa35e878a'
-  data.tar.gz: e64ca4cbe95a6416aa6e703b50d965dbb45fe11913d2a9bceb9a61cdacb97f8d
+SHA1:
+  metadata.gz: 66d4ba73c62582d5e221a6478f32359a8b31a8ef
+  data.tar.gz: c5b22c860d68d0414c425a2d2fcb678f40cfe07c
 SHA512:
-  metadata.gz: eabad90af13e794a0adfeaa2c450f591d9c1d934c6b7d08fb44950d8942057e713c4b9ea60fa6816a766b01036548693a5ce29bc01b3dc54e8843fa3a894fdf7
-  data.tar.gz: 80df270031008bbf159c71a7c693580a8aaab4147318deff24c695795f4db6732ba43218778899d6089b13a353171615a605c56e74ae1d9d39da32436c5a8af5
+  metadata.gz: 84fcdca7e3233e384672e363e71826c063fe51fa43debb79a83433f12765b39c54afa6d657da0fbd53400b4001c40abb62bc67f100cd2dd9046f4b6c5564129c
+  data.tar.gz: 65ebe9d587fa439db64ef22fb4e92fb2052cc647d05fc6874eb0dbaf02c40af4a5b574e694e6afaa3b0b24364a9b408181ef1527a850d093566fce151e444295

data/lib/itamae/plugin/recipe/selenium.rb

@@ -1,6 +1,5 @@
 require "itamae/plugin/recipe/selenium/version"
 
-include_recipe './selenium/tmp_directory'
 include_recipe './selenium/chrome'
 include_recipe './selenium/chromedriver'
 include_recipe './selenium/geckodriver'

data/lib/itamae/plugin/recipe/selenium/chromedriver.rb

@@ -1,29 +1,7 @@
-require 'net/http'
+include_recipe './tmp_directory'
 
-host = 'chromedriver.storage.googleapis.com'
-base_url = 'https://' + host
-browser_version = run_command('sudo yum list | grep google-chrome-stable').stdout.split[1]
-browser_version = Gem::Version.new(browser_version.to_s).segments[0..2].join('.')
-driver_version_url = base_url + "/LATEST_RELEASE_#{browser_version}"
-version = Net::HTTP.get_response(URI(driver_version_url)).body
-download_url = base_url + "/#{version}/chromedriver_linux64.zip"
-header = Net::HTTP.start(host) { |http| http.head("/#{version}/chromedriver_linux64.zip") }
-etag = header['etag'][1...-1]
-
-execute "download chromedriver-#{version}" do
-  cwd '/tmp/itamae-plugin-recipe-selenium'
-  command <<-EOF
-    rm -Rf chromedriver_linux64-#{version}*
-    curl -o chromedriver_linux64-#{version}.zip #{download_url}
-  EOF
-  not_if "echo '#{etag} chromedriver_linux64-#{version}.zip' | md5sum -c -"
-end
-
-execute "install chromedriver-#{version}" do
-  cwd '/tmp/itamae-plugin-recipe-selenium'
-  command <<-EOF
-    unzip chromedriver_linux64-#{version}.zip
-    sudo mv -f chromedriver /usr/local/bin/
-  EOF
-  not_if "/usr/local/bin/chromedriver -v | grep 'ChromeDriver #{version}'"
+setup_selenium_driver "setup chromedriver" do
+  type :chrome
+  tmp_dir '/tmp/itamae-plugin-recipe-selenium'
+  install_dir '/usr/local/bin'
 end

data/lib/itamae/plugin/recipe/selenium/geckodriver.rb

@@ -1,31 +1,7 @@
-require 'net/http'
+include_recipe './tmp_directory'
 
-base_url = "https://github.com/mozilla/geckodriver/releases"
-url = base_url + '/latest'
-response = nil
-5.times do
-  response = Net::HTTP.get_response(URI(url))
-
-  break unless response.is_a?(Net::HTTPRedirection) 
-  url = response['location']
-end
-version = response.uri.to_s[/[^v]*$/]
-download_url = base_url + "/download/v#{version}/geckodriver-v#{version}-linux64.tar.gz"
-
-execute "download geckodriver-#{version}" do
-  cwd '/tmp/itamae-plugin-recipe-selenium'
-  command <<-EOF
-    rm -Rf geckodriver-v#{version}-linux64*
-    curl -L -o geckodriver-v#{version}-linux64.tar.gz #{download_url}
-  EOF
-  not_if "test -e geckodriver-v#{version}-linux64.tar.gz"
-end
-
-execute "install geckodriver-#{version}" do
-  cwd '/tmp/itamae-plugin-recipe-selenium'
-  command <<-EOF
-    tar zxf geckodriver-v#{version}-linux64.tar.gz
-    sudo mv -f geckodriver /usr/local/bin/
-  EOF
-  not_if "/usr/local/bin/geckodriver -V | grep 'geckodriver #{version}'"
+setup_selenium_driver "setup geckodriver" do
+  type :gecko
+  tmp_dir '/tmp/itamae-plugin-recipe-selenium'
+  install_dir '/usr/local/bin'
 end

data/lib/itamae/plugin/recipe/selenium/version.rb

@@ -2,7 +2,7 @@ module Itamae
   module Plugin
     module Recipe
       module Selenium
-        VERSION = "0.1.1"
+        VERSION = "0.1.2"
       end
     end
   end

data/lib/itamae/plugin/resource/setup_selenium_driver.rb

@@ -0,0 +1,125 @@
+require 'net/http'
+
+module Itamae
+  module Plugin
+    module Resource
+      class SetupSeleniumDriver < Itamae::Resource::Base
+        define_attribute :action, default: :run
+        define_attribute :type, type: [String, Symbol], default: :chrome
+        define_attribute :tmp_dir, type: String, default: '/tmp'
+        define_attribute :install_dir, type: String, default: '/usr/local/bin/'
+
+        def pre_action
+          attributes.cwd = attributes.tmp_dir
+          attributes.install_dir = File.join(attributes.install_dir, '/')
+        end
+
+        def set_current_attributes
+          current.executed = false
+        end
+
+        def action_run(options)
+          case attributes.type.to_sym
+          when :chrome
+            run_setup_chromedriver
+          when :gecko
+            run_setup_geckodriver
+          end
+
+          updated!
+        end
+
+        private
+
+        def run_setup_chromedriver
+          host = 'chromedriver.storage.googleapis.com'
+          base_url = 'https://' + host
+          browser_version = run_command('sudo yum list | grep google-chrome-stable').stdout.split[1]
+          browser_version = Gem::Version.new(browser_version.to_s).segments[0..2].join('.')
+          driver_version_url = base_url + "/LATEST_RELEASE_#{browser_version}"
+          driver_version = Net::HTTP.get_response(URI(driver_version_url)).body
+          download_url = base_url + "/#{driver_version}/chromedriver_linux64.zip"
+          header = Net::HTTP.start(host) { |http| http.head("/#{driver_version}/chromedriver_linux64.zip") }
+          etag = header['etag'][1...-1]
+
+          Itamae.logger.debug "browser version: #{browser_version}"
+          Itamae.logger.debug "driver version: #{driver_version}"
+          Itamae.logger.debug "download url: #{download_url}"
+          Itamae.logger.debug "etag: #{etag}"
+
+          run_command_if_not(
+            'download chromedriver',
+            "echo '#{etag} chromedriver_linux64-#{driver_version}.zip' | md5sum -c -",
+            <<-COMMANDS
+              rm -Rf chromedriver_linux64-#{driver_version}*
+              curl -o chromedriver_linux64-#{driver_version}.zip #{download_url}
+            COMMANDS
+          )
+
+          run_command_if_not(
+            'install chromedriver',
+            "#{attributes.install_dir}chromedriver -v | grep 'ChromeDriver #{driver_version}'",
+            <<-COMMANDS
+              unzip chromedriver_linux64-#{driver_version}.zip
+              sudo mv -f chromedriver #{attributes.install_dir}
+            COMMANDS
+          )
+        end
+
+        def run_setup_geckodriver
+          base_url = "https://github.com/mozilla/geckodriver/releases"
+          url = base_url + '/latest'
+          response = nil
+          5.times do
+            response = Net::HTTP.get_response(URI(url))
+
+            break unless response.is_a?(Net::HTTPRedirection)
+            url = response['location']
+          end
+          driver_version = response.uri.to_s[/[^v]*$/]
+          download_url = base_url + "/download/v#{driver_version}/geckodriver-v#{driver_version}-linux64.tar.gz"
+
+          Itamae.logger.debug "driver version: #{driver_version}"
+          Itamae.logger.debug "download url: #{download_url}"
+
+          run_command_if_not(
+            'download geckodriver',
+            "test -e geckodriver-v#{driver_version}-linux64.tar.gz",
+            <<-COMMANDS
+              rm -Rf geckodriver-v#{driver_version}-linux64*
+              curl -L -o geckodriver-v#{driver_version}-linux64.tar.gz #{download_url}
+            COMMANDS
+          )
+
+          run_command_if_not(
+            'install geckodriver',
+            "#{attributes.install_dir}geckodriver -V | grep 'geckodriver #{driver_version}'",
+            <<-COMMANDS
+              tar zxf geckodriver-v#{driver_version}-linux64.tar.gz
+              sudo mv -f geckodriver #{attributes.install_dir}
+            COMMANDS
+          )
+        end
+
+        def run_command_if_not(name, if_not_command, command)
+          if do_not_run_command_because_of_not_if?(if_not_command)
+            Itamae.logger.debug  "#{resource_type}[#{name}] Execution skipped because already done"
+          else
+            run_command command
+            show_differences_message(name, 'executed', false, true)
+          end
+        end
+
+        def do_not_run_command_because_of_not_if?(command)
+          run_command(command, error: false).exit_status == 0
+        end
+
+        def show_differences_message(name, key, current_value, value)
+          Itamae.logger.color :green do
+            Itamae.logger.info "#{resource_type}[#{name}] #{key} will change from '#{current_value}' to '#{value}'"
+          end
+        end
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: itamae-plugin-recipe-selenium
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - y-matsuda
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-10-30 00:00:00.000000000 Z
+date: 2019-11-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: itamae
@@ -95,6 +95,7 @@ files:
 - lib/itamae/plugin/recipe/selenium/templates/etc/yum.repos.d/google-chrome.repo.erb
 - lib/itamae/plugin/recipe/selenium/tmp_directory.rb
 - lib/itamae/plugin/recipe/selenium/version.rb
+- lib/itamae/plugin/resource/setup_selenium_driver.rb
 homepage: https://github.com/maedadev/itamae-plugin-recipe-selenium
 licenses:
 - MIT
@@ -114,7 +115,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.6
+rubyforge_project: 
+rubygems_version: 2.5.2.3
 signing_key: 
 specification_version: 4
 summary: itamae recipe for god process monitoring framework