import { Budget, newBudgetFromDto } from "./budget";
import { Category } from "./category";

type BudgetMap<T extends { id: any }> = {
  [id in T["id"]]: Budget | null;
};

function newCategoryBudgetMapFromDto(dto: unknown): BudgetMap<Category> {
  return Object.fromEntries(
    (dto as { category_id: number; budget: unknown }[]).map(
      ({ category_id, budget }) => [category_id, newBudgetFromDto(budget)],
    ),
  );
}

export { type BudgetMap, newCategoryBudgetMapFromDto as newBudgetMapFromDto };