moose-inventory 0.1.0

data/lib/moose/inventory/cli/host_addvar.rb

@@ -0,0 +1,76 @@
+require 'thor'
+require 'json'
+
+require_relative './formatter.rb'
+require_relative '../db/exceptions.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # implementation of the "host addvar" method of the CLI
+      class Host
+        #==========================
+        desc 'addvar', 'Add a variable to the host'
+        # rubocop:disable Metrics/LineLength
+        def addvar(*args) # rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/MethodLength, Metrics/PerceivedComplexity
+          # rubocop:enable Metrics/LineLength
+          if args.length < 2
+            abort('ERROR: Wrong number of arguments, '\
+                  "#{args.length} for 2 or more.")
+          end
+
+          # Convenience
+          db = Moose::Inventory::DB
+          fmt = Moose::Inventory::Cli::Formatter
+
+          # Arguments
+          name = args[0].downcase
+          vars = args.slice(1, args.length - 1).uniq
+
+          # Transaction
+          db.transaction do # Transaction start
+            puts "Add variables '#{vars.join(",")}' to host '#{name}':"
+            fmt.puts 2,"- retrieve host '#{name}'..."
+            host = db.models[:host].find(name: name)
+            if host.nil?
+              fail db.exceptions[:moose],
+                   "The host '#{name}' does not exist."
+            end
+            fmt.puts 4, '- OK'
+
+            hostvars_ds = host.hostvars_dataset
+            vars.each do |v|
+              fmt.puts 2, "- add variable '#{v}'..."
+              vararray = v.split('=')
+              if v.start_with?('=') ||  v.end_with?('=') || vararray.length != 2
+                fail db.exceptions[:moose],
+                     "Incorrect format in '{#{v}}'. Expected 'key=value'."
+              end
+                
+              # Check against existing associations
+              hostvar = hostvars_ds[name: vararray[0]]
+              if !hostvar.nil?
+                unless hostvar[:value] == vararray[1]
+                  fmt.puts 4, '- already exists, applying as an update...'
+                  update = db.models[:hostvar].find(id: hostvar[:id])
+                  update[:value] = vararray[1]
+                  update.save
+                end
+              else
+                # hostvar doesn't exist, so create and associate
+                hostvar = db.models[:hostvar].create(name: vararray[0],
+                                                      value: vararray[1])
+                host.add_hostvar(hostvar)
+              end
+              fmt.puts 4, '- OK'
+            end
+            fmt.puts 2, '- all OK'
+          end # Transaction end
+
+          puts 'Succeeded.'
+        end
+      end
+    end
+  end
+end

data/lib/moose/inventory/cli/host_get.rb

@@ -0,0 +1,59 @@
+require 'thor'
+require 'json'
+
+require_relative './formatter.rb'
+require_relative '../db/exceptions.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Class implementing the "host get" method of the CLI
+      class Host
+        require_relative 'host_add'
+
+        #==========================
+        desc 'get HOST_1 [HOST_2 ...]',
+             'Get hosts HOST_n from the inventory'
+        def get(*argv) # rubocop:disable Metrics/AbcSize
+          if argv.length < 1
+            abort('ERROR: Wrong number of arguments, '\
+              "#{argv.length} for 1 or more")
+          end
+
+          # Convenience
+          db = Moose::Inventory::DB
+          fmt = Moose::Inventory::Cli::Formatter
+          
+          # Arguments
+          names = argv.uniq.map(&:downcase)
+
+          # Process
+          results = {}
+          names.each do |name| # rubocop:disable Style/Next
+            host = db.models[:host].find(name: name)
+
+            unless host.nil?
+              groups = host.groups_dataset.map(:name)
+
+              hostvars = {}
+              host.hostvars_dataset.each do |hv|
+                hostvars[hv[:name].to_sym] = hv[:value]
+              end
+
+              results[host[:name].to_sym] = {}
+              unless groups.length == 0
+                results[host[:name].to_sym][:groups] = groups
+              end
+              unless hostvars.length == 0
+                results[host[:name].to_sym][:hostvars] = hostvars
+              end
+            end
+          end
+
+          fmt.dump(results)
+        end
+      end
+    end
+  end
+end

data/lib/moose/inventory/cli/host_list.rb

