import { strict as assert } from "node:assert";
import test from "node:test";
import {
  buildDataGridRollbackStatements,
  buildDataGridSaveStatements,
  dataGridSaveExecutionSchema,
  normalizeDataGridSaveError,
  formatGridSqlLiteral,
  validateDataGridSave,
} from "../src/lib/dataGridSql.ts";
import { DBX_NEO4J_ELEMENT_ID_COLUMN } from "../src/lib/tableEditing.ts";

test("builds SQL Server grid save statements with schema and bracket quoting", () => {
  const statements = buildDataGridSaveStatements({
    databaseType: "sqlserver",
    tableMeta: {
      schema: "game",
      tableName: "player states",
      primaryKeys: ["role id"],
    },
    columns: ["role id", "state", "updated at"],
    rows: [[42, "old", "2026-05-03"]],
    dirtyRows: [
      [
        0,
        [
          [1, "ready"],
          [2, "2026-05-04"],
        ],
      ],
    ],
    deletedRows: [0],
    newRows: [[43, "new", "2026-05-05"]],
  });

  assert.deepEqual(statements, [
    "UPDATE [game].[player states] SET [state] = N'ready', [updated at] = N'2026-05-04' WHERE [role id] = 42;",
    "DELETE FROM [game].[player states] WHERE [role id] = 42;",
    "INSERT INTO [game].[player states] ([role id], [state], [updated at]) VALUES (43, N'new', N'2026-05-05');",
  ]);
});

test("builds Access grid save statements with backtick identifiers", () => {
  const statements = buildDataGridSaveStatements({
    databaseType: "access",
    tableMeta: {
      tableName: "Order Details",
      primaryKeys: ["Order ID"],
    },
    columns: ["Order ID", "Product Name", "Active"],
    rows: [[42, "Old", true]],
    dirtyRows: [[0, [[1, "Ready"]]]],
    deletedRows: [0],
    newRows: [[43, "New", false]],
  });

  assert.deepEqual(statements, [
    "UPDATE `Order Details` SET `Product Name` = 'Ready' WHERE `Order ID` = 42;",
    "DELETE FROM `Order Details` WHERE `Order ID` = 42;",
    "INSERT INTO `Order Details` (`Order ID`, `Product Name`, `Active`) VALUES (43, 'New', FALSE);",
  ]);
});

test("builds Access grid save statements with row predicates when primary keys are unavailable", () => {
  const statements = buildDataGridSaveStatements({
    databaseType: "access",
    tableMeta: {
      tableName: "orders",
      primaryKeys: [],
    },
    columns: ["id", "quantity", "status", "shipped_at"],
    rows: [[1, 3, "pending", null]],
    dirtyRows: [[0, [[1, 4]]]],
    deletedRows: [0],
    newRows: [],
  });

  assert.deepEqual(statements, [
    "UPDATE `orders` SET `quantity` = 4 WHERE `id` = 1 AND `quantity` = 3 AND `status` = 'pending' AND `shipped_at` IS NULL;",
    "DELETE FROM `orders` WHERE `id` = 1 AND `quantity` = 3 AND `status` = 'pending' AND `shipped_at` IS NULL;",
  ]);
});

test("builds grid save statements through source columns for aliased query results", () => {
  const statements = buildDataGridSaveStatements({
    databaseType: "postgres",
    tableMeta: {
      schema: "public",
      tableName: "ihli_data",
      primaryKeys: ["iso3", "year"],
    },
    columns: ["country_code", "report_year", "country_name", "ihli_rank"],
    sourceColumns: ["iso3", "year", "country_name", "rank"],
    rows: [["LUX", 2007, "Luxembourg", 1]],
    dirtyRows: [[0, [[2, "Luxembourg City"]]]],
    deletedRows: [0],
    newRows: [["USA", 2008, "United States", 2]],
  });

  assert.deepEqual(statements, [
    `UPDATE "public"."ihli_data" SET "country_name" = 'Luxembourg City' WHERE "iso3" = 'LUX' AND "year" = 2007;`,
    `DELETE FROM "public"."ihli_data" WHERE "iso3" = 'LUX' AND "year" = 2007;`,
    `INSERT INTO "public"."ihli_data" ("iso3", "year", "country_name", "rank") VALUES ('USA', 2008, 'United States', 2);`,
  ]);
});

