editor improvements

* add callout, youtube embed, image, video, table, detail, math
* fix attachments module
* other fixes

packages/editor-ext/src/lib/video/index.ts

@@ -0,0 +1,2 @@
+export { TiptapVideo } from "./video";
+export * from "./video-upload";

packages/editor-ext/src/lib/video/video-upload.ts

@@ -0,0 +1,125 @@
+import { type EditorState, Plugin, PluginKey } from "@tiptap/pm/state";
+import { Decoration, DecorationSet } from "@tiptap/pm/view";
+import { IAttachment } from "client/src/lib/types";
+import { MediaUploadOptions, UploadFn } from "../media-utils";
+
+const uploadKey = new PluginKey("video-upload");
+
+export const VideoUploadPlugin = ({
+  placeHolderClass,
+}: {
+  placeHolderClass: string;
+}) =>
+  new Plugin({
+    key: uploadKey,
+    state: {
+      init() {
+        return DecorationSet.empty;
+      },
+      apply(tr, set) {
+        set = set.map(tr.mapping, tr.doc);
+        // See if the transaction adds or removes any placeholders
+        //@-ts-expect-error - not yet sure what the type I need here
+        const action = tr.getMeta(this);
+        if (action?.add) {
+          const { id, pos, src } = action.add;
+
+          const placeholder = document.createElement("div");
+          placeholder.setAttribute("class", "video-placeholder");
+          const video = document.createElement("video");
+          video.setAttribute("class", placeHolderClass);
+          video.src = src;
+          placeholder.appendChild(video);
+          const deco = Decoration.widget(pos + 1, placeholder, {
+            id,
+          });
+          set = set.add(tr.doc, [deco]);
+        } else if (action?.remove) {
+          set = set.remove(
+            set.find(
+              undefined,
+              undefined,
+              (spec) => spec.id == action.remove.id,
+            ),
+          );
+        }
+        return set;
+      },
+    },
+    props: {
+      decorations(state) {
+        return this.getState(state);
+      },
+    },
+  });
+
+function findPlaceholder(state: EditorState, id: {}) {
+  const decos = uploadKey.getState(state) as DecorationSet;
+  const found = decos.find(undefined, undefined, (spec) => spec.id == id);
+  return found.length ? found[0]?.from : null;
+}
+
+export const handleVideoUpload =
+  ({ validateFn, onUpload }: MediaUploadOptions): UploadFn =>
+  async (file, view, pos, pageId) => {
+    // check if the file is an image
+    const validated = validateFn?.(file);
+    // @ts-ignore
+    if (!validated) return;
+    // A fresh object to act as the ID for this upload
+    const id = {};
+
+    // Replace the selection with a placeholder
+    const tr = view.state.tr;
+    if (!tr.selection.empty) tr.deleteSelection();
+
+    const reader = new FileReader();
+    reader.readAsDataURL(file);
+    reader.onload = () => {
+      tr.setMeta(uploadKey, {
+        add: {
+          id,
+          pos,
+          src: reader.result,
+        },
+      });
+      view.dispatch(tr);
+    };
+
+    await onUpload(file, pageId).then(
+      (attachment: IAttachment) => {
+        const { schema } = view.state;
+
+        const pos = findPlaceholder(view.state, id);
+
+        // If the content around the placeholder has been deleted, drop
+        // the image
+        if (pos == null) return;
+
+        // Otherwise, insert it at the placeholder's position, and remove
+        // the placeholder
+
+        if (!attachment) return;
+
+        const node = schema.nodes.video?.create({
+          src: `/files/${attachment.id}/${attachment.fileName}`,
+          attachmentId: attachment.id,
+          title: attachment.fileName,
+          size: attachment.fileSize,
+        });
+        if (!node) return;
+
+        const transaction = view.state.tr
+          .replaceWith(pos, pos, node)
+          .setMeta(uploadKey, { remove: { id } });
+        view.dispatch(transaction);
+      },
+      () => {
+        // Deletes the image placeholder on error
+        const transaction = view.state.tr
+          .delete(pos, pos)
+          .setMeta(uploadKey, { remove: { id } });
+        view.dispatch(transaction);
+      },
+    );
+  };

