# frozen_string_literal: true

require './lib/qx.rb'
require 'pg'
require 'minitest/autorun'
require 'pry'

ActiveRecord::Base.establish_connection('postgres://admin:password@localhost/qx_test')
tm = PG::BasicTypeMapForResults.new(ActiveRecord::Base.connection.raw_connection)
Qx.config(type_map: tm)
# Execute test schema
Qx.execute_file('./test/test_schema.sql')

class QxTest < Minitest::Test
  def setup; end

  # Let's just test that the schema was executed
  def test_execute_file
    email = 'uzzzr@example.com'
    result = Qx.execute_file('./test/test_insert_user.sql', email: email, id: 1).last
    Qx.delete_from(:users).where(id: 1).ex
    assert_equal email, result['email']
  end

  def test_select_from
    parsed = Qx.select(:id, 'name').from(:table_name).parse
    assert_equal parsed, %(SELECT id, name FROM table_name)
  end

  def test_select_distinct_on
    parsed = Qx.select(:id, 'name').distinct_on(:distinct_col1, :distinct_col2).from(:table_name).parse
    assert_equal parsed, %(SELECT DISTINCT ON (distinct_col1, distinct_col2) id, name FROM table_name)
  end

  def test_select_distinct
    parsed = Qx.select(:id, 'name').distinct.from(:table_name).parse
    assert_equal parsed, %(SELECT DISTINCT id, name FROM table_name)
  end

  def test_select_as
    parsed = Qx.select(:id, 'name').from(:table_name).as(:alias).parse
    assert_equal parsed, %((SELECT id, name FROM table_name) AS "alias")
  end

  def test_select_where
    parsed = Qx.select(:id, 'name').from(:table_name).where('x = $y OR a = $b', y: 1, b: 2).parse
    assert_equal parsed, %(SELECT id, name FROM table_name WHERE (x = 1 OR a = 2))
  end

  def test_select_where_hash_array
    parsed = Qx.select(:id, 'name').from(:table_name).where([x: 1], ['y = $n', { n: 2 }]).parse
    assert_equal parsed, %(SELECT id, name FROM table_name WHERE ("x" IN (1)) AND (y = 2))
  end

  def test_select_and_where
    parsed = Qx.select(:id, 'name').from(:table_name).where('x = $y', y: 1).and_where('a = $b', b: 2).parse
    assert_equal parsed, %(SELECT id, name FROM table_name WHERE (x = 1) AND (a = 2))
  end

  def test_select_and_where_hash
    parsed = Qx.select(:id, 'name').from(:table_name).where('x = $y', y: 1).and_where(a: 2).parse
    assert_equal parsed, %(SELECT id, name FROM table_name WHERE (x = 1) AND ("a" IN (2)))
  end

  def test_select_and_group_by
    parsed = Qx.select(:id, 'name').from(:table_name).group_by('col1', 'col2').parse
    assert_equal parsed, %(SELECT id, name FROM table_name GROUP BY col1, col2)
  end

  def test_select_and_order_by
    parsed = Qx.select(:id, 'name').from(:table_name).order_by('col1', ['col2', 'DESC NULLS LAST']).parse
    assert_equal parsed, %(SELECT id, name FROM table_name ORDER BY col1 , col2 DESC NULLS LAST)
  end

  def test_select_having
    parsed = Qx.select(:id, 'name').from(:table_name).having('COUNT(col1) > $n', n: 1).parse
    assert_equal parsed, %(SELECT id, name FROM table_name HAVING (COUNT(col1) > 1))
  end

  def test_select_and_having
    parsed = Qx.select(:id, 'name').from(:table_name).having('COUNT(col1) > $n', n: 1).and_having('SUM(col2) > $m', m: 2).parse
    assert_equal parsed, %(SELECT id, name FROM table_name HAVING (COUNT(col1) > 1) AND (SUM(col2) > 2))
  end

  def test_select_limit
    parsed = Qx.select(:id, 'name').from(:table_name).limit(10).parse
    assert_equal parsed, %(SELECT id, name FROM table_name LIMIT 10)
  end

  def test_select_offset
    parsed = Qx.select(:id, 'name').from(:table_name).offset(10).parse
    assert_equal parsed, %(SELECT id, name FROM table_name OFFSET 10)
  end

  def test_select_join
    parsed = Qx.select(:id, 'name').from(:table_name).join(['assoc1', 'assoc1.table_name_id=table_name.id']).parse
    assert_equal parsed, %(SELECT id, name FROM table_name JOIN assoc1 ON assoc1.table_name_id=table_name.id)
  end

  def test_select_add_join
    parsed = Qx.select(:id, 'name').from(:table_name).join('assoc1', 'assoc1.table_name_id=table_name.id')
               .add_join(['assoc2', 'assoc2.table_name_id=table_name.id']).parse
    assert_equal parsed, %(SELECT id, name FROM table_name JOIN assoc1 ON assoc1.table_name_id=table_name.id JOIN assoc2 ON assoc2.table_name_id=table_name.id)
  end

  def test_select_left_join
    parsed = Qx.select(:id, 'name').from(:table_name).left_join(['assoc1', 'assoc1.table_name_id=table_name.id']).parse
    assert_equal parsed, %(SELECT id, name FROM table_name LEFT JOIN assoc1 ON assoc1.table_name_id=table_name.id)
  end

  def test_select_add_left_join
    parsed = Qx.select(:id, 'name').from(:table_name).left_join('assoc1', 'assoc1.table_name_id=table_name.id')
               .add_left_join(['assoc2', 'assoc2.table_name_id=table_name.id']).parse
    assert_equal parsed, %(SELECT id, name FROM table_name LEFT JOIN assoc1 ON assoc1.table_name_id=table_name.id LEFT JOIN assoc2 ON assoc2.table_name_id=table_name.id)
  end

  def test_select_where_subquery
    parsed = Qx.select(:id, 'name').from(:table_name).where('id IN ($ids)', ids: Qx.select('id').from('assoc')).parse
    assert_equal parsed, %(SELECT id, name FROM table_name WHERE (id IN (SELECT id FROM assoc)))
  end

  def test_select_join_subquery
    parsed = Qx.select(:id).from(:table).join([Qx.select(:id).from(:assoc).as(:assoc), 'assoc.table_id=table.id']).parse
    assert_equal parsed, %(SELECT id FROM table JOIN (SELECT id FROM assoc) AS "assoc" ON assoc.table_id=table.id)
  end

  def test_select_from_subquery
    parsed = Qx.select(:id).from(Qx.select(:id).from(:table).as(:table)).parse
    assert_equal parsed, %(SELECT id FROM (SELECT id FROM table) AS "table")
  end

  def test_select_integration
    parsed = Qx.select(:id)
               .from(:table)
               .join([Qx.select(:id).from(:assoc).as(:assoc), 'assoc.table_id=table.id'])
               .left_join(['lefty', 'lefty.table_id=table.id'])
               .where('x = $n', n: 1)
               .and_where('y = $n', n: 1)
               .group_by(:x)
               .order_by(:y)
               .having('COUNT(x) > $n', n: 1)
               .and_having('COUNT(y) > $n', n: 1)
               .limit(10)
               .offset(10)
               .parse
    assert_equal parsed, %(SELECT id FROM table JOIN (SELECT id FROM assoc) AS "assoc" ON assoc.table_id=table.id LEFT JOIN lefty ON lefty.table_id=table.id WHERE (x = 1) AND (y = 1) GROUP BY x HAVING (COUNT(x) > 1) AND (COUNT(y) > 1) ORDER BY y  LIMIT 10 OFFSET 10)
  end

  def test_insert_into_values_hash
    parsed = Qx.insert_into(:table_name).values(x: 1).parse
    assert_equal parsed, %(INSERT INTO "table_name" ("x") VALUES (1))
  end

  def test_insert_into_values_hash_array
    parsed = Qx.insert_into(:table_name).values([{ x: 1 }, { x: 2 }]).parse
    assert_equal parsed, %(INSERT INTO "table_name" ("x") VALUES (1), (2))
  end

  def test_insert_into_values_csv_style
    parsed = Qx.insert_into(:table_name).values([['x'], [1], [2]]).parse
    assert_equal parsed, %(INSERT INTO "table_name" ("x") VALUES (1), (2))
  end

  def test_insert_into_values_common_values
    parsed = Qx.insert_into(:table_name).values([{ x: 'bye' }, { x: 'hi' }]).common_values(z: 1).parse
    assert_equal parsed, %(INSERT INTO "table_name" ("x", "z") VALUES ($Q$bye$Q$, 1), ($Q$hi$Q$, 1))
  end

  def test_insert_into_values_timestamps
    parsed = Qx.insert_into(:table_name).values(x: 1).ts.parse
    assert_equal parsed, %(INSERT INTO "table_name" ("x", created_at, updated_at) VALUES (1, '#{Time.now.utc}', '#{Time.now.utc}'))
  end

  def test_insert_into_values_returning
    parsed = Qx.insert_into(:table_name).values(x: 1).returning('*').parse
    assert_equal parsed, %(INSERT INTO "table_name" ("x") VALUES (1) RETURNING *)
  end

  def test_insert_into_select
    parsed = Qx.insert_into(:table_name, ['hi']).select('hi').from(:table2).where('x=y').parse
    assert_equal parsed, %(INSERT INTO "table_name" ("hi") SELECT hi FROM table2 WHERE (x=y))
  end

  def test_update_set
    parsed = Qx.update(:table_name).set(x: 1).where('y = 2').parse
    assert_equal parsed, %(UPDATE "table_name" SET "x" = 1 WHERE (y = 2))
  end

  def test_update_timestamps
    now = Time.now.utc
    parsed = Qx.update(:table_name).set(x: 1).where('y = 2').timestamps.parse
    assert_equal parsed, %(UPDATE "table_name" SET "x" = 1, updated_at = '#{now}' WHERE (y = 2))
  end

  def test_update_on_conflict
    Qx.update(:table_name).set(x: 1).where('y = 2').on_conflict(:nothing).parse
    assert_equal parsed, %(UPDATE "table_name" SET "x" = 1 WHERE (y = 2) ON CONFLICT DO NOTHING)
  end

  def test_insert_timestamps
    now = Time.now.utc
    parsed = Qx.insert_into(:table_name).values(x: 1).ts.parse
    assert_equal parsed, %(INSERT INTO "table_name" ("x", created_at, updated_at) VALUES (1, '#{now}', '#{now}'))
  end

  def test_delete_from
    parsed = Qx.delete_from(:table_name).where(x: 1).parse
    assert_equal parsed, %(DELETE FROM "table_name" WHERE ("x" IN (1)))
  end

  def test_pagination
    parsed = Qx.select(:x).from(:y).paginate(4, 30).parse
    assert_equal parsed, %(SELECT x FROM y LIMIT 30 OFFSET 90)
  end

  def test_execute_string
    result = Qx.execute('SELECT * FROM (VALUES ($x)) AS t', x: 'x')
    assert_equal result, [{ 'column1' => 'x' }]
  end

  def test_execute_format_csv
    result = Qx.execute('SELECT * FROM (VALUES ($x)) AS t', { x: 'x' }, format: 'csv')
    assert_equal result, [['column1'], ['x']]
  end

  def test_execute_on_instances
    result = Qx.insert_into(:users).values(id: 1, email: 'uzr@example.com').execute
    result = Qx.execute(Qx.select('*').from(:users).limit(1))
    assert_equal result, [{ 'id' => 1, 'email' => 'uzr@example.com' }]
    Qx.delete_from(:users).where(id: 1).execute
  end

  def test_explain
    parsed = Qx.select('*').from('table_name').explain.parse
    assert_equal parsed, %(EXPLAIN SELECT * FROM table_name)
  end

  # Manually test this one for now
  def test_pp_select
    pp = Qx.select('id, name').from('table_name').where(status: 'active').and_where(id: Qx.select('id').from('roles').where(name: 'admin')).pp
    pp2 = Qx.insert_into(:table_name).values([x: 1, y: 2]).pp
    pp3 = Qx.update(:table_name).set(x: 1, y: 2).where(z: 1, a: 22).pp
    pp_delete = Qx.delete_from(:table_name).where(id: 123).pp
    puts ''
    puts '--- pretty print'
    puts pp
    puts pp2
    puts pp3
    puts pp_delete
    puts '---'
  end

  def test_to_json
    parsed = Qx.select(:id).from(:users).to_json(:t).parse
    assert_equal parsed, %(SELECT array_to_json(array_agg(row_to_json(t))) FROM (SELECT id FROM users) AS "t")
  end

  def test_to_json_nested
    definitions = Qx.select(:part_of_speech, :body)
                    .from(:definitions)
                    .where('word_id=words.id')
                    .order_by('position ASC')
                    .to_json(:ds)
                    .as('definitions')
    parsed = Qx.select(:text, :pronunciation, definitions)
               .from(:words)
               .where("text='autumn'")
               .to_json(:ws)
               .parse
    assert_equal parsed, "SELECT array_to_json(array_agg(row_to_json(ws))) FROM (SELECT text, pronunciation, (SELECT array_to_json(array_agg(row_to_json(ds))) FROM (SELECT part_of_speech, body FROM definitions WHERE (word_id=words.id) ORDER BY position ASC ) AS \"ds\") AS \"definitions\" FROM words WHERE (text='autumn')) AS \"ws\""
  end

  def test_copy_csv_execution
    data = { 'id' => '1', 'email' => 'uzr@example.com' }
    filename = '/tmp/qx-test.csv'
    Qx.insert_into(:users).values(data).ex
    copy = Qx.select('*').from('users').execute(copy_csv: filename)
    contents = File.open(filename, 'r').read
    csv_data = contents.split("\n").map { |l| l.split(',') }
    headers = csv_data.first
    row = csv_data.last
    assert_equal data.keys, headers
    assert_equal data.values, row
  end

  def test_remove_clause
    expr = Qx.select('*').from('table').limit(1)
    expr = expr.remove_clause('limit')
    assert_equal 'SELECT * FROM table', expr.parse
  end
end