moose-inventory 0.1.0

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

@@ -0,0 +1,30 @@
+require 'thor'
+require_relative '../version.rb'
+require_relative '../config/config.rb'
+require_relative './group.rb'
+require_relative './host.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # TODO: Documentation
+      class Application < Thor
+        desc 'version', 'Get the code version'
+        def version
+          puts "Version #{Moose::Inventory::VERSION}"
+        end
+
+        desc 'group ACTION',
+             'Manipulate groups in the inventory. ' \
+             'ACTION can be add, rm, addhost, rmhost, addvar, rmvar'
+        subcommand 'group', Moose::Inventory::Cli::Group
+
+        desc 'host ACTION',
+             'Manipulate hosts in the inventory. ' \
+             'ACTION can be add, rm, addgroup, rmgroup, addvar, rmvar'
+        subcommand 'host', Moose::Inventory::Cli::Host
+      end
+    end
+  end
+end

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

@@ -0,0 +1,92 @@
+require 'json'
+require 'yaml'
+require 'indentation'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # TODO: Documentation
+      module Formatter
+        # rubocop:disable Style/ModuleFunction
+        extend self
+        # rubocop:enable Style/ModuleFunction
+
+        def self.dump(arg, format = nil)
+          out(arg, format)
+        end
+        
+        def self.out(arg, format = nil)
+          return if arg.nil?
+
+          if format.nil?
+            format = Moose::Inventory::Config._confopts[:format].downcase
+          end
+          
+          case format
+          when 'yaml','y'
+            $stdout.puts arg.to_yaml
+
+          when 'prettyjson','pjson','p'
+            $stdout.puts JSON.pretty_generate(arg)
+
+          when 'json','j'
+            $stdout.puts arg.to_json
+
+          else
+            abort("Output format '#{format}' is not yet supported.")
+          end
+        end
+        
+        #---------------
+        attr_accessor :indent
+
+        def reset_indent
+          @indent = 2
+        end
+        
+        def puts(indent, msg, stream='STDOUT')
+          case stream
+          when 'STDOUT'
+            $stdout.puts msg.indent(indent)
+          when 'STDERR'
+            $stderr.puts msg.indent(indent)
+          else
+            abort("Output stream '#{stream}' is not known.")
+          end
+        end
+        
+        def print(indent, msg, stream='STDOUT')
+          case stream
+          when 'STDOUT'
+            $stdout.print msg.indent(indent)
+          when 'STDERR'
+            $stderr.print msg.indent(indent)
+          else
+            abort("Output stream '#{stream}' is not known.")
+          end
+        end
+
+        def info(indent, msg, stream='STDOUT')
+          case stream
+          when 'STDOUT'
+            $stdout.print "INFO: {msg}"
+          when 'STDERR'
+            $stderr.print "INFO: {msg}"
+          else
+            abort("Output stream '#{stream}' is not known.")
+          end
+        end
+        
+        def warn(msg)
+            $stderr.print "WARNING: #{msg}"
+        end
+
+        def error(msg)
+            $stderr.print "ERROR: #{msg}"
+        end
+                        
+      end
+    end
+  end
+end

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

@@ -0,0 +1,23 @@
+require 'thor'
+require_relative './formatter.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Class implementing the "group" methods of the CLI
+      class Group < Thor 
+        require_relative 'group_add'
+        require_relative 'group_get'
+        require_relative 'group_list'
+        require_relative 'group_rm'
+        require_relative 'group_addchild'
+        require_relative 'group_rmchild'
+        require_relative 'group_addhost'
+        require_relative 'group_rmhost'
+        require_relative 'group_addvar'
+        require_relative 'group_rmvar'
+      end
+    end
+  end
+end

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

