Class: VagrantPlugins::VagrantNotifyForwarder::Utils

Inherits:
Object
  • Object
show all
Defined in:
lib/vagrant-notify-forwarder/utils.rb

Constant Summary collapse

@@OS_NAMES = 
{
    "Linux" => :linux,
    "Darwin" => :darwin,
    "FreeBSD" => :freebsd,
}
@@HARDWARE_NAMES = 
{
    "x86_64" => :x86_64,
    "amd64" => :x86_64,
    "arm64" => :arm64,
    "aarch64" => :arm64,
    "armv7l" => :armv7l,
}

Class Method Summary collapse

Class Method Details

.ensure_binary_downloaded(env, os, hardware) ⇒ Object 



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
 
# File 'lib/vagrant-notify-forwarder/utils.rb', line 29

def self.ensure_binary_downloaded(env, os, hardware)
  config = env.fetch(:machine).config.notify_forwarder
  download_map = config.binaries
  url, sha256sum = download_map[[os, hardware]]

  unless url && sha256sum
    env[:ui].error "Notify-forwarder: No binary configured for host '#{os}' '#{hardware}'"
    return
  end

  expanded_url = expand_binary_url(url)
  env[:ui].info "Notify-forwarder: Using binary '#{expanded_url}' (sha256 #{sha256sum[0,8]}...) for '#{os}' '#{hardware}'"

  uri = safe_parse_uri(expanded_url)
  basename = File.basename(uri ? uri.path : expanded_url)
  path = env[:tmp_path].join basename
  should_download = true

  if File.exist? path
    digest = Digest::SHA256.file(path).hexdigest

    if digest == sha256sum
      should_download = false
    end
  end

  if should_download
    env[:ui].detail 'Notify-forwarder: Downloading client'
    downloader = Vagrant::Util::Downloader.new expanded_url, path
    downloader.download!
  end

  File.chmod(0755, path)

  path
end

.expand_binary_url(url) ⇒ Object 



66
67
68
69
70
71
72
73
74
75
76
77
78
 
# File 'lib/vagrant-notify-forwarder/utils.rb', line 66

def self.expand_binary_url(url)
  expanded = expand_environment(url.to_s)

  if expanded.start_with?('file://')
    uri = URI.parse(expanded)
    uri.path = File.expand_path(uri.path)
    return uri.to_s
  end

  return expanded if expanded =~ %r{^[a-zA-Z][a-zA-Z0-9+.-]*://}

  File.expand_path(expanded)
end

.expand_environment(value) ⇒ Object 



80
81
82
83
84
85
 
# File 'lib/vagrant-notify-forwarder/utils.rb', line 80

def self.expand_environment(value)
  value.gsub(/\$(\{)?([A-Za-z0-9_]+)\}?/) do
    key = Regexp.last_match(2)
    ENV.fetch(key, '')
  end
end

.host_pidfile(env) ⇒ Object 



93
94
95
 
# File 'lib/vagrant-notify-forwarder/utils.rb', line 93

def self.host_pidfile(env)
  env[:machine].data_dir.join('notify_watcher_host_pid')
end

.parse_hardware_name(data) ⇒ Object 



25
26
27
 
# File 'lib/vagrant-notify-forwarder/utils.rb', line 25

def self.parse_hardware_name(data)
  @@HARDWARE_NAMES[data.strip] or :unsupported
end

.parse_os_name(data) ⇒ Object 



21
22
23
 
# File 'lib/vagrant-notify-forwarder/utils.rb', line 21

def self.parse_os_name(data)
  @@OS_NAMES[data.strip] or :unsupported
end

.safe_parse_uri(url) ⇒ Object 



87
88
89
90
91
 
# File 'lib/vagrant-notify-forwarder/utils.rb', line 87

def self.safe_parse_uri(url)
  URI.parse(url)
rescue URI::InvalidURIError
  nil
end