import { Kysely, sql } from "kysely";

// eslint-disable-next-line @typescript-eslint/no-explicit-any
export const up = async (db: Kysely<any>) => {
  // file.ts
  await db.schema
    .alterTable("file")
    .alterColumn("encrypted_content_iv", (col) => col.dropNotNull())
    .execute();

  // media.ts
  await db.schema
    .alterTable("thumbnail")
    .alterColumn("encrypted_content_iv", (col) => col.dropNotNull())
    .execute();

  // upload.ts
  await db.schema
    .createTable("upload_session")
    .addColumn("id", "uuid", (col) => col.primaryKey())
    .addColumn("type", "text", (col) => col.notNull())
    .addColumn("user_id", "integer", (col) => col.references("user.id").notNull())
    .addColumn("path", "text", (col) => col.notNull())
    .addColumn("bitmap", "bytea", (col) => col.notNull())
    .addColumn("total_chunks", "integer", (col) => col.notNull())
    .addColumn("uploaded_chunks", "integer", (col) =>
      col
        .generatedAlwaysAs(sql`bit_count(bitmap)`)
        .stored()
        .notNull(),
    )
    .addColumn("expires_at", "timestamp(3)", (col) => col.notNull())
    .addColumn("parent_id", "integer", (col) => col.references("directory.id"))
    .addColumn("master_encryption_key_version", "integer")
    .addColumn("encrypted_data_encryption_key", "text")
    .addColumn("data_encryption_key_version", "timestamp(3)")
    .addColumn("hmac_secret_key_version", "integer")
    .addColumn("content_type", "text")
    .addColumn("encrypted_name", "json")
    .addColumn("encrypted_created_at", "json")
    .addColumn("encrypted_last_modified_at", "json")
    .addColumn("file_id", "integer", (col) => col.references("file.id"))
    .addForeignKeyConstraint(
      "upload_session_fk01",
      ["user_id", "master_encryption_key_version"],
      "master_encryption_key",
      ["user_id", "version"],
    )
    .addForeignKeyConstraint(
      "upload_session_fk02",
      ["user_id", "hmac_secret_key_version"],
      "hmac_secret_key",
      ["user_id", "version"],
    )
    .addCheckConstraint(
      "upload_session_ck01",
      sql`length(bitmap) = ceil(total_chunks / 8.0)::integer`,
    )
    .addCheckConstraint("upload_session_ck02", sql`uploaded_chunks <= total_chunks`)
    .execute();
};

// eslint-disable-next-line @typescript-eslint/no-explicit-any
export const down = async (db: Kysely<any>) => {
  await db.deleteFrom("file_log").where("action", "=", "migrate").execute();

  await db.schema.dropTable("upload_session").execute();
  await db.schema
    .alterTable("thumbnail")
    .alterColumn("encrypted_content_iv", (col) => col.setNotNull())
    .execute();
  await db.schema
    .alterTable("file")
    .alterColumn("encrypted_content_iv", (col) => col.setNotNull())
    .execute();
};