vagrant-proxmox 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 57fd9b0a41cefd93d184deb4f2156c278b7ea0f6
+  data.tar.gz: 6aff0afae6d0cd851a8780060c25b20877d0f599
+SHA512:
+  metadata.gz: 460cd1faf092d13e2790d37ef1a7c40d6d7f289e5e15ffd2f6ddd42f44531f89b298e7462232133406f6529b04223a9913ecc1e751e17c8ea2d9fffa1cc89512
+  data.tar.gz: c63ab8bd9b932b52a06d50fb2181d0076e6251834955c60ccedabb6824459903be6e7a69140fca1faaeb91349f9f2a759603ccb198359af2a7f0663895451ea2

data/lib/sanity_checks.rb

@@ -0,0 +1,11 @@
+begin
+	require 'vagrant'
+rescue LoadError
+	raise 'The Vagrant Proxmox plugin must be run within Vagrant.'
+end
+
+# 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 Proxmox plugin is only compatible with Vagrant 1.2+'
+end

data/lib/vagrant-proxmox/action/cleanup_after_destroy.rb

@@ -0,0 +1,20 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			class CleanupAfterDestroy < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+				end
+
+				def call env
+					FileUtils.rm_rf ".vagrant/machines/#{env[:machine].name}/proxmox"
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/connect_proxmox.rb

@@ -0,0 +1,33 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			# This action connects to the Proxmox server and stores the access ticket
+			# and csrf prevention token in env[:proxmox_ticket] and
+			# env[:proxmox_csrf_prevention_token].
+			class ConnectProxmox < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+					@logger = Log4r::Logger.new 'vagrant_proxmox::action::connect_proxmox'
+				end
+
+				def call env
+					config = env[:machine].provider_config
+					response = RestClient.post "#{config.endpoint}/access/ticket",
+					                           username: config.user_name, password: config.password
+					begin
+						json_response = JSON.parse response.to_s, symbolize_names: true
+						env[:proxmox_ticket] = json_response[:data][:ticket]
+						env[:proxmox_csrf_prevention_token] = json_response[:data][:CSRFPreventionToken]
+					rescue => e
+						raise Errors::CommunicationError, error_msg: e.message
+					end
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/create_vm.rb

@@ -0,0 +1,63 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			# This action creates a new virtual machine on the Proxmox server and
+			# stores its node and vm_id env[:machine].id
+			class CreateVm < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+					@logger = Log4r::Logger.new 'vagrant_proxmox::action::create_vm'
+				end
+
+				def call env
+					env[:ui].info I18n.t('vagrant_proxmox.creating_vm')
+					config = env[:machine].provider_config
+					node = env[:proxmox_nodes].sample[:node]
+					vm_id = nil
+
+					begin
+						retryable(tries: config.task_timeout,
+						          sleep: config.task_status_check_interval) do
+							vm_id = get_free_vm_id env
+							params = {vmid: vm_id,
+							          ostemplate: config.os_template,
+							          hostname: env[:machine].config.vm.hostname || env[:machine].name.to_s,
+							          password: 'vagrant',
+							          memory: config.vm_memory,
+							          description: "#{config.vm_name_prefix}#{env[:machine].name}"}
+							get_machine_ip_address(env).try { |ip_address| params[:ip_address] = ip_address }
+
+							response = RestClient.post "#{config.endpoint}/nodes/#{node}/openvz", params,
+							                           {CSRFPreventionToken: env[:proxmox_csrf_prevention_token],
+							                            cookies: {PVEAuthCookie: env[:proxmox_ticket]}}
+							exit_status = wait_for_completion parse_task_id(response), node, env, 'vagrant_proxmox.errors.create_vm_timeout'
+							exit_status == 'OK' ? exit_status : raise(VagrantPlugins::Proxmox::Errors::ProxmoxTaskFailed, proxmox_exit_status: exit_status)
+						end
+					rescue StandardError => e
+						raise VagrantPlugins::Proxmox::Errors::VMCreationError, proxmox_exit_status: e.message
+					end
+
+					env[:machine].id = "#{node}/#{vm_id}"
+
+					env[:ui].info I18n.t('vagrant_proxmox.done')
+					next_action env
+				end
+
+				private
+				def get_free_vm_id env
+					response = RestClient.get "#{env[:machine].provider_config.endpoint}/cluster/resources?type=vm", {cookies: {PVEAuthCookie: env[:proxmox_ticket]}}
+					json_response = JSON.parse response.to_s, symbolize_names: true
+
+					allowed_vm_ids = env[:machine].provider_config.vm_id_range.to_set
+					used_vm_ids = json_response[:data].map { |vm| vm[:vmid] }
+					free_vm_ids = (allowed_vm_ids - used_vm_ids).sort
+					free_vm_ids.empty? ? raise(VagrantPlugins::Proxmox::Errors::NoVmIdAvailable) : free_vm_ids.first
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/destroy_vm.rb

