bcome 0.7.0 → 1.0.0

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

@@ -0,0 +1,25 @@
+module Bcome::Node::Resources
+  class Inventory < Bcome::Node::Resources::Base
+    def <<(node)
+      if existing_node = for_identifier(node.identifier)
+        if existing_node.static_server? && node.dynamic_server?
+          # We've got a duplicate, but we'll treat the remote node as authoritative
+          # We remove the static server from our selection
+          @nodes.delete(existing_node)
+        else
+          exception_message = "#{node.identifier} is not unique within namespace #{node.parent.namespace}"
+          raise Bcome::Exception::NodeIdentifiersMustBeUnique, exception_message
+        end
+      end
+      @nodes << node
+    end
+
+    def dynamic_nodes
+      active.select(&:dynamic_server?)
+    end
+
+    def unset!
+      @nodes = []
+    end
+  end
+end

data/lib/objects/node/resources/sub_inventory.rb

@@ -0,0 +1,56 @@
+module Bcome::Node::Resources
+  class SubselectInventory < Bcome::Node::Resources::Inventory
+    def initialize(config)
+      @config = config
+      super
+      run_subselect
+    end
+
+    def run_subselect
+      parent_inventory.load_nodes unless parent_inventory.nodes_loaded?
+      new_set = parent_inventory.resources.nodes
+
+      # Filter by ec2_filters
+      new_set = filter_by_tags(new_set)
+
+      @nodes = new_set
+    end
+
+    def update_nodes(inventory)
+      new_set = []
+
+      @nodes.collect do |node|
+        new_node = node.dup_with_new_parent(inventory)
+        if inventory.override_server_identifier?
+          new_node.identifier =~ /#{inventory.override_identifier}/
+          new_node.update_identifier(Regexp.last_match(1)) if Regexp.last_match(1)
+        end
+        new_set << new_node
+      end
+      @nodes = new_set
+    end
+
+    def filter_by_tags(nodes)
+      tag_filters.each do |key, values|
+        nodes = nodes.select { |node| node.has_tagged_value?(key, values) }
+      end
+      nodes
+    end
+
+    def parent_crumb
+      @config[:parent_crumb]
+    end
+
+    def filters
+      @config[:filters]
+    end
+
+    def tag_filters
+      filters[:by_tag] ? filters[:by_tag] : {}
+    end
+
+    def parent_inventory
+      @config[:parent_inventory]
+    end
+  end
+end

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

