import { ok, type ValueOf } from "@tailor-platform/erp-kit/core";
import { vi } from "vitest";
import type {
  InventoryCommands,
  InventoryQueries,
  ItemManagementQueries,
  PurchaseCommands,
  PurchaseQueries,
} from "../module";

export type Item = NonNullable<
  Awaited<ReturnType<ItemManagementQueries["getItem"]>>["value"]["item"]
>;

export function mockItemManagementQueries(
  item: Item | null,
): Pick<ItemManagementQueries, "getItem"> {
  return {
    getItem: () => Promise.resolve({ ok: true, value: { item } }),
  };
}

export type StorageLocation = NonNullable<
  Awaited<ReturnType<InventoryQueries["getStorageLocation"]>>["value"]["storageLocation"]
>;
export type PurchaseOrderLineForMatching = ValueOf<
  Awaited<ReturnType<PurchaseQueries["listPurchaseOrderLinesForMatching"]>>
>["purchaseOrderLines"][number];
export type InventoryLedgerEntry = ValueOf<
  Awaited<ReturnType<InventoryCommands["postInventoryLedger"]>>
>["inventoryLedgerEntries"][number];

export const mockInventoryQueries = (store: {
  storageLocation?: StorageLocation;
}): Pick<InventoryQueries, "getStorageLocation"> => ({
  getStorageLocation: vi.fn<InventoryQueries["getStorageLocation"]>(() =>
    Promise.resolve(
      store.storageLocation
        ? ok({ storageLocation: store.storageLocation, stockLevels: [] })
        : ok({ storageLocation: null, stockLevels: [] }),
    ),
  ),
});

export const mockInventoryCommands = (
  result: Awaited<ReturnType<InventoryCommands["postInventoryLedger"]>> = ok({
    inventoryLedgerEntries: [],
  }),
) => ({
  postInventoryLedger: vi.fn<InventoryCommands["postInventoryLedger"]>(() =>
    Promise.resolve(result),
  ),
});

export const mockPurchaseCommands = (
  result: Awaited<ReturnType<PurchaseCommands["recalculatePurchaseOrderReceiptStatus"]>> = ok({
    purchaseOrderIds: [],
  }),
) => ({
  recalculatePurchaseOrderReceiptStatus: vi.fn<
    PurchaseCommands["recalculatePurchaseOrderReceiptStatus"]
  >(() => Promise.resolve(result)),
});

export const mockPurchaseQueries = (store: {
  purchaseOrderLines?: PurchaseOrderLineForMatching[];
}) => ({
  listPurchaseOrderLinesForMatching: vi.fn<PurchaseQueries["listPurchaseOrderLinesForMatching"]>(
    (_db, input) =>
      Promise.resolve(
        ok({
          purchaseOrderLines: (store.purchaseOrderLines ?? []).filter((line) =>
            input.purchaseOrderLineIds.includes(line.id),
          ),
        }),
      ),
  ),
});