@@ -0,0 +1,29 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			class DestroyVm < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+				end
+
+				def call env
+					env[:ui].info I18n.t('vagrant_proxmox.destroying_vm')
+					node, vm_id = env[:machine].id.split '/'
+					endpoint = env[:machine].provider_config.endpoint
+					response = RestClient.delete "#{endpoint}/nodes/#{node}/openvz/#{vm_id}",
+																		 {CSRFPreventionToken: env[:proxmox_csrf_prevention_token],
+																		  cookies: {PVEAuthCookie: env[:proxmox_ticket]}}
+
+					wait_for_completion parse_task_id(response), node, env, 'vagrant_proxmox.errors.destroy_vm_timeout'
+					env[:ui].info I18n.t('vagrant_proxmox.done')
+
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/get_node_list.rb

@@ -0,0 +1,24 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			# This action gets a list of all the nodes of a Proxmox server cluster
+			# and stores it under env[:proxmox_nodes]
+			class GetNodeList < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+				end
+
+				def call env
+					endpoint = env[:machine].provider_config.endpoint
+					response = RestClient.get "#{endpoint}/nodes", {cookies: {PVEAuthCookie: env[:proxmox_ticket]}}
+					env[:proxmox_nodes] = JSON.parse(response.to_s, symbolize_names: true)[:data]
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/is_created.rb

@@ -0,0 +1,20 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			class IsCreated < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+				end
+
+				def call env
+					env[:result] = env[:machine].state.id != :not_created
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/is_stopped.rb

@@ -0,0 +1,20 @@
+module VagrantPlugins
+  module Proxmox
+    module Action
+
+      class IsStopped < ProxmoxAction
+
+        def initialize app, env
+          @app = app
+        end
+
+				def call env
+          env[:result] = env[:machine].state.id == :stopped
+					next_action env
+				end
+
+			end
+
+    end
+  end
+end

data/lib/vagrant-proxmox/action/message_already_running.rb

@@ -0,0 +1,20 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			class MessageAlreadyRunning < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+				end
+
+				def call env
+					env[:ui].info I18n.t('vagrant_proxmox.already_running')
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/message_already_stopped.rb

@@ -0,0 +1,20 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			class MessageAlreadyStopped < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+				end
+
+				def call env
+					env[:ui].info I18n.t('vagrant_proxmox.already_stopped')
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/message_not_created.rb

@@ -0,0 +1,20 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			class MessageNotCreated < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+				end
+
+ 				def call env
+					env[:ui].info I18n.t('vagrant_proxmox.not_created')
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/proxmox_action.rb

