import { err, ok, type CommandContext } from "@tailor-platform/erp-kit/core";
import { purchaseRequisitionLifecycle } from "../db/purchaseRequisition.lifecycle.generated";
import type { Transaction } from "../generated/kysely-tailordb";
import {
  PurchaseRequisitionNotFoundError,
  PurchaseRequisitionNotSubmittedError,
  RejectionReasonRequiredError,
} from "../lib/errors.generated";

export interface RejectPurchaseRequisitionInput {
  id: string;
  reason: string;
}

export async function run(
  db: Transaction,
  input: RejectPurchaseRequisitionInput,
  ctx: CommandContext,
) {
  void ctx;
  const requisition = await db
    .selectFrom("PurchaseRequisition")
    .selectAll()
    .where("id", "=", input.id)
    .forUpdate()
    .executeTakeFirst();
  if (!requisition) {
    return err(new PurchaseRequisitionNotFoundError(input.id));
  }
  const nextStatus = purchaseRequisitionLifecycle.tryTransition(requisition.status, "reject");
  if (!nextStatus) {
    return err(new PurchaseRequisitionNotSubmittedError(input.id));
  }
  if (!input.reason.trim()) {
    return err(new RejectionReasonRequiredError(input.id));
  }

  const updated = await db
    .updateTable("PurchaseRequisition")
    .set({
      status: nextStatus,
      rejectionReason: input.reason,
    })
    .where("id", "=", input.id)
    .returningAll()
    .executeTakeFirstOrThrow();

  return ok({ purchaseRequisition: updated });
}