@@ -0,0 +1,98 @@
+require 'thor'
+require_relative './formatter.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Implementation of the "group add" method of the CLI
+      class Group
+        #==========================
+        desc 'add NAME', 'Add a group NAME to the inventory'
+        option :hosts
+        # rubocop:disable Metrics/LineLength
+        def add(*argv) # rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity
+          # rubocop:enable Metrics/LineLength
+          if argv.length < 1
+            abort("ERROR: Wrong number of arguments, #{argv.length} for 1 or more.")
+          end
+
+          # Arguments
+          names = argv.uniq.map(&:downcase)
+          options[:hosts] = '' if options[:hosts].nil?
+          hosts = options[:hosts].downcase.split(',').uniq
+
+          # sanity
+          if names.include?('ungrouped')
+            abort("ERROR: Cannot manually manipulate the automatic group 'ungrouped'\n")
+          end
+
+            # Convenience
+            db    = Moose::Inventory::DB
+            fmt = Moose::Inventory::Cli::Formatter
+          
+          # Transaction
+          warn_count = 0
+          db.transaction do # Transaction start
+            names.each do |name|
+              # Add the group
+              puts "Add group '#{name}':"
+              group = db.models[:group].find(name: name)
+              hosts_ds = nil
+              fmt.puts 2, "- create group..."
+              if group.nil?
+                group = db.models[:group].create(name: name)
+                fmt.puts 4, '- OK'
+              else
+                warn_count += 1
+                fmt.warn "Group '#{name}' already exists, skipping creation.\n"
+                fmt.puts 4, "- already exists, skipping."
+                hosts_ds = group.hosts_dataset
+                fmt.puts 4, '- OK'
+              end
+              
+              # Associate with hosts
+              hosts.each do |h|
+                next if h.nil? || h.empty?
+                fmt.puts 2, "- add association {group:#{name} <-> host:#{ h }}..."
+                host = db.models[:host].find(name: h)
+                if host.nil?
+                  warn_count += 1
+                  fmt.warn "Host '#{h}' doesn't exist, but will be created.\n"
+                  fmt.puts 4, "- host doesn't exist, creating now..."
+                  host = db.models[:host].create(name: h)
+                  fmt.puts 6, "- OK"
+                end
+                if !hosts_ds.nil? && !hosts_ds[name: h].nil?
+                  warn_count += 1
+                  fmt.warn "Association {group:#{name} <-> host:#{ h }}"\
+                    " already exists, skipping creation.\n"
+                    fmt.puts 4, "- already exists, skipping."
+                else  
+                  group.add_host(host)
+                end
+                fmt.puts 4, '- OK'
+  
+                # Handle the host's automatic 'ungrouped' group
+                ungrouped = host.groups_dataset[name: 'ungrouped']
+                unless ungrouped.nil?
+                  fmt.puts 2, "- remove automatic association {group:ungrouped"\
+                    " <-> host:#{ h }}..."
+                  host.remove_group( ungrouped ) unless ungrouped.nil?
+                  fmt.puts 4, '- OK'
+                end
+              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/group_addchild.rb

