roflbot 0.0.1 → 0.0.3

data/README

@@ -1 +1 @@
-roflbot is an AIM bot.
+roflbot is an bot that says stuff on AIM and Twitter.

data/Rakefile

@@ -5,13 +5,14 @@ begin
   require 'jeweler'
   Jeweler::Tasks.new do |gem|
     gem.name = "roflbot"
-    gem.summary = %Q{roflbot is a simple AIM bot}
-    gem.description = %Q{roflbot is a simple AIM bot, yo}
+    gem.summary = %Q{roflbot is a bot that says stuff on AIM and Twitter}
+    gem.description = %Q{roflbot will make you a rofl waffle}
     gem.email = "viking415@gmail.com"
     gem.homepage = "http://github.com/viking/roflbot"
     gem.authors = ["Jeremy Stephens"]
     gem.add_dependency "net-toc"
     gem.add_dependency "treetop"
+    gem.add_dependency "twitter"
     # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
   end
   Jeweler::GemcutterTasks.new

data/VERSION

@@ -1 +1 @@
-0.0.1
+0.0.3

data/lib/roflbot.rb

@@ -4,6 +4,7 @@ require 'yaml'
 require 'rubygems'
 require 'net/toc'
 require 'treetop'
+require 'twitter'
 
 module Roflbot
 end

data/lib/roflbot/base.rb

@@ -21,19 +21,38 @@ module Roflbot
     end
 
     extend Forwardable
-    def initialize(username, password, options = {})
-      @username = username
-      @password = password
+    TWITTER = { :token => "7rfuoIhSGN7OQVvDLzJrcg", :secret => "I2o4SmRWVV4Yo7XhGgLrKGJq1KgpC8VrluyA9LLuH0" }
+
+    attr_reader :options
+    def initialize(options = {})
       @options = options
       @expectations = []
+      accounts = @options["accounts"]
+
+      @aim = Net::TOC.new(*accounts["AIM"].values_at("username", "password"))
+      @aim.on_im { |m, b, a| on_im(m, b, a) }
+
+      oauth = Twitter::OAuth.new(TWITTER[:token], TWITTER[:secret])
+      oauth.authorize_from_access(*accounts["Twitter"].values_at("token", "secret"))
+      @twitter = Twitter::Base.new(oauth)
+      @since_id = accounts['Twitter']['since_id']
+    end
 
-      @client = Net::TOC.new(@username, @password)
-      @client.on_im { |m, b, a| on_im(m, b, a) }
+    def_delegator :@aim, :disconnect
+
+    def start
+      @aim.connect
+      @thread = Thread.new { loop { respond_via_twitter; sleep 60 } }
+    end
+
+    def wait
+      @aim.wait
     end
 
-    def_delegator :@client, :connect
-    def_delegator :@client, :disconnect
-    def_delegator :@client, :wait
+    def stop
+      @aim.disconnect
+      @thread.kill  if @thread
+    end
 
     def expects(regexp)
       exp = Expectation.new(regexp)
@@ -53,5 +72,29 @@ module Roflbot
         break
       end
     end
+
+    def respond_via_twitter
+      mentions = @twitter.mentions(@since_id ? {:since_id => @since_id} : {})
+      if !mentions.empty?
+        @since_id = mentions[-1].id
+        @options['accounts']['Twitter']['since_id'] = @since_id
+      end
+
+      mentions.each do |tweet|
+        @expectations.each do |expectation|
+          # throw away beginning mention
+          message = tweet.text.sub(/^@[^\s]+\s+/, "")
+
+          next  if !expectation.matches?(message)
+          case expectation.response
+          when Proc
+            @twitter.update("@#{tweet.user.screen_name} #{expectation.response.call}")
+          when String
+            @twitter.update("@#{tweet.user.screen_name} #{expectation.response}")
+          end
+          break
+        end
+      end
+    end
   end
 end

data/lib/roflbot/runner.rb

