import {
  BigNumber,
  createPsqlIndexStatementHelper,
  generateEntityId,
} from "@medusajs/framework/utils"
import {
  BeforeCreate,
  Entity,
  ManyToOne,
  OnInit,
  PrimaryKey,
  Property,
  Rel,
} from "@mikro-orm/core"
import Order from "./order"

type OrderSummaryTotals = {
  total: BigNumber
  subtotal: BigNumber
  total_tax: BigNumber

  ordered_total: BigNumber
  fulfilled_total: BigNumber
  returned_total: BigNumber
  return_request_total: BigNumber
  write_off_total: BigNumber
  projected_total: BigNumber

  net_total: BigNumber
  net_subtotal: BigNumber
  net_total_tax: BigNumber

  balance: BigNumber

  paid_total: BigNumber
  refunded_total: BigNumber
}

const tableName = "order_summary"

const OrderIdVersionIndex = createPsqlIndexStatementHelper({
  tableName,
  columns: ["order_id", "version"],
  where: "deleted_at IS NOT NULL",
})

const DeletedAtIndex = createPsqlIndexStatementHelper({
  tableName,
  columns: "deleted_at",
  where: "deleted_at IS NOT NULL",
})

@Entity({ tableName })
@OrderIdVersionIndex.MikroORMIndex()
export default class OrderSummary {
  @PrimaryKey({ columnType: "text" })
  id: string

  @ManyToOne({
    entity: () => Order,
    columnType: "text",
    fieldName: "order_id",
    mapToPk: true,
    onDelete: "cascade",
  })
  order_id: string

  @ManyToOne(() => Order, {
    persist: false,
  })
  order: Rel<Order>

  @Property({
    columnType: "integer",
    defaultRaw: "1",
  })
  version: number = 1

  @Property({ columnType: "jsonb" })
  totals: OrderSummaryTotals | null = {} as OrderSummaryTotals

  @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 })
  @DeletedAtIndex.MikroORMIndex()
  deleted_at: Date | null = null

  @BeforeCreate()
  onCreate() {
    this.id = generateEntityId(this.id, "ordsum")
    this.order_id ??= this.order?.id
  }

  @OnInit()
  onInit() {
    this.id = generateEntityId(this.id, "ordsum")
    this.order_id ??= this.order?.id
  }
}