admin_module 0.1.4

data/lib/admin_module/rake/rule_tasks.rb

@@ -0,0 +1,157 @@
+##############################################################################
+# File::    rule_tasks.rb
+# Purpose:: RuleTasks definition
+#
+# Author::    Jeff McAffee 2014-04-24
+#
+##############################################################################
+
+require 'admin_module'
+require 'rake/dsl_definition'
+require 'rake'
+
+module AdminModule::Rake
+
+  class RuleTasks
+    include ::Rake::DSL if defined?(::Rake::DSL)
+
+    attr_accessor :env
+    attr_accessor :name
+    attr_accessor :to
+    attr_reader   :action
+    attr_reader   :valid_actions
+    attr_reader   :stop_on_exception
+
+    def initialize(task_name = 'rule_task', desc = "Modify a guideline rule")
+      @valid_actions = ['rename', 'delete']
+      @task_name, @desc = task_name, desc
+      @stop_on_exception = true
+
+      yield self if block_given?
+
+      define_task
+    end
+
+    def define_task #:nodoc:
+      desc @desc
+      task @task_name, required_args_for_action do |t,args|
+        set_vars args
+        commit  # Call method to perform when invoked.
+      end
+    end
+
+    def set_vars args
+      args.each do |arg,val|
+        instance_variable_set "@#{arg}", val
+      end
+
+      args
+    end
+
+    def action=(task_action)
+      raise "action must be one of #{valid_actions.join(', ')}" unless valid_actions.include?(task_action.downcase)
+      @action = task_action
+    end
+
+    def stop_on_exception=(do_stop)
+      raise ArgumentError, 'Expecting true or false' unless do_stop === true || do_stop === false
+      @stop_on_exception = do_stop
+    end
+
+    def commit
+      validate_params
+
+      client = AdminModule::Client.new
+      client.env = env
+
+      if self.respond_to? action
+        self.send(action, client)
+        return
+      else
+        raise "Unknown action - #{action}"
+      end
+
+    rescue Exception => e
+      raise e if stop_on_exception == true
+    ensure
+      client.quit unless client.nil?
+    end
+
+    def rename client
+      $stdout << client.rules.rename(name, to)
+    end
+
+    def delete client
+      $stdout << client.rules.delete(name)
+    end
+
+    def validate_params
+      assert_provided env, 'Missing "env"'
+      assert_provided action, 'Missing "action"'
+
+      case action
+      when 'rename'
+        assert_provided name, 'Missing "name"'
+        assert_provided to, 'Missing "to"'
+
+      when 'delete'
+        assert_provided name, 'Missing "name"'
+
+      end
+
+      assert_env_is_configured env
+    end
+
+    def assert_provided value, msg
+      if value.nil? || value.empty?
+        raise msg
+      end
+    end
+
+    def assert_env_is_configured arg
+      unless AdminModule.configuration.credentials.key? arg
+        init_msg = "Have you initialized your config file?\n Try: admin_module config init <filedir>"
+        env_msg = "Have you configured your environments?\n Try: admin_module config add env <envname> <url>"
+        raise "Unknown environment: #{arg}\n#{init_msg}\n\n#{env_msg}"
+      end
+    end
+
+    def required_args_for_action
+      args = []
+
+      case action
+      when 'delete'
+        args << :name
+
+      when 'rename'
+        args << :name
+        args << :to
+
+      else
+        # Noop
+      end
+
+      args
+    end
+
+    class << self
+      def install
+        new.install
+      end
+    end
+
+    def install
+      AdminModule.configuration.credentials.keys.each do |e|
+        valid_actions.each do |action|
+          AdminModule::Rake::RuleTasks.new("am:#{e}:rule:#{action}", "#{action} #{e} rule(s)") do |t|
+            t.env = e
+            t.action = action
+          end
+        end
+      end
+    end
+  end # class
+end # module AdminModule::Task
+
+AdminModule::Rake::RuleTasks.install
+

data/lib/admin_module/rake/snapshot_tasks.rb

