@blocknote/core 0.7.1-alpha.0 → 0.8.1

package/package.json

@@ -3,7 +3,7 @@
   "homepage": "https://github.com/TypeCellOS/BlockNote",
   "private": false,
   "license": "MPL-2.0",
-  "version": "0.7.1-alpha.0",
+  "version": "0.8.1",
   "files": [
     "dist",
     "types",
@@ -92,7 +92,7 @@
     "eslint-config-react-app": "^7.0.0",
     "jsdom": "^21.1.0",
     "prettier": "^2.7.1",
-    "typescript": "^4.5.4",
+    "typescript": "^5.0.4",
     "vite": "^4.1.2",
     "vite-plugin-eslint": "^1.8.1",
     "vitest": "^0.28.5"
@@ -110,5 +110,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "gitHead": "221da514c70e561fd8256d6bec6b825390060edf"
+  "gitHead": "2265d9068126df0881dd813ba6a265b82f48b466"
 }

package/src/BlockNoteEditor.ts

@@ -22,9 +22,14 @@ import styles from "./editor.module.css";
 import {
   Block,
   BlockIdentifier,
+  BlockSchema,
   PartialBlock,
 } from "./extensions/Blocks/api/blockTypes";
 import { TextCursorPosition } from "./extensions/Blocks/api/cursorPositionTypes";
