@dxos/react-ui-editor 0.3.10-main.cbe7692 → 0.3.10-main.cc91b2d

package/src/components/TextEditor/MarkdownEditor.stories.tsx

@@ -32,9 +32,15 @@ import {
   blast,
   demo,
   defaultOptions,
+  highlight,
+  code,
 } from './extensions';
 import { useTextModel } from '../../hooks';
 
+// Extensions:
+// TODO(burdon): Table of contents.
+// TODO(burdon): Front-matter
+
 const str = (...lines: string[]) => lines.join('\n');
 
 const num = () => faker.number.int({ min: 0, max: 9999 }).toLocaleString();
@@ -78,6 +84,7 @@ const text = {
     '```tsx',
     'const Component = () => {',
     '  const x = 100;',
+    '',
     '  return () => <div>Test</div>;',
     '};',
     '',
@@ -93,24 +100,26 @@ const text = {
     '',
   ),
 
-  headings: str(
-    ...[1, 2, 3, 4, 5, 6].map((level) => ['#'.repeat(level) + ` Heading ${level}`, faker.lorem.sentences(), '']).flat(),
-  ),
-
-  paragraphs: str(...faker.helpers.multiple(() => [faker.lorem.paragraph(), ''], { count: 3 }).flat()),
-
   table: str(
     '# Table',
     '',
-    `| ${faker.lorem.word()} | ${faker.lorem.word()} | ${faker.lorem.word()} |`,
-    '|---|---|---|',
-    `| ${num()} | ${num()} | ${num()} |`,
-    `| ${num()} | ${num()} | ${num()} |`,
-    `| ${num()} | ${num()} | ${num()} |`,
+    `| ${faker.lorem.word().padStart(8)} | ${faker.lorem.word().padStart(8)} | ${faker.lorem.word().padStart(8)} |`,
+    '|----------|----------|----------|',
+    `| ${num().padStart(8)} | ${num().padStart(8)} | ${num().padStart(8)} |`,
+    `| ${num().padStart(8)} | ${num().padStart(8)} | ${num().padStart(8)} |`,
+    `| ${num().padStart(8)} | ${num().padStart(8)} | ${num().padStart(8)} |`,
     '', // TODO(burdon): Possible GFM parsing bug if no newline?
   ),
 
   image: str('# Image', '', '![dxos](https://pbs.twimg.com/profile_banners/1268328127673044992/1684766689/1500x500)'),
+
+  headings: str(
+    ...[1, 2, 3, 4, 5, 6].map((level) => ['#'.repeat(level) + ` Heading ${level}`, faker.lorem.sentences(), '']).flat(),
+  ),
+
+  paragraphs: str(...faker.helpers.multiple(() => [faker.lorem.paragraph(), ''], { count: 3 }).flat()),
+
+  footer: str('', '', '', '', '')
 };
 
 const document = str(
@@ -138,6 +147,8 @@ const document = str(
   text.table,
   '---',
   text.image,
+  '',
+  text.footer,
 );
 
 const links = [
@@ -173,6 +184,7 @@ const onRender: LinkOptions['onRender'] = (el, url) => {
   );
 };
 
+// TODO(burdon): Pass in model.
 const Story = ({
   text,
   automerge,
@@ -205,14 +217,15 @@ export default {
 };
 
 const extensions = [
-  link({ onRender }),
-  tooltip({ onHover }),
-  tasklist(),
-  table(),
-  image(),
   autocomplete({
     onSearch: (text) => links.filter(({ label }) => label.toLowerCase().includes(text.toLowerCase())),
   }),
+  code(),
+  image(),
+  link({ onRender }),
+  table(),
+  tasklist(),
+  tooltip({ onHover }),
 ];
 
 export const Default = {
@@ -223,32 +236,30 @@ export const Readonly = {
   render: () => <Story text={document} extensions={extensions} readonly />,
 };
 
-export const Simple = {
-  render: () => <Story text={document} />,
-};
-
 export const Tooltips = {
-  render: () => <Story text={text.links} extensions={[tooltip({ onHover })]} />,
+  render: () => <Story text={str(text.links, text.footer)} extensions={[tooltip({ onHover })]} />,
 };
 
 export const Links = {
-  render: () => <Story text={text.links} extensions={[link({ onRender })]} />,
+  render: () => <Story text={str(text.links, text.footer)} extensions={[link({ onRender })]} />,
+};
+
+export const Code = {
+  render: () => <Story text={str(text.code, text.footer)} extensions={[code()]} readonly />,
 };
 
 export const Table = {
-  render: () => (
-    <Story text={str('This is a table', '', text.table, '', text.table, '', '')} readonly extensions={[table()]} />
-  ),
+  render: () => <Story text={str(text.table, text.footer)} extensions={[table()]} />,
 };
 
 export const Image = {
-  render: () => <Story text={str('This is an image', '', text.image, '', '')} readonly extensions={[image()]} />,
+  render: () => <Story text={str(text.image, text.footer)} readonly extensions={[image()]} />,
 };
 
 export const TaskList = {
   render: () => (
     <Story
-      text={str(text.tasks, '', text.list)}
+      text={str(text.tasks, '', text.list, text.footer)}
       extensions={[
         tasklist(),
         listener({
@@ -264,7 +275,7 @@ export const TaskList = {
 export const Autocomplete = {
   render: () => (
     <Story
-      text={str('# Autocomplete', '', '', '', '', '', '')}
+      text={str('# Autocomplete', '', 'Press CTRL-SPACE', text.footer)}
       extensions={[
         link({ onRender }),
         autocomplete({
@@ -280,7 +291,7 @@ const names = ['adam', 'alice', 'alison', 'bob', 'carol', 'charlie', 'sayuri', '
 export const Mention = {
   render: () => (
     <Story
-      text={str('# Mention', '', '', '', '', '', '')}
+      text={str('# Mention', '', 'Type @...', text.footer)}
       extensions={[
         mention({
           onSearch: (text) => names.filter((name) => name.toLowerCase().startsWith(text.toLowerCase())),
@@ -294,14 +305,15 @@ const mark = () => `[^${PublicKey.random().toHex()}]`;
 export const Comments = {
   render: () => (
     <Story
-      text={str(mark(), '', text.paragraphs, mark(), '', mark(), '')}
+      text={str(text.paragraphs, mark(), '', mark(), text.footer)}
       extensions={[
         comments({
           onCreate: () => PublicKey.random().toHex(),
           onUpdate: (info) => {
-            console.log('update', info);
+            // console.log('update', info);
           },
         }),
+        highlight(),
       ]}
     />
   ),
@@ -322,7 +334,7 @@ export const Diagnostics = {
 };
 
 export const Demo = {
-  render: () => <Story text={text.paragraphs} extensions={[demo()]} />,
+  render: () => <Story text={str(text.paragraphs, text.footer)} extensions={[demo()]} />,
 };
 
 export const Blast = {
@@ -330,6 +342,9 @@ export const Blast = {
     <Story
       text={str(text.paragraphs, text.code, text.paragraphs)}
       extensions={[
+        demo({
+          items: localStorage.getItem('dxos.composer.extension.demo')?.split(','),
+        }),
         blast(
           defaultsDeep(
             {
@@ -346,3 +361,7 @@ export const Blast = {
     />
   ),
 };
+
+export const NoExtensions = {
+  render: () => <Story text={document} />,
+};

package/src/components/TextEditor/TextEditor.stories.tsx

@@ -57,6 +57,17 @@ export default {
 };
 
 export const Default = {
+  render: () => (
+    <Story
+      slots={defaultsDeep(
+        { editor: { theme: textTheme, placeholder: 'Enter text...' } } satisfies TextEditorSlots,
+        defaultSlots,
+      )}
+    />
+  ),
+};
+
+export const Text = {
   render: () => (
     <Story
       text={initialText}
@@ -71,12 +82,12 @@ export const Default = {
 export const Automerge = {
   render: () => (
     <Story
+      automerge
       text={initialText}
       slots={defaultsDeep(
         { editor: { theme: textTheme, placeholder: 'Enter text...' } } satisfies TextEditorSlots,
         defaultSlots,
       )}
-      automerge
     />
   ),
 };

package/src/components/TextEditor/TextEditor.tsx

@@ -21,7 +21,7 @@ import { generateName } from '@dxos/display-name';
 import { useThemeContext } from '@dxos/react-ui';
 import { getColorForValue, inputSurface, mx } from '@dxos/react-ui-theme';
 
-import { basicBundle, demo, markdownBundle } from './extensions';
+import { basicBundle, markdownBundle } from './extensions';
 import { defaultTheme, markdownTheme, textTheme } from './themes';
 import { type EditorModel } from '../../hooks';
 import { type ThemeStyles } from '../../styles';
@@ -88,6 +88,12 @@ export const BaseTextEditor = forwardRef<TextEditorRef, TextEditorProps>(
       }
     }, [awareness, peer, themeMode]);
 
+    // TODO(burdon): Get from model.
+    // useHighlights(view, [
+    //   { id: 'c-1', from: 0, to: 20 },
+    //   { id: 'c-2', from: 100, to: 120 },
+    // ]);
+
     useEffect(() => {
       if (!root) {
         return;
@@ -102,6 +108,7 @@ export const BaseTextEditor = forwardRef<TextEditorRef, TextEditorProps>(
           editorMode === 'vim' && vim(),
 
           // Theme.
+          // TODO(burdon): Make optional.
           EditorView.baseTheme(defaultTheme),
           EditorView.theme(slots?.editor?.theme ?? {}),
           // TODO(burdon): themeMode doesn't change in storybooks.
@@ -119,7 +126,17 @@ export const BaseTextEditor = forwardRef<TextEditorRef, TextEditorProps>(
       // If the new state is derived from the old state, it will likely not be visible other than the cursor resetting.
       // Ideally this should not be hit except when changing between text objects.
       view?.destroy();
-      setView(new EditorView({ state, parent: root }));
+      setView(
+        new EditorView({
+          state,
+          parent: root,
+          // NOTE: Uncomment to spy on all transactions.
+          // https://codemirror.net/docs/ref/#view.EditorView.dispatch
+          // dispatch: (transaction, view) => {
+          //   view.update([transaction]);
+          // },
+        }),
+      );
       setState(state);
 
       return () => {
@@ -160,17 +177,6 @@ export const BaseTextEditor = forwardRef<TextEditorRef, TextEditorProps>(
   },
 );
 
-// TODO(burdon): Allow plugins to set extensions (factory).
-const maybeDebug = (): Extension => {
-  // TODO(burdon): Parse JSON script format (with key bindings?)
-  const items = localStorage.getItem('dxos.composer.demo');
-  if (items) {
-    return demo({ items: items.split(',') });
-  }
-
-  return [];
-};
-
 export const TextEditor = forwardRef<TextEditorRef, TextEditorProps>(
   ({ readonly, extensions = [], slots: _slots, ...props }, forwardedRef) => {
     const { themeMode } = useThemeContext();
@@ -179,11 +185,7 @@ export const TextEditor = forwardRef<TextEditorRef, TextEditorProps>(
       <BaseTextEditor
         ref={forwardedRef}
         readonly={readonly}
-        extensions={[
-          basicBundle({ readonly, themeMode, placeholder: slots?.editor?.placeholder }),
-          maybeDebug(),
-          ...extensions,
-        ]}
+        extensions={[basicBundle({ readonly, themeMode, placeholder: slots?.editor?.placeholder }), ...extensions]}
         slots={slots}
         {...props}
       />
@@ -199,11 +201,7 @@ export const MarkdownEditor = forwardRef<TextEditorRef, TextEditorProps>(
       <BaseTextEditor
         ref={forwardedRef}
         readonly={readonly}
-        extensions={[
-          markdownBundle({ readonly, themeMode, placeholder: slots?.editor?.placeholder }),
-          maybeDebug(),
-          ...extensions,
-        ]}
+        extensions={[markdownBundle({ readonly, themeMode, placeholder: slots?.editor?.placeholder }), ...extensions]}
         slots={slots}
         {...props}
       />

package/src/components/TextEditor/extensions/basic.ts

@@ -6,7 +6,7 @@ import { closeBrackets } from '@codemirror/autocomplete';
 import { bracketMatching, defaultHighlightStyle, syntaxHighlighting } from '@codemirror/language';
 import { type Extension } from '@codemirror/state';
 import { oneDarkHighlightStyle } from '@codemirror/theme-one-dark';
-import { EditorView, placeholder } from '@codemirror/view';
+import { drawSelection, EditorView, placeholder } from '@codemirror/view';
 
 import type { ThemeMode } from '@dxos/react-ui';
 
@@ -18,12 +18,13 @@ export type BasicBundleOptions = {
 
 export const basicBundle = ({ readonly, themeMode, placeholder: _placeholder }: BasicBundleOptions): Extension[] =>
   [
+    EditorView.lineWrapping,
+
     // TODO(burdon): Compare with minimalSetup.
     // https://codemirror.net/docs/ref/#codemirror.minimalSetup
     bracketMatching(),
     closeBrackets(),
-    EditorView.lineWrapping,
-
+    drawSelection(),
     _placeholder && placeholder(_placeholder),
 
     // TODO(burdon): Is this required?

package/src/components/TextEditor/extensions/blast.ts

@@ -5,7 +5,7 @@
 // https://github.com/chinchang/code-blast-codemirror/blob/master/code-blast.js
 //
 
-import { type Extension, StateField } from '@codemirror/state';
+import { type Extension } from '@codemirror/state';
 import { EditorView, keymap } from '@codemirror/view';
 import defaultsDeep from 'lodash.defaultsdeep';
 
@@ -25,6 +25,7 @@ export type BlastOptions = {
 };
 
 export const defaultOptions: BlastOptions = {
+  effect: 2,
   maxParticles: 200,
   particleGravity: 0.08,
   particleAlphaFadeout: 0.996,
@@ -35,7 +36,7 @@ export const defaultOptions: BlastOptions = {
   shakeIntensity: 5,
 };
 
-export const blast = (options: Partial<BlastOptions>): Extension => {
+export const blast = (options: Partial<BlastOptions> = defaultOptions): Extension => {
   let blaster: Blaster | undefined;
   let last = 0;
 
@@ -58,46 +59,43 @@ export const blast = (options: Partial<BlastOptions>): Extension => {
           blaster.destroy();
         }
 
-        blaster = new Blaster(update.view.scrollDOM, defaultsDeep({}, options, defaultOptions));
+        blaster = new Blaster(
+          update.view.scrollDOM,
+          defaultsDeep(
+            {
+              particleGravity: 0.2,
+              particleShrinkRate: 0.995,
+              color: () => [100 + Math.random() * 100, 0, 0],
+            },
+            options,
+            defaultOptions,
+          ),
+        );
         blaster.initialize();
         blaster.start(); // TODO(burdon): Stop/clean-up.
       } else {
         blaster.resize();
       }
 
-      const current = update.view.state.selection.main.head;
+      const current = update.state.selection.main.head;
       if (current !== last) {
         last = current;
         // TODO(burdon): Null if end of line.
-        const { element, point } = getPoint(update.view, current);
+        const { element, point } = getPoint(update.view, current - 1);
         if (element && point) {
           blaster.spawn({ element, point });
         }
       }
     }),
 
-    // Document changed.
-    StateField.define({
-      create: () => null,
-      update: (_value, transaction) => {
-        if (blaster && transaction.docChanged) {
-          // TODO(burdon): Only on delete.
-          const view = EditorView.findFromDOM(blaster.node)!;
-          const { element, point } = getPoint(view, transaction.selection!.main.head);
-          if (element && point) {
-            blaster.spawn({ element, point });
-          }
-        }
-
-        return null;
-      },
-    }),
-
     keymap.of([
       {
-        any: (view, event) => {
-          if (blaster && event.key === 'Enter' && event.shiftKey) {
-            blaster.shake({ time: 0.4 });
+        any: (_, event) => {
+          if (blaster) {
+            if (event.key === 'Enter' && event.shiftKey) {
+              blaster.shake({ time: 0.8 });
+              return true;
+            }
           }
 
           return false;
@@ -112,11 +110,11 @@ export const blast = (options: Partial<BlastOptions>): Extension => {
 //
 
 class Blaster {
+  private readonly _effect: Effect;
+
   _canvas: HTMLCanvasElement | undefined;
   _ctx: CanvasRenderingContext2D | undefined | null;
 
-  private readonly _effect: Effect;
-
   _running = false;
   _lastTime: number | undefined;
   _shakeTime = 0;
@@ -144,7 +142,7 @@ class Blaster {
     this._canvas.style.position = 'absolute';
     this._canvas.style.zIndex = '0';
     this._canvas.style.pointerEvents = 'none';
-    // this._canvas.style.border = '2px solid red';
+    // this._canvas.style.border = '2px dashed red';
 
     this._ctx = this._canvas.getContext('2d');
 
@@ -214,7 +212,7 @@ class Blaster {
   }
 
   shake = throttle<{ time: number }>(({ time }) => {
-    this._shakeTime = this._shakeTimeMax;
+    this._shakeTime = this._shakeTimeMax || time;
     this._shakeTimeMax = time;
   }, 100);
 

package/src/components/TextEditor/extensions/code.ts

@@ -0,0 +1,99 @@
+//
+// Copyright 2023 DXOS.org
+//
+
+import { type Range } from '@codemirror/state';
+import {
+  ViewPlugin,
+  type DecorationSet,
+  EditorView,
+  type ViewUpdate,
+  Decoration,
+  type BlockInfo,
+} from '@codemirror/view';
+import get from 'lodash.get';
+
+import { mx } from '@dxos/react-ui-theme';
+
+import { tokens } from '../../../styles';
+
+const CODE_REGEX = /```[\s\S]*?```/gs;
+
+// TODO(burdon): Reconcile with theme.
+const styles = EditorView.baseTheme({
+  '& .cm-code-block': {
+    display: 'block',
+    paddingInline: '4px !important',
+  },
+  '&light .cm-code-block': {
+    background: get(tokens, 'extend.colors.neutral.50'),
+  },
+  '&dark .cm-code-block': {
+    background: get(tokens, 'extend.colors.neutral.850'),
+  },
+  '& .cm-code-block-first': {
+    paddingTop: '4px !important',
+    borderTopLeftRadius: '4px',
+    borderTopRightRadius: '4px',
+  },
+  '& .cm-code-block-last': {
+    paddingBottom: '4px !important',
+    borderBottomLeftRadius: '4px',
+    borderBottomRightRadius: '4px',
+  },
+});
+
+const getLineRange = (lines: BlockInfo[], from: number, to: number) => {
+  const start = lines.findIndex((line) => line.from >= from);
+  const end = lines.findIndex((line) => line.to >= to);
+  return [start, end];
+};
+
+export const code = () => {
+  const getDecorations = (view: EditorView) => {
+    const decorations: Range<Decoration>[] = [];
+    if (view.state.readOnly) {
+      const text = view.state.doc.sliceString(0);
+      const matches = text.matchAll(CODE_REGEX);
+      const blocks = view.viewportLineBlocks;
+      for (const match of matches) {
+        const range = getLineRange(blocks, match.index!, match.index! + match[0].length);
+        for (let i = range[0]; i <= range[1]; i++) {
+          const block = blocks[i];
+          decorations.push(
+            Decoration.line({
+              class: mx(
+                'cm-code-block',
+                i === range[0] && 'cm-code-block-first',
+                i === range[1] && 'cm-code-block-last',
+              ),
+            }).range(block.from),
+          );
+        }
+      }
+    }
+
+    return Decoration.set(decorations);
+  };
+
+  return [
+    ViewPlugin.fromClass(
+      class {
+        decorations: DecorationSet;
+        constructor(view: EditorView) {
+          this.decorations = getDecorations(view);
+        }
+
+        update(update: ViewUpdate) {
+          if (update.docChanged || update.viewportChanged) {
+            this.decorations = getDecorations(update.view);
+          }
+        }
+      },
+      {
+        decorations: (value) => value.decorations,
+      },
+    ),
+    styles,
+  ];
+};