@dxos/plugin-automation 0.7.4 → 0.7.5-labs.071a3e2

package/src/components/Thread/Thread.tsx

@@ -0,0 +1,156 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import React, { type KeyboardEventHandler, useCallback, useMemo, useRef, useState } from 'react';
+
+import { type Message } from '@dxos/artifact';
+import { IconButton, Input, useTranslation } from '@dxos/react-ui';
+import { Spinner } from '@dxos/react-ui-sfx';
+import { mx } from '@dxos/react-ui-theme';
+
+import { ScrollContainer, type ScrollController } from './ScrollContainer';
+import { ThreadMessage, type ThreadMessageProps } from './ThreadMessage';
+import { AUTOMATION_PLUGIN } from '../../meta';
+
+export type ThreadProps = {
+  messages?: Message[];
+  streaming?: boolean;
+  onSubmit?: (message: string) => void;
+  onStop?: () => void;
+} & Pick<ThreadMessageProps, 'collapse' | 'debug' | 'onSuggest' | 'onDelete'>;
+
+// TODO(burdon): Factor out scroll logic.
+export const Thread = ({
+  messages,
+  streaming,
+  collapse,
+  debug,
+  onSubmit,
+  onStop,
+  onSuggest,
+  onDelete,
+}: ThreadProps) => {
+  const { t } = useTranslation(AUTOMATION_PLUGIN);
+  const scroller = useRef<ScrollController>(null);
+
+  const [text, setText] = useState('');
+  const handleKeyDown = useCallback<KeyboardEventHandler<HTMLInputElement>>(
+    (ev) => {
+      switch (ev.key) {
+        case 'Escape': {
+          setText('');
+          break;
+        }
+
+        case 'Enter': {
+          const value = text.trim();
+          if (value.length > 0) {
+            scroller.current?.scrollToBottom();
+            onSubmit?.(value);
+            setText('');
+          }
+          break;
+        }
+      }
+    },
+    [text],
+  );
+
+  /**
+   * Reduce message blocks into collections of messages that contain related contiguous blocks.
+   * For example, collapse all tool request/response pairs into a single message.
+   */
+  // TODO(dmaretskyi): This needs to be a separate type: `id` is not a valid ObjectId, this needs to accommodate messageId for deletion.
+  const { messages: lines = [] } = useMemo(() => {
+    if (!collapse) {
+      return { messages: messages ?? [] };
+    }
+
+    return (messages ?? []).reduce<{ messages: Message[]; current?: Message }>(
+      ({ current, messages }, message) => {
+        let i = 0;
+        for (const block of message.content) {
+          switch (block.type) {
+            case 'tool_use':
+            case 'tool_result': {
+              if (current) {
+                current.content.push(block);
+              } else {
+                current = {
+                  id: [message.id, i].join('_'),
+                  role: message.role,
+                  content: [block],
+                };
+                messages.push(current);
+              }
+              break;
+            }
+
+            case 'text':
+            default: {
+              current = undefined;
+              messages.push({
+                id: [message.id, i].join('_'),
+                role: message.role,
+                content: [block],
+              });
+              break;
+            }
+          }
+
+          i++;
+        }
+
+        return { current, messages };
+      },
+      { messages: [] as Message[] },
+    );
+  }, [messages, collapse]);
+
+  return (
+    <div className='flex flex-col grow overflow-hidden'>
+      <ScrollContainer ref={scroller} classNames='py-2 gap-2 overflow-x-hidden'>
+        {lines.map((message) => (
+          <ThreadMessage
+            key={message.id}
+            classNames='px-4'
+            message={message}
+            collapse={collapse}
+            debug={debug}
+            onSuggest={onSuggest}
+            onDelete={onDelete}
+          />
+        ))}
+      </ScrollContainer>
+
+      {onSubmit && (
+        <div className='flex p-4 gap-3 items-center'>
+          <Spinner active={streaming} />
+          <Input.Root>
+            <Input.TextInput
+              autoFocus
+              classNames='px-2 baseSurface rounded'
+              placeholder={t('chat input placeholder')}
+              value={text}
+              onChange={(ev) => setText(ev.target.value)}
+              onKeyDown={handleKeyDown}
+            />
+          </Input.Root>
+          {onStop && (
+            <IconButton
+              disabled={!streaming}
+              classNames={mx('!p-1 !opacity-20 transition', streaming && '!opacity-80')}
+              variant='ghost'
+              size={5}
+              onClick={onStop}
+              icon='ph--x--regular'
+              label={t('chat stop')}
+              iconOnly
+            />
+          )}
+        </div>
+      )}
+    </div>
+  );
+};

package/src/components/Thread/ThreadMessage.tsx

