import type { BusinessPartnerQueries, ItemManagementQueries, OrganizationQueries } from "../module";

export type Company = NonNullable<
  Awaited<ReturnType<OrganizationQueries["getCompany"]>>["value"]["company"]
>;
export type CustomerAccountReference = NonNullable<
  Awaited<ReturnType<BusinessPartnerQueries["getCustomerAccount"]>>["value"]["account"]
>;
export type Item = NonNullable<
  Awaited<ReturnType<ItemManagementQueries["getItem"]>>["value"]["item"]
>;

export const mockOrgQueries = (
  company: Company | null,
): Pick<OrganizationQueries, "getCompany"> => ({
  getCompany: () => Promise.resolve({ ok: true as const, value: { company } }),
});

export const mockBPQueries = (
  account: CustomerAccountReference | null,
): Pick<BusinessPartnerQueries, "getCustomerAccount"> => ({
  getCustomerAccount: () => Promise.resolve({ ok: true as const, value: { account } }),
});

// Item & Record<string, unknown>: requires a full Item row but permits off-model extras
// (companyId) that the command reads defensively.
export const mockItemQueries = (
  items: (Item & Record<string, unknown>)[],
): Pick<ItemManagementQueries, "getItem"> => ({
  getItem: (_db, input) =>
    Promise.resolve({
      ok: true as const,
      value: { item: "id" in input ? (items.find((item) => item.id === input.id) ?? null) : null },
    }),
});