@prosekit/extensions 0.14.1 → 0.14.2

package/dist/{table-UJVYsrB7.js → table-B81i9oH9.js}

@@ -1,6 +1,6 @@
 import { t as defineDropIndicator } from "./drop-indicator-DJq8pF92.js";
-import { defaultBlockAt, defineCommands, defineNodeSpec, definePlugin, getNodeType, insertNode, union } from "@prosekit/core";
 import { TextSelection } from "@prosekit/pm/state";
+import { defaultBlockAt, defineCommands, defineNodeSpec, definePlugin, getNodeType, insertNode, union } from "@prosekit/core";
 import { CellSelection, TableMap, addColumnAfter, addColumnBefore, addRowAfter, addRowBefore, columnResizing, deleteCellSelection, deleteColumn, deleteRow, deleteTable, findCellPos, findCellRange, findTable, mergeCells, moveTableColumn, moveTableRow, splitCell, tableEditing, tableNodes } from "prosemirror-tables";
 
 //#region src/table/table-commands/exit-table.ts
@@ -82,31 +82,6 @@ function isCellSelection(value) {
 	return value instanceof CellSelection;
 }
 
-//#endregion
-//#region src/table/table-commands/select-table.ts
-/**
-* @public
-*/
-function selectTable(options) {
-	return (state, dispatch) => {
-		const table = findTable(options?.pos ? state.doc.resolve(options.pos) : state.selection.$anchor);
-		if (!table) return false;
-		const map = TableMap.get(table.node);
-		if (map.map.length === 0) return false;
-		if (dispatch) {
-			let tr = state.tr;
-			const firstCellPosInTable = map.map[0];
-			const lastCellPosInTable = map.map[map.map.length - 1];
-			const firstCellPos = table.pos + firstCellPosInTable + 1;
-			const lastCellPos = table.pos + lastCellPosInTable + 1;
-			const selection = new CellSelection(tr.doc.resolve(firstCellPos), tr.doc.resolve(lastCellPos));
-			tr = tr.setSelection(selection);
-			dispatch?.(tr);
-		}
-		return true;
-	};
-}
-
 //#endregion
 //#region src/table/table-commands/select-table-cell.ts
 /**
@@ -160,6 +135,31 @@ function selectTableRow(options) {
 	};
 }
 
+//#endregion
+//#region src/table/table-commands/select-table.ts
+/**
+* @public
+*/
+function selectTable(options) {
+	return (state, dispatch) => {
+		const table = findTable(options?.pos ? state.doc.resolve(options.pos) : state.selection.$anchor);
+		if (!table) return false;
+		const map = TableMap.get(table.node);
+		if (map.map.length === 0) return false;
+		if (dispatch) {
+			let tr = state.tr;
+			const firstCellPosInTable = map.map[0];
+			const lastCellPosInTable = map.map[map.map.length - 1];
+			const firstCellPos = table.pos + firstCellPosInTable + 1;
+			const lastCellPos = table.pos + lastCellPosInTable + 1;
+			const selection = new CellSelection(tr.doc.resolve(firstCellPos), tr.doc.resolve(lastCellPos));
+			tr = tr.setSelection(selection);
+			dispatch?.(tr);
+		}
+		return true;
+	};
+}
+
 //#endregion
 //#region src/table/table-commands.ts
 /**
@@ -284,5 +284,5 @@ function defineTable() {
 }
 
 //#endregion
-export { insertTable as _, defineTableSpec as a, defineTableCommands as c, selectTableCell as d, selectTable as f, moveTableColumn as g, moveTableRow as h, defineTableRowSpec as i, selectTableRow as l, isCellSelection as m, defineTableCellSpec as n, defineTablePlugins as o, findTable as p, defineTableHeaderCellSpec as r, defineTableDropIndicator as s, defineTable as t, selectTableColumn as u, exitTable as v };
-//# sourceMappingURL=table-UJVYsrB7.js.map
+export { insertTable as _, defineTableSpec as a, defineTableCommands as c, selectTableColumn as d, selectTableCell as f, moveTableColumn as g, moveTableRow as h, defineTableRowSpec as i, selectTable as l, isCellSelection as m, defineTableCellSpec as n, defineTablePlugins as o, findTable as p, defineTableHeaderCellSpec as r, defineTableDropIndicator as s, defineTable as t, selectTableRow as u, exitTable as v };
+//# sourceMappingURL=table-B81i9oH9.js.map

package/dist/table-B81i9oH9.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"table-B81i9oH9.js","names":[],"sources":["../src/table/table-commands/exit-table.ts","../src/table/table-commands/insert-table.ts","../src/table/table-utils.ts","../src/table/table-commands/select-table-cell.ts","../src/table/table-commands/select-table-column.ts","../src/table/table-commands/select-table-row.ts","../src/table/table-commands/select-table.ts","../src/table/table-commands.ts","../src/table/table-drop-indicator.ts","../src/table/table-plugins.ts","../src/table/table-spec.ts","../src/table/table.ts"],"sourcesContent":["import { defaultBlockAt } from '@prosekit/core'\nimport { TextSelection } from '@prosekit/pm/state'\nimport type { Command } from '@prosekit/pm/state'\nimport type { TableRole } from 'prosemirror-tables'\n\n/**\n * When the selection is in a table node, create a default block after the table\n * table, and move the cursor there.\n *\n * @public\n */\nexport const exitTable: Command = (state, dispatch) => {\n  const { $head, $anchor } = state.selection\n\n  if (!$head.sameParent($anchor)) {\n    return false\n  }\n\n  let tableStart = -1\n  let tableDepth = -1\n  for (let depth = $head.depth; depth >= 0; depth--) {\n    const node = $head.node(depth)\n    if ((node.type.spec.tableRole as TableRole) === 'table') {\n      tableStart = $head.before(depth)\n      tableDepth = depth\n    }\n  }\n\n  if (tableStart < 0 || tableDepth <= 0) {\n    return false\n  }\n\n  const above = $head.node(tableDepth - 1)\n  const after = $head.indexAfter(tableDepth - 1)\n  const type = defaultBlockAt(above.contentMatchAt(after))\n  const node = type?.createAndFill()\n\n  if (!type || !node || !above.canReplaceWith(after, after, type)) {\n    return false\n  }\n\n  if (dispatch) {\n    const pos = $head.after(tableDepth)\n    const tr = state.tr.replaceWith(pos, pos, node)\n    tr.setSelection(TextSelection.near(tr.doc.resolve(pos), 1))\n    dispatch(tr.scrollIntoView())\n  }\n  return true\n}\n","import { getNodeType, insertNode } from '@prosekit/core'\nimport type { Schema } from '@prosekit/pm/model'\nimport type { Command } from '@prosekit/pm/state'\n\nfunction createEmptyTable(\n  schema: Schema,\n  row: number,\n  col: number,\n  header: boolean,\n) {\n  const tableType = getNodeType(schema, 'table')\n  const tableRowType = getNodeType(schema, 'tableRow')\n  const tableCellType = getNodeType(schema, 'tableCell')\n  const tableHeaderCellType = getNodeType(schema, 'tableHeaderCell')\n\n  if (header) {\n    const headerCell = tableHeaderCellType.createAndFill()!\n    const headerCells = repeat(headerCell, col)\n    const headerRow = tableRowType.createAndFill(null, headerCells)!\n\n    const bodyCell = tableCellType.createAndFill()!\n    const bodyCells = repeat(bodyCell, col)\n    const bodyRow = tableRowType.createAndFill(null, bodyCells)!\n    const bodyRows = repeat(bodyRow, row - 1)\n\n    return tableType.createAndFill(null, [headerRow, ...bodyRows])!\n  } else {\n    const bodyCell = tableCellType.createAndFill()!\n    const bodyCells = repeat(bodyCell, col)\n    const bodyRow = tableRowType.createAndFill(null, bodyCells)!\n    const bodyRows = repeat(bodyRow, row)\n\n    return tableType.createAndFill(null, bodyRows)!\n  }\n}\n\nfunction repeat<T>(node: T, length: number): T[] {\n  return Array<T>(length).fill(node)\n}\n\n/**\n * @public\n */\nexport interface InsertTableOptions {\n  /**\n   * The number of rows in the table.\n   */\n  row: number\n\n  /**\n   * The number of columns in the table.\n   */\n  col: number\n\n  /**\n   * Whether the table has a header row.\n   *\n   * @default false\n   */\n  header?: boolean\n}\n\n/**\n * Insert a table node with the given number of rows and columns, and optionally\n * a header row.\n *\n * @param options\n *\n * @public\n */\nexport function insertTable(options: InsertTableOptions): Command {\n  return (state, dispatch, view) => {\n    const { row, col, header = false } = options\n    const table = createEmptyTable(state.schema, row, col, header)\n    return insertNode({ node: table })(state, dispatch, view)\n  }\n}\n","import { CellSelection, findCellPos, findCellRange, findTable } from 'prosemirror-tables'\n\nexport { findCellPos, findCellRange, findTable }\n\n/**\n * Checks if the given object is a `CellSelection` instance.\n *\n * @public\n */\nexport function isCellSelection(value: unknown): value is CellSelection {\n  return value instanceof CellSelection\n}\n","import type { Command } from '@prosekit/pm/state'\nimport { CellSelection } from 'prosemirror-tables'\n\nimport { findCellPos } from '../table-utils.ts'\n\n/**\n * @public\n */\nexport interface SelectTableCellOptions {\n  /**\n   * A hit position of the table cell to select from. By default, the selection\n   * anchor will be used.\n   */\n  pos?: number\n}\n\n/**\n * @public\n */\nexport function selectTableCell(options?: SelectTableCellOptions): Command {\n  return (state, dispatch) => {\n    const $cellPos = findCellPos(\n      state.doc,\n      options?.pos ?? state.selection.anchor,\n    )\n    if (!$cellPos) {\n      return false\n    }\n    if (dispatch) {\n      const selection = new CellSelection($cellPos)\n      dispatch(state.tr.setSelection(selection))\n    }\n    return true\n  }\n}\n","import type { Command } from '@prosekit/pm/state'\nimport { CellSelection } from 'prosemirror-tables'\n\nimport { findCellRange } from '../table-utils.ts'\n\n/**\n * @public\n */\nexport interface SelectTableColumnOptions {\n  /**\n   * A hit position of the table cell to select from. By default, the selection\n   * anchor will be used.\n   */\n  anchor?: number\n\n  /**\n   * A hit position of the table cell to select to. By default, the selection\n   * head will be used.\n   */\n  head?: number\n}\n\n/**\n * @public\n */\nexport function selectTableColumn(options?: SelectTableColumnOptions): Command {\n  return (state, dispatch) => {\n    const range = findCellRange(state.selection, options?.anchor, options?.head)\n    if (!range) {\n      return false\n    }\n    if (dispatch) {\n      const [$anchorCell, $headCell] = range\n      const selection = CellSelection.colSelection($anchorCell, $headCell)\n      dispatch(state.tr.setSelection(selection))\n    }\n    return true\n  }\n}\n","import type { Command } from '@prosekit/pm/state'\nimport { CellSelection } from 'prosemirror-tables'\n\nimport { findCellRange } from '../table-utils.ts'\n\n/**\n * @public\n */\nexport interface SelectTableRowOptions {\n  /**\n   * A hit position of the table cell to select from. By default, the selection\n   * anchor will be used.\n   */\n  anchor?: number\n\n  /**\n   * A hit position of the table cell to select to. By default, the selection\n   * head will be used.\n   */\n  head?: number\n}\n\n/**\n * @public\n */\nexport function selectTableRow(options?: SelectTableRowOptions): Command {\n  return (state, dispatch) => {\n    const range = findCellRange(state.selection, options?.anchor, options?.head)\n    if (!range) {\n      return false\n    }\n    if (dispatch) {\n      const [$anchorCell, $headCell] = range\n      const selection = CellSelection.rowSelection($anchorCell, $headCell)\n      dispatch(state.tr.setSelection(selection))\n    }\n    return true\n  }\n}\n","import type { Command } from '@prosekit/pm/state'\nimport { CellSelection, TableMap } from 'prosemirror-tables'\n\nimport { findTable } from '../table-utils.ts'\n\n/**\n * @public\n */\nexport interface SelectTableOptions {\n  /**\n   * A hit position of the table to select from. By default, the selection\n   * anchor will be used.\n   */\n  pos?: number\n}\n\n/**\n * @public\n */\nexport function selectTable(options?: SelectTableOptions): Command {\n  return (state, dispatch) => {\n    const $pos = options?.pos\n      ? state.doc.resolve(options.pos)\n      : state.selection.$anchor\n    const table = findTable($pos)\n    if (!table) {\n      return false\n    }\n    const map = TableMap.get(table.node)\n    if (map.map.length === 0) {\n      return false\n    }\n    if (dispatch) {\n      let tr = state.tr\n      const firstCellPosInTable = map.map[0]\n      const lastCellPosInTable = map.map[map.map.length - 1]\n      const firstCellPos = table.pos + firstCellPosInTable + 1\n      const lastCellPos = table.pos + lastCellPosInTable + 1\n      const $firstCellPos = tr.doc.resolve(firstCellPos)\n      const $lastCellPos = tr.doc.resolve(lastCellPos)\n      const selection = new CellSelection($firstCellPos, $lastCellPos)\n      tr = tr.setSelection(selection)\n      dispatch?.(tr)\n    }\n    return true\n  }\n}\n","import { defineCommands, type Extension } from '@prosekit/core'\nimport {\n  addColumnAfter,\n  addColumnBefore,\n  addRowAfter,\n  addRowBefore,\n  deleteColumn,\n  deleteRow,\n  deleteTable,\n  mergeCells,\n  splitCell,\n} from 'prosemirror-tables'\n\nimport { deleteCellSelection } from './table-commands/delete-cell-selection.ts'\nimport { exitTable } from './table-commands/exit-table.ts'\nimport { insertTable, type InsertTableOptions } from './table-commands/insert-table.ts'\nimport { moveTableColumn, type MoveTableColumnOptions } from './table-commands/move-table-column.ts'\nimport { moveTableRow, type MoveTableRowOptions } from './table-commands/move-table-row.ts'\nimport { selectTableCell, type SelectTableCellOptions } from './table-commands/select-table-cell.ts'\nimport { selectTableColumn, type SelectTableColumnOptions } from './table-commands/select-table-column.ts'\nimport { selectTableRow, type SelectTableRowOptions } from './table-commands/select-table-row.ts'\nimport { selectTable, type SelectTableOptions } from './table-commands/select-table.ts'\n\n/**\n * @internal\n */\nexport type TableCommandsExtension = Extension<{\n  Commands: {\n    insertTable: [options: InsertTableOptions]\n    exitTable: []\n\n    selectTable: [options?: SelectTableOptions]\n    selectTableCell: [options?: SelectTableCellOptions]\n    selectTableColumn: [options?: SelectTableColumnOptions]\n    selectTableRow: [options?: SelectTableRowOptions]\n\n    addTableColumnBefore: []\n    addTableColumnAfter: []\n    addTableRowAbove: []\n    addTableRowBelow: []\n\n    deleteTable: []\n    deleteTableColumn: []\n    deleteTableRow: []\n    deleteCellSelection: []\n\n    mergeTableCells: []\n    splitTableCell: []\n\n    moveTableRow: [options: MoveTableRowOptions]\n    moveTableColumn: [options: MoveTableColumnOptions]\n  }\n}>\n\n/**\n * Adds commands for working with `table` nodes.\n *\n * @public\n */\nexport function defineTableCommands(): TableCommandsExtension {\n  return defineCommands({\n    insertTable,\n    exitTable: () => exitTable,\n\n    selectTable,\n    selectTableCell,\n    selectTableColumn,\n    selectTableRow,\n\n    addTableColumnBefore: () => addColumnBefore,\n    addTableColumnAfter: () => addColumnAfter,\n    addTableRowAbove: () => addRowBefore,\n    addTableRowBelow: () => addRowAfter,\n\n    deleteTable: () => deleteTable,\n    deleteTableColumn: () => deleteColumn,\n    deleteTableRow: () => deleteRow,\n    deleteCellSelection: () => deleteCellSelection,\n\n    mergeTableCells: () => mergeCells,\n    splitTableCell: () => splitCell,\n\n    moveTableRow,\n    moveTableColumn,\n  })\n}\n","import type { PlainExtension } from '@prosekit/core'\n\nimport type { DragEventHandler } from '../drop-indicator/index.ts'\nimport { defineDropIndicator } from '../drop-indicator/index.ts'\n\n/**\n * Hides the drop indicator when dragging a table column or row by using the\n * table handle.\n *\n * @internal\n */\nexport function defineTableDropIndicator(): PlainExtension {\n  return defineDropIndicator({\n    onDrag,\n  })\n}\n\nconst matchMap = new WeakMap<DataTransfer, boolean>()\n\nconst onDrag: DragEventHandler = ({ event }): boolean => {\n  const dataTransfer = event.dataTransfer\n  if (!dataTransfer) return true\n\n  let match: boolean\n\n  if (matchMap.has(dataTransfer)) {\n    match = matchMap.get(dataTransfer)!\n  } else {\n    // On Safari, accessing `dataTransfer.types` is more than 10x slower than in\n    // Chrome. This becomes a bottleneck when `onDrag` is called frequently, so\n    // we cache the result in a WeakMap.\n    const types = dataTransfer.types\n    match = types.includes('application/x-prosekit-table-handle-drag')\n    matchMap.set(dataTransfer, match)\n  }\n\n  // Don't show the drop indicator when the drag event has\n  // \"application/x-prosekit-table-handle-drag\" type.\n  return !match\n}\n","import { definePlugin, type PlainExtension } from '@prosekit/core'\nimport { columnResizing, tableEditing } from 'prosemirror-tables'\n\n/**\n * @public\n */\nexport function defineTablePlugins(): PlainExtension {\n  return definePlugin([tableEditing(), columnResizing()])\n}\n","import { defineNodeSpec, type Extension } from '@prosekit/core'\nimport type { AttributeSpec, Attrs } from '@prosekit/pm/model'\nimport { tableNodes } from 'prosemirror-tables'\n\nconst cellContent = 'block+'\n\n/**\n * @public\n */\nexport interface CellAttrs {\n  colspan?: number\n  rowspan?: number\n  colwidth?: number[] | null\n}\n\nconst cellAttrs = {\n  colspan: { default: 1 },\n  rowspan: { default: 1 },\n  colwidth: { default: null },\n} satisfies Record<string, AttributeSpec>\n\n/**\n * @internal\n */\nexport type TableSpecExtension = Extension<{\n  Nodes: {\n    table: Attrs\n  }\n}>\n\nconst specs = tableNodes({\n  tableGroup: 'block',\n  cellContent,\n  cellAttributes: {},\n})\n\n/**\n * @internal\n */\nexport function defineTableSpec(): TableSpecExtension {\n  return defineNodeSpec({\n    ...specs['table'],\n    content: 'tableRow+',\n    name: 'table',\n  })\n}\n\n/**\n * @internal\n */\nexport type TableRowSpecExtension = Extension<{\n  Nodes: {\n    tableRow: Attrs\n  }\n}>\n\n/**\n * @internal\n */\nexport function defineTableRowSpec(): TableRowSpecExtension {\n  return defineNodeSpec({\n    ...specs['table_row'],\n    content: '(tableCell | tableHeaderCell)*',\n    name: 'tableRow',\n  })\n}\n\n/**\n * @internal\n */\nexport type TableCellSpecExtension = Extension<{\n  Nodes: {\n    tableCell: CellAttrs\n  }\n}>\n\n/**\n * @internal\n */\nexport function defineTableCellSpec(): TableCellSpecExtension {\n  return defineNodeSpec({\n    ...specs['table_cell'],\n    name: 'tableCell',\n    attrs: cellAttrs,\n  })\n}\n\n/**\n * @internal\n */\nexport type TableHeaderCellSpecExtension = Extension<{\n  Nodes: {\n    tableHeaderCell: CellAttrs\n  }\n}>\n\nexport function defineTableHeaderCellSpec(): TableHeaderCellSpecExtension {\n  return defineNodeSpec({\n    ...specs['table_header'],\n    name: 'tableHeaderCell',\n    attrs: cellAttrs,\n  })\n}\n","import { union, type Union } from '@prosekit/core'\n\nimport { defineTableCommands, type TableCommandsExtension } from './table-commands.ts'\nimport { defineTableDropIndicator } from './table-drop-indicator.ts'\nimport { defineTablePlugins } from './table-plugins.ts'\nimport {\n  defineTableCellSpec,\n  defineTableHeaderCellSpec,\n  defineTableRowSpec,\n  defineTableSpec,\n  type TableCellSpecExtension,\n  type TableHeaderCellSpecExtension,\n  type TableRowSpecExtension,\n  type TableSpecExtension,\n} from './table-spec.ts'\n\n/**\n * @internal\n */\nexport type TableExtension = Union<\n  [\n    TableSpecExtension,\n    TableRowSpecExtension,\n    TableCellSpecExtension,\n    TableHeaderCellSpecExtension,\n    TableCommandsExtension,\n  ]\n>\n\n/**\n * @public\n */\nexport function defineTable(): TableExtension {\n  return union(\n    defineTableSpec(),\n    defineTableRowSpec(),\n    defineTableCellSpec(),\n    defineTableHeaderCellSpec(),\n    defineTablePlugins(),\n    defineTableCommands(),\n    defineTableDropIndicator(),\n  )\n}\n"],"mappings":";;;;;;;;;;;;AAWA,MAAa,aAAsB,OAAO,aAAa;CACrD,MAAM,EAAE,OAAO,YAAY,MAAM;AAEjC,KAAI,CAAC,MAAM,WAAW,QAAQ,CAC5B,QAAO;CAGT,IAAI,aAAa;CACjB,IAAI,aAAa;AACjB,MAAK,IAAI,QAAQ,MAAM,OAAO,SAAS,GAAG,QAExC,KADa,MAAM,KAAK,MAAM,CACpB,KAAK,KAAK,cAA4B,SAAS;AACvD,eAAa,MAAM,OAAO,MAAM;AAChC,eAAa;;AAIjB,KAAI,aAAa,KAAK,cAAc,EAClC,QAAO;CAGT,MAAM,QAAQ,MAAM,KAAK,aAAa,EAAE;CACxC,MAAM,QAAQ,MAAM,WAAW,aAAa,EAAE;CAC9C,MAAM,OAAO,eAAe,MAAM,eAAe,MAAM,CAAC;CACxD,MAAM,OAAO,MAAM,eAAe;AAElC,KAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,eAAe,OAAO,OAAO,KAAK,CAC7D,QAAO;AAGT,KAAI,UAAU;EACZ,MAAM,MAAM,MAAM,MAAM,WAAW;EACnC,MAAM,KAAK,MAAM,GAAG,YAAY,KAAK,KAAK,KAAK;AAC/C,KAAG,aAAa,cAAc,KAAK,GAAG,IAAI,QAAQ,IAAI,EAAE,EAAE,CAAC;AAC3D,WAAS,GAAG,gBAAgB,CAAC;;AAE/B,QAAO;;;;;AC3CT,SAAS,iBACP,QACA,KACA,KACA,QACA;CACA,MAAM,YAAY,YAAY,QAAQ,QAAQ;CAC9C,MAAM,eAAe,YAAY,QAAQ,WAAW;CACpD,MAAM,gBAAgB,YAAY,QAAQ,YAAY;CACtD,MAAM,sBAAsB,YAAY,QAAQ,kBAAkB;AAElE,KAAI,QAAQ;EAEV,MAAM,cAAc,OADD,oBAAoB,eAAe,EACf,IAAI;EAC3C,MAAM,YAAY,aAAa,cAAc,MAAM,YAAY;EAG/D,MAAM,YAAY,OADD,cAAc,eAAe,EACX,IAAI;EAEvC,MAAM,WAAW,OADD,aAAa,cAAc,MAAM,UAAU,EAC1B,MAAM,EAAE;AAEzC,SAAO,UAAU,cAAc,MAAM,CAAC,WAAW,GAAG,SAAS,CAAC;QACzD;EAEL,MAAM,YAAY,OADD,cAAc,eAAe,EACX,IAAI;EAEvC,MAAM,WAAW,OADD,aAAa,cAAc,MAAM,UAAU,EAC1B,IAAI;AAErC,SAAO,UAAU,cAAc,MAAM,SAAS;;;AAIlD,SAAS,OAAU,MAAS,QAAqB;AAC/C,QAAO,MAAS,OAAO,CAAC,KAAK,KAAK;;;;;;;;;;AAiCpC,SAAgB,YAAY,SAAsC;AAChE,SAAQ,OAAO,UAAU,SAAS;EAChC,MAAM,EAAE,KAAK,KAAK,SAAS,UAAU;AAErC,SAAO,WAAW,EAAE,MADN,iBAAiB,MAAM,QAAQ,KAAK,KAAK,OAAO,EAC7B,CAAC,CAAC,OAAO,UAAU,KAAK;;;;;;;;;;;ACjE7D,SAAgB,gBAAgB,OAAwC;AACtE,QAAO,iBAAiB;;;;;;;;ACS1B,SAAgB,gBAAgB,SAA2C;AACzE,SAAQ,OAAO,aAAa;EAC1B,MAAM,WAAW,YACf,MAAM,KACN,SAAS,OAAO,MAAM,UAAU,OACjC;AACD,MAAI,CAAC,SACH,QAAO;AAET,MAAI,UAAU;GACZ,MAAM,YAAY,IAAI,cAAc,SAAS;AAC7C,YAAS,MAAM,GAAG,aAAa,UAAU,CAAC;;AAE5C,SAAO;;;;;;;;;ACPX,SAAgB,kBAAkB,SAA6C;AAC7E,SAAQ,OAAO,aAAa;EAC1B,MAAM,QAAQ,cAAc,MAAM,WAAW,SAAS,QAAQ,SAAS,KAAK;AAC5E,MAAI,CAAC,MACH,QAAO;AAET,MAAI,UAAU;GACZ,MAAM,CAAC,aAAa,aAAa;GACjC,MAAM,YAAY,cAAc,aAAa,aAAa,UAAU;AACpE,YAAS,MAAM,GAAG,aAAa,UAAU,CAAC;;AAE5C,SAAO;;;;;;;;;ACXX,SAAgB,eAAe,SAA0C;AACvE,SAAQ,OAAO,aAAa;EAC1B,MAAM,QAAQ,cAAc,MAAM,WAAW,SAAS,QAAQ,SAAS,KAAK;AAC5E,MAAI,CAAC,MACH,QAAO;AAET,MAAI,UAAU;GACZ,MAAM,CAAC,aAAa,aAAa;GACjC,MAAM,YAAY,cAAc,aAAa,aAAa,UAAU;AACpE,YAAS,MAAM,GAAG,aAAa,UAAU,CAAC;;AAE5C,SAAO;;;;;;;;;ACjBX,SAAgB,YAAY,SAAuC;AACjE,SAAQ,OAAO,aAAa;EAI1B,MAAM,QAAQ,UAHD,SAAS,MAClB,MAAM,IAAI,QAAQ,QAAQ,IAAI,GAC9B,MAAM,UAAU,QACS;AAC7B,MAAI,CAAC,MACH,QAAO;EAET,MAAM,MAAM,SAAS,IAAI,MAAM,KAAK;AACpC,MAAI,IAAI,IAAI,WAAW,EACrB,QAAO;AAET,MAAI,UAAU;GACZ,IAAI,KAAK,MAAM;GACf,MAAM,sBAAsB,IAAI,IAAI;GACpC,MAAM,qBAAqB,IAAI,IAAI,IAAI,IAAI,SAAS;GACpD,MAAM,eAAe,MAAM,MAAM,sBAAsB;GACvD,MAAM,cAAc,MAAM,MAAM,qBAAqB;GAGrD,MAAM,YAAY,IAAI,cAFA,GAAG,IAAI,QAAQ,aAAa,EAC7B,GAAG,IAAI,QAAQ,YAAY,CACgB;AAChE,QAAK,GAAG,aAAa,UAAU;AAC/B,cAAW,GAAG;;AAEhB,SAAO;;;;;;;;;;;ACeX,SAAgB,sBAA8C;AAC5D,QAAO,eAAe;EACpB;EACA,iBAAiB;EAEjB;EACA;EACA;EACA;EAEA,4BAA4B;EAC5B,2BAA2B;EAC3B,wBAAwB;EACxB,wBAAwB;EAExB,mBAAmB;EACnB,yBAAyB;EACzB,sBAAsB;EACtB,2BAA2B;EAE3B,uBAAuB;EACvB,sBAAsB;EAEtB;EACA;EACD,CAAC;;;;;;;;;;;ACzEJ,SAAgB,2BAA2C;AACzD,QAAO,oBAAoB,EACzB,QACD,CAAC;;AAGJ,MAAM,2BAAW,IAAI,SAAgC;AAErD,MAAM,UAA4B,EAAE,YAAqB;CACvD,MAAM,eAAe,MAAM;AAC3B,KAAI,CAAC,aAAc,QAAO;CAE1B,IAAI;AAEJ,KAAI,SAAS,IAAI,aAAa,CAC5B,SAAQ,SAAS,IAAI,aAAa;MAC7B;AAKL,UADc,aAAa,MACb,SAAS,2CAA2C;AAClE,WAAS,IAAI,cAAc,MAAM;;AAKnC,QAAO,CAAC;;;;;;;;AChCV,SAAgB,qBAAqC;AACnD,QAAO,aAAa,CAAC,cAAc,EAAE,gBAAgB,CAAC,CAAC;;;;;ACHzD,MAAM,cAAc;AAWpB,MAAM,YAAY;CAChB,SAAS,EAAE,SAAS,GAAG;CACvB,SAAS,EAAE,SAAS,GAAG;CACvB,UAAU,EAAE,SAAS,MAAM;CAC5B;AAWD,MAAM,QAAQ,WAAW;CACvB,YAAY;CACZ;CACA,gBAAgB,EAAE;CACnB,CAAC;;;;AAKF,SAAgB,kBAAsC;AACpD,QAAO,eAAe;EACpB,GAAG,MAAM;EACT,SAAS;EACT,MAAM;EACP,CAAC;;;;;AAeJ,SAAgB,qBAA4C;AAC1D,QAAO,eAAe;EACpB,GAAG,MAAM;EACT,SAAS;EACT,MAAM;EACP,CAAC;;;;;AAeJ,SAAgB,sBAA8C;AAC5D,QAAO,eAAe;EACpB,GAAG,MAAM;EACT,MAAM;EACN,OAAO;EACR,CAAC;;AAYJ,SAAgB,4BAA0D;AACxE,QAAO,eAAe;EACpB,GAAG,MAAM;EACT,MAAM;EACN,OAAO;EACR,CAAC;;;;;;;;ACrEJ,SAAgB,cAA8B;AAC5C,QAAO,MACL,iBAAiB,EACjB,oBAAoB,EACpB,qBAAqB,EACrB,2BAA2B,EAC3B,oBAAoB,EACpB,qBAAqB,EACrB,0BAA0B,CAC3B"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@prosekit/extensions",
   "type": "module",
-  "version": "0.14.1",
+  "version": "0.14.2",
   "private": false,
   "description": "A collection of common extensions for ProseKit",
   "author": {
@@ -223,16 +223,16 @@
     "prosemirror-changeset": "^2.4.0",
     "prosemirror-drop-indicator": "^0.1.3",
     "prosemirror-dropcursor": "^1.8.2",
+    "prosemirror-enter-rules": "^0.1.5",
     "prosemirror-flat-list": "^0.5.8",
     "prosemirror-gapcursor": "^1.4.0",
     "prosemirror-highlight": "^0.15.0",
+    "prosemirror-math": "^0.2.2",
     "prosemirror-search": "^1.1.0",
     "prosemirror-tables": "^1.8.5",
     "shiki": "^3.22.0",
     "@prosekit/core": "^0.10.0",
-    "@prosekit/pm": "^0.1.15",
-    "prosemirror-enter-rules": "^0.1.3",
-    "prosemirror-math": "^0.1.0"
+    "@prosekit/pm": "^0.1.15"
   },
   "peerDependencies": {
     "loro-crdt": ">= 1.10.0",
@@ -259,7 +259,7 @@
     "just-pick": "^4.2.0",
     "katex": "^0.16.28",
     "loro-crdt": "^1.10.6",
-    "loro-prosemirror": "^0.4.2",
+    "loro-prosemirror": "^0.4.3",
     "rehype-parse": "^9.0.1",
     "rehype-remark": "^10.0.1",
     "remark-gfm": "^4.0.1",
@@ -270,12 +270,13 @@
     "type-fest": "^5.4.4",
     "typescript": "~5.9.3",
     "unified": "^11.0.5",
-    "vitest": "^4.0.18",
+    "vitest": "^4.1.0-beta.4",
     "vitest-browser-commands": "^0.2.0",
     "y-prosemirror": "^1.3.7",
     "yjs": "^13.6.29",
     "@prosekit/config-tsdown": "0.0.0",
-    "@prosekit/config-vitest": "0.0.0"
+    "@prosekit/config-vitest": "0.0.0",
+    "@prosekit/config-ts": "0.0.0"
   },
   "publishConfig": {
     "dev": {}

package/src/autocomplete/autocomplete-helpers.ts

@@ -1,7 +1,7 @@
 import type { ResolvedPos } from '@prosekit/pm/model'
 import { PluginKey, type EditorState, type Transaction } from '@prosekit/pm/state'
 
-import type { AutocompleteRule } from './autocomplete-rule'
+import type { AutocompleteRule } from './autocomplete-rule.ts'
 
 export function defaultCanMatch({ state }: { state: EditorState }): boolean {
   const $pos = state.selection.$from

package/src/autocomplete/autocomplete-plugin.ts

@@ -13,8 +13,8 @@ import {
   type PredictionPluginMatching,
   type PredictionPluginState,
   type PredictionTransactionMeta,
-} from './autocomplete-helpers'
-import type { AutocompleteRule } from './autocomplete-rule'
+} from './autocomplete-helpers.ts'
+import type { AutocompleteRule } from './autocomplete-rule.ts'
 
 /**
  * Creates a plugin that handles autocomplete functionality.

package/src/autocomplete/autocomplete-rule.ts

@@ -1,6 +1,6 @@
 import type { EditorState } from '@prosekit/pm/state'
 
-import { defaultCanMatch } from './autocomplete-helpers'
+import { defaultCanMatch } from './autocomplete-helpers.ts'
 
 /**
  * Options for the {@link MatchHandler} callback.

package/src/autocomplete/autocomplete.spec.ts

@@ -2,11 +2,11 @@ import { canUseRegexLookbehind, union } from '@prosekit/core'
 import { describe, expect, it, vi } from 'vitest'
 import { keyboard } from 'vitest-browser-commands/playwright'
 
-import { defineTestExtension, setupTestFromExtension } from '../testing'
-import { inputText } from '../testing/keyboard'
+import { defineTestExtension, setupTestFromExtension } from '../testing/index.ts'
+import { inputText } from '../testing/keyboard.ts'
 
-import { defineAutocomplete } from './autocomplete'
-import { AutocompleteRule, type MatchHandler, type MatchHandlerOptions } from './autocomplete-rule'
+import { AutocompleteRule, type MatchHandler, type MatchHandlerOptions } from './autocomplete-rule.ts'
+import { defineAutocomplete } from './autocomplete.ts'
 
 function setupSlashMenu() {
   const regex = canUseRegexLookbehind() ? /(?<!\S)\/(\S.*)?$/u : /\/(\S.*)?$/u

package/src/autocomplete/autocomplete.ts

@@ -1,7 +1,7 @@
 import { defineFacet, defineFacetPayload, pluginFacet, type Extension, type PluginPayload } from '@prosekit/core'
 
-import { createAutocompletePlugin } from './autocomplete-plugin'
-import type { AutocompleteRule } from './autocomplete-rule'
+import { createAutocompletePlugin } from './autocomplete-plugin.ts'
+import type { AutocompleteRule } from './autocomplete-rule.ts'
 
 /**
  * Defines an autocomplete extension that executes logic when the text before

package/src/autocomplete/index.ts

@@ -1,4 +1,3 @@
-export { defineAutocomplete } from './autocomplete'
 export {
   AutocompleteRule,
   type AutocompleteRuleOptions,
@@ -6,4 +5,5 @@ export {
   type CanMatchPredicate,
   type MatchHandler,
   type MatchHandlerOptions,
-} from './autocomplete-rule'
+} from './autocomplete-rule.ts'
+export { defineAutocomplete } from './autocomplete.ts'

package/src/background-color/background-color-commands.spec.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from 'vitest'
 
-import { setupTest } from '../testing'
+import { setupTest } from '../testing/index.ts'
 
 describe('addBackgroundColor', () => {
   it('can add background color to text', () => {

package/src/background-color/background-color-commands.ts

@@ -1,7 +1,7 @@
 import { addMark, defineCommands, removeMark, type Extension } from '@prosekit/core'
 import type { Command } from '@prosekit/pm/state'
 
-import type { BackgroundColorAttrs } from './background-color-spec'
+import type { BackgroundColorAttrs } from './background-color-spec.ts'
 
 /**
  * @internal

package/src/background-color/background-color-spec.spec.ts

@@ -1,7 +1,7 @@
 import { formatHTML } from 'diffable-html-snapshot'
 import { describe, expect, it } from 'vitest'
 
-import { setupTest } from '../testing'
+import { setupTest } from '../testing/index.ts'
 
 describe('defineBackgroundColorSpec', () => {
   it('should render background color as inline span with style attribute', () => {

package/src/background-color/background-color.ts

@@ -1,7 +1,7 @@
 import { union, type Union } from '@prosekit/core'
 
-import { defineBackgroundColorCommands, type BackgroundColorCommandsExtension } from './background-color-commands'
-import { defineBackgroundColorSpec, type BackgroundColorSpecExtension } from './background-color-spec'
+import { defineBackgroundColorCommands, type BackgroundColorCommandsExtension } from './background-color-commands.ts'
+import { defineBackgroundColorSpec, type BackgroundColorSpecExtension } from './background-color-spec.ts'
 
 /**
  * @internal

package/src/background-color/index.ts

@@ -1,8 +1,8 @@
-export { defineBackgroundColor, type BackgroundColorExtension } from './background-color'
 export {
   addBackgroundColor,
   defineBackgroundColorCommands,
   removeBackgroundColor,
   type BackgroundColorCommandsExtension,
-} from './background-color-commands'
-export { defineBackgroundColorSpec, type BackgroundColorAttrs, type BackgroundColorSpecExtension } from './background-color-spec'
+} from './background-color-commands.ts'
+export { defineBackgroundColorSpec, type BackgroundColorAttrs, type BackgroundColorSpecExtension } from './background-color-spec.ts'
+export { defineBackgroundColor, type BackgroundColorExtension } from './background-color.ts'

package/src/blockquote/blockquote-input-rule.ts

@@ -1,6 +1,6 @@
 import type { PlainExtension } from '@prosekit/core'
 
-import { defineWrappingInputRule } from '../input-rule'
+import { defineWrappingInputRule } from '../input-rule/index.ts'
 
 /**
  * Wraps the text block in a blockquote when `>` is typed at the start of a new

package/src/blockquote/blockquote-keymap.spec.ts

@@ -1,7 +1,7 @@
 import { describe, expect, it } from 'vitest'
 import { keyboard } from 'vitest-browser-commands/playwright'
 
-import { setupTest } from '../testing'
+import { setupTest } from '../testing/index.ts'
 
 describe('blockquote keymap', () => {
   it('should wrap paragraph into blockquote with shortcut', async () => {

package/src/blockquote/blockquote-keymap.ts

@@ -6,14 +6,16 @@ function toggleBlockquoteKeybinding(): Command {
   return toggleWrap({ type: 'blockquote' })
 }
 
-function backspaceUnsetBlockquote(): Command {
-  return (state, dispatch, view) => {
-    const $pos = isAtBlockStart(state, view)
-    if ($pos?.node(-1).type.name === 'blockquote') {
-      return joinBackward(state, dispatch, view)
-    }
-    return false
+const backspaceUnsetBlockquoteCommand: Command = (state, dispatch, view): boolean => {
+  const $pos = isAtBlockStart(state, view)
+  if ($pos?.node(-1).type.name === 'blockquote') {
+    return joinBackward(state, dispatch, view)
   }
+  return false
+}
+
+function backspaceUnsetBlockquote(): Command {
+  return backspaceUnsetBlockquoteCommand
 }
 /**
  * @internal

package/src/blockquote/blockquote.ts

@@ -1,9 +1,9 @@
 import { union, type Union } from '@prosekit/core'
 
-import { defineBlockquoteCommands, type BlockquoteCommandsExtension } from './blockquote-commands'
-import { defineBlockquoteInputRule } from './blockquote-input-rule'
-import { defineBlockquoteKeymap } from './blockquote-keymap'
-import { defineBlockquoteSpec, type BlockquoteSpecExtension } from './blockquote-spec'
+import { defineBlockquoteCommands, type BlockquoteCommandsExtension } from './blockquote-commands.ts'
+import { defineBlockquoteInputRule } from './blockquote-input-rule.ts'
+import { defineBlockquoteKeymap } from './blockquote-keymap.ts'
+import { defineBlockquoteSpec, type BlockquoteSpecExtension } from './blockquote-spec.ts'
 
 /**
  * @internal

package/src/blockquote/index.ts

@@ -1,5 +1,5 @@
-export { defineBlockquote, type BlockquoteExtension } from './blockquote'
-export { defineBlockquoteCommands, type BlockquoteCommandsExtension } from './blockquote-commands'
-export { defineBlockquoteInputRule } from './blockquote-input-rule'
-export { defineBlockquoteKeymap } from './blockquote-keymap'
-export { defineBlockquoteSpec, type BlockquoteSpecExtension } from './blockquote-spec'
+export { defineBlockquoteCommands, type BlockquoteCommandsExtension } from './blockquote-commands.ts'
+export { defineBlockquoteInputRule } from './blockquote-input-rule.ts'
+export { defineBlockquoteKeymap } from './blockquote-keymap.ts'
+export { defineBlockquoteSpec, type BlockquoteSpecExtension } from './blockquote-spec.ts'
+export { defineBlockquote, type BlockquoteExtension } from './blockquote.ts'

package/src/bold/bold-input-rule.spec.ts

@@ -1,7 +1,7 @@
 import { describe, expect, it } from 'vitest'
 
-import { setupTest } from '../testing'
-import { inputText } from '../testing/keyboard'
+import { setupTest } from '../testing/index.ts'
+import { inputText } from '../testing/keyboard.ts'
 
 describe('defineBoldInputRule', () => {
   const { editor, n, m } = setupTest()

package/src/bold/bold-input-rule.ts

@@ -1,6 +1,6 @@
 import { canUseRegexLookbehind, type PlainExtension } from '@prosekit/core'
 
-import { defineMarkInputRule } from '../input-rule'
+import { defineMarkInputRule } from '../input-rule/index.ts'
 
 /**
  * @internal

package/src/bold/bold.ts

@@ -1,9 +1,9 @@
 import { union, type Union } from '@prosekit/core'
 
-import { defineBoldCommands, type BoldCommandsExtension } from './bold-commands'
-import { defineBoldInputRule } from './bold-input-rule'
-import { defineBoldKeymap } from './bold-keymap'
-import { defineBoldSpec, type BoldSpecExtension } from './bold-spec'
+import { defineBoldCommands, type BoldCommandsExtension } from './bold-commands.ts'
+import { defineBoldInputRule } from './bold-input-rule.ts'
+import { defineBoldKeymap } from './bold-keymap.ts'
+import { defineBoldSpec, type BoldSpecExtension } from './bold-spec.ts'
 
 /**
  * @internal

package/src/bold/index.ts

@@ -1,5 +1,5 @@
-export { defineBold, type BoldExtension } from './bold'
-export { defineBoldCommands, type BoldCommandsExtension } from './bold-commands'
-export { defineBoldInputRule } from './bold-input-rule'
-export { defineBoldKeymap } from './bold-keymap'
-export { defineBoldSpec, type BoldSpecExtension } from './bold-spec'
+export { defineBoldCommands, type BoldCommandsExtension } from './bold-commands.ts'
+export { defineBoldInputRule } from './bold-input-rule.ts'
+export { defineBoldKeymap } from './bold-keymap.ts'
+export { defineBoldSpec, type BoldSpecExtension } from './bold-spec.ts'
+export { defineBold, type BoldExtension } from './bold.ts'

package/src/code/code-input-rule.ts

@@ -1,6 +1,6 @@
 import { canUseRegexLookbehind, type PlainExtension } from '@prosekit/core'
 
-import { defineMarkInputRule } from '../input-rule'
+import { defineMarkInputRule } from '../input-rule/index.ts'
 
 /**
  * @internal

package/src/code/code.ts

@@ -1,9 +1,9 @@
 import { union, type Union } from '@prosekit/core'
 
-import { defineCodeCommands, type CodeCommandsExtension } from './code-commands'
-import { defineCodeInputRule } from './code-input-rule'
-import { defineCodeKeymap } from './code-keymap'
-import { defineCodeSpec, type CodeSpecExtension } from './code-spec'
+import { defineCodeCommands, type CodeCommandsExtension } from './code-commands.ts'
+import { defineCodeInputRule } from './code-input-rule.ts'
+import { defineCodeKeymap } from './code-keymap.ts'
+import { defineCodeSpec, type CodeSpecExtension } from './code-spec.ts'
 
 /**
  * @internal

package/src/code/index.ts

@@ -1,5 +1,5 @@
-export { defineCode, type CodeExtension } from './code'
-export { defineCodeCommands, type CodeCommandsExtension } from './code-commands'
-export { defineCodeInputRule } from './code-input-rule'
-export { defineCodeKeymap } from './code-keymap'
-export { defineCodeSpec, type CodeSpecExtension } from './code-spec'
+export { defineCodeCommands, type CodeCommandsExtension } from './code-commands.ts'
+export { defineCodeInputRule } from './code-input-rule.ts'
+export { defineCodeKeymap } from './code-keymap.ts'
+export { defineCodeSpec, type CodeSpecExtension } from './code-spec.ts'
+export { defineCode, type CodeExtension } from './code.ts'

package/src/code-block/code-block-commands.ts

@@ -1,6 +1,6 @@
 import { defineCommands, insertNode, setBlockType, setNodeAttrs, toggleNode, type Extension } from '@prosekit/core'
 
-import type { CodeBlockAttrs } from './code-block-types'
+import type { CodeBlockAttrs } from './code-block-types.ts'
 
 /**
  * @internal

package/src/code-block/code-block-highlight.ts

@@ -12,7 +12,16 @@ export type HighlightParser = Parser
  * @public
  */
 export type CodeBlockHighlightOptions = {
+  /**
+   * A parser instance from the `prosemirror-highlight` package.
+   */
   parser: HighlightParser
+  /**
+   * ProseMirror node types to highlight.
+   *
+   * @default ['codeBlock', 'mathBlock']
+   */
+  nodeTypes?: string[]
 }
 
 /**
@@ -27,8 +36,9 @@ export type CodeBlockHighlightOptions = {
  */
 export function defineCodeBlockHighlight({
   parser,
+  nodeTypes = ['codeBlock', 'mathBlock'],
 }: CodeBlockHighlightOptions): Extension {
   return definePlugin(
-    createHighlightPlugin({ parser }),
+    createHighlightPlugin({ parser, nodeTypes }),
   )
 }

package/src/code-block/code-block-input-rule.ts

@@ -1,9 +1,9 @@
 import type { PlainExtension } from '@prosekit/core'
 
-import { defineTextBlockEnterRule } from '../enter-rule'
-import { defineTextBlockInputRule } from '../input-rule'
+import { defineTextBlockEnterRule } from '../enter-rule/index.ts'
+import { defineTextBlockInputRule } from '../input-rule/index.ts'
 
-import type { CodeBlockAttrs } from './code-block-types'
+import type { CodeBlockAttrs } from './code-block-types.ts'
 
 /**
  * Adds input rules for `codeBlock` nodes.

package/src/code-block/code-block-shiki.ts

@@ -1,10 +1,10 @@
 import type { Extension } from '@prosekit/core'
 import type { SpecialLanguage } from 'shiki'
 
-import { defineCodeBlockHighlight } from './code-block-highlight'
-import type { ShikiBundledLanguage, ShikiBundledTheme } from './shiki-bundle'
-import type { ShikiHighlighterOptions } from './shiki-highlighter-chunk'
-import { createLazyParser } from './shiki-parser'
+import { defineCodeBlockHighlight } from './code-block-highlight.ts'
+import type { ShikiBundledLanguage, ShikiBundledTheme } from './shiki-bundle.ts'
+import type { ShikiHighlighterOptions } from './shiki-highlighter-chunk.ts'
+import { createLazyParser } from './shiki-parser.ts'
 
 /**
  * The options to configure the Shiki highlighter.
@@ -12,6 +12,13 @@ import { createLazyParser } from './shiki-parser'
  * @public
  */
 export interface CodeBlockShikiOptions extends Omit<ShikiHighlighterOptions, 'themes' | 'langs' | 'engine'> {
+  /**
+   * ProseMirror node types to highlight.
+   *
+   * @default ['codeBlock', 'mathBlock']
+   */
+  nodeTypes?: string[]
+
   /**
    * A list of Shiki themes to pre-load. The first theme in the list will be
    * used to render the code block.
@@ -46,10 +53,11 @@ export interface CodeBlockShikiOptions extends Omit<ShikiHighlighterOptions, 'th
  * @public
  */
 export function defineCodeBlockShiki({
+  nodeTypes,
   themes = ['one-dark-pro'],
   langs = ['text'],
   ...rest
 }: CodeBlockShikiOptions = {}): Extension {
   const parser = createLazyParser({ themes, langs, ...rest })
-  return defineCodeBlockHighlight({ parser })
+  return defineCodeBlockHighlight({ parser, nodeTypes })
 }

package/src/code-block/code-block-spec.spec.ts

@@ -2,8 +2,8 @@ import { jsonFromHTML } from '@prosekit/core'
 import { formatHTML } from 'diffable-html-snapshot'
 import { describe, expect, it } from 'vitest'
 
-import { setupTest } from '../testing'
-import { htmlFromMarkdown, markdownFromHTML } from '../testing/markdown'
+import { setupTest } from '../testing/index.ts'
+import { htmlFromMarkdown, markdownFromHTML } from '../testing/markdown.ts'
 
 describe('defineCodeBlockSpec', () => {
   it('can parse and serialize code blocks', () => {

package/src/code-block/code-block-spec.ts

@@ -1,6 +1,6 @@
 import { defineNodeSpec, type Extension } from '@prosekit/core'
 
-import type { CodeBlockAttrs } from './code-block-types'
+import type { CodeBlockAttrs } from './code-block-types.ts'
 
 /**
  * @internal

package/src/code-block/code-block.ts

@@ -1,9 +1,9 @@
 import { union, type Union } from '@prosekit/core'
 
-import { defineCodeBlockCommands, type CodeBlockCommandsExtension } from './code-block-commands'
-import { defineCodeBlockEnterRule, defineCodeBlockInputRule } from './code-block-input-rule'
-import { defineCodeBlockKeymap } from './code-block-keymap'
-import { defineCodeBlockSpec, type CodeBlockSpecExtension } from './code-block-spec'
+import { defineCodeBlockCommands, type CodeBlockCommandsExtension } from './code-block-commands.ts'
+import { defineCodeBlockEnterRule, defineCodeBlockInputRule } from './code-block-input-rule.ts'
+import { defineCodeBlockKeymap } from './code-block-keymap.ts'
+import { defineCodeBlockSpec, type CodeBlockSpecExtension } from './code-block-spec.ts'
 
 /**
  * @internal

package/src/code-block/index.ts

@@ -1,11 +1,11 @@
-export { defineCodeBlock, type CodeBlockExtension } from './code-block'
-export { defineCodeBlockCommands, type CodeBlockCommandsExtension } from './code-block-commands'
-export { defineCodeBlockHighlight, type CodeBlockHighlightOptions, type HighlightParser } from './code-block-highlight'
-export { defineCodeBlockEnterRule, defineCodeBlockInputRule } from './code-block-input-rule'
-export { defineCodeBlockKeymap } from './code-block-keymap'
-export { defineCodeBlockShiki } from './code-block-shiki'
-export { defineCodeBlockSpec, type CodeBlockSpecExtension } from './code-block-spec'
-export type { CodeBlockAttrs } from './code-block-types'
+export { defineCodeBlockCommands, type CodeBlockCommandsExtension } from './code-block-commands.ts'
+export { defineCodeBlockHighlight, type CodeBlockHighlightOptions, type HighlightParser } from './code-block-highlight.ts'
+export { defineCodeBlockEnterRule, defineCodeBlockInputRule } from './code-block-input-rule.ts'
+export { defineCodeBlockKeymap } from './code-block-keymap.ts'
+export { defineCodeBlockShiki } from './code-block-shiki.ts'
+export { defineCodeBlockSpec, type CodeBlockSpecExtension } from './code-block-spec.ts'
+export type { CodeBlockAttrs } from './code-block-types.ts'
+export { defineCodeBlock, type CodeBlockExtension } from './code-block.ts'
 export {
   shikiBundledLanguagesInfo,
   shikiBundledThemesInfo,
@@ -13,4 +13,4 @@ export {
   type ShikiBundledLanguageInfo,
   type ShikiBundledTheme,
   type ShikiBundledThemeInfo,
-} from './shiki-bundle'
+} from './shiki-bundle.ts'