GetCategories

import { defineApi } from "@vsf-enterprise/unified-api-magento";
import type { NormalizersTypes } from "@vsf-enterprise/unified-api-magento";
import { type GetCategoriesArgs, getNormalizers } from "@vsf-enterprise/unified-api-magento/udl";

export const getCategories = defineApi.getCategories(async (context, args) => {
  const allCategories = await context.api.categories({});
  const { normalizeCategory } = getNormalizers(context);
  const rootCategories = (allCategories.data?.categories?.items?.[0]?.children ?? []).filter(
    Boolean,
  );
  const filteredData = filterCategories(rootCategories, {
    ids: args?.ids ?? [],
    slugs: args?.slugs ?? [],
  });

  return filteredData.map((category) => normalizeCategory(category));
});

function filterCategories(
  categories: NormalizersTypes.NormalizeCategoryInput[],
  args: Required<Pick<GetCategoriesArgs, "ids" | "slugs">>,
): NormalizersTypes.NormalizeCategoryInput[] {
  const { ids = [], slugs = [] } = args;

  let filteredData = flatCategoryHierarchy(categories);

  if (ids.length > 0 || slugs.length > 0) {
    const idsSet = ids ? new Set(ids) : null;
    const slugsSet = slugs ? new Set(slugs) : null;

    filteredData = filteredData.filter((category) => {
      const idMatches = !idsSet || idsSet.has(category.uid as string);
      const slugMatches = !slugsSet || slugsSet.has(category.url_path as string);

      return idMatches || slugMatches;
    });
  }

  return filteredData;
}

function flatCategoryHierarchy(
  categories: NormalizersTypes.NormalizeCategoryInput[],
  parentCategoryId?: string | undefined,
): NormalizersTypes.NormalizeCategoryInput[] {
  return categories.flatMap((category) => [
    { ...category, parentCategoryId },
    ...(category.children
      ? flatCategoryHierarchy(
          (category.children ?? []) as NormalizersTypes.NormalizeCategoryInput[],
          category.uid,
        )
      : []),
  ]);
}