packages/editor-ext/src/lib/video/video.ts

@@ -0,0 +1,146 @@
+import { ReactNodeViewRenderer } from "@tiptap/react";
+import { VideoUploadPlugin } from "./video-upload";
+import { mergeAttributes, Range, Node, nodeInputRule } from "@tiptap/core";
+
+export interface VideoOptions {
+  view: any;
+  HTMLAttributes: Record<string, any>;
+}
+export interface VideoAttributes {
+  src?: string;
+  title?: string;
+  align?: string;
+  attachmentId?: string;
+  size?: number;
+  width?: number;
+}
+
+declare module "@tiptap/core" {
+  interface Commands<ReturnType> {
+    videoBlock: {
+      setVideo: (attributes: VideoAttributes) => ReturnType;
+      setVideoAt: (
+        attributes: VideoAttributes & { pos: number | Range },
+      ) => ReturnType;
+      setVideoAlign: (align: "left" | "center" | "right") => ReturnType;
+      setVideoWidth: (width: number) => ReturnType;
+    };
+  }
+}
+
+const VIDEO_INPUT_REGEX = /!\[(.+|:?)]\((\S+)(?:(?:\s+)["'](\S+)["'])?\)/;
+
+export const TiptapVideo = Node.create<VideoOptions>({
+  name: "video",
+
+  group: "block",
+  isolating: true,
+  atom: true,
+  defining: true,
+  draggable: true,
+
+  addOptions() {
+    return {
+      view: null,
+      HTMLAttributes: {},
+    };
+  },
+
+  addAttributes() {
+    return {
+      src: {
+        default: "",
+        parseHTML: (element) => element.getAttribute("src"),
+        renderHTML: (attributes) => ({
+          src: attributes.src,
+        }),
+      },
+      attachmentId: {
+        default: undefined,
+        parseHTML: (element) => element.getAttribute("data-attachment-id"),
+        renderHTML: (attributes: VideoAttributes) => ({
+          "data-attachment-id": attributes.align,
+        }),
+      },
+      width: {
+        default: "100%",
+        parseHTML: (element) => element.getAttribute("data-width"),
+        renderHTML: (attributes: VideoAttributes) => ({
+          "data-width": attributes.width,
+        }),
+      },
+      size: {
+        default: null,
+        parseHTML: (element) => element.getAttribute("data-size"),
+        renderHTML: (attributes: VideoAttributes) => ({
+          "data-size": attributes.size,
+        }),
+      },
+      align: {
+        default: "center",
+        parseHTML: (element) => element.getAttribute("data-align"),
+        renderHTML: (attributes: VideoAttributes) => ({
+          "data-align": attributes.align,
+        }),
+      },
+    };
+  },
+
+  renderHTML({ HTMLAttributes }) {
+    return [
+      "video",
+      { controls: "true", ...HTMLAttributes },
+      ["source", HTMLAttributes],
+    ];
+  },
+
+  addCommands() {
+    return {
+      setVideo:
+        (attrs: VideoAttributes) =>
+        ({ commands }) => {
+          return commands.insertContent({
+            type: "video",
+            attrs: attrs,
+          });
+        },
+
+      setVideoAlign:
+        (align) =>
+        ({ commands }) =>
+          commands.updateAttributes("video", { align }),
+
+      setVideoWidth:
+        (width) =>
+        ({ commands }) =>
+          commands.updateAttributes("video", {
+            width: `${Math.max(0, Math.min(100, width))}%`,
+          }),
+    };
+  },
+
+  addNodeView() {
+    return ReactNodeViewRenderer(this.options.view);
+  },
+
+  addInputRules() {
+    return [
+      nodeInputRule({
+        find: VIDEO_INPUT_REGEX,
+        type: this.type,
+        getAttributes: (match) => {
+          const [, , src] = match;
+          return { src };
+        },
+      }),
+    ];
+  },
+
+  addProseMirrorPlugins() {
+    return [
+      VideoUploadPlugin({
+        placeHolderClass: "video-upload",
+      }),
+    ];
+  },
+});