1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
require "chromedriver/helper/version"
require "chromedriver/helper/google_code_parser"
require 'fileutils'
require 'rbconfig'
require 'open-uri'
require 'archive/zip'
module Chromedriver
def self.set_version(version)
Chromedriver::Helper.new.update(version)
end
class Helper
def run(*args)
download
exec binary_path, *args
end
def download(hit_network = false)
return if File.exist?(binary_path) && !hit_network
raise "Version not found for #{download_version}" unless download_url
filename = File.basename download_url
Dir.chdir platform_install_dir do
FileUtils.rm_f filename
File.open(filename, "wb") do |saved_file|
URI.parse(download_url).open("rb") do |read_file|
saved_file.write(read_file.read)
end
end
raise "Could not download #{download_url}" unless File.exists? filename
Archive::Zip.extract(filename, '.', :overwrite => :all)
end
raise "Could not unzip #{filename} to get #{binary_path}" unless File.exists? binary_path
FileUtils.chmod "ugo+rx", binary_path
File.open(version_path, 'w') { |file| file.write(download_version) }
end
def update(version = nil)
@download_version = version || google_code_parser.newest_download_version.to_s
hit_network = (current_version != download_version) ? true : false
download(hit_network)
end
def current_version
@current_version ||= if File.exist?(version_path)
File.read(version_path).strip
end
end
def download_version
@download_version ||= current_version || google_code_parser.newest_download_version.to_s
end
def download_url
@download_url ||= google_code_parser.version_download_url(download_version)
end
def google_code_parser
@google_code_parser ||= GoogleCodeParser.new(platform)
end
def version_path
@version_path ||= File.join(base_install_dir, ".chromedriver-version")
end
def binary_path
if platform == "win"
File.join platform_install_dir, "chromedriver.exe"
else
File.join platform_install_dir, "chromedriver"
end
end
def platform_install_dir
dir = File.join version_install_dir, platform
FileUtils.mkdir_p dir
dir
end
def version_install_dir
dir = File.expand_path File.join(base_install_dir, download_version)
FileUtils.mkdir_p dir
dir
end
def base_install_dir
@base_install_dir ||= begin
dir = File.expand_path File.join(ENV['HOME'], ".chromedriver-helper")
FileUtils.mkdir_p dir
dir
end
end
def platform
cfg = RbConfig::CONFIG
host_cpu = cfg["host_cpu"]
host_os = cfg["host_os"]
case host_os
when /linux/ then
host_cpu =~ /x86_64|amd64/ ? "linux64" : "linux32"
when /darwin/ then "mac"
when /mswin/ then "win"
when /mingw/ then "win"
else
raise("Unsupported host OS '#{host_os}'")
end
end
end
end
|
