moose-inventory 0.1.0

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

@@ -0,0 +1,41 @@
+require 'thor'
+require_relative './formatter.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Implementation of the "group list" method of the CLI
+      class Group
+        #==========================
+        desc 'list',
+             'List the groups, together with any associated hosts and groupvars'
+        option :ansiblestyle, type: :boolean
+        def list # rubocop:disable Metrics/AbcSize
+          # Convenience
+          db    = Moose::Inventory::DB
+
+          # Process
+          results = {}
+          db.models[:group].all.each do |group|
+            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
+          Formatter.out(results)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,77 @@
+require 'thor'
+require_relative './formatter.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Implementation of "group rm" methods of the CLI
+      class Group
+
+        #==========================
+        desc 'rm NAME',
+             'Remove a group NAME 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)
+
+          # sanity
+          if names.include?('ungrouped')
+            abort("Cannot manually manipulate the automatic group 'ungrouped'\n")
+          end
+            
+          # Transaction
+          warn_count = 0
+          db.transaction do # Transaction start
+            names.each do |name|
+              puts "Remove group '#{name}':"
+              fmt.puts 2, "- Retrieve group '#{name}'..."
+              group = db.models[:group].find(name: name)
+              if group.nil?
+                warn_count += 1
+                fmt.warn "Group '#{ name }' does not exist, skipping.\n"
+                fmt.puts 4, "- No such group, skipping."
+              end
+              fmt.puts 4, "- OK" 
+              unless group.nil?
+                # Handle automatic group for any associated hosts
+                hosts_ds = group.hosts_dataset
+                hosts_ds.each do |host|
+                  host_groups_ds = host.groups_dataset
+                  if host_groups_ds.count == 1 # We're the only group
+                    fmt.puts 2, "- Adding automatic association {group:ungrouped <-> host:#{host[:name]}}..."
+                    ungrouped = db.models[:group].find_or_create(name: 'ungrouped')
+                    host.add_group(ungrouped)
+                    fmt.puts 4, "- OK"
+                  end
+                end
+                # Remove the group
+                fmt.puts 2, "- Destroy group '#{name}'..."
+                group.remove_all_hosts
+                group.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/group_rmchild.rb

@@ -0,0 +1,20 @@
+require 'thor'
+require_relative './formatter.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Implemention of the "group rmchild" methods of the CLI
+      class Group < Thor # rubocop:disable ClassLength
+        #==========================
+        desc 'rmchild NAME CHILDNAME',
+             'Dissociate a child-group CHILDNAME from the group NAME'
+        def rmchild
+          abort("The 'groups rmchild GROUP' method is not yet implemented")
+          puts 'group rmchild'
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,89 @@
+require 'thor'
+require_relative './formatter.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Implementation of the "group rmhost" method of the CLI
+      class Group
+
+        #==========================
+        desc 'rmhost GROUPNAME HOSTNAME_1 [HOSTNAME_2 ...]',
+             'Dissociate the hosts HOSTNAME_n from the group NAME'
+        # rubocop:disable Metrics/LineLength
+        def rmhost(*args) # rubocop:disable Metrics/AbcSize, Metrics/PerceivedComplexity, Metrics/MethodLength, Metrics/CyclomaticComplexity
+          # rubocop:enable Metrics/LineLength
+          # 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
+              # Get the target group
+              puts "Dissociate group '#{name}' from host(s) '#{hosts.join(',')}':"
+              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'
+
+              # dissociate group from 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, "- remove association {group:#{name} <-> host:#{ h }}..."
+
+                # Check against existing associations
+                if hosts_ds[name: h].nil?
+                  warn_count += 1
+                  fmt.warn "Association {group:#{name} <-> host:#{ h }} doesn't"\
+                    " exist, skipping.\n"
+                  fmt.puts 4, '- doesn\'t exist, skipping.'
+                  fmt.puts 4, '- OK'
+                  next
+                end
+
+                host = db.models[:host].find(name: h)
+                group.remove_host(host) unless host.nil?
+                fmt.puts 4,'- OK'
+
+                # Add the host to the ungrouped group if not in any other group
+                if host.groups_dataset.count == 0
+                  fmt.puts 2, "- add automatic association {group:ungrouped <-> host:#{h}}..."
+                  host.add_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_rmvar.rb

