bcome 1.3.2 → 1.4.0

data/lib/objects/node/inventory/merge.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+module Bcome
+  module Node
+    module Inventory
+      class Merge < ::Bcome::Node::Inventory::Base
+        attr_reader :dynamic_nodes_loaded
+
+        def initialize(*params)
+          super
+          raise Bcome::Exception::MissingInventoryContributors, @views unless @views[:contributors]
+        end
+
+        def nodes_loaded?
+          true
+        end
+
+        def contributing_inventories
+          @inventories ||= @views[:contributors].collect { |inventory_key| load_inventory(inventory_key) }
+        end
+
+        def resources
+          @resources ||= do_set_resources
+        end
+
+        def update_nodes
+          resources.update_nodes(self)
+        end
+
+        def do_set_resources
+          ::Bcome::Node::Resources::Merged.new(inventory: self)
+        end
+
+        def load_inventory(from_crumb)
+          inventory = ::Bcome::Node::Factory.instance.bucket[from_crumb]
+          raise Bcome::Exception::CannotFindInventory, "for key '#{from_crumb}'" unless inventory
+
+          inventory
+        end
+      end
+    end
+  end
+end

data/lib/objects/node/inventory/subselect.rb

@@ -1,60 +1,78 @@
-module Bcome::Node::Inventory
-  class Subselect < ::Bcome::Node::Inventory::Base
-    def initialize(*params)
-      super
-      raise Bcome::Exception::MissingSubselectionKey, @views unless @views[:subselect_from]
-      update_nodes
-    end
+# frozen_string_literal: true
 
-    def resources
-      @resources ||= do_set_resources
-    end
+module Bcome
+  module Node
+    module Inventory
+      class Subselect < ::Bcome::Node::Inventory::Base
+        def initialize(*params)
+          super
+          raise Bcome::Exception::MissingSubselectionKey, @views unless @views[:subselect_from]
 
-    def update_nodes
-      resources.update_nodes(self)
-    end
+          update_nodes
+        end
 
-    def do_set_resources
-      ::Bcome::Node::Resources::SubselectInventory.new(parent_inventory: parent_inventory, filters: filters)
-    end
+        def enabled_menu_items
+          super + %i[reload]
+        end
 
-    def nodes_loaded?
-      true
-    end
+        def menu_items
+          base_items = super.dup
 
-    def filters
-      # Flex point for filters, as obviously we need to support more than just ec2 filtering eventually
-      @views[:filters] ? @views[:filters] : {}
-    end
+          base_items[:reload] = {
+            description: 'Restock this inventory from remote',
+            console_only: true
+          }
+          base_items
+        end
 
-    def self.to_s
-      'sub-inventory'
-    end
+        def resources
+          @resources ||= do_set_resources
+        end
 
-    def reload
-      do_reload
-    end
+        def update_nodes
+          resources.update_nodes(self)
+        end
 
-    def do_reload
-      parent_inventory.resources.reset_duplicate_nodes!
-      parent_inventory.do_reload
-      resources.run_subselect
-      update_nodes
-      return
-    end
+        def do_set_resources
+          ::Bcome::Node::Resources::SubselectInventory.new(parent_inventory: parent_inventory, filters: filters)
+        end
 
-    private
+        def nodes_loaded?
+          true
+        end
 
-    def parent_inventory
-      @parent_inventory ||= load_parent_inventory
-    end
+        def filters
+          # Flex point for filters, as obviously we need to support more than just ec2 filtering eventually
+          @views[:filters] || {}
+        end
+
+        def reload
+          do_reload
+        end
+
+        def do_reload
+          parent_inventory.resources.reset_duplicate_nodes!
+          parent_inventory.do_reload
+          resources.run_subselect
+          update_nodes
+          nil
+        end
+
+        private
+
+        def parent_inventory
+          @parent_inventory ||= load_parent_inventory
+        end
+
+        def load_parent_inventory
+          parent_crumb = @views[:subselect_from]
+          parent = ::Bcome::Node::Factory.instance.bucket[parent_crumb]
+          raise Bcome::Exception::CannotFindSubselectionParent, "for key '#{parent_crumb}'" unless parent
+          raise Bcome::Exception::CanOnlySubselectOnInventory, "breadcrumb'#{parent_crumb}' represents a #{parent.class}'" unless parent.inventory?
 
