tree-api.ts

import { useEffect, useState } from "react";

import startPollingPipelineStatus from "../pipeline-graph-view/pipeline-graph/main/support/startPollingPipelineStatus.ts";
import { getRunStatusFromPath, RunStatus } from "./RestClient.tsx";
import { mergeStageInfos } from "./utils/stage-merge.ts";

const onPollingError = (err: Error) =>
  console.log("There was an error when polling the pipeline status", err);

/**
 * Polls a run, stopping once the run has completed
 * Optionally retrieves data from the prior run and overlays the new run on top
 */
export default function useRunPoller({
  currentRunPath,
  previousRunPath,
}: RunPollerProps) {
  const [run, setRun] = useState<RunStatus>();
  const [loading, setLoading] = useState(true);

  useEffect(() => {
    let onPipelineDataReceived: (data: RunStatus) => void;
    if (previousRunPath) {
      let previousRun: RunStatus | null = null;
      onPipelineDataReceived = async (current: RunStatus) => {
        setLoading(false);
        if (current.complete) {
          setRun(current);
        } else {
          if (previousRun == null) {
            // only set the previous run if it is not yet set
            previousRun = await getRunStatusFromPath(previousRunPath);
          }
          // error getting previous run
          if (previousRun == null) {
            setRun(current);
          } else {
            setRun({
              stages: mergeStageInfos(previousRun!.stages, current.stages),
              complete: false,
            });
          }
        }
      };
    } else {
      onPipelineDataReceived = (data: RunStatus) => {
        setLoading(false);
        setRun(data);
      };
    }
    startPollingPipelineStatus(
      onPipelineDataReceived,
      onPollingError,
      () => setLoading(false),
      currentRunPath,
    );
  }, [currentRunPath, previousRunPath]);

  return {
    run,
    loading,
  };
}

interface RunPollerProps {
  currentRunPath: string;
  previousRunPath?: string;
}