@@ -1,19 +1,20 @@
 module Roflbot
   class Runner
     def initialize(argv = ARGV)
-      username = nil
-      password = nil
-      options = {}
+      @config = nil
       OptionParser.new do |opts|
-        opts.on("-u", "--username USERNAME") { |u| username = u }
-        opts.on("-p", "--password PASSWORD") { |p| password = p }
-        opts.on("-c", "--config FILENAME") { |c| options = YAML.load_file(c) }
+        opts.on("-c", "--config FILENAME") { |c| @config = c }
       end.parse!(argv)
 
-      bot = SentenceBot.new(username, password, options)
-      bot.connect
-      Signal.trap("INT") { bot.disconnect }
-      bot.wait
+      @bot = SentenceBot.new(YAML.load_file(@config))
+      Signal.trap("INT") { stop }
+      @bot.start
+      @bot.wait
+    end
+
+    def stop
+      @bot.stop
+      File.open(@config, "w") { |f| f.write(@bot.options.to_yaml) }
     end
   end
 end

data/lib/roflbot/sentence_bot.rb

@@ -6,7 +6,7 @@ module Roflbot
       super
       parser = SentenceParser.new
       @sentences = []
-      @options.delete("sentences").each do |sentence|
+      @options["sentences"].each do |sentence|
         node = parser.parse(sentence)
         @sentences << node
       end

data/roflbot.gemspec

@@ -5,13 +5,13 @@
 
 Gem::Specification.new do |s|
   s.name = %q{roflbot}
-  s.version = "0.0.1"
+  s.version = "0.0.3"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Jeremy Stephens"]
   s.date = %q{2010-06-30}
   s.default_executable = %q{roflbot}