-    def load_parent_inventory
-      parent_crumb = @views[:subselect_from]
-      parent = ::Bcome::Node::Factory.instance.bucket[parent_crumb]
-      raise Bcome::Exception::CannotFindSubselectionParent, "for key '#{parent_crumb}'" unless parent
-      raise Bcome::Exception::CanOnlySubselectOnInventory, "breadcrumb'#{parent_crumb}' represents a #{parent.class}'" unless parent.inventory?
-      parent
+          parent
+        end
+      end
     end
   end
 end

data/lib/objects/node/kube/base.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+## Bcome's kubernetes node namespaces
+module Bcome::Node
+  module Kube
+    class Base
+      attr_reader :resources
+
+      def initialize(kube_wrap, config = {})
+        @kube_wrap = kube_wrap
+        @config = config
+        @resources = []
+      end
+
+      def parent
+        @config[:parent]
+      end
+
+      def run(command)
+        @kube_wrap.run(command)
+      end
+
+      def build
+        child_node_lookup_result = run(get_child_node_command)
+        parse(child_node_lookup_result)
+      end
+
+      def parse(child_node_lookup_result)
+        # Get the output
+        output = child_node_lookup_result.stdout
+        # Remove the title & form array
+        data = output.remove_lines(1).split("\n")
+
+        structured_data = {}
+        data.each do |raw_namespace_line|
+          child_config = get_child_config_from_line(raw_namespace_line)
+          child_config.merge!(parent: self)
+          @resources << child_node_klass.new(@kube_wrap, child_config)
+        end
+      end
+
+      def get_child_node_command
+        raise 'should be overriden'
+      end
+
+      def get_child_config_from_line(*_params)
+        raise 'should be overidden'
+      end
+    end
+  end
+end

data/lib/objects/node/kube/container.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module Bcome::Node::Kube
+  class Container < Bcome::Node::Kube::Base
+    # Expose shell
+    # Enable Server::Spec testing
+    # Enable file handlers
+  end
+end

data/lib/objects/node/kube/estate.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Bcome::Node::Kube
+  class Estate < Bcome::Node::Kube::Base
+    # CONTAINS NAMESPACES
+    def get_child_node_command
+      'get namespaces'
+    end
+
+    def child_node_klass
+      ::Bcome::Kube::Node::Namespace
+    end
+
+    def get_child_config_from_line(raw_line)
+      name, active, age = raw_line.split("\s")
+      { name: name, active: active, age: age }
+    end
+  end
+end

data/lib/objects/node/kube/namespace.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Bcome::Node::Kube
+  class Namespace < Bcome::Node::Kube::Base
+    # CONTAINS PODS
+
+    def get_child_node_command
+      "get pods -n #{name}"
+    end
+
+    def child_node_klass
+      ::Bcome::Kube::Node::Pod
+    end
+
+    def get_child_config_from_line(raw_line)
+      name, ready, status, restarts, age = raw_line.split("\s")
+      { name: name, ready: ready, status: status, reastarts: restarts, age: age }
+    end
+
+    def name
+      @config[:name]
+    end
+  end
+end

data/lib/objects/node/kube/pod.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Bcome::Node::Kube
+  class Pod < Bcome::Node::Kube::Base
+    # CONTAINS CONTAINERS
+
+    def get_child_node_command
+      "get pods #{name} -n #{parent.name} -o jsonpath=‘{.spec.containers[*].name}’"
+    end
+
+    def child_node_klass
+      ::Bcome::Kube::Node::Container
+    end
+
+    def name
+      @config[:name]
+    end
+
+    def get_child_config_from_line(raw_line)
+      name = raw_line.split("\s")
+      { name: name }
+    end
+  end
+end

data/lib/objects/node/kube_wrap.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+## Kubectl command wrapper. This will integrate directly with bcome's GCP authentication driver
+class KubeWrap
+  KUBE_BIN = 'kubectl'
+
+  def initialize(access_token)
+    @access_token = access_token
+  end
+
+  def run(command_suffix)
+    full_command = create_command_for(command_suffix)
+    result = ::Bcome::Command::Local.run(full_command)
+
+    if result.failed?
+      puts "\n" + result.stderr.error + "\n"
+      raise
+    end
+
+    result
+  end
+
+  def create_command_for(suffix)
+    "#{KUBE_BIN} --token #{@access_token} #{suffix}"
+  end
+end

data/lib/objects/node/meta/base.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Bcome::Node::Meta
   class Base
     attr_reader :data
@@ -10,11 +12,16 @@ module Bcome::Node::Meta
       matchers[:values] = [matchers[:values]] if matchers[:values] && !matchers[:values].is_a?(Array)
 
       raise Bcome::Exception::InvalidMatcherQuery unless matchers[:key] && matchers[:values].is_a?(Array)