@@ -0,0 +1,47 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			class ProxmoxAction
+
+				protected
+				def next_action env
+					@app.call env
+				end
+
+				protected
+				def parse_task_id proxmox_response
+					JSON.parse(proxmox_response.to_s, symbolize_names: true)[:data]
+				end
+
+				# Wait for the completion of the proxmox task with the given upid.
+				protected
+				def wait_for_completion task_upid, node, env, timeout_message
+					begin
+						retryable(on: VagrantPlugins::Proxmox::ProxmoxTaskNotFinished,
+											tries: env[:machine].provider_config.task_timeout,
+											sleep: env[:machine].provider_config.task_status_check_interval) do
+							exit_status = get_task_exitstatus task_upid, node, env
+							exit_status.nil? ? raise(VagrantPlugins::Proxmox::ProxmoxTaskNotFinished) : exit_status
+						end
+					rescue VagrantPlugins::Proxmox::ProxmoxTaskNotFinished
+						raise Errors::Timeout.new timeout_message
+					end
+				end
+
+				protected
+				def get_machine_ip_address env
+					env[:machine].config.vm.networks.select { |type, _| type == :public_network }.first[1][:ip] rescue nil
+				end
+
+				private
+				def get_task_exitstatus task_upid, node, env
+					response = RestClient.get "#{env[:machine].provider_config.endpoint}/nodes/#{node}/tasks/#{task_upid}/status", {cookies: {PVEAuthCookie: env[:proxmox_ticket]}}
+					parse_task_id(response)[:exitstatus]
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/read_ssh_info.rb

@@ -0,0 +1,22 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			class ReadSSHInfo < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+				end
+
+				def call env
+					env[:machine_ssh_info] = get_machine_ip_address(env).try do |ip_address|
+						{host: ip_address, port: env[:machine].config.ssh.guest_port}
+					end
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/read_state.rb

@@ -0,0 +1,37 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			# This action reads the state of a Proxmox virtual machine and stores it
+			# in env[:machine_state_id].
+			class ReadState < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+					@logger = Log4r::Logger.new 'vagrant_proxmox::action::read_state'
+				end
+
+				def call env
+					begin
+						if env[:machine].id
+							node, vm_id = env[:machine].id.split '/'
+							endpoint = env[:machine].provider_config.endpoint
+							response = RestClient.get "#{endpoint}/nodes/#{node}/openvz/#{vm_id}/status/current",
+																				{cookies: {PVEAuthCookie: env[:proxmox_ticket]}}
+							states = {'running' => :running,
+												'stopped' => :stopped}
+							env[:machine_state_id] = states[JSON.parse(response.to_s)['data']['status']]
+						else
+							env[:machine_state_id] = :not_created
+						end
+					rescue RestClient::InternalServerError
+						env[:machine_state_id] = :not_created
+					end
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/shutdown_vm.rb

@@ -0,0 +1,31 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			# This action shuts down a Proxmox virtual machine.
+			class ShutdownVm < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+					@logger = Log4r::Logger.new 'vagrant_proxmox::action::shutdown_vm'
+				end
+
+				def call env
+					endpoint = env[:machine].provider_config.endpoint
+					node, vm_id = env[:machine].id.split '/'
+					env[:ui].info I18n.t('vagrant_proxmox.shut_down_vm')
+					response = RestClient.post "#{endpoint}/nodes/#{node}/openvz/#{vm_id}/status/shutdown", nil,
+																		 {CSRFPreventionToken: env[:proxmox_csrf_prevention_token],
+																		  cookies: {PVEAuthCookie: env[:proxmox_ticket]}}
+
+					wait_for_completion parse_task_id(response), node, env, 'vagrant_proxmox.errors.shutdown_vm_timeout'
+					env[:ui].info I18n.t('vagrant_proxmox.done')
+
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/start_vm.rb

