import { ObservationEvent } from "#/types/agent-server/core";
import { ACPToolCallEvent } from "#/types/agent-server/core/events/acp-tool-call-event";

export type ObservationResultStatus = "success" | "error" | "timeout";

/**
 * Map an ACPToolCallEvent's lifecycle + error flags to the same
 * success/error status the rest of the UI uses. A non-terminal call
 * (``pending`` / ``in_progress``) falls through to ``undefined`` so the
 * SuccessIndicator renders nothing — the card shows as "running" via the
 * absence of a check mark, matching how regular ActionEvents are displayed
 * before their ObservationEvent arrives.
 */
export const getACPToolCallResult = (
  event: ACPToolCallEvent,
): ObservationResultStatus | undefined => {
  if (event.is_error || event.status === "failed") return "error";
  if (event.status === "completed") return "success";
  return undefined;
};

export const getObservationResult = (
  event: ObservationEvent,
): ObservationResultStatus => {
  const { observation } = event;
  const observationType = observation.kind;

  switch (observationType) {
    case "ExecuteBashObservation": {
      const exitCode = observation.exit_code;
      const { metadata } = observation;

      if (exitCode === -1 || metadata.exit_code === -1) return "timeout"; // Command timed out
      if (exitCode === 0 || metadata.exit_code === 0) return "success"; // Command executed successfully
      return "error"; // Command failed
    }
    case "TerminalObservation": {
      const exitCode =
        observation.exit_code ?? observation.metadata.exit_code ?? null;

      if (observation.timeout || exitCode === -1) return "timeout";
      if (exitCode === 0) return "success";
      if (observation.is_error) return "error";
      return "success";
    }
    case "FileEditorObservation":
    case "StrReplaceEditorObservation":
      // Check if there's an error
      if (observation.error) return "error";
      return "success";
    case "MCPToolObservation":
      if (observation.is_error) return "error";
      return "success";
    case "SwitchLLMObservation":
      if (observation.is_error) return "error";
      return "success";
    case "InvokeSkillObservation":
      if (observation.is_error) return "error";
      return "success";
    case "TaskObservation":
      if (observation.is_error || observation.status === "failed")
        return "error";
      return "success";
    case "CanvasUIObservation":
      if (observation.is_error) return "error";
      return "success";
    default:
      return "success";
  }
};
