vagrant-nodemaster 0.0.1

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

@@ -0,0 +1,34 @@
+require 'vagrant-nodemaster/node/nodedbmanager'
+
+module Vagrant
+  module NodeMaster
+  
+		class NodeList < Vagrant.plugin(2, :command)
+			def execute
+					
+					options = {}
+					
+					opts = OptionParser.new do |opts|
+						opts.banner = "Usage: vagrant node list"
+					end
+					
+					argv = parse_options(opts)
+					return if !argv
+		  		raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if argv.length != 0
+		  		
+		  		dbmanager=DB::NodeDBManager.new
+		  		
+		  		@env.ui.info("#{"NODE".ljust(25)} ADDRESS (PORT) ",:prefix => false)
+					dbmanager.get_nodes.each do |entry|
+						@env.ui.info("#{entry[:name].ljust(25)} #{entry[:address]}(#{entry[:port]})", :prefix => false) 
+					end
+					@env.ui.info(" ", :prefix => false)
+					0
+			end
+  
+  	end
+  
+  
+  
+  end
+end

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

@@ -0,0 +1,43 @@
+require 'vagrant-nodemaster/node/nodedbmanager'
+
+module Vagrant
+  module NodeMaster
+  
+		class NodeRemove < Vagrant.plugin(2, :command)
+#			def initialize(sub_args, env,db)
+#					super(sub_args,env)					
+#					@db=db
+#			end
+			def execute
+					
+					options = {}
+					
+					opts = OptionParser.new do |opts|
+						opts.banner = "Usage: vagrant node remove [node-name] --clean"
+						opts.on("-c","--clean", "Remove all nodes from list") do |c|
+							options[:clean] = c
+						end
+					end
+					
+					argv = parse_options(opts)
+					
+					return if !argv
+		  		raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if (argv.length < 1 && !options[:clean]) || (argv.length >1)
+		  		
+		  		dbmanager=DB::NodeDBManager.new
+		  		
+		  		if (options[:clean])
+		  			dbmanager.remove_nodes
+		  		elsif
+		  			dbmanager.remove_node(argv[0])
+		  		end
+					
+					0
+			end
+  
+  	end
+  
+  
+  
+  end
+end

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

@@ -0,0 +1,73 @@
+require 'vagrant-nodemaster/node/nodedbmanager'
+require 'socket'
+require 'timeout'
+
+module Vagrant
+  module NodeMaster
+  
+		class NodeStatus < Vagrant.plugin(2, :command)
+
+			def execute
+					
+					options = {}
+					
+					opts = OptionParser.new do |opts|
+						opts.banner = "Usage: vagrant node status"
+					end
+					
+					argv = parse_options(opts)
+					return if !argv
+		  		raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if argv.length != 0
+		  		
+		  		dbmanager=DB::NodeDBManager.new
+
+					status = {}
+							  		
+
+
+					dbmanager.get_nodes.each do |entry|							
+							status[entry[:name]]=ERROR
+							if port_open?(entry[:address],entry[:port])
+								status[entry[:name]]=OK
+							end
+					end
+					
+					
+					#@env.ui.info("Client Status:", :prefix => false)
+					@env.ui.info("#{"NODE".ljust(25)} STATUS ",:prefix => false)
+					status.each do |name,status|
+							@env.ui.info("#{name.ljust(25)} #{status}",:prefix => false)
+					end
+					
+					@env.ui.info(" ",:prefix => false)
+					
+					0
+			end
+			
+			
+			private
+			OK="LISTENING"
+			ERROR= "HOST DOWN"
+			
+			def port_open?(ip, port, seconds=1)
+				begin
+					Timeout::timeout(seconds) do
+						begin
+							TCPSocket.new(ip, port).close
+							true
+						rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH
+							false
+						end
+					end
+		
+				rescue Timeout::Error
+					false
+				end
+			end
+  
+  	end
+  
+  
+  
+  end
+end

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

