bcome 1.3.2 → 1.4.0

data/lib/objects/exception/missing_identifier_on_view.rb

@@ -1,7 +1,11 @@
-module ::Bcome::Exception
-  class MissingIdentifierOnView < ::Bcome::Exception::Base
-    def message_prefix
-      'One of your views is missing an identifier field'
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class MissingIdentifierOnView < ::Bcome::Exception::Base
+      def message_prefix
+        'One of your views is missing an identifier field'
+      end
     end
   end
 end

data/lib/objects/exception/missing_inventory_contributors.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class MissingInventoryContributors < ::Bcome::Exception::Base
+      def message_prefix
+        'You must set the inventory contributors when defining a merged inventory'
+      end
+    end
+  end
+end

data/lib/objects/exception/missing_ip_address_on_server.rb

@@ -1,7 +1,11 @@
-module ::Bcome::Exception
-  class MissingIpaddressOnServer < ::Bcome::Exception::Base
-    def message_prefix
-      'A static server must specify either a public or internal ip address'
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class MissingIpaddressOnServer < ::Bcome::Exception::Base
+      def message_prefix
+        'A static server must specify either a public or internal ip address'
+      end
     end
   end
 end

data/lib/objects/exception/missing_network_config.rb

@@ -1,7 +1,11 @@
-module ::Bcome::Exception
-  class MissingNetworkConfig < ::Bcome::Exception::Base
-    def message_prefix
-      'Cannot find network config at'
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class MissingNetworkConfig < ::Bcome::Exception::Base
+      def message_prefix
+        'Cannot find network config at'
+      end
     end
   end
 end

data/lib/objects/exception/missing_or_invalid_client_secrets.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class MissingOrInvalidClientSecrets < ::Bcome::Exception::Base
+      def message_prefix
+        "Missing or invalid client secrets file\s"
+      end
+    end
+  end
+end

data/lib/objects/exception/missing_params_for_rsync.rb

@@ -1,7 +1,11 @@
-module ::Bcome::Exception
-  class MissingParamsForRsync < ::Bcome::Exception::Base
-    def message_prefix
-      ''
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class MissingParamsForRsync < ::Bcome::Exception::Base
+      def message_prefix
+        ''
+      end
     end
   end
 end

data/lib/objects/exception/missing_params_for_scp.rb

@@ -1,7 +1,11 @@
-module ::Bcome::Exception
-  class MissingParamsForScp < ::Bcome::Exception::Base
-    def message_prefix
-      ''
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class MissingParamsForScp < ::Bcome::Exception::Base
+      def message_prefix
+        ''
+      end
     end
   end
 end

data/lib/objects/exception/missing_subselection_key.rb

@@ -1,7 +1,11 @@
-module ::Bcome::Exception
-  class MissingSubselectionKey < ::Bcome::Exception::Base
-    def message_prefix
-      'You must set a subselection key when defining a subselection inventory'
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class MissingSubselectionKey < ::Bcome::Exception::Base
+      def message_prefix
+        'You must set a subselection key when defining a subselection inventory'
+      end
     end
   end
 end

data/lib/objects/exception/missing_type_on_view.rb

@@ -1,7 +1,11 @@
-module Bcome::Exception
-  class MissingTypeOnView < ::Bcome::Exception::Base
-    def message_prefix
-      'One of your views is missing a type field'
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class MissingTypeOnView < ::Bcome::Exception::Base
+      def message_prefix
+        'One of your views is missing a type field'
+      end
     end
   end
 end

data/lib/objects/exception/no_node_found_for_breadcrumb.rb

@@ -1,7 +1,11 @@
-module ::Bcome::Exception
-  class NoNodeFoundForBreadcrumb < ::Bcome::Exception::Base
-    def message_prefix
-      'No node exists for breadcrumb'
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class NoNodeFoundForBreadcrumb < ::Bcome::Exception::Base
+      def message_prefix
+        'No node exists for breadcrumb'
+      end
     end
   end
 end

data/lib/objects/exception/no_node_named_by_identifier.rb

@@ -1,7 +1,11 @@
-module Bcome::Exception
-  class NoNodeNamedByIdentifier < ::Bcome::Exception::Base
-    def message_prefix
-      'Cannot find identifier within inventory'
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class NoNodeNamedByIdentifier < ::Bcome::Exception::Base
+      def message_prefix
+        'Cannot find identifier within inventory'
+      end
     end
   end
 end

data/lib/objects/exception/node_identifiers_must_be_unique.rb

@@ -1,7 +1,11 @@
-module ::Bcome::Exception
-  class NodeIdentifiersMustBeUnique < ::Bcome::Exception::Base
-    def message_prefix
-      'Node identifiers cannot be ambiguous: '
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class NodeIdentifiersMustBeUnique < ::Bcome::Exception::Base
+      def message_prefix
+        'Node identifiers cannot be ambiguous: '
+      end
     end
   end
 end

