import { useGetV2GetLibraryAgent } from "@/app/api/__generated__/endpoints/library/library";
import { useGetV2GetASpecificPreset } from "@/app/api/__generated__/endpoints/presets/presets";
import { useGetV1ListExecutionSchedulesForAGraph } from "@/app/api/__generated__/endpoints/schedules/schedules";
import { GraphExecutionJobInfo } from "@/app/api/__generated__/models/graphExecutionJobInfo";
import { GraphExecutionMeta } from "@/app/api/__generated__/models/graphExecutionMeta";
import { LibraryAgentPreset } from "@/app/api/__generated__/models/libraryAgentPreset";
import { okData } from "@/app/api/helpers";
import { useParams } from "next/navigation";
import { parseAsString, useQueryStates } from "nuqs";
import { useCallback, useEffect, useMemo, useState } from "react";

function parseTab(
  value: string | null,
): "runs" | "scheduled" | "templates" | "triggers" {
  if (
    value === "runs" ||
    value === "scheduled" ||
    value === "templates" ||
    value === "triggers"
  ) {
    return value;
  }
  return "runs";
}

export function useNewAgentLibraryView() {
  const { id } = useParams();
  const agentId = id as string;

  const {
    data: agent,
    isSuccess,
    error,
  } = useGetV2GetLibraryAgent(agentId, { query: { select: okData } });

  const [{ activeItem, activeTab: activeTabRaw }, setQueryStates] =
    useQueryStates({
      activeItem: parseAsString,
      activeTab: parseAsString,
    });

  const activeTab = useMemo(() => parseTab(activeTabRaw), [activeTabRaw]);

  const {
    data: _template,
    isSuccess: isTemplateLoaded,
    isLoading: isTemplateLoading,
    error: templateError,
  } = useGetV2GetASpecificPreset(activeItem ?? "", {
    query: {
      enabled: Boolean(activeTab === "templates" && activeItem),
      select: okData,
    },
  });
  const activeTemplate =
    isTemplateLoaded &&
    activeTab === "templates" &&
    _template?.id === activeItem
      ? _template
      : null;

  useEffect(() => {
    if (!activeTabRaw && !activeItem) {
      setQueryStates({
        activeTab: "runs",
      });
    }
  }, [activeTabRaw, activeItem, setQueryStates]);

  const [sidebarCounts, setSidebarCounts] = useState({
    runsCount: 0,
    schedulesCount: 0,
    templatesCount: 0,
    triggersCount: 0,
  });

  const [sidebarLoading, setSidebarLoading] = useState(true);

  const hasAnyItems = useMemo(
    () =>
      (sidebarCounts.runsCount ?? 0) > 0 ||
      (sidebarCounts.schedulesCount ?? 0) > 0 ||
      (sidebarCounts.templatesCount ?? 0) > 0 ||
      (sidebarCounts.triggersCount ?? 0) > 0,
    [sidebarCounts],
  );

  // Show sidebar layout while loading or when there are items or settings is selected
  const showSidebarLayout =
    sidebarLoading || hasAnyItems || activeItem === "settings";

  useEffect(() => {
    if (agent) {
      document.title = `${agent.name} - Library - AutoGPT Platform`;
    }
  }, [agent]);

  useEffect(() => {
    if (
      activeTab === "triggers" &&
      sidebarCounts.triggersCount === 0 &&
      !sidebarLoading
    ) {
      setQueryStates({
        activeTab: "runs",
      });
    }
  }, [activeTab, sidebarCounts.triggersCount, sidebarLoading, setQueryStates]);

  function handleSelectRun(
    id: string,
    tab?: "runs" | "scheduled" | "templates" | "triggers",
  ) {
    setQueryStates({
      activeItem: id,
      activeTab: tab ?? "runs",
    });
  }

  function handleClearSelectedRun() {
    setQueryStates({
      activeItem: null,
    });
  }

  const { data: schedules } = useGetV1ListExecutionSchedulesForAGraph(
    agent?.graph_id || "",
    {
      query: {
        enabled: !!agent?.graph_id,
        select: okData,
      },
    },
  );

  function handleScheduleDeleted(deletedScheduleId: string) {
    if (activeItem !== deletedScheduleId) {
      return;
    }

    if (!schedules) {
      handleClearSelectedRun();
      return;
    }

    const remainingSchedules = schedules.filter(
      (s) => s.id !== deletedScheduleId,
    );

    if (remainingSchedules.length > 0) {
      handleSelectRun(remainingSchedules[0].id, "scheduled");
    } else {
      handleClearSelectedRun();
    }
  }

  function handleSetActiveTab(
    tab: "runs" | "scheduled" | "templates" | "triggers",
  ) {
    setQueryStates({
      activeTab: tab,
    });
  }

  function handleSelectSettings() {
    setQueryStates({
      activeItem: "settings",
      activeTab: "runs", // Reset to runs tab when going to settings
    });
  }

  const handleCountsChange = useCallback(
    (counts: {
      runsCount: number;
      schedulesCount: number;
      templatesCount: number;
      triggersCount: number;
      loading?: boolean;
    }) => {
      setSidebarCounts({
        runsCount: counts.runsCount,
        schedulesCount: counts.schedulesCount,
        templatesCount: counts.templatesCount,
        triggersCount: counts.triggersCount,
      });
      if (counts.loading !== undefined) {
        setSidebarLoading(counts.loading);
      }
    },
    [],
  );

  function onItemCreated(
    createEvent:
      | { type: "runs"; item: GraphExecutionMeta }
      | { type: "triggers"; item: LibraryAgentPreset }
      | { type: "scheduled"; item: GraphExecutionJobInfo },
  ) {
    if (!hasAnyItems) {
      // Manually increment item count to flip hasAnyItems and showSidebarLayout
      const counts = {
        runsCount: createEvent.type === "runs" ? 1 : 0,
        triggersCount: createEvent.type === "triggers" ? 1 : 0,
        schedulesCount: createEvent.type === "scheduled" ? 1 : 0,
        templatesCount: 0,
      };
      handleCountsChange(counts);
    }
  }

  function onRunInitiated(newRun: GraphExecutionMeta) {
    if (!agent) return;
    onItemCreated({ item: newRun, type: "runs" });
  }

  function onTriggerSetup(newTrigger: LibraryAgentPreset) {
    if (!agent) return;
    onItemCreated({ item: newTrigger, type: "triggers" });
  }

  function onScheduleCreated(newSchedule: GraphExecutionJobInfo) {
    if (!agent) return;
    onItemCreated({ item: newSchedule, type: "scheduled" });
  }

  return {
    agentId: id,
    agent,
    ready: isSuccess,
    activeTemplate,
    isTemplateLoading,
    error: error || templateError,
    hasAnyItems,
    showSidebarLayout,
    activeItem,
    sidebarLoading,
    activeTab,
    setActiveTab: handleSetActiveTab,
    handleClearSelectedRun,
    handleScheduleDeleted,
    handleCountsChange,
    handleSelectRun,
    handleSelectSettings,
    onRunInitiated,
    onTriggerSetup,
    onScheduleCreated,
  };
}