@@ -0,0 +1,43 @@
+require 'vagrant-nodemaster/node/nodedbmanager'
+
+module Vagrant
+  module NodeMaster
+  
+		class NodeUpdate < Vagrant.plugin(2, :command)
+#			def initialize(sub_args, env,db)
+#					super(sub_args,env)					
+#					@db=db
+#			end
+			def execute
+					
+					options = {}
+					
+					#TODO CAMBIAR LA FORMA DE INVOCAR A ESTE COMANDO PARA NO TENER
+					#QUE INTRODUCIR TODOS LOS DATOS SI SE QUIERE MODIFICAR LA INFORMACION
+					#DEL CLIENTE
+					opts = OptionParser.new do |opts|
+						opts.banner = "Usage: vagrant node update <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.update_node(argv[0],argv[1],argv[2].to_i,options[:dns])
+					
+					0
+			end
+  
+  	end
+  
+  
+  
+  end
+end

data/lib/vagrant-nodemaster/nodecommand.rb

@@ -0,0 +1,102 @@
+require 'vagrant/plugin'
+require 'sqlite3'
+
+module Vagrant
+  module NodeMaster
+   
+		class Command < Vagrant.plugin(2, :command)
+			
+			def initialize(argv, env)
+				super
+				
+				@main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
+				
+#				puts "MAIN ARGS #{@main_args}"
+#				puts "SUB COMMAND #{@sub_command}"
+#				puts "SUB ARGS #{@sub_args}"
+				
+				@subcommands = Vagrant::Registry.new
+				
+				@subcommands.register(:list) do
+					require File.expand_path("../node/nodelist", __FILE__)
+					NodeList
+				end       
+				
+				@subcommands.register(:add) do
+					require File.expand_path("../node/nodeadd", __FILE__)
+					NodeAdd
+				end
+				
+				@subcommands.register(:remove) do
+					require File.expand_path("../node/noderemove", __FILE__)
+					NodeRemove
+				end
+			
+				@subcommands.register(:update) do
+					require File.expand_path("../node/nodeupdate", __FILE__)
+					NodeUpdate
+				end
+
+				@subcommands.register(:status) do
+					require File.expand_path("../node/nodestatus", __FILE__)
+					NodeStatus
+				end
+				
+			end
+
+			
+			
+			
+			
+			def execute
+				if @main_args.include?("-h") || @main_args.include?("--help")
+					# Print the help for all the box commands.
+					return help
+				end
+				
+				
+				
+			  # If we reached this far then we must have a subcommand. If not,
+				# then we also just print the help and exit.
+				command_class = @subcommands.get(@sub_command.to_sym) if @sub_command
+				return help if !command_class || !@sub_command
+				@logger.debug("Invoking command class: #{command_class} #{@sub_args.inspect}")	  
+			  
+			  begin
+					# Initialize and execute the command class
+					command_class.new(@sub_args, @env).execute
+				rescue Exception => e
+					@env.ui.error(e.message)
+				end
+			  
+    
+				0
+			end
+			
+			def help
+				opts = OptionParser.new do |opts|
+					opts.banner = "Usage: vagrant node <command> [<args>]"
+					opts.separator ""
+					opts.separator "Available subcommands:"
+			
+					# Add the available subcommands as separators in order to print them
+					# out as well.
+					keys = []
+					@subcommands.each { |key, value| keys << key.to_s }
+			
+					keys.sort.each do |key|
+						opts.separator "     #{key}"
+					end
+			
+					opts.separator ""
+					opts.separator "For help on any individual command run `vagrant node COMMAND -h`"
+				end
+			
+				@env.ui.info(opts.help, :prefix => false)
+			end
+			
+			
+		end
+		
+	end
+end

data/lib/vagrant-nodemaster/remote/remotebackupcommand.rb