@@ -0,0 +1,40 @@
+require 'thor'
+require 'json'
+
+require_relative './formatter.rb'
+require_relative '../db/exceptions.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Implementation of the "host list" method of the CLI
+      class Host
+        desc 'list', 'List the contents of the inventory by host'
+        def list # rubocop:disable Metrics/AbcSize
+          # Convenience
+          db = Moose::Inventory::DB
+          fmt = Moose::Inventory::Cli::Formatter
+
+          # Process
+          results = {}
+          db.models[:host].all.each do |host|
+            groups = host.groups_dataset.map(:name)
+            results[host[:name].to_sym] = {}
+            results[host[:name].to_sym][:groups] = groups
+
+            hostvars = {}
+            host.hostvars_dataset.each do |hv| 
+              hostvars[hv[:name].to_sym] = hv[:value] 
+            end
+
+            unless hostvars.length == 0
+              results[host[:name].to_sym][:hostvars] = hostvars
+            end
+          end
+          fmt.dump(results)
+        end
+      end
+    end
+  end
+end

data/lib/moose/inventory/cli/host_rm.rb

@@ -0,0 +1,62 @@
+require 'thor'
+require 'json'
+
+require_relative './formatter.rb'
+require_relative '../db/exceptions.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # implementation of the "host rm" method of the CLI
+      class Host
+        #==========================
+        desc 'rm HOSTNAME_1 [HOSTNAME_2 ...]',
+             'Remove hosts HOSTNAME_n from the inventory'
+        def rm(*argv) # rubocop:disable Metrics/AbcSize
+          #
+          # Sanity
+          if argv.length < 1
+            abort('ERROR: Wrong number of arguments, '\
+              "#{argv.length} for 1 or more.")
+          end
+
+          # Convenience
+          db = Moose::Inventory::DB
+          fmt = Moose::Inventory::Cli::Formatter
+          
+          # Arguments
+          names = argv.uniq.map(&:downcase)
+
+          # Transaction
+          warn_count = 0
+          db.transaction do # Transaction start
+            names.each do |name|
+              puts "Remove host '#{name}':"
+              fmt.puts 2, "- Retrieve host '#{name}'..."
+              host = db.models[:host].find(name: name)
+              if host.nil?
+                warn_count += 1
+                fmt.warn "Host '#{name}' does not exist, skipping.\n"
+                fmt.puts 4, "- No such host, skipping."
+              end
+              fmt.puts 4, "- OK" 
+              unless host.nil?
+                fmt.puts 2, "- Destroy host '#{name}'..."
+                host.remove_all_groups
+                host.destroy
+                fmt.puts 4, "- OK"
+              end
+              fmt.puts 2, "- All OK"
+            end
+          end # Transaction end
+          if warn_count == 0 
+            puts "Succeeded."
+          else
+            puts "Succeeded, with warnings."
+          end
+        end
+      end
+    end
+  end
+end

data/lib/moose/inventory/cli/host_rmgroup.rb

@@ -0,0 +1,80 @@
+require 'thor'
+require 'json'
+
+require_relative './formatter.rb'
+require_relative '../db/exceptions.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # implementation the "host rmgroup" methods of the CLI
+      class Host
+        #==========================
+        desc 'rmgroup HOSTNAME GROUPNAME [GROUPNAME ...]',
+             'dissociation the host from a group'
+        # rubocop:disable Metrics/LineLength
+        def rmgroup(*args) # rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/MethodLength, Metrics/PerceivedComplexity
+          # rubocop:enable Metrics/LineLength
+          if args.length < 2
+            abort('ERROR: Wrong number of arguments, '\
+                  "#{args.length} for 2 or more.")
+          end
+          
+          # Convenience
+          db = Moose::Inventory::DB
+          fmt = Moose::Inventory::Cli::Formatter
+
+          # arguments
+          name   = args[0].downcase
+          groups = args.slice(1, args.length - 1).uniq.map(&:downcase)
+
+          # Sanity
+          if groups.include?('ungrouped')
+            abort 'ERROR: Cannot manually manipulate the automatic '\
+              'group \'ungrouped\'.'
+          end
+          
+          # Transaction
+          db.transaction do # Transaction start
+            puts "Dissociate host '#{name}' from groups '#{groups.join(',')}':" 
+            fmt.puts 2, "- Retrieve host '#{name}'..."
+            host = db.models[:host].find(name: name)
+            if host.nil?
+              fail db.exceptions[:moose],
+                   "The host '#{name}' was not found in the database."
+            end
+            fmt.puts 4, '- OK'
+
+            # dissociate host from the groups
+            groups_ds = host.groups_dataset
+            groups.each do |g|
+              fmt.puts 2, "- Remove association {host:#{name} <-> group:#{g}}..."
+
+              # Check against existing associations
+              if groups_ds[name: g].nil?
+                fmt.warn "Association {host:#{name} <-> group:#{g}} doesn't exist, skipping.\n"
+                fmt.puts 4,  "- Doesn't exist, skipping."
+              else
+                group = db.models[:group].find(name: g)
+                host.remove_group(group) unless group.nil?
+              end
+              fmt.puts 4, '- OK'
+            end
+
+            # Handle 'ungrouped' group automation
+            if host.groups_dataset.count == 0
+              fmt.puts 2, '- Add automatic association '\
+                "{host:#{name} <-> group:ungrouped}..."
+              ungrouped  = db.models[:group].find_or_create(name: 'ungrouped')
+              host.add_group(ungrouped) unless ungrouped.nil?
+              fmt.puts 4, '- OK'
+            end
+            fmt.puts 2, '- All OK'
+          end # End transaction
+          puts 'Succeeded'
+        end
+      end
+    end
+  end
+end

