con_ssh 0.0.1

data/.ruby-version

@@ -0,0 +1 @@
+2.3.0

data/README.md

@@ -0,0 +1,3 @@
+con_ssh
+=================
+

data/Rakefile

@@ -0,0 +1,8 @@
+require 'rake/testtask'
+
+Rake::TestTask.new do |t|
+  t.libs << 'test'
+end
+
+desc "Run tests"
+task default: :test

data/bin/con

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+
+require 'ostruct'
+require 'con_ssh'
+
+SSHCon.new.run *ARGV 

data/con_ssh.gemspec

@@ -0,0 +1,18 @@
+Gem::Specification.new do |s|
+  s.name          = "con_ssh"
+  s.version       = "0.0.1"
+  s.authors       = ["Tom Lobato"]
+  s.email         = "lobato@bettercall.io"
+  # s.homepage      = "http://sys-watchdog.bettercall.io/"
+  s.summary       = "SSH cli wrapper."
+  s.description   = "#{s.summary}."
+  s.licenses      = ["MIT"]
+  s.platform      = Gem::Platform::RUBY
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- test/*`.split("\n")
+  s.require_paths = ["lib"]
+  s.executables   = %w(con)
+  s.required_ruby_version = '>= 2.3.0'
+end
+

data/gempush

@@ -0,0 +1,5 @@
+#!/bin/bash
+
+rm con_ssh-*.gem
+gem build con_ssh.gemspec 
+gem push con_ssh-*.gem

data/lib/con_ssh/core_extensions.rb

@@ -0,0 +1,24 @@
+
+class String
+  def strip_text
+    split("\n")
+    .map{|l| l.trip}
+    .join("\n")
+  end
+end
+
+class Exception
+  def desc
+    "#{ message } #{ backtrace&.join "\n" }"
+  end
+end
+
+def log_ex e
+  STDERR.puts e.desc
+end
+
+def match_root_or_current_user stat
+  (stat.uid == 0           and stat.gid == 0) ||
+  (stat.uid == Process.uid and stat.gid == Process.gid)
+end
+

data/lib/con_ssh.rb

@@ -0,0 +1,173 @@
+
+require 'ostruct'
+require 'con_ssh/core_extensions'
+
+class SSHCon
+  CONF_PATH = File.expand_path '~/.con'
+  CONF_LINE_FIELDS = %w(shortcut conn_desc host user port knock unknock)
+  DEFAULT_PORT = "22"
+
+  def initialize
+    @conn_confs = {}
+  end
+
+  def run *args
+    # setup
+    if args[0] == 'setup'
+      warn "Ignoring arguments after #{args[0]}." if args.length > 1
+      install_sample_conf
+
+    # knock / unknock
+    elsif ['knock', 'unknock'].include? args[1] 
+      warn "Ignoring arguments after #{args[1]}." if args.length > 2
+      parse_conf
+      conn_conf = @conn_confs[args[0]]
+      unless conn_confs
+        warn "Shorcut #{args[0]} not found. See #{CONF_PATH}."
+        exit 1
+      end
+      ports = conn_conf[args[1]]
+      if ports.nil? || ports.empty?
+        warn "#{conn_conf.conn_desc} has no #{args[1]} configuration."
+        exit 1
+      end
+      knock conn_conf.host, ports
+
+    # connect
+    elsif conn_conf = @conn_confs[args[0]]
+      warn "Ignoring arguments after #{args[0]}." if args.length > 1      
+      parse_conf
+      ssh conn_conf
+
+    # help
+    elsif ['-h'].include? args[0]
+      warn "Ignoring arguments after #{args[0]}." if args.length > 1      
+      print_help
+
+    # fail
+    else
+      warn "Invalid arguments."
+      print_help
+      exit 1
+    end
+  end
+
+  private
+
+  def ssh c
+    port = (c.port && c.port != DEFAULT_PORT) ? "-p #{ c.port } " : ''
+    user = c.user ? "#{ c.user }@" : ''
+    knock c.host, c.knock if c.knock
+    system "ssh #{port}#{user}#{c.host}"
+    knock c.host, c.unknock if c.unknock
+  end
+
+  def knock host, ports
+    unless has_knock
+        msg = "Skipping knock! 
+             The connection configuration has ports knock set but you don`t have 'knock' installed on your system. 
+             Please install: 'apt-get install knockd', 'brew install knock', ...".strip_text
+        warn msg
+      return
+      end
+    run_cmd "knock #{ host } #{ ports.join ' ' }", false
+  end
+
+
+  # Conf
+
+  def parse_conf
+    unless File.exists? CONF_PATH
+      raise "#{CONF_PATH} not found. Install a sample with 'con setup'."
+    end
+    File.open(CONF_PATH).readlines.each do |line|
+      add_connection line.strip     
+    end
+  end
+
+  def add_connection line
+    return if skip_line? line
+    
+    values = line.split /\s+/
+    
+    conn_conf = OpenStruct.new
+    CONF_LINE_FIELDS.each_with_index do |field, idx|
+      conn_conf[field.to_sym] = values[idx]
+    end
+
+    unless valid? conn_conf
+      warn "WARNING: Invalid line, skipping: '#{line}'"
+      return
+    end
+
+    adjust_conn_conf conn_conf
+
+    if @conn_confs[conn_conf.shortcut]
+      warn "WARNING: Shorcut '#{conn_conf.shortcut} is duplicated, skipping line: '#{line}'"
+      return
+    end
+
+    @conn_confs[conn_conf.shortcut] = conn_conf
+  end
+
+  def valid? c
+    c.shortcut &&
+    c.conn_desc &&
+    c.host
+  end
+
+  def adjust_conn_conf c
+    c.knock  = c.knock.split  ',' if c.knock 
+    c.unknock = c.unknock.split ',' if c.unknock 
+
+    if c.host =~ /^(.*?)@(.*?)$/
+      c.host = $2
+      c.user ||= $1
+    end
+  end
+
+  def install_sample_conf
+    if File.exists? CONF_PATH
+      warn "#{CONF_PATH} already exists. Skipping install."
+      return false
+    end
+    sample_conf = "# #{ CONF_LINE_FIELDS.join ' ' }
+    # *connection_desc must have no spaces.
+    # *user, port, knock, unknock are optionals.
+
+    s1  server_1  123.234.35.456
+    s1r server_1r root@123.234.35.456
+    s3  server_3  123.234.35.456  username  2222  knock unknock"
+    File.write CONF_PATH, sample_conf.strip_text
+    true
+  end
+
+
+  # Util
+  
+  def warn desc
+    $stderr.puts desc
+  end
+
+  def skip_line? line
+    line =~ /^\s*[$#]/
+  end
+
+  def has_knock
+    !`which knock`.strip.empty?
+  end
+
+  def print_help
+    puts "Usage: con <shortcut>|setup [knock|unknock]".strip_text
+  end
+
+  def run_cmd cmd, print = true
+    puts "run: #{cmd}" if print
+      output = `#{cmd} 2>&1`
+      exit_status = $?.to_i
+      unless exit_status == 0
+      warn "Non-zero output: #{output}"
+    end
+    [exit_status, output]
+  end
+end

data/test/con_ssh_test.rb

@@ -0,0 +1,8 @@
+require 'test/unit'
+require 'ssh_con'
+
+class SSHConTest < Test::Unit::TestCase
+  def test_run
+    SSHCon.new
+  end
+end

metadata

@@ -0,0 +1,58 @@
+--- !ruby/object:Gem::Specification
+name: con_ssh
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Tom Lobato
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-06-23 00:00:00.000000000 Z
+dependencies: []
+description: SSH cli wrapper..
+email: lobato@bettercall.io
+executables:
+- con
+extensions: []
+extra_rdoc_files: []
+files:
+- ".codeclimate.yml"
+- ".gitignore"
+- ".rubocop-custom.yml"
+- ".ruby-version"
+- README.md
+- Rakefile
+- TODO
+- bin/con
+- con_ssh.gemspec
+- gempush
+- lib/con_ssh.rb
+- lib/con_ssh/core_extensions.rb
+- test/con_ssh_test.rb
+homepage: 
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.3.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: SSH cli wrapper.
+test_files:
+- test/con_ssh_test.rb