Class: VagrantPlugins::ScriptRock::Provisioner

Inherits:
Object
  • Object
show all
Defined in:
lib/vagrant-scriptrock/provisioner.rb

Defined Under Namespace

Classes: GuardrailPrivateKeyInstallError

Instance Method Summary collapse

Constructor Details

#initialize(machine, config) ⇒ Provisioner

Returns a new instance of Provisioner.



11
12
13
14
15
16
 
# File 'lib/vagrant-scriptrock/provisioner.rb', line 11

def initialize(machine, config)
  @debug = false
  @machine = machine
  @root_config = machine.config
  puts "provision initialize config" if @debug
end

Instance Method Details

#cleanup ⇒ Object 



129
130
131
132
 
# File 'lib/vagrant-scriptrock/provisioner.rb', line 129

def cleanup
  puts "provision cleanup" if @debug
  guardrail_delete
end

#configure(root_config) ⇒ Object 



18
19
20
21
 
# File 'lib/vagrant-scriptrock/provisioner.rb', line 18

def configure(root_config)
  puts "provision configure" if @debug
  @root_config = root_config
end

#guardrail_auth_headers ⇒ Object 



27
28
29
 
# File 'lib/vagrant-scriptrock/provisioner.rb', line 27

def guardrail_auth_headers
  return { "Authorization" => "Token token=\"#{@root_config.scriptrock.api_key}#{@root_config.scriptrock.secret_key}\"" }
end

#guardrail_create ⇒ Object 



47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/vagrant-scriptrock/provisioner.rb', line 47

def guardrail_create
  url = "#{@root_config.scriptrock.connect_url}/api/v1/nodes.json"
  response = HTTParty.post(url, :headers => guardrail_auth_headers, :body => {
      :node => {
        "name" => guardrail_name,
        "node_type" => "SV",
      },
    })
  responseJson = JSON.parse(response.body)
  if response.code == 201
    puts "ScriptRock: created new node, id #{responseJson["id"]} name '#{guardrail_name}'"
    return responseJson
  else
    throw "ScriptRock Guardrail create node error code #{response.code} body: #{response.body}"
  end
end

#guardrail_create_update ⇒ Object 



100
101
102
103
104
105
106
107
108
109
110
 
# File 'lib/vagrant-scriptrock/provisioner.rb', line 100

def guardrail_create_update
  begin
    node = guardrail_lookup_and_show
    if node == nil
      node = guardrail_create
    end
    guardrail_update(node)
  rescue => e
    puts "Error contacting guardrail api: #{e.class}: #{e.message}"
  end
end

#guardrail_delete ⇒ Object 



80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
 
# File 'lib/vagrant-scriptrock/provisioner.rb', line 80

def guardrail_delete
  begin
    node = guardrail_lookup_and_show
    if node == nil
      puts "ScriptRock: node with name '#{guardrail_name}' not found"
    else
      url = "#{@root_config.scriptrock.connect_url}/api/v1/nodes/#{node["id"]}.json"
      response = HTTParty.delete(url, :headers => guardrail_auth_headers)
      if response.code == 204
        puts "ScriptRock: deleted node, id #{node["id"]} name #{guardrail_name}"
        return true
      else
        throw "ScriptRock Guardrail delete node error code #{response.code} body: #{response.body}"
      end      
    end
  rescue => e
    puts "Error contacting guardrail api: #{e.class}: #{e.message}"
  end
end

#guardrail_lookup_and_show ⇒ Object 



31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
 
# File 'lib/vagrant-scriptrock/provisioner.rb', line 31

def guardrail_lookup_and_show
  url = "#{@root_config.scriptrock.connect_url}/api/v1/nodes/lookup.json?name=#{CGI.escape(guardrail_name)}"
  response = HTTParty.get(url, :headers => guardrail_auth_headers)
  if response.code == 200
    responseJson = JSON.parse(response.body)
    url = "#{@root_config.scriptrock.connect_url}/api/v1/nodes/#{responseJson["node_id"]}.json"
    response = HTTParty.get(url, :headers => guardrail_auth_headers)
    responseJson = JSON.parse(response.body)
    if response.code == 200
      puts "ScriptRock: node already exists, id #{responseJson["id"]} name '#{guardrail_name}'"
      return responseJson
    end
  end
  return nil
end

#guardrail_name ⇒ Object 



23
24
25
 
# File 'lib/vagrant-scriptrock/provisioner.rb', line 23

def guardrail_name
  return "#{@root_config.scriptrock.name_prefix} #{@machine.name}"
end

#guardrail_update(node) ⇒ Object 



64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
 
# File 'lib/vagrant-scriptrock/provisioner.rb', line 64

def guardrail_update(node)
  url = "#{@root_config.scriptrock.connect_url}/api/v1/nodes/#{node["id"]}.json"
  ssh_info = @machine.ssh_info
  node = {
    :medium_type => 3,
    :description => "#{@machine.name} (vagrant)",
    :medium_hostname => "#{@root_config.scriptrock.first_hop} ssh://#{ssh_info[:username]}@#{ssh_info[:host]}:#{ssh_info[:port]}".strip,
  }
  response = HTTParty.put(url, :headers => guardrail_auth_headers, :body => { :node => node })
  if response.code == 204
    return true
  else
    throw "ScriptRock Guardrail update node error code #{response.code} body: #{response.body}"
  end        
end

#provision ⇒ Object 



112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
 
# File 'lib/vagrant-scriptrock/provisioner.rb', line 112

def provision
  puts "provision provision" if @debug

  # insert the guardrail public key onto the target node if it is not already present
  puts "Checking for and possibly installing Guardrail public key in ~/.ssh/authorized_keys..."
  pk = @root_config.scriptrock.ssh_pubkey
  key_install_script = "mkdir -p ~/.ssh && " +
    "(grep -q -s '#{pk}' ~/.ssh/authorized_keys || echo '#{pk}' >> ~/.ssh/authorized_keys) && "+
    "grep -q -s '#{pk}' ~/.ssh/authorized_keys"
  @machine.communicate.tap do |comm|
    comm.execute(key_install_script, error_class: GuardrailPrivateKeyInstallError)
  end

  # add this node to guardrail if not already present, then update to use the current credentials + forwarded port
  guardrail_create_update
end