spectre-core 1.8.3 → 1.11.0

data/lib/spectre/ftp.rb

@@ -1,195 +0,0 @@
-require 'net/ftp'
-require 'net/sftp'
-require 'logger'
-require 'json'
-
-
-module Spectre
-  module FTP
-    @@cfg = {}
-
-    class FTPConnection < DslClass
-      def initialize host, username, password, opts, logger
-        @__logger = logger
-        @__session = nil
-
-        @__host = host
-        @__username = username
-        @__password = password
-        @__opts = opts
-      end
-
-      def connect!
-        return unless @__session == nil or @__session.closed?
-        @__logger.info "Connecting to '#{@__host}' with user '#{@__username}'"
-        @__session = Net::FTP.new(@__host, @__opts)
-        @__session.login @__username, @__password
-      end
-
-      def close
-        return unless @__session and not @__session.closed?
-        @__session.close
-      end
-
-      def can_connect?
-        begin
-          connect!
-          return true
-        rescue
-          return false
-        end
-      end
-
-      def download remotefile, to: File.basename(remotefile)
-        connect!
-        @__logger.info "Downloading '#{@__username}@#{@__host}:#{File.join @__session.pwd, remotefile}' to '#{File.expand_path to}'"
-        @__session.getbinaryfile(remotefile, to)
-      end
-
-      def upload localfile, to: File.basename(localfile)
-        connect!
-        @__logger.info "Uploading '#{File.expand_path localfile}' to '#{@__username}@#{@__host}:#{File.join @__session.pwd, to}'"
-        @__session.putbinaryfile(localfile, to)
-      end
-
-      def list
-        connect!
-        file_list = @__session.list
-        @__logger.info "Listing file in #{@__session.pwd}\n#{file_list}"
-        file_list
-      end
-    end
-
-
-    class SFTPConnection < DslClass
-      def initialize host, username, opts, logger
-        opts[:non_interactive] = true
-
-        @__logger = logger
-        @__session = nil
-        @__host = host
-        @__username = username
-        @__opts = opts
-      end
-
-      def connect!
-        return unless @__session == nil or @__session.closed?
-        @__logger.info "Connecting to '#{@__host}' with user '#{@__username}'"
-        @__session = Net::SFTP.start(@__host, @__username, @__opts)
-        @__session.connect!
-      end
-
-      def close
-        return unless @__session and not @__session.closed?
-        # @__session.close!
-      end
-
-      def can_connect?
-        begin
-          connect!
-          return true
-        rescue
-          return false
-        end
-      end
-
-      def download remotefile, to: File.basename(remotefile)
-        @__logger.info "Downloading '#{@__username}@#{@__host}:#{remotefile}' to '#{File.expand_path to}'"
-        connect!
-        @__session.download!(remotefile, to)
-      end
-
-      def upload localfile, to: File.basename(localfile)
-        @__logger.info "Uploading '#{File.expand_path localfile}' to '#{@__username}@#{@__host}:#{to}'"
-        connect!
-        @__session.upload!(localfile, to)
-      end
-
-      def stat path
-        connect!
-        file_info = @__session.stat! path
-        @__logger.info "Stat '#{path}'\n#{JSON.pretty_generate file_info.attributes}"
-        file_info.attributes
-      end
-
-      def exists path
-        begin
-          file_info = @__session.stat! path
-
-        rescue Net::SFTP::StatusException => e
-          return false if e.description == 'no such file'
-          raise e
-        end
-
-        return true
-      end
-    end
-
-
-    class << self
-      def ftp name, config={}, &block
-        raise "FTP connection '#{name}' not configured" unless @@cfg.key?(name) or config.count > 0
-        cfg = @@cfg[name] || {}
-
-        host = config[:host] || cfg['host'] || name
-        username = config[:username] || cfg['username']
-        password = config[:password] || cfg['password']
-
-        opts = {}
-        opts[:ssl] = config[:ssl]
-        opts[:port] = config[:port] || cfg['port'] || 21
-
-        @@logger.info "Connecting to #{host} with user #{username}"
-
-        ftp_conn = FTPConnection.new(host, username, password, opts, @@logger)
-
-        begin
-          ftp_conn.instance_eval &block
-        ensure
-          ftp_conn.close
-        end
-      end
-
-      def sftp name, config={}, &block
-        raise "FTP connection '#{name}' not configured" unless @@cfg.key?(name) or config.count > 0
-
-        cfg = @@cfg[name] || {}
-
-        host = config[:host] || cfg['host'] || name
-        username = config[:username] || cfg['username']
-        password = config[:password] || cfg['password']
-
-        opts = {}
-        opts[:password] = password
-        opts[:port] = config[:port] || cfg['port'] || 22
-        opts[:keys] = [cfg['key']] if cfg.key? 'key'
-        opts[:passphrase] = cfg['passphrase'] if cfg.key? 'passphrase'
-
-        opts[:auth_methods] = []
-        opts[:auth_methods].push 'publickey' if opts[:keys]
-        opts[:auth_methods].push 'password' if opts[:password]
-
-        sftp_con = SFTPConnection.new(host, username, opts, @@logger)
-
-        begin
-          sftp_con.instance_eval &block
-        ensure
-          sftp_con.close
-        end
-      end
-    end
-
-    Spectre.register do |config|
-      @@logger = ::Logger.new config['log_file'], progname: 'spectre/ftp'
-
-      if config.key? 'ftp'
-
-        config['ftp'].each do |name, cfg|
-          @@cfg[name] = cfg
-        end
-      end
-    end
-
-    Spectre.delegate :ftp, :sftp, to: self
-  end
-end

