wesabot 1.0.1

data/lib/campfire/polling_bot/plugins/kibitz/kibitz_plugin.rb

@@ -0,0 +1,90 @@
+# Plugin to make Wes chatty (or annoying)
+class KibitzPlugin < Campfire::PollingBot::Plugin
+  accepts :text_message, :addressed_to_me => true
+  priority -1
+
+  def process(message)
+    person = message.person
+    case message.command
+    when /^\s*$/
+      # question questioners, exclaim at exclaimers, dot dotters
+      message.body.match(/(\?|\!|\.)$/)
+      bot.say(person + $1.to_s)
+    when /say\s+(.*)/
+      bot.say($1)
+    when /^(hey|hi|hello|sup|howdy)/i
+      bot.say("#{$1} #{person}")
+    when /(^later|(?:good\s*)?bye)/i
+      bot.say("#{$1} #{person}")
+    when /you rock|awesome|cool|nice/i
+      sayings = ["Thanks, #{person}, you're pretty cool yourself.",
+                 "I try.",
+                 "Aw, shucks. Thanks, #{person}."]
+      bot.say_random(sayings)
+    when /(^|you|still)\s*there/i, /\byt\b/i
+      bot.say_random(%w{Yup y})
+    when /(wake up|you awake)/i
+      bot.say("Yo.")
+    when /zod/i
+      bot.say_random [
+        "Zod's a wanker.",
+        "I'd tell you about Zod, but you wouldn't listen. No one ever does.",
+        "somebody send Zod back to the Phantom Zone",
+        "Zod and I were friends, once. It all ended one awful night in El Paso over a bottle of shitty scotch and a ten-dollar whore. Christ, those were the days."
+      ]
+    when /thanks|thank you/i
+      bot.say_random ["No problem.", "np", "any time", "that's what I'm here for", "You're welcome."]
+    when /^(good\s?night|(?:g')?night)$/i
+      bot.say_random [
+        "see you later, #{person}",
+        "later, #{person}",
+        "night",
+        "goodnight",
+        "bye",
+        "have a good night"
+      ]
+    when /^(see you(?: later)?)$/i
+      bot.say_random [
+        "see you later, #{person}",
+        "later, #{person}",
+        "bye",
+        "later",
+        "see ya",
+      ]
+    when /^(?:get|grab|fetch|bring)(?: (.*?))?(?: (?:a|some))? coffee$/i
+      person = $1
+      if person =~ /(me|us)/
+        person, do_they = "you", "do you"
+      else
+        do_they = "does #{person}"
+      end
+
+      bot.say_random [
+        "would #{person} like cream or sugar?",
+        "how #{do_they} take it?",
+        "coming right up",
+        "It is by caffeine alone I set my mind in motion",
+        "It is by the beans of Java that thoughts acquire speed",
+        "The hands acquire shakes, the shakes become a warning",
+        "It is by caffeine alone I set my mind in motion"
+      ]
+    else
+      bot.say_random [
+        "I have no idea what you're talking about, #{person}.",
+        "eh?",
+        "oh, interesting",
+        "say more, #{person}",
+        "#{person}, you do realize that you're talking to a bot with a very limited vocabulary, don't you?",
+        "Whatever, #{person}.",
+        "#{bot.other_person(person) || 'Marc'}, tell #{person} to leave me alone.",
+        "Not now, #{person}.",
+        "brb crying",
+        "what do you think, #{person}?",
+        "That's really something.",
+        "but what can I do? I'm just a lowly bot",
+        "I'll get some electrons on that right away",
+        "How do you feel when someone says '#{message.command}' to you, #{person}?"
+      ]
+    end
+  end
+end

data/lib/campfire/polling_bot/plugins/kibitz/spec/kibitz_plugin_spec.rb

@@ -0,0 +1,56 @@
+require 'spec_helper'
+
+describe KibitzPlugin do
+  before do
+    @plugin = KibitzPlugin.new
+  end
+
+  describe "when not addressing wes" do
+    it "does not say anything" do
+      saying("should we deploy?").
+        should_not make_wes_say_anything
+    end
+  end
+
+  describe "when addressing wes" do
+    it "responds with the name of the sender when the message has no command" do
+      saying("wes?").
+        should make_wes_say("John?")
+    end
+
+    it "says whatever the sender tells wes to when using the 'say' command" do
+      saying("wes, say Hi There").
+        should make_wes_say("Hi There")
+    end
+
+    it "responds with the same greeting the sender uses" do
+      saying("hey wes").
+        should make_wes_say("hey John")
+    end
+
+    it "responds with the same goodbye the sender uses" do
+      saying("bye, wes").
+        should make_wes_say("bye John")
+    end
+
+    it "responds politely to praise" do
+      saying("yay! you rock, wes").
+        should make_wes_say("Thanks, John, you're pretty cool yourself.")
+    end
+
+    it "says yes when the sender asks if wes is there" do
+      asking("wes, yt?").
+        should make_wes_say("Yup")
+    end
+
+    it "says hi informally when told to wake up" do
+      asking("you awake wes?").
+        should make_wes_say("Yo.")
+    end
+
+    it "responds politely to thanks" do
+      asking("thanks, wes!").
+        should make_wes_say("No problem.")
+    end
+  end
+end

data/lib/campfire/polling_bot/plugins/reload/reload_plugin.rb

@@ -0,0 +1,24 @@
+# Plugin to allow Wes to update and reload himself
+class ReloadPlugin < Campfire::PollingBot::Plugin
+  accepts :text_message, :addressed_to_me => true
+
+  def process(message)
+    case message.command
+    when /^reload/i
+      bot.say("k")
+      system("git pull origin master && bundle install")
+      if bot.debug
+        head = `git rev-parse HEAD`
+        bot.say("updated to: https://github.com/hackarts/wesabot/commit/#{head}")
+      end
+      bot.say("restarting...")
+      exec *INVOCATION
+      return HALT
+    end
+  end
+
+  # return array of available commands and descriptions
+  def help
+    [['reload', "update and reload #{bot.name}"]]
+  end
+end

data/lib/campfire/polling_bot/plugins/remind_me/remind_me_plugin.rb

@@ -0,0 +1,149 @@
+# Reminders plugin
+require 'chronic'
+class RemindMePlugin < Campfire::PollingBot::Plugin
+  accepts :text_message, :addressed_to_me => true
+  priority 10
+  
+  NUMBER_WORDS = %w{zero one two three four five six seven eight nine ten}
+  PRONOUN_SUBS = [
+    ["I'm", "you're"],
+    ['my', 'your'],
+    ['I', 'you'],
+  ]
+  
+  def process(message)
+    case message.command
+    when /remind\s+(.*?)\s+(in\s+(\S+)\s+(?:seconds|minutes|hours|days|weeks|years))\s+to\s+(.*)/i
+      reminder_person = $1
+      time_string = $2
+      time_value = $3
+      action = $4
+      if i = NUMBER_WORDS.index(time_value.downcase)    
+        time_string.sub!(time_value, i.to_s)     
+      end
+    when /remind\s+(.*?)\s+to\s+(.*?)\s+((?:in|on|at|next|this).*)/i
+      reminder_person = $1
+      action = $2
+      time_string = $3
+    when /remind\s+(.*?)\s+(.*?)\s+to\s+(.*)/i
+      reminder_person = $1
+      time_string = $2
+      action = $3
+    when /(?:list|show) (\S+) reminders/i, /(?:list|show) reminders for (\S+)/i
+      list_reminders(message.person, $1)
+      return HALT
+    when /(list|show) reminders$/
+      list_reminders(message.person, message.person)
+      return HALT
+    when /delete reminder (?:#\s*)?(\d+)/
+      delete_reminder(message.person, $1.to_i)
+      return HALT
+    else
+      return
+    end
+
+    PRONOUN_SUBS.each do |sub|
+      action.gsub!(/\b#{sub[0]}\b/i, sub[1])
+    end
+    
+    begin
+      time = Chronic.parse(time_string.sub(/^at/, ''))
+    rescue RangeError
+      bot.say("Whatever, #{message.person}.")
+      return HALT
+    end
+    
+    if !time || time.kind_of?(Chronic::Span)
+      bot.say("sorry, #{message.person}, but I don't know what '#{time_string}' means.")
+      return HALT
+    end
+    
+    if reminder_person.downcase == 'me'
+      reminder_person = message.person      
+    else
+      reminder_person.gsub!(/^(\w)/) {$1.upcase}
+    end
+    
+    Reminder.create(:person => reminder_person, :action => action, :reminder_time => time)
+    time_now = Time.now
+    if [time.year,time.month,time.day] == [time_now.year,time_now.month,time_now.day]
+      converted_time_string = time.strftime('%I:%M %p').gsub(/^0/,'')
+    else
+      converted_time_string = time.strftime('%c')
+    end
+    
+    bot.say("ok, #{message.person}, I will remind #{reminder_person == message.person ? 'you' : reminder_person} #{time_string} (#{converted_time_string}) to #{action}")
+    return HALT
+  end
+
+  def heartbeat
+    # check reminders every 9 seconds (heartbeat is called every 3 sec)
+    @heartbeat_counter ||= 0
+    @heartbeat_counter += 1
+    return unless (@heartbeat_counter % 3) == 1
+    due_reminders.each do |reminder|
+      bot.say("#{reminder.person}, I'm reminding you to #{reminder.action}")
+      reminder.destroy!
+    end
+  end
+  
+  # return array of available commands and descriptions
+  def help
+    [['remind (me|<person>) [in] <time string> to <message>', "set up a reminder"],
+     ['remind (me|<person>) to <message> (in|on|at|next|this) <time string>', "set up a reminder"],
+     ["[list|show] [person]['s] reminders", "display current reminders for yourself or person"],
+     ["delete reminder <n>", "delete your reminder #n"],
+    ]
+  end
+    
+  private
+
+  def due_reminders
+    Reminder.all(:reminder_time.lte => Time.now)
+  end
+  
+  def list_reminders(current_person, request_person)
+    case request_person.downcase
+    when 'my', 'me'
+      request_person = current_person
+    else
+      request_person.gsub!(/'s$/i,'')
+    end
+    request_person.capitalize!
+    
+    case request_person
+    when /everyone/i, /all/i      
+      if (reminders = Reminder.all).any?
+        bot.say("Here are all the reminders I have:")
+        reminders.each do |reminder|
+          bot.say("#{reminder.id} - [#{reminder.reminder_time}] - #{reminder.person} - #{reminder.action}")
+        end
+      end
+      return
+    else
+      reminders = Reminder.all(:conditions => {:person => request_person}, :order => [:reminder_time])
+      if reminders.any?
+        if request_person == current_person
+          bot.say("Here are the reminders I have for you, #{current_person}:")
+        else
+          bot.say("Here are the reminders for #{request_person}:")
+        end
+        reminders.each do |reminder|
+          bot.say("#{reminder.id} - [#{reminder.reminder_time}] - #{reminder.action}")
+        end
+        return
+      end
+    end
+    bot.say("I couldn't find any reminders for #{request_person}")
+  end 
+  
+  def delete_reminder(current_person, id)
+    reminder = Reminder.get(id)
+    if reminder.person == current_person
+      reminder.destroy
+      bot.say("Ok, I've deleted reminder ##{id}.")
+    else
+      bot.say("Sorry, #{current_person}, but I couldn't find a reminder with that id that belongs to you.")
+    end
+  end    
+end

data/lib/campfire/polling_bot/plugins/remind_me/reminder.rb

@@ -0,0 +1,8 @@
+# used by RemindMe plugin
+class Reminder
+  include DataMapper::Resource
+  property :id,            Serial
+  property :person,        String, :index => true
+  property :action,        Text, :lazy => false
+  property :reminder_time, Time, :required => true, :index => true
+end

data/lib/campfire/polling_bot/plugins/shared/message.rb

@@ -0,0 +1,37 @@
+# used by HistoryPlugin, among others
+class Message
+  include DataMapper::Resource
+  property :id,           Serial
+  property :room,         Integer, :required => true, :index => true
+  property :message_id,   Integer, :required => true
+  property :message_type, String, :length => 20, :required => true, :index => true
+  property :person,       String, :index => true
+  property :user_id,      Integer, :required => false, :index => true
+  property :link,         Text, :lazy => false
+  property :body,         Text, :lazy => false
+  property :timestamp,    Integer, :required => true, :index => true
+
+  belongs_to :user, :required => false
+
+  def self.last_message(name, time = nil)
+    first(:conditions => {
+        :person => name,
+        :message_type.not => 'Enter',
+        :timestamp.lt => (time || Time.now)
+      }, :order => [:timestamp.desc])
+  end
+
+  def self.last_left(name, time = nil)
+    first(:conditions => {
+        :person => name,
+        :message_type => ['Leave','Kick'],
+        :timestamp.lt => (time || Time.now)
+      }, :order => [:timestamp.desc])
+  end
+  
+  # guess whether the message is still on the screen by counting
+  # how many messages there are between it and now
+  def visible?
+    self.class.count(:id.gt => id) < 20
+  end
+end

data/lib/campfire/polling_bot/plugins/shared/user.rb

@@ -0,0 +1,32 @@
+class User
+  include DataMapper::Resource
+  property :id,          Serial
+  property :name,        String, :index => true
+  property :campfire_id, Integer, :index => true
+
+  has n, :messages
+
+  def short_name
+    name && name.split(' ').first
+  end
+
+  def wants_greeting?
+    greeting_setting = GreetingSetting.for_user(self)
+
+    if greeting_setting
+      greeting_setting.wants_greeting
+    else
+      true
+    end
+  end
+
+  def wants_greeting=(yesno)
+    greeting_setting = GreetingSetting.for_user(self)
+
+    if greeting_setting
+      greeting_setting.update(:wants_greeting => yesno)
+    else
+      GreetingSetting.create(:user => self, :wants_greeting => yesno)
+    end
+  end
+end

data/lib/campfire/polling_bot/plugins/sms/sms_plugin.rb

@@ -0,0 +1,88 @@
+# Plugin to send SMS (well, email) messages
+require 'net/smtp'
+class SMSPlugin < Campfire::PollingBot::Plugin
+  accepts :text_message, :addressed_to_me => true
+  
+  def process(message)
+    case message.command
+    when /(?:set)?\s+my\s+sms\s+address\s*(?:is|to|:)\s*(.+)/i
+      address = $1
+      if address =~ /"mailto:(.*?)"/
+        address = $1
+      end
+      update_sms_address(message.person, address)
+      bot.say("OK, #{message.person}, I've set your SMS address to #{address}")
+      return HALT
+    when /(?:set)?\s+(\w+?)'s\s+sms\s+address\s*(?:is|to|:)\s*(.+)/i
+      person = $1
+      address = $2
+      if address =~ /"mailto:(.*?)"/
+        address = $1
+      end
+      update_sms_address(person, address)
+      bot.say("OK, #{message.person}, I've set #{$1}'s SMS address to #{$2}")
+      return HALT     
+    when /list sms addresses/i
+      if (settings = SMSSetting.all(:order => [:person])).any?
+        bot.say("Here are the SMS addresses I have:")
+        settings.each { |setting| bot.say("  #{setting.person} - #{setting.address}") }
+      else
+        bot.say("Sorry, I don't have any SMS addresses yet.")
+      end
+      return HALT
+    when /(?:sms|text|txt)\s+([^\s:]+):?\s*(.*)/i
+      if address = send_sms(message.person, $1, $2)
+        bot.say("Sent an SMS to #{$1} at #{address}")
+      end
+      return HALT
+    end
+  end
+  
+  # return array of available commands and descriptions
+  def help
+    [['set my sms address to: <address>', "set your sms address"],
+     ["set <person>'s sms address to", "set someone else's sms address"],
+     ['(sms|text|txt) <person>: <message>', "send an sms message"],
+     ["list sms addresses", "list all sms addresses"]
+    ]
+  end
+  
+  private
+  
+  def update_sms_address(person, address)
+    person.downcase!
+    if setting = find_sms_setting(person)
+      setting.update_attributes(:address => address)
+    else
+      SMSSetting.create(:person => person, :address => address)
+    end
+    return address
+  end
+  
+  def send_sms(sender, recipient, message)
+    if setting = find_sms_setting(recipient)
+      send_email(sender, setting.address, message)
+      return setting.address
+    else
+      bot.say("Sorry, I don't have an SMS address for #{recipient}.")
+      return nil
+    end
+  end
+  
+  def find_sms_setting(person)
+    SMSSetting.first(:person => person.downcase)
+  end  
+  
+  def send_email(from, to, message)
+    from += "@wesabe.com"
+    msg = "From: #{from}\nTo: #{to}\n\n#{message}"
+    begin
+      Net::SMTP.start('localhost', 25, 'wesabe.com') { |smtp| smtp.send_message(msg, from, to) }
+      return true
+    rescue Exception => e
+      bot.say("Hmm...couldn't send mail. Here's what happened when I tried:")
+      bot.paste(e.message)
+      return false
+    end
+  end
+end

data/lib/campfire/polling_bot/plugins/sms/sms_setting.rb

@@ -0,0 +1,7 @@
+# used by SMSPlugin
+class SMSSetting
+  include DataMapper::Resource
+  property :id,            Serial
+  property :person,        String, :index => true
+  property :address,       String
+end

data/lib/campfire/polling_bot/plugins/time/time_plugin.rb

@@ -0,0 +1,17 @@
+# Simple sample plugin to just give the current time
+class TimePlugin < Campfire::PollingBot::Plugin
+  accepts :text_message, :addressed_to_me => true
+  
+  def process(message)
+    case message.command
+    when /time/i
+      bot.say("#{message.person}, the time is #{Time.now}")
+      return HALT
+    end
+  end
+
+  # return array of available commands and descriptions
+  def help
+    [['time', "say the current time"]]
+  end
+end

data/lib/campfire/polling_bot/plugins/tweet/tweet.rb

@@ -0,0 +1,52 @@
+# used by TweetPlugin
+class Tweet
+  class TwitterError < RuntimeError
+    attr_reader :http_result
+
+    def initialize(http_result)
+      @http_result = http_result
+    end
+  end
+
+  include DataMapper::Resource
+  property :id,        Serial
+  property :message,   String, :required => true
+  property :person,    String
+  property :timestamp, Integer, :required => true
+
+  def tweet(user, pass, proxy=nil)
+    self.class.tweet(message, user, pass, proxy)
+  end
+
+  def self.tweet(message, user, pass, proxy=nil)
+    result = post(message, user, pass, proxy)
+    if result.code == 200
+      return true
+    else
+      raise TwitterError.new(result.body)
+    end
+  end
+
+  def self.list
+    all(:order => [:timestamp.asc])
+  end
+
+  def to_s
+    message
+  end
+
+  private
+
+  # post a message to twitter
+  def self.post(message, user, pass, proxy=nil)
+    options = { 
+      :query => {:status => message}, 
+      :basic_auth => {:username => user, :password => pass} 
+    }
+    if proxy ||= ENV['HTTP_PROXY'] 
+      proxy_uri = URI.parse(proxy)
+      options.update(:http_proxyaddr => proxy_uri.host, :http_proxyport => proxy_uri.port)
+    end
+    HTTParty.post("http://twitter.com/statuses/update.xml", options)
+  end
+end

data/lib/campfire/polling_bot/plugins/tweet/tweet_plugin.rb

@@ -0,0 +1,117 @@
+# plugin to send a tweet to a Twitter account
+class TweetPlugin < Campfire::PollingBot::Plugin
+  accepts :text_message, :addressed_to_me => true
+  priority 20
+
+  def process(message)
+    case message.command
+    when /^(?:tweet|twitter):?\s*("?)(.*?)\1$/i
+      begin
+        text = strip_links($2)
+        post_tweet(text)
+      rescue TwitterError => ex
+        bot.say("Hmm...didn't work. Got this response:")
+        bot.paste(ex.http_result.content)
+      end
+      return HALT
+    when /^(?:save|queue)\s+(?:tweet|twitter):?\s*("?)(.*?)\1$/i
+      tweet = Tweet.create(:person => message.person, :message => strip_links($2), :timestamp => message.timestamp.to_i)
+      bot.say("Ok, saved for later: #{tweet}")
+      return HALT
+    when /^(?:show|list)\s+(?:all\s+)?(?:queued\s+)?(?:tweets|twitters)$/i
+      act_on_list do |list|
+        bot.say("Here are the tweets in the queue:")
+        i = 0
+        bot.paste(list.map{|tweet| "#{i+=1}. #{tweet}"}.join("\n"))
+      end
+      return HALT
+    when /^show\s+next\s+(?:tweet|twitter)$/i
+      act_on_list do |list|
+        bot.say("Next tweet to post: #{list.first}")
+      end
+      return HALT
+    when /^(?:post|send)\s+next\s+(?:tweet|twitter)$/i
+      act_on_list do |list|
+        tweet = list.first
+        post_tweet(tweet)
+        tweet.destroy
+      end
+      return HALT
+    when /^(?:post|send)\s+tweet\s+(\d+)$/i
+      act_on_tweet($1.to_i-1) do |tweet|
+        post_tweet(tweet)
+        tweet.destroy
+      end
+      return HALT
+    when /^delete\s+tweet\s+(\d+)$/i
+      act_on_tweet($1.to_i-1) do |tweet|
+        tweet.destroy
+        bot.say("Okay, deleted tweet #{$1}: #{tweet}")
+      end
+      return HALT
+    end
+  end
+
+  # return array of available commands and descriptions
+  def help
+    user = username || 'the configured user'
+    [['tweet: <message>', "post <message> to #{user}'s twitter account"],
+     ['save tweet: <message>', "save <message> for later"],
+     ['show tweets', "shows the queued tweets for #{user}'s twitter account"],
+     ['show next tweet', "shows the oldest queued twitter message"],
+     ['post next tweet', "sends the oldest queued twitter message"],
+     ['post tweet <n>', "sends the <n>th tweet from the list"],
+     ['delete tweet <n>', "deletes the <n>th tweet from the list"]]
+  end
+
+  private
+
+  def username
+    config && config['username']
+  end
+
+  def password
+    config && config['password']
+  end
+
+  def proxy
+    config && config['proxy']
+  end
+
+  def post_tweet(tweet)
+    if username.nil? || password.nil?
+      bot.say("The twitter plugin is not properly configured!")
+      return
+    end
+
+    begin
+      Tweet.tweet(tweet.to_s, username, password, proxy)
+      bot.say("Ok, tweeted: #{tweet}")
+    rescue Tweet::TwitterError => ex
+      bot.say("Hmm...didn't work. Got this response:")
+      bot.paste(ex.http_result.content)
+    end
+  end
+
+  def act_on_list
+    list = Tweet.list
+    if list.empty?
+      bot.say("There are no queued tweets.")
+    else
+      yield list
+    end
+  end
+
+  def act_on_tweet(n)
+    if tweet = Tweet.list[n]
+      yield tweet
+    else
+      bot.say("Could not find tweet #{n}")
+    end
+  end
+
+  # strip links from CF messages
+  def strip_links(msg)
+    msg.gsub(/<a href="([^"]*)".*?>.*?<\/a>/, '\1')
+  end
+end

data/lib/campfire/polling_bot/plugins/tweet/tweet_plugin.yml.sample

@@ -0,0 +1,4 @@
+# Configuration file for TweetPlugin
+username: your_twitter_username
+password: your_twitter_password
+proxy: http://your.proxy.com:8080 # optional