xlogin 0.16.2 → 0.16.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 342f00cdb5f923c597cf7f40c1d4cf35c8cb7ecb596caa8457c05b09c9b7bd2d
-  data.tar.gz: 4a7f6f3041a53b942ea434a8417b6b150397fed622e1a2f7a3b7dc0148f87865
+  metadata.gz: a8d97fd12ca6f889f47992398783c38f3ebf172823c941696f3c012804440bb2
+  data.tar.gz: 80111e038a1302cbcae48b528d74c759021f9820cbb34d5e307ef5cdb19065f2
 SHA512:
-  metadata.gz: 7bb500f47f1b95b5f29c3f10083df805a1dd09cb23ce3696719a8bcffe00812cd947b3001743085663e698628c1d7ed78aa1c2db9c77e434791ad2bf32f0acb9
-  data.tar.gz: abeccd10d1efca609233477172d1926bd3003f05ac4faec3d81b8750c096f7745f0c886fe3801143a52d09afabfcafecbd5c253480ec4c899e0b60bf1d5f7236
+  metadata.gz: 7cf57105505b9611989006139c592a6a36577eefdd619b582929d88568f3f2d9baf86d0896a6b2a407c7f411bb1d0e2c517b06f243d49140dafdf8d388c242f6
+  data.tar.gz: 6fca3d40be5b12a7cfcebd8e1d19a78f980a5991302b536f0b90de0e3b07cbfaebca8782e2335c817e117a1aeb8b445eee2a95b2c68985e9877c0dbba62ce4a9

data/lib/xlogin.rb

@@ -30,22 +30,16 @@ module Xlogin
     end
 
     def get(args, **opts, &block)
-      session = case args
-                when Hash   then factory.build(**args.merge(**opts))
-                when String then factory.build_from_hostname(args, **opts)
-                else
-                  raise Xlogin::Error.new("Invalid argument: '#{args}'")
-                end
-
-      return session unless block
-      begin block.call(session) ensure session.close end
+      case args
+      when Hash   then factory.build(**args.merge(**opts), &block)
+      when String then factory.build_from_hostname(args, **opts, &block)
+      else
+        raise Xlogin::Error.new("Invalid argument: '#{args}'")
+      end
     end
 
     def pool(args, **opts, &block)
-      pool = factory.build_pool(args, **opts)
-
-      return pool unless block
-      begin block.call(pool) ensure pool.close end
+      factory.build_pool(args, **opts, &block)
     end
 
     def configure(&block)

data/lib/xlogin/cli.rb

@@ -42,7 +42,7 @@ module Xlogin
           loggers << ((jobs > 1)? buffer : $stdout)
           loggers << File.expand_path(File.join(config[:logdir], "#{info[:name]}.log"), ENV['PWD']) if config[:logdir]
 
-          session = Xlogin.get(info.merge(log: loggers))
+          session = Xlogin.get(info.merge(log: loggers, **config[:env]))
           session.enable(session.config.enable) if session.config.enable && Xlogin.settings.enable?
 
           command_lines = config[:command].flat_map { |e| e.to_s.split(';').map(&:strip) }
@@ -60,7 +60,7 @@ module Xlogin
 
     def run(args)
       config = Hash.new
-      config[:env]       = []
+      config[:env]       = {}
       config[:inventory] = []
       config[:template]  = []
       config[:command]   = []
@@ -70,23 +70,21 @@ module Xlogin
       parser.banner  = "#{File.basename($0)} HOST_PATTERN [Options]"
       parser.version = Xlogin::VERSION
 
-      parser.on('-i PATH', '--inventory', String, 'The PATH to the inventory file.')            { |v| config[:inventory] << v }
-      parser.on('-t PATH', '--template',  String, 'The PATH to the template file or directory.'){ |v| config[:template]  << v }
-      parser.on('-L PATH', '--log-dir',   String, 'The PATH to the log directory.')             { |v| config[:logdir]     = v }
-
-      parser.on('-l',         '--list',   TrueClass, 'List the inventory.')       { |v| config[:runner] = self.method(:list) }
+      parser.on('-i PATH', '--inventory', String,  'The PATH to the inventory file.')            { |v| config[:inventory] << v }
+      parser.on('-t PATH', '--template',  String,  'The PATH to the template file or directory.'){ |v| config[:template]  << v }
+      parser.on('-L PATH', '--log-dir',   String,  'The PATH to the log directory.')             { |v| config[:logdir]     = v }
+      parser.on('-j NUM',  '--jobs',      Integer, 'The NUM of jobs to execute in parallel.')    { |v| config[:jobs] = v }
+      parser.on(           '--enable',    TrueClass, 'Automatically enable privilege mode.')     { |v| config[:enable] = v }
+      parser.on('-l',      '--list',      TrueClass, 'List the inventory.')       { |v| config[:runner] = self.method(:list) }
       parser.on('-e COMMAND', '--exec',   String,    'Execute commands and quit.'){ |v| config[:runner] = self.method(:exec); config[:command] << v }
