alert_in_ua_2_twitter.rb

require './alert_in_ua_2_twitter/version'
require './alert_in_ua_2_twitter/db'
require './alert_in_ua_2_twitter/alert_retriever'
require './alert_in_ua_2_twitter/alert_image_retriever'
require './alert_in_ua_2_twitter/translation'
require './alert_in_ua_2_twitter/tweeter'

API_TOKEN = ENV['ALERTS_IN_UA_TOKEN']
TWITTER_CONSUMER_KEY = ENV['TWITTER_CONSUMER_KEY']
TWITTER_CONSUMER_SECRET = ENV['TWITTER_CONSUMER_SECRET']
TWITTER_ACCESS_TOKEN = ENV['TWITTER_ACCESS_TOKEN']
TWITTER_ACCESS_SECRET = ENV['TWITTER_ACCESS_SECRET']
SNITCH_URL = ENV['SNITCH_URL']

class AlertInUa2Twitter
  def initialize(lang = :ja)
    MiniI18n.locale = lang
  end

  def alert
    started_alerts, active_alerts, terminated_alerts = AlertInUa2Twitter::AlertRetriever.new.get
    image_path = AlertInUa2Twitter::AlertImageRetriever.new.get if started_alerts.any? || terminated_alerts.any?

    format_and_notify(started_alerts, :started, image_path) if started_alerts.any?
    format_and_notify(terminated_alerts, :finished) if terminated_alerts.any?
  end

private

  def format_and_notify(alerts, kind, image = nil)
    m = []
    now = Time.now.getlocal('+09:00')

    alerts.each do |a|
      m << (MiniI18n.t(a.location_title.sub('.', '')) || a.location_title)
    end

    message = [
      MiniI18n.t(kind),
      m.sort.join(MiniI18n.t(:comma)),
      MiniI18n.t(:period),
      MiniI18n.l(now, format: :short)
    ].join

    tweet_message(message, image)
  end

  def tweeter
    @tweeter ||= AlertInUa2Twitter::Tweeter.new(
      TWITTER_CONSUMER_KEY,
      TWITTER_CONSUMER_SECRET,
      TWITTER_ACCESS_TOKEN,
      TWITTER_ACCESS_SECRET
    )
  end

  def tweet_message(message, image_path = nil)
    puts "Tweeting: #{message}"
    tweeter.create_tweet(message, image_path)
  end
end