@btst/stack 3.0.0-rc.2 → 3.0.0-rc.4

package/dist/packages/stack/src/plugins/comments/client/hooks/use-comments.mjs

@@ -1,24 +1,40 @@
 "use client";
-import { useQuery, useSuspenseQuery, useInfiniteQuery, useQueryClient, useMutation } from '@tanstack/react-query';
+import { useQuery, useSuspenseQuery, useInfiniteQuery, useMutation, useQueryClient } from '@tanstack/react-query';
 import { runResourceMutation, createApiClient } from '@btst/stack/plugins/client';
-import { createCommentsQueryKeys, commentsResources } from '../../../../../../../plugins/comments/query-keys.mjs';
+import { useStack } from '@btst/stack/context';
+import { commentsResources, createCommentsQueryKeys } from '../../../../../../../plugins/comments/query-keys.mjs';
+import { COMMENTS_PLUGIN_ID } from '../constants.mjs';
 
 const commentMutations = commentsResources.comments.mutations;
-function getClient(config) {
-  return createApiClient({
-    baseURL: config.apiBaseURL,
-    basePath: config.apiBasePath
+function useCommentsRuntime() {
+  const { api, plugins, queryClient: stackQueryClient } = useStack();
+  const pluginApi = plugins?.[COMMENTS_PLUGIN_ID]?.api;
+  const headers = pluginApi?.browserHeaders;
+  const client = createApiClient({
+    baseURL: pluginApi?.baseURL ?? api?.baseURL ?? "",
+    basePath: pluginApi?.basePath ?? api?.basePath ?? "",
+    headers,
+    credentials: pluginApi?.credentials
   });
+  const queryClient = useQueryClient(stackQueryClient);
+  return {
+    client,
+    headers,
+    queryClient,
+    queries: createCommentsQueryKeys(client)
+  };
 }
-function useComments(config, params, options) {
-  const client = getClient(config);
-  const queries = createCommentsQueryKeys(client, config.headers);
-  const query = useQuery({
-    ...queries.comments.list(params),
-    staleTime: 3e4,
-    retry: false,
-    enabled: options?.enabled ?? true
-  });
+function useComments(params, options) {
+  const { queries, queryClient } = useCommentsRuntime();
+  const query = useQuery(
+    {
+      ...queries.comments.list(params),
+      staleTime: 3e4,
+      retry: false,
+      enabled: options?.enabled ?? true
+    },
+    queryClient
+  );
   return {
     data: query.data,
     comments: query.data?.items ?? [],
@@ -29,14 +45,16 @@ function useComments(config, params, options) {
     refetch: query.refetch
   };
 }
-function useSuspenseComments(config, params) {
-  const client = getClient(config);
-  const queries = createCommentsQueryKeys(client, config.headers);
-  const { data, refetch, error, isFetching } = useSuspenseQuery({
-    ...queries.comments.list(params),
-    staleTime: 3e4,
-    retry: false
-  });
+function useSuspenseComments(params) {
+  const { queries, queryClient } = useCommentsRuntime();
+  const { data, refetch, error, isFetching } = useSuspenseQuery(
+    {
+      ...queries.comments.list(params),
+      staleTime: 3e4,
+      retry: false
+    },
+    queryClient
+  );
   if (error && !isFetching) {
     throw error;
   }
@@ -46,17 +64,19 @@ function useSuspenseComments(config, params) {
     refetch
   };
 }
-function useSuspenseModerationComments(config, params) {
+function useSuspenseModerationComments(params) {
   const limit = params.limit ?? 20;
   const page = params.page ?? 1;
   const offset = (page - 1) * limit;
-  const client = getClient(config);
-  const queries = createCommentsQueryKeys(client, config.headers);
-  const { data, refetch, error, isFetching } = useSuspenseQuery({
-    ...queries.comments.list({ status: params.status, limit, offset }),
-    staleTime: 3e4,
-    retry: false
-  });
+  const { queries, queryClient } = useCommentsRuntime();
+  const { data, refetch, error, isFetching } = useSuspenseQuery(
+    {
+      ...queries.comments.list({ status: params.status, limit, offset }),
+      staleTime: 3e4,
+      retry: false
+    },
+    queryClient
+  );
   if (error && !isFetching) {
     throw error;
   }
@@ -72,10 +92,9 @@ function useSuspenseModerationComments(config, params) {
     refetch
   };
 }
-function useInfiniteComments(config, params, options) {
+function useInfiniteComments(params, options) {
   const pageSize = params.pageSize ?? 10;
-  const client = getClient(config);
-  const queries = createCommentsQueryKeys(client, config.headers);
+  const { queries, queryClient } = useCommentsRuntime();
   const baseQuery = queries.commentsThread.list({
     resourceId: params.resourceId,
     resourceType: params.resourceType,
@@ -85,17 +104,20 @@ function useInfiniteComments(config, params, options) {
     sort: params.sort,
     limit: pageSize
   });
-  const query = useInfiniteQuery({
-    ...baseQuery,
-    initialPageParam: 0,
-    getNextPageParam: (lastPage) => {
-      const nextOffset = lastPage.offset + lastPage.limit;
-      return nextOffset < lastPage.total ? nextOffset : void 0;
+  const query = useInfiniteQuery(
+    {
+      ...baseQuery,
+      initialPageParam: 0,
+      getNextPageParam: (lastPage) => {
+        const nextOffset = lastPage.offset + lastPage.limit;
+        return nextOffset < lastPage.total ? nextOffset : void 0;
+      },
+      staleTime: 3e4,
+      retry: false,
+      enabled: options?.enabled ?? true
     },
-    staleTime: 3e4,
-    retry: false,
-    enabled: options?.enabled ?? true
-  });
+    queryClient
+  );
   const comments = query.data?.pages.flatMap((p) => p.items) ?? [];
   const total = query.data?.pages[0]?.total ?? 0;
   return {
@@ -110,24 +132,24 @@ function useInfiniteComments(config, params, options) {
     error: query.error
   };
 }
-function useCommentCount(config, params) {
-  const client = getClient(config);
-  const queries = createCommentsQueryKeys(client, config.headers);
-  const query = useQuery({
-    ...queries.commentCount.byResource(params),
-    staleTime: 6e4,
-    retry: false
-  });
+function useCommentCount(params) {
+  const { queries, queryClient } = useCommentsRuntime();
+  const query = useQuery(
+    {
+      ...queries.commentCount.byResource(params),
+      staleTime: 6e4,
+      retry: false
+    },
+    queryClient
+  );
   return {
     count: query.data ?? 0,
     isLoading: query.isLoading,
     error: query.error
   };
 }
-function usePostComment(config, params) {
-  const queryClient = useQueryClient();
-  const client = getClient(config);
-  const queries = createCommentsQueryKeys(client, config.headers);
+function usePostComment(params) {
+  const { client, headers, queries, queryClient } = useCommentsRuntime();
   const getListKey = (parentId, offset, limit) => {
     if (params.infiniteKey && (parentId ?? null) === null) {
       return params.infiniteKey;
@@ -143,232 +165,240 @@ function usePostComment(config, params) {
     }).queryKey;
   };
   const isInfinitePost = (parentId) => !!params.infiniteKey && (parentId ?? null) === null;
-  return useMutation({
-    mutationFn: (input) => runResourceMutation(
-      client,
-      commentMutations.create,
-      {
-        resourceId: params.resourceId,
-        resourceType: params.resourceType,
-        parentId: input.parentId ?? null,
-        body: input.body
-      },
-      config.headers
-    ),
-    onMutate: async (input) => {
-      const listKey = getListKey(input.parentId, input.offset, input.limit);
-      await queryClient.cancelQueries({ queryKey: listKey });
-      const optimisticId = `optimistic-${Date.now()}`;
-      const optimistic = {
-        id: optimisticId,
-        resourceId: params.resourceId,
-        resourceType: params.resourceType,
-        parentId: input.parentId ?? null,
-        authorId: params.currentUserId ?? "",
-        resolvedAuthorName: "You",
-        resolvedAvatarUrl: null,
-        body: input.body,
-        status: "pending",
-        likes: 0,
-        isLikedByCurrentUser: false,
-        editedAt: null,
-        createdAt: (/* @__PURE__ */ new Date()).toISOString(),
-        updatedAt: (/* @__PURE__ */ new Date()).toISOString(),
-        replyCount: 0
-      };
-      if (isInfinitePost(input.parentId)) {
-        const previous2 = queryClient.getQueryData(listKey);
-        const isDesc = params.sort === "desc";
-        queryClient.setQueryData(
-          listKey,
-          (old) => {
-            if (!old) {
-              return {
-                pages: [
-                  {
-                    items: [optimistic],
-                    total: 1,
-                    limit: params.pageSize ?? 10,
-                    offset: 0
-                  }
-                ],
-                pageParams: [0]
-              };
-            }
-            const targetIdx = isDesc ? 0 : old.pages.length - 1;
-            return {
-              ...old,
-              // Increment `total` on every page so the header count (which reads
-              // pages[0].total) stays in sync even after multiple pages are loaded.
-              pages: old.pages.map(
-                (page, idx) => idx === targetIdx ? {
-                  ...page,
-                  items: isDesc ? [optimistic, ...page.items] : [...page.items, optimistic],
-                  total: page.total + 1
-                } : { ...page, total: page.total + 1 }
-              )
-            };
-          }
-        );
-        return { previous: previous2, isInfinite: true, listKey, optimisticId };
-      }
-      const previous = queryClient.getQueryData(listKey);
-      queryClient.setQueryData(listKey, (old) => {
-        if (!old) {
-          return { items: [optimistic], total: 1, limit: 20, offset: 0 };
-        }
-        return {
-          ...old,
-          items: [...old.items, optimistic],
-          total: old.total + 1
+  return useMutation(
+    {
+      mutationFn: (input) => runResourceMutation(
+        client,
+        commentMutations.create,
+        {
+          resourceId: params.resourceId,
+          resourceType: params.resourceType,
+          parentId: input.parentId ?? null,
+          body: input.body
+        },
+        headers
+      ),
+      onMutate: async (input) => {
+        const listKey = getListKey(input.parentId, input.offset, input.limit);
+        await queryClient.cancelQueries({ queryKey: listKey });
+        const optimisticId = `optimistic-${Date.now()}`;
+        const optimistic = {
+          id: optimisticId,
+          resourceId: params.resourceId,
+          resourceType: params.resourceType,
+          parentId: input.parentId ?? null,
+          authorId: params.currentUserId ?? "",
+          resolvedAuthorName: "You",
+          resolvedAvatarUrl: null,
+          body: input.body,
+          status: "pending",
+          likes: 0,
+          isLikedByCurrentUser: false,
+          editedAt: null,
+          createdAt: (/* @__PURE__ */ new Date()).toISOString(),
+          updatedAt: (/* @__PURE__ */ new Date()).toISOString(),
+          replyCount: 0
         };
-      });
-      return { previous, isInfinite: false, listKey, optimisticId };
-    },
-    onSuccess: (data, _input, context) => {
-      if (!context) return;
-      if (context.isInfinite) {
-        queryClient.setQueryData(
-          context.listKey,
-          (old) => {
-            if (!old) {
+        if (isInfinitePost(input.parentId)) {
+          const previous2 = queryClient.getQueryData(listKey);
+          const isDesc = params.sort === "desc";
+          queryClient.setQueryData(
+            listKey,
+            (old) => {
+              if (!old) {
+                return {
+                  pages: [
+                    {
+                      items: [optimistic],
+                      total: 1,
+                      limit: params.pageSize ?? 10,
+                      offset: 0
+                    }
+                  ],
+                  pageParams: [0]
+                };
+              }
+              const targetIdx = isDesc ? 0 : old.pages.length - 1;
               return {
-                pages: [
-                  {
-                    items: [data],
-                    total: 1,
-                    limit: _input.limit ?? params.pageSize ?? 10,
-                    offset: _input.offset ?? 0
-                  }
-                ],
-                pageParams: [_input.offset ?? 0]
+                ...old,
+                // Increment `total` on every page so the header count (which reads
+                // pages[0].total) stays in sync even after multiple pages are loaded.
+                pages: old.pages.map(
+                  (page, idx) => idx === targetIdx ? {
+                    ...page,
+                    items: isDesc ? [optimistic, ...page.items] : [...page.items, optimistic],
+                    total: page.total + 1
+                  } : { ...page, total: page.total + 1 }
+                )
               };
             }
-            return {
-              ...old,
-              pages: old.pages.map((page) => ({
-                ...page,
-                items: page.items.map(
-                  (item) => item.id === context.optimisticId ? data : item
-                )
-              }))
-            };
-          }
-        );
-      } else {
-        queryClient.setQueryData(context.listKey, (old) => {
+          );
+          return { previous: previous2, isInfinite: true, listKey, optimisticId };
+        }
+        const previous = queryClient.getQueryData(listKey);
+        queryClient.setQueryData(listKey, (old) => {
           if (!old) {
-            return {
-              items: [data],
-              total: 1,
-              limit: _input.limit ?? params.pageSize ?? 20,
-              offset: _input.offset ?? 0
-            };
+            return { items: [optimistic], total: 1, limit: 20, offset: 0 };
           }
           return {
             ...old,
-            items: old.items.map(
-              (item) => item.id === context.optimisticId ? data : item
-            )
+            items: [...old.items, optimistic],
+            total: old.total + 1
           };
         });
+        return { previous, isInfinite: false, listKey, optimisticId };
+      },
+      onSuccess: (data, _input, context) => {
+        if (!context) return;
+        if (context.isInfinite) {
+          queryClient.setQueryData(
+            context.listKey,
+            (old) => {
+              if (!old) {
+                return {
+                  pages: [
+                    {
+                      items: [data],
+                      total: 1,
+                      limit: _input.limit ?? params.pageSize ?? 10,
+                      offset: _input.offset ?? 0
+                    }
+                  ],
+                  pageParams: [_input.offset ?? 0]
+                };
+              }
+              return {
+                ...old,
+                pages: old.pages.map((page) => ({
+                  ...page,
+                  items: page.items.map(
+                    (item) => item.id === context.optimisticId ? data : item
+                  )
+                }))
+              };
+            }
+          );
+        } else {
+          queryClient.setQueryData(
+            context.listKey,
+            (old) => {
+              if (!old) {
+                return {
+                  items: [data],
+                  total: 1,
+                  limit: _input.limit ?? params.pageSize ?? 20,
+                  offset: _input.offset ?? 0
+                };
+              }
+              return {
+                ...old,
+                items: old.items.map(
+                  (item) => item.id === context.optimisticId ? data : item
+                )
+              };
+            }
+          );
+        }
+      },
+      onError: (_err, _input, context) => {
+        if (!context) return;
+        queryClient.setQueryData(context.listKey, context.previous);
       }
     },
-    onError: (_err, _input, context) => {
-      if (!context) return;
-      queryClient.setQueryData(context.listKey, context.previous);
-    }
-  });
+    queryClient
+  );
 }
-function useUpdateComment(config) {
-  const queryClient = useQueryClient();
-  const client = getClient(config);
-  const queries = createCommentsQueryKeys(client, config.headers);
-  return useMutation({
-    mutationFn: (input) => runResourceMutation(
-      client,
-      commentMutations.update,
-      input,
-      config.headers
-    ),
-    onSettled: () => {
-      queryClient.invalidateQueries({
-        queryKey: queries.comments.list._def
-      });
-      queryClient.invalidateQueries({ queryKey: ["commentsThread"] });
-    }
-  });
+function useUpdateComment() {
+  const { client, headers, queries, queryClient } = useCommentsRuntime();
+  return useMutation(
+    {
+      mutationFn: (input) => runResourceMutation(
+        client,
+        commentMutations.update,
+        input,
+        headers
+      ),
+      onSettled: () => {
+        queryClient.invalidateQueries({
+          queryKey: queries.comments.list._def
+        });
+        queryClient.invalidateQueries({ queryKey: ["commentsThread"] });
+      }
+    },
+    queryClient
+  );
 }
-function useApproveComment(config) {
-  const queryClient = useQueryClient();
-  const client = getClient(config);
-  const queries = createCommentsQueryKeys(client, config.headers);
-  return useMutation({
-    mutationFn: (id) => runResourceMutation(
-      client,
-      commentMutations.updateStatus,
-      { id, status: "approved" },
-      config.headers
-    ),
-    onSettled: () => {
-      queryClient.invalidateQueries({
-        queryKey: queries.comments.list._def
-      });
-      queryClient.invalidateQueries({
-        queryKey: queries.commentCount.byResource._def
-      });
-      queryClient.invalidateQueries({ queryKey: ["commentsThread"] });
-    }
-  });
+function useApproveComment() {
+  const { client, headers, queries, queryClient } = useCommentsRuntime();
+  return useMutation(
+    {
+      mutationFn: (id) => runResourceMutation(
+        client,
+        commentMutations.updateStatus,
+        { id, status: "approved" },
+        headers
+      ),
+      onSettled: () => {
+        queryClient.invalidateQueries({
+          queryKey: queries.comments.list._def
+        });
+        queryClient.invalidateQueries({
+          queryKey: queries.commentCount.byResource._def
+        });
+        queryClient.invalidateQueries({ queryKey: ["commentsThread"] });
+      }
+    },
+    queryClient
+  );
 }
-function useUpdateCommentStatus(config) {
-  const queryClient = useQueryClient();
-  const client = getClient(config);
-  const queries = createCommentsQueryKeys(client, config.headers);
-  return useMutation({
-    mutationFn: (input) => runResourceMutation(
-      client,
-      commentMutations.updateStatus,
-      input,
-      config.headers
-    ),
-    onSettled: () => {
-      queryClient.invalidateQueries({
-        queryKey: queries.comments.list._def
-      });
-      queryClient.invalidateQueries({
-        queryKey: queries.commentCount.byResource._def
-      });
-      queryClient.invalidateQueries({ queryKey: ["commentsThread"] });
-    }
-  });
+function useUpdateCommentStatus() {
+  const { client, headers, queries, queryClient } = useCommentsRuntime();
+  return useMutation(
+    {
+      mutationFn: (input) => runResourceMutation(
+        client,
+        commentMutations.updateStatus,
+        input,
+        headers
+      ),
+      onSettled: () => {
+        queryClient.invalidateQueries({
+          queryKey: queries.comments.list._def
+        });
+        queryClient.invalidateQueries({
+          queryKey: queries.commentCount.byResource._def
+        });
+        queryClient.invalidateQueries({ queryKey: ["commentsThread"] });
+      }
+    },
+    queryClient
+  );
 }
-function useDeleteComment(config) {
-  const queryClient = useQueryClient();
-  const client = getClient(config);
-  const queries = createCommentsQueryKeys(client, config.headers);
-  return useMutation({
-    mutationFn: (id) => runResourceMutation(
-      client,
-      commentMutations.delete,
-      id,
-      config.headers
-    ),
-    onSettled: () => {
-      queryClient.invalidateQueries({
-        queryKey: queries.comments.list._def
-      });
-      queryClient.invalidateQueries({
-        queryKey: queries.commentCount.byResource._def
-      });
-      queryClient.invalidateQueries({ queryKey: ["commentsThread"] });
-    }
-  });
+function useDeleteComment() {
+  const { client, headers, queries, queryClient } = useCommentsRuntime();
+  return useMutation(
+    {
+      mutationFn: (id) => runResourceMutation(
+        client,
+        commentMutations.delete,
+        id,
+        headers
+      ),
+      onSettled: () => {
+        queryClient.invalidateQueries({
+          queryKey: queries.comments.list._def
+        });
+        queryClient.invalidateQueries({
+          queryKey: queries.commentCount.byResource._def
+        });
+        queryClient.invalidateQueries({ queryKey: ["commentsThread"] });
+      }
+    },
+    queryClient
+  );
 }
-function useToggleLike(config, params) {
-  const queryClient = useQueryClient();
-  const client = getClient(config);
-  const queries = createCommentsQueryKeys(client, config.headers);
+function useToggleLike(params) {
+  const { client, headers, queries, queryClient } = useCommentsRuntime();
   const isInfinite = !!params.infiniteKey && (params.parentId ?? null) === null;
   const listKey = isInfinite ? params.infiniteKey : queries.comments.list({
     resourceId: params.resourceId,
@@ -404,35 +434,38 @@ function useToggleLike(config, params) {
       });
     }
   }
-  return useMutation({
-    mutationFn: (input) => runResourceMutation(
-      client,
-      commentMutations.like,
-      input,
-      config.headers
-    ),
-    onMutate: async (input) => {
-      await queryClient.cancelQueries({ queryKey: listKey });
-      const previous = isInfinite ? queryClient.getQueryData(listKey) : queryClient.getQueryData(listKey);
-      applyLikeUpdate(input.commentId, (c) => {
-        const wasLiked = c.isLikedByCurrentUser;
-        return {
-          ...c,
-          isLikedByCurrentUser: !wasLiked,
-          likes: wasLiked ? Math.max(0, c.likes - 1) : c.likes + 1
-        };
-      });
-      return { previous };
-    },
-    onError: (_err, _input, context) => {
-      if (context?.previous !== void 0) {
-        queryClient.setQueryData(listKey, context.previous);
+  return useMutation(
+    {
+      mutationFn: (input) => runResourceMutation(
+        client,
+        commentMutations.like,
+        input,
+        headers
+      ),
+      onMutate: async (input) => {
+        await queryClient.cancelQueries({ queryKey: listKey });
+        const previous = isInfinite ? queryClient.getQueryData(listKey) : queryClient.getQueryData(listKey);
+        applyLikeUpdate(input.commentId, (c) => {
+          const wasLiked = c.isLikedByCurrentUser;
+          return {
+            ...c,
+            isLikedByCurrentUser: !wasLiked,
+            likes: wasLiked ? Math.max(0, c.likes - 1) : c.likes + 1
+          };
+        });
+        return { previous };
+      },
+      onError: (_err, _input, context) => {
+        if (context?.previous !== void 0) {
+          queryClient.setQueryData(listKey, context.previous);
+        }
+      },
+      onSettled: () => {
+        queryClient.invalidateQueries({ queryKey: listKey });
       }
     },
-    onSettled: () => {
-      queryClient.invalidateQueries({ queryKey: listKey });
-    }
-  });
+    queryClient
+  );
 }
 
 export { useApproveComment, useCommentCount, useComments, useDeleteComment, useInfiniteComments, usePostComment, useSuspenseComments, useSuspenseModerationComments, useToggleLike, useUpdateComment, useUpdateCommentStatus };