data/lib/moose/inventory/cli/host_rmvar.rb

@@ -0,0 +1,69 @@
+require 'thor'
+require 'json'
+
+require_relative './formatter.rb'
+require_relative '../db/exceptions.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # implementation of the "host rmvar" method of the CLI
+      class Host
+        #==========================
+        desc 'rmvar', 'Remove a variable from the host'
+        # rubocop:disable Metrics/LineLength
+        def rmvar(*args) # rubocop:disable Metrics/CyclomaticComplexity, Metrics/AbcSize
+          # rubocop:enableMetrics/LineLength
+          if args.length < 2
+            abort('ERROR: Wrong number of arguments, ' \
+                  "#{args.length} for 2 or more.")
+          end
+
+          # Convenience
+          db  = Moose::Inventory::DB
+          fmt = Moose::Inventory::Cli::Formatter
+
+          # Arguments
+          name = args[0].downcase
+          vars = args.slice(1, args.length - 1).uniq
+
+          # Transaction
+          db.transaction do # Transaction start
+            puts "Remove variable(s) '#{vars.join(",")}' from host '#{name}':"
+            
+            fmt.puts 2, "- retrieve host '#{name}'..."
+            host = db.models[:host].find(name: name)
+            if host.nil?
+              fail db.exceptions[:moose],
+                   "The host '#{name}' does not exist."
+            end
+            fmt.puts 4, '- OK'
+
+            hostvars_ds = host.hostvars_dataset
+            vars.each do |v|
+              fmt.puts 2, "- remove variable '#{v}'..."
+              vararray = v.split('=')
+              if v.start_with?('=') || v.scan('=').count > 1
+                fail db.exceptions[:moose],
+                     "Incorrect format in {#{v}}. " \
+                     'Expected \'key\' or \'key=value\'.'
+              end
+
+              # Check against existing associations
+              hostvar = hostvars_ds[name: vararray[0]]
+              unless hostvar.nil?
+                # remove the association
+                host.remove_hostvar(hostvar)
+                hostvar.destroy
+              end
+              fmt.puts 4, '- OK'
+            end
+            fmt.puts 2, '- all OK'
+          end # Transaction end
+          puts 'Succeeded.'
+        end
+      end
+    end
+  end
+end

data/lib/moose/inventory/config/config.rb