@@ -0,0 +1,218 @@
+module Bcome::Node::Server
+  class Base < Bcome::Node::Base
+    
+    attr_reader :origin_object_id
+
+    def initialize(*params)
+      super
+      # Set the object_id - sub inventories dup servers into new collections. This allows us to spot duplicates when interacting with collections
+      @origin_object_id = object_id
+      @bootstrap = false
+    end
+
+    def bootstrap?
+      @bootstrap ? true : false
+    end  
+
+    def toggle_bootstrap(set_to = (@bootstrap ? false : true))
+      @bootstrap = set_to
+      puts "Bootstrap is #{bootstrap? ? "on" : "off" }".informational
+    end
+
+    def dup_with_new_parent(new_parent)
+      new_node = self.clone
+      new_node.update_parent(new_parent)
+      new_node
+    end  
+
+    def update_parent(new_parent)
+      @parent = new_parent
+    end
+
+    def tags
+      data_print_from_hash(cloud_tags.data, "Tags")
+    end
+  
+    def cloud_tags
+      @generated_tags ||= do_generate_cloud_tags
+    end
+
+    def has_tagged_value?(key, values)
+      matchers = { :key => key, :values => values }
+      cloud_tags.has_key_and_value?(matchers)
+    end
+
+    def update_identifier(new_identifier)
+      @identifier = new_identifier
+    end
+
+    def do_generate_cloud_tags
+      raise "Should be overidden"
+    end
+
+    def type
+      "server"
+    end
+
+    def machines
+      [self]
+    end
+
+    def server?
+      true
+    end
+
+    def requires_description?
+      false
+    end
+
+    def requires_type?
+      false
+    end
+
+    def enabled_menu_items
+      (super + [:get, :ssh, :tags]) - [:enable, :disable, :enable!, :disable!]
+    end
+
+    def menu_items
+      base_items = super.dup
+      base_items[:tags] = {
+        description: "print out remote EC2 tags"
+      }
+      base_items[:ssh] = {
+        description: "initiate an ssh connection to this server",
+      }
+      base_items[:get] = {
+        description: "Download a file or directory",
+        console_only: false,
+        usage: "get \"/remote/path\", \"/local/path\"",
+        terminal_usage: "get \"/remote/path\" \"/local/path\""
+      }
+
+
+      base_items
+    end
+
+    def open_ssh_connection
+      ssh_driver.ssh_connection  
+    end
+
+    def close_ssh_connection
+      ssh_driver.close_ssh_connection
+    end
+
+    def has_ssh_connection?
+      ssh_driver.has_open_ssh_con?
+    end
+
+    def has_no_ssh_connection?
+      !has_ssh_connection?
+    end
+ 
+    def ssh
+      ssh_driver.do_ssh
+    end
+
+    def execute_script(script_name)
+      command_result = ::Bcome::Ssh::ScriptExec.execute(self, script_name)
+      return command_result
+    end
+
+    def rsync(local_path, remote_path)
+      ssh_driver.rsync(local_path, remote_path)
+    end
+  
+    def put(local_path, remote_path)
+      ssh_driver.put(local_path, remote_path)
+    end
+
+    def get(remote_path, local_path)
+      ssh_driver.get(remote_path, local_path)
+    end
+
+    def ls
+      puts "\n" + visual_hierarchy.hierarchy + "\n"
+      puts pretty_description
+    end
+    alias :lsa :ls
+
+    def ping
+      ping_result = ssh_driver.ping
+      print_ping_result(ping_result)
+    end
+
+    def print_ping_result(ping_result = { success: true })
+      result = {
+        namespace => {
+          "connection" => ping_result[:success] ? "success" : "failed",
+          "ssh_config" => ssh_driver.pretty_config_details
+        }
+      }
+
+      unless ping_result[:success]
+        result[namespace]["error"] = ping_result[:error].message
+      end
+
+      colour = ping_result[:success] ? :green : :red
+
+      ap result, {
+        :color => {
+           hash:  colour,
+           symbol: colour,
+           string: colour,
+           keyword: colour,
+           variable: colour,
+           array: colour
+        }
+      }
+    end
+
+    def list_attributes
+      attribs = {
+        "identifier": :identifier,
+        "internal ip": :internal_ip_address,
+        "public ip": :public_ip_address,
+      }
+
+      attribs.merge!("description": :description ) if has_description?
+      attribs
+    end
+
+    def cache_data
+      d = { identifier: identifier }
+      d[:internal_ip_address] = internal_ip_address if internal_ip_address
+      d[:public_ip_address] = public_ip_address if public_ip_address
+      d[:description] = description if description
+      d[:cloud_tags] = cloud_tags 
+      d
+    end
+
+    def do_run(raw_commands)
+      raw_commands = raw_commands.is_a?(String) ? [raw_commands] : raw_commands
+      commands = raw_commands.collect{|raw_command| ::Bcome::Ssh::Command.new({ :node => self, :raw => raw_command }) }
+      command_exec = ::Bcome::Ssh::CommandExec.new(commands)
+      command_exec.execute!
+      commands.each {|c| c.unset_node }
+      return commands
+    end
+
+    def run(*raw_commands)
+      raise ::Bcome::Exception::MethodInvocationRequiresParameter.new "Please specify commands when invoking 'run'" if raw_commands.empty?
+      commands = do_run(raw_commands)
+      commands
+    end
+
+    def has_description?
+      !@description.nil?
+    end
+
+    def static_server?
+      false
+    end
+
+    def dynamic_server?
+      !static_server?
+    end
+
+  end  
+end

data/lib/objects/node/server/dynamic.rb