@@ -0,0 +1,223 @@
+##############################################################################
+# File::    snapshot_tasks.rb
+# Purpose:: SnapshotTasks definition
+#
+# Author::    Jeff McAffee 2015-04-05
+#
+##############################################################################
+
+require 'admin_module'
+require 'rake/dsl_definition'
+require 'rake'
+
+module AdminModule::Rake
+  class SnapshotTasks
+    include ::Rake::DSL if defined?(::Rake::DSL)
+
+    attr_accessor :env
+    attr_accessor :name
+    attr_accessor :to
+    attr_accessor :path
+    attr_reader   :action
+    attr_reader   :valid_actions
+    attr_reader   :stop_on_exception
+
+    def initialize(task_name = 'snapshots_task', desc = "modify snapshot definition(s)")
+      @valid_actions = ['import', 'export', 'read', 'list', 'rename']
+      @task_name, @desc = task_name, desc
+
+      @stop_on_exception = true
+
+      yield self if block_given?
+
+      define_task
+    end
+
+    def define_task #:nodoc:
+      desc @desc
+      task(@task_name, required_args_for_action) do |t,args|
+        set_vars args
+        commit  # Call method to perform when invoked.
+      end
+    end
+
+    #
+    # Add each arg passed to the task, as an instance variable of the task
+    #
+
+    def set_vars args
+      args.each do |arg,val|
+        instance_variable_set "@#{arg}", val
+      end
+
+      args
+    end
+
+    def action=(task_action)
+      raise "action must be one of #{valid_types.join(', ')}" unless valid_actions.include?(task_action.downcase)
+      @action = task_action
+    end
+
+    def stop_on_exception=(do_stop)
+      raise ArgumentError, 'Expecting true or false' unless do_stop === true || do_stop === false
+      @stop_on_exception = do_stop
+    end
+
+    #
+    # Execute the task (action)
+    #
+
+    def commit
+      validate_params
+
+      client = AdminModule::Client.new
+      client.env = env
+
+      if self.respond_to? action
+        self.send(action, client)
+        return
+      else
+        raise "Unknown action - #{action}"
+      end
+
+    rescue Exception => e
+      raise e if stop_on_exception == true
+    ensure
+      client.quit
+    end
+
+    #
+    # Actions
+    #
+
+    def list client
+      $stdout << client.snapshots.list.join("\n")
+      $stdout << "\n"
+    end
+
+    def read client
+      result = {}
+      result[name] = client.snapshots.read(name)
+      $stdout << result.to_yaml
+    end
+
+    def import client
+      $stdout << client.snapshots.import(path)
+    end
+
+    def export client
+      $stdout << client.snapshots.export(path)
+    end
+
+    def rename client
+      $stdout << client.snapshots.rename(name, to)
+    end
+
+    #
+    # Verify we have the needed parameters (arguments) to perform
+    # the task action.
+    #
+
+    def validate_params
+      assert_provided env, 'Missing "env"'
+      assert_provided action, 'Missing "action"'
+
+      case action
+      when 'import'
+        assert_provided path, 'Missing "path"'
+
+      when 'export'
+        assert_provided path, 'Missing "path"'
+
+      when 'read'
+        assert_provided name, 'Missing "name"'
+
+      when 'rename'
+        assert_provided name, 'Missing "name"'
+        assert_provided to, 'Missing "to"'
+
+      end
+
+      assert_env_is_configured env
+    end
+
+    def assert_provided value, msg
+      if value.nil? || value.empty?
+        raise msg
+      end
+    end
+
+    def assert_env_is_configured arg
+      unless AdminModule.configuration.credentials.key? arg
+        init_msg = "Have you initialized your config file?\n Try: admin_module config init <filedir>"
+        env_msg = "Have you configured your environments?\n Try: admin_module config add env <envname> <url>"
+        raise "Unknown environment: #{arg}\n#{init_msg}\n\n#{env_msg}"
+      end
+    end
+
+    #
+    # Define the task args, based on the action.
+    # Used when programatically generating the tasks.
+    #
+
+    def required_args_for_action
+      args = []
+
+      case action
+      when 'read'
+        args << :name
+
+      when 'rename'
+        args << :name
+        args << :to
+
+      when 'import'
+        args << :path
+
+      when 'export'
+        args << :path
+
+      else
+        # Noop
+      end
+
+      args
+    end
+
+    #
+    # Class method to create an instance of the task generator,
+    # and call the instance's install method.
+    #
+
+    class << self
+      def install
+        new.install
+      end
+    end
+
+    #
+    # Instance method to generate the tasks;
+    # For each environment where we have configured credentials,
+    # generate a task for each available action.
+    #
+
+    def install
+      AdminModule.configuration.credentials.keys.each do |e|
+        valid_actions.each do |action|
+          AdminModule::Rake::SnapshotTasks.new("am:#{e}:snapshot:#{action}", "#{action} #{e} snapshot defn(s)") do |t|
+            t.env = e
+            t.action = action
+          end
+        end
+      end
+    end
+  end # class
+end # module
+
+
+#
+# Install (generate) the tasks
+#
+
+AdminModule::Rake::SnapshotTasks.install
+

data/lib/admin_module/rake/stage_tasks.rb