-
-      parser.on('-E KEY=VAL', '--env',    /(\w+=\w+)+/, 'Environment variables.')                 { |v| config[:env] << v }
-      parser.on('-j NUM',     '--jobs',   Integer,      'The NUM of jobs to execute in parallel.'){ |v| config[:jobs] = v }
+      parser.on('-E KEY=VAL', '--env',    /\w+=[^=]+/, 'Environment variables.')  { |v| v.split('=').tap{ |k, v| config[:env].update(k.to_sym => v) } }
 
       config[:patterns]  = parser.parse!(args)
       config[:inventory] << DEFAULT_INVENTORY if config[:inventory].empty?
       config[:template]  << DEFAULT_TEMPLATE  if config[:template].empty?
 
       Xlogin.configure do
-        set Hash[config[:env].map{ |v| v.split('=') }]
-
+        set enable: config[:enable]
         source   *config[:inventory].map{ |e| File.expand_path(e, ENV['PWD']) }
         template *config[:template].map { |e| File.expand_path(e, ENV['PWD']) }
       end

data/lib/xlogin/factory.rb

@@ -93,22 +93,26 @@ module Xlogin
       end
     end
 
-    def build(type:, **opts)
+    def build(type:, **opts, &block)
       template = get_template(type)
       raise Xlogin::Error.new("Template not found: '#{type}'") unless template
 
-      template.build(uri(opts), **opts)
+      session = template.build(uri(opts), **opts)
+      return session unless block
+      begin block.call(session) ensure session.close end
     end
 
-    def build_pool(args, **opts)
-      Xlogin::SessionPool.new(args, **opts)
+    def build_pool(args, **opts, &block)
+      pool = Xlogin::SessionPool.new(args, **opts)
+      return pool unless block
+      begin block.call(pool) ensure pool.close end
     end
 
-    def build_from_hostname(args, **opts)
+    def build_from_hostname(args, **opts, &block)
       hostinfo = get_hostinfo(args)
       raise Xlogin::Error.new("Host not found: '#{args}'") unless hostinfo
 
-      build(**hostinfo.merge(**opts))
+      build(**hostinfo.merge(**opts), &block)
     end
 
     def method_missing(method_name, *args, **opts, &block)

data/lib/xlogin/rake_task.rb

@@ -10,21 +10,21 @@ module Xlogin
     class << self
       include Rake::DSL
 
-      def all(*patterns, &block)
+      def all(*patterns, **opts, &block)
         description = Rake.application.last_description
         task all: Xlogin.list(*patterns).map{ |e| e[:name] }
 
         desc description
-        generate(*patterns, &block)
+        generate(*patterns, **opts, &block)
       end
 
-      def generate(*patterns, &block)
+      def generate(*patterns, **opts, &block)
         description = Rake.application.last_description
 
         hostnames = Xlogin.list(*patterns).map{ |e| e[:name] }
         hostnames.each do |hostname|
           desc "#{description} - #{hostname}" if description
-          RakeTask.new(hostname, &block)
+          RakeTask.new(hostname, **opts, &block)
         end
       end
 
@@ -40,14 +40,13 @@ module Xlogin
     attr_reader   :name
     attr_accessor :log
     attr_accessor :lock
-    attr_accessor :timeout
     attr_accessor :silent
     attr_accessor :fail_on_error
 
-    def initialize(name)
+    def initialize(name, **opts)
       @name     = name
+      @opts     = opts
       @runner   = nil
-      @timeout  = nil
       @silent ||= Rake.application.options.silent
       @fail_on_error = true
 
@@ -66,8 +65,10 @@ module Xlogin
 
     private
     def define
