module Gnip
  class Activity < Gnip::Message
    attr_reader :id, :object_type, :actor, :verb, :posted_time, :generator, :provider, :link,
    :body, :object, :favorites_count, :twitter_entities, :twitter_filter_level, :twitter_lang,
    :retweet_count, :gnip

    def initialize params = {}
      @id = params['id']
      @object_type = params['objectType']
      @actor = Gnip::Actor.new params['actor']
      @verb = params['verb']
      @posted_time = params['postedTime']
      @generator = params['generator']
      @provider = params['provider']
      @link = params['link']
      @body = params['body']
      @object = retweet? ? Gnip::Activity.new(params['object']) : params['object']
      @favorites_count = params['favoritesCount']
      @twitter_entities = params['twitter_entities']
      @twitter_filter_level = params['twitter_filter_level']
      @twitter_lang = params['twitter_lang']
      @retweet_count = params['retweetCount']
      @gnip = Gnip::GnipData.new(params['gnip']) if params['gnip']
    end

    def original_attributes
      {
        :id => @id,
        :objectType => @object_type,
        :actor => @actor.original_attributes,
        :verb => @verb,
        :postedTime => @posted_time,
        :generator => @generator,
        :provider => @provider,
        :link => @link,
        :body => @body,
        :object => @object.kind_of?(Gnip::Activity) ? @object.original_attributes : @object,
        :favoritesCount => @favorites_count,
        :twitter_entities => @twitter_entities,
        :twitter_filter_level => @twitter_filter_level,
        :twitter_lang => @twitter_lang,
        :retweetCount => @retweet_count,
        :gnip => @gnip ? @gnip.original_attributes : nil
      }
    end

    def posted_time
      DateTime.parse(@posted_time)
    end

    def link
      URI(@link)
    end

    def tweet_id
      @id.split(':').last
    end

    def to_json
      generate_json(original_attributes)
    end

    def author
      actor.display_name
    end

    def retweet? 
      verb == 'share'
    end
  end
end