test("skips expression-only result columns when saving aliased query results", () => {
  const statements = buildDataGridSaveStatements({
    databaseType: "postgres",
    tableMeta: {
      schema: "public",
      tableName: "ihli_data",
      primaryKeys: ["iso3", "year"],
    },
    columns: ["iso3", "year", "country_name", "score"],
    sourceColumns: ["iso3", "year", "country_name", undefined],
    rows: [["LUX", 2007, "Luxembourg", 50242.1]],
    dirtyRows: [
      [
        0,
        [
          [2, "Luxembourg City"],
          [3, 999],
        ],
      ],
    ],
    deletedRows: [],
    newRows: [["USA", 2008, "United States", 43000]],
  });

  assert.deepEqual(statements, [
    `UPDATE "public"."ihli_data" SET "country_name" = 'Luxembourg City' WHERE "iso3" = 'LUX' AND "year" = 2007;`,
    `INSERT INTO "public"."ihli_data" ("iso3", "year", "country_name") VALUES ('USA', 2008, 'United States');`,
  ]);
});

test("builds Hive grid save statements with backtick identifiers", () => {
  const statements = buildDataGridSaveStatements({
    databaseType: "hive",
    tableMeta: {
      tableName: "department states",
      primaryKeys: ["dept id"],
    },
    columns: ["dept id", "display name"],
    rows: [[10, "Sales"]],
    dirtyRows: [[0, [[1, "Marketing"]]]],
    deletedRows: [0],
    newRows: [[20, "Engineering"]],
  });

  assert.deepEqual(statements, [
    "UPDATE `department states` SET `display name` = 'Marketing' WHERE `dept id` = 10;",
    "DELETE FROM `department states` WHERE `dept id` = 10;",
    "INSERT INTO `department states` (`dept id`, `display name`) VALUES (20, 'Engineering');",
  ]);
});

test("builds Hive grid save statements without primary keys using row predicates", () => {
  const statements = buildDataGridSaveStatements({
    databaseType: "hive",
    tableMeta: {
      tableName: "departments",
      primaryKeys: [],
    },
    columns: ["id", "name", "location"],
    rows: [[10, "Sales", null]],
    dirtyRows: [[0, [[1, "Marketing"]]]],
    deletedRows: [0],
    newRows: [],
  });

  assert.deepEqual(statements, [
    "UPDATE `departments` SET `name` = 'Marketing' WHERE `id` = 10 AND `name` = 'Sales' AND `location` IS NULL;",
    "DELETE FROM `departments` WHERE `id` = 10 AND `name` = 'Sales' AND `location` IS NULL;",
  ]);
});

test("builds TDengine append-only grid save statements", () => {
  const newTs = tdengineTimestampLiteral("2026-05-16 09:40:00.000");
  const statements = buildDataGridSaveStatements({
    databaseType: "tdengine",
    tableMeta: {
      schema: "test_db",
      tableName: "meters",
      primaryKeys: ["ts"],
    },
    columns: ["ts", "current", "voltage", "location", "groupid"],
    rows: [["2026-05-16 09:35:57.000", 10.3, 219, "Beijing", 1]],
    dirtyRows: [[0, [[1, 11.1]]]],
    deletedRows: [0],
    newRows: [["2026-05-16 09:40:00.000", 12.1, 220, "Shenzhen", 3]],
  });

  assert.deepEqual(statements, [
    `INSERT INTO \`test_db\`.\`meters\` (\`ts\`, \`current\`, \`voltage\`, \`location\`, \`groupid\`) VALUES (${newTs}, 12.1, 220, 'Shenzhen', 3);`,
  ]);
});