+import {
+  DefaultBlockSchema,
+  defaultBlockSchema,
+} from "./extensions/Blocks/api/defaultBlocks";
 import {
   ColorStyle,
   Styles,
@@ -37,21 +42,23 @@ import {
   defaultSlashMenuItems,
 } from "./extensions/SlashMenu";
 
-export type BlockNoteEditorOptions = {
-  // TODO: Figure out if enableBlockNoteExtensions is needed and document them.
+export type BlockNoteEditorOptions<BSchema extends BlockSchema> = {
+  // TODO: Figure out if enableBlockNoteExtensions/disableHistoryExtension are needed and document them.
   enableBlockNoteExtensions: boolean;
 
   /**
    * UI element factories for creating a custom UI, including custom positioning
    * & rendering.
    */
-  uiFactories: UiFactories;
+  uiFactories: UiFactories<BSchema>;
   /**
    * TODO: why is this called slashCommands and not slashMenuItems?
    *
+   * (couldn't fix any type, see https://github.com/TypeCellOS/BlockNote/pull/191#discussion_r1210708771)
+   *
    * @default defaultSlashMenuItems from `./extensions/SlashMenu`
    */
-  slashCommands: BaseSlashMenuItem[];
+  slashCommands: BaseSlashMenuItem<any>[];
 
   /**
    * The HTML element that should be used as the parent element for the editor.
@@ -68,15 +75,15 @@ export type BlockNoteEditorOptions = {
   /**
    *  A callback function that runs when the editor is ready to be used.
    */
-  onEditorReady: (editor: BlockNoteEditor) => void;
+  onEditorReady: (editor: BlockNoteEditor<BSchema>) => void;
   /**
    * A callback function that runs whenever the editor's contents change.
    */
-  onEditorContentChange: (editor: BlockNoteEditor) => void;
+  onEditorContentChange: (editor: BlockNoteEditor<BSchema>) => void;
   /**
    * A callback function that runs whenever the text cursor position changes.
    */
-  onTextCursorPositionChange: (editor: BlockNoteEditor) => void;
+  onTextCursorPositionChange: (editor: BlockNoteEditor<BSchema>) => void;
   /**
    * Locks the editor from being editable by the user if set to `false`.
    */
@@ -84,7 +91,7 @@ export type BlockNoteEditorOptions = {
   /**
    * The content that should be in the editor when it's created, represented as an array of partial block objects.
    */
-  initialContent: PartialBlock[];
+  initialContent: PartialBlock<BSchema>[];
   /**
    * Use default BlockNote font and reset the styles of <p> <li> <h1> elements etc., that are used in BlockNote.
    *
@@ -98,6 +105,11 @@ export type BlockNoteEditorOptions = {
    */
   theme: "light" | "dark";
 
+  /**
+   * A list of block types that should be available in the editor.
+   */
+  blockSchema: BSchema;
+
   /**
    * When enabled, allows for collaboration between multiple users.
    */
@@ -133,32 +145,51 @@ const blockNoteTipTapOptions = {
   enableCoreExtensions: false,
 };
 
-export class BlockNoteEditor {
+export class BlockNoteEditor<BSchema extends BlockSchema = DefaultBlockSchema> {
   public readonly _tiptapEditor: TiptapEditor & { contentComponent: any };
-  private blockCache = new WeakMap<Node, Block>();
+  public blockCache = new WeakMap<Node, Block<BSchema>>();
+  public readonly schema: BSchema;
 
   public get domElement() {
     return this._tiptapEditor.view.dom as HTMLDivElement;
   }
 
+  public isFocused() {
+    return this._tiptapEditor.view.hasFocus();
+  }
+
   public focus() {
     this._tiptapEditor.view.focus();
   }
 
-  constructor(private readonly options: Partial<BlockNoteEditorOptions> = {}) {
+  constructor(
+    private readonly options: Partial<BlockNoteEditorOptions<BSchema>> = {}
+  ) {
     // apply defaults
-    options = {
+    const newOptions: Omit<typeof options, "defaultStyles" | "blockSchema"> & {
+      defaultStyles: boolean;
+      blockSchema: BSchema;
+    } = {
       defaultStyles: true,
+      // TODO: There's a lot of annoying typing stuff to deal with here. If
+      //  BSchema is specified, then options.blockSchema should also be required.
+      //  If BSchema is not specified, then options.blockSchema should also not
+      //  be defined. Unfortunately, trying to implement these constraints seems
+      //  to be a huge pain, hence the `as any` casts.
+      blockSchema: options.blockSchema || (defaultBlockSchema as any),
       ...options,
     };
 
-    const extensions = getBlockNoteExtensions({
+    const extensions = getBlockNoteExtensions<BSchema>({
       editor: this,
-      uiFactories: options.uiFactories || {},
-      slashCommands: options.slashCommands || defaultSlashMenuItems,
-      collaboration: options.collaboration,
+      uiFactories: newOptions.uiFactories || {},
+      slashCommands: newOptions.slashCommands || defaultSlashMenuItems,
+      blockSchema: newOptions.blockSchema,
+      collaboration: newOptions.collaboration,
     });
 
+    this.schema = newOptions.blockSchema;
+
     const tiptapOptions: EditorOptions = {
       // TODO: This approach to setting initial content is "cleaner" but requires the PM editor schema, which is only
       //  created after initializing the TipTap editor. Not sure it's feasible.
@@ -168,39 +199,39 @@ export class BlockNoteEditor {
       //     blockToNode(block, this._tiptapEditor.schema).toJSON()
       //   ),
       ...blockNoteTipTapOptions,
-      ...options._tiptapOptions,
+      ...newOptions._tiptapOptions,
       onCreate: () => {
-        options.onEditorReady?.(this);
-        options.initialContent &&
-          this.replaceBlocks(this.topLevelBlocks, options.initialContent);
+        newOptions.onEditorReady?.(this);
+        newOptions.initialContent &&
+          this.replaceBlocks(this.topLevelBlocks, newOptions.initialContent);
       },
       onUpdate: () => {
-        options.onEditorContentChange?.(this);
+        newOptions.onEditorContentChange?.(this);
       },
       onSelectionUpdate: () => {
-        options.onTextCursorPositionChange?.(this);
+        newOptions.onTextCursorPositionChange?.(this);
       },
       editable: options.editable === undefined ? true : options.editable,
       extensions:
-        options.enableBlockNoteExtensions === false
-          ? options._tiptapOptions?.extensions
-          : [...(options._tiptapOptions?.extensions || []), ...extensions],
+        newOptions.enableBlockNoteExtensions === false
+          ? newOptions._tiptapOptions?.extensions
+          : [...(newOptions._tiptapOptions?.extensions || []), ...extensions],
       editorProps: {
         attributes: {
           "data-theme": options.theme || "light",
-          ...(options.editorDOMAttributes || {}),
+          ...(newOptions.editorDOMAttributes || {}),
           class: [
             styles.bnEditor,
             styles.bnRoot,
-            options.defaultStyles ? styles.defaultStyles : "",
-            options.editorDOMAttributes?.class || "",
+            newOptions.defaultStyles ? styles.defaultStyles : "",
+            newOptions.editorDOMAttributes?.class || "",
           ].join(" "),
         },
       },
     };
 
-    if (options.parentElement) {
-      tiptapOptions.element = options.parentElement;
+    if (newOptions.parentElement) {
+      tiptapOptions.element = newOptions.parentElement;
     }
 
     this._tiptapEditor = new Editor(tiptapOptions) as Editor & {
@@ -212,11 +243,11 @@ export class BlockNoteEditor {
    * Gets a snapshot of all top-level (non-nested) blocks in the editor.
    * @returns A snapshot of all top-level (non-nested) blocks in the editor.
    */
-  public get topLevelBlocks(): Block[] {
-    const blocks: Block[] = [];
+  public get topLevelBlocks(): Block<BSchema>[] {
+    const blocks: Block<BSchema>[] = [];
 
     this._tiptapEditor.state.doc.firstChild!.descendants((node) => {
-      blocks.push(nodeToBlock(node, this.blockCache));
+      blocks.push(nodeToBlock(node, this.schema, this.blockCache));
 
       return false;
     });
@@ -229,12 +260,14 @@ export class BlockNoteEditor {
    * @param blockIdentifier The identifier of an existing block that should be retrieved.
    * @returns The block that matches the identifier, or `undefined` if no matching block was found.
    */
-  public getBlock(blockIdentifier: BlockIdentifier): Block | undefined {
+  public getBlock(
+    blockIdentifier: BlockIdentifier
+  ): Block<BSchema> | undefined {
     const id =
       typeof blockIdentifier === "string"
         ? blockIdentifier
         : blockIdentifier.id;
-    let newBlock: Block | undefined = undefined;
+    let newBlock: Block<BSchema> | undefined = undefined;
 
     this._tiptapEditor.state.doc.firstChild!.descendants((node) => {
       if (typeof newBlock !== "undefined") {
@@ -245,7 +278,7 @@ export class BlockNoteEditor {
         return true;
       }
 
-      newBlock = nodeToBlock(node, this.blockCache);
+      newBlock = nodeToBlock(node, this.schema, this.blockCache);
 
       return false;
     });
@@ -259,7 +292,7 @@ export class BlockNoteEditor {
    * @param reverse Whether the blocks should be traversed in reverse order.
    */
   public forEachBlock(
-    callback: (block: Block) => boolean,
+    callback: (block: Block<BSchema>) => boolean,
     reverse: boolean = false
   ): void {
     const blocks = this.topLevelBlocks.slice();
@@ -268,7 +301,7 @@ export class BlockNoteEditor {
       blocks.reverse();
     }
 
-    function traverseBlockArray(blockArray: Block[]): boolean {
+    function traverseBlockArray(blockArray: Block<BSchema>[]): boolean {
       for (const block of blockArray) {
         if (!callback(block)) {
           return false;
@@ -289,11 +322,19 @@ export class BlockNoteEditor {
     traverseBlockArray(blocks);
   }
 
+  /**
+   * Executes a callback whenever the editor's contents change.
+   * @param callback The callback to execute.
+   */
+  public onEditorContentChange(callback: () => void) {
+    this._tiptapEditor.on("update", callback);
+  }
+
   /**
    * Gets a snapshot of the current text cursor position.
    * @returns A snapshot of the current text cursor position.
    */
-  public getTextCursorPosition(): TextCursorPosition {
+  public getTextCursorPosition(): TextCursorPosition<BSchema> {
     const { node, depth, startPos, endPos } = getBlockInfoFromPos(
       this._tiptapEditor.state.doc,
       this._tiptapEditor.state.selection.from
@@ -321,15 +362,15 @@ export class BlockNoteEditor {
     }
 
     return {
-      block: nodeToBlock(node, this.blockCache),
+      block: nodeToBlock(node, this.schema, this.blockCache),
       prevBlock:
         prevNode === undefined
           ? undefined
-          : nodeToBlock(prevNode, this.blockCache),
+          : nodeToBlock(prevNode, this.schema, this.blockCache),
       nextBlock:
         nextNode === undefined
           ? undefined
-          : nodeToBlock(nextNode, this.blockCache),
+          : nodeToBlock(nextNode, this.schema, this.blockCache),
     };
   }
 
@@ -363,7 +404,7 @@ export class BlockNoteEditor {
   /**
    * Gets a snapshot of the current selection.
    */
-  public getSelection(): Selection | undefined {
+  public getSelection(): Selection<BSchema> | undefined {
     if (
       this._tiptapEditor.state.selection.from ===
       this._tiptapEditor.state.selection.to
@@ -371,7 +412,7 @@ export class BlockNoteEditor {
       return undefined;
     }
 
-    const blocks: Block[] = [];
+    const blocks: Block<BSchema>[] = [];
 
     this._tiptapEditor.state.doc.descendants((node, pos) => {
       if (node.type.spec.group !== "blockContent") {
@@ -388,6 +429,7 @@ export class BlockNoteEditor {
       blocks.push(
         nodeToBlock(
           this._tiptapEditor.state.doc.resolve(pos).node(),
+          this.schema,
           this.blockCache
         )
       );
@@ -423,7 +465,7 @@ export class BlockNoteEditor {
    * `referenceBlock`. Inserts the blocks at the start of the existing block's children if "nested" is used.
    */
   public insertBlocks(
-    blocksToInsert: PartialBlock[],
+    blocksToInsert: PartialBlock<BSchema>[],
     referenceBlock: BlockIdentifier,
     placement: "before" | "after" | "nested" = "before"
   ): void {
@@ -437,7 +479,10 @@ export class BlockNoteEditor {
    * @param blockToUpdate The block that should be updated.
    * @param update A partial block which defines how the existing block should be changed.
    */
-  public updateBlock(blockToUpdate: BlockIdentifier, update: PartialBlock) {
+  public updateBlock(
+    blockToUpdate: BlockIdentifier,
+    update: PartialBlock<BSchema>
+  ) {
     updateBlock(blockToUpdate, update, this._tiptapEditor);
   }
 
@@ -458,7 +503,7 @@ export class BlockNoteEditor {
    */
   public replaceBlocks(
     blocksToRemove: BlockIdentifier[],
-    blocksToInsert: PartialBlock[]
+    blocksToInsert: PartialBlock<BSchema>[]
   ) {
     replaceBlocks(blocksToRemove, blocksToInsert, this._tiptapEditor);
   }
@@ -504,6 +549,8 @@ export class BlockNoteEditor {
     ]);
     const colorStyles = new Set<ColorStyle>(["textColor", "backgroundColor"]);
 
+    this._tiptapEditor.view.focus();
+
     for (const [style, value] of Object.entries(styles)) {
       if (toggleStyles.has(style as ToggledStyle)) {
         this._tiptapEditor.commands.setMark(style);
@@ -518,6 +565,8 @@ export class BlockNoteEditor {
    * @param styles The styles to remove.
    */
   public removeStyles(styles: Styles) {
+    this._tiptapEditor.view.focus();
+
     for (const style of Object.keys(styles)) {
       this._tiptapEditor.commands.unsetMark(style);
     }
@@ -537,6 +586,8 @@ export class BlockNoteEditor {
     ]);
     const colorStyles = new Set<ColorStyle>(["textColor", "backgroundColor"]);
 
+    this._tiptapEditor.view.focus();
+
     for (const [style, value] of Object.entries(styles)) {
       if (toggleStyles.has(style as ToggledStyle)) {
         this._tiptapEditor.commands.toggleMark(style);
@@ -632,7 +683,7 @@ export class BlockNoteEditor {
    * @param blocks An array of blocks that should be serialized into HTML.
    * @returns The blocks, serialized as an HTML string.
    */
-  public async blocksToHTML(blocks: Block[]): Promise<string> {
+  public async blocksToHTML(blocks: Block<BSchema>[]): Promise<string> {
     return blocksToHTML(blocks, this._tiptapEditor.schema);
   }
 
@@ -643,8 +694,8 @@ export class BlockNoteEditor {
    * @param html The HTML string to parse blocks from.
    * @returns The blocks parsed from the HTML string.
    */
-  public async HTMLToBlocks(html: string): Promise<Block[]> {
-    return HTMLToBlocks(html, this._tiptapEditor.schema);
+  public async HTMLToBlocks(html: string): Promise<Block<BSchema>[]> {
+    return HTMLToBlocks(html, this.schema, this._tiptapEditor.schema);
   }
 
   /**
@@ -653,7 +704,7 @@ export class BlockNoteEditor {
    * @param blocks An array of blocks that should be serialized into Markdown.
    * @returns The blocks, serialized as a Markdown string.
    */
-  public async blocksToMarkdown(blocks: Block[]): Promise<string> {
+  public async blocksToMarkdown(blocks: Block<BSchema>[]): Promise<string> {
     return blocksToMarkdown(blocks, this._tiptapEditor.schema);
   }
 
@@ -664,8 +715,8 @@ export class BlockNoteEditor {
    * @param markdown The Markdown string to parse blocks from.
    * @returns The blocks parsed from the Markdown string.
    */
-  public async markdownToBlocks(markdown: string): Promise<Block[]> {
-    return markdownToBlocks(markdown, this._tiptapEditor.schema);
+  public async markdownToBlocks(markdown: string): Promise<Block<BSchema>[]> {
+    return markdownToBlocks(markdown, this.schema, this._tiptapEditor.schema);
   }
 
   /**

package/src/BlockNoteExtensions.ts

@@ -20,15 +20,20 @@ import styles from "./editor.module.css";
 import { BackgroundColorExtension } from "./extensions/BackgroundColor/BackgroundColorExtension";
 import { BackgroundColorMark } from "./extensions/BackgroundColor/BackgroundColorMark";
 import { blocks } from "./extensions/Blocks";
+import { BlockSchema } from "./extensions/Blocks/api/blockTypes";
+import { CustomBlockSerializerExtension } from "./extensions/Blocks/api/serialization";
 import blockStyles from "./extensions/Blocks/nodes/Block.module.css";
 import { BlockSideMenuFactory } from "./extensions/DraggableBlocks/BlockSideMenuFactoryTypes";
-import { DraggableBlocksExtension } from "./extensions/DraggableBlocks/DraggableBlocksExtension";
-import { FormattingToolbarExtension } from "./extensions/FormattingToolbar/FormattingToolbarExtension";
+import { createDraggableBlocksExtension } from "./extensions/DraggableBlocks/DraggableBlocksExtension";
+import { createFormattingToolbarExtension } from "./extensions/FormattingToolbar/FormattingToolbarExtension";
 import { FormattingToolbarFactory } from "./extensions/FormattingToolbar/FormattingToolbarFactoryTypes";
 import HyperlinkMark from "./extensions/HyperlinkToolbar/HyperlinkMark";
 import { HyperlinkToolbarFactory } from "./extensions/HyperlinkToolbar/HyperlinkToolbarFactoryTypes";
 import { Placeholder } from "./extensions/Placeholder/PlaceholderExtension";
-import { BaseSlashMenuItem, SlashMenuExtension } from "./extensions/SlashMenu";
+import {
+  BaseSlashMenuItem,
+  createSlashMenuExtension,
+} from "./extensions/SlashMenu";
 import { TextAlignmentExtension } from "./extensions/TextAlignment/TextAlignmentExtension";
 import { TextColorExtension } from "./extensions/TextColor/TextColorExtension";
 import { TextColorMark } from "./extensions/TextColor/TextColorMark";
@@ -36,20 +41,21 @@ import { TrailingNode } from "./extensions/TrailingNode/TrailingNodeExtension";
 import UniqueID from "./extensions/UniqueID/UniqueID";
 import { SuggestionsMenuFactory } from "./shared/plugins/suggestion/SuggestionsMenuFactoryTypes";
 
-export type UiFactories = Partial<{
-  formattingToolbarFactory: FormattingToolbarFactory;
+export type UiFactories<BSchema extends BlockSchema> = Partial<{
+  formattingToolbarFactory: FormattingToolbarFactory<BSchema>;
   hyperlinkToolbarFactory: HyperlinkToolbarFactory;
-  slashMenuFactory: SuggestionsMenuFactory<BaseSlashMenuItem>;
-  blockSideMenuFactory: BlockSideMenuFactory;
+  slashMenuFactory: SuggestionsMenuFactory<BaseSlashMenuItem<BSchema>>;
+  blockSideMenuFactory: BlockSideMenuFactory<BSchema>;
 }>;
 
 /**
  * Get all the Tiptap extensions BlockNote is configured with by default
  */
-export const getBlockNoteExtensions = (opts: {
-  editor: BlockNoteEditor;
-  uiFactories: UiFactories;
-  slashCommands: BaseSlashMenuItem[];
+export const getBlockNoteExtensions = <BSchema extends BlockSchema>(opts: {
+  editor: BlockNoteEditor<BSchema>;
+  uiFactories: UiFactories<BSchema>;
+  slashCommands: BaseSlashMenuItem<any>[]; // couldn't fix type, see https://github.com/TypeCellOS/BlockNote/pull/191#discussion_r1210708771
+  blockSchema: BSchema;
   collaboration?: {
     fragment: Y.XmlFragment;
     user: {
@@ -101,6 +107,10 @@ export const getBlockNoteExtensions = (opts: {
 
     // custom blocks:
     ...blocks,
+    ...Object.values(opts.blockSchema).map((blockSpec) =>
+      blockSpec.node.configure({ editor: opts.editor })
+    ),
+    CustomBlockSerializerExtension,
 
     Dropcursor.configure({ width: 5, color: "#ddeeff" }),
     // This needs to be at the bottom of this list, because Key events (such as enter, when selecting a /command),
@@ -147,7 +157,7 @@ export const getBlockNoteExtensions = (opts: {
 
   if (opts.uiFactories.blockSideMenuFactory) {
     ret.push(
-      DraggableBlocksExtension.configure({
+      createDraggableBlocksExtension<BSchema>().configure({
         editor: opts.editor,
         blockSideMenuFactory: opts.uiFactories.blockSideMenuFactory,
       })
@@ -156,7 +166,7 @@ export const getBlockNoteExtensions = (opts: {
 
   if (opts.uiFactories.formattingToolbarFactory) {
     ret.push(
-      FormattingToolbarExtension.configure({
+      createFormattingToolbarExtension<BSchema>().configure({
         editor: opts.editor,
         formattingToolbarFactory: opts.uiFactories.formattingToolbarFactory,
       })
@@ -175,7 +185,7 @@ export const getBlockNoteExtensions = (opts: {
 
   if (opts.uiFactories.slashMenuFactory) {
     ret.push(
-      SlashMenuExtension.configure({
+      createSlashMenuExtension<BSchema>().configure({
         editor: opts.editor,
         commands: opts.slashCommands,
         slashMenuFactory: opts.uiFactories.slashMenuFactory,

package/src/api/blockManipulation/blockManipulation.test.ts

@@ -1,5 +1,6 @@
 import { afterEach, beforeEach, describe, expect, it } from "vitest";
 import { Block, BlockNoteEditor, PartialBlock } from "../..";
+import { DefaultBlockSchema } from "../../extensions/Blocks/api/defaultBlocks";
 
 let editor: BlockNoteEditor;
 
@@ -14,11 +15,13 @@ function waitForEditor() {
   });
 }
 
-let singleBlock: PartialBlock;
+let singleBlock: PartialBlock<DefaultBlockSchema>;
 
-let multipleBlocks: PartialBlock[];
+let multipleBlocks: PartialBlock<DefaultBlockSchema>[];
 
-let insert: (placement: "before" | "nested" | "after") => Block[];
+let insert: (
+  placement: "before" | "nested" | "after"
+) => Block<DefaultBlockSchema>[];
 
 beforeEach(() => {
   (window as Window & { __TEST_OPTIONS?: {} }).__TEST_OPTIONS = {};

package/src/api/blockManipulation/blockManipulation.ts

@@ -2,13 +2,14 @@ import { Editor } from "@tiptap/core";
 import { Node } from "prosemirror-model";
 import {
   BlockIdentifier,
+  BlockSchema,
   PartialBlock,
 } from "../../extensions/Blocks/api/blockTypes";
 import { blockToNode } from "../nodeConversions/nodeConversions";
 import { getNodeById } from "../util/nodeUtil";
 
-export function insertBlocks(
-  blocksToInsert: PartialBlock[],
+export function insertBlocks<BSchema extends BlockSchema>(
+  blocksToInsert: PartialBlock<BSchema>[],
   referenceBlock: BlockIdentifier,
   placement: "before" | "after" | "nested" = "before",
   editor: Editor
@@ -56,9 +57,9 @@ export function insertBlocks(
   editor.view.dispatch(editor.state.tr.insert(insertionPos, nodesToInsert));
 }
 
-export function updateBlock(
+export function updateBlock<BSchema extends BlockSchema>(
   blockToUpdate: BlockIdentifier,
-  update: PartialBlock,
+  update: PartialBlock<BSchema>,
   editor: Editor
 ) {
   const id =
@@ -115,9 +116,9 @@ export function removeBlocks(
   }
 }
 
-export function replaceBlocks(
+export function replaceBlocks<BSchema extends BlockSchema>(
   blocksToRemove: BlockIdentifier[],
-  blocksToInsert: PartialBlock[],
+  blocksToInsert: PartialBlock<BSchema>[],
   editor: Editor
 ) {
   insertBlocks(blocksToInsert, blocksToRemove[0], "before", editor);

package/src/api/formatConversions/formatConversions.test.ts

@@ -1,25 +1,30 @@
 import { afterEach, beforeEach, describe, expect, it } from "vitest";
 import { Block, BlockNoteEditor } from "../..";
 import UniqueID from "../../extensions/UniqueID/UniqueID";
+import { DefaultBlockSchema } from "../../extensions/Blocks/api/defaultBlocks";
 
 let editor: BlockNoteEditor;
 
-let nonNestedBlocks: Block[];
+let nonNestedBlocks: Block<DefaultBlockSchema>[];
 let nonNestedHTML: string;
 let nonNestedMarkdown: string;
 
-let nestedBlocks: Block[];
+let nestedBlocks: Block<DefaultBlockSchema>[];
 // let nestedHTML: string;
 // let nestedMarkdown: string;
 
-let styledBlocks: Block[];
+let styledBlocks: Block<DefaultBlockSchema>[];
 let styledHTML: string;
 let styledMarkdown: string;
 
-let complexBlocks: Block[];
+let complexBlocks: Block<DefaultBlockSchema>[];
 // let complexHTML: string;
 // let complexMarkdown: string;
 
+function removeInlineContentClass(html: string) {
+  return html.replace(/ class="_inlineContent_1c48ad"/g, "");
+}
+
 beforeEach(() => {
   (window as Window & { __TEST_OPTIONS?: {} }).__TEST_OPTIONS = {};
 
@@ -665,7 +670,7 @@ describe("Non-Nested Block/HTML/Markdown Conversions", () => {
   it("Convert non-nested blocks to HTML", async () => {
     const output = await editor.blocksToHTML(nonNestedBlocks);
 
-    expect(output).toMatchSnapshot();
+    expect(removeInlineContentClass(output)).toMatchSnapshot();
   });
 
   it("Convert non-nested blocks to Markdown", async () => {
@@ -691,7 +696,7 @@ describe("Nested Block/HTML/Markdown Conversions", () => {
   it("Convert nested blocks to HTML", async () => {
     const output = await editor.blocksToHTML(nestedBlocks);
 
-    expect(output).toMatchSnapshot();
+    expect(removeInlineContentClass(output)).toMatchSnapshot();
   });
 
   it("Convert nested blocks to Markdown", async () => {
@@ -717,7 +722,7 @@ describe("Styled Block/HTML/Markdown Conversions", () => {
   it("Convert styled blocks to HTML", async () => {
     const output = await editor.blocksToHTML(styledBlocks);
 
-    expect(output).toMatchSnapshot();
+    expect(removeInlineContentClass(output)).toMatchSnapshot();
   });
 
   it("Convert styled blocks to Markdown", async () => {
@@ -743,7 +748,7 @@ describe("Complex Block/HTML/Markdown Conversions", () => {
   it("Convert complex blocks to HTML", async () => {
     const output = await editor.blocksToHTML(complexBlocks);
 
-    expect(output).toMatchSnapshot();
+    expect(removeInlineContentClass(output)).toMatchSnapshot();
   });
 
   it("Convert complex blocks to Markdown", async () => {