@@ -0,0 +1,197 @@
+##############################################################################
+# File::    stage_tasks.rb
+# Purpose:: StageTasks definition
+# 
+# Author::    Jeff McAffee 2015-03-10
+# Copyright:: Copyright (c) 2015, kTech Systems LLC. All rights reserved.
+# Website::   http://ktechsystems.com
+##############################################################################
+
+require 'admin_module'
+require 'rake/dsl_definition'
+require 'rake'
+
+module AdminModule::Rake
+
+  class StageTasks
+    include ::Rake::DSL if defined?(::Rake::DSL)
+
+    attr_accessor :env
+    attr_accessor :name
+    attr_accessor :to
+    attr_accessor :path
+    attr_accessor :allow_create
+    attr_reader   :action
+    attr_reader   :valid_actions
+    attr_reader   :stop_on_exception
+
+    def initialize(task_name = 'stages_task', desc = "Modify a stage or stages")
+      @valid_actions = ['import', 'export', 'read', 'list', 'rename', 'delete']
+      @task_name, @desc = task_name, desc
+
+      @stop_on_exception = true
+      @allow_create = false
+
+      yield self if block_given?
+
+      define_task
+    end
+
+    def define_task #:nodoc:
+      desc @desc
+      task(@task_name, required_args_for_action) do |t,args|
+        set_vars args
+        commit  # Call method to perform when invoked.
+      end
+    end
+
+    def set_vars args
+      args.each do |arg,val|
+        instance_variable_set "@#{arg}", val
+      end
+
+      args
+    end
+
+    def action=(task_action)
+      raise "action must be one of #{valid_types.join(', ')}" unless valid_actions.include?(task_action.downcase)
+      @action = task_action
+    end
+
+    def stop_on_exception=(do_stop)
+      raise ArgumentError, 'Expecting true or false' unless do_stop === true || do_stop === false
+      @stop_on_exception = do_stop
+    end
+
+    def commit
+      validate_params
+
+      client = AdminModule::Client.new
+      client.env = env
+
+      if self.respond_to? action
+        self.send(action, client)
+        return
+      else
+        raise "Unknown action - #{action}"
+      end
+
+    rescue Exception => e
+      raise e if stop_on_exception == true
+    ensure
+      client.quit
+    end
+
+    def list client
+      $stdout << client.stages.list.join("\n")
+      $stdout << "\n"
+    end
+
+    def read client
+      result = {}
+      result[name] = client.stages.read(name)
+      $stdout << result.to_yaml
+    end
+
+    def import client
+      $stdout << client.stages.import(path, allow_create)
+    end
+
+    def export client
+      $stdout << client.stages.export(path)
+    end
+
+    def rename client
+      $stdout << client.stages.rename(name, to)
+    end
+
+    def delete client
+      $stdout << client.stages.delete(name)
+    end
+
+    def validate_params
+      assert_provided env, 'Missing "env"'
+      assert_provided action, 'Missing "action"'
+
+      case action
+      when 'import'
+        assert_provided path, 'Missing "path"'
+
+      when 'export'
+        assert_provided path, 'Missing "path"'
+
+      when 'read'
+        assert_provided name, 'Missing "name"'
+
+      when 'rename'
+        assert_provided name, 'Missing "name"'
+        assert_provided to, 'Missing "to"'
+
+      when 'delete'
+        assert_provided name, 'Missing "name"'
+
+      end
+
+      assert_env_is_configured env
+    end
+
+    def assert_provided value, msg
+      if value.nil? || value.empty?
+        raise msg
+      end
+    end
+
+    def assert_env_is_configured arg
+      unless AdminModule.configuration.credentials.key? arg
+        init_msg = "Have you initialized your config file?\n Try: admin_module config init <filedir>"
+        env_msg = "Have you configured your environments?\n Try: admin_module config add env <envname> <url>"
+        raise "Unknown environment: #{arg}\n#{init_msg}\n\n#{env_msg}"
+      end
+    end
+
+    def required_args_for_action
+      args = []
+
+      case action
+      when 'read','delete'
+        args << :name
+
+      when 'rename'
+        args << :name
+        args << :to
+
+      when 'import'
+        args << :path
+        args << :allow_create
+
+      when 'export'
+        args << :path
+
+      else
+        # Noop
+      end
+
+      args
+    end
+
+    class << self
+      def install
+        new.install
+      end
+    end
+
+    def install
+      AdminModule.configuration.credentials.keys.each do |e|
+        valid_actions.each do |action|
+          AdminModule::Rake::StageTasks.new("am:#{e}:stage:#{action}", "#{action} #{e} stage(s)") do |t|
+            t.env = e
+            t.action = action
+          end
+        end
+      end
+    end
+  end # class
+end # module
+
+AdminModule::Rake::StageTasks.install
+