test("builds TDengine supertable overwrites as full-row inserts by tbname and timestamp", () => {
  const existingTs = tdengineTimestampLiteral("2026-05-16 09:35:57.000");
  const newTs = tdengineTimestampLiteral("2026-05-16 09:40:00.000");
  const statements = buildDataGridSaveStatements({
    databaseType: "tdengine",
    tableMeta: {
      schema: "test_db",
      tableName: "meters",
      primaryKeys: ["tbname", "ts"],
      columns: [
        { name: "ts", data_type: "TIMESTAMP", is_nullable: false, is_primary_key: true },
        { name: "current", data_type: "FLOAT", is_nullable: true, is_primary_key: false },
        { name: "voltage", data_type: "INT", is_nullable: true, is_primary_key: false },
        { name: "location", data_type: "NCHAR", is_nullable: true, is_primary_key: false, extra: "TAG" },
        { name: "groupid", data_type: "INT", is_nullable: true, is_primary_key: false, extra: "TAG" },
      ],
    },
    columns: ["tbname", "ts", "current", "voltage", "location", "groupid"],
    rows: [["d1001", "2026-05-16 09:35:57.000", 10.3, 219, "Beijing", 1]],
    dirtyRows: [[0, [[2, 11.1]]]],
    deletedRows: [0],
    newRows: [["d1003", "2026-05-16 09:40:00.000", 12.1, 220, "Shenzhen", 3]],
  });

  assert.deepEqual(statements, [
    `INSERT INTO \`test_db\`.\`meters\` (\`tbname\`, \`ts\`, \`current\`, \`voltage\`, \`location\`, \`groupid\`) VALUES ('d1001', ${existingTs}, 11.1, 219, 'Beijing', 1);`,
    `INSERT INTO \`test_db\`.\`meters\` (\`tbname\`, \`ts\`, \`current\`, \`voltage\`, \`location\`, \`groupid\`) VALUES ('d1003', ${newTs}, 12.1, 220, 'Shenzhen', 3);`,
  ]);
});

test("builds TDengine child table overwrites without pseudo or tag columns", () => {
  const existingTs = tdengineTimestampLiteral("2026-05-16 09:35:57.000");
  const newTs = tdengineTimestampLiteral("2026-05-16 09:40:00.000");
  const statements = buildDataGridSaveStatements({
    databaseType: "tdengine",
    tableMeta: {
      schema: "test_db",
      tableName: "d1001",
      primaryKeys: ["tbname", "ts"],
      columns: [
        { name: "ts", data_type: "TIMESTAMP", is_nullable: false, is_primary_key: true },
        { name: "current", data_type: "FLOAT", is_nullable: true, is_primary_key: false },
        { name: "voltage", data_type: "INT", is_nullable: true, is_primary_key: false },
        { name: "location", data_type: "NCHAR", is_nullable: true, is_primary_key: false, extra: "TAG" },
        { name: "groupid", data_type: "INT", is_nullable: true, is_primary_key: false, extra: "TAG" },
      ],
    },
    columns: ["tbname", "ts", "current", "voltage", "location", "groupid"],
    rows: [["d1001", "2026-05-16 09:35:57.000", 10.3, 219, "Beijing", 1]],
    dirtyRows: [[0, [[2, 11.1]]]],
    deletedRows: [],
    newRows: [["d1001", "2026-05-16 09:40:00.000", 12.1, 220, "Beijing", 1]],
  });

  assert.deepEqual(statements, [
    `INSERT INTO \`test_db\`.\`d1001\` (\`ts\`, \`current\`, \`voltage\`) VALUES (${existingTs}, 11.1, 219);`,
    `INSERT INTO \`test_db\`.\`d1001\` (\`ts\`, \`current\`, \`voltage\`) VALUES (${newTs}, 12.1, 220);`,
  ]);
});

test("formats TDengine timestamp literals with the local timezone offset", () => {
  assert.equal(
    formatGridSqlLiteral("2026-05-16 09:35:57.975", "tdengine"),
    tdengineTimestampLiteral("2026-05-16 09:35:57.975"),
  );
});

function tdengineTimestampLiteral(text: string): string {
  const [datePart, timePart] = text.split(" ");
  const [time, rawFraction = ""] = timePart.split(".");
  const fraction = `.${rawFraction.padEnd(3, "0").slice(0, 3)}`;
  const date = new Date(`${datePart}T${time}${fraction}`);
  const offsetMinutes = date.getTimezoneOffset();
  const sign = offsetMinutes <= 0 ? "+" : "-";
  const abs = Math.abs(offsetMinutes);
  const hours = String(Math.floor(abs / 60)).padStart(2, "0");
  const minutes = String(abs % 60).padStart(2, "0");
  return `'${datePart}T${time}${fraction}${sign}${hours}:${minutes}'`;
}