@@ -0,0 +1,21 @@
+require 'thor'
+require_relative './formatter.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Implemention of the "group addchild" methods of the CLI
+      class Group < Thor # rubocop:disable ClassLength
+        #==========================
+        desc 'addchild [options] NAME CHILDNAME',
+             'Associate a child-group CHILDNAME with the group NAME'
+        option :allowcreate, type: :boolean
+        def addchild
+          abort("The 'groups addchild GROUP' method is not yet implemented")
+          puts 'group addchild'
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,97 @@
+require 'thor'
+require_relative './formatter.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Implementation of the "group addhost" method of the CLI
+      class Group
+        #==========================
+        desc 'addhost NAME HOSTNAME',
+             'Associate a host HOSTNAME with the group NAME'
+        def addhost(*args) # rubocop:disable Metrics/AbcSize
+          # Sanity
+          if args.length < 2
+            abort("ERROR: Wrong number of arguments, #{args.length} for 2 or more.")
+          end
+
+          # Arguments
+          name  = args[0].downcase
+          hosts = args.slice(1, args.length - 1).uniq.map(&:downcase)
+
+          # Sanity
+          if name == 'ungrouped'
+            abort("ERROR: Cannot manually manipulate the automatic group 'ungrouped'.")
+          end
+
+          # Convenience
+          db = Moose::Inventory::DB
+          fmt = Moose::Inventory::Cli::Formatter
+
+          # Transaction
+          warn_count = 0
+          begin 
+            db.transaction do # Transaction start
+              puts "Associate group '#{name}' with host(s) '#{hosts.join(',')}':"
+              # Get the target group
+              fmt.puts 2, "- retrieve group '#{name}'..."
+              group = db.models[:group].find(name: name)
+              if group.nil?
+                abort("ERROR: The group '#{name}' does not exist.")
+              end
+              fmt.puts 4,  '- OK'
+
+              # Associate group with the hosts
+              ungrouped  = db.models[:group].find_or_create(name: 'ungrouped')
+              hosts_ds = group.hosts_dataset
+              hosts.each do |h| # rubocop:disable Style/Next
+                fmt.puts 2, "- add association {group:#{name} <-> host:#{ h }}..."
+
+                # Check against existing associations
+                unless hosts_ds[name: h].nil?
+                  warn_count += 1
+                  fmt.warn "Association {group:#{name} <-> host:#{ h }} already"\
+                    " exists, skipping.\n"
+                  fmt.puts 4, '- already exists, skipping.'
+                  fmt.puts 4, '- OK'
+                  next
+                end
+
+                # Add new association
+                host = db.models[:host].find(name: h)
+                if host.nil?
+                  warn_count += 1
+                  fmt.warn "Host '#{h}' does not exist and will be created.\n"
+                  fmt.puts 4, '- host does not exist, creating now...'
+                  host = db.models[:host].create(name: h)
+                  fmt.puts 6, '- OK'
+                end
+                  
+                group.add_host(host)
+                fmt.puts 4, '- OK'
+
+                # Remove the host from the ungrouped group, if necessary
+                unless host.groups_dataset[name: 'ungrouped'].nil?
+                  fmt.puts 2,'- remove automatic association '\
+                    "{group:ungrouped <-> host:#{h}}..."
+                  host.remove_group(ungrouped)
+                  fmt.puts 4, '- OK'
+                end
+              end
+              fmt.puts 2, '- all OK'
+            end # Transaction end
+          rescue db.exceptions[:moose] => e
+            abort("ERROR: #{e.message}")
+          end
+          if warn_count == 0
+            puts 'Succeeded.'
+          else
+            puts 'Succeeded, with warnings.'
+          end
+        end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,72 @@
+require 'thor'
+require_relative './formatter.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Implementation of the "group addvar" method of the CLI
+      class Group
+        #==========================
+        desc 'addvar NAME VARNAME=VALUE',
+             'Add a variable VARNAME with value VALUE to the group NAME'
+        def addvar(*args)
+          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 group '#{name}':"
+          fmt.puts 2,"- retrieve group '#{name}'..."
+          group = db.models[:group].find(name: name)
+          if group.nil?
+            fail db.exceptions[:moose],
+                 "The group '#{name}' does not exist."
+          end
+          fmt.puts 4, '- OK'
+
+          groupvars_ds = group.groupvars_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
+            groupvar = groupvars_ds[name: vararray[0]]
+            if !groupvar.nil?
+              unless groupvar[:value] == vararray[1]
+                fmt.puts 4, '- already exists, applying as an update...'
+                update = db.models[:groupvar].find(id: groupvar[:id])
+                update[:value] = vararray[1]
+                update.save
+              end
+            else
+              # groupvar doesn't exist, so create and associate
+              groupvar = db.models[:groupvar].create(name: vararray[0],
+                                                    value: vararray[1])
+              group.add_groupvar(groupvar)
+            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/group_get.rb

@@ -0,0 +1,52 @@
+require 'thor'
+require_relative './formatter.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Implementation of the "group get" method of the CLI
+      class Group
+        desc 'get GROUP_1 [GROUP_2 ...]', 'Get groups GROUP_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|
+            group = db.models[:group].find(name: name)
+              
+            unless group.nil?
+              hosts = group.hosts_dataset.map(:name)
+            
+              groupvars = {}
+              group.groupvars_dataset.each do |gv|
+                groupvars[gv[:name].to_sym] = gv[:value]
+              end
+
+              results[group[:name].to_sym] = {}
+              unless hosts.length == 0
+                results[group[:name].to_sym][:hosts]     = hosts
+              end
+              unless groupvars.length == 0
+                results[group[:name].to_sym][:groupvars] = groupvars
+              end
+            end
+          end
+          
+          fmt.dump(results)
+        end
+      end
+    end
+  end
+end