fix: space overview favorites (#2110)

apps/client/src/features/favorite/queries/favorite-query.ts

@@ -14,11 +14,11 @@ import {
 } from "../services/favorite-service";
 import { FavoriteType } from "../types/favorite.types";
 
-export function useFavoritesQuery(type?: FavoriteType) {
+export function useFavoritesQuery(type?: FavoriteType, spaceId?: string) {
   return useInfiniteQuery({
-    queryKey: ["favorites", type],
+    queryKey: ["favorites", type, spaceId],
     queryFn: ({ pageParam }) =>
-      getFavorites({ type, cursor: pageParam, limit: 15 }),
+      getFavorites({ type, spaceId, cursor: pageParam, limit: 15 }),
     initialPageParam: undefined as string | undefined,
     getNextPageParam: (lastPage) =>
       lastPage.meta.hasNextPage ? lastPage.meta.nextCursor : undefined,
@@ -26,10 +26,10 @@ export function useFavoritesQuery(type?: FavoriteType) {
   });
 }
 
-export function useFavoriteIds(type: FavoriteType): Set<string> {
+export function useFavoriteIds(type: FavoriteType, spaceId?: string): Set<string> {
   const { data } = useQuery({
-    queryKey: ["favorite-ids", type],
-    queryFn: () => getFavoriteIds(type),
+    queryKey: ["favorite-ids", type, spaceId],
+    queryFn: () => getFavoriteIds(type, spaceId),
     refetchOnMount: true,
   });
 
@@ -52,9 +52,9 @@ export function useAddFavoriteMutation() {
     onSuccess: (_result, variables) => {
       const entityId = getEntityId(variables);
       if (entityId) {
-        queryClient.setQueryData(
-          ["favorite-ids", variables.type],
-          (old: { items: string[]; meta: any } | undefined) => {
+        queryClient.setQueriesData<{ items: string[]; meta: any }>(
+          { queryKey: ["favorite-ids", variables.type] },
+          (old) => {
             if (!old) return old;
             if (old.items.includes(entityId)) return old;
             return { ...old, items: [...old.items, entityId] };
@@ -76,9 +76,9 @@ export function useRemoveFavoriteMutation() {
     onSuccess: (_result, variables) => {
       const entityId = getEntityId(variables);
       if (entityId) {
-        queryClient.setQueryData(
-          ["favorite-ids", variables.type],
-          (old: { items: string[]; meta: any } | undefined) => {
+        queryClient.setQueriesData<{ items: string[]; meta: any }>(
+          { queryKey: ["favorite-ids", variables.type] },
+          (old) => {
             if (!old) return old;
             return { ...old, items: old.items.filter((id) => id !== entityId) };
           },

apps/client/src/features/favorite/services/favorite-service.ts

@@ -21,13 +21,14 @@ export async function removeFavorite(
   await api.post("/favorites/remove", params);
 }
 
-export async function getFavoriteIds(type: FavoriteType): Promise<IPagination<string>> {
-  const req = await api.post<IPagination<string>>("/favorites/ids", { type });
+export async function getFavoriteIds(type: FavoriteType, spaceId?: string): Promise<IPagination<string>> {
+  const req = await api.post<IPagination<string>>("/favorites/ids", { type, spaceId });
   return req.data;
 }
 
 export async function getFavorites(params?: {
   type?: FavoriteType;
+  spaceId?: string;
   limit?: number;
   cursor?: string;
 }): Promise<IPagination<IFavorite>> {

apps/client/src/features/home/components/favorites-pages.tsx

@@ -18,7 +18,11 @@ import { getSpaceUrl } from "@/lib/config";
 import { useTranslation } from "react-i18next";
 import { getInitialsColor } from "@/lib/get-initials-color";
 
-export default function FavoritesPages() {
+interface Props {
+  spaceId?: string;
+}
+
+export default function FavoritesPages({ spaceId }: Props) {
   const { t } = useTranslation();
   const {
     data,
@@ -27,7 +31,7 @@ export default function FavoritesPages() {
     hasNextPage,
     fetchNextPage,
     isFetchingNextPage,
-  } = useFavoritesQuery("page");
+  } = useFavoritesQuery("page", spaceId);
 
   const favorites = data?.pages.flatMap((p) => p.items) ?? [];
 
@@ -72,19 +76,21 @@ export default function FavoritesPages() {
                       </Group>
                     </UnstyledButton>
                   </Table.Td>
-                  <Table.Td>
-                    {fav.space && (
-                      <Badge
-                        color={getInitialsColor(fav.space.name)}
-                        variant="light"
-                        component={Link}
-                        to={getSpaceUrl(fav.space.slug)}
-                        style={{ cursor: "pointer" }}
-                      >
-                        {fav.space.name}
-                      </Badge>
-                    )}
-                  </Table.Td>
+                  {!spaceId && (
+                    <Table.Td>
+                      {fav.space && (
+                        <Badge
+                          color={getInitialsColor(fav.space.name)}
+                          variant="light"
+                          component={Link}
+                          to={getSpaceUrl(fav.space.slug)}
+                          style={{ cursor: "pointer" }}
+                        >
+                          {fav.space.name}
+                        </Badge>
+                      )}
+                    </Table.Td>
+                  )}
                   <Table.Td>
                     <Text
                       c="dimmed"

apps/client/src/features/page/components/header/page-header-menu.tsx

@@ -145,7 +145,7 @@ function PageActionMenu({ readOnly }: PageActionMenuProps) {
   ] = useDisclosure(false);
   const [pageEditor] = useAtom(pageEditorAtom);
   const pageUpdatedAt = useTimeAgo(page?.updatedAt);
-  const favoriteIds = useFavoriteIds("page");
+  const favoriteIds = useFavoriteIds("page", page?.spaceId);
   const addFavoriteMutation = useAddFavoriteMutation();
   const removeFavoriteMutation = useRemoveFavoriteMutation();
   const isFavorited = page?.id ? favoriteIds.has(page.id) : false;

apps/client/src/features/page/tree/components/space-tree.tsx

@@ -509,7 +509,7 @@ function NodeMenu({ node, treeApi, spaceId }: NodeMenuProps) {
     copyPageModalOpened,
     { open: openCopyPageModal, close: closeCopySpaceModal },
   ] = useDisclosure(false);
-  const favoriteIds = useFavoriteIds("page");
+  const favoriteIds = useFavoriteIds("page", spaceId);
   const addFavorite = useAddFavoriteMutation();
   const removeFavorite = useRemoveFavoriteMutation();
   const isFavorited = favoriteIds.has(node.data.id);

apps/client/src/features/space/components/space-home-tabs.tsx

@@ -47,7 +47,7 @@ export default function SpaceHomeTabs() {
         {space?.id && <RecentChanges spaceId={space.id} />}
       </Tabs.Panel>
       <Tabs.Panel value="favorites">
-        <FavoritesPages />
+        {space?.id && <FavoritesPages spaceId={space.id} />}
       </Tabs.Panel>
       <Tabs.Panel value="created">
         {space?.id && <CreatedByMe spaceId={space.id} />}