import type { Transaction } from "../generated/kysely-tailordb";
import {
  InvalidOrderStatusError,
  OrderNotClosableError,
  SalesOrderNotFoundError,
} from "../lib/errors.generated";
import { err, ok, type CommandContext } from "@tailor-platform/erp-kit/module";

export interface CloseSalesOrderInput {
  salesOrderId: string;
  closeReason?: string;
}

export async function run(db: Transaction, input: CloseSalesOrderInput, ctx: CommandContext) {
  void ctx;
  const salesOrder = await db
    .selectFrom("SalesOrder")
    .selectAll()
    .where("id", "=", input.salesOrderId)
    .forUpdate()
    .executeTakeFirst();
  if (!salesOrder) {
    return err(new SalesOrderNotFoundError(input.salesOrderId));
  }
  if (
    !["SHIPPED", "INVOICED", "PARTIALLY_SHIPPED", "PARTIALLY_INVOICED", "CONFIRMED"].includes(
      salesOrder.status,
    )
  ) {
    return err(new InvalidOrderStatusError(input.salesOrderId));
  }
  const lines = await db
    .selectFrom("SalesOrderLine")
    .selectAll()
    .where("salesOrderId", "=", input.salesOrderId)
    .execute();
  if (
    lines.some(
      (line) => line.shippedQuantity < line.quantity || line.invoicedQuantity < line.quantity,
    )
  ) {
    return err(new OrderNotClosableError(input.salesOrderId));
  }
  const updated = await db
    .updateTable("SalesOrder")
    .set({ status: "CLOSED", closeReason: input.closeReason ?? null, updatedAt: new Date() })
    .where("id", "=", input.salesOrderId)
    .returningAll()
    .executeTakeFirst();
  return ok({ salesOrder: updated! });
}
