linkwarden/apps/web/components/CollectionListing.tsx

import React, { useEffect, useMemo, useState } from "react";
import Tree, {
  mutateTree,
  moveItemOnTree,
  RenderItemParams,
  TreeItem,
  TreeData,
  ItemId,
  TreeSourcePosition,
  TreeDestinationPosition,
} from "@atlaskit/tree";
import { Collection } from "@linkwarden/prisma/client";
import Link from "next/link";
import { CollectionIncludingMembersAndLinkCount } from "@linkwarden/types/global";
import { useRouter } from "next/router";
import toast from "react-hot-toast";
import { useTranslation } from "next-i18next";
import {
  useCollections,
  useUpdateCollection,
} from "@linkwarden/router/collections";
import { useUpdateUser, useUser } from "@linkwarden/router/user";
import Icon from "./Icon";
import { IconWeight } from "@phosphor-icons/react";
import Droppable from "./Droppable";
import { cn } from "@linkwarden/lib/utils";
import { Active, useDndContext } from "@dnd-kit/core";

interface ExtendedTreeItem extends TreeItem {
  data: Collection;
}

const CollectionListing = () => {
  const { active: droppableActive } = useDndContext();
  const { t } = useTranslation();
  const updateCollection = useUpdateCollection();
  const { data: collections = [], isLoading } = useCollections();

  const { data: user } = useUser();
  const updateUser = useUpdateUser();

  const router = useRouter();

  const [tree, setTree] = useState<TreeData | undefined>();

  const initialTree = useMemo(() => {
    if (collections.length > 0) {
      return buildTreeFromCollections(
        collections,
        router,
        tree,
        user?.collectionOrder
      );
    } else return undefined;
  }, [collections, user]);

  useEffect(() => {
    setTree(initialTree);
  }, [initialTree]);

  useEffect(() => {
    if (user?.username) {
      // refetch();
      if (
        (!user.collectionOrder || user.collectionOrder.length === 0) &&
        collections.length > 0
      )
        updateUser.mutate({
          ...user,
          collectionOrder: collections
            .filter((e) => e.parentId === null)
            .map((e) => e.id as number),
        });
      else {
        const newCollectionOrder: number[] = [...(user.collectionOrder || [])];

        // Start with collections that are in both account.collectionOrder and collections
        const existingCollectionIds = collections.map((c) => c.id as number);
        const filteredCollectionOrder = user.collectionOrder.filter((id: any) =>
          existingCollectionIds.includes(id)
        );

        // Add new collections that are not in account.collectionOrder and meet the specific conditions
        collections.forEach((collection) => {
          if (
            !filteredCollectionOrder.includes(collection.id as number) &&
            (!collection.parentId || collection.ownerId === user.id)
          ) {
            filteredCollectionOrder.push(collection.id as number);
          }
        });

        // check if the newCollectionOrder is the same as the old one
        if (
          JSON.stringify(newCollectionOrder) !==
          JSON.stringify(user.collectionOrder)
        ) {
          updateUser.mutateAsync({
            ...user,
            collectionOrder: newCollectionOrder,
          });
        }
      }
    }
  }, [user, collections]);

  const onExpand = (movedCollectionId: ItemId) => {
    setTree((currentTree) =>
      mutateTree(currentTree!, movedCollectionId, { isExpanded: true })
    );
  };

  const onCollapse = (movedCollectionId: ItemId) => {
    setTree((currentTree) =>
      mutateTree(currentTree as TreeData, movedCollectionId, {
        isExpanded: false,
      })
    );
  };

  function reorderTreeItems(
    tree: TreeData,
    movedCollectionId: ItemId,
    source: TreeSourcePosition,
    destination: TreeDestinationPosition
  ) {
    // Same parent reordering
    if (source.parentId === destination.parentId) {
      const parent = tree.items[source.parentId];
      const children = [...parent.children];

      // Remove from source index
      children.splice(source.index, 1);
      // Insert at destination index
      if (destination.index !== undefined) {
        children.splice(destination.index, 0, movedCollectionId);
      }

      parent.children = children;
      return tree;
    }

    // Different parent move
    const sourceParent = tree.items[source.parentId];
    const destinationParent = tree.items[destination.parentId];

    // Remove from source parent
    sourceParent.children = sourceParent.children.filter(
      (id) => id !== movedCollectionId
    );

    // Initialize children array if it doesn't exist
    if (!destinationParent.children) {
      destinationParent.children = [];
    }

    // If destination index is not specified, add to the end
    const destinationIndex =
      destination.index !== undefined
        ? destination.index
        : destinationParent.children.length;

    // Add to destination parent
    destinationParent.children.splice(destinationIndex, 0, movedCollectionId);

    // Update destination parent properties
    destinationParent.hasChildren = true;
    destinationParent.isExpanded = true;

    // Update the moved item's parent ID
    tree.items[movedCollectionId].data.parentId = destination.parentId;

    return tree;
  }

  function flattenTreeIds(
    tree: TreeData,
    nodeId: ItemId = "root",
    result: Array<ItemId> = []
  ) {
    const node = tree.items[nodeId];

    if (nodeId !== "root") {
      result.push(node.id);
    }

    if (node.children && node.children.length > 0) {
      node.children.forEach((childId) => {
        flattenTreeIds(tree, childId, result);
      });
    }

    return result;
  }

  const onDragEnd = async (
    source: TreeSourcePosition,
    destination: TreeDestinationPosition | undefined
  ) => {
    if (!destination || !tree) {
      return;
    }

    if (
      source.index === destination.index &&
      source.parentId === destination.parentId
    ) {
      return;
    }

    const movedCollectionId = Number(
      tree.items[source.parentId].children[source.index]
    );

    const movedCollection = collections.find((c) => c.id === movedCollectionId);

    const destinationCollection = collections.find(
      (c) => c.id === Number(destination.parentId)
    );

    if (
      (movedCollection?.ownerId !== user?.id &&
        destination.parentId !== source.parentId) ||
      (destinationCollection?.ownerId !== user?.id &&
        destination.parentId !== "root")
    ) {
      return toast.error(t("cant_change_collection_you_dont_own"));
    }

    setTree((currentTree) => moveItemOnTree(currentTree!, source, destination));

    const newTree = reorderTreeItems(
      tree,
      movedCollectionId,
      source,
      destination
    );

    if (source.parentId !== destination.parentId) {
      await updateCollection.mutateAsync(
        {
          ...movedCollection,
          parentId:
            destination.parentId && destination.parentId !== "root"
              ? Number(destination.parentId)
              : destination.parentId === "root"
                ? "root"
                : null,
        },
        {
          onError: (error) => {
            toast.error(error.message);
          },
        }
      );
    }

    await updateUser.mutateAsync({
      ...user,
      collectionOrder: flattenTreeIds(newTree),
    });
  };

  if (isLoading) {
    return (
      <div className="flex flex-col gap-4">
        <div className="skeleton h-4 w-full"></div>
        <div className="skeleton h-4 w-full"></div>
        <div className="skeleton h-4 w-full"></div>
      </div>
    );
  } else if (!tree) {
    return (
      <p className="text-neutral text-xs font-semibold truncate w-full px-2 mt-5 mb-8">
        {t("you_have_no_collections")}
      </p>
    );
  } else
    return (
      <Tree
        tree={tree}
        renderItem={(itemProps) =>
          renderItem({ ...itemProps }, router.asPath, droppableActive)
        }
        onExpand={onExpand}
        onCollapse={onCollapse}
        onDragEnd={onDragEnd}
        isDragEnabled
        isNestingEnabled
      />
    );
};

