bcome 0.7.0 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 75e06da0e2c27915d8ca0a306ef24c0ce217c0f7
-  data.tar.gz: ec8e78f0956c494335ea297d32f86917b827ce43
+  metadata.gz: 6b13dea68c96da6c50cd9f911836c141f6b27720
+  data.tar.gz: 9cbb87e356a39489d29dac89e2a5720131d442a2
 SHA512:
-  metadata.gz: da2ca953c5f9d2c6d3778e68311f61b0e7e74f6cc512819c231d89595349f76e5beb25900888db306d4ea3f51cb3003ae83de1737547973d0e83cb94b3feea9e
-  data.tar.gz: 7a7587be892d152d337f81b112926ed5c075391c27b80e92fa74571f2b2605c44f53af0c2a93532a2ae7b0f9a98dbbf8bf2c0ce46fa476526ac81f45ce435fa4
+  metadata.gz: ef0390c136034e31889cc7f18727340106f713448696403e163b33cf10b55d02a27d00123017a9c1b20055cade92ce23aa87fcd9a98e25b38a99e17e5211225b
+  data.tar.gz: 7be2d171f7e899da877709792b6f6750e5db8dc8477a4649eef4b44250cffbc52c6cc5542db9d351f5ae567c66552bbbe58e5d1545c71304d5b171f45cba4e72

data/bin/bcome

@@ -1,46 +1,15 @@
 #!/usr/bin/env ruby
 
-# (C) Guillaume Roderick (webzakimbo)
-# MIT licence.
-
-## Load & Boot into namespace
-require 'irb'
-require "irb/completion"
 require 'bcome'
 
-load Bcome::Boot.boot_path
-
-## Load the shell
-unless @no_shell
-  require 'irb/ext/multi-irb'
-  IRB.parse_opts_with_ignoring_script
-  irb_method = (@direct_command && @direct_shell_commands.include?(@direct_command)) ? @direct_command.to_sym : nil
-  IRB.irb irb_method, @context_object
-end
+begin
+  spawn_into_console = true
+  arguments = ARGV - [ARGV[0]]
+  ::Bcome::Bootup.set_and_do({ breadcrumbs: ARGV[0], arguments: arguments }, spawn_into_console)
 
-if @direct_command && @direct_command_on
-  if @direct_command == "ssh"
-    ## We're SSH'ing directly to a node, indicated by an 'ssh' key at the end of our context breadcrumb
-    # called with, e.g. bcome platform:environment:node_name:ssh
-    @direct_command_on.node.ssh
-  elsif @direct_command == "run"
-    ## We're directly excuting a command on a n node indicated by a 'run' key at the end of our context breadrumb
-    # called with, e.g. bcome platform:environment:node_name:run "command to execute"
-    params = ARGV[(1..(ARGV.size))].join("\s")
-    @direct_command_on.node.run(params)
-   else 
-    @direct_command_on.send(@direct_command)
-  end
-end
+  # close any trailing connections
+  Bcome::Bootup.instance.estate.close_ssh_connections
 
-## We're going to directly execute a command on a node, indicated by the method name  at the end of our context breadcrumb.
-## This works for commands specified in the users own personal framework installation, either from a context command or from an orchestration hook
-## e,g, bcome platform:environment:node_name:bootstrap
-if @context_object && @direct_execute_method
-  if ARGV && ARGV.size > 1
-    arguments = ARGV[1..ARGV.size]
-    @context_object.send(@direct_execute_method, arguments)
-  else
-    @context_object.send(@direct_execute_method) 
-  end
+rescue ::Bcome::Exception::Base
+  puts "bcome terminated unexpectedly".warning
 end

data/lib/bcome.rb

@@ -1,11 +1,20 @@
+require 'irb'
+require 'irb/completion'
 require 'rainbow'
-require 'aws-sdk'
 require 'net/scp'
 require 'net/ssh/proxy/command'
 require 'fog/aws'
 require 'require_all'
 require 'pmap'
+require 'singleton'
+require 'active_support'
+require 'active_support/core_ext'
+require 'pp'
+require 'awesome_print'
 
-require_all "#{File.dirname(__FILE__)}/../lib"
-require_all "#{File.dirname(__FILE__)}/../filters"
+require_all "#{File.dirname(__FILE__)}/../patches"
+require_all "#{File.dirname(__FILE__)}/../lib/objects"
 
+# Load in any user defined orchestration
+path_to_bcome_orchestration_configs = "#{Dir.getwd}/bcome/orchestration"
+require_all path_to_bcome_orchestration_configs if File.directory?(path_to_bcome_orchestration_configs)

data/lib/objects/bcome/version.rb

@@ -0,0 +1,3 @@
+module Bcome
+  VERSION = '1.0.0'.freeze
+end

data/lib/objects/bootup.rb

