root/lang/ruby/flagtter/flagtter.rb @ 34281

#!/usr/local/bin/ruby -Ku
# -*- coding: utf-8 -*-

require 'net/http'
require 'rexml/document'
require 'time'
require 'timeout'
require 'yaml'

LOG_FILE    = ENV["HOME"] + "/var/log/flagtter.log"
CONFIG_FILE = ENV["HOME"] + "/.flagtterrc"

config = YAML.load_file(CONFIG_FILE)
USERNAME = config["username"]
PASSWORD = config["password"]

Net::HTTP.version_1_2

log = File.open(LOG_FILE, "a")
result = ""

# 検索結果の RSS を入手
begin
  timeout(60) {
    http = Net::HTTP.new("pcod.no-ip.org");
    response = http.get("/yats/search?query=俺+たら+んだ+-user:flagtter,linux_cafe,akiba_twit,sairo_bot,gakukino&lang=ja&rss&fast",
                      {'User-Agent' => "Ruby/#{RUBY_VERSION}"});
    result = response.body;
  }
rescue
  # RSS が入手できなかったら終了
  log.close
  exit(0)
rescue TimeoutError
  # RSS が入手できなかったら終了
  log.close
  exit(0)
end

# RSS から文字列を抽出 → 投稿
rexml = REXML::Document.new(result)
rexml.elements.each("//feed/entry") { |item|
  name = item.elements["title"].text
  time = Time.parse(item.elements["updated"].text.gsub(/T/, " ").gsub(/Z/, ""))
  if ((log.mtime - 9 * 60 * 60) < time)
    text = item.elements["summary"].text
    if (regex = text.match(/俺.*たら.*んだ/))
      text = regex.to_a[0];
      status = "『#{text}』とつぶやいた @ #{name} にフラグがセットされました。"
      # 投稿
      begin
        timeout(120) {
          request1 = Net::HTTP::Post.new('/statuses/update.json')
          request1.basic_auth(USERNAME, PASSWORD)
          request1.body = "status=" + status
          Net::HTTP.start("twitter.com", 80) { |http|
            response1 = http.request(request1)
          }
          # 発言者をフォロー
          # 投稿に成功した場合だけフォローするようにする
          request2 =
            Net::HTTP::Post.new("/friendships/create/#{name}.json")
          request2.basic_auth(USERNAME, PASSWORD)
          Net::HTTP.start("twitter.com", 80) { |http|
            response2 = http.request(request2)
          }
        }
      rescue
        # ignore
      rescue Timeout::Error
        # ignore
      end
      log.puts "#{time} - #{status}" # 失敗してもログには追記
    end
    break # 1 回につき 1 投稿のみ
  end
}

log.flush
log.close