@@ -0,0 +1,203 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import React, { type FC } from 'react';
+
+import { type MessageContentBlock, type Message } from '@dxos/artifact';
+import { invariant } from '@dxos/invariant';
+import { log } from '@dxos/log';
+import { Button, ButtonGroup, Icon, type ThemedClassName } from '@dxos/react-ui';
+import { Json } from '@dxos/react-ui-syntax-highlighter';
+import { mx } from '@dxos/react-ui-theme';
+import { safeParseJson } from '@dxos/util';
+
+import { StatusLine } from './StatusLine';
+import { ToggleContainer } from './ToggleContainer';
+import { MarkdownViewer } from '../MarkdownViewer';
+
+export type ThreadMessageProps = ThemedClassName<{
+  message: Message;
+  collapse?: boolean;
+  debug?: boolean;
+  onSuggest?: (text: string) => void;
+  onDelete?: (id: string) => void;
+}>;
+
+export const ThreadMessage: FC<ThreadMessageProps> = ({
+  classNames,
+  message,
+  collapse,
+  debug,
+  onSuggest,
+  onDelete,
+}) => {
+  if (typeof message !== 'object') {
+    return <div className={mx(classNames)}>{message}</div>;
+  }
+
+  const { role, content = [] } = message;
+
+  // TODO(burdon): Factor out tool blocks.
+  const toolBlocks = content.filter((block) => block.type === 'tool_use' || block.type === 'tool_result');
+  if (collapse && toolBlocks.length > 0) {
+    let request: (MessageContentBlock & { type: 'tool_use' }) | undefined;
+    const json: any[] = [];
+    const lines = toolBlocks.map((block) => {
+      switch (block.type) {
+        case 'tool_use': {
+          request = block;
+          json.push(block);
+          // TODO(burdon): Get plugin name.
+          return `Calling ${block.name}...`;
+        }
+
+        case 'tool_result': {
+          if (!request) {
+            log.warn('unexpected message', { tool: block });
+            return 'Error';
+          }
+
+          json.push(block);
+          return `Processed ${request.name}`;
+        }
+
+        default: {
+          request = undefined;
+          return 'Error';
+        }
+      }
+    });
+
+    return (
+      <div className={mx('flex', classNames)}>
+        <div className='w-full p-1 px-2 overflow-hidden rounded-md bg-baseSurface'>
+          <ToggleContainer title={<StatusLine lines={lines} autoAdvance />} toggle>
+            <Json data={json} classNames='!p-1 text-xs' />
+          </ToggleContainer>
+        </div>
+      </div>
+    );
+  }
+
+  return (
+    <div className={mx('flex flex-col shrink-0 gap-2')}>
+      {debug && (
+        <div className='text-xs text-subdued'>
+          {message.id}{' '}
+          {onDelete && (
+            <span className='cursor-pointer underline' onClick={() => onDelete(message.id)}>
+              delete
+            </span>
+          )}
+        </div>
+      )}
+      {content.map((block, idx) => (
+        <div key={idx} className={mx('flex', classNames, block.type === 'text' && role === 'user' && 'justify-end')}>
+          <Block role={role} block={block} onSuggest={onSuggest ?? (() => {})} />
+        </div>
+      ))}
+    </div>
+  );
+};
+
+const Block = ({
+  block,
+  role,
+  onSuggest,
+}: Pick<Message, 'role'> & { block: MessageContentBlock; onSuggest: (text: string) => void }) => {
+  const Component = componentMap[block.type] ?? componentMap.default;
+  return (
+    <div
+      className={mx(
+        'p-1 px-2 overflow-hidden rounded-md',
+        (block.type !== 'text' || block.disposition) && 'w-full bg-baseSurface',
+        block.type === 'text' && role === 'user' && 'bg-primary-200 dark:bg-primary-500',
+      )}
+    >
+      <Component block={block} onSuggest={onSuggest} />
+    </div>
+  );
+};
+
+const titles: Record<string, string> = {
+  ['cot' as const]: 'Chain of thought',
+  ['artifact' as const]: 'Artifact',
+  ['tool_use' as const]: 'Tool request',
+  ['tool_result' as const]: 'Tool result',
+};
+
+type BlockComponent = FC<{ block: MessageContentBlock; onSuggest: (text: string) => void }>;
+
+const componentMap: Record<string, BlockComponent> = {
+  text: ({ block }) => {
+    invariant(block.type === 'text');
+    const title = block.disposition ? titles[block.disposition] : undefined;
+    if (!title) {
+      return (
+        <MarkdownViewer content={block.text} classNames={[block.disposition === 'cot' && 'text-sm text-subdued']} />
+      );
+    }
+
+    return (
+      <ToggleContainer
+        title={title}
+        icon={
+          block.pending ? (
+            <Icon icon={'ph--circle-notch--regular'} classNames='text-subdued ml-2 animate-spin' size={4} />
+          ) : undefined
+        }
+        defaultOpen={block.disposition === 'cot'}
+        toggle
+      >
+        <MarkdownViewer content={block.text} classNames={[block.disposition === 'cot' && 'text-sm text-subdued']} />
+      </ToggleContainer>
+    );
+  },
+
+  json: ({ block, onSuggest }) => {
+    invariant(block.type === 'json');
+
+    switch (block.disposition) {
+      case 'suggest': {
+        const { text = '' }: { text: string } = safeParseJson(block.json ?? '{}') ?? ({} as any);
+        return <Button onClick={() => onSuggest(text)}>{text}</Button>;
+      }
+
+      case 'select': {
+        const { options = [] }: { options: string[] } = safeParseJson(block.json ?? '{}') ?? ({} as any);
+        return (
+          <ButtonGroup>
+            {options.map((option) => (
+              <Button key={option} onClick={() => onSuggest(option)}>
+                {option}
+              </Button>
+            ))}
+          </ButtonGroup>
+        );
+      }
+
+      default: {
+        const title = block.disposition ? titles[block.disposition] : undefined;
+        return (
+          <ToggleContainer title={title ?? 'JSON'} toggle>
+            <Json data={safeParseJson(block.json ?? block)} classNames='!p-1 text-xs' />
+          </ToggleContainer>
+        );
+      }
+    }
+  },
+
+  default: ({ block }) => {
+    let title = titles[block.type];
+    if (block.type === 'tool_use') {
+      title = `Tool [${block.name}]`; // TODO(burdon): Get label from tool.
+    }
+
+    return (
+      <ToggleContainer title={title ?? 'JSON'} toggle>
+        <Json data={block} classNames='!p-1 text-xs' />
+      </ToggleContainer>
+    );
+  },
+};

