@stack-spot/ai-chat-widget 2.10.0 → 2.11.0

package/src/views/MessageInput/QuickCommandSelector.tsx

@@ -1,6 +1,6 @@
 import { Icon } from '@stack-spot/citric-icons'
-import { aiClient, workspaceAiClient } from '@stack-spot/portal-network'
-import { QuickCommandResponse } from '@stack-spot/portal-network/api/ai'
+import { AgentVisibilityLevel, aiClient, workspaceAiClient } from '@stack-spot/portal-network'
+import { QuickCommandResponse, VisibilityLevelEnum } from '@stack-spot/portal-network/api/ai'
 import { useCallback } from 'react'
 import { Selector } from '../../components/Selector'
 import { useCurrentChat, useCurrentChatState, useWidgetState } from '../../context/hooks'
@@ -8,17 +8,16 @@ import { quickCommandRegex } from '../../regex'
 
 type QuickCommandResponseWithSpaceName = QuickCommandResponse & { spaceName?: string }
 
-export const QuickCommandSelector = ({ inputRef, isTrial }: 
+export const QuickCommandSelector = ({ inputRef, isTrial }:
   { isTrial: boolean, inputRef: React.RefObject<HTMLTextAreaElement | HTMLInputElement> }) => {
   const chat = useCurrentChat()
-  const isQuickCommandEnabled =  useCurrentChatState('features').quickCommands
+  const isQuickCommandEnabled = useCurrentChatState('features').quickCommands
   const spotId = useWidgetState('features')?.workspaceId
 
-  const useFavorites = () => aiClient.allQuickCommands.useQuery({ visibility: 'favorite' })
   const [addFavorite, pendingAddFav] = aiClient.addFavoriteQuickCommand.useMutation()
   const [removeFavorite, pendingRemoveFav] = aiClient.removeFavoriteQuickCommand.useMutation()
 
-  const addFavoriteQc = async(idOrSlug?: string) => {
+  const addFavoriteQc = async (idOrSlug?: string) => {
     try {
       await addFavorite({ slug: idOrSlug || '' })
       await aiClient.allQuickCommands.invalidate()
@@ -32,7 +31,7 @@ export const QuickCommandSelector = ({ inputRef, isTrial }:
   const onAddFavorite = (idOrSlug?: string) => new Promise<boolean>(async (resolve, reject) => {
     try {
       await addFavoriteQc(idOrSlug)
-      if (!pendingAddFav){
+      if (!pendingAddFav) {
         resolve(true)
       }
 
@@ -40,10 +39,10 @@ export const QuickCommandSelector = ({ inputRef, isTrial }:
       // eslint-disable-next-line no-console
       console.error(error)
       reject(error)
-    } 
-  }) 
+    }
+  })
 
-  const removeFavoriteQc = async(idOrSlug?: string) => {
+  const removeFavoriteQc = async (idOrSlug?: string) => {
     try {
       await removeFavorite({ slug: idOrSlug || '' })
       await aiClient.allQuickCommands.invalidate()
@@ -57,7 +56,7 @@ export const QuickCommandSelector = ({ inputRef, isTrial }:
   const onRemoveFavorite = (idOrSlug?: string) => new Promise<boolean>(async (resolve, reject) => {
     try {
       await removeFavoriteQc(idOrSlug)
-      if (!pendingRemoveFav){
+      if (!pendingRemoveFav) {
         resolve(true)
       }
 
@@ -65,9 +64,9 @@ export const QuickCommandSelector = ({ inputRef, isTrial }:
       // eslint-disable-next-line no-console
       console.error(error)
       reject(error)
-    } 
+    }
   })
-  
+
   const onSelectItem = useCallback((qc: QuickCommandResponseWithSpaceName) => {
     const newValue = `/${qc.slug}`
     chat.set('nextMessage', newValue)
@@ -77,12 +76,19 @@ export const QuickCommandSelector = ({ inputRef, isTrial }:
     inputRef.current.focus()
   }, [chat, inputRef])
 
-  const getQuickCommands = () => {
+  const getQuickCommands = ({ filter, visibility }: { filter?: string, visibility?: AgentVisibilityLevel[] | VisibilityLevelEnum[] }) => {
+
     if (spotId) {
-      return workspaceAiClient.getQCFromWorkspaceAi.useQuery({ workspaceId: spotId })
+      return { data: workspaceAiClient.getQCFromWorkspaceAi.useQuery({ workspaceId: spotId, name: filter }) }
     }
 
-    const quickCommands = aiClient.allQuickCommands.useQuery({ order: 'a-to-z' })
+    const [quickCommands, { fetchNextPage, hasNextPage }] = aiClient.allQuickCommandsV3.useInfiniteQuery({
+      order: 'a-to-z',
+      visibilityList: visibility as VisibilityLevelEnum[],
+      name: filter,
+      page: 1, size: 20,
+    })
+
     const quickCommandsFiltered = quickCommands.filter(
       (qc) => qc.visibility_level.toLowerCase() !== 'workspace',
     )
@@ -92,11 +98,17 @@ export const QuickCommandSelector = ({ inputRef, isTrial }:
     })
 
     const workspaceQuickCommandsWithWorkspaceName: QuickCommandResponseWithSpaceName[] =
-    workspaceQuickCommands.flatMap(({ qcs, space_name }) =>
-      qcs?.map((qc) => ({ ...qc, spaceName: space_name })),
-    ) as QuickCommandResponseWithSpaceName[]
+      workspaceQuickCommands.flatMap(({ qcs, space_name }) =>
+        qcs?.map((qc) => ({ ...qc, spaceName: space_name })),
+      ) as QuickCommandResponseWithSpaceName[]
+
+    const filteredWorkspaceQuickCommand = filter ? workspaceQuickCommandsWithWorkspaceName?.filter((qc) => 
+      qc.name?.toLowerCase().includes(filter!)) : workspaceQuickCommandsWithWorkspaceName
 
-    return [...quickCommandsFiltered, ...workspaceQuickCommandsWithWorkspaceName]
+    const data = !visibility ? [...quickCommandsFiltered, ...filteredWorkspaceQuickCommand] : (
+      visibility.includes('workspace') ? filteredWorkspaceQuickCommand : quickCommandsFiltered
+    )
+    return { data, fetchNextPage: fetchNextPage, hasNextPage }
   }
 
   const QuickCommandItem = ({ slug, description, spaceName }: QuickCommandResponseWithSpaceName) => <>
@@ -107,9 +119,7 @@ export const QuickCommandSelector = ({ inputRef, isTrial }:
 
   return <Selector
     inputRef={inputRef}
-    favorite={{ 
-      useFavorites, onAddFavorite, onRemoveFavorite, favoriteIsSlug: true,
-    }}
+    favorite={{ onAddFavorite, onRemoveFavorite, favoriteIsSlug: true }}
     selectorConfig={{
       resourceName: 'Quick Command',
       shortcut: '/',