@dxos/plugin-assistant 0.8.3 → 0.8.4-main.03d5cd7b56

package/src/components/Prompt/Prompt.tsx

@@ -1,124 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type Extension, Prec } from '@codemirror/state';
-import React, { forwardRef, useImperativeHandle } from 'react';
-
-import { type ThemedClassName, useThemeContext } from '@dxos/react-ui';
-import {
-  createBasicExtensions,
-  createThemeExtensions,
-  keymap,
-  useTextEditor,
-  type BasicExtensionsOptions,
-  type UseTextEditorProps,
-} from '@dxos/react-ui-editor';
-import { mx } from '@dxos/react-ui-theme';
-import { isNonNullable } from '@dxos/util';
-
-import { autocompleteExtension, type AutocompleteOptions } from './autocomplete';
-import { promptReferences, type ReferencesProvider } from './references';
-
-// TODO(burdon): Handle object references.
-
-export interface PromptController {
-  focus(): void;
-  setText(text: string): void;
-}
-
-export type PromptProps = ThemedClassName<
-  {
-    extensions?: Extension;
-    references?: ReferencesProvider;
-    onOpenChange?: (open: boolean) => void;
-  } & AutocompleteOptions &
-    Pick<UseTextEditorProps, 'autoFocus'> &
-    Pick<BasicExtensionsOptions, 'lineWrapping' | 'placeholder'>
->;
-
-export const Prompt = forwardRef<PromptController, PromptProps>(
-  (
-    {
-      classNames,
-      extensions,
-      references,
-      autoFocus,
-      lineWrapping = false,
-      placeholder,
-      onSubmit,
-      onSuggest,
-      onCancel,
-      onOpenChange,
-    },
-    forwardRef,
-  ) => {
-    const { themeMode } = useThemeContext();
-    const { parentRef, view } = useTextEditor(
-      {
-        debug: true,
-        autoFocus,
-        extensions: [
-          autocompleteExtension({ onSubmit, onSuggest, onCancel }),
-          createBasicExtensions({
-            bracketMatching: false,
-            lineWrapping,
-            placeholder,
-          }),
-          createThemeExtensions({ themeMode }),
-          references ? promptReferences({ provider: references }) : [],
-          Prec.highest(
-            keymap.of([
-              {
-                key: 'cmd-ArrowUp',
-                preventDefault: true,
-                run: () => {
-                  onOpenChange?.(true);
-                  return true;
-                },
-              },
-              {
-                key: 'cmd-ArrowDown',
-                preventDefault: true,
-                run: () => {
-                  onOpenChange?.(false);
-                  return true;
-                },
-              },
-            ]),
-          ),
-          extensions,
-        ].filter(isNonNullable),
-      },
-      [themeMode, extensions, onSubmit, onSuggest],
-    );
-
-    // Expose editor view.
-    useImperativeHandle(
-      forwardRef,
-      () => {
-        return {
-          focus: () => {
-            view?.focus();
-          },
-          setText: (text: string) => {
-            view?.dispatch({
-              changes: {
-                from: 0,
-                to: view.state.doc.length,
-                insert: text,
-              },
-              selection: {
-                anchor: text.length,
-                head: text.length,
-              },
-            });
-          },
-        };
-      },
-      [view, onSubmit],
-    );
-
-    return <div ref={parentRef} className={mx('w-full', classNames)} />;
-  },
-);

package/src/components/Prompt/PromptBar.stories.tsx