@@ -0,0 +1,169 @@
+# Configuration
+
+require 'yaml'
+
+module Moose
+  module Inventory
+    ##
+    # This Modules manages application-wide configuration options
+    module Config
+      # rubocop:disable Style/ModuleFunction
+      extend self
+      # rubocop:enable Style/ModuleFunction
+
+      @_argv     = []
+      @_confopts = {}
+      @_settings = {}
+
+      attr_reader :_argv
+      attr_reader :_confopts
+      attr_reader :_settings
+
+      #----------------------
+      def self.init(args)
+        @_argv = args.dup
+
+        top_level_args
+        # ansible_args
+        resolve_config_file
+        load
+      end
+
+      #----------------------
+      def self.top_level_args
+        # Certain top level flags affect the configuration.
+        # --config FILE  => sets the configuration file to be used.
+        #                   Default is to search standard locations.
+        # --env ENV      => sets the section to be used as the configuration.
+        #                   Defaults to ""
+        # --format FORMAT=> See formatter for supported types.
+        #                   Defaults to json.
+
+        @_confopts = { env: '', format: 'json', trace: false }
+
+        # The following are a O(n^m) approach.  TODO: O(n) version?
+          
+        # Check for two-part flags   
+        %w(env config format).each do |var|
+          @_argv.each_with_index do |val, index|
+            next if val != "--#{var}"
+            @_confopts[var.to_sym] = @_argv[index + 1]
+            1.downto(0) { |offset| @_argv.delete_at(index + offset) }
+            break
+          end
+        end
+        
+        # Check for one-part flags
+        %w(trace).each do |var|
+          @_argv.each_with_index do |val, index|
+            next if val != "--#{var}"
+            @_confopts[var.to_sym] = true
+            @_argv.delete_at(index)
+            break
+          end
+        end
+        
+      end
+
+      #----------------------
+      def self.ansible_args  # rubocop:disable Metrics/AbcSize
+        # Handle Ansible flags
+        # --hosts           => list all hosts
+        # --hosts HOSTNAME  => get host name
+        # --groups          => list all groups
+
+        if @_argv[0] == '--hosts'
+          host = @_argv[1]
+          if !host.nil?
+            @_argv.clear
+            ['host', 'get', "#{host}", '--ansiblestyle'].each do |arg|
+              @_argv << arg
+            end
+          else
+            @_argv.clear
+            ['host', 'list', '--ansiblestyle'].each do |arg|
+              @_argv << arg
+            end
+          end
+        elsif @_argv[0] == '--groups'
+          ['group', 'list', '--ansiblestyle'].each do |arg|
+            @_argv << arg
+          end
+        end
+      end
+
+      #----------------------
+      def self.resolve_config_file # rubocop:disable Metrics/AbcSize
+        if ! @_confopts[:config].nil?
+          path = File.expand_path(@_confopts[:config])
+          if File.exist?(path)
+            @_confopts[:config] = path
+          else
+            fail("The configuration file #{path} does not exist")
+          end
+        else
+          possibles = ['./.moose-tools/inventory/config',
+                       '~/.moose-tools/inventory/config',
+                       '~/local/etc/moose-tools/inventory/config',
+                       '/etc/moose-tools/inventory/config'
+                      ]
+          possibles.each do |f|
+            file = File.expand_path(f)
+            @_confopts[:config] = file if File.exist?(file)
+          end
+        end
+
+        if @_confopts[:config].nil?
+          fail('No configuration either given or found in standard locations.')
+        end
+      end
+
+      #----------------------
+      def self.symbolize_keys(hash)
+        # rubocop:disable Style/EachWithObject
+        hash.inject({}) do |result, (key, value)|
+          new_key = case key
+                    when String then key.to_sym
+                    else key
+                    end
+          new_value = case value
+                      when Hash then symbolize_keys(value)
+                      else value
+                      end
+          result[new_key] = new_value
+          result
+        end
+        # rubocop:enable Style/EachWithObject
+      end
+
+      #----------------------
+      # rubocop:disable PerceivedComplexity
+      # rubocop:disable Metrics/CyclomaticComplexity, Metrics/AbcSize
+      def self.load
+        newsets = symbolize_keys(YAML.load_file(@_confopts[:config]))
+
+        path = @_confopts[:config]
+
+        # Get the "general" section
+        @_settings[:general] = newsets[:general]
+        @_settings[:general].nil? && fail("Missing 'general' root in #{path}")
+
+        # Get the config for the correct environment
+
+        if @_confopts[:env] && !@_confopts[:env].empty?
+          env = @_confopts[:env]
+          @_settings[:config] = newsets[@_confopts[:env].to_sym]
+        else
+          env  = @_settings[:general][:defaultenv]
+          (env.nil? || env.empty?) && fail("No defaultenv set in #{path}")
+          @_settings[:config] = newsets[env.to_sym]
+        end
+
+        @_settings[:config].nil? && fail("Missing '#{env}' root in #{path}")
+
+        # And now we should have a valid config stuffed into @_options[:config]
+      end
+      # rubocop:enable Metrics/CyclomaticComplexity, Metrics/AbcSize
+    end
+  end
+end