+
       key = matchers[:key].to_sym
       values = matchers[:values]
       @data.key?(key) && values.include?(@data[key])
     end
 
+    def keys
+      @data.keys
+    end
+
     def dump
       @data.each do |k, v|
         puts "#{k} => #{v.resource_value}"
@@ -29,7 +36,7 @@ module Bcome::Node::Meta
       if @data.key?(key)
         @data[key]
       elsif default
-        return default
+        default
       else
         raise Bcome::Exception::CantFindKeyInMetadata, key unless @data.key?(key)
       end

data/lib/objects/node/meta/cloud.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Bcome::Node::Meta
   class Cloud < Bcome::Node::Meta::Base
   end

data/lib/objects/node/meta/local.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Bcome::Node::Meta
   class Local < Bcome::Node::Meta::Base
   end

data/lib/objects/node/meta_data_factory.rb

@@ -1,5 +1,7 @@
+# frozen_string_literal: true
+
 module Bcome::Node::LocalMetaDataFactory
-  META_DATA_FILE_PATH_PREFIX = 'bcome/metadata'.freeze
+  META_DATA_FILE_PATH_PREFIX = 'bcome/metadata'
 
   def metadata
     @node_metadata ||= do_create_metadata

data/lib/objects/node/meta_data_loader.rb

@@ -1,58 +1,64 @@
+# frozen_string_literal: true
+
 module Bcome::Node
   class MetaDataLoader
     include ::Singleton
 
-    META_DATA_FILE_PATH_PREFIX = 'bcome/metadata'.freeze
+    META_DATA_FILE_PATH_PREFIX = 'bcome/metadata'
 
     def initialize
       @all_metadata_filenames = Dir["#{META_DATA_FILE_PATH_PREFIX}/*"]
     end
 
-    def decryption_key
-      @decryption_key
-    end
- 
+    attr_accessor :decryption_key
+
     def data
       @data ||= do_load
     end
 
     def data_for_namespace(namespace)
-      static_data = data[namespace.to_sym] ? data[namespace.to_sym] : {}
+      static_data = data[namespace.to_sym] || {}
       static_data.merge(terraform_data_for_namespace(namespace))
     end
 
     def terraform_data_for_namespace(namespace)
-      parser = Bcome::Terraform::Parser.new(namespace)
-      attributes = parser.attributes
-      if attributes.keys.any?
-        {
-          "terraform_attributes" => attributes
-        }
-      else
-        {}
-      end
+      # Radical departure II - all we care is the outputs. This will then work across any terraform backend, and any version
+      tf_state = ::Bcome::Terraform::Output.new(namespace)
+      terraform_data = {}
+      terraform_data['terraform_outputs'] = tf_state.output
+      terraform_data
     end
 
     def prompt_for_decryption_key
-      print "\nEnter your decryption key: ".informational
+      decryption_key_prompt = 'Enter your metadata decryption key: '.informational
+
+      print "\n#{decryption_key_prompt}"
       @decryption_key = STDIN.noecho(&:gets).chomp
+      print "\r#{decryption_key_prompt}" + "**********\n\n"
     end
 
     def load_file_data_for(filepath)
-     if filepath =~ /.enc/  # encrypted file contents 
+      if filepath =~ /.enc/ # encrypted file contents
         prompt_for_decryption_key unless decryption_key
-        encrypted_contents = File.read(filepath)     
-        decrypted_contents = encrypted_contents.decrypt(decryption_key) 
-        return YAML.load(decrypted_contents)
+        encrypted_contents = File.read(filepath)
+        decrypted_contents = encrypted_contents.decrypt(decryption_key)
+
+        begin
+          YAML.safe_load(decrypted_contents, [Symbol], [], true)
+        rescue Exception => e
+          @decryption_key = nil
+          raise ::Bcome::Exception::InvalidMetaDataConfig, "#{e.class} #{e.message} - " + decrypted_contents
+        end
+
       else # unencrypted
-        return YAML.load_file(filepath)
+        YAML.load_file(filepath)
       end
     end
 
     def do_load
       all_meta_data = {}
       @all_metadata_filenames.each do |filepath|
-        next if filepath =~ /-unenc/  # we only read from the encrypted, packed files. 
+        next if filepath =~ /-unenc/ # we only read from the encrypted, packed files.
 
         begin
           filedata = load_file_data_for(filepath)
@@ -61,7 +67,7 @@ module Bcome::Node
           raise Bcome::Exception::InvalidMetaDataConfig, "Error: #{e.message}"
         end
       end
-      return all_meta_data
+      all_meta_data
     end
   end
 end