@@ -0,0 +1,65 @@
+require 'thor'
+require_relative './formatter.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Implementation of the "group rmvar" method of the CLI
+      class Group
+        #==========================
+        desc 'rmvar NAME VARNAME',
+             'Remove a variable VARNAME from the group NAME'
+        def rmvar(*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 "Remove variable(s) '#{vars.join(",")}' from 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, "- 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
+              groupvar = groupvars_ds[name: vararray[0]]
+              unless groupvar.nil?
+                # remove the association
+                group.remove_groupvar(groupvar)
+                groupvar.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/cli/host.rb

@@ -0,0 +1,24 @@
+require 'thor'
+require 'json'
+
+require_relative './formatter.rb'
+require_relative '../db/exceptions.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Class implementing the "host" methods of the CLI
+      class Host < Thor # rubocop:disable Metrics/ClassLength
+        require_relative 'host_add'
+        require_relative 'host_get'
+        require_relative 'host_list'
+        require_relative 'host_rm'
+        require_relative 'host_addgroup'
+        require_relative 'host_rmgroup'
+        require_relative 'host_addvar'
+        require_relative 'host_rmvar'
+      end
+    end
+  end
+end

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

@@ -0,0 +1,93 @@
+require 'thor'
+require 'json'
+require 'indentation'
+
+require_relative './formatter.rb'
+require_relative '../db/exceptions.rb'
+ 
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # Class implementing the "host" methods of the CLI
+      class Host
+        #==========================
+        desc 'add HOSTNAME_1 [HOSTNAME_2 ...]',
+             'Add a hosts HOSTNAME_n to the inventory'
+        option :groups
+        # rubocop:disable Metrics/LineLength
+        def add(*argv) # rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity
+          # rubocop:enable Metrics/LineLength
+          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)
+
+          # split(/\W+/) splits on hyphens too, which is not what we want
+          #groups = options[:groups].downcase.split(/\W+/).uniq
+          options[:groups].nil? && options[:groups] = ''  
+          groups = options[:groups].downcase.split(',').uniq
+
+          # Sanity
+          if groups.include?('ungrouped')
+            abort("ERROR: Cannot manually manipulate "\
+              "the automatic group 'ungrouped'.")
+          end
+
+          # Process
+          db.transaction do # Transaction start
+            fmt.reset_indent
+              
+            names.each do |name|
+              puts "Add host '#{name}':"
+              fmt.puts 2, "- Creating host '#{name}'..."
+              host = db.models[:host].find(name: name)
+              groups_ds = nil
+              if host.nil?
+                host = db.models[:host].create(name: name)
+              else
+                fmt.warn "The host '#{name}' already exists, skipping creation.\n"
+                groups_ds = host.groups_dataset
+              end
+              fmt.puts 4, "- OK"
+              
+              groups.each do |g|
+                next if g.nil? || g.empty?
+                fmt.puts 2, "- Adding association {host:#{name} <-> group:#{g}}..."
+                group = db.models[:group].find(name: g)
+                if group.nil?
+                  fmt.warn "The group '#{g}' doesn't exist, but will be created.\n"
+                  group = db.models[:group].create(name: g)
+                end
+                if !groups_ds.nil? && groups_ds[name: g].nil?
+                  fmt.warn "Association {host:#{name} <-> group:#{ g }} already exists, skipping creation.\n"
+                else
+                  host.add_group(group)
+                end
+                fmt.puts 4, '- OK'
+              end
+              
+              # Handle the automatic 'ungrouped' group
+              groups_ds = host.groups_dataset
+              if !groups_ds.nil?  && groups_ds.count == 0
+                  fmt.puts 2, "- Adding automatic association {host:#{name} <-> group:ungrouped}..."
+                  ungrouped = db.models[:group].find_or_create(name: 'ungrouped')
+                  host.add_group(ungrouped)
+                  fmt.puts 4, "- OK"
+              end
+              fmt.puts 2, "- All OK"
+            end
+          end # Transaction end
+          puts 'Succeeded'
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,88 @@
+require 'thor'
+require 'json'
+
+require_relative './formatter.rb'
+require_relative '../db/exceptions.rb'
+
+module Moose
+  module Inventory
+    module Cli
+      ##
+      # implementation of the "addgroup" method of the CLI
+      class Host
+        desc 'addgroup HOSTNAME GROUPNAME [GROUPNAME ...]',
+             'Associate the host with a group'
+        # rubocop:disable Metrics/LineLength
+        def addgroup(*args) # rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/MethodLength, Metrics/PerceivedComplexity
+          # rubocop:enable Metrics/LineLength
+          # Sanity
+          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 "Associate host '#{name}' with groups '#{groups.join(',')}':" 
+            # Get the target host
+            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'
+
+            # Associate host with the groups
+            groups_ds = host.groups_dataset
+            groups.each do |g|
+              fmt.puts 2,  "- Add association {host:#{name} <-> group:#{ g }}..."
+
+              # Check against existing associations
+              if !groups_ds[name: g].nil?
+                fmt.warn "Association {host:#{name} <-> group:#{g}} already exists, skipping."
+                fmt.puts 4,  "- Already exists, skipping."
+              else
+                # Add new association
+                group = db.models[:group].find(name: g)
+                if group.nil?
+                  fmt.warn "Group '#{g}' does not exist and will be created."
+                  fmt.puts 4,  "- Group does not exist, creating now..."
+                  group = db.models[:group].create(name: g)
+                  fmt.puts 6,  "- OK"
+                end
+                host.add_group(group)
+              end
+              fmt.puts 4, '- OK'
+            end
+
+            # Handle 'ungrouped' group automation
+            unless groups_ds[name: 'ungrouped'].nil?
+              fmt.puts 2,  '- Remove automatic association '\
+                "{host:#{name} <-> group:ungrouped}..."
+              ungrouped  = db.models[:group].find(name: 'ungrouped')
+              host.remove_group(ungrouped) unless ungrouped.nil?
+              fmt.puts 4, '- OK'
+            end
+            fmt.puts 2, '- All OK'
+          end # Transaction end
+          puts 'Succeeded'
+        end
+      end
+    end
+  end
+end