data/lib/objects/exception/orchestration_script_does_not_exist.rb

@@ -1,7 +1,11 @@
-module Bcome::Exception
-  class OrchestrationScriptDoesNotExist < ::Bcome::Exception::Base
-    def message_prefix
-      'Cannot find orchestration file at path: '
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class OrchestrationScriptDoesNotExist < ::Bcome::Exception::Base
+      def message_prefix
+        'Cannot find orchestration file at path: '
+      end
     end
   end
 end

data/lib/objects/exception/proxy_host_node_does_not_have_public_ip_address.rb

@@ -1,7 +1,11 @@
-module ::Bcome::Exception
-  class ProxyHostNodeDoesNotHavePublicIp < ::Bcome::Exception::Base
-    def message_prefix
-      'Node missing public ip address, and so cannot be used as a proxy ssh server'
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class ProxyHostNodeDoesNotHavePublicIp < ::Bcome::Exception::Base
+      def message_prefix
+        'Node missing public ip address, and so cannot be used as a proxy ssh server'
+      end
     end
   end
 end

data/lib/objects/exception/unknown_dynamic_server_type.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class UnknownDynamicServerType < ::Bcome::Exception::Base
+      def message_prefix
+        ''
+      end
+    end
+  end
+end

data/lib/objects/exception/unknown_method_for_namespace.rb

@@ -1,7 +1,11 @@
-module Bcome::Exception
-  class UnknownMethodForNamespace < ::Bcome::Exception::Base
-    def message_prefix
-      'unknown method '
+# frozen_string_literal: true
+
+module Bcome
+  module Exception
+    class UnknownMethodForNamespace < ::Bcome::Exception::Base
+      def message_prefix
+        'unknown method or namespace'
+      end
     end
   end
 end

data/lib/objects/interactive/session.rb

@@ -1,10 +1,13 @@
+# frozen_string_literal: true
+
 module Bcome::Interactive
   class Session
     class << self
       def run(node, session_type, init_data = {})
         session_end_message = "\ninteractive session ended\n".informational
         begin
-          session = ::Bcome::Interactive::Session.new(node, session_type_to_klass[session_type], init_data)
+          session_klass = session_type.is_a?(Symbol) ? session_type_to_klass[session_type] : session_type
+          session = ::Bcome::Interactive::Session.new(node, session_klass, init_data)
           session.start
         rescue ::Bcome::Exception::InteractiveSessionHalt => e
           puts session_end_message

data/lib/objects/interactive/session_item/base.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Bcome::Interactive::SessionItem
   class Base
     def initialize(session, init_data)

data/lib/objects/interactive/session_item/capture_input.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Bcome::Interactive::SessionItem
   class CaptureInput < ::Bcome::Interactive::SessionItem::Base
     def do

data/lib/objects/interactive/session_item/transparent_ssh.rb

@@ -1,18 +1,22 @@
+# frozen_string_literal: true
+
 module Bcome::Interactive::SessionItem
   class TransparentSsh < ::Bcome::Interactive::SessionItem::Base
-    END_SESSION_KEY = '\\q'.freeze
-    HELP_KEY = '\\?'.freeze
-    LIST_KEY = '\\l'.freeze
-
-    DANGER_CMD = "rm\s+-r|rm\s+-f|rm\s+-fr|rm\s+-rf|rm".freeze
+    END_SESSION_KEY = '\\q'
+    HELP_KEY = '\\?'
+    LIST_KEY = '\\l'
+    RECONNECT_CMD = '\\r'
+    DANGER_CMD = "rm\s+-r|rm\s+-f|rm\s+-fr|rm\s+-rf|rm"
 
     def machines
-      node.server? ? [node] : node.machines
+      skip_for_hidden = true
+      node.server? ? [node] : node.machines(skip_for_hidden)
     end
 
     def do
       puts ''
       open_ssh_connections
+      puts "\nINTERACTIVE COMMAND SESSION".underline
       show_menu
       puts ''
       list_machines
@@ -22,10 +26,7 @@ module Bcome::Interactive::SessionItem
     def action
       input = get_input
 
-      if exit?(input)
-        close_ssh_connections
-        return
-      end
+      return if exit?(input)
 
       if show_menu?(input)
         show_menu
@@ -40,16 +41,13 @@ module Bcome::Interactive::SessionItem
     end
 
     def show_menu
-      warning = "\n\sCommands entered here will be executed on " + 'EVERY'.underline + ' machine in your selection.'
-      second_warning = "\n\n\s" + 'Use with CAUTION.'.warning
-      info = "\n\n\s\\l list machines\n\s\\q to quit\n\s\\? this message".informational
-      puts "#{warning}#{second_warning}#{info}\n"
+      warning = "\nCommands entered here will be executed on" + "\severy\s".warning + "machine in your selection. \n\nUse with caution or hit \\q if you're unsure what this is."
+      info = "\n\n\\l list machines\n\\q to quit\n\\? this message".informational
+      puts warning + "#{info}\n"
     end
 
     def handle_the_unwise(input)