export default CollectionListing;

const renderItem = (
  { item, onExpand, onCollapse, provided }: RenderItemParams,
  currentPath: string,
  droppableActive: Active | null
) => {
  const collection = item.data;

  return (
    <Droppable
      id={`side-bar-collection-${collection.id}`}
      data={{
        name: collection.name,
        id: collection.id,
        ownerId: collection.ownerId,
      }}
      className="group"
    >
      <div
        ref={provided.innerRef}
        {...provided.draggableProps}
        className="mb-1"
      >
        <div
          className={cn(
            currentPath === `/collections/${collection.id}`
              ? "bg-primary/20 is-active"
              : droppableActive
                ? "select-none"
                : "hover:bg-neutral/20",
            "duration-100 flex gap-1 items-center pr-2 pl-1 rounded-md"
          )}
        >
          {Dropdown(item as ExtendedTreeItem, onExpand, onCollapse)}

          <Link
            href={`/collections/${collection.id}`}
            className="w-full"
            {...provided.dragHandleProps}
          >
            <div
              className={`py-1 cursor-pointer flex items-center gap-2 w-full rounded-md h-8`}
            >
              {collection.icon ? (
                <Icon
                  icon={collection.icon}
                  size={30}
                  weight={(collection.iconWeight || "regular") as IconWeight}
                  color={collection.color}
                  className="-mr-[0.15rem]"
                />
              ) : (
                <i
                  className="bi-folder-fill text-xl"
                  style={{ color: collection.color }}
                ></i>
              )}

              <p className="truncate w-full">{collection.name}</p>

              {collection.isPublic && (
                <i
                  className="bi-globe2 text-sm text-black/50 dark:text-white/50 drop-shadow"
                  title="This collection is being shared publicly."
                ></i>
              )}
              <div className="drop-shadow text-neutral text-xs">
                {collection._count?.links}
              </div>
            </div>
          </Link>
        </div>
      </div>
    </Droppable>
  );
};