@@ -0,0 +1,81 @@
+module Bcome
+  class Bootup
+    def self.set_and_do(params, spawn_into_console = true)
+      instance.set(params, spawn_into_console)
+      instance.do
+    rescue Bcome::Exception::Base => e
+      puts e.pretty_display
+    rescue Excon::Error::Socket => e
+      puts "\nNo network access - please check your connection and try again\n".red
+    end
+
+    def self.traverse(breadcrumbs = nil, _spawn_into_console = false)
+      spawn_into_console = false
+      ::Bcome::Bootup.set_and_do({ breadcrumbs: breadcrumbs }, spawn_into_console)
+    end
+
+    include Singleton
+
+    attr_reader :breadcrumbs, :arguments
+
+    def set(params, spawn_into_console = false)
+      @breadcrumbs = params[:breadcrumbs]
+      @arguments = params[:arguments]
+      @spawn_into_console = spawn_into_console
+    end
+
+    def do
+      context = crumbs.empty? ? init_context(estate) : traverse(estate)
+      context
+    end
+
+    def init_context(context)
+      if @spawn_into_console
+        ::Bcome::Workspace.instance.set(context: context)
+      else
+        context
+      end
+    end
+
+    def traverse(_starting_context)
+      starting_context = estate
+      crumbs.each_with_index do |crumb, _index|
+        # Some contexts' resources are loaded dynamically and do not come from the estate config. As we're traversing, we'll need to load
+        # them if necessary
+        starting_context.load_nodes if starting_context.inventory? && !starting_context.nodes_loaded?
+
+        # Attempt to load our next context resource
+        next_context = starting_context.resource_for_identifier(crumb)
+
+        # Our current breadcrumb is not a node, and so we'll attempt to invoke a method call on the previous
+        # e.g. given resource:foo, then invoke 'foo' on 'resource'
+        unless next_context
+          starting_context.invoke(crumb, @arguments)
+          return
+        end
+        starting_context = next_context
+      end
+
+      # Set our workspace to our last context - we're not invoking a method call and so we're entering a console session
+      init_context(starting_context)
+    end
+
+    def estate
+      @estate ||= ::Bcome::Node::Factory.instance.init_tree
+    end
+
+    def parser
+      ::Bcome::Parser::BreadCrumb.new(@breadcrumbs)
+    end
+
+    def crumbs
+      parser.crumbs
+    end
+
+    private
+
+    def teardown!
+      @estate = nil
+    end
+  end
+end

data/lib/objects/command/local.rb

@@ -0,0 +1,40 @@
+require 'open3'
+
+module Bcome::Command
+  class Local
+    def self.run(raw_command, _success_exit_codes = [0])
+      command = new(raw_command)
+      command.syscall
+      command
+    end
+
+    attr_reader :stdout, :stderr
+
+    def initialize(command, success_exit_codes = [0])
+      @command = command
+      @success_exit_codes = success_exit_codes
+    end
+
+    def syscall
+      @stdout, @stderr, @process_status = Open3.capture3(@command)
+    rescue Exception => e
+      @stderr = e.message
+    end
+
+    def exit_code
+      @process_status ? @process_status.exitstatus : 1
+    end
+
+    def is_success?
+      @success_exit_codes.include?(exit_code)
+    end
+
+    def failed?
+      !is_success?
+    end
+
+    def pretty_result
+      is_success? ? "success / exit code: #{exit_code}".success : "failure / exit code: #{exit_code}".error
+    end
+  end
+end

data/lib/objects/config_factory.rb

@@ -0,0 +1,36 @@
+class Bcome::ConfigFactory
+  attr_reader :tree
+
+  def initialize
+    @tree = { views: [] }
+    @collections = []
+  end
+
+  def flattened
+    @tree[:views].first
+  end
+
+  def add_crumbs(crumbs, data)
+    views = @tree
+    number_crumbs = crumbs.size
+
+    crumbs.each_with_index do |crumb, index|
+      is_last_crumb = number_crumbs == (index + 1)
+      if this_view = hash_for_identifier_from_view(crumb, views)
+        views = this_view
+      else
+        this_view = { identifier: crumb }
+        this_view[:views] = [] unless is_last_crumb && data[:type].to_sym == :inventory
+        views[:views] << this_view
+        views = hash_for_identifier_from_view(crumb, views)
+      end
+
+      this_view.merge!(data) if is_last_crumb
+    end
+  end
+
+  def hash_for_identifier_from_view(identifier, views)
+    raise Bcome::Exception::InventoriesCannotHaveSubViews, 'Inventories cannot hold other inventories - invalid network config' unless views.key?(:views)
+    views[:views].select { |v| v[:identifier].to_s == identifier.to_s }.first
+  end
+end

data/lib/objects/driver/base.rb

@@ -0,0 +1,30 @@
+module Bcome::Driver
+  class Base
+    class << self
+      def create_from_config(config)
+        raise Bcome::Exception::InvalidNetworkDriverType, 'Your network configurtion is invalid' unless config.is_a?(Hash)
+        raise Bcome::Exception::InvalidNetworkDriverType, "Missing config parameter 'type'" unless config[:type]
+        config_klass_key = config[:type].to_sym
+        driver_klass = klass_for_type[config_klass_key]
+        raise Bcome::Exception::InvalidNetworkDriverType, config unless driver_klass
+        driver = driver_klass.new(config)
+        driver
+      end
+
+      def klass_for_type
+        {
+          static: ::Bcome::Driver::Static,
+          ec2: ::Bcome::Driver::Ec2
+        }
+      end
+    end
+
+    def initialize(params)
+      @params = params
+    end
+
+    def config
+      @params
+    end
+  end
+end