test("builds Trino insert statements with schema-qualified identifiers", () => {
  const statements = buildDataGridSaveStatements({
    databaseType: "trino",
    tableMeta: {
      schema: "tiny",
      tableName: "nation",
      primaryKeys: [],
    },
    columns: ["nationkey", "name"],
    rows: [],
    dirtyRows: [],
    deletedRows: [],
    newRows: [[100, "Atlantis"]],
  });

  assert.deepEqual(statements, ['INSERT INTO "tiny"."nation" ("nationkey", "name") VALUES (100, \'Atlantis\');']);
});

test("builds Informix grid save statements without delimited identifiers", () => {
  const statements = buildDataGridSaveStatements({
    databaseType: "informix",
    tableMeta: {
      schema: "testdb",
      tableName: "dbx_grid_edit_probe",
      primaryKeys: ["id"],
    },
    columns: ["id", "name"],
    rows: [[1, "before"]],
    dirtyRows: [[0, [[1, "after"]]]],
    deletedRows: [0],
    newRows: [[2, "new"]],
  });

  assert.deepEqual(statements, [
    "UPDATE dbx_grid_edit_probe SET name = 'after' WHERE id = 1;",
    "DELETE FROM dbx_grid_edit_probe WHERE id = 1;",
    "INSERT INTO dbx_grid_edit_probe (id, name) VALUES (2, 'new');",
  ]);
});

test("uses Oracle ROWID as a synthetic key without writing it as a normal column", () => {
  const statements = buildDataGridSaveStatements({
    databaseType: "oracle",
    tableMeta: {
      schema: "DBXTEST",
      tableName: "DBX_LOAD_TABLE_006",
      primaryKeys: ["__DBX_ROWID"],
    },
    columns: ["__DBX_ROWID", "ID", "CITY", "NOTE"],
    rows: [["AAATiBAABAAABrXAAA", 1, "上海", "old"]],
    dirtyRows: [[0, [[2, "北京"]]]],
    deletedRows: [0],
    newRows: [[null, 2, "广州", "new"]],
  });

  assert.deepEqual(statements, [
    `UPDATE "DBXTEST"."DBX_LOAD_TABLE_006" SET "CITY" = '北京' WHERE ROWIDTOCHAR(ROWID) = 'AAATiBAABAAABrXAAA';`,
    `DELETE FROM "DBXTEST"."DBX_LOAD_TABLE_006" WHERE ROWIDTOCHAR(ROWID) = 'AAATiBAABAAABrXAAA';`,
    `INSERT INTO "DBXTEST"."DBX_LOAD_TABLE_006" ("ID", "CITY", "NOTE") VALUES (2, '广州', 'new');`,
  ]);
});

test("builds Neo4j grid save statements with elementId keys", () => {
  const statements = buildDataGridSaveStatements({
    databaseType: "neo4j",
    tableMeta: {
      tableName: "Employee",
      primaryKeys: [DBX_NEO4J_ELEMENT_ID_COLUMN],
    },
    columns: [DBX_NEO4J_ELEMENT_ID_COLUMN, "name", "active"],
    rows: [["4:abc:7", "Ada", true]],
    dirtyRows: [[0, [[1, "Grace"]]]],
    deletedRows: [0],
    newRows: [[null, "Linus", false]],
  });

  assert.deepEqual(statements, [
    "MATCH (n:`Employee`) WHERE elementId(n) = '4:abc:7' SET n.`name` = 'Grace';",
    "MATCH (n:`Employee`) WHERE elementId(n) = '4:abc:7' DETACH DELETE n;",
    "CREATE (n:`Employee` {`name`: 'Linus', `active`: FALSE});",
  ]);
});

test("builds best-effort Neo4j rollback statements", () => {
  const statements = buildDataGridRollbackStatements({
    databaseType: "neo4j",
    tableMeta: {
      tableName: "Employee",
      primaryKeys: [DBX_NEO4J_ELEMENT_ID_COLUMN],
    },
    columns: [DBX_NEO4J_ELEMENT_ID_COLUMN, "name", "active"],
    rows: [["4:abc:7", "Ada", true]],
    dirtyRows: [[0, [[1, "Grace"]]]],
    deletedRows: [0],
    newRows: [[null, "Linus", false]],
  });

  assert.deepEqual(statements, [
    "MATCH (n:`Employee`) WHERE n.`name` = 'Linus' AND n.`active` = FALSE DETACH DELETE n;",
    "CREATE (n:`Employee` {`name`: 'Ada', `active`: TRUE});",
    "MATCH (n:`Employee`) WHERE elementId(n) = '4:abc:7' SET n.`name` = 'Ada';",
  ]);
});

