eduardosasso/bullish

# frozen_string_literal: true

require './bullish.rb'
require './services/news/crawler'
require './services/twitter'

task default: %w[test]

require 'rake/testtask'

Rake::TestTask.new do |t|
  t.pattern = 'test/**/*_test.rb'
  t.warning = false
end

task :send_free_edition do
  day = ARGV[1]

  send_email(Bullish.free_edition, day)

  exit
end

task :send_morning_edition do
  day = ARGV[1]

  send_email(Bullish.morning_edition, day)

  exit
end

task :send_afternoon_edition do
  day = ARGV[1]

  send_email(Bullish.afternoon_edition, day)

  exit
end

task :preview_free_email do
  day = ARGV[1]

  bullish = Bullish.free_edition

  preview_email(bullish, day)

  exit
end

task :preview_free_email_template do
  day = ARGV[1]

  bullish = Bullish.free_edition

  preview_email_template(bullish, day)

  exit
end

task :preview_morning_email do
  day = ARGV[1]

  bullish = Bullish.morning_edition

  preview_email(bullish, day)

  exit
end

task :preview_morning_email_template do
  day = ARGV[1]

  bullish = Bullish.morning_edition

  preview_email_template(bullish, day)

  exit
end

task :preview_afternoon_email do
  day = ARGV[1]

  bullish = Bullish.afternoon_edition

  preview_email(bullish, day)

  exit
end

task :preview_afternoon_email_template do
  day = ARGV[1]

  bullish = Bullish.afternoon_edition

  preview_email_template(bullish, day)

  exit
end

task :preview_all do
  Editions::Edition::DAY_ELEMENTS.each do |day, _value|
    puts '- ' + day.to_s + ' free'
    free = Bullish.free_edition
    free.edition.day_of_the_week = day
    free.edition.save(day.to_s + '-free ' + free.edition.subject)

    puts '- ' + day.to_s + ' morning'
    morning = Bullish.morning_edition
    morning.edition.day_of_the_week = day
    morning.edition.save(day.to_s + '-morning ' + morning.edition.subject)

    puts '- ' + day.to_s + ' afternoon'
    afternoon = Bullish.afternoon_edition
    afternoon.edition.day_of_the_week = day
    afternoon.edition.save(day.to_s + '-afternoon ' + afternoon.edition.subject)
  end
end

task :build_archive do
  archive = Services::Archive.new

  archive.build_index
  archive.build_directory # TODO: can be optimized to run once a month
end

task :update_news do
  # TODO: temp run separately when general news
  Services::News::DB.reset

  stock_news = Services::Trending.new.stocks.map do |stock|
    Services::News::Crawler.stock(stock.symbol)
  end.flatten.compact

  market_news = Services::News::Crawler.reuters

  all_news = stock_news + market_news

  Services::News::DB.save(items: all_news) if all_news.any?
end

task :news do
  pp Services::News::DB.all.collect(&:headline)
end

task :reset_news do
  Services::News::DB.new.reset
end

task :prepare_tweets do
  Services::Twitter.reset_tweets

  twitter = Services::Twitter.new

  Services::Trending.new.stocks.each do |stock|
    twitter.add_tweet(stock)
  end

  twitter.save_tweets
end

def send_email(bullish, day = nil)
  bullish.edition.day_of_the_week = day if day

  bullish.post
end

def preview_email(bullish, day = nil)
  bullish.edition.day_of_the_week = day if day

  bullish.save
end

def preview_email_template(bullish, day = nil)
  bullish.edition.day_of_the_week = day if day

  bullish.save_template
end