Revision 09b29d2c
Added by Marc Dequènes about 13 years ago
- ID 09b29d2ca895315082be96521c37365f9c52e7e9
| lib/cyborghood/cyborg/task.rb | ||
|---|---|---|
|
|
||
|
module CyborgHood
|
||
|
module TaskAspect
|
||
|
def task(name, &block)
|
||
|
def task(name, options = {}, &block)
|
||
|
the_bot = self.is_a?(Cyborg) ? self : self.bot
|
||
|
DSL::Task.new(the_bot, name, &block)
|
||
|
DSL::Task.new(the_bot, name, options, &block)
|
||
|
end
|
||
|
|
||
|
def schedule_task(callback = nil, &task)
|
||
| ... | ... | |
|
|
||
|
module DSL
|
||
|
class TaskBase
|
||
|
attr_reader :bot, :name, :errors, :results, :preferred_locales, :locale, :user
|
||
|
attr_reader :bot, :name, :errors, :results, :preferred_locales, :locale, :user, :store
|
||
|
|
||
|
include I18nTranslation
|
||
|
|
||
| ... | ... | |
|
end
|
||
|
|
||
|
# the name MUST be unique
|
||
|
def initialize(bot, name, &block)
|
||
|
def initialize(bot, name, options = {}, &block)
|
||
|
@bot = bot
|
||
|
@name = name
|
||
|
|
||
| ... | ... | |
|
@locale = nil
|
||
|
@user = nil
|
||
|
|
||
|
@store = OpenStruct.new
|
||
|
|
||
|
_setup
|
||
|
_start_dsl &block
|
||
|
|
||
|
step = {:options => options, :cb => block}
|
||
|
step_code = get_step_code(step)
|
||
|
_start_dsl(step_code)
|
||
|
end
|
||
|
|
||
|
def set_preferred_locales(prefs)
|
||
| ... | ... | |
|
@cancel_on_start_error = true
|
||
|
end
|
||
|
|
||
|
def on_error(&cb)
|
||
|
@error_cb = cb
|
||
|
def on_error(options = {}, &cb)
|
||
|
@error_step = {:options => options, :cb => cb}
|
||
|
end
|
||
|
|
||
|
def on_success(&cb)
|
||
|
@success_cb = cb
|
||
|
def on_success(options = {}, &cb)
|
||
|
@success_step = {:options => options, :cb => cb}
|
||
|
end
|
||
|
|
||
|
def stop_bot(condition)
|
||
| ... | ... | |
|
subtasks_running.empty?
|
||
|
end
|
||
|
|
||
|
def get_step_code(step)
|
||
|
return if step.nil?
|
||
|
|
||
|
cb = step[:cb]
|
||
|
return cb unless cb.nil?
|
||
|
|
||
|
file = step[:options][:file]
|
||
|
if file.nil?
|
||
|
logger.error "Task '#{@name}': error in DSL: neither block nor file provided for next step definition"
|
||
|
return
|
||
|
end
|
||
|
|
||
|
begin
|
||
|
step_code = File.read(File.join(self.base_lpath, file + ".rb"))
|
||
|
rescue
|
||
|
logger.error "Task '#{@name}': error in DSL: cannot open step definition file: " + $!.message
|
||
|
return
|
||
|
end
|
||
|
|
||
|
return step_code
|
||
|
end
|
||
|
|
||
|
def _check_finished
|
||
|
return unless _subtasks_finished?
|
||
|
|
||
| ... | ... | |
|
@result.freeze
|
||
|
|
||
|
# next step
|
||
|
cb = @errors.empty? ? @success_cb : @error_cb
|
||
|
if cb
|
||
|
step = @errors.empty? ? @success_step : @error_step
|
||
|
step_code = get_step_code(step)
|
||
|
if step_code
|
||
|
logger.debug "Task '#{@name}': step result: " + (@errors.empty? ? "success" : "error")
|
||
|
_start_dsl(&cb)
|
||
|
_start_dsl(step_code)
|
||
|
else
|
||
|
_finished
|
||
|
end
|
||
| ... | ... | |
|
logger.debug "Task '#{@name}': created"
|
||
|
end
|
||
|
|
||
|
def _start_dsl(&block)
|
||
|
def base_lpath
|
||
|
File.join(Config::LIB_DIR, 'cyborghood-' + @bot.model.downcase, 'tasks')
|
||
|
end
|
||
|
|
||
|
def _start_dsl(step_code)
|
||
|
@dsl_runing = false
|
||
|
@subtasks = []
|
||
|
@cancel_on_start_error = false
|
||
|
@error_cb = nil
|
||
|
@success_cb = nil
|
||
|
@error_step = nil
|
||
|
@success_step = nil
|
||
|
|
||
|
begin
|
||
|
instance_eval(&block)
|
||
|
if step_code.is_a? String
|
||
|
instance_eval step_code
|
||
|
else
|
||
|
instance_eval &step_code
|
||
|
end
|
||
|
rescue
|
||
|
logger.error "Task '#{@name}': error in DSL: " + $!.to_s
|
||
|
logger.debug $!.backtrace.join("\n")
|
||
Also available in: Unified diff
[evol] Task: implement loading step from file and a store to help communicate between files