All files / src/pages/main-page/project-folders useProjectFolderWorkspace.ts

78.57% Statements 33/42
83.33% Branches 35/42
100% Functions 2/2
80.48% Lines 33/41

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151                                          585x 585x 585x 585x 585x 585x 585x             585x             585x 585x   585x         585x 190x       190x 22x   190x   585x 585x     585x     585x           5x 5x   1x   4x   5x       4x     4x       2x   2x 2x           1x       1x                                                   585x                                          
import { useMemo, useRef, useState } from "react";
import { useSearchParams } from "react-router-dom";
import { toast } from "sonner";
import {
  useCoreProjectFoldersBinCreateMutation,
  useCoreProjectFoldersCreateMutation,
  useCoreProjectFoldersPartialUpdateMutation,
  useCoreProjectFoldersWorkspaceRetrieveQuery,
  useCoreProjectsPartialUpdateMutation,
} from "@/api/apiStore.gen";
import {
  buildProjectFolderIndex,
  buildProjectFolderItems,
  buildProjectFolderPathItems,
  type ProjectFolderId,
  parseProjectFolderId,
  ROOT_PROJECT_FOLDER_ID,
} from "@/pages/main-page/project-folders/model";
 
/** Coordinate folder API state, URL navigation, and user-triggered operations. */
export function useProjectFolderWorkspace({ enabled }: { enabled: boolean }) {
  const [searchParams, setSearchParams] = useSearchParams();
  const [createProjectFolder] = useCoreProjectFoldersCreateMutation();
  const [updateProjectFolder] = useCoreProjectFoldersPartialUpdateMutation();
  const [binProjectFolder] = useCoreProjectFoldersBinCreateMutation();
  const [updateProject] = useCoreProjectsPartialUpdateMutation();
  const [requestedFolderPage, setRequestedFolderPage] = useState(1);
  const requestedFolder = searchParams.get("folder");
  const {
    currentData: folderResponse,
    isLoading,
    isFetching,
    isError,
    refetch,
  } = useCoreProjectFoldersWorkspaceRetrieveQuery(
    {
      parent: requestedFolder ?? "root",
      page: requestedFolderPage,
    },
    { skip: !enabled },
  );
  const requestedFolderId = parseProjectFolderId(requestedFolder);
  const isMissing = folderResponse?.requested_folder_missing ?? false;
  const currentFolderId =
    isError || isMissing
      ? ROOT_PROJECT_FOLDER_ID
      : (folderResponse?.current_folder?.id ??
        requestedFolderId ??
        ROOT_PROJECT_FOLDER_ID);
  const folderIndex = useMemo(() => {
    const folderItems = [
      ...buildProjectFolderPathItems(folderResponse?.breadcrumbs ?? []),
      ...buildProjectFolderItems(folderResponse?.folders.results ?? []),
    ];
    const uniqueItems = Array.from(
      new Map(folderItems.map((folder) => [folder.id, folder])).values(),
    );
    return buildProjectFolderIndex(enabled && !isError ? uniqueItems : []);
  }, [enabled, folderResponse, isError]);
  const pendingProjectMovesRef = useRef(new Set<number>());
  const [pendingProjectMoveIds, setPendingProjectMoveIds] = useState<
    ReadonlySet<number>
  >(new Set());
  const currentFolders = buildProjectFolderItems(
    isError ? [] : (folderResponse?.folders.results ?? []),
  );
  const folderPage = Math.min(
    requestedFolderPage,
    folderResponse?.folders.pages ?? 1,
  );
 
  const navigateToFolder = (folderId: ProjectFolderId) => {
    setRequestedFolderPage(1);
    const nextSearchParams = new URLSearchParams(searchParams);
    if (folderId === ROOT_PROJECT_FOLDER_ID) {
      nextSearchParams.delete("folder");
    } else {
      nextSearchParams.set("folder", folderId.toString());
    }
    setSearchParams(nextSearchParams);
  };
 
  const createFolder = async (parentId: ProjectFolderId, name: string) => {
    await createProjectFolder({
      projectFolder: { parent: parentId, name },
    }).unwrap();
    toast.success(`${name} created.`);
  };
 
  const binFolder = async (folderId: number) => {
    const result = await binProjectFolder({ id: folderId }).unwrap();
    const projectLabel =
      result.binned_project_count === 1 ? "project" : "projects";
    toast.success(
      `Folder moved to Bin with ${result.binned_project_count} ${projectLabel}.`,
    );
  };
 
  const renameFolder = async (folderId: number, name: string) => {
    await updateProjectFolder({
      id: folderId.toString(),
      patchedProjectFolder: { name },
    }).unwrap();
    toast.success(`Folder renamed to ${name}.`);
  };
 
  const moveProject = async (projectId: number, folderId: ProjectFolderId) => {
    Iif (pendingProjectMovesRef.current.has(projectId)) return;
 
    pendingProjectMovesRef.current.add(projectId);
    setPendingProjectMoveIds(new Set(pendingProjectMovesRef.current));
    try {
      await updateProject({
        id: projectId.toString(),
        patchedProject: { folder: folderId },
      }).unwrap();
      toast.success(
        folderId === ROOT_PROJECT_FOLDER_ID
          ? "Project moved to My Projects."
          : "Project moved to folder.",
      );
    } catch {
      toast.error("Failed to move project.");
    } finally {
      pendingProjectMovesRef.current.delete(projectId);
      setPendingProjectMoveIds(new Set(pendingProjectMovesRef.current));
    }
  };
 
  return {
    currentFolderId,
    isMissing,
    isUnavailable: isError,
    isLoading: enabled && (isLoading || isFetching),
    canOrganize: enabled && !isLoading && !isFetching && !isError && !isMissing,
    folderIndex,
    currentFolders,
    folderPage,
    folderPageCount: folderResponse?.folders.pages ?? 1,
    folderCount: folderResponse?.folders.count ?? 0,
    setFolderPage: setRequestedFolderPage,
    navigateToFolder,
    createFolder,
    renameFolder,
    binFolder,
    moveProject,
    pendingProjectMoveIds,
    refetchFolders: refetch,
  };
}