inprovise 0.2.2

data/lib/inprovise/cli/node.rb

@@ -0,0 +1,95 @@
+# CLI Node commands for Inprovise
+#
+# Author::    Martin Corino
+# License::   Distributes under the same license as Ruby
+
+class Inprovise::Cli
+
+  desc 'Manage infrastructure nodes'
+  command :node do |cnod|
+
+    cnod.desc 'Add an infrastructure node'
+    cnod.arg_name 'NODE'
+    cnod.command :add do |cnod_add|
+
+      cnod_add.flag [:a, :address], :arg_name => 'ADDRESS', :desc => 'Set the node address (hostname or IP). If not set node name is used as hostname.'
+      cnod_add.flag [:c, :config], :arg_name => 'CFGKEY=CFGVAL', :multiple => true, :desc => 'Specify a configuration setting for the node.'
+      cnod_add.flag [:C, :credential], :arg_name => 'CFGKEY=CFGVAL', :multiple => true, :desc => 'Specify a security credential setting for the node.'
+      cnod_add.switch [:sniff], :default_value => true, :desc => 'Enable or disable running sniffers'
+      cnod_add.flag [:g, :group], :arg_name => 'GROUP', :multiple => true, :desc => 'Existing infrastructure group to add new node to.'
+
+      cnod_add.action do |global,options,args|
+        raise ArgumentError, 'Missing or too many arguments!' unless args.size == 1
+        Inprovise::Controller.run(:add, options, :node, *args)
+        Inprovise::Controller.wait!
+      end
+
+    end
+
+    cnod.desc 'Remove (an) infrastructure node(s)'
+    cnod.arg_name 'NODE[ NODE [...]]'
+    cnod.command :remove do |cnod_del|
+
+      cnod_del.action do |global,options,args|
+        raise ArgumentError, 'Missing argument!' if args.empty?
+        Inprovise::Controller.run(:remove, options, :node, *args)
+        Inprovise::Controller.wait!
+      end
+
+    end
+
+    cnod.desc 'Update node configuration for the given infrastructure node(s) or group(s).'
+    cnod.arg_name 'NAME[ NAME [...]]'
+    cnod.command :update do |cnod_update|
+
+      cnod_update.flag [:c, :config], :arg_name => 'CFGKEY=CFGVAL', :multiple => true, :desc => 'Specify a configuration setting for the node(s)'
+      cnod_update.flag [:C, :credential], :arg_name => 'CFGKEY=CFGVAL', :multiple => true, :desc => 'Specify a security credential setting for the node.'
+      cnod_update.switch [:r, :reset], negatable: false, :desc => 'Reset configuration before update (default is to merge updates)'
+      cnod_update.switch [:sniff], :default_value => true, :desc => 'Enable or disable running sniffers'
+      cnod_update.flag [:g, :group], :arg_name => 'GROUP', :multiple => true, :desc => 'Existing infrastructure group to add node(s) to.'
+
+      cnod_update.action do |global,options,args|
+        raise ArgumentError, 'Missing argument!' if args.empty?
+        Inprovise::Controller.run(:update, options, :node, *args)
+        Inprovise::Controller.wait!
+      end
+    end
+
+    cnod.desc 'List infrastructure nodes (all or for specified nodes/groups)'
+    cnod.arg_name '[NAME[ NAME [...]]]'
+    cnod.command :list do |cnod_list|
+      cnod_list.switch [:d, :details], negatable: false, :desc => 'Show node details'
+
+      cnod_list.action do |global_options,options,args|
+        $stdout.puts
+        $stdout.puts "   INFRASTRUCTURE NODES"
+        $stdout.puts "   ===================="
+        if args.empty?
+          Inprovise::Infrastructure.list(Inprovise::Infrastructure::Node).each do |n|
+            Inprovise::Cli.show_target(n, options[:details])
+          end
+        else
+          args.each do |a|
+            tgt = Inprovise::Infrastructure.find(a)
+            case tgt
+              when Inprovise::Infrastructure::Node
+                Inprovise::Cli.show_target(tgt, options[:details])
+              when Inprovise::Infrastructure::Group
+                $stdout.puts "   #{tgt.to_s}"
+                $stdout.puts "   #{'-' * tgt.to_s.size}"
+                tgt.targets.each {|n| Inprovise::Cli.show_target(n, options[:details]) }
+                $stdout.puts "   #{'-' * tgt.to_s.size}"
+              else
+                $stdout.puts "ERROR: #{a} is unknown".red
+            end
+          end
+        end
+        $stdout.puts
+      end
+    end
+
+    cnod.default_command :list
+
+  end
+
+end

