vagrant-scriptrock 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 6426244896b029fec557201523eef0dd47259345
+  data.tar.gz: 7ee812b8f4a6b06bc12c082a655ac0dc75dcf3d6
+SHA512:
+  metadata.gz: 19f803cd69f5fe11c5c73d2802fdff3ac535b06b84f1793c00ec006f6db0b12b2842a401dbb1584fd9756de3f11015d9881bd8191840688db6d5986eff0cdde2
+  data.tar.gz: 8fd8e862f6f28184eb7f58277c6c2e81ec906665e7e28766a42b644d6cf53b4967bc154e61851dbeaea06d8e488b6f66848b28995c1b911dfec31cf2e99d381c

data/.gitignore

@@ -0,0 +1,21 @@
+.DS_Store
+
+.bundle
+pkg/*
+Gemfile.lock
+Gemfile.dev
+Gemfile.dev.lock
+
+coverage
+
+.rspec
+
+.vagrant
+.vagrant.d/*
+Vagrantfile
+
+*~
+.*~
+.*.swo
+.*.swp
+

data/CHANGELOG.md

@@ -0,0 +1 @@
+

data/Gemfile

@@ -0,0 +1,10 @@
+source "https://rubygems.org"
+
+group :development do
+	gem "vagrant", git: "https://github.com/mitchellh/vagrant.git"
+end
+
+group :plugins do
+	gem "vagrant-scriptrock", path: "."
+end
+

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2015 Mark Sheahan
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,29 @@
+# Vagrant::Scriptrock
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'vagrant-scriptrock'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install vagrant-scriptrock
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it ( https://github.com/[my-github-username]/vagrant-scriptrock/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create a new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+

data/lib/vagrant-scriptrock/config.rb

@@ -0,0 +1,116 @@
+require 'httparty'
+require 'json'
+require 'yaml'
+
+module VagrantPlugins
+	module ScriptRock
+		class Config < Vagrant.plugin(2, :config)
+			attr_accessor :scriptrock_yml_path
+			attr_accessor :first_hop
+			attr_accessor :api_key
+			attr_accessor :secret_key
+			attr_accessor :connect_url
+			attr_accessor :ssh_pubkey
+
+			def initialize
+				@debug = true#false
+				puts "config initialize" if @debug
+				@scriptrock_yml_path = "~/.scriptrock/scriptrock.yml"
+				@first_hop = UNSET_VALUE
+				@api_key = UNSET_VALUE
+				@secret_key = UNSET_VALUE
+				@connect_url = UNSET_VALUE
+				@ssh_pubkey = UNSET_VALUE
+			end
+
+			def load_vars_from_yml
+				if unset(@api_key) || unset(@secret_key) || unset(@connect_url)
+					if unset(@scriptrock_yml_path)
+						puts "ScriptRock yml config path un-set, not loading values from yml"
+					elsif !File.exist?(File.expand_path(@scriptrock_yml_path))
+						puts "ScriptRock yml file '#{@scriptrock_yml_path}' doesn't exist, not loading values from yml"
+					else
+						yml = YAML.load(File.read(File.expand_path(@scriptrock_yml_path)))
+						puts yml if @debug
+						if unset(@api_key)
+							@api_key = yml["api_key"]
+						end
+						if unset(@secret_key)
+							@secret_key = yml["secret_key"]
+						end
+						if unset(@connect_url)
+							@connect_url = yml["connect_url"]
+						end
+					end
+				end
+			end
+
+			def get_ssh_pubkey
+				if @ssh_pubkey != UNSET_VALUE
+					return @ssh_pubkey
+				end
+
+				load_vars_from_yml()
+				begin
+					puts "ScriptRock API connect_url #{@connect_url}" if @debug
+					response = HTTParty.get(
+						"#{@connect_url}/api/v1/users/ssh_key.json",
+						:headers => {
+							"Authorization" => "Token token=\"#{@api_key}\""
+						})
+					if response.code == 200
+						h = JSON.parse(response.body)
+						@ssh_pubkey = h["public_key"]
+					else
+						puts "ScriptRock get_ssh_pubkey error code = #{response.code}"
+						puts "ScriptRock get_ssh_pubkey error body = #{response.body}"
+					end
+				rescue => e
+					puts "ScriptRock get_ssh_pubkey error = #{e.class}: #{e.message}"
+				end
+				return @ssh_pubkey	
+			end
+
+			def finalize!
+				puts "finalize!" if @debug
+				get_ssh_pubkey()
+				if unset(@first_hop)
+					@first_hop = ""
+				end
+			end
+
+			def unset(v)
+				return v == UNSET_VALUE || v == nil || v == ""
+			end
+
+			def dump
+				puts "config dump"
+				puts "yml_path #{@scriptrock_yml_path}"
+				puts "api_key #{@api_key}"
+				puts "connect_url #{@connect_url}"
+				puts "ssh_pubkey #{@ssh_pubkey}"
+			end
+
+			def validate(machine)
+				puts "config validate" if @debug
+				dump if @debug
+
+				errors = _detected_errors
+				if unset(@api_key)
+					errors << "ScriptRock Guardrail api_key is not set"
+				end
+				if unset(@secret_key)
+					errors << "ScriptRock Guardrail secret_key is not set"
+				end
+				if unset(@connect_url)
+					errors << "ScriptRock Guardrail connect_url is not set"
+				end
+				if unset(@ssh_pubkey)
+					errors << "ScriptRock Guardrail ssh public key is not set"
+				end
+
+				{ "errors" => errors }
+			end			
+		end
+	end
+end

data/lib/vagrant-scriptrock/plugin.rb

@@ -0,0 +1,30 @@
+require "vagrant"
+
+# This is a sanity check to make sure no one is attempting to install
+# this into an early Vagrant version.
+if Vagrant::VERSION < "1.2.0"
+	raise "The Vagrant ScriptRock plugin is only compatible with Vagrant 1.2+"
+end
+
+module VagrantPlugins
+	module ScriptRock
+		class Plugin < Vagrant.plugin("2")
+			name "ScriptRock"
+			description <<-DESC
+This plugin will install a Guardrail public key ~/.ssh/authorized_keys
+on the instantiated vm, register the new node in the target Guardrail site,
+and delete the node from the Guardrail when the vm is destroyed.
+DESC
+
+			config(:scriptrock) do
+				require_relative "config"
+				Config
+			end
+
+			provisioner(:scriptrock) do
+				require_relative "provisioner"
+				Provisioner
+			end
+		end
+	end
+end

data/lib/vagrant-scriptrock/provisioner.rb

@@ -0,0 +1,125 @@
+require 'cgi'
+require 'httparty'
+
+module VagrantPlugins
+	module ScriptRock
+		class Provisioner < Vagrant.plugin("2", :provisioner)
+			def initialize(machine, config)
+				@debug = true#false
+				@machine = machine
+				@root_config = machine.config
+				puts "provision initialize config" if @debug
+			end
+
+			def configure(root_config)
+				puts "provision configure" if @debug
+				@root_config = root_config
+			end
+
+			def guardrail_name
+				return "vagrant #{@machine.name} #{@machine.id}"
+			end
+
+			def guardrail_auth_headers
+				return { "Authorization" => "Token token=\"#{@root_config.scriptrock.api_key}#{@root_config.scriptrock.secret_key}\"" }
+			end
+
+			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
+
+			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
+
+			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
+
+			def guardrail_delete
+				begin
+					node = guardrail_lookup_and_show
+					if node != nil
+						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
+
+			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
+
+			def provision
+				puts "provision provision" if @debug
+
+				# insert the guardrail public key onto the target node if it is not already present
+				@machine.communicate.tap do |comm|
+					pk = @root_config.scriptrock.ssh_pubkey
+					comm.execute("mkdir -p ~/.ssh && grep -q -s '#{pk}' ~/.ssh/authorized_keys || echo '#{pk}' >> ~/.ssh/authorized_keys")
+				end
+
+				# add this node to guardrail if not already present, then update to use the current credentials + forwarded port
+				guardrail_create_update
+			end
+
+			def cleanup
+				puts "provision cleanup" if @debug
+				guardrail_delete
+			end
+		end
+	end
+end

data/lib/vagrant-scriptrock/version.rb

@@ -0,0 +1,5 @@
+module VagrantPlugins
+	module ScriptRock
+		VERSION = "0.0.1"
+	end
+end

data/lib/vagrant-scriptrock.rb

@@ -0,0 +1,17 @@
+require "pathname"
+require "vagrant-scriptrock/plugin"
+
+module VagrantPlugins
+	module ScriptRock
+		lib_path = Pathname.new(File.expand_path("../vagrant-scriptrock", __FILE__))
+		autoload :Config,      lib_path.join("config")
+		autoload :Provisioner, lib_path.join("provisioner")
+
+		# This returns the path to the source of this plugin.
+		#
+		# @return [Pathname]
+		def self.source_root
+			@source_root ||= Pathname.new(File.expand_path("../../", __FILE__))
+		end
+	end
+end

data/vagrant-scriptrock.gemspec

@@ -0,0 +1,26 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "vagrant-scriptrock/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "vagrant-scriptrock"
+  spec.version       = VagrantPlugins::ScriptRock::VERSION
+  spec.authors       = ["Mark Sheahan"]
+  spec.email         = ["mark.sheahan@scriptrock.com"]
+  spec.homepage      = "https://github.com/ScriptRock/vagrant-scriptrock"
+  spec.summary       = "Vagrant plugin for ScriptRock Guardrail node registry/deletion when Vagrant VMs are provisioned/destroyed"
+  spec.description   = '''This plugin will install a Guardrail public key ~/.ssh/authorized_keys
+on the instantiated vm, register the VM as a new node on the target Guardrail site,
+and delete the Guardrail node when the VM is destroyed.'''
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.6"
+  spec.add_development_dependency "rake"
+end
+

metadata

@@ -0,0 +1,88 @@
+--- !ruby/object:Gem::Specification
+name: vagrant-scriptrock
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Mark Sheahan
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-01-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: |-
+  This plugin will install a Guardrail public key ~/.ssh/authorized_keys
+  on the instantiated vm, register the VM as a new node on the target Guardrail site,
+  and delete the Guardrail node when the VM is destroyed.
+email:
+- mark.sheahan@scriptrock.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- CHANGELOG.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/vagrant-scriptrock.rb
+- lib/vagrant-scriptrock/config.rb
+- lib/vagrant-scriptrock/plugin.rb
+- lib/vagrant-scriptrock/provisioner.rb
+- lib/vagrant-scriptrock/version.rb
+- vagrant-scriptrock.gemspec
+homepage: https://github.com/ScriptRock/vagrant-scriptrock
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Vagrant plugin for ScriptRock Guardrail node registry/deletion when Vagrant
+  VMs are provisioned/destroyed
+test_files: []