@stack-spot/ai-chat-widget 1.38.2-beta.2 → 2.0.0-beta.1

package/src/components/TabManager.tsx

@@ -1,8 +1,7 @@
 /* Tabs to select chat and add new chats */
 
-import { Button, IconBox } from '@citric/core'
-import { TimesMini } from '@citric/icons'
-import { IconButton } from '@citric/ui'
+import { Icon } from '@stack-spot/citric-icons'
+import { Button, IconButton } from '@stack-spot/citric-react'
 import { listToClass, theme } from '@stack-spot/portal-theme'
 import { Dictionary, useTranslate } from '@stack-spot/portal-translate'
 import { last } from 'lodash'
@@ -71,14 +70,6 @@ const Tabs = styled.nav<{ $numberOfExtraButtons: number }>`
     }
   }
 
-  ${IconButton} {
-    display: flex;
-    align-items: center;
-    justify-content: center;
-    background-color: transparent;
-    border: none;
-  }
-
   .list-overflow {
     max-width: ${({ $numberOfExtraButtons }) => `calc(100% - ${$numberOfExtraButtons * (EXTRA_BTN_SIZE + EXTRA_BTN_MARGIN * 2)}px)`};
   }
@@ -145,21 +136,6 @@ const Tabs = styled.nav<{ $numberOfExtraButtons: number }>`
           mask-image: linear-gradient(to right, rgba(0, 0, 0, 1) 0%, rgba(0, 0, 0, 1) 70%, rgba(0, 0, 0, 0) 100%);
         }
       }