@@ -0,0 +1,40 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			# This action starts a Proxmox virtual machine.
+			class StartVm < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+					@logger = Log4r::Logger.new 'vagrant_proxmox::action::start_vm'
+				end
+
+				def call env
+					endpoint = env[:machine].provider_config.endpoint
+					node, vm_id = env[:machine].id.split '/'
+					env[:ui].info I18n.t('vagrant_proxmox.starting_vm')
+					response = RestClient.post "#{endpoint}/nodes/#{node}/openvz/#{vm_id}/status/start", nil,
+																		 {CSRFPreventionToken: env[:proxmox_csrf_prevention_token],
+																		  cookies: {PVEAuthCookie: env[:proxmox_ticket]}}
+
+					wait_for_completion parse_task_id(response), node, env, 'vagrant_proxmox.errors.start_vm_timeout'
+					env[:ui].info I18n.t('vagrant_proxmox.done')
+
+					env[:ui].info I18n.t('vagrant_proxmox.waiting_for_ssh_connection')
+					loop do
+						# If we're interrupted then just back out
+						break if env[:interrupted]
+						break if env[:machine].communicate.ready?
+						sleep env[:machine].provider_config.task_status_check_interval
+					end
+					env[:ui].info I18n.t('vagrant_proxmox.done')
+
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/stop_vm.rb

@@ -0,0 +1,31 @@
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			# This action stops a Proxmox virtual machine.
+			class StopVm < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+					@logger = Log4r::Logger.new 'vagrant_proxmox::action::stop_vm'
+				end
+
+				def call env
+					endpoint = env[:machine].provider_config.endpoint
+					node, vm_id = env[:machine].id.split '/'
+					env[:ui].info I18n.t('vagrant_proxmox.stopping_vm')
+					response = RestClient.post "#{endpoint}/nodes/#{node}/openvz/#{vm_id}/status/stop", nil,
+																		 {CSRFPreventionToken: env[:proxmox_csrf_prevention_token],
+																		  cookies: {PVEAuthCookie: env[:proxmox_ticket]}}
+
+					wait_for_completion parse_task_id(response), node, env, 'vagrant_proxmox.errors.stop_vm_timeout'
+					env[:ui].info I18n.t('vagrant_proxmox.done')
+
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end

data/lib/vagrant-proxmox/action/sync_folders.rb

@@ -0,0 +1,52 @@
+require 'vagrant/util/subprocess'
+
+module VagrantPlugins
+	module Proxmox
+		module Action
+
+			# This action uses 'rsync' to sync the folders over to the virtual machine.
+			class SyncFolders < ProxmoxAction
+
+				def initialize app, env
+					@app = app
+					@logger = Log4r::Logger.new 'vagrant_proxmox::action::sync_folders'
+				end
+
+				def call env
+					ssh_info = env[:machine].ssh_info
+
+					env[:machine].config.vm.synced_folders.each do |_, data|
+						hostpath = File.expand_path data[:hostpath], env[:root_path]
+						guestpath = data[:guestpath]
+						next if data[:disabled]
+
+						# Make sure there is a trailing slash on the host path to
+						# avoid creating an additional directory with rsync
+						hostpath = "#{hostpath}/" if hostpath !~ /\/$/
+
+						env[:ui].info I18n.t('vagrant_proxmox.rsync_folder', hostpath: hostpath, guestpath: guestpath)
+
+						# Create the guest path
+						env[:machine].communicate.sudo "mkdir -p '#{guestpath}'"
+						env[:machine].communicate.sudo "chown #{ssh_info[:username]} '#{guestpath}'"
+
+						# rsync over to the guest path using the SSH info
+						command = [
+								'rsync', '--verbose', '--archive', '--compress', '--delete',
+								'-e', "ssh -p #{ssh_info[:port]} -i '#{ssh_info[:private_key_path]}' -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null",
+								hostpath, "#{ssh_info[:username]}@#{ssh_info[:host]}:#{guestpath}"]
+
+						rsync_process = Vagrant::Util::Subprocess.execute *command
+						if rsync_process.exit_code != 0
+							raise Errors::RsyncError, guestpath: guestpath, hostpath: hostpath, stderr: rsync_process.stderr
+						end
+					end
+
+					next_action env
+				end
+
+			end
+
+		end
+	end
+end