require File.join(File.dirname(__FILE__), '..', 'chatterbot') require 'optparse' module Chatterbot # # very basic DSL to handle the common stuff you would want to do with a bot. module DSL # # generate a Bot object. if the DSL is being called from a Bot object, just return it # otherwise create a bot and return that def bot return @bot unless @bot.nil? # # parse any command-line options and use them to initialize the bot # params = {} opts = OptionParser.new opts.banner = "Usage: #{File.basename($0)} [options]" opts.separator "" opts.separator "Specific options:" opts.on('-d', '--db [ARG]', "Specify a DB connection URI") { |d| ENV["chatterbot_db"] = d } opts.on('-c', '--config [ARG]', "Specify a config file to use") { |c| ENV["chatterbot_config"] = c } opts.on('-t', '--test', "Run the bot without actually sending any tweets") { params[:debug_mode] = true } opts.on('--dry-run', "Run the bot in test mode, and also don't update the database") { params[:debug_mode] = true ; params[:no_update] = true } opts.on('-s', '--since_id [ARG]', "Check for tweets since tweet id #[ARG]") { |s| params[:since_id] = s } opts.on_tail("-h", "--help", "Show this message") do puts opts exit end opts.parse!(ARGV) @bot = Chatterbot::Bot.new(params) end # # specify a bot-specific blacklist of users. accepts an array, or a # comma-delimited string def blacklist(b=nil) if b.is_a?(String) b = b.split(",").collect { |s| s.strip } end if b.nil? || b.empty? bot.blacklist = [] else bot.blacklist += b end end # # specify list of strings we will check when deciding to respond to a tweet or not def exclude(e=nil) if e.is_a?(String) e = e.split(",").collect { |s| s.strip } end if e.nil? || e.empty? bot.exclude = [] else bot.exclude += e end end # # search twitter for the specified terms def search(query, opts = {}, &block) @bot.search(query, opts, &block) end # # handle replies to the bot def replies(&block) @bot.replies(&block) end # # send a tweet def tweet(txt, params = {}, original = nil) @bot.tweet(txt, params, original) end # # reply to a tweet def reply(txt, source) @bot.reply(txt, source) end end end include Chatterbot::DSL include Chatterbot::Helpers