import { err, ok, type CommandContext } from "@tailor-platform/erp-kit/core";
import { salesOrderLifecycle } from "../db/salesOrder.lifecycle.generated";
import type { Transaction } from "../generated/kysely-tailordb";
import {
  InvalidOrderStatusError,
  RejectionReasonRequiredError,
  SalesOrderNotFoundError,
} from "../lib/errors.generated";

export interface RejectSalesOrderInput {
  salesOrderId: string;
  rejectionReason: string;
}

export async function run(db: Transaction, input: RejectSalesOrderInput, 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));
  }
  const nextStatus = salesOrderLifecycle.tryTransition(salesOrder.status, "reject");
  if (!nextStatus) {
    return err(new InvalidOrderStatusError(input.salesOrderId));
  }
  if (!input.rejectionReason.trim()) {
    return err(new RejectionReasonRequiredError(input.salesOrderId));
  }
  const updated = await db
    .updateTable("SalesOrder")
    .set({
      status: nextStatus,
      rejectionReason: input.rejectionReason,
    })
    .where("id", "=", input.salesOrderId)
    .returningAll()
    .executeTakeFirstOrThrow();
  return ok({ salesOrder: updated });
}
