import {
  RdbmsSessionStorageMigrator,
  RdbmsSessionStorageMigratorOptions,
  MigrationOperation,
} from '@shopify/shopify-app-session-storage';

import {SqliteConnection} from './sqlite-connection';

export class SqliteSessionStorageMigrator extends RdbmsSessionStorageMigrator {
  constructor(
    dbConnection: SqliteConnection,
    opts: Partial<RdbmsSessionStorageMigratorOptions> = {},
    migrations: MigrationOperation[],
  ) {
    super(dbConnection, opts, migrations);
  }

  async initMigrationPersistence(): Promise<void> {
    const discardCreateTable = await this.connection.hasTable(
      this.options.migrationDBIdentifier,
    );

    if (!discardCreateTable) {
      const migration = `
        CREATE TABLE ${this.options.migrationDBIdentifier} (
          ${
            this.getOptions().migrationNameColumnName
          } varchar(255) NOT NULL PRIMARY KEY
        );`;

      await this.connection.query(migration, []);
    }
  }
}