@@ -0,0 +1,39 @@
+module Bcome::Node::Server
+  class Dynamic < Bcome::Node::Server::Base
+    class << self
+      def to_s
+        'dynamic server'
+      end
+
+      def new_from_fog_instance(fog_instance, parent)
+        identifier = fog_instance.tags['Name']
+
+        if parent.override_server_identifier?
+          identifier =~ /#{parent.override_identifier}/
+          identifier = Regexp.last_match(1) if Regexp.last_match(1)
+        end
+
+        params = {
+          identifier: identifier,
+          internal_ip_address: fog_instance.private_ip_address,
+          public_ip_address: fog_instance.public_ip_address,
+          role: fog_instance.tags['function'],
+          description: "EC2 server - #{identifier}",
+          ec2_server: fog_instance
+        }
+
+        new(parent: parent,
+            views: params)
+      end
+    end
+
+    def do_generate_cloud_tags
+      raw_tags = ec2_server ? ec2_server.tags.deep_symbolize_keys : {}
+      ::Bcome::Node::Meta::Cloud.new(raw_tags)
+    end
+
+    def ec2_server
+      views[:ec2_server]
+    end
+  end
+end

data/lib/objects/node/server/static.rb

@@ -0,0 +1,31 @@
+module Bcome::Node::Server
+  class Static < Bcome::Node::Server::Base
+    def self.to_s
+      'static server'
+    end
+
+    def initialize(params)
+      config = params[:views]
+      @identifier = config[:identifier]
+      @public_ip_address = config[:public_ip_address]
+      @internal_ip_address = config[:internal_ip_address]
+      @cloud_tags = config[:cloud_tags]
+      verify_we_have_at_least_one_interface(config)
+      super
+    end
+
+    attr_reader :cloud_tags
+
+    def verify_we_have_at_least_one_interface(config)
+      raise Bcome::Exception::MissingIpaddressOnServer, config unless has_at_least_one_interface?
+    end
+
+    def has_at_least_one_interface?
+      @public_ip_address || @internal_ip_address
+    end
+
+    def static_server?
+      true
+    end
+  end
+end

data/lib/objects/orchestration/base.rb

@@ -0,0 +1,13 @@
+module Bcome::Orchestration
+  class Base
+    def initialize(node, arguments)
+      @node = node
+      @arguments = arguments
+    end
+
+    def do_execute
+      raise Bcome::Exception::MissingExecuteOnRegistryObject, self.class.to_s unless respond_to?(:execute)
+      execute
+    end
+  end
+end

data/lib/objects/orchestrator.rb

@@ -0,0 +1,24 @@
+class ::Bcome::Orchestrator
+  include ::Singleton
+
+  attr_reader :context
+
+  def initialize
+    @silence = false
+  end
+
+  def silence_command_output!
+    @silence = true
+  end
+
+  def command_output_silenced?
+    @silence == true
+  end
+
+  def get(breadcrumb = nil)
+    context = ::Bcome::Bootup.traverse(breadcrumb)
+    raise Bcome::Exception::NoNodeFoundForBreadcrumb, breadcrumb unless context
+    context.load_nodes if context.inventory? && !context.nodes_loaded?
+    context
+  end
+end

data/lib/objects/parser/bread_crumb.rb

@@ -0,0 +1,31 @@
+module Bcome::Parser
+  class BreadCrumb
+    attr_reader :crumbs
+
+    class << self
+      def parse(raw_crumbs)
+        parser = new(raw_crumbs)
+        parser.parse!
+      end
+    end
+
+    def initialize(raw_crumbs)
+      @raw_crumbs = raw_crumbs.to_s
+
+      if @raw_crumbs.empty?
+        @crumbs = []
+      else
+        validate!
+        parse!
+      end
+    end
+
+    def parse!
+      @crumbs ||= @raw_crumbs.split(':')
+    end
+
+    def validate!
+      raise Bcome::Exception::InvalidBcomeBreadcrumb.new "- letters, numbers & underscores only" unless @raw_crumbs =~ /^([a-z0-9A-Z_]+)(:\s*[a-z0-9A-Z_]+)*:?$/i
+    end
+  end
+end