@stack-spot/ai-chat-widget 0.2.0 → 0.4.0

package/src/components/HistoryList.tsx

@@ -1,16 +1,89 @@
 /* eslint-disable no-empty-pattern */
-import { WithStyle } from '@stack-spot/portal-theme'
-import { ButtonAction } from '../types'
+import { Text } from '@citric/core'
+import { theme, WithStyle } from '@stack-spot/portal-theme'
+import { Dictionary, useTranslate } from '@stack-spot/portal-translate'
+import { groupBy, map } from 'lodash'
+import { useMemo } from 'react'
+import { styled } from 'styled-components'
+import { subtractDays } from '../utils/date'
+
+type HistorySectionName = 'today' | 'yesterday' | 'last7' | 'last30' | 'older'
 
 interface Props<T> extends WithStyle {
   items: T[],
-  renderLabel: (item: T) => React.ReactElement,
+  renderItem: (item: T) => React.ReactNode,
   getDate: (item: T) => Date,
   keygen: (item: T) => React.Key,
-  getActions?: (item: T) => ButtonAction[],
-  onSelect?: (item: T) => void,
 }
 
-export function HistoryList<T>({}: Props<T>) {
-  return null
+const HistoryBox = styled.div`
+  display: flex;
+  flex-direction: column;
+  gap: 4px;
+
+  > section {
+    > header {
+      background-color: ${theme.color.light[500]};
+      padding: 8px;
+      border-radius: 4px;
+      color: ${theme.color.light[700]};
+      margin-bottom: 4px;
+    }
+
+    > ul {
+      margin: 0;
+      padding: 0;
+      list-style: none;
+      display: flex;
+      flex-direction: column;
+      gap: 2px;
+    }
+  }
+`
+
+function dateToSectionName(date: Date): HistorySectionName {
+  const now = new Date()
+  if (date.toDateString() === now.toDateString()) return 'today'
+  const yesterday = subtractDays(now, 1)
+  if (date.toDateString() === yesterday.toDateString()) return 'yesterday'
+  const todayAtMidnight = new Date(now.toDateString())
+  const last7Days = subtractDays(todayAtMidnight, 7)
+  if (date.getTime() >= last7Days.getTime()) return 'last7'
+  const last30Days = subtractDays(todayAtMidnight, 30)
+  return date.getTime() >= last30Days.getTime() ? 'last30' : 'older'
 }
+
+export function HistoryList<T>({ getDate, items, keygen, className, style, renderItem }: Props<T>) {
+  const t = useTranslate(dictionary)
+  const sections = useMemo(() => {
+    const byDate = groupBy(items, item => dateToSectionName(getDate(item)))
+    return map(byDate, (value: T[], key: HistorySectionName) => (
+      <section key={key}>
+        <header>
+          <Text>{t[key]}</Text>
+        </header>
+        <ul>{value.map(item => <li key={keygen(item)}>{renderItem(item)}</li>)}</ul>
+      </section>
+    ))
+  }, [items])
+
+  return <HistoryBox className={className} style={style}>{sections}</HistoryBox>
+}
+
+const dictionary = {
+  en: {
+    today: 'Today',
+    yesterday: 'Yesterday',
+    last7: 'Last 7 days',
+    last30: 'Last 30 days',
+    older: 'Older',
+  },
+  pt: {
+    today: 'Hoje',
+    yesterday: 'Ontem',
+    last7: 'Últimos 7 dias',
+    last30: 'Últimos 30 dias',
+    older: 'Mais antigo',
+  },
+} satisfies Dictionary
+

package/src/components/OverlayMenu.tsx

@@ -1,10 +1,77 @@
 /* eslint-disable no-empty-pattern */
-import { WithStyle } from '@stack-spot/portal-theme'
+import { IconBox, Text } from '@citric/core'
+import { theme, WithStyle } from '@stack-spot/portal-theme'
+import { useMemo } from 'react'
+import { styled } from 'styled-components'
 import { ButtonAction, WithChildren } from '../types'
+import { Tooltip } from './Tooltip'
+import { useTooltip } from './Tooltip/context'
+import { TooltipPosition } from './Tooltip/types'
 
 interface Props extends WithStyle, WithChildren {
-  position?: 'left' | 'top' | 'bottom' | 'right',
+  position?: TooltipPosition,
   actions: ButtonAction[],
 }
 
-export const OverlayMenu = ({}: Props) => null
+const MenuList = styled.ul`
+  margin: 0;
+  padding: 0;
+  list-style: none;
+  border-radius: 8px;
+  background-color: ${theme.color.light[400]};
+  overflow: hidden;
+  display: flex;
+  flex-direction: column;
+
+  > li {
+    display: flex;
+    flex-direction: column;
+
+    > button {
+      padding: 8px 12px;
+      transition: background-color 0.2s;
+      cursor: pointer;
+      display: flex;
+      flex-direction: row;
+      align-items: center;
+      gap: 8px;
+      background-color: transparent;
+      border: none;
+
+      &:hover {
+        background-color: ${theme.color.light[500]};
+      }
+    }
+  }
+`
+
+const StyledButton = styled.button<{ $color: string | undefined }>`
+  color: ${({ $color }) => $color || theme.color.light.contrastText};
+
+  svg {
+    fill: ${({ $color }) => $color || theme.color.light.contrastText};
+  }
+`
+
+export const OverlayMenu = ({ actions, children, className, position, style }: Props) => {
+  const tooltip = useTooltip()
+  const menu = useMemo(() => {
+    const items = actions.map(({ label, onClick, className, color, icon, style }) => (
+      <li key={label} className={className} style={style}>
+        <StyledButton $color={color} onClick={() => {
+          onClick()
+          tooltip.hide()
+        }}>
+          <IconBox>{icon}</IconBox>
+          <Text>{label}</Text>
+        </StyledButton>
+      </li>
+    ))
+    return <MenuList>{items}</MenuList>
+  }, [actions])
+  return (
+    <Tooltip content={menu} custom position={position} className={className} style={style} triggeredBy="click">
+      {children}
+    </Tooltip>
+  )
+}

package/src/components/Tooltip/Tooltip.tsx

@@ -8,21 +8,27 @@ import { TooltipPosition } from './types'
 interface Props extends WithChildren, WithStyle {
   content: React.ReactNode,
   position?: TooltipPosition,
+  triggeredBy?: 'click' | 'hover',
   custom?: boolean,
 }
 
-export const Tooltip = ({ content, custom, position = 'bottom', children, className, style }: Props) => {
+export const Tooltip = ({ content, custom, position, triggeredBy = 'hover', children, className, style }: Props) => {
   const api = useTooltip()
+
+  function show(e: React.MouseEvent<HTMLDivElement, MouseEvent>, hideOnClickOutside?: boolean) {
+    api.show({
+      content: custom ? content : <DefaultTooltip><Text appearance="microtext1">{content}</Text></DefaultTooltip>,
+      anchor: e.target as HTMLElement,
+      position,
+      hideOnClickOutside,
+    })
+  }
   return (
     <div
-      onMouseEnter={e => api.show(
-        custom ? content : <DefaultTooltip><Text appearance="microtext1">{content}</Text></DefaultTooltip>,
-        e.target as HTMLElement,
-        position,
-      )}
-      onMouseLeave={() => api.hide()}
+      {...(triggeredBy === 'hover' ? { onMouseEnter: show, onMouseLeave: () => api.hide() } : { onClick: (e) => show(e, true) })}
       className={className}
       style={style}
+      tabIndex={triggeredBy === 'click' ? 0 : undefined}
     >
       {children}
     </div>

package/src/components/Tooltip/TooltipAPI.ts

@@ -1,40 +1,77 @@
 import { animationTimeMS } from './style'
-import { TooltipPosition } from './types'
+import { ShowOptions } from './types'
 
+const MARGIN_TO_CORNERS_PX = 10
+
+function isRelative(element: HTMLElement) {
+  return ['relative', 'absolute', 'fixed'].includes(element.computedStyleMap().get('position')?.toString() ?? '')
+}
 
 export class TooltipAPI {
   private tooltipRef: React.RefObject<HTMLDivElement>
   private setContent: React.Dispatch<React.SetStateAction<React.ReactNode>>
   private hideTimeoutId: number | undefined
+  private clickListener: ((e: MouseEvent) => void) | undefined
+  private relativeTo: HTMLElement | undefined
 
   constructor(tooltipRef: React.RefObject<HTMLDivElement>, setContent: React.Dispatch<React.SetStateAction<React.ReactNode>>) {
     this.tooltipRef = tooltipRef
     this.setContent = setContent
   }
 
-  show(content: React.ReactNode, anchor: HTMLElement, position: TooltipPosition): void {
+  private computeRelativeTo() {
+    if (this.relativeTo) return
+    this.relativeTo = this.tooltipRef.current?.parentElement as HTMLElement
+    while (this.relativeTo && this.relativeTo !== document.body && !isRelative(this.relativeTo)) {
+      this.relativeTo = this.relativeTo.parentElement as HTMLElement
+    }
+  }
+
+  show({ content, anchor, position = 'bottom', hideOnClickOutside }: ShowOptions): void {
     window.clearTimeout(this.hideTimeoutId)
     this.hideTimeoutId = undefined
+    if (this.clickListener) document.removeEventListener('click', this.clickListener)
     this.setContent(content)
     setTimeout(() => {
       if (!this.tooltipRef.current) return
-      const rect = anchor.getClientRects()[0]
+      const anchorRect = anchor.getClientRects()[0]
       this.tooltipRef.current.classList.add('visible')
       const tooltipWidth = this.tooltipRef.current.clientWidth
       const tooltipHeight = this.tooltipRef.current.clientHeight
       let top = 0
       let left = 0
       if (position === 'left' || position === 'right') {
-        top = rect.top + rect.height / 2 - tooltipHeight / 2
-        if (position === 'left') left = rect.left - tooltipWidth
-        else left = rect.left + rect.width
+        top = anchorRect.top + anchorRect.height / 2 - tooltipHeight / 2
+        if (position === 'left') left = anchorRect.left - tooltipWidth
+        else left = anchorRect.left + anchorRect.width
       } else {
-        left = rect.left + rect.width / 2 - tooltipWidth / 2
-        if (position === 'top') top = rect.top - tooltipHeight
-        else top = rect.top + rect.height
+        left = anchorRect.left + anchorRect.width / 2 - tooltipWidth / 2
+        if (position === 'top') top = anchorRect.top - tooltipHeight
+        else top = anchorRect.top + anchorRect.height
+      }
+      // takes the parent the tooltip is positioned relative to into consideration
+      this.computeRelativeTo()
+      const relativeRect = this.relativeTo?.getClientRects()[0] ?? { top: 0, left: 0 }
+      top -= relativeRect.top
+      left -= relativeRect.left
+      // adjusts positions in order to avoid overflowing the window and leaving a margin to the corners
+      if (top <= 0) top += MARGIN_TO_CORNERS_PX
+      else if (top + tooltipHeight >= document.body.clientHeight - MARGIN_TO_CORNERS_PX) {
+        top = document.body.clientHeight - MARGIN_TO_CORNERS_PX + tooltipHeight
+      }
+      if (left <= 0) left += MARGIN_TO_CORNERS_PX
+      else if (left + tooltipWidth >= document.body.clientWidth - MARGIN_TO_CORNERS_PX) {
+        left = document.body.clientWidth - MARGIN_TO_CORNERS_PX - tooltipWidth
       }
       this.tooltipRef.current.style.top = `${top}px`
       this.tooltipRef.current.style.left = `${left}px`
+      if (hideOnClickOutside) {
+        this.clickListener = (e: MouseEvent) => {
+          if (this.tooltipRef.current?.contains(e.target as HTMLElement)) return
+          this.hide()
+        }
+        document.addEventListener('click', this.clickListener)
+      }
     }, 10)
   }
 
@@ -42,5 +79,6 @@ export class TooltipAPI {
     if (!this.tooltipRef.current) return
     this.tooltipRef.current.classList.remove('visible')
     this.hideTimeoutId = window.setTimeout(() => this.setContent(undefined), animationTimeMS)
+    if (this.clickListener) document.removeEventListener('click', this.clickListener)
   }
 }

package/src/components/Tooltip/context.tsx

@@ -12,7 +12,7 @@ export const TooltipProvider = ({ children }: Required<WithChildren>) => {
   return (
     <Context.Provider value={api}>
       {children}
-      <TooltipBox ref={ref} aria-hidden>{content}</TooltipBox>
+      <TooltipBox ref={ref}>{content}</TooltipBox>
     </Context.Provider>
   )
 }

package/src/components/Tooltip/style.tsx

@@ -5,7 +5,6 @@ export const animationTimeMS = 300
 
 export const TooltipBox = styled.div`
   position: absolute;
-  pointer-events: none;
   opacity: 0;
   transition: opacity ${animationTimeMS/ 1000}s;
   top: 0;

package/src/components/Tooltip/types.ts

@@ -6,3 +6,10 @@ export interface BoxPosition {
   y1: number,
   y2: number,
 }
+
+export interface ShowOptions {
+  content: React.ReactNode,
+  anchor: HTMLElement,
+  position?: TooltipPosition,
+  hideOnClickOutside?: boolean,
+}

package/src/features.ts

@@ -16,4 +16,5 @@ export const defaultFeatures: AIWidgetFeatures = {
   workspace: true,
   knowledgeSource: true,
   agent: true,
+  chatHistory: true,
 }

package/src/layout.css

@@ -44,6 +44,12 @@
       padding: 0 10px;
       width: calc(100% - 20px);
     }
+    .chat-window.narrow {
+      width: auto;
+    }
+    .chat-right-panel {
+      display: none;
+    }
     .home-page {
       .title {
         margin-top: 40px;

package/src/state/ChatEntry.ts

@@ -1,4 +1,5 @@
 import { pull } from 'lodash'
+import { LabeledWithImage } from './types'
 
 export interface SerializableAction {
   title: string,
@@ -25,14 +26,14 @@ export interface KnowledgeSource {
 
 export interface TextChatEntry {
   type: 'text' | 'md',
-  agent: 'bot' | 'user' | 'system',
+  agentType: 'bot' | 'user' | 'system',
   // image?: string,
   actions?: ChatAction[],
   subtitle?: string,
   content: string,
   knowledgeSources?: KnowledgeSource[],
   updated?: string,
-  agentId?: string,
+  agent?: LabeledWithImage,
   messageId?: string,
   error?: string,
   // customInput?: CustomInputResponse,
@@ -63,7 +64,7 @@ export class ChatEntry {
 
   static createUserEntry(content: string) {
     return new ChatEntry({
-      agent: 'user',
+      agentType: 'user',
       type: 'text',
       content,
       updated: new Date().toISOString(),
@@ -71,7 +72,7 @@ export class ChatEntry {
   }
 
   static createStreamedBotEntry(abort: () => void) {
-    return new ChatEntry({ agent: 'bot', type: 'md', content: '' }, true, abort)
+    return new ChatEntry({ agentType: 'bot', type: 'md', content: '' }, true, abort)
   }
 
   setValue(value: TextChatEntry) {

package/src/state/ChatState.ts

@@ -1,15 +1,7 @@
 import { dropRight, last, pull } from 'lodash'
 import { ChatEntry } from './ChatEntry'
 import { ObservableState } from './ObservableState'
-
-interface Labeled {
-  id: string,
-  label: string,
-}
-
-interface LabeledWithImage extends Labeled {
-  image?: string,
-}
+import { Labeled, LabeledWithImage } from './types'
 
 export interface ChatProperties {
   label: string,

package/src/state/types.ts

@@ -0,0 +1,8 @@
+export interface Labeled {
+  id: string,
+  label: string,
+}
+
+export interface LabeledWithImage extends Labeled {
+  image?: string,
+}

package/src/types.ts

@@ -10,7 +10,7 @@ export interface ButtonAction extends WithStyle {
   icon?: React.ReactElement,
   color?: string,
   label: string,
-  onClick: () => void,
+  onClick: () => any,
 }
 
 export interface MinimizedActions {

package/src/utils/date.ts

@@ -28,6 +28,10 @@ export function useDateFormatter() {
   }
 }
 
+export function subtractDays(date: Date, numberOfDays: number) {
+  return new Date(date.getTime() - 24 * numberOfDays * 60 * 60000)
+}
+
 const dictionary = {
   en: {
     today: 'Today',

package/src/utils/download.ts

@@ -0,0 +1,12 @@
+export function download(filename: string, text: string) {
+  const element = document.createElement('a')
+  element.setAttribute('href', 'data:text/plain;charset=utf-8,' + encodeURIComponent(text))
+  element.setAttribute('download', filename)
+
+  element.style.display = 'none'
+  document.body.appendChild(element)
+
+  element.click()
+
+  document.body.removeChild(element)
+}

package/src/utils/knowledge-source.ts

@@ -1,4 +1,5 @@
-import { DocumentResponse } from '@stack-spot/portal-network/api/ai'
+import { DocumentResponse, SourceKnowledgeSource4, SourceProjectFile4, SourceStackAi } from '@stack-spot/portal-network/api/ai'
+import { KnowledgeSource } from '../state/ChatEntry'
 
 function attemptToParseMalFormedJson(str: string) {
   try {
@@ -41,3 +42,14 @@ export function extractCodeFromKSDocument(document: DocumentResponse): { languag
 
   return typeof document === 'object' ? { language: 'json', snippet: JSON.stringify(document, null, 2) } : { snippet: String(document) }
 }
+
+export function genericSourcesToKnowledgeSources(
+  sources: (SourceStackAi | SourceKnowledgeSource4 | SourceProjectFile4)[] | undefined,
+): KnowledgeSource[] | undefined {
+  return sources?.filter(s => s.type === 'knowledge_source').map(ks => ({
+    documentId: ks.document_id,
+    documentScore: ks.document_score,
+    name: ks.name,
+    slug: ks.slug,
+  }))
+}

package/src/views/Agents.tsx

@@ -83,8 +83,9 @@ export const Agents = () => {
 
 const AgentsPanel = () => {
   const t = useTranslate(dictionary)
+  const chat = useCurrentChat()
   
-  return <RightPanelTabs tabs={[
+  return <RightPanelTabs key={chat.id} tabs={[
     { title: t.personal, content: <AgentsTab key="personal" visibility="PERSONAL" /> },
     { title: t.builtin, content: <AgentsTab key="builtin" visibility="BUILT-IN" /> },
     { title: t.shared, content: <AgentsTab key="shared" visibility="SHARED" /> },

package/src/views/Chat/AgentInfo.tsx

@@ -1,17 +1,17 @@
 import { Text } from '@citric/core'
 import { MiniLogo } from '@stack-spot/portal-components/svg'
+import { LabeledWithImage } from '../../state/types'
 
 interface Props {
-  agentId?: string,
+  agent?: LabeledWithImage,
 }
 
-// todo: retrieve agent data and render accordingly
-// eslint-disable-next-line @typescript-eslint/no-unused-vars
-export const AgentInfo = ({ agentId: _agentId }: Props) => (
+export const AgentInfo = ({ agent }: Props) => (
   <>
-    <div className="agent-image-wrapper">
-      <MiniLogo className="agent-image" />
-    </div>
-    <Text appearance="body2">Stackspot AI</Text>
+    {agent?.image
+      ? <img src={agent.image} className="custom-agent-image" />
+      : <div className="default-image-wrapper"><MiniLogo className="agent-image" /></div>
+    }
+    <Text appearance="body2">{agent?.label || 'Stackspot AI'}</Text>
   </>
 )

package/src/views/Chat/ChatMessage.tsx

@@ -6,17 +6,17 @@ import { Dictionary, useTranslate } from '@stack-spot/portal-translate'
 import { useCallback, useMemo, useRef, useState } from 'react'
 import { Markdown } from '../../components/Markdown'
 import { useChatEntry, useWidget } from '../../context/hooks'
-import { useChatScrollToBottomEffect } from '../../hooks/chat-scroll'
 import { ChatEntry, TextChatEntry } from '../../state/ChatEntry'
 import { useDateFormatter } from '../../utils/date'
 import { AgentInfo } from './AgentInfo'
+import { useChatScrollToBottomEffect } from './chat-scroll'
 
 export const ChatMessage = ({ message, username }: { message: ChatEntry, username: string }) => {
   const t = useTranslate(dictionary)
   const [liked, setLiked] = useState<boolean | undefined>()
   const entry = useChatEntry(message)
   const dateFormatter = useDateFormatter()
-  const userInfo = entry.agent === 'user' ? <Avatar size="xs">{username}</Avatar> : <AgentInfo agentId={entry.agentId} />
+  const userInfo = entry.agentType === 'user' ? <Avatar size="xs">{username}</Avatar> : <AgentInfo agent={entry.agent} />
   const date = new Date(entry.updated ?? '')
   const shouldShowDate = entry.updated && !isNaN(date.getTime())
   const ref = useRef<HTMLLIElement>(null)
@@ -55,7 +55,7 @@ export const ChatMessage = ({ message, username }: { message: ChatEntry, usernam
   }, [entry.messageId, liked])
   
   return (entry.content || entry.error) && (
-    <li className={entry.agent} ref={ref}>
+    <li className={entry.agentType} ref={ref}>
       <div className="chat-message">
         <div className="user-info">{userInfo}</div>
         {entry.content && <div className="message-content">
@@ -77,7 +77,7 @@ export const ChatMessage = ({ message, username }: { message: ChatEntry, usernam
         ))}</ul>
       </div>}
       <div className="message-footer">
-        {entry.agent === 'bot' && entry.messageId && !entry.error && <div className="message-actions">
+        {entry.agentType === 'bot' && entry.messageId && !entry.error && <div className="message-actions">
           <IconButton title={t.like} aria-label={t.like} onClick={like}>
             {liked === true ? <LikeFill /> : <Like />}
           </IconButton>

package/src/{hooks → views/Chat}/chat-scroll.ts

@@ -1,4 +1,4 @@
-import { useEffect } from 'react'
+import { useEffect, useRef } from 'react'
 
 /**
  * Scrolls the closest chat (upwards in the tree) to its bottom.
@@ -6,9 +6,12 @@ import { useEffect } from 'react'
  * @param deps when the deps changes, the chat is scrolled.
  */
 export function useChatScrollToBottomEffect(ref: React.RefObject<HTMLElement>, deps: any[]) {
+  const prevScrollTop = useRef(0)
+
   useEffect(() => {
     const chat = ref.current?.closest('.chat-content')
-    if (!chat) return
-    chat.scrollTop = chat.scrollHeight
+    if (!chat || chat.scrollTop < prevScrollTop.current) return
+    chat.scrollTop = chat.scrollHeight - chat.clientHeight
+    prevScrollTop.current = chat.scrollTop
   }, deps)
 }

package/src/views/Chat/styled.ts

@@ -76,7 +76,7 @@ export const ChatList = styled.ul`
         flex-direction: column;
         gap: 4px;
     
-        .agent-image-wrapper {
+        .default-image-wrapper {
           width: 24px;
           height: 24px;
           border-radius: 4px;
@@ -90,6 +90,12 @@ export const ChatList = styled.ul`
             height: 18px;
           }
         }
+
+        .custom-agent-image {
+          width: 24px;
+          height: 24px;
+          border-radius: 50%;
+        }
       }
     }
 
@@ -100,6 +106,7 @@ export const ChatList = styled.ul`
         display: flex;
         flex-direction: row;
         gap: 8px;
+        align-items: center;
         
         .message-content {
           padding: 10px;

package/src/views/ChatHistory/ChatHistoryPanel.tsx

@@ -0,0 +1,28 @@
+import { aiClient } from '@stack-spot/portal-network'
+import InfiniteScroll from 'react-infinite-scroll-component'
+import { HistoryList } from '../../components/HistoryList'
+import { MessageInterceptor } from '../../state/ChatState'
+import { HistoryItem } from './HistoryItem'
+
+export const ChatHistoryPanel = ({ interceptors }: { interceptors: MessageInterceptor[] }) => {
+  const [chats, { fetchNextPage, hasNextPage }] = aiClient.chats.useInfiniteQuery({ size: 40 })
+  return (
+    <div id="chatHistoryList" style={{ height: '100%', overflow: 'auto' }}>
+      <InfiniteScroll
+        scrollableTarget="chatHistoryList"
+        dataLength={chats.length}
+        next={fetchNextPage}
+        hasMore={hasNextPage}
+        loader={<div></div>}
+      >
+        <HistoryList
+          items={chats}
+          getDate={c => new Date(c.updated || c.created || '')}
+          keygen={c => c.id}
+          renderItem={c => <HistoryItem item={c} interceptors={interceptors} />}
+          style={{ marginRight: '6px' }}
+        />
+      </InfiniteScroll>
+    </div>
+  )
+}