@@ -0,0 +1,90 @@
+
+require 'vagrant/plugin'
+ 
+
+module Vagrant
+  module NodeMaster
+ 
+	class BackupCommand < Vagrant.plugin(2, :command)
+		def initialize(argv, env)
+			super
+
+			@main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
+
+#			puts "MAIN ARGS #{@main_args}"
+#			puts "SUB COMMAND #{@sub_command}"
+#			puts "SUB ARGS #{@sub_args}"
+
+			@subcommands = Vagrant::Registry.new
+
+			@subcommands.register(:take) do
+				require File.expand_path("../remotebackuptake", __FILE__)
+				BackupTake
+			end
+			
+			@subcommands.register(:log) do
+				require File.expand_path("../remotebackuplog", __FILE__)
+				BackupLog
+			end
+							
+
+		end
+
+		def execute
+#			if @main_args.include?("-h") || @main_args.include?("--help")
+#				 Print the help for all the box commands.
+#				return help
+#			end
+			
+			# If we reached this far then we must have a subcommand. If not,
+			# then we also just print the help and exit.
+			command_class = @subcommands.get(@sub_command.to_sym) if @sub_command
+			return help if !command_class || !@sub_command
+			
+			@logger.debug("Invoking command class: #{command_class} #{@sub_args.inspect}")
+			
+				
+			begin	
+				# Initialize and execute the command class
+				command_class.new(@sub_args, @env).execute
+			rescue RestClient::RequestFailed => e
+				@env.ui.error("Remote Client \"#{@sub_args[0]}\": Request Failed")
+			rescue RestClient::ResourceNotFound => e          
+				@env.ui.error("Remote Client \"#{@sub_args[0]}\": Box \"#{@sub_args[1]}\" could not be found")
+			rescue RestClient::ExceptionWithResponse=> e          
+				@env.ui.error(e.response)
+			rescue Exception => e
+				@env.ui.error(e.message)
+			end
+
+			
+		end
+		
+		def help
+			opts = OptionParser.new do |opts|
+				opts.banner = "Usage: vagrant remote backup <command> [<args>]"
+				opts.separator ""
+				opts.separator "Available subcommands:"
+	
+				# Add the available subcommands as separators in order to print them
+				# out as well.
+				keys = []
+				@subcommands.each { |key, value| keys << key.to_s }
+	
+				keys.sort.each do |key|
+					opts.separator "     #{key}"
+				end
+	
+				opts.separator ""
+				opts.separator "For help on any individual command run `vagrant remote COMMAND -h`"
+			end
+	
+			@env.ui.info(opts.help, :prefix => false)
+		end
+
+
+
+		
+	end
+  end
+end

data/lib/vagrant-nodemaster/remote/remotebackuplog.rb

@@ -0,0 +1,62 @@
+require 'optparse'
+require 'vagrant-nodemaster/requestcontroller'
+
+
+module Vagrant
+  module NodeMaster
+  	
+		class BackupLog < Vagrant.plugin(2, :command)
+			def execute
+				options = {}
+	
+				
+				#FIXME add date params
+				opts = OptionParser.new do |opts|
+					opts.banner = "Usage: vagrant remote backup log <node-name> [vmname] [-h]"
+#					opts.separator ""
+#          opts.on("-b", "--background", "Take backup in background") do |b|
+#                  options[:background] = b
+#          end
+
+				end
+				
+				
+				argv = parse_options(opts)          
+									
+				return if !argv 
+				raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if argv.length < 1 || argv.length > 2		
+
+				
+				begin
+					log_entries=RequestController.node_backup_log(@env.ui,argv[0],argv[1])
+					
+					
+					
+					if !log_entries.empty?
+						@env.ui.info("--------------------------------------------------------------------------------")
+						@env.ui.info("|        DATE       |          VM NAME       |              STATUS             |")
+						@env.ui.info("--------------------------------------------------------------------------------")
+						
+						log_entries.each do |date,vm,status|						
+							@env.ui.info("|#{date}|#{vm.rjust(13).ljust(23)} |#{status.rjust(20).ljust(33)}|")
+							@env.ui.info("--------------------------------------------------------------------------------")		
+						end
+					end
+				rescue RestClient::ResourceNotFound => e
+					@env.ui.error("Remote Client \"#{argv[0]}\": No Records found ", :new_line => false)
+					if (argv[1])
+						@env.ui.error("for Virtual Machine \"#{argv[1]}\"")
+					else	
+						@env.ui.error("for this node.")
+					end 
+				end
+				
+				
+						
+				0
+			end
+			
+			
+		end
+  end
+end