antimony-ssh 0.1.4

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e13880e85fe266cd7cff4b9eeb56076c71d0b0c3
+  data.tar.gz: 8f63105db29c061a17f537ba3c845d669f4b514d
+SHA512:
+  metadata.gz: 9359b2f40479e62d65c1f814cc6e0b4921d60a455582f41cc3ae2ce93cda93e895c09cbcd5c8171422f637a4383cee015ada0d10d4b997ecb43cc35f7643a1a9
+  data.tar.gz: 85a45f33ed26cf25fbbf97bdcedfb748694c3364d8aee6fecdac6e139e00d08119db269e760449ed806857f250760545de689412f31366a12f53bc6ea55f864a

data/bin/sb

@@ -0,0 +1,23 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+lib_path = File.expand_path('../../lib'.freeze, __FILE__)
+($LOAD_PATH.unshift lib_path) unless $LOAD_PATH.include? lib_path
+
+require 'thor'
+require 'logging'
+require 'antimony'
+
+class AntimonyCLI < Thor
+  desc 'formula FILE [--inputs my_input:hello other_input:goodbye]', 'run the specified formula'
+  method_option :inputs, type: :hash
+  def formula(file)
+    inputs = options[:inputs] || {}
+    formula = Antimony::Formula.new(file, inputs)
+    formula.run
+  end
+
+  default_task :formula
+end
+
+AntimonyCLI.start(ARGV)

data/lib/antimony/config.rb

@@ -0,0 +1,6 @@
+# encoding: UTF-8
+module Antimony
+  class Config
+    attr_accessor :show_output, :pretty
+  end
+end

data/lib/antimony/formula.rb

@@ -0,0 +1,63 @@
+# encoding: UTF-8
+module Antimony
+  class Formula
+    LOG_PARENT_PATH = Dir.pwd + '/log'
+    LOG_PATH = "#{LOG_PARENT_PATH}/formulas".freeze
+    FORMULAS_PATH = Dir.pwd + '/formulas'
+
+    attr_accessor :inputs, :outputs, :formula_log
+
+    def initialize(name, inputs = {})
+      ($LOAD_PATH.unshift FORMULAS_PATH) unless $LOAD_PATH.include? FORMULAS_PATH
+
+      @inputs = indifferent_hash(inputs)
+      @outputs = {}
+
+      init_log(name)
+
+      helper_path = "#{FORMULAS_PATH}/formula_helper.rb"
+
+      eval File.read(helper_path) if File.exist?(helper_path) # rubocop:disable Lint/Eval
+
+      @formula = File.read("#{FORMULAS_PATH}/#{name}.rb")
+    end
+
+    def run
+      eval(@formula) # rubocop:disable Lint/Eval
+    end
+
+    def session(host, username, password)
+      @connection = Antimony::Session.new(host, username, password)
+      yield if block_given?
+      @log.info @connection.session_log
+      @connection.close
+      @connection = nil
+    end
+
+    private
+
+    def init_log(name)
+      Dir.mkdir LOG_PARENT_PATH unless Dir.exist? LOG_PARENT_PATH
+      Dir.mkdir LOG_PATH unless Dir.exist? LOG_PATH
+      @log = Logging.logger[name]
+      layout = Logging.layouts.pattern(pattern: "%m\n")
+      @log.add_appenders(Logging.appenders.file("#{LOG_PATH}/#{name}.log",           truncate: true,
+                                                                                     layout: layout))
+    end
+
+    def method_missing(name, *args, &block)
+      raise 'No active connection!' unless @connection
+      @connection.send(name, *args, &block)
+    end
+
+    def indifferent_hash(hash)
+      {}.tap do |new_hash|
+        hash.each do |key, value|
+          value = indifferent_hash(value) if value.is_a?(Hash)
+          new_hash[key.to_sym] = value
+          new_hash[key.to_s] = value
+        end
+      end
+    end
+  end
+end

data/lib/antimony/session.rb

