# frozen_string_literal: true

#
# Copyright (C) 2015 - present Instructure, Inc.
#
# This file is part of Canvas.
#
# Canvas is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the Free
# Software Foundation, version 3 of the License.
#
# Canvas is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License along
# with this program. If not, see <http://www.gnu.org/licenses/>.

class StubbedClient
  def self.put_records(records:, stream_name:)
    events = records.map { |e| JSON.parse(e[:data]).dig("attributes", "event_name") }.join(" | ")
    puts "Events #{events} put to stream #{stream_name}: #{records}" # rubocop:disable Rails/Output
    Aws::Kinesis::Types::PutRecordsOutput.new(
      records: records.map { Aws::Kinesis::Types::PutRecordsResultEntry.new }
    )
  end

  def self.stream_name
    "stubbed_kinesis_stream"
  end
end

Rails.configuration.to_prepare do
  LiveEvents.logger = Rails.logger
  LiveEvents.cache = Rails.cache
  LiveEvents.statsd = InstStatsd::Statsd
  LiveEvents.max_queue_size = -> { Setting.get("live_events_max_queue_size", 5000).to_i }
  LiveEvents.retry_throttled_events = -> { Setting.get("live_events_retry_throttled_events", true) == true }
  LiveEvents.settings = -> { YAML.safe_load(DynamicSettings.find(tree: :private, default_ttl: 2.hours)["live_events.yml", failsafe_cache: Rails.root.join("config")] || "{}") }
  LiveEvents.aws_credentials = lambda do |settings|
    if settings["vault_credential_path"]
      Canvas::Vault::AwsCredentialProvider.new(settings["vault_credential_path"])
    else
      nil
    end
  end
  LiveEvents.stream_client = ->(settings) { StubbedClient if settings["stub_kinesis"] }
  # sometimes this async worker thread grabs a connection on a Setting read or similar.
  # We need it to be released or the main thread can have a real problem.
  LiveEvents.on_work_unit_end = -> { ActiveRecord::Base.connection_handler.clear_active_connections!(:all) }
end