data/lib/inprovise/cli/provision.rb

@@ -0,0 +1,84 @@
+# CLI provisioning commands for Inprovise
+#
+# Author::    Martin Corino
+# License::   Distributes under the same license as Ruby
+
+class Inprovise::Cli
+
+  desc 'Apply the given script/package to the specified infrastructure nodes and/or groups.'
+  arg_name 'SCRIPT TARGET[ TARGET[...]]'
+  command :apply do |capply|
+
+    capply.desc 'Path to a provisioning scheme to load'
+    capply.flag [:s,:scheme], :arg_name => 'FILE', :multiple => true, :default_value => Inprovise.default_scheme
+    capply.flag [:c, :config], :arg_name => 'CFGKEY=CFGVAL', :multiple => true, :desc => 'Specify a configuration setting for the script execution'
+
+    capply.action do |global, options, args|
+      raise ArgumentError, 'Missing arguments!' if args.empty?
+      raise ArgumentError, 'Missing targets!' if args.size < 2
+      Inprovise::Controller.run(:apply, options, *args)
+      Inprovise::Controller.wait!
+    end
+  end
+
+  desc 'Revert the given script/package on the specified infrastructure nodes and/or groups.'
+  arg_name 'SCRIPT NAME[ NAME[...]]'
+  command :revert do |crevert|
+
+    crevert.desc 'Path to a provisioning scheme to load'
+    crevert.flag [:s,:scheme], :arg_name => 'FILE', :multiple => true, :default_value => Inprovise.default_scheme
+    crevert.flag [:c, :config], :arg_name => 'CFGKEY=CFGVAL', :multiple => true, :desc => 'Specify a configuration setting for the script execution'
+
+    crevert.action do |global, options, args|
+      raise ArgumentError, 'Missing arguments!' if args.empty?
+      raise ArgumentError, 'Missing targets!' if args.size < 2
+      Inprovise::Controller.run(:revert, options, *args)
+      Inprovise::Controller.wait!
+    end
+  end
+
+  desc 'Validate the given script/package on the specified infrastructure nodes and/or groups.'
+  arg_name 'SCRIPT NAME[ NAME[...]]'
+  command :validate do |cvalid|
+
+    cvalid.desc 'Path to a provisioning scheme to load'
+    cvalid.flag [:s,:scheme], :arg_name => 'FILE', :multiple => true, :default_value => Inprovise.default_scheme
+    cvalid.flag [:c, :config], :arg_name => 'CFGKEY=CFGVAL', :multiple => true, :desc => 'Specify a configuration setting for the script execution'
+
+    cvalid.action do |global, options, args|
+      raise ArgumentError, 'Missing arguments!' if args.empty?
+      raise ArgumentError, 'Missing targets!' if args.size < 2
+      Inprovise::Controller.run(:validate, options, *args)
+      Inprovise::Controller.wait!
+    end
+  end
+
+  desc 'Trigger a specific action on the specified infrastructure nodes and/or groups.'
+  arg_name 'ACTION NAME[ NAME[...]]'
+  command :trigger do |ctrigger|
+
+    ctrigger.desc 'Path to a provisioning scheme to load'
+    ctrigger.flag [:s,:scheme], :arg_name => 'FILE', :multiple => true, :default_value => Inprovise.default_scheme
+    ctrigger.flag [:c, :config], :arg_name => 'CFGKEY=CFGVAL', :multiple => true, :desc => 'Specify a configuration setting for the script execution'
+
+    ctrigger.action do |global, options, args|
+      raise ArgumentError, 'Missing arguments!' if args.empty?
+      raise ArgumentError, 'Missing targets!' if args.size < 2
+      Inprovise::Controller.run(:trigger, options, *args)
+      Inprovise::Controller.wait!
+    end
+  end
+
+  desc 'List the available scripts. By default lists only described scripts.'
+  command :list do |clist|
+
+    clist.desc 'Path to a provisioning scheme to load'
+    clist.flag [:s,:scheme], :arg_name => 'FILE', :multiple => true, :default_value => Inprovise.default_scheme
+    clist.switch [:a, :all], negatable: false, :desc => 'List all scripts (with or without description)'
+
+    clist.action do |global, options, args|
+      Inprovise::Controller.list_scripts(options)
+    end
+  end
+
+end

