import { TransactionState } from "@medusajs/orchestration"
import { DALUtils, generateEntityId } from "@medusajs/utils"
import {
  BeforeCreate,
  Entity,
  Enum,
  Filter,
  Index,
  OnInit,
  OptionalProps,
  PrimaryKey,
  Property,
  Unique,
} from "@mikro-orm/core"

type OptionalFields = "deleted_at"

@Entity()
@Unique({
  name: "IDX_workflow_execution_workflow_id_transaction_id_unique",
  properties: ["workflow_id", "transaction_id"],
})
@Filter(DALUtils.mikroOrmSoftDeletableFilterOptions)
export default class WorkflowExecution {
  [OptionalProps]?: OptionalFields

  @Property({ columnType: "text", nullable: false })
  @Index({ name: "IDX_workflow_execution_id" })
  id!: string

  @Index({ name: "IDX_workflow_execution_workflow_id" })
  @PrimaryKey({ columnType: "text" })
  workflow_id: string

  @Index({ name: "IDX_workflow_execution_transaction_id" })
  @PrimaryKey({ columnType: "text" })
  transaction_id: string

  @Property({ columnType: "jsonb", nullable: true })
  execution: Record<string, unknown> | null = null

  @Property({ columnType: "jsonb", nullable: true })
  context: Record<string, unknown> | null = null

  @Index({ name: "IDX_workflow_execution_state" })
  @Enum(() => TransactionState)
  state: TransactionState

  @Property({
    onCreate: () => new Date(),
    columnType: "timestamptz",
    defaultRaw: "now()",
  })
  created_at: Date

  @Property({
    onCreate: () => new Date(),
    onUpdate: () => new Date(),
    columnType: "timestamptz",
    defaultRaw: "now()",
  })
  updated_at: Date

  @Property({ columnType: "timestamptz", nullable: true })
  deleted_at: Date | null = null

  @BeforeCreate()
  onCreate() {
    this.id = generateEntityId(this.id, "wf_exec")
  }

  @OnInit()
  onInit() {
    this.id = generateEntityId(this.id, "wf_exec")
  }
}