require 'yaml'

require 'sinatra'

require 'rest_client'

require 'json'

require 'rack/utils'

#require 'oniguruma'

require 'oauth'

#RestClient.proxy = "http://localhost:8118"

#RestClient.proxy = "http://bess-proxy.mdeca.org:7033"

set :host, 'localhost'

disable :logging

set :views, File.dirname(__FILE__) + "/views"

set :public, File.dirname(__FILE__) + '/public'

configure do

  if File.exists? 'timeline.yml'

    @@data = YAML.load_file 'timeline.yml'

  else

    @@data = ""

  end

  @@config = YAML.load_file ENV['HOME'] + "/.coefficient.yml"

  consumer = OAuth::Consumer.new(@@config['credentials']['twitter']['consumer_key'],

                                 @@config['credentials']['twitter']['consumer_secret'],

                                 { :site => "https://twitter.com" })

  @@sites = {}

  @@sites['twitter'] = {}

  @@sites['twitter']['api'] = "https://twitter.com/"

  @@sites['twitter']['url'] = "https://twitter.com/"

  @@sites['twitter']['consumer'] = consumer

  @@sites['twitter']['access_token'] = OAuth::AccessToken.new(consumer,

                                                              @@config['credentials']['twitter']['access_token'],

                                                              @@config['credentials']['twitter']['access_secret'])

  @@sites['identica'] = {}

  @@sites['identica']['api'] = "https://identi.ca/api/"

  @@sites['identica']['url'] = "https://identi.ca/"

end

def make_user_link(origin, name)

#  origin = data['origin']