data/lib/inprovise/cli.rb

@@ -0,0 +1,105 @@
+# CLI for Inprovise
+#
+# Author::    Martin Corino
+# License::   Distributes under the same license as Ruby
+
+require 'gli'
+require 'fileutils'
+
+class Inprovise::Cli
+
+  class << self
+    include GLI::App
+  end
+
+
+  program_desc 'CLI for Inprovise'
+
+  version Inprovise::VERSION
+
+  subcommand_option_handling :normal
+  arguments :strict
+  sort_help :manually
+
+  desc 'Don\'t actually run any commands on the group, just pretend.'
+  switch [:n,:'dry-run'], {negatable: false}
+
+  desc 'Increase verbosity, useful for debugging.'
+  flag [:v, :verbose], :arg_name => 'LEVEL', :default_value => 0, :type => Integer
+
+  desc 'Show exception backtraces on exit.'
+  switch [:x, :'show-backtrace'], {negatable: false}
+
+  desc 'Don\'t run tasks in parrallel across nodes.'
+  switch [:sequential], {negatable: false}
+
+  desc 'Don\'t validate and run dependencies.'
+  switch [:'skip-dependencies'], {negatable: false}
+
+  require_relative './cli/node'
+  require_relative './cli/group'
+  require_relative './cli/provision'
+
+  desc 'Initialize Inprovise project.'
+  command :init do |cinit|
+    cinit.action do |global,options,args|
+      raise RuntimeError, 'Cannot initialize existing project directory.' if File.exists?(Inprovise::INFRA_FILE)
+      raise RuntimeError, "Default scheme #{Inprovise.default_scheme} already exists." if File.exists?(Inprovise.default_scheme)
+      begin
+        Inprovise::Infrastructure.init(Inprovise::INFRA_FILE)
+        path = Inprovise::Template.new(File.join(File.dirname(__FILE__),'template','inprovise.rb.erb')).render_to_tempfile
+        FileUtils.mv(path, Inprovise.default_scheme)
+      rescue
+        File.delete(Inprovise::INFRA_FILE) if File.exists?(Inprovise::INFRA_FILE)
+        File.delete(Inprovise.default_scheme) if File.exists?(Inprovise.default_scheme)
+        raise
+      end
+    end
+  end
+
+  pre do |global,command,options,args|
+    # Pre logic here
+    # Return true to proceed; false to abort and not call the
+    # chosen command
+    # Use skips_pre before a command to skip this block
+    # on that command only
+    Inprovise.verbosity = global[:verbose] || 0
+    Inprovise.show_backtrace = global[:'show-backtrace']
+    Inprovise.sequential = global[:sequential]
+    Inprovise.demonstrate = global[:demonstrate]
+    Inprovise.skip_dependencies = global[:'skip-dependencies']
+    unless command.name == :init
+      if File.readable?(File.join(Inprovise.root, Inprovise::RC_FILE))
+        Inprovise.log.local("Loading #{Inprovise::RC_FILE}") if Inprovise.verbosity > 1
+        load File.join(Inprovise.root, Inprovise::RC_FILE)
+      end
+
+      Inprovise::Infrastructure.load
+    end
+    true
+  end
+
+  post do |global,command,options,args|
+    # Post logic here
+    # Use skips_post before a command to skip this
+    # block on that command only
+  end
+
+  on_error do |exception|
+    # Error logic here
+    # return false to skip default error handling
+    $stderr.puts "ERROR: #{exception.message}".red
+    if Inprovise.show_backtrace
+      $stderr.puts "#{exception}\n#{exception.backtrace.join("\n")}"
+    end
+    exit 1
+  end
+
+  def self.show_target(tgt, details=false)
+    $stdout.puts "   #{tgt.to_s}"
+    if details
+      $stdout.puts "   \t"+JSON.pretty_generate(tgt.config).split("\n").join("\n   \t")
+    end
+  end
+
+end

data/lib/inprovise/cmd_channel.rb