data/lib/objects/driver/bucket.rb

@@ -0,0 +1,20 @@
+module Bcome::Driver
+  class Bucket
+    include Singleton
+
+    def initialize
+      @drivers = []
+    end
+
+    def driver_for_network_data(network_data)
+      found_driver = @drivers.select { |driver| driver.config == network_data }.first
+      found_driver ? found_driver : create_network_driver(network_data)
+    end
+
+    def create_network_driver(network_data)
+      driver = ::Bcome::Driver::Base.create_from_config(network_data)
+      @drivers << driver
+      driver
+    end
+  end
+end

data/lib/objects/driver/ec2.rb

@@ -0,0 +1,44 @@
+module Bcome::Driver
+  class Ec2 < Bcome::Driver::Base
+    def initialize(*params)
+      super
+      raise Bcome::Exception::Ec2DriverMissingProvisioningRegion, params.inspect unless provisioning_region
+    end
+
+    def fog_client
+      @fog_client ||= get_fog_client
+    end
+
+    def fetch_server_list(filters)
+      servers = unfiltered_server_list.all(filters)
+      servers
+    end
+
+    def unfiltered_server_list
+      @unfiltered_server_list ||= fog_client.servers.all({})
+    end
+
+    def loading
+      fog_client.servers.all({})
+    end
+
+    def credentials_key
+      @params[:credentials_key]
+    end
+
+    def provisioning_region
+      @params[:provisioning_region]
+    end
+
+    protected
+
+    def get_fog_client
+      ::Fog.credential = credentials_key
+      client = ::Fog::Compute.new(
+        provider: 'AWS',
+        region: provisioning_region
+      )
+      client
+    end
+  end
+end

data/lib/objects/driver/static.rb

@@ -0,0 +1,4 @@
+module Bcome::Driver
+  class Static < Bcome::Driver::Base
+  end
+end

data/lib/objects/exception/argument_error_invoking_method_from_command_line.rb

@@ -0,0 +1,7 @@
+module ::Bcome::Exception
+  class ArgumentErrorInvokingMethodFromCommmandLine < ::Bcome::Exception::Base
+    def message_prefix
+      'Argument error invoking method. Please refer to documentation for invoking this command from the command line: '
+    end
+  end
+end

data/lib/objects/exception/base.rb

@@ -0,0 +1,15 @@
+module Bcome::Exception
+  class Base < RuntimeError
+    def initialize(message_suffix = nil)
+      @message_suffix = message_suffix
+    end
+
+    def message
+      "#{message_prefix}#{@message_suffix ? ": #{@message_suffix}" : ''}"
+    end
+
+    def pretty_display
+      puts "\n#{message}\n".error
+    end
+  end
+end

data/lib/objects/exception/can_only_subselect_on_inventory.rb

@@ -0,0 +1,7 @@
+module ::Bcome::Exception
+  class CanOnlySubselectOnInventory < ::Bcome::Exception::Base
+    def message_prefix
+      'Invalid subselect - can only subselect on another inventory'
+    end
+  end
+end

data/lib/objects/exception/cannot_find_internal_registry_klass.rb

@@ -0,0 +1,7 @@
+module ::Bcome::Exception
+  class CannotFindInternalRegistryKlass < ::Bcome::Exception::Base
+    def message_prefix
+      'Could not initialize orchestration method '
+    end
+  end
+end

data/lib/objects/exception/cannot_find_subselection_parent.rb

@@ -0,0 +1,7 @@
+module ::Bcome::Exception
+  class CannotFindSubselectionParent < ::Bcome::Exception::Base
+    def message_prefix
+      'Cannot find subselection parent'
+    end
+  end
+end

data/lib/objects/exception/cant_find_key_in_cloud_tags.rb

@@ -0,0 +1,7 @@
+module ::Bcome::Exception
+  class CantFindKeyInCloudTags < ::Bcome::Exception::Base
+    def message_prefix
+      'Cannot find key in cloud tags '
+    end
+  end
+end

data/lib/objects/exception/cant_find_key_in_metadata.rb

@@ -0,0 +1,7 @@
+module ::Bcome::Exception
+  class CantFindKeyInMetadata < ::Bcome::Exception::Base
+    def message_prefix
+      'Cannot find key in bcome metadata '
+    end
+  end
+end

data/lib/objects/exception/cant_find_proxy_host_by_identifier.rb

@@ -0,0 +1,7 @@
+module ::Bcome::Exception
+  class CantFindProxyHostByIdentifier < ::Bcome::Exception::Base
+    def message_prefix
+      'Can\'t find proxy host by identifier'
+    end
+  end
+end