twitter_reudy.rb

# Copyright (C) 2011 Glass_saga <glass.saga@gmail.com>

INTERVAL = 60 # タイムラインを取得する間隔
ABORT_ON_API_LIMIT = false # API制限に引っかかった時にabortするかどうか

trap(:INT) { exit }

require 'optparse'
require 'rubytter'
require 'highline'
require 'time'
require_relative 'lib/reudy/bot_irc_client'
require_relative 'lib/reudy/reudy'
require_relative 'lib/reudy/reudy_common'

module Gimite
  class TwitterClient
    include(Gimite)

    def initialize(user)
      @user = user
      @user.client = self
      @last_tweet = Time.now

      key = user.settings[:twitter][:key]
      secret = user.settings[:twitter][:secret]
      cons = OAuth::Consumer.new(key, secret, site: "http://api.twitter.com")

      unless File.exist?(File.dirname(__FILE__) + "/token")
        request_token = cons.get_request_token
        puts "Access This URL and press 'Allow' => #{request_token.authorize_url}"
        pin = HighLine.new.ask('Input key shown by twitter: ')
        access_token = request_token.get_access_token(oauth_verifier: pin)
        open(File.dirname(__FILE__) + "/token", "w") do |f|
          f.puts access_token.token
          f.puts access_token.secret
        end
      end

      keys = File.read(File.dirname(__FILE__) + "/token").split(/\r?\n/).map(&:chomp)

      token = OAuth::AccessToken.new(cons, keys[0], keys[1])

      @r = OAuthRubytter.new(token)
    end

    attr_accessor :r

    def onTweet(status)
      @user.onOtherSpeak(status.user.screen_name, status.text)
    end

    # 補助情報を出力
    def outputInfo(s)
      puts "(#{s})"
    end

    # 発言する
    def speak(s)
      time = Time.now
      if time - @last_tweet > INTERVAL
        @r.update(s)
        puts "tweeted: #{s}"
        @last_tweet = time
      end
    end
  end

  opt = OptionParser.new

  directory = 'public'
  opt.on('-d DIRECTORY') do |v|
    directory = v
  end

  db = 'pstore'
  opt.on('--db DB_TYPE') do |v|
    db = v
  end

  mecab = nil
  opt.on('-m', '--mecab') do
    mecab = true
  end

  opt.parse!(ARGV)

  # twitter用ロイディを作成
  client = TwitterClient.new(Reudy.new(directory, {}, db, mecab))

  loop do
    begin
      since_id = -1
      client.r.friends_timeline(since_id: since_id).each do |status|
        puts "#{status.user.screen_name}: #{status.text}"
        since_id = status.id
        client.onTweet(status)
      end
      sleep(INTERVAL)
    rescue => ex
      case ex.message
      when "Could not authenticate with OAuth."
        abort ex.message
      when /Rate limit exceeded./
        if ABORT_ON_API_LIMIT
          abort ex.message
        else
          reset_time = Time.parse(r.limit_status[:reset_time])
          puts ex.message
          puts "API制限は#{reset_time}に解除されます。"
          sleep(reset_time - Time.now)
        end
      else
        puts ex.message
      end
    end
  end
end