#!/usr/bin/env ruby
require 'rubygems'
require 'open-uri'
require 'json'
require 'net/http'

TWITTER_USER = ""
TWITTER_PASS = ""
RETWEET_CODE = "" # This code is given out to trusted people allowed to retweet

LAST_TWEET_FILE = ".last_tweet_code"

def last_tweet
  @last_tweet ||= open(LAST_TWEET_FILE).read rescue 0
end

def url
  "http://twitter.com/direct_messages.json?since_id=#{last_tweet}"
end

def dm_from_twitter
  open(url, :http_basic_authentication => [TWITTER_USER, TWITTER_PASS]).read
end

def direct_messages
  @direct_messages ||= JSON.parse(dm_from_twitter)
end

def update_last_tweet(tweet_id)
  File.open(LAST_TWEET_FILE, "w") do |f|
    f.puts tweet_id
  end
end

def send_update(tweet)
  Net::HTTP.post_form(URI.parse("http://#{TWITTER_USER}:#{TWITTER_PASS}@twitter.com/statuses/update.json"),
    {'status' => tweet}
  )
end

def valid?(tweet)
  !tweet.match(/\b#{RETWEET_CODE}\b/).nil?
end

def strip(tweet)
  tweet.gsub(/\b#{RETWEET_CODE}\b\s*/, '')
end

direct_messages.reverse.each do |dm|
  next if dm["id"] == last_tweet
  tweet = "from @#{dm["sender_screen_name"]}: #{strip(dm["text"])}" if valid?(dm["text"])
  send_update(tweet)
end

update_last_tweet(direct_messages.first["id"]) if direct_messages.first