-  s.description = %q{roflbot is a simple AIM bot, yo}
+  s.description = %q{roflbot will make you a rofl waffle}
   s.email = %q{viking415@gmail.com}
   s.executables = ["roflbot"]
   s.extra_rdoc_files = [
@@ -41,7 +41,7 @@ Gem::Specification.new do |s|
   s.rdoc_options = ["--charset=UTF-8"]
   s.require_paths = ["lib"]
   s.rubygems_version = %q{1.3.6}
-  s.summary = %q{roflbot is a simple AIM bot}
+  s.summary = %q{roflbot is a bot that says stuff on AIM and Twitter}
   s.test_files = [
     "test/helper.rb",
      "test/roflbot/test_sentence_bot.rb",
@@ -56,13 +56,16 @@ Gem::Specification.new do |s|
     if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
       s.add_runtime_dependency(%q<net-toc>, [">= 0"])
       s.add_runtime_dependency(%q<treetop>, [">= 0"])
+      s.add_runtime_dependency(%q<twitter>, [">= 0"])
     else
       s.add_dependency(%q<net-toc>, [">= 0"])
       s.add_dependency(%q<treetop>, [">= 0"])
+      s.add_dependency(%q<twitter>, [">= 0"])
     end
   else
     s.add_dependency(%q<net-toc>, [">= 0"])
     s.add_dependency(%q<treetop>, [">= 0"])
+    s.add_dependency(%q<twitter>, [">= 0"])
   end
 end
 

data/test/helper.rb

@@ -3,6 +3,7 @@ require 'rubygems'
 require 'mocha'
 require 'delegate'
 require 'ruby-debug'
+require 'tempfile'
 require File.dirname(__FILE__) + "/../lib/roflbot"
 
 class FakeClient < Mocha::Mock
@@ -17,13 +18,31 @@ class Test::Unit::TestCase
     @client = FakeClient.new
     @buddy = stub("Buddy")
     Net::TOC.stubs(:new).returns(@client)
+
+    @oauth = stub("OAuth", :authorize_from_access => [])
+    @twitter = stub("Twitter", :mentions => [])
+    Twitter::OAuth.stubs(:new).returns(@oauth)
+    Twitter::Base.stubs(:new).returns(@twitter)
   end
 
   def receive_im(message, auto = false)
     @client.on_im_block.call(message, @buddy, auto)
   end
 
+  def fake_tweet(message, id)
+    Hashie::Mash.new("user" => { "screen_name" => "dudeguy" }, "text" => "@roflbot #{message}", "id" => id)
+  end
+
   def fixture_filename(name)
     File.dirname(__FILE__) + "/fixtures/#{name}"
   end
+
+  def create_config(name)
+    tmp = Tempfile.open("roflbot_config")
+    file = File.open(fixture_filename(name), "r")
+    tmp.write(file.read)
+    file.close
+    tmp.close
+    tmp.path
+  end
 end

data/test/roflbot/test_base.rb

@@ -2,39 +2,90 @@ require File.dirname(__FILE__) + "/../helper"
 
 module Roflbot
   class TestBase < Test::Unit::TestCase
+    def setup
+      super
+      @options = {
+        "accounts" => {
+          "AIM" => { "username" => "roflbot", "password" => "password" },
+          "Twitter" => { "token" => "abcdef", "secret" => "123456" }
+        }
+      }
+    end
+
     def test_new
       Net::TOC.expects(:new).with("roflbot", "password").returns(@client)
-      roflbot = Base.new("roflbot", "password")
+      Twitter::OAuth.expects(:new).with(Base::TWITTER[:token], Base::TWITTER[:secret]).returns(@oauth)
+      @oauth.expects(:authorize_from_access).with("abcdef", "123456")
+      Twitter::Base.expects(:new).with(@oauth).returns(@twitter)
+      roflbot = Base.new(@options)
     end
 
-    def test_connect
-      roflbot = Base.new("roflbot", "password")
+    def test_start
+      roflbot = Base.new(@options)
       @client.expects(:connect)
-      roflbot.connect
+      roflbot.start
     end
 
     def test_wait
-      roflbot = Base.new("roflbot", "password")
+      roflbot = Base.new(@options)
       @client.expects(:wait)
       roflbot.wait
     end
 
-    def test_disconnect
-      roflbot = Base.new("roflbot", "password")
+    def test_stop
+      roflbot = Base.new(@options)
       @client.expects(:disconnect)
-      roflbot.disconnect
+      roflbot.stop
     end
 
-    def test_responds_with_string
-      roflbot = Base.new("roflbot", "password")
+    def test_responds_to_aim_with_string
+      roflbot = Base.new(@options)
       roflbot.expects(/^hey$/).responds("oh hai")
 
       @buddy.expects(:send_im).with("oh hai")
       receive_im("hey")
     end
 
-    def test_responds_with_block
-      roflbot = Base.new("roflbot", "password")
+    def test_responds_to_twitter_with_string
+      roflbot = Base.new(@options)
+      roflbot.expects(/^hey$/).responds("oh hai")
+
+      seq = sequence("foo")
+      @twitter.expects(:mentions).returns([fake_tweet("hey", 12345)]).in_sequence(seq)
+      @twitter.expects(:update).with("@dudeguy oh hai").in_sequence(seq)
+      roflbot.respond_via_twitter
+
+      seq = sequence("bar")
+      @twitter.expects(:mentions).with(:since_id => 12345).returns([fake_tweet("hey", 23456)]).in_sequence(seq)
+      @twitter.expects(:update).with("@dudeguy oh hai").in_sequence(seq)
+      roflbot.respond_via_twitter
+    end
+
+    def test_saves_since_id
+      roflbot = Base.new(@options)
+      roflbot.expects(/^hey$/).responds("oh hai")
+
+      seq = sequence("foo")
+      @twitter.expects(:mentions).returns([fake_tweet("hey", 12345)]).in_sequence(seq)
+      @twitter.expects(:update).with("@dudeguy oh hai").in_sequence(seq)
+      roflbot.respond_via_twitter
+
+      assert_equal 12345, roflbot.options['accounts']['Twitter']['since_id']
+    end
+
+    def test_uses_saved_since_id
+      @options['accounts']['Twitter']['since_id'] = 1337
+      roflbot = Base.new(@options)
+      roflbot.expects(/^hey$/).responds("oh hai")
+
+      seq = sequence("foo")
+      @twitter.expects(:mentions).with(:since_id => 1337).returns([fake_tweet("hey", 12345)]).in_sequence(seq)
+      @twitter.expects(:update).with("@dudeguy oh hai").in_sequence(seq)
+      roflbot.respond_via_twitter
+    end
+
+    def test_responds_to_aim_with_block
+      roflbot = Base.new(@options)
       roflbot.expects(/^hey$/).responds { "omg wtf" }
 
       @buddy.expects(:send_im).with("omg wtf")

data/test/roflbot/test_runner.rb

@@ -2,10 +2,29 @@ require File.dirname(__FILE__) + "/../helper"
 
 module Roflbot
   class TestRunner < Test::Unit::TestCase
+    def setup
+      super
+      @bot = stub("SentenceBot", :start => nil, :wait => nil, :options => {})
+      SentenceBot.stubs(:new).returns(@bot)
+    end
+
     def test_runs_sentence_bot
-      bot = mock("SentenceBot", :connect => nil, :wait => nil)
-      SentenceBot.expects(:new).with("foo", "bar", { "foo" => "bar" }).returns(bot)
-      Runner.new(%W{-u foo -p bar -c #{fixture_filename("bogus.yml")}})
+      @bot.expects(:start)
+      @bot.expects(:wait)
+      SentenceBot.expects(:new).with({ "foo" => "bar" }).returns(@bot)
+      Runner.new(%W{-c #{create_config("bogus.yml")}})
+    end
+
+    def test_stop
+      config = create_config("bogus.yml")
+      runner = Runner.new(%W{-c #{config}})
+      @bot.stubs(:options).returns({ "omg" => "whoa" })
+
+      @bot.expects(:stop)
+      runner.stop
+
+      hash = YAML.load_file(config)
+      assert_equal({ "omg" => "whoa" }, hash)
     end
   end
 end

data/test/roflbot/test_sentence_bot.rb

@@ -2,16 +2,25 @@ require File.dirname(__FILE__) + "/../helper"
 
 module Roflbot
   class TestSentenceBot < Test::Unit::TestCase
+    def setup
+      super
+      @options = {
+        "accounts" => {
+          "AIM" => {"username" => 'dudeguy', "password" => 'password'},
+          "Twitter" => { "token" => "abcdef", "secret" => "123456" }
+        },
+        "sentences" => ["Hey (noun), (ending)"],
+        "noun" => %w{dude guy},
+        "ending" => ["sup?", "I love you."]
+      }
+    end
+
     def test_base_superclass
       assert_equal Base, SentenceBot.superclass
     end
 
     def test_responses
-      bot = SentenceBot.new('dudeguy', 'password', {
-        "sentences" => ["Hey (noun), (ending)"],
-        "noun" => %w{dude guy},
-        "ending" => ["sup?", "I love you."]
-      })
+      bot = SentenceBot.new(@options)
 
       @buddy.expects(:send_im).times(100).with do |message|
         message =~ /^Hey (dude|guy), (sup\?|I love you.)$/

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 1
-  version: 0.0.1
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - Jeremy Stephens
@@ -41,7 +41,19 @@ dependencies:
         version: "0"
   type: :runtime
   version_requirements: *id002
-description: roflbot is a simple AIM bot, yo
+- !ruby/object:Gem::Dependency 
+  name: twitter
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id003
+description: roflbot will make you a rofl waffle
 email: viking415@gmail.com
 executables: 
 - roflbot
@@ -97,7 +109,7 @@ rubyforge_project:
 rubygems_version: 1.3.6
 signing_key: 
 specification_version: 3
-summary: roflbot is a simple AIM bot
+summary: roflbot is a bot that says stuff on AIM and Twitter
 test_files: 
 - test/helper.rb
 - test/roflbot/test_sentence_bot.rb