@@ -0,0 +1,100 @@
+# Command channel for Inprovise
+#
+# Author::    Martin Corino
+# License::   Distributes under the same license as Ruby
+
+module Inprovise::CmdChannel
+
+  class << self
+
+    def implementations
+      @implementations ||= {}
+    end
+
+    def default_implementation
+      @default ||= 'ssh'
+    end
+
+    def default_implementation=(impl)
+      @default = impl
+    end
+
+    def define(impl, base=::Object, &definition)
+      implklass = Class.new(base) do
+        include Inprovise::CmdChannel
+      end
+      implklass.class_eval(&definition)
+      implementations[impl.to_s] = implklass
+      implklass
+    end
+
+    def open(node, impl)
+      implementations[impl || default_implementation].new(node)
+    end
+
+  end
+
+  attr_reader :node
+
+  def initialize(node)
+    @node = node
+  end
+
+  # session management
+
+  def close
+    # noop
+  end
+
+  # command execution (MANDATORY)
+
+  def run(command, forcelog=false)
+    raise RuntimeError, 'UNIMPLEMENTED'
+  end
+
+  # MANDATORY file management routines
+
+  def upload(from, to)
+    raise RuntimeError, 'UNIMPLEMENTED'
+  end
+
+  def download(from, to)
+    raise RuntimeError, 'UNIMPLEMENTED'
+  end
+
+  # OPTIONAL file management routines
+
+  # not recursive
+  # def mkdir(path)
+  # end
+
+  # def exists?(path)
+  # end
+
+  # def file?
+  # end
+  #
+  # def directory?
+  # end
+
+  # def content
+  # end
+
+  # def delete(path)
+  # end
+
+  # def permissions(path)
+  # end
+  #
+  # def set_permissions(path, perm)
+  # end
+
+  # def owner(path)
+  # end
+  #
+  # def set_owner(path, user, group=nil)
+  # end
+
+end
+
+require_relative './channel/ssh'

data/lib/inprovise/cmd_helper.rb

@@ -0,0 +1,150 @@
+# Command helper for Inprovise
+#
+# Author::    Martin Corino
+# License::   Distributes under the same license as Ruby
+
+module Inprovise::CmdHelper
+
+  class << self
+
+    def implementations
+      @implementations ||= {}
+    end
+
+    def default_implementation
+      @default ||= 'linux'
+    end
+
+    def default_implementation=(impl)
+      @default = impl
+    end
+
+    def define(impl, base=::Object, &definition)
+      implklass = Class.new(base) do
+        include Inprovise::CmdHelper
+      end
+      implklass.class_eval(&definition)
+      implementations[impl.to_s] = implklass
+      implklass
+    end
+
+    def get(node, impl)
+      implementations[impl || default_implementation].new(node.channel)
+    end
+
+  end
+
+  attr_reader :channel
+
+  # default init
+  def initialize(channel)
+    @channel = channel
+  end
+
+  # platform properties
+
+  def admin_user
+    nil
+  end
+
+  def env_reference(varname)
+    nil
+  end
+
+  def cwd
+    nil
+  end
+
+  # *must* return previous value
+  def set_cwd(path)
+    nil
+  end
+
+  # generic command execution
+
+  def run(cmd, forcelog=false)
+    @channel.run(cmd,forcelog)
+  end
+
+  # return sudo helper
+  def sudo
+    nil
+  end
+
+  # file management
+
+  def upload(from, to)
+    @channel.upload(from, to)
+  end
+
+  def download(from, to)
+    @channel.download(from, to)
+  end
+
+  # basic commands
+
+  def echo(arg)
+    nil
+  end
+
+  def env(var)
+    echo(env_reference(var))
+  end
+
+  def cat(path)
+    nil
+  end
+
+  def hash_for(path)
+    nil
+  end
+
+  def mkdir(path)
+    nil
+  end
+
+  def exists?(path)
+    false
+  end
+
+  def file?(path)
+    false
+  end
+
+  def directory?(path)
+    false
+  end
+
+  def copy(from, to)
+    nil
+  end
+
+  def delete(path)
+    nil
+  end
+
+  def permissions(path)
+    0
+  end
+
+  def set_permissions(path, perm)
+    nil
+  end
+
+  def owner(path)
+    nil
+  end
+
+  def set_owner(path, user, group=nil)
+    nil
+  end
+
+  def binary_exists?(bin)
+    false
+  end
+
+end
+
+require_relative './helper/linux'
+require_relative './helper/cygwin'
+require_relative './helper/windows'