import { Database } from "./database.js";
import { assertEqualRows, getConfig } from "./helpers.js";
import {
  doltAddFields,
  doltCheckoutFields,
  doltCommitFields,
  countFields,
} from "./fields.js";
import { mergeMatcher } from "./workbenchTests/matchers.js";

const tests = [
  {
    q: "create table test (pk int, value int, primary key(pk))",
    res: {
      command: "CREATE",
      rowCount: null,
      oid: null,
      rows: [],
      fields: [],
    },
  },
  {
    q: "select * from test",
    res: {
      command: "SELECT",
      rowCount: 0,
      oid: null,
      rows: [],
      fields: [
        {
          name: "pk",
          tableID: 0, // TODO: need to be filled? Got 16859 from Postgres
          columnID: 1,
          dataTypeID: 23,
          dataTypeSize: 4,
          dataTypeModifier: -1,
          format: "text",
        },
        {
          name: "value",
          tableID: 0, // TODO: need to be filled? Got 16859 from Postgres
          columnID: 2,
          dataTypeID: 23,
          dataTypeSize: 4,
          dataTypeModifier: -1,
          format: "text",
        },
      ],
    },
  },
  {
    q: "insert into test (pk, value) values (0,0)",
    res: {
      command: "INSERT",
      rowCount: 1,
      oid: 0,
      rows: [],
      fields: [],
    },
  },
  {
    q: "select * from test",
    res: {
      command: "SELECT",
      rowCount: 1,
      oid: null,
      rows: [{ pk: 0, value: 0 }],
      fields: [
        {
          name: "pk",
          tableID: 0,
          columnID: 1,
          dataTypeID: 23,
          dataTypeSize: 4,
          dataTypeModifier: -1,
          format: "text",
        },
        {
          name: "value",
          tableID: 0,
          columnID: 2,
          dataTypeID: 23,
          dataTypeSize: 4,
          dataTypeModifier: -1,
          format: "text",
        },
      ],
    },
  },
  {
    q: "select dolt_add('-A');",
    res: {
      command: "SELECT",
      rowCount: 1,
      oid: null,
      rows: [{ dolt_add: ["0"] }],
      fields: doltAddFields,
    },
  },
  {
    q: "select dolt_commit('-m', 'my commit')",
    res: {
      command: "SELECT",
      rowCount: 1,
      oid: null,
      rows: [{ dolt_commit: [""] }],
      fields: doltCommitFields,
    },
  },
  {
    q: "select COUNT(*) FROM dolt.log",
    res: {
      command: "SELECT",
      rowCount: 1,
      oid: null,
      rows: [{ count: "3" }],
      fields: countFields,
    },
  },
  {
    q: "select dolt_checkout('-b', 'mybranch')",
    res: {
      command: "SELECT",
      rowCount: 1,
      oid: null,
      rows: [{ dolt_checkout: ["0", "Switched to branch 'mybranch'"] }],
      fields: doltCheckoutFields,
    },
  },
  {
    q: "insert into test (pk, value) values (1,1),(2,3)",
    res: {
      command: "INSERT",
      rowCount: 2,
      oid: 0,
      rows: [],
      fields: [],
    },
  },
  {
    q: "select dolt_commit('-a', '-m', 'my commit2')",
    res: {
      command: "SELECT",
      rowCount: 1,
      oid: null,
      rows: [{ dolt_commit: [""] }],
      fields: doltCommitFields,
    },
  },
  {
    q: "select dolt_checkout('main')",
    res: {
      command: "SELECT",
      rowCount: 1,
      oid: null,
      rows: [{ dolt_checkout: ["0", "Switched to branch 'main'"] }],
      fields: doltCheckoutFields,
    },
  },
  {
    q: "select dolt_merge('mybranch')",
    res: {
      command: "SELECT",
      rowCount: 1,
      oid: null,
      rows: [
        {
          dolt_merge: ["", "1", "0", "merge successful"],
        },
      ],
      fields: [],
    },
    matcher: mergeMatcher,
  },
  {
    q: "select COUNT(*) FROM dolt.log",
    res: {
      command: "SELECT",
      rowCount: 1,
      oid: null,
      rows: [{ count: "4" }],
      fields: countFields,
    },
  },
];

async function main() {
  const database = new Database(getConfig());

  await Promise.all(
    tests.map((test) => {
      return database
        .query(test.q)
        .then((data) => {
          assertEqualRows(test, data);
        })
        .catch((err) => {
          console.error(err);
          process.exit(1);
        });
    })
  );

  database.close();
  process.exit(0);
}

main();