import { describe, test } from "vitest";
import type { Kysely, Transaction } from "kysely";

export { describe, beforeAll, afterAll } from "vitest";

// This is our custom test context. Read more about test context here in official docs: https://vitest.dev/guide/test-context
interface DatabaseFixture {
  trx: Transaction<Database>;
}

const withDatabaseTransaction = test.extend<DatabaseFixture>({
  // eslint-disable-next-line no-empty-pattern
  trx: async ({}, use) => {
    const database: Kysely<Database> = await getDatabase();
    try {
      await database.transaction().execute(async (trx: Transaction<Database>) => {
        await use(trx);
        // there is no way to rollback in kysely, so we throw an error to simulate a rollback
        throw new KyseleyRollbackError();
      });
    } catch (err) {
      const isRollbackError = err instanceof KyseleyRollbackError;
      if (!isRollbackError) {
        throw err;
      }
    }
  }
});

describe("Test 1", () => {
  withDatabaseTransaction("run in transaction", async ({ trx }) => {
    // use trx here
  });
});

// eslint-disable-next-line @typescript-eslint/no-empty-interface
interface Database {} // Kyseley database interface

function getDatabase(): Promise<Kysely<Database>> {
  // create a singleton Kyseley instance and return it
  throw new Error("Function not implemented");
}

class KyseleyRollbackError extends Error {}