ccsh 0.0.3

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e18fe57923e83f6f801b3c17cffab2d0928df142
+  data.tar.gz: 8fa3fb86f5626fa09a4f657379626bcc1e4c4d72
+SHA512:
+  metadata.gz: a5b6525d2f5d690e88d2cb9322a42400d5909e625d797d24b7a7c64bf31607c7026bc34de478936bae625ec89dee37b936b8d1075e35b17ab21b5f8c9f72fede
+  data.tar.gz: f40ee7dcf33c3071a3ff83c5b105ac66ae3b07cfc2d79109aecc96d864331d4dd389daa932c74faecb90b5a21440989b603f6e004cb169c44b4c0a779c947b10

data/bin/ccsh

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'CCSH'
+
+CCSH.execute!

data/lib/ccsh.rb

@@ -0,0 +1,78 @@
+require 'rubygems' unless defined?(Gem)
+
+module CCSH
+    require 'ccsh/version.rb'
+    require 'ccsh/utils.rb'
+    require 'ccsh/options.rb'
+    require 'ccsh/host'
+    require 'ccsh/hosts.rb'
+    require 'ccsh/ssh.rb'
+
+    def self.execute!
+        options = CCSH::Options.parse_options ARGV
+
+        ENV['CCSH_DEBUG']   = "true" if options[:debug]
+        ENV['CCSH_VERBOSE'] = "true" if options[:verbose]
+
+        raise "You must to specified a hostname or group" if options[:targets].empty?
+
+        filename = options[:hosts]
+        targets = options[:targets]
+
+        self.start_cli CCSH::Hosts.new.parser!(filename).filter_by(targets)
+    end
+
+    def self.with_info
+        cmd_start = Time.now
+        cmd = yield
+        cmd_end = Time.now
+
+        info = {
+            'rc'   => cmd.return_code,
+            'time' => cmd_end - cmd_start,
+        }.inspect
+
+        puts ">>> #{cmd.hostname} #{info}"
+        STDERR.puts cmd.stderr if cmd.stderr != nil && cmd.stderr != ''
+        puts cmd.stdout if cmd.stdout != nil && cmd.stdout != ''
+        puts
+    end
+
+    def self.start_cli(hosts)
+        quit = false
+        loop do
+            begin
+                CCSH::Utils.handle_signals
+                printf "ccsh> "
+                command = STDIN.gets.chomp
+
+                if command != ''
+                    CCSH::Utils.exit_console 0 if command == 'quit'
+                    CCSH::Utils.exit_console 0 if command == 'exit'
+
+                    if command == 'clear'
+                        CCSH::Utils.clear_console
+                    elsif command == 'reset'
+                        CCSH::Utils.reset_console
+                    else
+                        threads = []
+                        hosts.each do |host|
+                            threads << Thread.new do
+
+                                with_info do
+                                    host.run command
+                                end
+                            end
+                        end
+
+                        threads.each(&:join)
+                    end
+                end
+            rescue Exception => exception
+                STDERR.puts exception.message
+                STDERR.puts
+                raise exception
+            end
+        end
+    end
+end

data/lib/ccsh/host.rb

@@ -0,0 +1,45 @@
+module CCSH
+    class Host        
+        attr_accessor :name
+        attr_accessor :port
+        attr_accessor :user
+        attr_accessor :groups
+        attr_accessor :hostname
+        attr_accessor :password
+        attr_accessor :private_key
+        attr_accessor :ssh_options
+        attr_accessor :timeout
+
+        def initialize
+            @name = ''
+            @port = 22
+            @user = 'root'
+            @groups = ''
+            @hostname = '127.0.0.1'
+            @password = nil
+            @private_key = nil
+            @ssh_options = []
+        end
+
+        def run(command)
+            CCSH::Utils.verbose("Running command '#{@command}' on #{@hostname}")
+            command = CCSH::SSH.start do |ssh|
+                ssh.user = @user
+                ssh.port = @port
+                ssh.options = @ssh_options
+                ssh.password = @password
+                ssh.private_key = @private_key
+                ssh.options = @ssh_options
+
+                ssh.command = command
+                ssh.hostname = @hostname || @name
+            end
+
+            return command
+        end
+
+        def contain?(target)
+            return @groups.include? target
+        end
+    end
+end

data/lib/ccsh/hosts.rb