data/lib/spectre/mysql.rb

@@ -1,97 +0,0 @@
-require 'mysql2'
-
-module Spectre
-  module MySQL
-
-    class MySqlQuery < DslClass
-      def initialize query
-        @__query = query
-      end
-
-      def host hostname
-        @__query['host'] = hostname
-      end
-
-      def username user
-        @__query['username'] = user
-      end
-
-      def password pass
-        @__query['password'] = pass
-      end
-
-      def database name
-        @__query['database'] = name
-      end
-
-      def query statement
-        @__query['query'] = [] if not @__query.key? 'query'
-        @__query['query'].append(statement)
-      end
-    end
-
-    class << self
-      @@mysql_cfg = {}
-      @@result = nil
-      @@last_conn = nil
-
-      def mysql name = nil, &block
-        query = {}
-
-        if name != nil and @@mysql_cfg.key? name
-          query.merge! @@mysql_cfg[name]
-          raise "No `host' set for MySQL client '#{name}'. Check your MySQL config in your environment." if !query['host']
-        elsif name != nil
-          query['host'] = name
-        elsif @@last_conn == nil
-          raise 'No name given and there was no previous MySQL connection to use'
-        end
-
-        MySqlQuery.new(query).instance_eval(&block) if block_given?
-
-        if name != nil
-          @@last_conn = {
-            host: query['host'],
-            username: query['username'],
-            password: query['password'],
-            database: query['database']
-          }
-        end
-
-        @@logger.info "Connecting to database #{query['username']}@#{query['host']}:#{query['database']}"
-
-        client = ::Mysql2::Client.new(**@@last_conn)
-
-        res = []
-
-        query['query'].each do |statement|
-          @@logger.info 'Executing statement "' + statement + '"'
-          res = client.query(statement, cast_booleans: true)
-        end if query['query']
-
-        @@result = res.map { |row| OpenStruct.new row } if res
-
-        client.close
-      end
-
-      def result
-        raise 'No MySQL query has been executed yet' unless @@result
-        @@result
-      end
-    end
-
-    Spectre.register do |config|
-      @@logger = ::Logger.new config['log_file'], progname: 'spectre/mysql'
-
-      if config.key? 'mysql'
-        @@mysql_cfg = {}
-
-        config['mysql'].each do |name, cfg|
-          @@mysql_cfg[name] = cfg
-        end
-      end
-    end
-
-    Spectre.delegate :mysql, :result, to: self
-  end
-end

