18F/identity-idp

# frozen_string_literal: true

require 'identity/hostdata'

module Reports
  class BaseReport < ApplicationJob
    queue_as :long_running

    def self.transaction_with_timeout(rails_env = Rails.env)
      # rspec-rails's use_transactional_tests does not seem to act as expected when switching
      # connections mid-test, so we just skip for now :[
      return yield if rails_env.test?

      ActiveRecord::Base.connected_to(role: :reading, shard: :read_replica) do
        ActiveRecord::Base.transaction do
          quoted_timeout = ActiveRecord::Base.connection.quote(IdentityConfig.store.report_timeout)
          ActiveRecord::Base.connection.execute("SET LOCAL statement_timeout = #{quoted_timeout}")
          yield
        end
      end
    end

    private

    def public_bucket_name
      if (prefix = IdentityConfig.store.s3_report_public_bucket_prefix)
        Identity::Hostdata.bucket_name("#{prefix}-#{Identity::Hostdata.env}")
      end
    end

    def first_of_this_month
      Time.zone.now.beginning_of_month
    end

    def end_of_today
      Time.zone.now.end_of_day
    end

    def transaction_with_timeout(...)
      self.class.transaction_with_timeout(...)
    end

    def save_report(report_name, body, extension:)
      if !IdentityConfig.store.s3_reports_enabled
        logger.info('Not uploading report to S3, s3_reports_enabled is false')
        return body
      end
      upload_file_to_s3_timestamped_and_latest(report_name, body, extension)
    end

    def upload_file_to_s3_timestamped_and_latest(report_name, body, extension)
      latest_path, path = generate_s3_paths(report_name, extension)
      content_type = Mime::Type.lookup_by_extension(extension).to_s
      url = upload_file_to_s3_bucket(path: path, body: body, content_type: content_type)
      upload_file_to_s3_bucket(path: latest_path, body: body, content_type: content_type)
      url
    end

    def generate_s3_paths(name, extension, subname: nil, now: Time.zone.now)
      host_data_env = Identity::Hostdata.env
      name_subdir_ext = "#{name}#{subname ? '/' : ''}#{subname}.#{extension}"
      latest = "#{host_data_env}/#{name}/latest.#{name_subdir_ext}"
      [latest, "#{host_data_env}/#{name}/#{now.year}/#{now.strftime('%F')}.#{name_subdir_ext}"]
    end

    def logger
      Rails.logger
    end

    def class_name
      self.class.name
    end

    def bucket_name
      Identity::Hostdata.bucket_name(IdentityConfig.store.s3_report_bucket_prefix)
    end

    def s3_client
      @s3_client ||= JobHelpers::S3Helper.new.s3_client
    end

    def upload_file_to_s3_bucket(path:, body:, content_type:, bucket: bucket_name)
      url = "s3://#{bucket}/#{path}"
      logger.info("#{class_name}: uploading to #{url}")
      obj = Aws::S3::Resource.new(client: s3_client).bucket(bucket).object(path)
      obj.put(body: body, acl: 'private', content_type: content_type)
      logger.debug("#{class_name}: upload completed to #{url}")
      url
    end
  end
end