module Tw
  class Client

    def mentions
      Twitter.mentions.map{|m|
        Tw::Tweet.new(:id => m.id,
                      :user => m.user.screen_name,
                      :text => m.text,
                      :time => m.created_at)
      }
    end

    def search(word)
      Twitter.search(word).results.map{|m|
        Tw::Tweet.new(:id => m.id,
                      :user => m.from_user,
                      :text => m.text,
                      :time => m.created_at)
      }
    end

    def home_timeline
      Twitter.home_timeline.map{|m|
        Tw::Tweet.new(:id => m.id,
                      :user => m.user.screen_name,
                      :text => m.text,
                      :time => m.created_at)
      }
    end

    def user_timeline(user)
      Twitter.user_timeline(user).map{|m|
        Tw::Tweet.new(:id => m.id,
                      :user => m.user.screen_name,
                      :text => m.text,
                      :time => m.created_at)
      }
    end

    def list_timeline(user,list)
      Twitter.list_timeline(user, list).map{|m|
        Tw::Tweet.new(:id => m.id,
                      :user => m.user.screen_name,
                      :text => m.text,
                      :time => m.created_at)
      }
    end

    def direct_messages
      [Twitter.direct_messages.map{|m|
         Tw::Tweet.new(:id => m.id,
                       :user => m.sender.screen_name,
                       :text => m.text,
                       :time => m.created_at)
       }, Twitter.direct_messages_sent.map{|m|
         Tw::Tweet.new(:id => m.id,
                       :user => {
                         :from => m.sender.screen_name,
                         :to => m.recipient.screen_name
                       },
                       :text => m.text,
                       :time => m.created_at)
       }].flatten
    end

    def tweet(message)
      res = Twitter.update message
      puts res.text
      puts "http://twitter.com/#{res.user.screen_name}/status/#{res.id}"
      puts res.created_at
    end

    def direct_message_create(to, message)
      res = Twitter.direct_message_create to, message
      puts res.text
      puts res.created_at
    end

  end
end