data/lib/spectre/ssh.rb

@@ -1,149 +0,0 @@
-require 'net/ssh'
-require 'logger'
-
-
-module Spectre
-  module SSH
-    @@cfg = {}
-
-    class SSHConnection < DslClass
-      def initialize host, username, opts, logger
-        opts[:non_interactive] = true
-
-        @__logger = logger
-        @__host = host
-        @__username = username
-        @__opts = opts
-        @__session = nil
-        @__exit_code = nil
-        @__output = ''
-      end
-
-      def file_exists path
-        exec "ls #{path}"
-        exit_code == 0
-      end
-
-      def owner_of path
-        exec "stat -c %U #{path}"
-        output.chomp
-      end
-
-      def connect!
-        return unless @__session == nil or @__session.closed?
-        @__session = Net::SSH.start(@__host, @__username, @__opts)
-      end
-
-      def close
-        return unless @__session and not @__session.closed?
-        @__session.close
-      end
-
-      def can_connect?
-        @__output = nil
-
-        begin
-          connect!
-          @__session.open_channel.close
-          @__output = "successfully connected to #{@__host} with user #{@__username}"
-          @__exit_code = 0
-          return true
-        rescue Exception => e
-          @__logger.error e.message
-          @__output = "unable to connect to #{@__host} with user #{@__username}"
-          @__exit_code = 1
-        end
-
-        return false
-      end
-
-      def exec command
-        connect!
-
-        log_str = "#{@__session.options[:user]}@#{@__session.host} -p #{@__session.options[:port]} #{command}"
-
-        @channel = @__session.open_channel do |channel|
-          channel.exec(command) do |ch, success|
-            abort "could not execute #{command} on #{@__session.host}" unless success
-
-            @__output = ''
-
-            channel.on_data do |ch, data|
-              @__output += data
-            end
-
-            channel.on_extended_data do |ch,type,data|
-              @__output += data
-            end
-
-            channel.on_request('exit-status') do |ch, data|
-              @__exit_code = data.read_long
-            end
-
-            # channel.on_request('exit-signal') do |ch, data|
-            #   exit_code = data.read_long
-            # end
-          end
-
-        end
-
-        @channel.wait
-        @__session.loop
-
-        log_str += "\n" + @__output
-        @__logger.info log_str
-      end
-
-      def output
-        @__output
-      end
-
-      def exit_code
-        @__exit_code
-      end
-    end
-
-
-    class << self
-      def ssh name, config = {}, &block
-        raise "SSH connection '#{name}' not configured" unless @@cfg.key?(name) or config.count > 0
-
-        cfg = @@cfg[name] || {}
-
-        host = cfg['host'] || name
-        username = config[:username] || cfg['username']
-        password = config[:password] || cfg['password']
-
-        opts = {}
-        opts[:password] = password
-        opts[:port] = config[:port] || cfg['port'] || 22
-        opts[:keys] = [cfg['key']] if cfg.key? 'key'
-        opts[:passphrase] = cfg['passphrase'] if cfg.key? 'passphrase'
-
-        opts[:auth_methods] = []
-        opts[:auth_methods].push 'publickey' if opts[:keys]
-        opts[:auth_methods].push 'password' if opts[:password]
-
-        ssh_con = SSHConnection.new(host, username, opts, @@logger)
-
-        begin
-          ssh_con.instance_eval &block
-        ensure
-          ssh_con.close
-        end
-      end
-    end
-
-    Spectre.register do |config|
-      @@logger = ::Logger.new config['log_file'], progname: 'spectre/ssh'
-
-      if config.key? 'ssh'
-        config['ssh'].each do |name, cfg|
-          @@cfg[name] = cfg
-        end
-      end
-    end
-
-    Spectre.delegate :ssh, to: self
-  end
-end