package/src/components/Thread/ToggleContainer.stories.tsx

@@ -0,0 +1,111 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import '@dxos-theme';
+
+import { computed, type ReadonlySignal, signal } from '@preact/signals-core';
+import { type StoryObj, type Meta } from '@storybook/react';
+import React, { useEffect, useMemo, useState } from 'react';
+
+import { faker } from '@dxos/random';
+import { Icon, Input, Toolbar } from '@dxos/react-ui';
+import { mx } from '@dxos/react-ui-theme';
+import { withLayout, withTheme, withSignals } from '@dxos/storybook-utils';
+
+import { ToggleContainer, type ToggleContainerProps } from './ToggleContainer';
+import { MarkdownViewer } from '../MarkdownViewer';
+
+class Generator {
+  private readonly _current = signal<string>(faker.lorem.sentence(5));
+  private readonly _lines = signal<string[]>([]);
+  private _running: NodeJS.Timeout | undefined;
+
+  readonly count = computed(() => this._lines.value.length);
+  readonly text: ReadonlySignal<string[]> = computed(() => [...this._lines.value, this._current.value]);
+
+  start() {
+    this.stop();
+    this._running = setInterval(() => {
+      if (this._current.value.length > 0) {
+        this._current.value += ' ';
+      }
+      this._current.value += faker.lorem.words(Math.ceil(Math.random() * 2));
+      if (Math.random() > 0.95) {
+        this._lines.value = [...this._lines.value, this._current.value + '.'];
+        this._current.value = '';
+      }
+    }, 100);
+  }
+
+  stop() {
+    if (this._running) {
+      clearInterval(this._running);
+      this._running = undefined;
+    }
+  }
+}
+
+const Render = ({ shrinkX, ...props }: ToggleContainerProps) => {
+  const generator = useMemo(() => new Generator(), []);
+  const [running, setRunning] = useState(false);
+  useEffect(() => {
+    if (running) {
+      generator.start();
+    } else {
+      generator.stop();
+    }
+  }, [running]);
+
+  return (
+    <div className='flex flex-col w-[500px]'>
+      <Toolbar.Root classNames='p-4'>
+        <Input.Root>
+          <Input.Switch checked={running} onCheckedChange={(checked) => setRunning(checked)} />
+        </Input.Root>
+        <div className='grow' />
+        <div>{generator.count.value}</div>
+      </Toolbar.Root>
+      <div className='flex p-4'>
+        <div className={mx('border border-border rounded-md p-2', !shrinkX && 'w-full')}>
+          <ToggleContainer
+            {...props}
+            shrinkX={shrinkX}
+            toggle
+            icon={
+              running ? (
+                <Icon icon={'ph--circle-notch--regular'} classNames='text-subdued ml-2 animate-spin' size={4} />
+              ) : undefined
+            }
+          >
+            <MarkdownViewer classNames='text-sm' content={generator.text.value.join('\n\n')} />
+          </ToggleContainer>
+        </div>
+      </div>
+    </div>
+  );
+};
+
+const meta: Meta<typeof ToggleContainer> = {
+  title: 'plugins/plugin-automation/ToggleContainer',
+  component: ToggleContainer,
+  render: Render,
+  decorators: [withSignals, withTheme, withLayout({ fullscreen: true, classNames: 'justify-center bg-baseSurface' })],
+};
+
+export default meta;
+
+type Story = StoryObj<typeof ToggleContainer>;
+
+export const Default: Story = {
+  args: {
+    title: 'Markdown',
+  },
+};
+
+export const Shrink: Story = {
+  args: {
+    title: 'Markdown',
+    shrinkX: true,
+  },
+};