-      if prompt_for_confirmation('Command may be dangerous to run on all machines. Are you sure you want to proceed? [Y|N] > '.error)
-        execute_on_machines(input)
-      end
+      execute_on_machines(input) if prompt_for_confirmation('Command may be dangerous to run on all machines. Are you sure you want to proceed? [Y|N] > '.error)
     end
 
     def command_may_be_unwise?(input)
@@ -65,7 +63,7 @@ module Bcome::Interactive::SessionItem
     def start_message; end
 
     def terminal_prompt
-      ">\s".bold.terminal_prompt
+      "enter a command>\s"
     end
 
     def exit?(input)
@@ -81,8 +79,8 @@ module Bcome::Interactive::SessionItem
     end
 
     def open_ssh_connections
-      ::Bcome::Ssh::ConnectionHandler.connect(node, show_progress: true)
-      system('clear')
+      ::Bcome::Ssh::Connector.connect(node, show_progress: true)
+      # system('clear')
     end
 
     def close_ssh_connections
@@ -91,20 +89,28 @@ module Bcome::Interactive::SessionItem
 
     def indicate_failed_nodes(unconnected_nodes)
       unconnected_nodes.each do |node|
-        puts "\s\s - #{node.namespace}".bc_cyan
+        puts "\s\s - #{node.namespace}"
       end
     end
 
     def list_machines
       puts "\n"
       machines.each do |machine|
-        puts "\s- #{machine.namespace}"
+        puts "- #{machine.namespace}"
       end
     end
 
     def execute_on_machines(user_input)
       machines.pmap do |machine|
-        machine.run(user_input)
+        begin
+          machine.run(user_input)
+        rescue IOError => e
+          puts "Reopening connection to\s".informational + machine.identifier
+          machine.reopen_ssh_connection
+          machine.run(user_input)
+        rescue Exception => e
+          puts "Error connecting to #{machine.identifier} (#{e.message})".error
+        end
       end
     end
   end

data/lib/objects/loading_bar/handler.rb

@@ -0,0 +1,80 @@
+# frozen_string_literal: true
+
+module Bcome
+  module LoadingBar
+    module Handler
+      def start_indicator(config)
+        klass = config[:type] == :progress ? ::Bcome::LoadingBar::Indicator::Progress : ::Bcome::LoadingBar::Indicator::Basic
+        @indicator = klass.new(config)
+        fork_process
+      end
+
+      def stop_indicator
+        signal_stop
+      end
+
+      def cursor
+        ::TTY::Cursor
+      end
+
+      def wrap_indicator(config, &block)
+        begin
+          start_indicator(config)
+          cursor.invisible do
+            block.call
+          end
+        rescue IRB::Abort
+          stop_indicator
+          raise ::Bcome::Exception::Generic, 'Interrupt'
+        rescue StandardError => e
+          stop_indicator
+          raise ::Bcome::Exception::Generic, e.message if config[:type] == :basic
+        end
+        stop_indicator
+      end
+
+      def fork_process
+        @pid = fork do
+          Signal.trap(::Bcome::LoadingBar::Indicator::Base::SIGNAL_SUCCESS) do
+            @indicator.increment_success
+          end
+
+          Signal.trap(::Bcome::LoadingBar::Indicator::Base::SIGNAL_FAILURE) do
+            @indicator.increment_failure
+          end
+
+          @indicator.indicate
+        end
+        ::Bcome::LoadingBar::PidBucket.instance << @pid
+      end
+
+      def do_signal(signal)
+        ::Process.kill(signal, @pid)
+      end
+
+      def signal_stop
+        # Stop the loader
+        do_signal(::Bcome::LoadingBar::Indicator::Base::SIGNAL_STOP)
+
+        # De-register its pid
+        ::Bcome::LoadingBar::PidBucket.instance - @pid
+
+        # Show the PARENT process indicator, which has been kept in sync with the forked process. We do this due to race condition where the forked indicator
+        # does not gets final status before it is killed, so that it looks as if it has not completed
+        @indicator.show
+      end
+
+      def signal_success
+        do_signal(::Bcome::LoadingBar::Indicator::Base::SIGNAL_SUCCESS)
+        # Keep parent indicator in sync (see #signal_stop)
+        @indicator.increment_success
+      end
+
+      def signal_failure
+        do_signal(::Bcome::LoadingBar::Indicator::Base::SIGNAL_FAILURE)
+        # Keeo parent indicator in sync (see #signal_stop)
+        @indicator.increment_failure
+      end
+    end
+  end
+end