RSpec.describe "End to End" do
  include WaitForPop

  let(:tmp_dir) { Dir.mktmpdir }
  after { FileUtils.remove_entry(tmp_dir) }

  let(:log_path) { "#{tmp_dir}/test_tag.log" }
  let(:log_data) { File.read(log_path) }

  let(:log_hash) do
    hash = {}
    log_data.split("\n").each do |line|
      sha = line.split("|", 2).last
      hash[sha] ||= 0
      hash[sha] += 1
    end
    hash
  end

  let(:log_reverse_hash) do
    Hash[log_hash.map(&:reverse)]
  end

  let(:log_def_path) { "#{tmp_dir}/test_tag_definitions.log" }
  let(:log_def_data) { File.read(log_def_path) }
  let(:log_def_hash) do
    hash = {}
    log_def_data.split("\n").each do |line|
      sha, event = line.split("|", 2)
      hash[sha] = JSON.parse(event)
    end

    hash
  end

  before do
    ActiveRecord::SqlAnalyzer.configure do |c|
      c.logger_root_path tmp_dir
      c.log_sample_proc Proc.new { |_name| true }

      c.add_analyzer(
        name: :test_tag,
        tables: %w(matching_table)
      )
    end

    ActiveRecord::SqlAnalyzer.config[:analyzers].each do |analyzer|
      analyzer[:logger_instance].definition_log_file.sync = true
      analyzer[:logger_instance].log_file.sync = true
    end
  end

  def execute(sql)
    DBConnection.connection.execute(sql)
    wait_for_pop
  end

  it "does not log with a non-matching table" do
    execute "SELECT * FROM nonmatching_table"

    expect(log_data).to eq("")
    expect(log_def_data).to eq("")
  end

  it "logs with a matching + non-matching table in one query" do
    execute "SELECT nmt.id FROM nonmatching_table AS nmt JOIN matching_table AS mt ON mt.id = nmt.id WHERE mt.id = 1234"
    execute "SELECT nmt.id FROM nonmatching_table AS nmt JOIN matching_table AS mt ON mt.id = nmt.id WHERE mt.id = 4321"
    execute "SELECT nmt.id FROM nonmatching_table AS nmt JOIN matching_table AS mt ON mt.id = nmt.id WHERE mt.test_string = 'abc'"

    expect(log_hash.length).to eq(2)

    id_sha = log_reverse_hash[2]
    str_sha = log_reverse_hash[1]

    expect(log_def_hash.length).to eq(2)

    expect(log_def_hash[id_sha]["sql"]).to include("mt.id = [REDACTED]")
    expect(log_def_hash[str_sha]["sql"]).to include("mt.test_string = '[REDACTED]'")
  end

  it "logs with only a matching table in a query" do
    execute "SELECT * FROM matching_table WHERE id = 1234"
    execute "SELECT * FROM matching_table WHERE id = 4321"
    execute "SELECT * FROM matching_table WHERE test_string = 'abc'"

    expect(log_hash.length).to eq(2)

    id_sha = log_reverse_hash[2]
    str_sha = log_reverse_hash[1]

    expect(log_def_hash.length).to eq(2)

    expect(log_def_hash[id_sha]["sql"]).to include("id = [REDACTED]")
    expect(log_def_hash[str_sha]["sql"]).to include("test_string = '[REDACTED]'")
  end

  it "handles invalid UTF-8" do
    execute "SELECT * FROM matching_table WHERE test_string = '\xe5'"
    execute "SELECT * FROM matching_table WHERE test_string = 'foobar'"

    expect(log_reverse_hash.first.first).to eq(2)

    sha = log_reverse_hash[2]
    expect(log_def_hash[sha]["sql"]).to include("test_string = '[REDACTED]'")
  end

  context "when sampling is disabled" do
    before do
      ActiveRecord::SqlAnalyzer.configure do |c|
        c.log_sample_proc Proc.new { |_name| false }
      end
    end

    it "does not log" do
      execute "SELECT * FROM matching_table"

      expect(log_data).to eq("")
      expect(log_def_data).to eq("")
    end
  end
end