require 'business_time'
require 'haml'
require 'mongoid'

require "iro/engine"

class Iro::Iro

  def self.get_coins
    out = HTTParty.get( "https://pro-api.coinmarketcap.com/v2/cryptocurrency/quotes/latest?slug=bitcoin,ethereum", {
      headers: { 'X-CMC_PRO_API_KEY' => COINMARKETCAP[:key] },
    })
    out = out.parsed_response.deep_symbolize_keys
    out[:data].each do |k, item|
      opi = Iro::Datapoint.new({
        date:     Time.now.to_date,
        kind:     Iro::Datapoint::KIND_CRYPTO,
        symbol:   item[:symbol],
        quote_at: item[:quote][:USD][:last_updated],
        value:    item[:quote][:USD][:price],
        volume:   item[:quote][:USD][:volume_24h],
      })
      opi.save!
    end
  end

  def self.get_treasuries
    response = HTTParty.get( "https://home.treasury.gov/resource-center/data-chart-center/interest-rates/daily-treasury-rates.csv/all/#{Time.now.strftime('%Y%m')}?type=daily_treasury_yield_curve&field_tdr_date_value_month=#{Time.now.strftime('%Y%m')}&page&_format=csv")
    outs = CSV.parse( response.body, { headers: true })
    out = outs[0]
    date = Date.strptime(out['Date'], '%m/%d/%Y')
    {
      '1 Mo'  => Iro::Datapoint::SYMBOL_T1MO,
      '2 Mo'  => Iro::Datapoint::SYMBOL_T2MO,
      '3 Mo'  => Iro::Datapoint::SYMBOL_T3MO,
      '4 Mo'  => Iro::Datapoint::SYMBOL_T4MO,
      '6 Mo'  => Iro::Datapoint::SYMBOL_T6MO,
      '1 Yr'  => Iro::Datapoint::SYMBOL_T1YR,
      '2 Yr'  => Iro::Datapoint::SYMBOL_T2YR,
      '3 Yr'  => Iro::Datapoint::SYMBOL_T3YR,
      '5 Yr'  => Iro::Datapoint::SYMBOL_T5YR,
      '7 Yr'  => Iro::Datapoint::SYMBOL_T7YR,
      '10 Yr' => Iro::Datapoint::SYMBOL_T10YR,
      '20 Yr' => Iro::Datapoint::SYMBOL_T20YR,
      '30 Yr' => Iro::Datapoint::SYMBOL_T30YR,
    }.each do |k, v|
      opi = Iro::Datapoint.new({
        date:     date,
        quote_at: date,
        kind:     Iro::Datapoint::KIND_TREASURY,
        symbol:   v,
        value:    out[k],
      })
      opi.save!
      print '^'
    end
  end

  def self.import_1990_2023_treasuries
    outs = CSV.parse( File.read( Rails.root.join('data', 'treasuries', '1990..23 daily-treasury-rates.csv') ), { headers: true })
    outs.each do |out|
      date = Date.strptime(out['Date'], '%m/%d/%y')
      {
        '1 Mo'  => Iro::Datapoint::SYMBOL_T1MO,
        '2 Mo'  => Iro::Datapoint::SYMBOL_T2MO,
        '3 Mo'  => Iro::Datapoint::SYMBOL_T3MO,
        '4 Mo'  => Iro::Datapoint::SYMBOL_T4MO,
        '6 Mo'  => Iro::Datapoint::SYMBOL_T6MO,
        '1 Yr'  => Iro::Datapoint::SYMBOL_T1YR,
        '2 Yr'  => Iro::Datapoint::SYMBOL_T2YR,
        '3 Yr'  => Iro::Datapoint::SYMBOL_T3YR,
        '5 Yr'  => Iro::Datapoint::SYMBOL_T5YR,
        '7 Yr'  => Iro::Datapoint::SYMBOL_T7YR,
        '10 Yr' => Iro::Datapoint::SYMBOL_T10YR,
        '20 Yr' => Iro::Datapoint::SYMBOL_T20YR,
        '30 Yr' => Iro::Datapoint::SYMBOL_T30YR,
      }.each do |k, v|
        if out[k]
          opi = Iro::Datapoint.new({
            date:     date,
            quote_at: date,
            kind:     Iro::Datapoint::KIND_TREASURY,
            symbol:   v,
            value:    out[k],
          })
          begin
            opi.save!
          rescue Mongoid::Errors::Validations => err
            puts! err, 'err'
          end
          print '^'
        end
      end
    end
  end

  def self.import_2024_treasuries
    outs = CSV.parse( File.read( Rails.root.join('data', 'treasuries', '2024 daily-treasury-rates.csv') ), { headers: true })
    outs.each do |out|
      date = Date.strptime(out['Date'], '%m/%d/%Y')
      {
        '1 Mo'  => Iro::Datapoint::SYMBOL_T1MO,
        '2 Mo'  => Iro::Datapoint::SYMBOL_T2MO,
        '3 Mo'  => Iro::Datapoint::SYMBOL_T3MO,
        '4 Mo'  => Iro::Datapoint::SYMBOL_T4MO,
        '6 Mo'  => Iro::Datapoint::SYMBOL_T6MO,
        '1 Yr'  => Iro::Datapoint::SYMBOL_T1YR,
        '2 Yr'  => Iro::Datapoint::SYMBOL_T2YR,
        '3 Yr'  => Iro::Datapoint::SYMBOL_T3YR,
        '5 Yr'  => Iro::Datapoint::SYMBOL_T5YR,
        '7 Yr'  => Iro::Datapoint::SYMBOL_T7YR,
        '10 Yr' => Iro::Datapoint::SYMBOL_T10YR,
        '20 Yr' => Iro::Datapoint::SYMBOL_T20YR,
        '30 Yr' => Iro::Datapoint::SYMBOL_T30YR,
      }.each do |k, v|
        if out[k]
          opi = Iro::Datapoint.new({
            date:     date,
            quote_at: date,
            kind:     Iro::Datapoint::KIND_TREASURY,
            symbol:   v,
            value:    out[k],
          })
          begin
            opi.save!
          rescue Mongoid::Errors::Validations => err
            puts! err, 'err'
          end
          print '^'
        end
      end
    end
  end

end