@@ -1,69 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import '@dxos-theme';
-
-import { type StoryObj, type Meta } from '@storybook/react';
-import React, { useEffect, useState } from 'react';
-
-import { IntentPlugin, SettingsPlugin } from '@dxos/app-framework';
-import { withPluginManager } from '@dxos/app-framework/testing';
-import { log } from '@dxos/log';
-import { TranscriptionPlugin } from '@dxos/plugin-transcription';
-import { withTheme, withLayout } from '@dxos/storybook-utils';
-
-import { PromptBar } from './PromptBar';
-import translations from '../../translations';
-
-const meta: Meta<typeof PromptBar> = {
-  title: 'plugins/plugin-assistant/PromptBar',
-  component: PromptBar,
-  decorators: [
-    withPluginManager({
-      plugins: [IntentPlugin(), SettingsPlugin(), TranscriptionPlugin()],
-    }),
-    withTheme,
-    withLayout(),
-  ],
-  parameters: {
-    layout: 'centered',
-    translations,
-    controls: { disable: true },
-  },
-};
-
-export default meta;
-
-type Story = StoryObj<typeof PromptBar>;
-
-export const Default: Story = {
-  // args: {
-  //   classNames: 'w-96 p-4 rounded outline outline-gray-200',
-  // },
-};
-
-export const Toolbar: Story = {
-  render: (args) => {
-    const [processing, setProcessing] = useState(false);
-    useEffect(() => {
-      let t: NodeJS.Timeout;
-      if (processing) {
-        t = setTimeout(() => setProcessing(false), 10_000);
-      }
-      return () => clearTimeout(t);
-    }, [processing]);
-    log.info('processing', { processing });
-
-    return (
-      <PromptBar
-        classNames='w-[25rem] p-1 overflow-hidden border border-gray-200 rounded'
-        microphone
-        processing={processing}
-        onSubmit={() => setProcessing(true)}
-        onCancel={() => setProcessing(false)}
-        {...args}
-      />
-    );
-  },
-};

package/src/components/Prompt/PromptBar.tsx

@@ -1,105 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { forwardRef, useState } from 'react';
-
-import { useVoiceInput } from '@dxos/plugin-transcription';
-import { Icon, IconButton, type ThemedClassName, Tooltip, useForwardedRef, useTranslation } from '@dxos/react-ui';
-import { Spinner } from '@dxos/react-ui-sfx';
-import { errorText, mx } from '@dxos/react-ui-theme';
-
-import { Prompt, type PromptController, type PromptProps } from './Prompt';
-import { ASSISTANT_PLUGIN } from '../../meta';
-
-export type PromptBarProps = ThemedClassName<
-  Pick<
-    PromptProps,
-    | 'extensions'
-    | 'references'
-    | 'placeholder'
-    | 'lineWrapping'
-    | 'onSubmit'
-    | 'onSuggest'
-    | 'onCancel'
-    | 'onOpenChange'
-  > & {
-    processing?: boolean;
-    error?: Error;
-    microphone?: boolean;
-  }
->;
-
-export const PromptBar = forwardRef<PromptController, PromptBarProps>(
-  ({ classNames, placeholder, processing, error, microphone, references, onCancel, ...props }, forwardedRef) => {
-    const { t } = useTranslation(ASSISTANT_PLUGIN);
-    const promptRef = useForwardedRef<PromptController>(forwardedRef);
-    const [active, setActive] = useState(false);
-
-    // TODO(burdon): Configure capability in TranscriptionPlugin.
-    const { recording } = useVoiceInput({
-      active,
-      onUpdate: (text) => {
-        promptRef.current?.setText(text);
-        promptRef.current?.focus();
-      },
-    });
-
-    return (
-      <div
-        className={mx(
-          'shrink-0 w-full grid grid-cols-[var(--rail-action)_1fr_var(--rail-action)] overflow-hidden',
-          classNames,
-        )}
-      >
-        <div className='flex w-[--rail-action] h-[--rail-action] items-center justify-center'>
-          {(error && (
-            <Tooltip.Trigger content={error.message} delayDuration={0}>
-              <Icon icon='ph--warning-circle--regular' classNames={errorText} size={5} />
-            </Tooltip.Trigger>
-          )) || <Spinner active={processing} />}
-        </div>
-        <Prompt
-          ref={promptRef}
-          autoFocus
-          classNames='pbs-2'
-          lineWrapping={true}
-          placeholder={placeholder ?? t('prompt placeholder')}
-          references={references}
-          onCancel={onCancel}
-          {...props}
-        />
-        {(onCancel || microphone) && (
-          <div className='flex w-[--rail-action] h-[--rail-action] items-center justify-center'>
-            {processing && onCancel && (
-              <IconButton
-                classNames='px-1.5'
-                variant='ghost'
-                size={5}
-                icon='ph--x--regular'
-                iconOnly
-                label={t('cancel processing button')}
-                onClick={onCancel}
-              />
-            )}
-            {!processing && microphone && (
-              <IconButton
-                classNames={mx('px-1.5', recording && 'bg-primary-500')}
-                variant='ghost'
-                size={5}
-                icon='ph--microphone--regular'
-                iconOnly
-                noTooltip
-                label={t('microphone button')}
-                onMouseDown={() => setActive(true)}
-                onMouseUp={() => setActive(false)}
-                onTouchStart={() => setActive(true)}
-                onTouchEnd={() => setActive(false)}
-              />
-            )}
-          </div>
-        )}
-      </div>
-    );
-  },
-);

