feat: enhancements (#2107)

* refactor
* fix
* update packages

apps/client/src/ee/ai-chat/components/ai-chat-layout.tsx

@@ -55,7 +55,7 @@ export default function AiChatLayout() {
     navigate(location.pathname, { replace: true, state: null });
   }, [chatId, location, navigate, sendMessage]);
 
-  const hasMessages = messages.length > 0 || isStreaming;
+  const hasMessages = messages.length > 0 || isStreaming || !!chatId;
 
   // While the redirect effect is running (or if the user is still on this
   // component for any reason) never render the chat UI for a forbidden chat.
@@ -65,18 +65,6 @@ export default function AiChatLayout() {
 
   return (
     <div className={classes.main}>
-      {error && (
-        <div
-          style={{
-            padding: "var(--mantine-spacing-sm) var(--mantine-spacing-lg)",
-            color: "var(--mantine-color-red-6)",
-            fontSize: "var(--mantine-font-size-sm)",
-          }}
-        >
-          {error}
-        </div>
-      )}
-
       {hasMessages ? (
         <>
           <ChatMessageList
@@ -85,6 +73,17 @@ export default function AiChatLayout() {
             streamingContent={streamingContent}
             streamingToolCalls={streamingToolCalls}
           />
+          {error && (
+            <div
+              style={{
+                padding: "var(--mantine-spacing-sm) var(--mantine-spacing-lg)",
+                color: "var(--mantine-color-red-6)",
+                fontSize: "var(--mantine-font-size-sm)",
+              }}
+            >
+              {error}
+            </div>
+          )}
           <div className={classes.inputArea}>
             <ChatInput
               isStreaming={isStreaming}

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

@@ -1,3 +1,4 @@
+import { useMemo } from "react";
 import {
   useQuery,
   useInfiniteQuery,
@@ -8,10 +9,10 @@ import {
   addFavorite,
   removeFavorite,
   getFavorites,
+  getFavoriteIds,
   ToggleFavoriteParams,
 } from "../services/favorite-service";
-import { IPagination } from "@/lib/types.ts";
-import { IFavorite, FavoriteType } from "../types/favorite.types";
+import { FavoriteType } from "../types/favorite.types";
 
 export function useFavoritesQuery(type?: FavoriteType) {
   return useInfiniteQuery({
@@ -26,23 +27,21 @@ export function useFavoritesQuery(type?: FavoriteType) {
 }
 
 export function useFavoriteIds(type: FavoriteType): Set<string> {
-  const { data } = useQuery<IPagination<IFavorite>>({
+  const { data } = useQuery({
     queryKey: ["favorite-ids", type],
-    queryFn: () => getFavorites({ type, limit: 50 }),
+    queryFn: () => getFavoriteIds(type),
     refetchOnMount: true,
   });
 
-  const ids = new Set<string>();
-  if (data?.items) {
-    for (const fav of data.items) {
-      let id: string | undefined;
-      if (type === "page") id = fav.pageId;
-      else if (type === "space") id = fav.spaceId;
-      else if (type === "template") id = fav.templateId;
-      if (id) ids.add(id);
-    }
-  }
-  return ids;
+  const items = data?.items;
+  return useMemo(() => new Set(items ?? []), [items]);
+}
+
+function getEntityId(variables: ToggleFavoriteParams): string | undefined {
+  if (variables.type === "page") return variables.pageId;
+  if (variables.type === "space") return variables.spaceId;
+  if (variables.type === "template") return variables.templateId;
+  return undefined;
 }
 
 export function useAddFavoriteMutation() {
@@ -51,9 +50,17 @@ export function useAddFavoriteMutation() {
   return useMutation<void, Error, ToggleFavoriteParams>({
     mutationFn: (data) => addFavorite(data),
     onSuccess: (_result, variables) => {
-      queryClient.invalidateQueries({
-        queryKey: ["favorite-ids", variables.type],
-      });
+      const entityId = getEntityId(variables);
+      if (entityId) {
+        queryClient.setQueryData(
+          ["favorite-ids", variables.type],
+          (old: { items: string[]; meta: any } | undefined) => {
+            if (!old) return old;
+            if (old.items.includes(entityId)) return old;
+            return { ...old, items: [...old.items, entityId] };
+          },
+        );
+      }
       queryClient.invalidateQueries({
         queryKey: ["favorites", variables.type],
       });
@@ -67,9 +74,16 @@ export function useRemoveFavoriteMutation() {
   return useMutation<void, Error, ToggleFavoriteParams>({
     mutationFn: (data) => removeFavorite(data),
     onSuccess: (_result, variables) => {
-      queryClient.invalidateQueries({
-        queryKey: ["favorite-ids", variables.type],
-      });
+      const entityId = getEntityId(variables);
+      if (entityId) {
+        queryClient.setQueryData(
+          ["favorite-ids", variables.type],
+          (old: { items: string[]; meta: any } | undefined) => {
+            if (!old) return old;
+            return { ...old, items: old.items.filter((id) => id !== entityId) };
+          },
+        );
+      }
       queryClient.invalidateQueries({
         queryKey: ["favorites", variables.type],
       });

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

@@ -21,6 +21,11 @@ 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 });
+  return req.data;
+}
+
 export async function getFavorites(params?: {
   type?: FavoriteType;
   limit?: number;

apps/client/src/features/space/components/sidebar/space-sidebar.tsx

@@ -16,6 +16,8 @@ import {
   IconPlus,
   IconSearch,
   IconSettings,
+  IconStar,
+  IconStarFilled,
   IconTrash,
 } from "@tabler/icons-react";
 import {
@@ -43,6 +45,11 @@ import PageImportModal from "@/features/page/components/page-import-modal.tsx";
 import { useTranslation } from "react-i18next";
 import { SwitchSpace } from "./switch-space";
 import ExportModal from "@/components/common/export-modal";
+import {
+  useFavoriteIds,
+  useAddFavoriteMutation,
+  useRemoveFavoriteMutation,
+} from "@/features/favorite/queries/favorite-query";
 import { mobileSidebarAtom } from "@/components/layouts/global/hooks/atoms/sidebar-atom.ts";
 import { useToggleSidebar } from "@/components/layouts/global/hooks/hooks/use-toggle-sidebar.ts";
 import { searchSpotlight } from "@/features/search/constants";
@@ -56,7 +63,6 @@ export function SpaceSidebar() {
   const [mobileSidebarOpened] = useAtom(mobileSidebarAtom);
   const toggleMobileSidebar = useToggleSidebar(mobileSidebarAtom);
 
-
   const { spaceSlug } = useParams();
   const { data: space } = useGetSpaceBySlugQuery(spaceSlug);
 
@@ -82,7 +88,12 @@ export function SpaceSidebar() {
             marginBottom: 3,
           }}
         >
-          <Group gap={4} wrap="nowrap" justify="space-between" style={{ width: "100%" }}>
+          <Group
+            gap={4}
+            wrap="nowrap"
+            justify="space-between"
+            style={{ width: "100%" }}
+          >
             <SwitchSpace
               spaceName={space?.name}
               spaceSlug={space?.slug}
@@ -241,6 +252,20 @@ function SpaceMenu({
   const unwatchMutation = useUnwatchSpaceMutation();
   const isWatching = watchStatus?.watching ?? false;
 
+  const favoriteIds = useFavoriteIds("space");
+  const addFavoriteMutation = useAddFavoriteMutation();
+  const removeFavoriteMutation = useRemoveFavoriteMutation();
+  const isFavorited = favoriteIds.has(spaceId);
+
+  const handleToggleFavorite = () => {
+    const params = { type: "space" as const, spaceId };
+    if (isFavorited) {
+      removeFavoriteMutation.mutate(params);
+    } else {
+      addFavoriteMutation.mutate(params);
+    }
+  };
+
   const handleToggleWatch = () => {
     if (isWatching) {
       unwatchMutation.mutate(spaceId);
@@ -265,6 +290,22 @@ function SpaceMenu({
         </Menu.Target>
 
         <Menu.Dropdown>
+          <Menu.Item
+            onClick={handleToggleFavorite}
+            leftSection={
+              isFavorited ? (
+                <IconStarFilled
+                  size={16}
+                  color="var(--mantine-color-yellow-filled)"
+                />
+              ) : (
+                <IconStar size={16} />
+              )
+            }
+          >
+            {isFavorited ? t("Remove from favorites") : t("Add to favorites")}
+          </Menu.Item>
+
           <Menu.Item
             onClick={handleToggleWatch}
             leftSection={

apps/client/src/features/space/components/spaces-page/all-spaces-list.tsx

@@ -7,9 +7,15 @@ import {
   Space,
   Menu,
   Anchor,
+  Tooltip,
 } from "@mantine/core";
-import { IconDots, IconSettings } from "@tabler/icons-react";
+import { IconDots, IconSettings, IconEye, IconEyeOff } from "@tabler/icons-react";
 import StarButton from "@/features/favorite/components/star-button";
+import {
+  useWatchedSpaceIds,
+  useWatchSpaceMutation,
+  useUnwatchSpaceMutation,
+} from "@/features/space/queries/space-watcher-query";
 import { Link } from "react-router-dom";
 import { useTranslation } from "react-i18next";
 import React, { useState } from "react";
@@ -26,6 +32,45 @@ import { CustomAvatar } from "@/components/ui/custom-avatar.tsx";
 import { AvatarIconType } from "@/features/attachments/types/attachment.types.ts";
 import { AutoTooltipText } from "@/components/ui/auto-tooltip-text.tsx";
 
+function WatchButton({ spaceId, watchedIds, size = 16 }: { spaceId: string; watchedIds: Set<string>; size?: number }) {
+  const { t } = useTranslation();
+  const watchMutation = useWatchSpaceMutation();
+  const unwatchMutation = useUnwatchSpaceMutation();
+  const isWatching = watchedIds.has(spaceId);
+  const isPending = watchMutation.isPending || unwatchMutation.isPending;
+
+  const handleToggle = (e: React.MouseEvent) => {
+    e.stopPropagation();
+    e.preventDefault();
+    if (isWatching) {
+      unwatchMutation.mutate(spaceId);
+    } else {
+      watchMutation.mutate(spaceId);
+    }
+  };
+
+  return (
+    <Tooltip
+      label={isWatching ? t("Stop watching space") : t("Watch space")}
+      openDelay={250}
+      withArrow
+    >
+      <ActionIcon
+        variant="subtle"
+        color={isWatching ? "blue" : "gray"}
+        onClick={handleToggle}
+        loading={isPending}
+      >
+        {isWatching ? (
+          <IconEyeOff size={size} stroke={2} />
+        ) : (
+          <IconEye size={size} stroke={2} />
+        )}
+      </ActionIcon>
+    </Tooltip>
+  );
+}
+
 interface AllSpacesListProps {
   spaces: any[];
   onSearch: (query: string) => void;
@@ -44,6 +89,7 @@ export default function AllSpacesList({
   onPrev,
 }: AllSpacesListProps) {
   const { t } = useTranslation();
+  const watchedIds = useWatchedSpaceIds();
   const [settingsOpened, { open: openSettings, close: closeSettings }] =
     useDisclosure(false);
   const [selectedSpaceId, setSelectedSpaceId] = useState<string | null>(null);
@@ -65,7 +111,7 @@ export default function AllSpacesList({
             <Table.Tr>
               <Table.Th>{t("Space")}</Table.Th>
               <Table.Th>{t("Members")}</Table.Th>
-              <Table.Th w={100}></Table.Th>
+              <Table.Th w={130}></Table.Th>
             </Table.Tr>
           </Table.Thead>
 
@@ -117,8 +163,9 @@ export default function AllSpacesList({
                     </Text>
                   </Table.Td>
                   <Table.Td>
-                    <Group gap="xs" justify="flex-end">
+                    <Group gap="xs" justify="flex-end" wrap="nowrap">
                       <StarButton type="space" spaceId={space.id} size={16} />
+                      <WatchButton spaceId={space.id} watchedIds={watchedIds} size={16} />
                       <Menu position="bottom-end">
                         <Menu.Target>
                           <ActionIcon variant="subtle" color="gray">

apps/client/src/features/space/queries/space-watcher-query.ts

@@ -1,13 +1,27 @@
+import { useMemo } from "react";
 import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
 import {
   watchSpace,
   unwatchSpace,
   getSpaceWatchStatus,
+  getWatchedSpaceIds,
 } from "@/features/space/services/space-watcher-service";
 import { notifications } from "@mantine/notifications";
 import { useTranslation } from "react-i18next";
 
 const SPACE_WATCHER_KEY = "space-watcher";
+const WATCHED_SPACE_IDS_KEY = "watched-space-ids";
+
+export function useWatchedSpaceIds(): Set<string> {
+  const { data } = useQuery({
+    queryKey: [WATCHED_SPACE_IDS_KEY],
+    queryFn: () => getWatchedSpaceIds(),
+    refetchOnMount: true,
+  });
+
+  const items = data?.items;
+  return useMemo(() => new Set(items ?? []), [items]);
+}
 
 export function useSpaceWatchStatusQuery(spaceId: string) {
   return useQuery({
@@ -27,6 +41,14 @@ export function useWatchSpaceMutation() {
       queryClient.setQueryData([SPACE_WATCHER_KEY, spaceId], {
         watching: true,
       });
+      queryClient.setQueryData(
+        [WATCHED_SPACE_IDS_KEY],
+        (old: { items: string[]; meta: any } | undefined) => {
+          if (!old) return old;
+          if (old.items.includes(spaceId)) return old;
+          return { ...old, items: [...old.items, spaceId] };
+        },
+      );
       notifications.show({ message: t("You are now watching this space") });
     },
   });
@@ -41,6 +63,13 @@ export function useUnwatchSpaceMutation() {
       queryClient.setQueryData([SPACE_WATCHER_KEY, spaceId], {
         watching: false,
       });
+      queryClient.setQueryData(
+        [WATCHED_SPACE_IDS_KEY],
+        (old: { items: string[]; meta: any } | undefined) => {
+          if (!old) return old;
+          return { ...old, items: old.items.filter((id) => id !== spaceId) };
+        },
+      );
       notifications.show({
         message: t("You are no longer watching this space"),
       });

apps/client/src/features/space/services/space-watcher-service.ts

@@ -1,4 +1,5 @@
 import api from "@/lib/api-client";
+import { IPagination } from "@/lib/types";
 
 export async function watchSpace(
   spaceId: string,
@@ -18,6 +19,11 @@ export async function unwatchSpace(
   return req.data;
 }
 
+export async function getWatchedSpaceIds(): Promise<IPagination<string>> {
+  const req = await api.post<IPagination<string>>("/spaces/watched-ids");
+  return req.data;
+}
+
 export async function getSpaceWatchStatus(
   spaceId: string,
 ): Promise<{ watching: boolean }> {