-
-      ${IconButton} {
-        padding: 0;
-        width: 18px;
-        height: 18px;
-        transition: background-color 0.3s;
-
-        svg {
-          height: 14px;
-        }
-
-        &:hover {
-          background-color: ${theme.color.light[600]};
-        }
-      }
     }
   }
 `
@@ -185,37 +161,37 @@ export function TabManager<T, Key extends React.Key>(
   const tabItems = useMemo(() => tabs.map((tab) => (
     <li key={keygen(tab)} className={keygen(tab) === active ? 'active' : undefined} onClick={onClickTab}>
       <button className="label" onClick={() => onSelect(tab)}><FadingOverflow>{renderLabel(tab)}</FadingOverflow></button>
-      {tabs.length > 1 && <IconButton aria-label={t.close} title={t.close} onClick={() => onRemove(tab)}><TimesMini /></IconButton>}
+      {tabs.length > 1 &&
+        <IconButton appearance="text" icon="TimesMini" aria-label={t.close} title={t.close} onClick={() => onRemove(tab)} />}
     </li>
   )), [tabs, active])
 
-  const extras = useMemo(() => buttons.map(({ ariaLabel, label, onClick, icon, className, style }) => (
-    <>
-      {label ? 
-        <Button 
-          sx={{ paddingLeft: 0, marginRight: '4px', fontWeight: '400' }}
-          colorScheme="light"
-          size="md"
-          key={ariaLabel} 
-          title={label} 
-          style={style} 
-          onClick={onClick}>
-          <IconBox size="sm" sx={{ width: '32px', height: '16px', paddingInline: '8px' }} >
-            {icon} 
-          </IconBox>
-          {label} 
-        </Button> : 
-        <IconButton
-          key={ariaLabel}
-          aria-label={ariaLabel}
-          title={label}
-          className={listToClass([className, 'extra'])}
-          style={style}
-          onClick={onClick}
-        >
-          {icon}
-        </IconButton>}
-    </>)), [buttons])
+  const extras = useMemo(() => buttons.map(({
+    ariaLabel, title, label, onClick, group, icon, appearance, size, className, style, disabled }) => label
+    ? <Button 
+      style={{ marginRight: '4px', ...style }}
+      colorScheme="light"
+      size="md"
+      key={ariaLabel} 
+      title={title || label} 
+      onClick={onClick}
+      disabled={disabled}
+    >
+      <Icon group={group} icon={icon as any} />
+      {label} 
+    </Button>
+    : <IconButton
+      key={ariaLabel}
+      group={group}
+      icon={icon as any}
+      aria-label={ariaLabel}
+      title={title}
+      size={size}
+      className={listToClass([className, 'extra'])}
+      style={style}
+      onClick={onClick}
+      appearance={appearance}
+    />), [buttons])
 
   // when a new tab is added, we should scroll to it if there are more tabs that we can show.
   useEffect(() => {
@@ -227,7 +203,7 @@ export function TabManager<T, Key extends React.Key>(
   }, [tabs])
 
   return (
-    <Tabs $numberOfExtraButtons={buttons.length}>
+    <Tabs $numberOfExtraButtons={buttons.length} className="tabs">
       <FadingOverflow className="list-overflow" scroll="arrows" enableHorizontalScrollWithVerticalWheel>
         <ul ref={tabList}>{tabItems}</ul>
       </FadingOverflow>

package/src/components/ToolBadge.tsx

@@ -1,148 +1,67 @@
-import { IconBox, Text } from '@citric/core'
-import { ChevronDown, Cog } from '@citric/icons'
-import { IconButton } from '@citric/ui'
-import { AnimatedHeight } from '@stack-spot/portal-components/AnimatedHeight'
-import { ColorSchemeKey, listToClass, theme } from '@stack-spot/portal-theme'
-import { Dictionary, useTranslate } from '@stack-spot/portal-translate'
-import { get } from 'lodash'
-import { DetailedHTMLProps, HTMLAttributes, useState } from 'react'
+import { Icon } from '@stack-spot/citric-icons'
+import { Accordion, Card, ImageBox, ImageWithFallback, Text } from '@stack-spot/citric-react'
+import { theme } from '@stack-spot/portal-theme'
 import { styled } from 'styled-components'
-import { FastOmit } from 'styled-components/dist/types'
+import { useCurrentChatState } from '../context/hooks'
 import { toPrecision } from '../views/Steps/utils'
 
-interface Props extends FastOmit<DetailedHTMLProps<HTMLAttributes<HTMLDivElement>, HTMLDivElement>, never> {
+type Props = Omit<React.JSX.IntrinsicElements['div'], 'onChange'> & {
   name: string,
   duration?: number,
   image?: string,
   description?: string,
-  appearance?: 'round' | 'square',
-  backgroundColor?: ColorSchemeKey,
 }
 
-const Styled = styled.div<{ $backgroundColor: ColorSchemeKey }>`
- .tool-header-wrapper {
-    background-color: ${({ $backgroundColor }) => get(theme.color, $backgroundColor)};
-    border-radius: 8px;
+const Styled = styled.div`
+  [data-citric="card"] {
+    flex-direction: row;
+    gap: 8px;
+    border-radius: ${theme.radius.xl};
+    border: none;
+    align-items: center;
+    padding: 4px 16px 4px 4px;
   }
- 
-  .tool-header {
-    padding: 4px 8px 4px 4px;
-    display: flex;
-    flex-direction: column;
-    gap: 12px;
-    position: relative;
 
-    .title {
-      display: flex;
-      gap: 8px;
-      align-items: center;
-      .tool-image {
-        width: 24px;
-        height: 24px;
-        border-radius: 50%;
-        overflow: hidden;
-        flex-shrink: 0;
-      }
-      .name {
-        flex: 1;
-        overflow: hidden;
-        text-overflow: ellipsis;
-      }
-    }
-
-    .btn-expand {
-      display: none;
-    }
-
-    .duration {
-      opacity: 0.7;
-      text-align: right;
-    }
-  }
-
-  &.with-description {
-    .tool-header {
-      &:before {
-        content: '';
-        top: 32px;
-        bottom: 8px;
-        left: 15px;
-        width: 1px;
-        background-color: ${theme.color.light[700]};
-        opacity: 0.3;
-        position: absolute;
-      }
-
-      .btn-expand {
-        display: flex;
-        transition: transform 0.3s ease-in-out;
-        &.open {
-          transform: rotate(180deg);
-        }
-      }
-
-      .description {
-        padding: 0 0 10px 20px;
-        opacity: 0.7;
-      }
-    }
-  }
-
-  &.round {
-    .tool-header-wrapper {
-      border-radius: 30px;
-    }
-    &:not(.with-description) {
-      .tool-header-wrapper {
-        padding-right: 10px;
+  [data-citric="accordion"] {
+    background-color: ${theme.color.light[500]};
+    > header > label {
+      padding: 6px 10px 6px 6px;
+      font: ${theme.font.body2};
+      input:before {
+        font-size: 12px !important;
       }
     }
   }
 `
 
 export const ToolBadge = (
-  { name, duration, image, description, appearance = 'square', backgroundColor = 'light.300', className, ...props }: Props,
+  { name, duration, image, description, ...props }: Props,
 ) => {
-  const t = useTranslate(dictionary)
-  const [showDescription, setShowDescription] = useState(false)
-  const rootClass = listToClass([className, appearance, description && 'with-description'])
-  const wrapperClass = 'tool-header-wrapper'
-
-  const content = (
-    <div className="tool-header">
-      <div className="title">
-        {image ? <img src={image} className="tool-image" /> : <IconBox className="tool-image"><Cog /></IconBox>}
-        <Text colorScheme="light.700" className="name">{name}</Text>
-        {duration !== undefined && <Text colorScheme="light.700" className="duration">
-          {toPrecision(duration)}s
-        </Text>}
-        <IconButton
-          size="xs"
-          className={listToClass(['btn-expand', showDescription && 'open'])}
-          onClick={() => setShowDescription(v => !v)}
-          aria-label={showDescription ? t.close : t.open}
-        >
-          <ChevronDown />
-        </IconButton>
-      </div>
-      {showDescription && <Text className="description" colorScheme="light.700">{description}</Text>}
-    </div>
-  )
-
-  return (
-    <Styled {...props} $backgroundColor={backgroundColor} className={rootClass}>
-      {description ? <AnimatedHeight className={wrapperClass}>{content}</AnimatedHeight> : <div className={wrapperClass}>{content}</div>}
-    </Styled>
-  )
+  const currentAgent = useCurrentChatState('agent')
+  
+  const header = <>
+    <ImageBox><ImageWithFallback src={image} fallback={
+      <ImageWithFallback
+        src={currentAgent?.image ?? ''}
+        className="image"
+        fallback={<Icon icon="Cog" className="tool-image" />}
+      />
+    } /></ImageBox>
+    <Text color="light.700" className="name" style={{ flex: 1 }}>{name}</Text>
+    {duration !== undefined && <Text color="light.700" className="duration">
+      {toPrecision(duration)}s
+    </Text>}
+  </>
+    
+  return <Styled>
+    {description ? (
+      <Accordion
+        appearance="card"
+        header={header}
+        {...props}
+      >
+        <Text className="description" color="light.700">{description}</Text>
+      </Accordion>
+    ) : <Card {...props}>{header}</Card>}
+  </Styled>
 }
-
-const dictionary = {
-  en: {
-    open: 'Open',
-    close: 'Close',
-  },
-  pt: {
-    open: 'Abrir',
-    close: 'Fechar',
-  },
-} satisfies Dictionary

package/src/components/WorkspaceTabNavigator.tsx

@@ -1,6 +1,5 @@
-import { Box, Flex, IconBox, Image, Text } from '@citric/core'
-import { ArrowRight, Search, Spaces, Times } from '@citric/icons'
-import { Avatar, IconButton } from '@citric/ui'
+import { Icon } from '@stack-spot/citric-icons'
+import { FieldGroup, IconButton, ImageBox, Input, Row, Text } from '@stack-spot/citric-react'
 import { Placeholder } from '@stack-spot/portal-components/Placeholder'
 import { workspaceAiClient } from '@stack-spot/portal-network'
 import { WorkspaceResponse, WorkspaceVisibilityLevelEnum } from '@stack-spot/portal-network/api/workspace-ai'
@@ -9,7 +8,6 @@ import { memo, useMemo, useState } from 'react'
 import { useRightPanel } from '../right-panel/hooks'
 import { ButtonFavorite } from './ButtonFavorite'
 import { ComponentNavigator, ComponentNavigatorProps, NavigationComponent, NavigationMap, useComponentNavigation } from './ComponentNavigator'
-import { IconInput } from './IconInput'
 import { ListResource } from './ListResource'
 
 interface CardSpaceProps {
@@ -20,25 +18,24 @@ interface CardSpaceProps {
 }
 
 export const CardSpace = ({ onClick, name, icon, logoUrl }: CardSpaceProps) =>
-  <Flex
+  <Row
     onClick={onClick}
     flex={1}
-    alignItems="center"
     justifyContent="space-between"
     mr={2}
     bg="light.400"
-    r="sm"
+    radius="sm"
     p={3}
-    sx={{ cursor: 'pointer' }}
+    style={{ cursor: 'pointer' }}
   >
-    <Flex alignContent="center" alignItems="center" sx={{ gap: '8px', m: 1 }} >
-      <Avatar size="xxs" appearance="square" sx={{ bg: 'light.600', r: 'xxs' }}>
-        {logoUrl ? <Image src={logoUrl} /> : <IconBox> {icon} </IconBox>}
-      </Avatar>
+    <Row gap="8px">
+      <ImageBox appearance="square">
+        {logoUrl ? <img src={logoUrl} /> : icon}
+      </ImageBox>
       <Text appearance="body2">{name}</Text>
-    </Flex>
-    <IconButton><ArrowRight /></IconButton>
-  </Flex>
+    </Row>
+    <IconButton icon="ArrowRight" />
+  </Row>
 
 interface WorkspaceSourcesTabProps {
   visibility: WorkspaceVisibilityLevelEnum,
@@ -91,16 +88,17 @@ const WorkspaceSourcesTab = ({ visibility, onClick }: WorkspaceSourcesTabProps)
 
   return (
     <>
-      <Box w={12}>
-        <IconInput icon={<Search />} value={filter} onChange={setFilter} className="search" />
-      </Box>
+      <FieldGroup fullWidth style={{ alignSelf: 'stretch' }}>
+        <Icon icon="Search" />
+        <Input type="search" value={filter} onChange={setFilter} />
+      </FieldGroup>
 
       {!!filtered.length &&
         <ListResource
           list={filtered}
           keygen={w => w.id}
           style={{ gap: '6px', display: 'flex', flexDirection: 'column' }}
-          renderLabel={w => <CardSpace name={w.name} logoUrl={w.logo} icon={<Spaces />} onClick={() => onClick(w)} />}
+          renderLabel={w => <CardSpace name={w.name} logoUrl={w.logo} icon={<Icon icon="Spaces" />} onClick={() => onClick(w)} />}
           renderAfterElement={(w) =>
             <ButtonFavorite favorite={{ idOrSlug: w?.id, listFavorites, onAddFavorite, onRemoveFavorite }} />}
           optionClassName={w => (filter && !w.name.toLocaleLowerCase().includes(filter.toLocaleLowerCase()))
@@ -110,26 +108,25 @@ const WorkspaceSourcesTab = ({ visibility, onClick }: WorkspaceSourcesTabProps)
           className="option-list"
         />
       }
-      <Box w={12}>
+      <div style={{ width: '100%' }}>
         {!!workspaces.length && !filtered.length &&
           <Placeholder title={t.noSearchResults} description={t.noSearchResultsDescription} className="no-data-placeholder" />}
         {!workspaces.length && <Placeholder title={t.noData} description={t.noDataDescription} />}
-      </Box>
+      </div>
     </>
   )
 }
 
-
 const WorkspaceHeader = <T extends NavigationMap, K extends keyof T>({ data }: { data: NavigationComponent<T, K> }) => {
   const { close: closeRightPanel } = useRightPanel()
   const workspaceId = (data.props as any)['workspaceId']
   if (!workspaceId) return
 
   const workspace = workspaceAiClient.workspaceAi.useQuery({ id: workspaceId })
-  return <Flex justifyContent="space-between" alignItems="center" flex={1}>
+  return <Row justifyContent="space-between" flex={1}>
     {data.component === 'workspaceResource' ? 'Spots' : workspace.name}
-    {data.fullScreen && <IconButton title={'t.close'} aria-label={'t.close'} onClick={closeRightPanel}>  <Times /> </IconButton>}
-  </Flex>
+    {data.fullScreen && <IconButton icon="Times" title={'t.close'} aria-label={'t.close'} onClick={closeRightPanel} />}
+  </Row>
 }
 
 interface WorkspaceTabNavigatorProps {

package/src/components/form/DescribedCheckboxGroup.tsx

@@ -1,81 +1,106 @@
-import { Checkbox, Flex, Text } from '@citric/core'
-import { listToClass } from '@stack-spot/portal-theme'
+import { Icon } from '@stack-spot/citric-icons'
+import { Accordion, Checkbox, CheckboxGroup, FieldGroup, Input, Row, Text, useCheckboxGroupControls } from '@stack-spot/citric-react'
 import { Dictionary, useTranslate } from '@stack-spot/portal-translate'
-import { useMemo } from 'react'
-import { Accordion } from '../Accordion'
-import { RadioCheckBox } from './styled'
-import { CheckProps } from './types'
+import { difference, uniqBy } from 'lodash'
+import { useCallback, useEffect, useState } from 'react'
+import { Labeled } from '../../state/types'
+import { ButtonFavorite, Favorite } from '../ButtonFavorite'
+
+interface Props<T> {
+  initialValue?: T[],
+  options: T[],
+  /**
+   * Must return the data needed by the component, given the option.
+   */
+  data: (option: T) => { name: string, description: string } & Favorite<any>,
+  /**
+   * Use this when the selection must be kept at a global level, managed by a React Ref.
+   */
+  globalSelection?: React.MutableRefObject<Labeled[] | undefined>,
+  emptyResults: React.ReactNode,
+  emptyDataset: React.ReactNode,
+  onChange?: (value: T[]) => void,
+}
 
 /**
  * Renders a checkbox group where each option has a label and a description.
  * The description in placed under the label and checkbox as an accordion.
+ * 
+ * Also renders a search input and a select all checkbox.
  */
-export function DescribedCheckboxGroup<T>({
-  keygen, 
-  onChange, 
-  options, 
-  renderDescription, 
-  renderLabel, 
-  renderBeforeElement, 
-  renderAfterElement, 
-  optionClassName, 
-  optionStyle, 
-  value, 
-  className, 
-  style }: CheckProps<T>) {
+export function DescribedCheckboxGroup<T>(
+  { initialValue, options: opt, data, globalSelection, onChange: change, emptyDataset, emptyResults }: Props<T>,
+) {
   const t = useTranslate(dictionary)
-  const allSelected = options.length > 0 && options.every(option => value.includes(option))
+  const [options, setOptions] = useState(opt)
 
-  const handleSelectAll = (e: React.ChangeEvent<HTMLInputElement>) => {
-    if (e.target.checked) {
-      onChange(options)
-    } else {
-      onChange([])
-    }
-  }
+  useEffect(() => {
+    if (controls.value.length === 0 && (initialValue?.length ?? 0) > 0) controls.setValue(initialValue!)
+  }, [initialValue])
 
-  const items = useMemo(() => options.map((option) => {
-    const label = renderLabel(option)
-    const description = renderDescription(option)
-    const header = (
-      <label>
-        <Checkbox
-          checked={value.includes(option)}
-          onChange={(e) => {
-            if (e.target.checked && !value.includes(option)) onChange([...value, option])
-            else onChange(value.filter(item => item !== option))
-          }}
-        />
-        {typeof label === 'string' ? <Text>{label}</Text> : label}
-      </label>
+  const onChange = useCallback((newValue: T[], previousValue: T[]) => {
+    change?.(newValue)
+    if (!globalSelection?.current) return
+    const added = difference(newValue, previousValue)
+    const removed = difference(previousValue, newValue)
+    globalSelection.current = globalSelection.current?.filter(o => !removed.some(r => data(r).idOrSlug === o.id)) ?? []
+    globalSelection.current = uniqBy(
+      [...globalSelection.current, ...added.map(o => ({ id: data(o).idOrSlug, label: data(o).name }))],
+      'id',
     )
+  }, [])
 
-    return (
-      <>
-        <li key={keygen(option)} className={listToClass(['radio-group-item', optionClassName?.(option)])} style={optionStyle?.(option)}>
-          {renderBeforeElement?.(option)}
-          <Accordion header={header}>
-            {typeof description === 'string'
-              ? <Text appearance="microtext1" colorScheme="light.700">{description}</Text>
-              : description as any
-            }
-          </Accordion>
-          {renderAfterElement?.(option)}
-        </li>
-      </>
-    )
-  }), [options, value])
+  const controls = useCheckboxGroupControls({
+    initialValue,
+    options,
+    applyFilter: (filter, o) => data(o).name.toLocaleLowerCase().includes(filter.toLocaleLowerCase()),
+    renderKey: o => data(o).idOrSlug,
+    onChange,
+  })
 
-  return <RadioCheckBox style={style} className={className}>
-    <Flex as="li" alignItems="center" ml={3} pl={1}>
-      <Checkbox
-        checked={allSelected}
-        onChange={handleSelectAll}
+  return options.length ? <>
+    <FieldGroup fullWidth>
+      <Icon icon="Search" />
+      <Input type="search" value={controls.filter} onChange={controls.setFilter} />
+    </FieldGroup>
+    {controls.options.length ? <>
+      <Row tag="label" gap="10px" p="11px">
+        <Checkbox value={controls.isAllSelected} onChange={v => v ? controls.selectAll() : controls.removeSelection()} />
+        <Text>{controls.isAllSelected ? t.removeAll : t.selectAll}</Text>
+      </Row>
+      <CheckboxGroup
+        options={controls.options}
+        value={controls.value}
+        onChange={controls.setValue}
+        renderKey={controls.renderKey}
+        className="option-list"
+        renderItem={(checkbox, o) => {
+          const { description, idOrSlug, name, listFavorites, onAddFavorite, onRemoveFavorite } = data(o)
+          return (
+            <Row className={controls.isUnfilteredButChecked(o) ? 'filtered-out' : ''}>
+              <Accordion header={btn => <>{checkbox}<Text>{name}</Text>{btn}</>} appearance="card" maxHeight={100}>
+                <Text appearance="microtext1" color="light.700">{description}</Text>
+              </Accordion>
+              {onAddFavorite && <ButtonFavorite favorite={{
+                idOrSlug: idOrSlug,
+                listFavorites,
+                onAddFavorite: async (...args) => {
+                  const res = await onAddFavorite(...args)
+                  setOptions([...options]) // forces options re-rendering
+                  return res
+                },
+                onRemoveFavorite: async (...args) => {
+                  const res = await onRemoveFavorite?.(...args)
+                  setOptions([...options]) // forces options re-rendering
+                  return res ?? false
+                },
+              }} />}
+            </Row>
+          )
+        }}
       />
-      <Text>{allSelected ? t.removeAll : t.selectAll}</Text>
-    </Flex>
-    {items}
-  </RadioCheckBox>
+    </> : emptyResults}
+  </> : emptyDataset
 }
 
 const dictionary = {