#  name = data['user']['screen_name']

  string = %Q(<a href="#{@@sites[origin]['url']}#{name}">#{name}</a>)

  return string

end

def parse_text(origin, text)

  if origin == "identica"

    return identicaify text

  else

    return twitterify text

  end

end

def escape_amp(text)

  return "???" if text.nil?

  return text.gsub(/&(?!(amp|lt|gt|#39|quot);)/, "&")

end

def html(text)

  return escape_amp(text).

    gsub("<", "&lt;").

    gsub(">", ">").

    gsub("'", "'").

    gsub("\"", """)

end

def linkify(text)

  new = text.dup

  re = Regexp.new(/(https?:\/\/([\w%\-+?#\/=~;]|&(?!quot;)|[.,](?!\z|\s))+)/)

  new.gsub!(re, '<a href="\1">\1</a>')

  #ugh = Oniguruma::ORegexp.new('(?<!http:\/\/)(ur1.ca\/([a-zA-Z0-9_%\-+?#\/=~]|&(?!quot;)|[.,](?!\z|\s))+)')

  ugh = Regexp.new('(?<!http:\/\/)(ur1.ca\/([a-zA-Z0-9_%\-+?#\/=~]|&(?!quot;)|[.,](?!\z|\s))+)', Regexp::EXTENDED)

  new.gsub!(ugh, '<a href="http://\1">\1</a>')

#  new.gsub!(/(?<!http:\/\/)(ur1.ca\/([a-zA-Z0-9_%\-+?#\/=~]|&(?!quot;)|[.,](?!\z|\s))+)/, '<a href="http://\1">\1</a>')

  return new

end

def username_link(origin, text)

  if origin == 'twitter'

    return text.

      gsub(/@([0-9a-zA-Z_]+)/, '@<a href="http://twitter.com/\1">\1</a>')

  elsif origin == 'identica'

    return text.

      gsub(/@([0-9a-zA-Z_]+)/, '@<a href="http://identi.ca/\1">\1</a>')

  end

end

def hashtag_link(origin, text)

  if origin == 'twitter'

    return text.

      gsub(/(^|\s)#(?:(?!39;)([A-Za-z0-9_\-]+|\.(?!\z|\s)))/, '\1#<a href="http://twitter.com/search?q=\2">\2</a>')

  elsif origin == 'identica'

    return text.

      gsub(/(^|\s)#(?:(?!39;)([A-Za-z0-9_\-]+|\.(?!\z|\s)))/, '\1#<a href="http://identi.ca/tag/\2">\2</a>')

  end

end

def twitterify(text)

  text = html(text)

  text = linkify(text)

  text = username_link('twitter', text)

  text = hashtag_link('twitter', text)

  return text

end

def identicaify(text)

  text = html(text)

  text = linkify(text)

  text = username_link('identica', text)

  text = hashtag_link('identica', text)

  text.gsub!(/(?:!([A-Za-z0-9]+))/, '!<a href="http://identi.ca/group/\1">\1</a>')

  return text

end

def convert_time_to_est(time)

  new_time = DateTime.parse(time).new_offset(Rational(-1,6))

  return new_time.strftime("%A, %B %e %Y at %l:%M %P")

end

def request_status(origin, id)

  api_url = @@sites[origin]['api'].dup

  api_url.gsub!(/(https?:\/\/)(.+)/, '\1' <<

                @@config['credentials']['username'] << ":" <<

                @@config['credentials']['password'] << "@" <<

                '\2')

  url = api_url

  path = ""

  path << "/statuses/show/" << id << ".json"

  url << path

  if origin == 'twitter'

    puts "OAuth: getting #{path}"

    get = @@sites[origin]['access_token'].get(path)

  else

    puts "get #{url.sub(/\w+:\w+@/, "****:****@")}..."

    get = RestClient.get url, { :accept => "text/json" }

  end

  json = get.body

  parsed_data = JSON.parse json

  return parsed_data

end

def get_since(origin, id)

  page = 1

  received_data = Array.new

  api_url = @@sites[origin]['api'].dup

  api_url.gsub!(/(https?:\/\/)(.+)/, '\1' <<

                @@config['credentials']['username'] << ":" <<

                @@config['credentials']['password'] << "@" <<

                '\2')

  while (true) do

    url = api_url.dup

    path = ""

    path << "/statuses/friends_timeline.json"

    path << "?since_id=#{id}&include_rts=true"

    if page > 1

      path << "&page=#{page}"

    end

    url << path

    if origin == 'twitter'

      puts "OAuth: getting #{path}"

      get = @@sites[origin]['access_token'].get(path)

    else

      puts "get #{url.sub(/\w+:\w+@/, "****:****@")}..."

      get = RestClient.get url, { :accept => "text/json" }

    end

    json = get.body

    json.gsub!(/\003/, '')

    parsed_data = JSON.parse json

    if parsed_data.length == 0

      break

    end

    received_data.concat parsed_data

    # less than 20 updates means we don't need to fetch page 2

    #    if parsed_data.length  20

    #      break

    #    end

    page += 1

  end

  return received_data

end

def load_metadata

  return YAML.load_file("metadata.yml")

end

def save_metadata(metadata)

  File.open("metadata.yml", "w") do |file|

    file.puts metadata.to_yaml

  end

end

def save_timeline(data)

  File.open("timeline.yml", "w") do |file|

    file.puts data.to_yaml

  end

end

def update_metadata(metadata, data)

  metadata.keys.each do |site|

    first = data.find {|obj| obj['origin'] == site }

    if first.nil?

      metadata[site]['new_since'] = metadata[site]['last_id']

      next

    end

    metadata[site]['last_update'] = convert_time_to_est(first['created_at'])

    metadata[site]['last_request'] = convert_time_to_est(Time.now.to_s)

    metadata[site]['new_since'] = metadata[site]['last_id']

    metadata[site]['last_id'] = first['id']

  end

  return metadata

end

get '/' do

  @metadata = load_metadata()

  @data = @@data

  erb :"timeline.html"

#  "whatever"

end

post '/update' do

  metadata = load_metadata()

  new_entries = []

  req_status = {}

  metadata.keys.each do |site|

    req_status[site] = {}

    begin

      new_data = get_since(site, metadata[site]['last_id'])

      # XXX Does this catch 502's?

      # XXX on 28 April 2009 we got a Errno::ECONNRESET - Connection reset

      # by peer - SSL_connect

      # XXX: OpenSSL::SSL::SSLError - SSL_connect SYSCALL returned=5

      # errno=0 state=SSLv2/v3 read server hello A:

      # sometimes, twitter generates

      # x = JSON.parse('{"request":NULL,"error":"Twitter is down for maintenance. It will return in about an hour."}')

      # which throws JSON::ParserError because NULL is different than null.

      # RestClient::RequestTimeout: RestClient::RequestTimeout

    rescue RestClient::Exception => e

      req_status[site]['msg'] = "#{site}: error: #{e.http_code}"

      req_status[site]['error'] = true

    rescue SocketError => e

      req_status[site]['msg'] = "#{site}: error: #{e}"

      req_status[site]['error'] = true

    rescue Errno::ETIMEDOUT => e

      req_status[site]['msg'] = "#{site}: timed out: #{e}"

      req_status[site]['error'] = true

    rescue Errno::ECONNREFUSED => e

      req_status[site]['msg'] = "#{site}: connection refused"

      req_status[site]['error'] = true

    rescue Net::HTTPFatalError => e

      req_status[site]['msg'] = "#{site}: fatal error (#{e})"

      req_status[site]['error'] = true

    end

    # XXX What happens if page 1 comes back, but page 2 errors out?

    if not new_data.nil? and new_data.length > 0

      new_data.each do |entry|

        entry['origin'] = site

      end

      req_status[site]['msg'] = "#{site}: success"

      req_status[site]['error'] = false

      new_entries.concat new_data

    elsif not new_data.nil? and new_data.length == 0

      req_status[site]['msg'] = "#{site}: nothing new"

      req_status[site]['error'] = false

    end

  end

#  result = RubyProf.stop

#  printer = RubyProf::FlatPrinter.new(result)

#  printer.print(STDOUT, 0)

  if new_entries.length == 0

    # for whatever reason, there was no new data

    @req_status = req_status

    p @req_status

    return erb :"update.html"

  end

  new_data = new_entries.sort_by { |entry|

    DateTime.parse(entry['created_at'])

  }.reverse

  metadata = update_metadata(metadata, new_data)

  new_data.concat(@@data)

  @@data = new_data

  save_timeline(@@data)

  save_metadata(metadata)

  @req_status = req_status

  return erb :"update.html"

end

get '/trim' do

  new_data = @@data.slice(0..74)

  @@data = new_data

  save_timeline(@@data)

  redirect '/'

end

get '/status/:origin/:id' do

  puts "something got /status/..."

  # XXX This does no exception handling at all!

  @status = request_status params[:origin], params[:id]

  if @status.has_key? 'error'

    return "Error!"

  else

    erb :"status.html"

  end

end

post '/post' do

  text = params[:post_textarea]

  in_reply_to = params[:in_reply_to]

  destination = params[:destination]

  url = ""

  if text.length > 140

    halt 403, "text too long: #{text.length} characters."

  end

  if in_reply_to.empty?

    send_in_reply_to = false

  else

    send_in_reply_to = true

  end

  url = @@sites[destination]['api'].dup

  url.gsub!(/(https?:\/\/)(.+)/, '\1' <<

            @@config['credentials']['username'] << ":" <<

            @@config['credentials']['password'] << "@" <<

            '\2')

  url << "/statuses/update.json"

  payload = {}

  payload[:status] = text

  if send_in_reply_to

    payload[:in_reply_to_status_id] = in_reply_to

  end

  begin

    res = RestClient.post url, payload

    result = res.body

  rescue Exception => e

    halt 403, "Exception from RestClient: #{e}"

  end

  begin

    JSON.parse(result)

  rescue Exception => e

    halt 403, "JSON failed to parse, may have submitted okay (#{e})"

  end

  return "success"

end