-      mkdir_p(File.dirname(log),  verbose: Rake.application.options.trace) if log
-      mkdir_p(File.dirname(lock), verbose: Rake.application.options.trace) if lock
+      self.log  = File.join(Dir.pwd, 'log',  name + '.log')   if self.log  == true
+      self.lock = File.join(Dir.pwd, 'lock', name_with_scope) if self.lock == true
+      mkdir_p(File.dirname(self.log),  verbose: Rake.application.options.trace) if self.log
+      mkdir_p(File.dirname(self.lock), verbose: Rake.application.options.trace) if self.lock
 
       if lock
         task(name => lock)
@@ -89,7 +90,7 @@ module Xlogin
       loggers << log    if log
       loggers << STDOUT if !silent && !Rake.application.options.always_multitask
 
-      session = Xlogin.get(name, log: loggers, timeout: timeout)
+      session = Xlogin.get(name, log: loggers, **@opts)
       instance_exec(session, &@runner)
 
       print(buffer.string) if !silent && Rake.application.options.always_multitask
@@ -98,27 +99,25 @@ module Xlogin
       RakeTask.shutdown! if fail_on_error
 
       session.log_message(e.to_s.colorize(color: :red)) if session
-      print(buffer.string + "\n", color: :red) if Rake.application.options.always_multitask
+      print(buffer.string.colorize(color: :red) + "\n") if Rake.application.options.always_multitask
       return false
     ensure
       session.close rescue nil
     end
 
-    def puts(text, **opts)
-      return text.each { |e| puts(e, **opts) } if text.kind_of?(Array)
-
-      text = text.to_s
-      text = text + "\n" unless text[-1] == "\n"
-      print(text, **opts)
+    def puts(text)
+      strio = StringIO.new.tap{ |io| io.puts text }
+      print(strio.string)
     end
 
-    def print(text, **opts)
-      text = text.to_s
+    def print(text)
+      text = text.to_s.gsub("\r", '')
       return if text.empty?
 
-      text = text.gsub("\r", '')
-      text = text.lines.map{ |text| "#{name}\t|#{text}" }.join if Rake.application.options.always_multitask
-      text = text.colorize(**opts)
+      if Rake.application.options.always_multitask
+        strio = StringIO.new.tap{ |io| io.puts text.lines.map{ |line| "#{name}\t|#{line}" } }
+        text = strio.string
+      end
       $stdout.print(text)
     end
 

data/lib/xlogin/session.rb

@@ -10,9 +10,9 @@ module Xlogin
 
     def initialize(template, uri, **opts)
       @uri  = uri
-      @host = uri.host
-      @port = uri.port
-      @port ||= case @uri.scheme
+      @host = opts[:host] || uri.host
+      @port = opts[:port] || uri.port
+      @port ||= case opts[:scheme] || uri.scheme
                 when 'ssh'    then 22
                 when 'telnet' then 23
                 end
@@ -25,7 +25,7 @@ module Xlogin
       @host, @port = Xlogin.factory.open_tunnel(@tunnel, @host, @port) if @tunnel
 
       num_try = 0
-      username, password = uri.userinfo.to_s.split(':')
+      username, password = (opts[:userinfo] || ENV['USERINFO'] || uri.userinfo).to_s.split(':')
 
       begin
         args = Hash.new
@@ -64,11 +64,9 @@ module Xlogin
       Regexp.union(*@template.prompts.map(&:first))
     end
 
-    def duplicate(type: @template.name, **args)
-      template = Xlogin::Factory.instance.get_template(type)
-      raise Xlogin::Error.new("Template not found: '#{type}'") unless template
-
-      template.build(@uri, **@config.to_h.merge(args))
+    def duplicate(type: @template.name, uri: @uri.dup, **args, &block)
+      args = @config.to_h.merge(args)
+      Xlogin::Factory.instance.build(type: type, uri: uri, **args, &block)
     end
 
     def puts(string = '', &block)

data/lib/xlogin/version.rb

@@ -1,3 +1,3 @@
 module Xlogin
-	VERSION = "0.16.2"
+	VERSION = "0.16.7"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: xlogin
 version: !ruby/object:Gem::Version
-  version: 0.16.2
+  version: 0.16.7
 platform: ruby
 authors:
 - haccht
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-05-11 00:00:00.000000000 Z
+date: 2021-06-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: net-telnet
@@ -193,7 +193,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
+rubygems_version: 3.0.0
 signing_key: 
 specification_version: 4
 summary: rancid clogin alternative