############################################################################## # 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 " env_msg = "Have you configured your environments?\n Try: admin_module config add env " 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