import type { EntityCompanionDefinition, ViewerContext } from '@expo/entity';
import {
  AlwaysAllowPrivacyPolicyRule,
  Entity,
  EntityConfiguration,
  EntityPrivacyPolicy,
  IntField,
  StringField,
} from '@expo/entity';
import type { Knex } from 'knex';

type InvalidTestEntityFields = {
  id: number;
  name: string | null;
};

export class InvalidTestEntity extends Entity<InvalidTestEntityFields, 'id', ViewerContext> {
  static defineCompanionDefinition(): EntityCompanionDefinition<
    InvalidTestEntityFields,
    'id',
    ViewerContext,
    InvalidTestEntity,
    InvalidTestEntityPrivacyPolicy
  > {
    return {
      entityClass: InvalidTestEntity,
      entityConfiguration: invalidTestEntityConfiguration,
      privacyPolicyClass: InvalidTestEntityPrivacyPolicy,
    };
  }

  public static async createOrTruncatePostgresTableAsync(knex: Knex): Promise<void> {
    const tableName = 'postgres_test_entities';
    const hasTable = await knex.schema.hasTable(tableName);
    if (!hasTable) {
      await knex.schema.createTable(tableName, (table) => {
        table.integer('id'); // invalid because ID is not a primary key or unique
        table.string('name');
      });
    }
    await knex.into(tableName).truncate();
  }

  public static async dropPostgresTableAsync(knex: Knex): Promise<void> {
    const tableName = 'postgres_test_entities';
    const hasTable = await knex.schema.hasTable(tableName);
    if (hasTable) {
      await knex.schema.dropTable(tableName);
    }
  }
}

class InvalidTestEntityPrivacyPolicy extends EntityPrivacyPolicy<
  InvalidTestEntityFields,
  'id',
  ViewerContext,
  InvalidTestEntity
> {
  protected override readonly createRules = [
    new AlwaysAllowPrivacyPolicyRule<
      InvalidTestEntityFields,
      'id',
      ViewerContext,
      InvalidTestEntity
    >(),
  ];
  protected override readonly readRules = [
    new AlwaysAllowPrivacyPolicyRule<
      InvalidTestEntityFields,
      'id',
      ViewerContext,
      InvalidTestEntity
    >(),
  ];
  protected override readonly updateRules = [
    new AlwaysAllowPrivacyPolicyRule<
      InvalidTestEntityFields,
      'id',
      ViewerContext,
      InvalidTestEntity
    >(),
  ];
  protected override readonly deleteRules = [
    new AlwaysAllowPrivacyPolicyRule<
      InvalidTestEntityFields,
      'id',
      ViewerContext,
      InvalidTestEntity
    >(),
  ];
}

export const invalidTestEntityConfiguration = new EntityConfiguration<
  InvalidTestEntityFields,
  'id'
>({
  idField: 'id',
  tableName: 'postgres_test_entities',
  schema: {
    id: new IntField({
      columnName: 'id',
      cache: false,
    }),
    name: new StringField({
      columnName: 'name',
    }),
  },
  databaseAdapterFlavor: 'postgres',
  cacheAdapterFlavor: 'redis',
});