PaperClipAI/ui/src/hooks/useTaskListCollection.ts

import { useState } from "react";
import { includeTaskListAncestors, type TaskListItem } from "@/components/TaskList";
import { applyIssueFilters, countActiveIssueFilters, defaultIssueFilterState, type IssueFilterState } from "@/lib/issue-filters";
import { loadTaskListView, matchesTaskListView, saveTaskListView, type TaskListView } from "@/lib/task-list-views";

type CreatorOption = { id: string; label: string; kind: "agent" | "user" };
const expanded: ReadonlySet<string> = new Set();

/** Shared collection behavior for the proposed task UI. Supply accessible data;
 * this hook scopes presentation, and does not replace server authorization. */
export function useTaskListCollection({ items, companyId, userId, viewStorageKey, initialQuery = "", query: controlledQuery, onQueryChange, view: controlledView, layout: controlledLayout, onLayoutChange, initialFilters, searchMode = "local", liveIssueIds }: {
  items: readonly TaskListItem[];
  companyId: string | null;
  userId: string | null;
  viewStorageKey: string;
  initialQuery?: string;
  query?: string;
  onQueryChange?: (query: string) => void;
  view?: TaskListView;
  layout?: "list" | "board";
  onLayoutChange?: (layout: "list" | "board") => void;
  initialFilters?: Partial<IssueFilterState>;
  /** Server results may match comments, documents, or normalized identifiers. */
  searchMode?: "local" | "server";
  liveIssueIds?: ReadonlySet<string>;
}) {
  const [queryState, setQueryState] = useState(initialQuery);
  const query = controlledQuery ?? queryState;
  const [preference, setPreference] = useState(() => ({ key: viewStorageKey, view: loadTaskListView(viewStorageKey) }));
  const view = controlledView ?? (preference.key === viewStorageKey ? preference.view : loadTaskListView(viewStorageKey));
  const [layoutState, setLayoutState] = useState<"list" | "board">("list");
  const layout = controlledLayout ?? layoutState;
  const [filters, setFilters] = useState<IssueFilterState>(() => ({ ...defaultIssueFilterState, ...initialFilters }));
  const [collapse, setCollapse] = useState({ companyId, ids: expanded });
  const collapsedIds = collapse.companyId === companyId ? collapse.ids : expanded;
  const setCollapsedIds = (ids: ReadonlySet<string>) => setCollapse({ companyId, ids });
  const resetExpansion = () => setCollapsedIds(expanded);

  const companyItems = items.filter(item => item.issue.companyId === companyId);
  const creatorsById = new Map<string, CreatorOption>();
  for (const item of companyItems) {
    const { createdByAgentId, createdByUserId } = item.issue;
    const rawId = createdByAgentId ?? createdByUserId;
    if (!rawId) continue;
    const kind = createdByAgentId ? "agent" : "user";
    const id = `${kind}:${rawId}`;
    creatorsById.set(id, { id, kind, label: item.createdBy || rawId });
  }
  const filteredIds = new Set(applyIssueFilters(companyItems.map(item => item.issue), filters, userId, false, liveIssueIds).map(issue => issue.id));
  const normalizedQuery = query.trim().toLocaleLowerCase();
  const matchingItems = companyItems.filter(item => filteredIds.has(item.issue.id)
    && matchesTaskListView(view, item.issue, Boolean(item.unread), userId)
    && (searchMode === "server" || [item.issue.title, item.issue.identifier, item.issue.description, item.createdBy, item.assignee, item.issue.project?.name,
      ...(item.issue.labels ?? []).map(label => label.name)].some(value => value?.toLocaleLowerCase().includes(normalizedQuery))));
  const visibleItems = includeTaskListAncestors(companyItems, matchingItems);
  const parentIds = new Set(companyItems.flatMap(item => item.issue.parentId ? [item.issue.parentId] : []));
  const allExpanded = ![...parentIds].some(id => collapsedIds.has(id));
  const activeFilterCount = countActiveIssueFilters(filters);

  const setQuery = (next: string) => { setQueryState(next); onQueryChange?.(next); resetExpansion(); };
  const setView = (next: TaskListView) => {
    setPreference({ key: viewStorageKey, view: next });
    saveTaskListView(viewStorageKey, next);
    resetExpansion();
  };
  const updateFilters = (patch: Partial<IssueFilterState>) => {
    setFilters(current => ({ ...current, ...patch }));
    resetExpansion();
  };
  const clear = () => { setQuery(""); setFilters(defaultIssueFilterState); setView("all"); };

  return {
    query, setQuery, view, setView, layout, setLayout: onLayoutChange ?? setLayoutState, filters, updateFilters, clear,
    companyItems, matchingItems, visibleItems, creators: [...creatorsById.values()],
    collapsedIds, setCollapsedIds, allExpanded,
    toggleExpanded: () => setCollapsedIds(allExpanded ? parentIds : expanded),
    activeFilterCount, narrowed: Boolean(normalizedQuery || activeFilterCount || view !== "all"),
  };
}