@@ -0,0 +1,66 @@
+module CCSH
+    require 'yaml'
+    require 'pathname'
+
+    require 'ccsh/host'
+
+    class Hosts
+        attr_accessor :defaults
+        attr_accessor :hosts
+
+        def initialize
+            # TODO
+        end
+
+        def parser!(filename)
+            begin
+                file = Pathname.new(filename)
+
+                raise "The file does not exists: '#{file.realpath}'" if not file.exist?
+                raise "The host file is not regular file: '#{file.realpath}'" if not file.file?
+
+                hostfile = YAML.load_file(file.realpath)
+
+                @hosts = []
+                @defaults = CCSH::Utils.merge_defaults(hostfile['defaults'])
+
+                hostfile['hosts'].each do |h|
+                    host = Host.new
+
+                    host.name = CCSH::Utils.get_options('name', h, @defaults['name'])
+                    host.user = CCSH::Utils.get_options('user', h, @defaults['user'])
+                    host.port = CCSH::Utils.get_options('port', h, @defaults['port'])
+
+                    host.hostname = CCSH::Utils.get_options('hostname', h, @defaults['hostname'])
+                    host.password = CCSH::Utils.get_options('password', h, @defaults['password'])
+                    host.private_key = CCSH::Utils.get_options('private_key', h, @defaults['private_key'])
+                    host.ssh_options = CCSH::Utils.get_options('ssh_options', h, @defaults['ssh_options'])
+                    host.timeout = CCSH::Utils.get_options('timeout', h, @defaults['timeout'])
+
+                    host.groups = h['groups']
+
+                    @hosts << host
+                end
+            rescue Exception => e
+                puts e.message
+                puts e.backtrace
+                puts
+            end
+
+            return self
+        end
+
+        def filter_by(targets)
+            return @hosts if targets.include? 'all'
+
+            hosts = []
+            @hosts.each do |host|
+                targets.each do |target|
+                    hosts << host if host.groups.include? target
+                end
+            end
+
+            return hosts
+        end
+    end
+end

data/lib/ccsh/options.rb

@@ -0,0 +1,57 @@
+module CCSH
+    module Options
+        require 'ostruct'
+        require 'optparse'
+
+        def self.parse_options(args)
+            options = OpenStruct.new
+
+            # default values
+            user_home = File.expand_path('~')
+
+            options.config  = "#{user_home}/.ccsh/config.yaml"
+            options.hosts   = "#{user_home}/.ccsh/hosts.yaml"
+            options.output  = ""
+            options.debug   = false
+            options.verbose = false
+            options.check   = false
+
+            # open parser
+            opt_parser = OptionParser.new do |opts|
+                opts.banner = "Usage: ccsh [options] GROUP1 GROUP2 ... "
+                opts.separator ""
+                opts.separator "Options: "
+
+                opts.on("-c", "--config CONFIG_FILE", "Configuration file (default: ~/.ccsh/config.yaml)") do |cfg|
+                    options.config = cfg
+                end
+
+                opts.on("-d", "--[no-]debug", "Run Debug mode") do |d|
+                    options.debug = d
+                end
+
+                opts.on("-h", "--hosts HOST_FILE", "Specified hosts file") do |h|
+                    options.hosts = h
+                end
+
+                opts.on("-k", "--[no-]check", "Check host connection before continuing") do |k|
+                    options.check = k
+                end
+
+                opts.on("-v", "--[no-]verbose", "Run verbosely") do |v|
+                    options.verbose = v
+                end
+
+                opts.on("--version", "Display version") do |v|
+                    build = "build #{CCSH::BUILD_NUMBER}" if CCSH::BUILD_NUMBER != nil
+                    puts "CCSH version #{CCSH::VERSION} #{build}"
+                    exit
+                end
+            end
+            opt_parser.parse!(args)
+
+            options.targets = args
+            return options
+        end
+    end
+end

data/lib/ccsh/ssh.rb