@@ -0,0 +1,132 @@
+# encoding: UTF-8
+module Antimony
+  class Session
+    def initialize(host, username, password)
+      @cursor_position = 0
+      @screen_text = blank_screen
+      @session_log = []
+
+      puts "Establishing SSH tunnel ..."
+      @gateway = Net::SSH::Gateway.new(host, username, :password => password)
+      tunnel_port = @gateway.open(host, 23)
+
+      puts "Connecting to #{host} using local port #{tunnel_port} ..."
+      @connection = Net::Telnet.new(
+        'Host' => 'localhost',
+        'Port' => tunnel_port,
+        'Timeout' => 10
+      )
+
+      update_screen
+    end
+
+    def close
+      @connection.close
+      @gateway.shutdown!
+    end
+
+    def send_keys(keys, count = 1, text = true)
+      count.times { @connection.print keys }
+      update_screen
+      add_text(keys) if text
+      log_screen
+      print_screen if Antimony.configuration.show_output
+    end
+
+    def value_at(row, column, length)
+      start_index = ((row - 1) * 80) + (column - 1)
+      end_index = start_index + length - 1
+      @screen_text[start_index..end_index].join
+    end
+
+    def screen_text
+      @screen_text.join
+    end
+
+    def printable_screen_text
+      @screen_text.join.scan(LINE)
+    end
+
+    def print_screen
+      puts SCREEN_SEPARATOR
+      puts printable_screen_text
+      puts SCREEN_SEPARATOR
+    end
+
+    def session_log
+      txt = EMPTY.clone
+      @session_log.each_with_index do |entry, i|
+        txt += LOG_SEPARATOR + ENTER
+        txt += "#{i}: #{ENTER}"
+        txt += entry + ENTER
+      end
+      txt += LOG_SEPARATOR
+      txt
+    end
+
+    def log(text)
+      @session_log.push(text)
+    end
+
+    def log_screen
+      log(printable_screen_text.join("\n"))
+    end
+
+    private
+
+    def update_screen
+      @screen_data = receive_data
+      @screen_text = blank_screen if @screen_data.include?('[2J')
+      parse_ansi
+    end
+
+    def add_text(text)
+      text.chars.each do |char|
+        @screen_text[@cursor_position] = char
+        @cursor_position += 1
+      end
+    end
+
+    def receive_data
+      whole = []
+      whole.push(@chunk) while chunk
+      whole.join
+    end
+
+    def chunk
+      @chunk = @connection.waitfor RECEIVE_OPTS
+    rescue # rubocop:disable Lint/HandleExceptions
+    end
+
+    def cursor_position(esc)
+      esc_code = /\e\[/
+      pos = esc.gsub(esc_code, EMPTY).gsub(H, EMPTY).split(SEMICOLON)
+      row_index = pos[0].to_i - 1
+      col_index = pos[1].to_i - 1
+      (row_index * 80) + col_index
+    end
+
+    def parse_ansi
+      ansi = @screen_data.scan(ANSI_REGEX).map do |e|
+        {
+          value: e[0],
+          type: e[0].include?(ESCAPE) ? :esc : :chr
+        }
+      end
+
+      ansi.each do |e|
+        if (e[:type] == :esc) && (e[:value].end_with? H)
+          @cursor_position = cursor_position(e[:value])
+        elsif e[:type] == :chr
+          @screen_text[@cursor_position] = e[:value]
+          @cursor_position += 1
+        end
+      end
+      @screen_text
+    end
+
+    def blank_screen
+      (1..1920).to_a.map { |_i| SPACE }
+    end
+  end
+end

data/lib/antimony.rb

@@ -0,0 +1,90 @@
+# encoding: UTF-8
+# encoding 'ASCII-8bit'
+lib_path = File.expand_path('../antimony', __FILE__)
+($LOAD_PATH.unshift lib_path) unless $LOAD_PATH.include? lib_path
+
+require 'bundler'
+Bundler.require(:default)
+
+# Misc
+require 'net/telnet'
+require 'net/ssh/gateway'
+require 'logging'
+
+# Gem
+require "#{lib_path}/config.rb"
+require "#{lib_path}/formula.rb"
+require "#{lib_path}/session.rb"
+
+module Antimony
+  class << self
+    attr_writer :configuration
+  end
+
+  def self.configuration
+    @configuration ||= Antimony::Config.new.tap { |config| config.show_output = false }
+  end
+
+  def self.configure
+    yield configuration if block_given?
+  end
+
+  # Constants
+  ANSI_REGEX = /(\e\[.{1,2};?.{0,2}m|\e\[.{1}J|\e\[.{1,2}A|\e\[.{1,2}B|\e\[.{1,2}C|\e\[.{1,2}D|\e\[K|\e\[.{1,2};.{1,2}H|\e\[.{3}|.)/ # rubocop:disable Metrics/LineLength
+  SCREEN_SEPARATOR = '################################################################################'.freeze
+  LOG_SEPARATOR    = '================================================================================'.freeze
+
+  SPACE = ' '.freeze
+  EMPTY = ''.freeze
+  RECEIVE_OPTS = { 'Match' => /.{20}/, 'Timeout' => 1 }.freeze
+
+  ENTER = "\n".freeze
+  TAB = "\t".freeze
+  ESCAPE = "\e".freeze
+
+  H = 'H'.freeze
+
+  SEMICOLON = ';'.freeze
+
+  LINE = /.{80}/
+
+  class Session
+    KEYBOARD_METHODS = {
+      f1: "\x1B\x31".freeze,
+      f2: "\x1B\x32".freeze,
+      f3: "\x1B\x33".freeze,
+      f4: "\x1B\x34".freeze,
+      f5: "\x1B\x35".freeze,
+      f6: "\x1B\x36".freeze,
+      f7: "\x1B\x37".freeze,
+      f8: "\x1B\x38".freeze,
+      f9: "\x1B\x39".freeze,
+      f10: "\x1B\x30".freeze,
+      f11: "\x1B\x2D".freeze,
+      f12: "\x1B\x3D".freeze,
+      f13: "\x1B\x21".freeze,
+      f14: "\x1B\x40".freeze,
+      f15: "\x1B\x23".freeze,
+      f16: "\x1B\x24".freeze,
+      f17: "\x1B\x25".freeze,
+      f18: "\x1B\x5E".freeze,
+      f19: "\x1B\x26".freeze,
+      f20: "\x1B\x2A".freeze,
+      f21: "\x1B\x28".freeze,
+      f22: "\x1B\x29".freeze,
+      f23: "\x1B\x5F".freeze,
+      f24: "\x1B\x2B".freeze,
+      enter: "\n".freeze,
+      tab: "\t".freeze,
+      backspace: "\177".freeze,
+      arrow_up: "\e[A".freeze,
+      arrow_down: "\e[B".freeze,
+      arrow_right: "\e[C".freeze,
+      arrow_left: "\e[D".freeze
+    }.freeze
+
+    KEYBOARD_METHODS.each do |key, val|
+      define_method(key) { |count = 1| send_keys val, count, false }
+    end
+  end
+end

metadata

@@ -0,0 +1,166 @@
+--- !ruby/object:Gem::Specification
+name: antimony-ssh
+version: !ruby/object:Gem::Version
+  version: 0.1.4
+platform: ruby
+authors:
+- Andrew Terris
+- Adrienne Hisbrook
+- Lesley Dennison
+- Ryan Rosenblum
+- Vinh Luong
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2020-09-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  name: logging
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: thor
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 4.2.0
+  name: net-ssh
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 4.2.0
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.3.0
+  name: net-ssh-gateway
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.3.0
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.3'
+  name: rake
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.3'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  name: rspec
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.10.1
+  name: pry
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.10.1
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.37.0
+  name: rubocop
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.37.0
+description: DSL for writing scripts to drive TN5250 green-screen applications using
+  Telnet via SSH tunneling
+email: ''
+executables:
+- sb
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/sb
+- lib/antimony.rb
+- lib/antimony/config.rb
+- lib/antimony/formula.rb
+- lib/antimony/session.rb
+homepage: https://github.com/manheim/antimony
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project:
+rubygems_version: 2.6.14.1
+signing_key:
+specification_version: 4
+summary: TN5250/AS400 automation
+test_files: []