import { commandIds } from "@commands/ids";
import type { AppStore } from "@editor/store";
import { fetchGenerationOptions } from "@platform/comfy/generationClient";
import type { GenerationOptions } from "@editor/state";

export async function loadGenerationResources(store: AppStore): Promise<void> {
  const current = store.getState().editor.generation.resources.status;
  if (current === "loading" || current === "ready") return;
  store.dispatch(commandIds.generationLoadResources, undefined);
  try {
    const options = await fetchGenerationOptions() as GenerationOptions;
    store.dispatch(commandIds.generationSetResources, { options });
  } catch (reason: unknown) {
    store.dispatch(commandIds.generationFailResources, { error: reason instanceof Error ? reason.message : "Unable to load ComfyUI models" });
  }
}