package/src/components/Prompt/autocomplete.ts

@@ -1,218 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Prec, type Extension } from '@codemirror/state';
-import {
-  EditorView,
-  Decoration,
-  ViewPlugin,
-  keymap,
-  type DecorationSet,
-  type ViewUpdate,
-  WidgetType,
-} from '@codemirror/view';
-
-export type AutocompleteOptions = {
-  /**
-   * Callback triggered when Enter is pressed.
-   * @param text The current text in the editor
-   * @returns true if the editor should reset the document.
-   */
-  onSubmit?: (text: string) => boolean | void;
-
-  /**
-   * Function that returns a list of suggestions based on the current text.
-   * @param text The current text before the cursor
-   * @returns Array of suggestion strings
-   */
-  onSuggest?: (text: string) => string[];
-
-  /**
-   * ESC pressed.
-   */
-  onCancel?: () => void;
-};
-
-/**
- * Creates an autocomplete extension that shows inline suggestions.
- * Pressing Tab will complete the suggestion.
- */
-// TODO(burdon): Move to react-ui-editor.
-export const autocompleteExtension = ({ onSubmit, onSuggest, onCancel }: AutocompleteOptions): Extension => {
-  const suggest = ViewPlugin.fromClass(
-    class {
-      _decorations: DecorationSet;
-      _currentSuggestion: string | null = null;
-
-      constructor(view: EditorView) {
-        this._decorations = this.computeDecorations(view);
-      }
-
-      update(update: ViewUpdate) {
-        if (update.docChanged || update.selectionSet) {
-          this._decorations = this.computeDecorations(update.view);
-        }
-      }
-
-      private computeDecorations(view: EditorView): DecorationSet {
-        const text = view.state.doc.toString();
-        const suggestions = onSuggest?.(text) ?? [];
-        if (!suggestions.length) {
-          this._currentSuggestion = null;
-          return Decoration.none;
-        }
-
-        // Get the first suggestion.
-        this._currentSuggestion = suggestions[0];
-        const suffix = this._currentSuggestion.slice(text.length);
-        if (!suffix) {
-          return Decoration.none;
-        }
-
-        // Always show ghost text at the end of the document.
-        return Decoration.set([
-          Decoration.widget({
-            widget: new InlineSuggestionWidget(suffix),
-            side: 1,
-          }).range(view.state.doc.length),
-        ]);
-      }
-
-      completeSuggestion(view: EditorView): boolean {
-        if (!this._currentSuggestion) {
-          return false;
-        }
-
-        const text = view.state.doc.toString();
-        const suffix = this._currentSuggestion.slice(text.length);
-        if (!suffix) {
-          return false;
-        }
-
-        view.dispatch({
-          changes: {
-            from: view.state.doc.length,
-            insert: suffix,
-          },
-          selection: {
-            anchor: view.state.doc.length + suffix.length,
-          },
-        });
-
-        return true;
-      }
-    },
-    {
-      decorations: (v) => v._decorations,
-    },
-  );
-
-  return [
-    suggest,
-    EditorView.theme({
-      '.cm-inline-suggestion': {
-        opacity: 0.4,
-      },
-    }),
-
-    Prec.highest(
-      keymap.of([
-        {
-          key: 'Tab',
-          preventDefault: true,
-          run: (view) => {
-            const plugin = view.plugin(suggest);
-            return plugin?.completeSuggestion(view) ?? false;
-          },
-        },
-        {
-          key: 'ArrowRight',
-          preventDefault: true,
-          run: (view) => {
-            // Only complete if cursor is at the end
-            if (view.state.selection.main.head !== view.state.doc.length) {
-              return false;
-            }
-
-            const plugin = view.plugin(suggest);
-            return plugin?.completeSuggestion(view) ?? false;
-          },
-        },
-        {
-          key: 'Enter',
-          preventDefault: true,
-          run: (view) => {
-            const text = view.state.doc.toString().trim();
-            if (text.length > 0 && onSubmit) {
-              const reset = onSubmit(text);
-
-              // Clear the document after calling onEnter.
-              if (reset) {
-                view.dispatch({
-                  changes: {
-                    from: 0,
-                    to: view.state.doc.length,
-                    insert: '',
-                  },
-                });
-              }
-            }
-
-            return true;
-          },
-        },
-        {
-          key: 'Shift-Enter',
-          preventDefault: true,
-          run: (view) => {
-            view.dispatch({
-              changes: {
-                from: view.state.selection.main.head,
-                insert: '\n',
-              },
-              selection: {
-                anchor: view.state.selection.main.head + 1,
-                head: view.state.selection.main.head + 1,
-              },
-            });
-            return true;
-          },
-        },
-        {
-          key: 'Escape',
-          preventDefault: true,
-          run: (view) => {
-            // Clear the entire document.
-            view.dispatch({
-              changes: {
-                from: 0,
-                to: view.state.doc.length,
-                insert: '',
-              },
-            });
-            onCancel?.();
-            return true;
-          },
-        },
-      ]),
-    ),
-  ];
-};
-
-class InlineSuggestionWidget extends WidgetType {
-  constructor(private suffix: string) {
-    super();
-  }
-
-  override toDOM(): HTMLSpanElement {
-    const span = document.createElement('span');
-    span.textContent = this.suffix;
-    span.className = 'cm-inline-suggestion';
-    return span;
-  }
-
-  override eq(other: InlineSuggestionWidget): boolean {
-    return other.suffix === this.suffix;
-  }
-}