const Dropdown = (
  item: ExtendedTreeItem,
  onExpand: (id: ItemId) => void,
  onCollapse: (id: ItemId) => void
) => {
  if (item.children && item.children.length > 0) {
    return item.isExpanded ? (
      <button onClick={() => onCollapse(item.id)}>
        <div className="bi-caret-down-fill opacity-50 hover:opacity-100 duration-200"></div>
      </button>
    ) : (
      <button onClick={() => onExpand(item.id)}>
        <div className="bi-caret-right-fill opacity-40 hover:opacity-100 duration-200"></div>
      </button>
    );
  }
  // return <span>&bull;</span>;
  return <div></div>;
};

const buildTreeFromCollections = (
  collections: CollectionIncludingMembersAndLinkCount[],
  router: ReturnType<typeof useRouter>,
  tree?: TreeData,
  order?: number[]
): TreeData => {
  if (order) {
    collections.sort((a: any, b: any) => {
      return order.indexOf(a.id) - order.indexOf(b.id);
    });
  }

  function getTotalLinkCount(collectionId: number): number {
    const collection = items[collectionId];
    if (!collection) {
      return 0;
    }

    let totalLinkCount = (collection.data as any)._count?.links || 0;

    if (collection.hasChildren) {
      collection.children.forEach((childId) => {
        totalLinkCount += getTotalLinkCount(childId as number);
      });
    }

    return totalLinkCount;
  }

  const items: { [key: string]: ExtendedTreeItem } = collections.reduce(
    (acc: any, collection) => {
      acc[collection.id as number] = {
        id: collection.id,
        children: [],
        hasChildren: false,
        isExpanded: tree?.items[collection.id as number]?.isExpanded || false,
        data: {
          id: collection.id,
          parentId: collection.parentId,
          name: collection.name,
          description: collection.description,
          color: collection.color,
          icon: collection.icon,
          iconWeight: collection.iconWeight,
          isPublic: collection.isPublic,
          ownerId: collection.ownerId,
          createdAt: collection.createdAt,
          updatedAt: collection.updatedAt,
          _count: {
            links: collection._count?.links,
          },
        },
      };
      return acc;
    },
    {}
  );

  const activeCollectionId = Number(router.asPath.split("/collections/")[1]);

  if (activeCollectionId) {
    for (const item in items) {
      const collection = items[item];
      if (Number(item) === activeCollectionId && collection.data.parentId) {
        // get all the parents of the active collection recursively until root and set isExpanded to true
        let parentId = collection.data.parentId || null;
        while (parentId && items[parentId]) {
          items[parentId].isExpanded = true;
          parentId = items[parentId].data.parentId;
        }
      }
    }
  }

  collections.forEach((collection) => {
    const parentId = collection.parentId;
    if (parentId && items[parentId] && collection.id) {
      items[parentId].children.push(collection.id);
      items[parentId].hasChildren = true;
    }
  });

  collections.forEach((collection) => {
    const collectionId = collection.id;
    if (items[collectionId as number] && collection.id) {
      const linkCount = getTotalLinkCount(collectionId as number);
      (items[collectionId as number].data as any)._count.links = linkCount;
    }
  });

  const rootId = "root";
  items[rootId] = {
    id: rootId,
    children: (collections
      .filter(
        (c) =>
          c.parentId === null || !collections.find((i) => i.id === c.parentId)
      )
      .map((c) => c.id) || "") as unknown as string[],
    hasChildren: true,
    isExpanded: true,
    data: { name: "Root" } as Collection,
  };

  return { rootId, items };
};