@stack-spot/ai-chat-widget 1.31.2-beta.1 → 1.32.0

package/src/views/Stacks.tsx

@@ -5,18 +5,22 @@ import { aiClient, workspaceAiClient } from '@stack-spot/portal-network'
 import { GetAiStackResponse, VisibilityLevelEnum } from '@stack-spot/portal-network/api/ai'
 import { WorkspaceResponse } from '@stack-spot/portal-network/api/workspace-ai'
 import { Dictionary, useTranslate } from '@stack-spot/portal-translate'
-import { useCallback, useEffect, useMemo, useRef, useState } from 'react'
+import { ReactElement, useCallback, useEffect, useMemo, useRef, useState } from 'react'
 import { ButtonFavorite } from '../components/ButtonFavorite'
 import { NavigationComponent } from '../components/ComponentNavigator'
 import { DescribedRadioGroup } from '../components/form/DescribedRadioGroup'
 import { IconInput } from '../components/IconInput'
+import { RightPanelContentList } from '../components/RightPanelContentList'
 import { RightPanelTabs } from '../components/RightPanelTabs'
 import { WorkspaceTabNavigator } from '../components/WorkspaceTabNavigator'
 import { useCurrentChat, useWidget, useWidgetState } from '../context/hooks'
+import { Scope } from '../features'
 import { useRightPanel } from '../right-panel/hooks'
 import { ChatProperties } from '../state/ChatState'
 import { checkIsTrial } from '../utils/check-is-trial'
 
+type TabElement = { title: string, content: ReactElement }
+
 /**
  * Renders the Stack selection form in the Right Panel if this is the panel that is currently opened.
  */
@@ -41,27 +45,43 @@ const StacksPanel =  () => {
   const chat = useCurrentChat()
   const isTrial = checkIsTrial()
   const stack = useRef(chat.get('stack'))
+  const features = useWidgetState('features')
+  const isGroupResourcesByScope = features?.groupResourcesByScope
+  const scopes = features?.scopes ?? []
+  const spotId = features?.workspaceId
 
   useEffect(() => {
     stack.current = chat.get('stack')
   }, [chat])
 
-  const tabs = useMemo(() => isTrial ? [
-    { title: t.favorites, content: <StacksTab key="favorites" visibility="favorite" stack={stack} /> },
-    { title: t.personal, content: <StacksTab key="personal" visibility="personal" stack={stack} /> },
-  ]: [
-    { title: t.favorites, content: <StacksTab key="favorites" visibility="favorite" stack={stack} /> },
-    { title: t.personal, content: <StacksTab key="personal" visibility="personal" stack={stack} /> },
-    { title: t.shared, content: <StacksTab key="shared" visibility="shared" stack={stack} /> },
-    { title: t.spots, content: <StacksTabWorkspace key="workspace" visibility="workspace" stack={stack} /> },
-    { title: t.account, content: <StacksTab key="account" visibility="account" stack={stack} /> },
-  ], [t, isTrial])
-  
-  return <RightPanelTabs key={chat.id} tabs={tabs} />
+  const allTabsMap: Partial<Record<Scope, TabElement>> = {
+    favorite: { title: t.favorites, content: <StacksTab key="favorites" visibility="favorite" stack={stack} /> },
+    personal: { title: t.personal, content: <StacksTab key="personal" visibility="personal" stack={stack} /> },
+    shared: { title: t.shared, content: <StacksTab key="shared" visibility="shared" stack={stack} /> },
+    workspace: { title: t.spots, content: <StacksTabWorkspace key="workspace" visibility="workspace" stack={stack} /> },
+    account: { title: t.account, content: <StacksTab key="account" visibility="account" stack={stack} /> },
+  }
+  const defaultScopes: Scope[] = ['favorite', 'personal', 'shared', 'workspace', 'account']
+
+  const rawScopes: Scope[] = scopes?.length ? scopes : defaultScopes
+
+  const scopesToRender: Scope[] = isTrial
+    ? ['favorite', 'personal']
+    : rawScopes
+
+  const tabs = scopesToRender
+    .map(scope => allTabsMap[scope])
+    .filter(Boolean) as TabElement[]
+
+  return (isGroupResourcesByScope ? (
+    <RightPanelTabs key={chat.id} tabs={tabs} />
+  ) : (
+    <RightPanelContentList><StacksTab stack={stack} workspaceId={spotId} /></RightPanelContentList>
+  ))
 }
 
 export interface StacksTabProps {
-  visibility: VisibilityLevelEnum,
+  visibility?: VisibilityLevelEnum,
   workspaceId?: string,
   stack: React.MutableRefObject<ChatProperties['stack']>,
   showSubmitButton?: boolean,