vagrant-nodemaster 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in vagrant-nodemaster.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1 @@
+require 'bundler/gem_tasks'

data/lib/vagrant-nodemaster.rb

@@ -0,0 +1,26 @@
+require "vagrant-nodemaster/version"
+
+module Vagrant
+  module NodeMaster
+    class Plugin < Vagrant.plugin("2")
+    
+    	name "remote"
+    	description <<-DESC
+    	Este es mi primer plugin y estoy probando la estructura
+    	DESC
+
+
+    	command ('remote') do
+    		require_relative "vagrant-nodemaster/remotecommand"
+    		Command
+			end
+
+    	command ('node') do
+    		require_relative "vagrant-nodemaster/nodecommand"
+    		Command
+			end
+		
+				
+    end
+  end
+end

data/lib/vagrant-nodemaster/apidesc.rb

@@ -0,0 +1,220 @@
+module RestRoutes		
+	#FIXME Figure how to manage remote port
+#	REMOTE_PORT = 3333
+	
+	class RouteManager
+	
+		def self.box_list_route
+			BOX_LIST_ROUTE
+		end
+		
+		def self.box_delete_route
+			BOX_DELETE_ROUTE
+		end
+		
+		def self.box_add_route
+			BOX_ADD_ROUTE
+		end
+		
+		def self.vm_up_route
+			VM_UP_ROUTE
+		end
+		
+		def self.vm_suspend_route 
+			VM_SUSPEND_ROUTE
+		end
+		
+		def self.vm_resume_route
+			VM_RESUME_ROUTE
+		end
+		
+		def self.vm_halt_route
+			VM_HALT_ROUTE
+		end
+		
+		def self.vm_destroy_route
+			VM_DESTROY_ROUTE
+		end
+		
+		def self.vm_status_route
+			VM_STATUS_ROUTE
+		end
+		
+		def self.vm_provision_route
+			VM_PROVISION_ROUTE
+		end
+		
+		def self.vm_status_all_route
+			VM_STATUS_ALL_ROUTE
+		end
+		
+		def self.vm_sshconfig_route
+			SSH_CONFIG_ROUTE
+		end
+		
+		def self.snapshots_all_route
+			SNAPSHOTS_ALL_ROUTE
+		end
+		
+		def self.vm_snapshots_route
+			VM_SNAPSHOTS_ROUTE
+		end
+		
+		def self.vm_snapshot_take_route
+			VM_SNAPSHOT_TAKE_ROUTE
+		end
+		
+		def self.vm_snapshot_restore_route
+			VM_SNAPSHOT_RESTORE_ROUTE
+		end
+		
+		def self.node_backup_log_route
+			NODE_BACKUP_LOG_ROUTE
+		end
+		
+		def self.vm_backup_log_route
+			VM_BACKUP_LOG_ROUTE
+		end
+		
+		
+		def self.box_list_url(host,port)
+			"http://#{host}:#{port}#{box_list_route}"
+		end
+		
+		def self.box_add_url(host,port)			
+			"http://#{host}:#{port}#{box_add_route}"
+		end
+		
+		def self.box_delete_url(host,port,box,provider)
+			
+			url=String.new(box_delete_route)
+			url[":box"]=box
+			url[":provider"]=provider
+			url="http://#{host}:#{port}#{url}"
+			
+			url
+			
+		end
+
+		
+		
+		def self.vm_up_url(host,port)
+			"http://#{host}:#{port}#{vm_up_route}"
+		end
+		
+		def self.vm_halt_url(host,port)
+			"http://#{host}:#{port}#{vm_halt_route}"
+		end
+	
+		def self.vm_destroy_url(host,port)
+			"http://#{host}:#{port}#{vm_destroy_route}"
+		end			
+		
+		def self.vm_suspend_url(host,port)
+			"http://#{host}:#{port}#{vm_suspend_route}"
+		end
+		
+		def self.vm_resume_url(host,port)
+			"http://#{host}:#{port}#{vm_resume_route}"
+		end
+		
+		def self.vm_provision_url(host,port)
+			"http://#{host}:#{port}#{vm_provision_route}"
+		end
+		
+			
+		
+		
+		def self.vm_status_url(host,port,vmname=nil)	
+			url="http://#{host}:#{port}#{vm_status_all_route}"
+		
+			if (vmname!=nil)
+				url=String.new(vm_status_route)
+				url[":vm"]=vmname
+				url="http://#{host}:#{port}#{url}"		
+			end
+			
+			url
+			
+		end
+		
+		def self.vm_sshconfig_url(host,port,vmname)
+			url=String.new(vm_sshconfig_route)
+			url[":vm"]=vmname
+			
+			"http://#{host}:#{port}#{url}"
+								
+		end
+		
+		def self.snapshot_list_url(host,port,vmname=nil)	
+			url="http://#{host}:#{port}#{snapshots_all_route}"
+		
+			if (vmname!=nil)
+				url=String.new(vm_snapshots_route)
+				url[":vm"]=vmname
+				url="http://#{host}:#{port}#{url}"		
+			end
+			
+			url
+			
+		end
+		
+		def self.vm_snapshot_take_url(host,port,vmname)				
+			url=String.new(vm_snapshot_take_route)				
+			url[":vm"]=vmname
+			url="http://#{host}:#{port}#{url}"						
+			
+			url
+			
+		end
+		
+		def self.vm_snapshot_restore_url(host,port,vmname)
+			url=String.new(vm_snapshot_restore_route)
+			url[":vm"]=vmname
+			url="http://#{host}:#{port}#{url}"		
+			
+			url
+			
+		end
+		
+		def self.backup_log_url(host,port,vmname=nil)
+			url="http://#{host}:#{port}#{node_backup_log_route}"
+		
+			if (vmname!=nil)
+				url=String.new(vm_backup_log_route)
+				url[":vm"]=vmname
+				url="http://#{host}:#{port}#{url}"		
+			end
+			
+			url
+			
+		end
+		
+		
+		
+		private
+			BOX_LIST_ROUTE =	"/api/box/list"
+			BOX_DELETE_ROUTE =	"/api/box/:box/:provider/delete"
+			BOX_ADD_ROUTE = "/api/box/add" 
+			
+			VM_UP_ROUTE = "/api/vm/up"
+			VM_HALT_ROUTE = "/api/vm/halt"
+			VM_DESTROY_ROUTE =	"/api/vm/destroy"
+			VM_SUSPEND_ROUTE = "/api/vm/suspend"
+			VM_RESUME_ROUTE = "/api/vm/resume"
+			VM_PROVISION_ROUTE = "/api/vm/provision"
+			VM_STATUS_ALL_ROUTE = "/api/vm/status"
+			VM_STATUS_ROUTE = "/api/vm/:vm/status"			
+			SSH_CONFIG_ROUTE = "/api/vm/:vm/sshconfig"
+			
+			SNAPSHOTS_ALL_ROUTE = "/api/vm/snapshots"
+			VM_SNAPSHOTS_ROUTE = "/api/vm/:vm/snapshots"
+			VM_SNAPSHOT_TAKE_ROUTE = "/api/vm/:vm/take"
+			VM_SNAPSHOT_RESTORE_ROUTE = "/api/vm/:vm/restore"
+			
+			VM_BACKUP_LOG_ROUTE = "/api/vm/:vm/backuplog"
+			NODE_BACKUP_LOG_ROUTE = "/api/backuplog"
+			
+	end
+	
+end