package/src/components/Prompt/index.ts

@@ -1,6 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export * from './Prompt';
-export * from './PromptBar';

package/src/components/Prompt/references.ts

@@ -1,180 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { autocompletion, completionKeymap, type CompletionResult } from '@codemirror/autocomplete';
-import { type Extension, RangeSet } from '@codemirror/state';
-import {
-  Decoration,
-  EditorView,
-  keymap,
-  ViewPlugin,
-  WidgetType,
-  type DecorationSet,
-  type ViewUpdate,
-} from '@codemirror/view';
-
-import { Mutex } from '@dxos/async';
-
-export type ReferenceData = {
-  uri: string;
-  label: string;
-  // TODO(dmaretskyi): Consider adding details renderer for when you hover over the reference.
-};
-
-export interface ReferencesProvider {
-  getReferences({ query }: { query: string }): Promise<ReferenceData[]>;
-
-  resolveReference({ uri }: { uri: string }): Promise<ReferenceData | null>;
-}
-
-export type PromptReferencesOptions = {
-  provider: ReferencesProvider;
-  /**
-   * Will prevent the autocomplete from closing when the user blurs the editor.
-   * @default false
-   */
-  debug?: boolean;
-  /**
-   * @default '@'
-   */
-  triggerCharacter?: string;
-};
-
-/**
- * Include references into text.
- */
-export const promptReferences = ({
-  provider,
-  debug = false,
-  triggerCharacter = '@',
-}: PromptReferencesOptions): Extension => {
-  if (triggerCharacter.length !== 1) {
-    throw new Error('triggerCharacter must be a single character');
-  }
-
-  const decorationField = ViewPlugin.fromClass(
-    class ReferenceView {
-      private _mutex = new Mutex();
-
-      decorations: DecorationSet = Decoration.set([]);
-
-      constructor(view: EditorView) {
-        queueMicrotask(async () => {
-          const guard = await this._mutex.acquire();
-          try {
-            this.decorations = await this._computeDecorations(view);
-          } finally {
-            guard.release();
-          }
-        });
-      }
-
-      update(update: ViewUpdate) {
-        if (update.docChanged) {
-          queueMicrotask(async () => {
-            const guard = await this._mutex.acquire();
-            try {
-              this.decorations = await this._computeDecorations(update.view);
-            } finally {
-              guard.release();
-            }
-          });
-        }
-      }
-
-      private async _computeDecorations(view: EditorView): Promise<DecorationSet> {
-        const text = view.state.doc.toString();
-        const references = text.matchAll(new RegExp(`${triggerCharacter}[a-zA-Z0-9@:]+\\s`, 'g'));
-
-        const decorations = [];
-        for (const match of references) {
-          const reference = match[0];
-          const uri = reference.slice(1, -1);
-          const data = await provider.resolveReference({ uri });
-          if (data) {
-            decorations.push(
-              Decoration.replace({
-                widget: new ReferenceWidget(data),
-              }).range(match.index!, match.index! + reference.length),
-            );
-          }
-        }
-
-        return Decoration.set(decorations);
-      }
-    },
-    {
-      decorations: (v) => v.decorations,
-      provide: (plugin) => [
-        EditorView.atomicRanges.of(
-          (view): DecorationSet => view.plugin(decorationField)?.decorations ?? RangeSet.empty,
-        ),
-      ],
-    },
-  );
-
-  return [
-    decorationField,
-
-    EditorView.theme({
-      '.cm-reference-pill': {
-        borderRadius: '0.25rem',
-        borderWidth: '1px',
-        marginRight: '0.25rem',
-        marginLeft: '0.25rem',
-      },
-    }),
-
-    autocompletion({
-      activateOnTyping: true,
-      override: [
-        async (context): Promise<CompletionResult | null> => {
-          const match = context.matchBefore(new RegExp(`${triggerCharacter}[a-zA-Z0-9]+`));
-
-          if (!match || match?.to === match?.from) {
-            return null;
-          }
-
-          const query = match.text.slice(1);
-          const references = await provider.getReferences({ query });
-
-          return {
-            from: match.from,
-            filter: false,
-            options: references.map((reference) => ({
-              label: reference.label,
-              apply: `${triggerCharacter}${reference.uri} `,
-            })),
-          };
-        },
-      ],
-      closeOnBlur: !debug,
-      tooltipClass: () => 'shadow rounded',
-      aboveCursor: true,
-    }),
-
-    keymap.of(completionKeymap),
-  ];
-};
-
-class ReferenceWidget extends WidgetType {
-  constructor(private data: ReferenceData) {
-    super();
-  }
-
-  override toDOM(): HTMLSpanElement {
-    const span = document.createElement('span');
-    span.textContent = `@ ${this.data.label}`;
-    span.className = 'cm-reference-pill';
-    return span;
-  }
-
-  override eq(other: ReferenceWidget): boolean {
-    return other.data.uri === this.data.uri;
-  }
-
-  override ignoreEvent(): boolean {
-    return true;
-  }
-}