# frozen_string_literal: true

require 'cgi'
require 'facets/string/word_wrap'
require 'twitter'

module Augury
  class Fortune
    def initialize(username, path, config)
      @username = username
      @path = path
      @config = config
      @tweets = []
    end

    def collect_with_max_id(collection = [], max_id = nil, &block)
      response = yield(max_id)
      collection += response
      if response.empty?
        collection
      elsif !@config[:count].zero? && collection.length >= @config[:count]
        # Get everything or trim the results to the count
        @config[:count].zero? ? collection : collection[0..@config[:count] - 1]
      else
        collect_with_max_id(collection, response.last.id - 1, &block)
      end
    end

    def retrieve_tweets
      @tweets = collect_with_max_id do |max_id|
        options = {
          count: 200,
          include_rts: @config[:retweets],
          exclude_replies: !@config[:replies],
        }
        options[:max_id] = max_id unless max_id.nil?
        @twitter.user_timeline(@username, options)
      end
    rescue Twitter::Error::TooManyRequests => e
      reset_length = e.rate_limit.reset_in + 1
      puts "Twitter rate limit exceeded. Waiting #{reset_length} minute(s)"
      sleep reset_length
    end

    def format_fortune
      filtered = @tweets.flat_map(&:full_text).reject do |tweet|
        tweet.match(/https?:/) unless @config[:links]
      end
      formatted = filtered.flat_map { |tweet| CGI.unescapeHTML(tweet).word_wrap(@config[:width]) }
      author = @config[:attribution] ? "\n-- #{@twitter.user(@username).name}\n" : ''
      formatted.join("#{author}%\n")
    end

    def write_fortune
      # Write out the file
      begin
        mode = @config[:append] ? 'a' : 'w'
        file = File.open(@path, mode)
        file.write("%\n") if @config[:append]
        file.write(format_fortune)
      ensure
        file&.close
      end
      # Create the dat file too
      `strfile '#{@path}' '#{@path}.dat'`
    end

    def twitter_setup
      raise Augury::TwitterConfigError unless @config[:twitter]

      @twitter = Twitter::REST::Client.new do |cfg|
        cfg.consumer_key = @config[:twitter]['consumer_key']
        cfg.consumer_secret = @config[:twitter]['consumer_secret']
        cfg.access_token = @config[:twitter]['access_token']
        cfg.access_token_secret = @config[:twitter]['access_token_secret']
      end
    end
  end
end