evOo-local/apps/web/src/hooks/useArtifactBytes.ts

import { useLingui } from "@lingui/react/macro";
import { useEffect, useState } from "react";
import { fetchArtifactBytes } from "../lib/artifact-open";

export type ArtifactBytesState =
  | { status: "loading" }
  | { status: "ready"; bytes: Uint8Array }
  | { status: "error"; message: string };

export function useArtifactBytes(artifactId: string): ArtifactBytesState {
  const { t } = useLingui();
  const [state, setState] = useState<ArtifactBytesState>({ status: "loading" });

  useEffect(() => {
    let cancelled = false;
    setState({ status: "loading" });
    void fetchArtifactBytes(artifactId)
      .then((bytes) => {
        if (!cancelled) setState({ status: "ready", bytes });
      })
      .catch((error) => {
        if (cancelled) return;
        setState({
          status: "error",
          message: error instanceof Error ? error.message : t`Could not load this file.`,
        });
      });
    return () => {
      cancelled = true;
    };
  }, [artifactId, t]);

  return state;
}