view app/models/tweet.rb @ 144:1925b08153dc

Record 404s
author nanaya <me@nanaya.pro>
date Wed, 13 Dec 2017 06:10:13 +0900
parents 8a858aec1905
children c791b6bfeeda
line wrap: on
line source

class Tweet
  TIMELINE_OPTIONS = {
    :count => 100,
    :exclude_replies => false,
    :include_rts => true,
    :tweet_mode => :extended,
  }

  def initialize(twitter_id)
    @clients = {}
    @twitter_id = twitter_id
  end

  def cache_expires_time
    (15 + rand(15)).minutes
  end

  def timeline
    initial_config_id = client_config_id

    @timeline ||=
      Rails.cache.fetch("timeline:v1:#{@twitter_id}", :expires_in => cache_expires_time) do
        begin
          timeline = client.user_timeline(@twitter_id, TIMELINE_OPTIONS)
        rescue Twitter::Error::TooManyRequests
          @client_config_id += 1

          if initial_config_id == client_config_id
            raise
          else
            retry
          end
        rescue Twitter::Error::NotFound
          data = { :result => :not_found }
        end

        data || {
          :result => :ok,
          :timeline => timeline.select do |tweet|
            tweet.retweeted_status.nil? || tweet.user.id != tweet.retweeted_status.user.id
          end.map do |tweet|
            # Fails when there's Twitter::NullObject initiated somewhere in previous select
            # Reference: https://github.com/sferik/twitter/issues/892
            tweet.to_h
          end,
        }
      end.tap do |data|
        raise Twitter::Error::NotFound if data[:result] == :not_found

        data[:timeline_parsed] = data[:timeline].map do |tweet_hash|
          Twitter::Tweet.new(tweet_hash)
        end
      end[:timeline_parsed]
  end

  def user
    @user ||=
      if timeline.any?
        timeline.first.user
      else
        Rails.cache.fetch("user:v1:#{@twitter_id}", :expires_in => cache_expires_time) do
          client.user(@twitter_id)
        end
      end
  end

  def client
    @clients[client_config_id] ||=
      Twitter::REST::Client.new do |config|
        $cfg[:twitter][client_config_id].each do |cfg_key, cfg_value|
          config.public_send(:"#{cfg_key}=", cfg_value)
        end
      end
  end

  def client_config_id
    @client_config_id ||= 0

    @client_config_id %= $cfg[:twitter].size
  end
end