18F/identity-idp

View on GitHub
app/jobs/reports/combined_invoice_supplement_report_v2.rb

Summary

Maintainability
A
0 mins
Test Coverage
# frozen_string_literal: true

require 'csv'

module Reports
  class CombinedInvoiceSupplementReportV2 < BaseReport
    REPORT_NAME = 'combined-invoice-supplement-report-v2'

    def perform(_date)
      csv = build_csv(IaaReportingHelper.iaas, IaaReportingHelper.partner_accounts)
      save_report(REPORT_NAME, csv, extension: 'csv')
    end

    # @param [Array<IaaReportingHelper::IaaConfig>] iaas
    # @param [Array<IaaReportingHelper::PartnerConfig>] partner_accounts
    # @return [String] CSV report
    def build_csv(iaas, partner_accounts)
      by_iaa_results = iaas.flat_map do |iaa|
        Db::MonthlySpAuthCount::UniqueMonthlyAuthCountsByIaa.call(
          key: iaa.key,
          issuers: iaa.issuers,
          start_date: iaa.start_date,
          end_date: iaa.end_date,
        )
      end

      by_issuer_results = iaas.flat_map do |iaa|
        iaa.issuers.flat_map do |issuer|
          Db::MonthlySpAuthCount::TotalMonthlyAuthCountsWithinIaaWindow.call(
            issuer: issuer,
            iaa_start_date: iaa.start_date,
            iaa_end_date: iaa.end_date,
            iaa: iaa.key,
          )
        end
      end

      by_partner_results = partner_accounts.flat_map do |partner_account|
        Db::MonthlySpAuthCount::NewUniqueMonthlyUserCountsByPartner.call(
          partner: partner_account.partner,
          issuers: partner_account.issuers,
          start_date: partner_account.start_date,
          end_date: partner_account.end_date,
        )
      end

      by_issuer_profile_age_results = partner_accounts.flat_map do |partner_account|
        partner_account.issuers.flat_map do |issuer|
          Db::MonthlySpAuthCount::NewUniqueMonthlyUserCountsByPartner.call(
            partner: partner_account.partner,
            issuers: [issuer],
            start_date: partner_account.start_date,
            end_date: partner_account.end_date,
          )
        end
      end

      combine_by_iaa_month(
        by_iaa_results: by_iaa_results,
        by_issuer_results: by_issuer_results,
        by_partner_results: by_partner_results,
        by_issuer_profile_age_results: by_issuer_profile_age_results,
      )
    end

    def combine_by_iaa_month(
      by_iaa_results:,
      by_issuer_results:,
      by_partner_results:,
      by_issuer_profile_age_results:
    )
      by_iaa_and_year_month = by_iaa_results.group_by do |result|
        [result[:key], result[:year_month]]
      end

      by_issuer_iaa_issuer_year_months = by_issuer_results.
        group_by { |r| r[:iaa] }.
        transform_values do |iaa|
          iaa.group_by { |r| r[:issuer] }.
            transform_values { |issuer| issuer.group_by { |r| r[:year_month] } }
        end

      # rubocop:disable Metrics/BlockLength
      CSV.generate do |csv|
        csv << [
          'iaa_order_number',
          'partner',
          'iaa_start_date',
          'iaa_end_date',

          'issuer',
          'friendly_name',

          'year_month',
          'year_month_readable',

          'iaa_ial1_unique_users',
          'iaa_ial2_unique_users',
          'iaa_unique_users',
          'partner_ial2_unique_user_events_year1',
          'partner_ial2_unique_user_events_year2',
          'partner_ial2_unique_user_events_year3',
          'partner_ial2_unique_user_events_year4',
          'partner_ial2_unique_user_events_year5',
          'partner_ial2_unique_user_events_year_greater_than_5',
          'partner_ial2_unique_user_events_unknown',
          'partner_ial2_new_unique_user_events_year1',
          'partner_ial2_new_unique_user_events_year2',
          'partner_ial2_new_unique_user_events_year3',
          'partner_ial2_new_unique_user_events_year4',
          'partner_ial2_new_unique_user_events_year5',
          'partner_ial2_new_unique_user_events_year_greater_than_5',
          'partner_ial2_new_unique_user_events_unknown',

          'issuer_ial2_unique_user_events_year1',
          'issuer_ial2_unique_user_events_year2',
          'issuer_ial2_unique_user_events_year3',
          'issuer_ial2_unique_user_events_year4',
          'issuer_ial2_unique_user_events_year5',
          'issuer_ial2_unique_user_events_year_greater_than_5',
          'issuer_ial2_unique_user_events_unknown',
          'issuer_ial2_new_unique_user_events_year1',
          'issuer_ial2_new_unique_user_events_year2',
          'issuer_ial2_new_unique_user_events_year3',
          'issuer_ial2_new_unique_user_events_year4',
          'issuer_ial2_new_unique_user_events_year5',
          'issuer_ial2_new_unique_user_events_year_greater_than_5',
          'issuer_ial2_new_unique_user_events_unknown',

          'issuer_ial1_total_auth_count',
          'issuer_ial2_total_auth_count',
          'issuer_ial1_plus_2_total_auth_count',

          'issuer_ial1_unique_users',
          'issuer_ial2_unique_users',
          'issuer_unique_users',
        ]
        by_issuer_iaa_issuer_year_months.each do |iaa_key, issuer_year_months|
          issuer_year_months.each do |issuer, year_months_data|
            friendly_name = ServiceProvider.find_by(issuer: issuer).friendly_name
            year_months = year_months_data.keys.sort

            year_months.each do |year_month|
              iaa_results = by_iaa_and_year_month[ [iaa_key, year_month] ]
              if !iaa_results
                logger.warn(
                  {
                    level: 'warning',
                    name: 'missing iaa_results',
                    iaa: iaa_key,
                    year_month: year_month,
                  }.to_json,
                )
                next
              end

              issuer_results = year_months_data[year_month]
              year_month_start = Date.strptime(year_month, '%Y%m')
              iaa_start_date = Date.parse(iaa_results.first[:iaa_start_date])
              iaa_end_date = Date.parse(iaa_results.first[:iaa_end_date])

              partner_results = by_partner_results.find do |result|
                result[:year_month] == year_month && result[:issuers]&.include?(issuer)
              end || {}

              issuer_profile_age_results = by_issuer_profile_age_results.find do |result|
                result[:year_month] == year_month && result[:issuers]&.include?(issuer)
              end || {}

              csv << [
                iaa_key,
                partner_results[:partner],
                iaa_start_date,
                iaa_end_date,

                issuer,
                friendly_name,

                year_month,
                year_month_start.strftime('%B %Y'),

                extract(iaa_results, :unique_users, ial: 1),
                extract(iaa_results, :unique_users, ial: 2),
                extract(iaa_results, :unique_users, ial: :all),
                partner_results[:partner_ial2_unique_user_events_year1] || 0,
                partner_results[:partner_ial2_unique_user_events_year2] || 0,
                partner_results[:partner_ial2_unique_user_events_year3] || 0,
                partner_results[:partner_ial2_unique_user_events_year4] || 0,
                partner_results[:partner_ial2_unique_user_events_year5] || 0,
                partner_results[:partner_ial2_unique_user_events_year_greater_than_5] || 0,
                partner_results[:partner_ial2_unique_user_events_unknown] || 0,
                partner_results[:partner_ial2_new_unique_user_events_year1] || 0,
                partner_results[:partner_ial2_new_unique_user_events_year2] || 0,
                partner_results[:partner_ial2_new_unique_user_events_year3] || 0,
                partner_results[:partner_ial2_new_unique_user_events_year4] || 0,
                partner_results[:partner_ial2_new_unique_user_events_year5] || 0,
                partner_results[:partner_ial2_new_unique_user_events_year_greater_than_5] || 0,
                partner_results[:partner_ial2_new_unique_user_events_unknown] || 0,

                issuer_profile_age_results[:partner_ial2_unique_user_events_year1] || 0,
                issuer_profile_age_results[:partner_ial2_unique_user_events_year2] || 0,
                issuer_profile_age_results[:partner_ial2_unique_user_events_year3] || 0,
                issuer_profile_age_results[:partner_ial2_unique_user_events_year4] || 0,
                issuer_profile_age_results[:partner_ial2_unique_user_events_year5] || 0,
                issuer_profile_age_results[:partner_ial2_unique_user_events_year_greater_than_5] || 0, # rubocop:disable Layout/LineLength
                issuer_profile_age_results[:partner_ial2_unique_user_events_unknown] || 0,
                issuer_profile_age_results[:partner_ial2_new_unique_user_events_year1] || 0,
                issuer_profile_age_results[:partner_ial2_new_unique_user_events_year2] || 0,
                issuer_profile_age_results[:partner_ial2_new_unique_user_events_year3] || 0,
                issuer_profile_age_results[:partner_ial2_new_unique_user_events_year4] || 0,
                issuer_profile_age_results[:partner_ial2_new_unique_user_events_year5] || 0,
                issuer_profile_age_results[:partner_ial2_new_unique_user_events_year_greater_than_5] || 0, # rubocop:disable Layout/LineLength
                issuer_profile_age_results[:partner_ial2_new_unique_user_events_unknown] || 0,

                (ial1_total_auth_count = extract(issuer_results, :total_auth_count, ial: 1)),
                (ial2_total_auth_count = extract(issuer_results, :total_auth_count, ial: 2)),
                ial1_total_auth_count + ial2_total_auth_count,

                extract(issuer_results, :unique_users, ial: 1),
                extract(issuer_results, :unique_users, ial: 2),
                extract(issuer_results, :unique_users, ial: :all),
              ]
            end
          end
        end
      end
      # rubocop:enable Metrics/BlockLength
    end

    def extract(arr, key, ial:)
      arr.find { |elem| elem[:ial] == ial && elem[key] }&.dig(key) || 0
    end
  end
end