data/lib/vagrant-nodemaster/node/nodeadd.rb

@@ -0,0 +1,33 @@
+require 'vagrant-nodemaster/node/nodedbmanager'
+
+module Vagrant
+  module NodeMaster
+  
+		class NodeAdd < Vagrant.plugin(2, :command)
+
+			def execute
+					
+					options = {}
+					
+					opts = OptionParser.new do |opts|
+						opts.banner = "Usage: vagrant node add <node-name> <node-address> <node-port> --hostname"
+						opts.on("--hostname", "Address in dns format") do |d|
+							options[:dns] = d
+						end
+					end
+					
+					argv = parse_options(opts)
+					return if !argv
+		  		raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if argv.length != 3
+		  		
+		  		dbmanager=DB::NodeDBManager.new
+		  		
+		  		dbmanager.add_node(argv[0],argv[1],argv[2].to_i,options[:dns])
+					
+					0
+			end
+  
+  	end
+  
+  end
+end

data/lib/vagrant-nodemaster/node/nodedbmanager.rb

@@ -0,0 +1,176 @@
+require 'sqlite3'
+
+module Vagrant
+	module NodeMaster
+		module DB
+			class NodeDBManager
+				
+				def initialize
+					@db=check_database					
+				end
+				
+				def add_node(node_name,node_address,node_port,dnsformat)
+					
+					if (!portvalid?(node_port))		
+						raise "Node port must be above 1023"
+					end
+					
+					
+					if (!namevalid?(node_name))
+						raise "Node name Invalid (Spectial characters allowed: '_' and '-')." 
+					end
+					
+					if (dnsformat && !dnsvalid?(node_address)) || 
+						(!dnsformat && !ipvalid?(node_address))						
+					
+						raise "Invalid ip or dns address."
+						 
+					end				
+														
+					
+					sql="INSERT INTO #{TABLE_NAME} VALUES ( ? , ? , ?)"
+					@db.execute(sql,node_name,node_address,node_port)
+					
+				end
+				
+				def update_node(node_name,node_address,node_port,dnsformat)
+					
+					if (!portvalid?(node_port))						
+					
+						raise "Node port must be above 1023"
+						 
+					end
+					
+					if (dnsformat && !dnsvalid?(node_address)) || 
+						(!dnsformat && !ipvalid?(node_address))						
+					
+						raise "Invalid ip or dns address."
+						 
+					end			
+					
+					sql="UPDATE #{TABLE_NAME} SET #{NODE_ADDRESS_COLUMN} = ?,#{NODE_PORT_COLUMN} = ?  WHERE #{NODE_NAME_COLUMN}= ?"
+					@db.execute(sql,node_address,node_port,node_name)
+					
+				end
+				
+
+				
+				
+				def remove_node(node_name)
+					sql="DELETE FROM #{TABLE_NAME} WHERE #{NODE_NAME_COLUMN}=?"
+					@db.execute(sql,node_name)
+				end
+				
+				def remove_nodes
+					sql="DELETE FROM #{TABLE_NAME}"
+					@db.execute(sql)
+				end
+
+				
+				def get_node_names
+					sql="SELECT #{NODE_NAME_COLUMN} FROM #{TABLE_NAME}"
+					return @db.execute(sql);
+				end				
+				
+				def get_nodes				
+					sql="SELECT * FROM #{TABLE_NAME}"
+					rows=@db.execute(sql);
+					nodes = []
+					rows.each do |row|
+						node = {}
+						node[:name]=row[0]
+						node[:address]=row[1]
+						node[:port]=row[2]
+						nodes << node
+					end
+
+					return nodes
+				end
+				
+				def get_node(node_name)
+					sql="SELECT * FROM #{TABLE_NAME} WHERE #{NODE_NAME_COLUMN}=?"
+					raise "Node not found" if (nodear=@db.execute(sql,node_name)).empty?
+					node = {}
+					node[:name]=nodear[0][0]
+					node[:address]=nodear[0][1]
+					node[:port]=nodear[0][2]
+					return node
+				end
+				
+				private
+			
+				TABLE_NAME='node_table'
+				NODE_NAME_COLUMN = 'node_name'
+				NODE_ADDRESS_COLUMN = 'node_address'
+				NODE_PORT_COLUMN = 'node_port'
+				
+				def check_database					
+					#Creates and/or open the database
+					db = SQLite3::Database.new( "/tmp/test.db" )
+									
+					if db.execute("SELECT name FROM sqlite_master 
+											 WHERE type='table' AND name='#{TABLE_NAME}';").length==0						
+#						db.execute( "create table '#{TABLE_NAME}' (id INTEGER PRIMARY KEY AUTOINCREMENT, 
+#												{NODE_NAME_COLUMN} TEXT, #{NODE_ADDRESS_COLUMN} TEXT);" )				
+						db.execute( "create table '#{TABLE_NAME}' (#{NODE_NAME_COLUMN} TEXT PRIMARY KEY, 
+												 																#{NODE_ADDRESS_COLUMN} TEXT NOT NULL,
+												 																#{NODE_PORT_COLUMN} INTEGER NOT NULL);" )
+					end
+					return db
+				end
+		
+				def ipvalid?(node_address)
+						
+						ipreg= '^(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])$'
+						
+						
+						ipregex = Regexp.new(ipreg)
+						
+
+						if (node_address =~ ipregex) 
+							return true
+						end
+						
+						return false 
+						
+				end
+				
+				def dnsvalid?(node_address)
+						
+						
+						hostreg='^(([a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9\-]*[a-zA-Z0-9])\.)*([A-Za-z0-9]|[A-Za-z0-9][A-Za-z0-9\-]*[A-Za-z0-9])$'
+						
+						
+						hregex = Regexp.new(hostreg)
+
+						if (node_address =~ hregex)
+							return true
+						end
+						
+						return false 
+						
+				end
+				
+						
+				def namevalid?(node_name)
+					#only allowing '_' and '-'
+					namereg = '^[a-z0-9_-]*$'
+					nameregex = Regexp.new(namereg)
+					if (node_name =~nameregex)
+						return true
+					end
+						return false
+						
+				end
+				
+				def portvalid?(node_port)
+					if (node_port>=1024)
+						return true
+					end
+					return false
+				end
+						
+			end
+		end
+	end
+end