package/src/components/Thread/ToggleContainer.tsx

@@ -0,0 +1,103 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import React, { type JSX, type PropsWithChildren, useEffect, useState } from 'react';
+
+import { Icon, type ThemedClassName } from '@dxos/react-ui';
+import { mx } from '@dxos/react-ui-theme';
+
+export type ToggleContainerProps = ThemedClassName<
+  PropsWithChildren<{
+    title?: string | JSX.Element;
+    icon?: JSX.Element;
+    toggle?: boolean;
+    defaultOpen?: boolean;
+    duration?: number;
+    /** Should shrink the width when closed. */
+    shrinkX?: boolean;
+  }>
+>;
+
+// TODO(burdon): Externalize toggle state.
+export const ToggleContainer = ({
+  title,
+  icon,
+  toggle,
+  defaultOpen,
+  duration = 400,
+  shrinkX = false,
+  children,
+  classNames,
+}: ToggleContainerProps) => {
+  const [expand, setExpand] = useState(defaultOpen || !toggle);
+  const [expandX, setExpandX] = useState(shrinkX ? expand : true);
+  const [expandY, setExpandY] = useState(expand);
+
+  useEffect(() => {
+    let t: NodeJS.Timeout;
+    if (expand) {
+      if (shrinkX) {
+        setExpandX(true);
+      }
+      t = setTimeout(
+        () => {
+          setExpandY(true);
+        },
+        shrinkX ? duration : 0,
+      );
+    } else {
+      setExpandY(false);
+      if (shrinkX) {
+        t = setTimeout(() => {
+          setExpandX(false);
+        }, duration);
+      }
+    }
+
+    return () => clearTimeout(t);
+  }, [expand]);
+
+  return (
+    <div className={mx('overflow-hidden', classNames)}>
+      {title && (
+        <div
+          className='flex gap-1 py-1 items-center text-sm text-subdued cursor-pointer select-none'
+          onClick={toggle ? () => setExpand((open) => !open) : undefined}
+        >
+          {toggle && (
+            <div className='flex w-[24px] h-[24px] items-center justify-center'>
+              <Icon
+                size={4}
+                icon={'ph--caret-right--regular'}
+                style={{ transitionDuration: `${shrinkX ? duration * 2 : duration}ms` }}
+                classNames={['transition transition-transform ease-in-out', expand ? 'rotate-90' : 'transform-none']}
+              />
+            </div>
+          )}
+          <div className='flex-1 pis-1 pie-1 truncate'>{title}</div>
+          {icon}
+        </div>
+      )}
+      <div
+        style={{ transitionDuration: `${duration}ms` }}
+        className={mx(
+          'grid transition-[grid-template-columns] ease-in-out',
+          expandX ? 'grid-cols-[1fr]' : 'grid-cols-[0fr]',
+        )}
+      >
+        <div className='overflow-hidden'>
+          <div
+            style={{ transitionDuration: `${duration}ms` }}
+            className={mx(
+              'grid transition-[grid-template-rows] ease-in-out',
+              expandY ? 'grid-rows-[1fr]' : 'grid-rows-[0fr]',
+            )}
+          >
+            <div className={mx('flex overflow-hidden transition-opacity')}>{children}</div>
+          </div>
+        </div>
+      </div>
+    </div>
+  );
+};

package/src/components/Thread/index.ts

@@ -0,0 +1,5 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+export * from './Thread';

package/src/components/TriggerEditor/TriggerEditor.stories.tsx

@@ -7,9 +7,8 @@ import '@dxos-theme';
 import { type Meta } from '@storybook/react';
 import React, { useEffect, useState } from 'react';
 
-import { FunctionTrigger, TriggerKind } from '@dxos/functions';
+import { FunctionType, FunctionTrigger, TriggerKind } from '@dxos/functions';
 import { create } from '@dxos/live-object';
-import { FunctionType } from '@dxos/plugin-script/types';
 import { useSpaces } from '@dxos/react-client/echo';
 import { withClientProvider } from '@dxos/react-client/testing';
 import { withLayout, withTheme } from '@dxos/storybook-utils';