@@ -0,0 +1,78 @@
+module CCSH
+    module SSH
+        require 'net/ssh'
+
+        def self.start
+            cmd = RemoteCommand.new
+            yield cmd
+
+            raise 'Could not parser the remote command' if cmd.command == nil
+            raise 'Could not parser the hostname' if cmd.hostname == nil
+
+            return cmd.execute!
+        end
+
+        class RemoteCommand
+            attr_accessor :command
+            attr_accessor :hostname
+            attr_accessor :options
+            attr_accessor :user
+            attr_accessor :port
+            attr_accessor :password
+            attr_accessor :private_key
+            attr_accessor :options
+
+            attr_accessor :stdout
+            attr_accessor :stderr
+            attr_accessor :return_code
+            attr_accessor :return_signal
+
+            def initialize
+                @command  = nil
+                @hostname = nil
+                @options  = []
+                @stdout = ''
+                @stderr = ''
+            end
+
+            def execute!
+                @options = {
+                    :password => @password,
+                    :keys     => [@private_key],
+                    :port     => @port,
+                    :host_key => @options['host_key'],
+                    :timeout  => @options['timeout'],
+                }.select {|key,value| value != nil}
+
+                raise "something" unless CCSH::Utils.valid_ssh(@options)
+
+                Net::SSH.start(@hostname, @user, @options) do |ssh|
+                    ssh.open_channel do |ch|
+                        ch.exec @command do |ch, success|
+                            raise "Could execute command #{command} on #{host}" unless success
+                        end
+
+                        ch.on_data do |c, data|
+                            @stdout << data
+                        end
+
+                        ch.on_extended_data do |c, type, data|
+                            @stderr << data
+                        end
+
+                        ch.on_request("exit-status") do |c,data|
+                            @return_code = data.read_long
+                        end
+                    
+                        ch.on_request("exit-signal") do |c, data|
+                            @return_signal = data.read_long
+                        end
+
+                    end.wait    
+                end
+
+                return self
+            end
+        end
+    end
+end

data/lib/ccsh/utils.rb

@@ -0,0 +1,70 @@
+module CCSH
+    module Utils
+        def self.merge_defaults(defaults)
+            defaultsValues = {
+                'user'        => 'root',
+                'port'        => '22',
+                'private_key' => '~/.ssh/id_rsa',
+            }
+            defaultsValues.merge!(defaults) if defaults != nil
+
+            ssh_options = {
+                'timeout' => 720,
+                'ssh-rsa' => 'ssh-rsa',
+            }
+
+            if defaults['ssh_options'] != nil
+                defaultsValues['ssh_options'] = ssh_options.merge!(defaults['ssh_options'])
+            else
+                defaultsValues['ssh_options'] = ssh_options
+            end
+
+            return defaultsValues
+        end
+
+        def self.get_options(item, object, default)
+            if object.key? item
+                return object[item]
+            end
+
+            return default
+        end
+
+        def self.valid_ssh(options)
+            return true
+        end
+
+        def self.debug(msg)
+            puts "DEBUG", msg if ENV['CCSH_DEBUG'] == "true"
+        end
+
+        def self.verbose(msg)
+            puts msg if ENV['CCSH_VERBOSE'] == "true"
+        end
+
+        def self.handle_signals
+            Signal.trap('INT') do
+                self.exit_console 0
+            end
+
+            Signal.trap('TERM') do
+                self.exit_console 0
+            end
+        end
+
+        def self.exit_console(code, msg = nil)
+            puts msg if msg != nil
+
+            puts "\nBye..."
+            exit code || 0
+        end
+
+        def self.clear_console
+            printf "\e[H\e[2J"
+        end
+
+        def self.clear_console
+            printf("\033c");
+        end
+    end
+end

data/lib/ccsh/version.rb

@@ -0,0 +1,8 @@
+module CCSH
+
+    # ccsh relased version
+    VERSION   = "0.0.3"
+
+    # the git commit tag number that generate this build
+    BUILD_NUMBER = "dev-#{VERSION}"
+end

metadata

@@ -0,0 +1,109 @@
+--- !ruby/object:Gem::Specification
+name: ccsh
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+platform: ruby
+authors:
+- Rafael Silva <raffs>
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-11-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.2'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: Interactive shell console, connected into multiple clients
+email:
+- rafaeloliveira.cs@gmail.com
+executables:
+- ccsh
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/ccsh
+- lib/ccsh.rb
+- lib/ccsh/host.rb
+- lib/ccsh/hosts.rb
+- lib/ccsh/options.rb
+- lib/ccsh/ssh.rb
+- lib/ccsh/utils.rb
+- lib/ccsh/version.rb
+homepage: https://github.com/raffs/ccsh
+licenses:
+- Apache-2.0
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - "~>"
+    - !ruby/object:Gem::Version
+      version: '2.0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.14
+signing_key: 
+specification_version: 4
+summary: Interactive multiple client shell command
+test_files: []