test("skips schema setup for Neo4j data grid saves", () => {
  assert.equal(
    dataGridSaveExecutionSchema("neo4j", { schema: "ignored", tableName: "Employee", primaryKeys: [] }),
    undefined,
  );
});

test("skips current_schema setup for Oracle data grid saves", () => {
  assert.equal(
    dataGridSaveExecutionSchema("oracle", { schema: "DBXTEST", tableName: "T", primaryKeys: [] }),
    undefined,
  );
  assert.equal(
    dataGridSaveExecutionSchema("postgres", { schema: "public", tableName: "T", primaryKeys: [] }),
    "public",
  );
});

test("normalizes Hive ACID update and delete errors", () => {
  const error = normalizeDataGridSaveError(
    "hive",
    "Statement 1 failed: Agent RPC error (-1): Error while compiling statement: FAILED: SemanticException [Error 10294]: Attempt to do update or delete using transaction manager that does not support these operations.. Previous 0 statement(s) may have been committed.",
  );

  assert.equal(
    error,
    "Hive UPDATE/DELETE are not enabled for this table or server. Add rows with INSERT, or enable ACID transactional tables in Hive before editing/deleting existing rows.",
  );
});

test("rejects NULL writes to non-null table columns", () => {
  const error = validateDataGridSave({
    columns: ["ID", "CREATED_AT", "CITY"],
    columnInfo: [
      { name: "ID", is_nullable: false, is_primary_key: true },
      { name: "CREATED_AT", is_nullable: false, is_primary_key: false },
      { name: "CITY", is_nullable: true, is_primary_key: false },
    ],
    dirtyRows: [[0, [[1, null]]]],
    newRows: [[2, null, "上海"]],
  });

  assert.equal(error, 'Column "CREATED_AT" does not allow NULL.');
});

test("allows NULL for MySQL auto increment columns when inserting rows", () => {
  const error = validateDataGridSave({
    databaseType: "mysql",
    tableMeta: {
      tableName: "people",
      primaryKeys: ["id"],
    },
    columns: ["id", "name"],
    rows: [],
    columnInfo: [
      { name: "id", is_nullable: false, column_default: null, is_primary_key: true, extra: "auto_increment" },
      { name: "name", is_nullable: false, column_default: null, is_primary_key: false, extra: null },
    ],
    dirtyRows: [],
    newRows: [[null, "Ada"]],
  });

  assert.equal(error, undefined);
});

test("rejects inserted rows that duplicate existing primary key values", () => {
  const error = validateDataGridSave({
    databaseType: "postgres",
    tableMeta: {
      tableName: "education_data",
      primaryKeys: ["country_code", "year"],
    },
    columns: ["country_code", "year", "value"],
    rows: [["ALB", 2021, 0.812]],
    columnInfo: [
      { name: "country_code", is_nullable: false, column_default: null, is_primary_key: true, extra: null },
      { name: "year", is_nullable: false, column_default: null, is_primary_key: true, extra: null },
      { name: "value", is_nullable: true, column_default: null, is_primary_key: false, extra: null },
    ],
    dirtyRows: [],
    newRows: [["ALB", 2021, 0.812]],
  });

  assert.equal(
    error,
    'New row duplicates the existing primary key (country_code = "ALB", year = 2021). Change the key before saving.',
  );
});

test("rejects inserted rows that duplicate another new row primary key", () => {
  const error = validateDataGridSave({
    databaseType: "postgres",
    tableMeta: {
      tableName: "education_data",
      primaryKeys: ["country_code", "year"],
    },
    columns: ["country_code", "year", "value"],
    rows: [],
    columnInfo: [
      { name: "country_code", is_nullable: false, column_default: null, is_primary_key: true, extra: null },
      { name: "year", is_nullable: false, column_default: null, is_primary_key: true, extra: null },
      { name: "value", is_nullable: true, column_default: null, is_primary_key: false, extra: null },
    ],
    dirtyRows: [],
    newRows: [
      ["ALB", 2021, 0.812],
      ["ALB", 2021, 0.913],
    ],
  });

  assert.equal(
    error,
    'New row duplicates another new row\'s primary key (country_code = "ALB", year = 2021). Change the key before saving.',
  );
});