#require_relative "../manager" require 'optparse' require_relative 'db' require_relative 'client' require_relative 'version' module SSH module Manager class Client attr_accessor :options CODES = %w[iso-2022-jp shift_jis euc-jp utf8 binary] CODE_ALIASES = { "jis" => "iso-2022-jp", "sjis" => "shift_jis" } def initialize(argv) @options = {} @argv = argv extract_options end def execute! cli = SSH::Manager::Cli # TODO id.to_i is not good enough. we want to support hostnames too # Checking and casting in the methods itself could solve the problem # futhermore this could reside in a separate method if @options[:add] puts 'Adding ..' cli.new(@options).add_connection elsif @options[:connect] puts 'Connecting ..' cli.new(@options).connect_to(@options[:connect]) elsif @options[:info] puts 'Printing info ..' cli.new(@options).show_info(@options[:info]) elsif @options[:execute] puts 'Executing command..' cli.new(@options).execute_command(@options[:execute]) elsif @options[:transfer_file] puts 'Transfering file..' cli.new(@options).transfer_file(@options[:transfer_file].to_i, @argv[2], @argv[3]) elsif @options[:ping] puts 'Pinging..' cli.new(@options).ping(@options[:ping]) elsif @options[:delete] puts 'Deleting ..' cli.new(@options).delete(@options[:delete]) elsif @options[:list] puts 'Listing ..' cli.new(@options).list_all elsif @options[:upgrade] puts 'Checking for new updates ..' cli.new(@options).update_available elsif @options[:update] puts 'Updating ..' cli.new(@options).update(@options[:update]) elsif @options[:multi] puts 'Connecting to multiple ips..' cli.new(@options).multiple_connection(@options[:multi]) elsif @options[:transfer_key] puts 'Transfering key..' cli.new(@options).transfer_key(@options[:transfer_key]) elsif @options[:encoding] puts 'coding key..' cli.new(@options).test(@options[:encoding].to_i) elsif @options[:settings] puts 'Settings' cli.new(@options).settings elsif @options[:search] puts 'Searching ..' cli.new(@options).search_for(@options[:search]) else if @argv.count == 1 cli.new(@options).connect_to(@argv.first.split(',')) if @argv != [] else cli.new(@options).connect_to(@argv) if @argv != [] end puts @optparse if @argv ==[] exit end end def extract_options begin @optparse = OptionParser.new do |opts| opts.banner = "Usage: sshm [options] ..." @options[:add] = false opts.on( '-a', '--add', 'add a new connection' ) do @options[:add] = true end @options[:transfer_key] = false opts.on( '-t', '--transferkey id,id,id', Array, 'transfer key to ' ) do |opt| @options[:transfer_key] = opt end @options[:transfer_file] = false opts.on( '-r', '--transferfile filename', 'file or dir / connection_ID / dest_path(default is /home/user/)' ) do |opt| @options[:transfer_file] = opt end @options[:encoding] = false code_list = (CODE_ALIASES.keys + CODES).join(',') opts.on("--code CODE", CODES, CODE_ALIASES, "Select encoding"," (#{code_list})") do |encoding| @options[:encoding] = encoding end @options[:connect] = false opts.on( '-c', '--connect id1, id2, id3', Array, 'connect to ' ) do |opt| @options[:connect] = opt end @options[:info] = false opts.on( '-i', '--info id1 id2 id3',Array, 'info about to ' ) do |opt| @options[:info] = opt end @options[:execute] = false opts.on( '-x', '--exec id1 id2 id3',Array, 'exec command on ' ) do |opt| @options[:execute] = opt end @options[:ping] = false opts.on( '-p', '--ping id1, id2, i3', Array, 'test connection of ' ) do |opt| @options[:ping] = opt end @options[:delete] = false opts.on( '-d', '--delete id1 id2 id3',Array, 'delete connection ' ) do |opt| @options[:delete] = opt end @options[:update] = false opts.on( '-u', '--update id,id,id',Array ,'update connection ' ) do |opt| @options[:update] = opt end @options[:search] = false opts.on( '-s', '--search string', 'search connection for given criteria' ) do |opt| @options[:search] = opt end @options[:multi] = false opts.on( '-m', '--multi searchterm', 'connect to multiple ips with given criteria' ) do |opt| @options[:multi] = opt end @options[:list] = false opts.on( '-l', '--list', 'list all connections' ) do @options[:list] = true end @options[:upgrade] = false opts.on( '-g', '--upgrade', 'checks for upgrade' ) do @options[:upgrade] = true end @options[:settings] = false opts.on( '--settings', 'update settings' ) do @options[:settings] = true end opts.on( '-h', '--help', 'display this screen' ) do puts opts exit end opts.on( '-v', '--version', 'print programs version' ) do puts SSH::Manager::VERSION exit end end @optparse.parse(@argv) rescue => e catch "That didnt work: #{e.to_s}" end end end end end