prosemirror-highlight 0.14.0 → 0.15.1

package/README.md

@@ -157,6 +157,44 @@ export const refractorPlugin = createHighlightPlugin({ parser })
 
 </details>
 
+### With [Lezer]
+
+<details>
+<summary>Using Lezer parsers with classHighlighter</summary>
+
+```ts
+import type { Tree } from '@lezer/common'
+import { parser as cssParser } from '@lezer/css'
+import { classHighlighter } from '@lezer/highlight'
+import { parser as javascriptParser } from '@lezer/javascript'
+import { createHighlightPlugin } from 'prosemirror-highlight'
+import { createParser, type ParserOptions } from 'prosemirror-highlight/lezer'
+
+function parse({ content, language }: ParserOptions): Tree | undefined {
+  const lang = language?.toLowerCase() || ''
+  switch (lang) {
+    case 'js':
+    case 'javascript':
+    case 'ts':
+    case 'typescript':
+    case 'jsx':
+    case 'tsx':
+      return javascriptParser.parse(content)
+    case 'css':
+    case 'scss':
+    case 'sass':
+      return cssParser.parse(content)
+    default:
+      return undefined
+  }
+}
+
+const parser = createParser({ highlighter: classHighlighter, parse })
+export const lezerPlugin = createHighlightPlugin({ parser })
+```
+
+</details>
+
 ### With [Sugar high]
 
 <details>
@@ -186,6 +224,25 @@ export const sugarHighPlugin = createHighlightPlugin({ parser })
 
 </details>
 
+## Line Numbers
+
+You can add line numbers to code blocks by wrapping any parser with `withLineNumbers`:
+
+```ts
+import { createHighlightPlugin, withLineNumbers } from 'prosemirror-highlight'
+import { createParser } from 'prosemirror-highlight/shiki'
+import { getSingletonHighlighter } from 'shiki'
+
+const highlighter = await getSingletonHighlighter({
+  themes: ['github-light'],
+  langs: ['javascript', 'typescript', 'python'],
+})
+const parser = withLineNumbers(createParser(highlighter))
+export const shikiPlugin = createHighlightPlugin({ parser })
+```
+
+This inserts `<span class="line-number">` elements with class name `line-number` at the start of each line.
+
 ## Online demo
 
 [![Open in StackBlitz](https://developer.stackblitz.com/img/open_in_stackblitz.svg)](https://stackblitz.com/github/ocavue/prosemirror-highlight?file=playground%2Fmain.ts)
@@ -204,5 +261,6 @@ MIT
 [Highlight.js]: https://github.com/highlightjs/highlight.js
 [Shiki]: https://github.com/shikijs/shiki
 [refractor]: https://github.com/wooorm/refractor
+[Lezer]: https://lezer.codemirror.net
 [Prism]: https://github.com/PrismJS/prism
 [Sugar high]: https://github.com/huozhi/sugar-high

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { n as Parser, t as LanguageExtractor } from "./types-DCbyVqHc.js";
+import { n as Parser, r as ParserOptions, t as LanguageExtractor } from "./types-jXJYKXGQ.js";
 import { Plugin, Transaction } from "prosemirror-state";
 import { Decoration, DecorationSet } from "prosemirror-view";
 import { Node } from "prosemirror-model";
@@ -83,5 +83,14 @@ declare function createHighlightPlugin({
   languageExtractor?: LanguageExtractor;
 }): Plugin<HighlightPluginState>;
 //#endregion
-export { DecorationCache, type HighlightPluginState, type LanguageExtractor, type Parser, createHighlightPlugin };
+//#region src/line-number.d.ts
+/**
+ * Returns a new parser that adds line numbers to the parsed decorations.
+ *
+ * Line numbers are added as `<span class="line-number">` elements with
+ * the line number as the text content.
+ */
+declare function withLineNumbers(parser: Parser): Parser;
+//#endregion
+export { DecorationCache, type HighlightPluginState, type LanguageExtractor, type Parser, type ParserOptions, createHighlightPlugin, withLineNumbers };
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","names":[],"sources":["../src/cache.ts","../src/plugin.ts"],"mappings":";;;;;;;;;cAOa,eAAA;EAAA,QACH,KAAA;cAGN,KAAA,GAAQ,GAAA,UAAa,IAAA,EAAM,IAAA,EAAiB,WAAA,EAAa,UAAA;EAA9B;;;;EAS7B,GAAA,CAAI,GAAA,YAAW,IAAA,EAAA,IAAA,EAAA,WAAA,EAAA,UAAA;EAWQ;;;;;;;EAAvB,GAAA,CAAI,GAAA,UAAa,IAAA,EAAM,IAAA,EAAiB,WAAA,EAAa,UAAA;EApB3C;;;;;;;;EAAA,QAoCF,OAAA;EA3BO;;;;EAyCf,MAAA,CAAO,GAAA;EA9BgB;;;;;;;EAyCvB,UAAA,CAAW,EAAA,EAAI,WAAA,GAAc,eAAA;AAAA;;;;;AAjE/B;UCGiB,oBAAA;EACf,KAAA,EAAO,eAAA;EACP,WAAA,EAAa,aAAA;EACb,QAAA,EAAU,OAAA;AAAA;;;;;iBAOI,qBAAA,CAAA;EACd,MAAA;EACA,SAAA;EACA;AAAA;EDiD6B;;;;EC3C7B,MAAA,EAAQ,MAAA;EDlBE;;;;;ECyBV,SAAA;EDhBA;;;;ECsBA,iBAAA,GAAoB,iBAAA;AAAA,IAClB,MAAA,CAAO,oBAAA"}
+{"version":3,"file":"index.d.ts","names":[],"sources":["../src/cache.ts","../src/plugin.ts","../src/line-number.ts"],"mappings":";;;;;;;;;cAOa,eAAA;EAAA,QACH,KAAA;cAGN,KAAA,GAAQ,GAAA,UAAa,IAAA,EAAM,IAAA,EAAiB,WAAA,EAAa,UAAA;EAA9B;;;;EAS7B,GAAA,CAAI,GAAA,YAAW,IAAA,EAAA,IAAA,EAAA,WAAA,EAAA,UAAA;EAWQ;;;;;;;EAAvB,GAAA,CAAI,GAAA,UAAa,IAAA,EAAM,IAAA,EAAiB,WAAA,EAAa,UAAA;EApB3C;;;;;;;;EAAA,QAoCF,OAAA;EA3BO;;;;EAyCf,MAAA,CAAO,GAAA;EA9BgB;;;;;;;EAyCvB,UAAA,CAAW,EAAA,EAAI,WAAA,GAAc,eAAA;AAAA;;;;;AAjE/B;UCGiB,oBAAA;EACf,KAAA,EAAO,eAAA;EACP,WAAA,EAAa,aAAA;EACb,QAAA,EAAU,OAAA;AAAA;;;;;iBAOI,qBAAA,CAAA;EACd,MAAA;EACA,SAAA;EACA;AAAA;EDiD6B;;;;EC3C7B,MAAA,EAAQ,MAAA;EDlBE;;;;;ECyBV,SAAA;EDhBA;;;;ECsBA,iBAAA,GAAoB,iBAAA;AAAA,IAClB,MAAA,CAAO,oBAAA;;;;;;;ADpCX;;iBEGgB,eAAA,CAAgB,MAAA,EAAQ,MAAA,GAAS,MAAA"}

package/dist/index.js

@@ -1,5 +1,5 @@
 import { Plugin, PluginKey } from "prosemirror-state";
-import { DecorationSet } from "prosemirror-view";
+import { Decoration, DecorationSet } from "prosemirror-view";
 
 //#region src/cache.ts
 /**
@@ -182,5 +182,41 @@ function collectCodeBlocks(doc, nodeTypes) {
 }
 
 //#endregion
-export { DecorationCache, createHighlightPlugin };
+//#region src/line-number.ts
+/**
+* Returns a new parser that adds line numbers to the parsed decorations.
+*
+* Line numbers are added as `<span class="line-number">` elements with
+* the line number as the text content.
+*/
+function withLineNumbers(parser) {
+	return function parserWithLineNumbers(options) {
+		const parsed = parser(options);
+		if (parsed && Array.isArray(parsed)) {
+			const { pos, content } = options;
+			const start = pos + 1;
+			const lineStarts = [start];
+			for (const match of content.matchAll(/(?:\r?\n)/g)) lineStarts.push(start + match.index + match[0].length);
+			const decorations = [];
+			for (const [index, lineStart] of lineStarts.entries()) decorations.push(createLineStartWidget(lineStart, index + 1));
+			return [...decorations, ...parsed];
+		}
+		return parsed;
+	};
+}
+function createLineStartWidget(from, lineNumber) {
+	return Decoration.widget(from, () => createLineStartSpan(lineNumber), {
+		key: "line-number",
+		side: -20
+	});
+}
+function createLineStartSpan(lineNumber) {
+	const span = document.createElement("span");
+	span.className = "line-number";
+	span.textContent = lineNumber.toString();
+	return span;
+}
+
+//#endregion
+export { DecorationCache, createHighlightPlugin, withLineNumbers };
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":[],"sources":["../src/cache.ts","../src/plugin.ts"],"sourcesContent":["import type { Node as ProseMirrorNode } from 'prosemirror-model'\nimport type { Transaction } from 'prosemirror-state'\nimport type { Decoration } from 'prosemirror-view'\n\n/**\n * Represents a cache of doc positions to the node and decorations at that position\n */\nexport class DecorationCache {\n  private cache: Map<number, [node: ProseMirrorNode, decorations: Decoration[]]>\n\n  constructor(\n    cache?: Map<number, [node: ProseMirrorNode, decorations: Decoration[]]>,\n  ) {\n    this.cache = new Map(cache)\n  }\n\n  /**\n   * Gets the cache entry at the given doc position, or null if it doesn't exist\n   * @param pos The doc position of the node you want the cache for\n   */\n  get(pos: number) {\n    return this.cache.get(pos)\n  }\n\n  /**\n   * Sets the cache entry at the given position with the give node/decoration\n   * values\n   * @param pos The doc position of the node to set the cache for\n   * @param node The node to place in cache\n   * @param decorations The decorations to place in cache\n   */\n  set(pos: number, node: ProseMirrorNode, decorations: Decoration[]): void {\n    if (pos < 0) {\n      return\n    }\n\n    this.cache.set(pos, [node, decorations])\n  }\n\n  /**\n   * Removes the value at the oldPos (if it exists) and sets the new position to\n   * the given values\n   * @param oldPos The old node position to overwrite\n   * @param newPos The new node position to set the cache for\n   * @param node The new node to place in cache\n   * @param decorations The new decorations to place in cache\n   */\n  private replace(\n    oldPos: number,\n    newPos: number,\n    node: ProseMirrorNode,\n    decorations: Decoration[],\n  ): void {\n    this.remove(oldPos)\n    this.set(newPos, node, decorations)\n  }\n\n  /**\n   * Removes the cache entry at the given position\n   * @param pos The doc position to remove from cache\n   */\n  remove(pos: number): void {\n    this.cache.delete(pos)\n  }\n\n  /**\n   * Invalidates the cache by removing all decoration entries on nodes that have\n   * changed, updating the positions of the nodes that haven't and removing all\n   * the entries that have been deleted; NOTE: this does not affect the current\n   * cache, but returns an entirely new one\n   * @param tr A transaction to map the current cache to\n   */\n  invalidate(tr: Transaction): DecorationCache {\n    const returnCache = new DecorationCache(this.cache)\n    const mapping = tr.mapping\n\n    this.cache.forEach(([node, decorations], pos) => {\n      if (pos < 0) {\n        return\n      }\n\n      const result = mapping.mapResult(pos)\n      const mappedNode = tr.doc.nodeAt(result.pos)\n\n      if (result.deleted || !mappedNode?.eq(node)) {\n        returnCache.remove(pos)\n      } else if (pos !== result.pos) {\n        // update the decorations' from/to values to match the new node position\n        const updatedDecorations = decorations\n          .map((d): Decoration | null => {\n            // @ts-expect-error: internal api\n            // eslint-disable-next-line @typescript-eslint/no-unsafe-call\n            return d.map(mapping, 0, 0) as Decoration | null\n          })\n          .filter((d): d is Decoration => d != null)\n        returnCache.replace(pos, result.pos, mappedNode, updatedDecorations)\n      }\n    })\n\n    return returnCache\n  }\n}\n","import type { Node as ProseMirrorNode } from 'prosemirror-model'\nimport { Plugin, PluginKey } from 'prosemirror-state'\nimport { type Decoration, DecorationSet } from 'prosemirror-view'\n\nimport { DecorationCache } from './cache'\nimport type { LanguageExtractor, Parser } from './types'\n\n/**\n * Describes the current state of the highlightPlugin\n */\nexport interface HighlightPluginState {\n  cache: DecorationCache\n  decorations: DecorationSet | undefined\n  promises: Promise<void>[]\n}\n\n/**\n * Creates a plugin that highlights the contents of all nodes (via Decorations)\n * with a type passed in blockTypes\n */\nexport function createHighlightPlugin({\n  parser,\n  nodeTypes = ['code_block', 'codeBlock'],\n  languageExtractor = (node) => node.attrs.language as string | undefined,\n}: {\n  /**\n   * A function that returns an array of decorations for the given node text\n   * content, language, and position.\n   */\n  parser: Parser\n\n  /**\n   * An array containing all the node type name to target for highlighting.\n   *\n   * @default ['code_block', 'codeBlock']\n   */\n  nodeTypes?: string[]\n\n  /**\n   * A function that returns the language string to use when highlighting that\n   * node. By default, it returns `node.attrs.language`.\n   */\n  languageExtractor?: LanguageExtractor\n}): Plugin<HighlightPluginState> {\n  const key = new PluginKey<HighlightPluginState>('prosemirror-highlight')\n\n  return new Plugin<HighlightPluginState>({\n    key,\n    state: {\n      init(_, instance) {\n        const cache = new DecorationCache()\n        const [decorations, promises] = calculateDecoration(\n          instance.doc,\n          parser,\n          nodeTypes,\n          languageExtractor,\n          cache,\n        )\n\n        return { cache, decorations, promises }\n      },\n      apply: (tr, data) => {\n        const cache = data.cache.invalidate(tr)\n        const refresh = !!tr.getMeta('prosemirror-highlight-refresh')\n\n        if (!tr.docChanged && !refresh) {\n          const decorations = data.decorations?.map(tr.mapping, tr.doc)\n          const promises = data.promises\n          return { cache, decorations, promises }\n        }\n\n        const [decorations, promises] = calculateDecoration(\n          tr.doc,\n          parser,\n          nodeTypes,\n          languageExtractor,\n          cache,\n        )\n        return { cache, decorations, promises }\n      },\n    },\n    view: (view) => {\n      const promises = new Set<Promise<void>>()\n\n      // Refresh the decorations when all promises resolve\n      const refresh = () => {\n        if (promises.size > 0) {\n          return\n        }\n        const tr = view.state.tr.setMeta('prosemirror-highlight-refresh', true)\n        view.dispatch(tr)\n      }\n\n      const check = () => {\n        const state = key.getState(view.state)\n\n        for (const promise of state?.promises ?? []) {\n          promises.add(promise)\n          promise\n            .then(() => {\n              promises.delete(promise)\n              refresh()\n            })\n            .catch((error) => {\n              console.error(\n                '[prosemirror-highlight] Error resolving parser:',\n                error,\n              )\n              promises.delete(promise)\n            })\n        }\n      }\n\n      check()\n\n      return {\n        update: () => {\n          check()\n        },\n      }\n    },\n    props: {\n      decorations(this, state) {\n        return this.getState(state)?.decorations\n      },\n    },\n  })\n}\n\nfunction calculateDecoration(\n  doc: ProseMirrorNode,\n  parser: Parser,\n  nodeTypes: string[],\n  languageExtractor: LanguageExtractor,\n  cache: DecorationCache,\n): [DecorationSet | undefined, Promise<void>[]] {\n  const allDecorations: Decoration[][] = []\n  const promises: Promise<void>[] = []\n  const nodes = collectCodeBlocks(doc, nodeTypes)\n\n  try {\n    for (const [node, pos] of nodes) {\n      const language = languageExtractor(node)\n      const cached = cache.get(pos)\n\n      if (cached) {\n        const [_, decorations] = cached\n        if (decorations.length > 0) {\n          allDecorations.push(decorations)\n        }\n      } else {\n        const parsed = parser({\n          content: node.textContent,\n          language: language || undefined,\n          pos,\n          size: node.nodeSize,\n        })\n        if (parsed && Array.isArray(parsed)) {\n          cache.set(pos, node, parsed)\n          if (parsed.length > 0) {\n            allDecorations.push(parsed)\n          }\n        } else if (parsed instanceof Promise) {\n          cache.remove(pos)\n          promises.push(parsed)\n        } else {\n          console.error(\n            `[prosemirror-highlight] Invalid parser result:`,\n            parsed,\n          )\n        }\n      }\n    }\n  } catch (error) {\n    console.error(`[prosemirror-highlight] Error parsing code blocks:`, error)\n  }\n\n  const decorationSet =\n    allDecorations.length > 0\n      ? DecorationSet.create(doc, allDecorations.flat())\n      : undefined\n  return [decorationSet, promises]\n}\n\nfunction collectCodeBlocks(\n  doc: ProseMirrorNode,\n  nodeTypes: string[],\n): Array<[node: ProseMirrorNode, pos: number]> {\n  const nodes: Array<[node: ProseMirrorNode, pos: number]> = []\n  doc.descendants((node, pos) => {\n    if (node.type.isTextblock && nodeTypes.includes(node.type.name)) {\n      nodes.push([node, pos])\n      return false\n    }\n  })\n  return nodes\n}\n"],"mappings":";;;;;;;AAOA,IAAa,kBAAb,MAAa,gBAAgB;CAG3B,YACE,OACA;AACA,OAAK,QAAQ,IAAI,IAAI,MAAM;;;;;;CAO7B,IAAI,KAAa;AACf,SAAO,KAAK,MAAM,IAAI,IAAI;;;;;;;;;CAU5B,IAAI,KAAa,MAAuB,aAAiC;AACvE,MAAI,MAAM,EACR;AAGF,OAAK,MAAM,IAAI,KAAK,CAAC,MAAM,YAAY,CAAC;;;;;;;;;;CAW1C,AAAQ,QACN,QACA,QACA,MACA,aACM;AACN,OAAK,OAAO,OAAO;AACnB,OAAK,IAAI,QAAQ,MAAM,YAAY;;;;;;CAOrC,OAAO,KAAmB;AACxB,OAAK,MAAM,OAAO,IAAI;;;;;;;;;CAUxB,WAAW,IAAkC;EAC3C,MAAM,cAAc,IAAI,gBAAgB,KAAK,MAAM;EACnD,MAAM,UAAU,GAAG;AAEnB,OAAK,MAAM,SAAS,CAAC,MAAM,cAAc,QAAQ;AAC/C,OAAI,MAAM,EACR;GAGF,MAAM,SAAS,QAAQ,UAAU,IAAI;GACrC,MAAM,aAAa,GAAG,IAAI,OAAO,OAAO,IAAI;AAE5C,OAAI,OAAO,WAAW,CAAC,YAAY,GAAG,KAAK,CACzC,aAAY,OAAO,IAAI;YACd,QAAQ,OAAO,KAAK;IAE7B,MAAM,qBAAqB,YACxB,KAAK,MAAyB;AAG7B,YAAO,EAAE,IAAI,SAAS,GAAG,EAAE;MAC3B,CACD,QAAQ,MAAuB,KAAK,KAAK;AAC5C,gBAAY,QAAQ,KAAK,OAAO,KAAK,YAAY,mBAAmB;;IAEtE;AAEF,SAAO;;;;;;;;;;AC/EX,SAAgB,sBAAsB,EACpC,QACA,YAAY,CAAC,cAAc,YAAY,EACvC,qBAAqB,SAAS,KAAK,MAAM,YAoBV;CAC/B,MAAM,MAAM,IAAI,UAAgC,wBAAwB;AAExE,QAAO,IAAI,OAA6B;EACtC;EACA,OAAO;GACL,KAAK,GAAG,UAAU;IAChB,MAAM,QAAQ,IAAI,iBAAiB;IACnC,MAAM,CAAC,aAAa,YAAY,oBAC9B,SAAS,KACT,QACA,WACA,mBACA,MACD;AAED,WAAO;KAAE;KAAO;KAAa;KAAU;;GAEzC,QAAQ,IAAI,SAAS;IACnB,MAAM,QAAQ,KAAK,MAAM,WAAW,GAAG;IACvC,MAAM,UAAU,CAAC,CAAC,GAAG,QAAQ,gCAAgC;AAE7D,QAAI,CAAC,GAAG,cAAc,CAAC,QAGrB,QAAO;KAAE;KAAO,aAFI,KAAK,aAAa,IAAI,GAAG,SAAS,GAAG,IAAI;KAEhC,UADZ,KAAK;KACiB;IAGzC,MAAM,CAAC,aAAa,YAAY,oBAC9B,GAAG,KACH,QACA,WACA,mBACA,MACD;AACD,WAAO;KAAE;KAAO;KAAa;KAAU;;GAE1C;EACD,OAAO,SAAS;GACd,MAAM,2BAAW,IAAI,KAAoB;GAGzC,MAAM,gBAAgB;AACpB,QAAI,SAAS,OAAO,EAClB;IAEF,MAAM,KAAK,KAAK,MAAM,GAAG,QAAQ,iCAAiC,KAAK;AACvE,SAAK,SAAS,GAAG;;GAGnB,MAAM,cAAc;IAClB,MAAM,QAAQ,IAAI,SAAS,KAAK,MAAM;AAEtC,SAAK,MAAM,WAAW,OAAO,YAAY,EAAE,EAAE;AAC3C,cAAS,IAAI,QAAQ;AACrB,aACG,WAAW;AACV,eAAS,OAAO,QAAQ;AACxB,eAAS;OACT,CACD,OAAO,UAAU;AAChB,cAAQ,MACN,mDACA,MACD;AACD,eAAS,OAAO,QAAQ;OACxB;;;AAIR,UAAO;AAEP,UAAO,EACL,cAAc;AACZ,WAAO;MAEV;;EAEH,OAAO,EACL,YAAkB,OAAO;AACvB,UAAO,KAAK,SAAS,MAAM,EAAE;KAEhC;EACF,CAAC;;AAGJ,SAAS,oBACP,KACA,QACA,WACA,mBACA,OAC8C;CAC9C,MAAM,iBAAiC,EAAE;CACzC,MAAM,WAA4B,EAAE;CACpC,MAAM,QAAQ,kBAAkB,KAAK,UAAU;AAE/C,KAAI;AACF,OAAK,MAAM,CAAC,MAAM,QAAQ,OAAO;GAC/B,MAAM,WAAW,kBAAkB,KAAK;GACxC,MAAM,SAAS,MAAM,IAAI,IAAI;AAE7B,OAAI,QAAQ;IACV,MAAM,CAAC,GAAG,eAAe;AACzB,QAAI,YAAY,SAAS,EACvB,gBAAe,KAAK,YAAY;UAE7B;IACL,MAAM,SAAS,OAAO;KACpB,SAAS,KAAK;KACd,UAAU,YAAY;KACtB;KACA,MAAM,KAAK;KACZ,CAAC;AACF,QAAI,UAAU,MAAM,QAAQ,OAAO,EAAE;AACnC,WAAM,IAAI,KAAK,MAAM,OAAO;AAC5B,SAAI,OAAO,SAAS,EAClB,gBAAe,KAAK,OAAO;eAEpB,kBAAkB,SAAS;AACpC,WAAM,OAAO,IAAI;AACjB,cAAS,KAAK,OAAO;UAErB,SAAQ,MACN,kDACA,OACD;;;UAIA,OAAO;AACd,UAAQ,MAAM,sDAAsD,MAAM;;AAO5E,QAAO,CAHL,eAAe,SAAS,IACpB,cAAc,OAAO,KAAK,eAAe,MAAM,CAAC,GAChD,QACiB,SAAS;;AAGlC,SAAS,kBACP,KACA,WAC6C;CAC7C,MAAM,QAAqD,EAAE;AAC7D,KAAI,aAAa,MAAM,QAAQ;AAC7B,MAAI,KAAK,KAAK,eAAe,UAAU,SAAS,KAAK,KAAK,KAAK,EAAE;AAC/D,SAAM,KAAK,CAAC,MAAM,IAAI,CAAC;AACvB,UAAO;;GAET;AACF,QAAO"}
+{"version":3,"file":"index.js","names":[],"sources":["../src/cache.ts","../src/plugin.ts","../src/line-number.ts"],"sourcesContent":["import type { Node as ProseMirrorNode } from 'prosemirror-model'\nimport type { Transaction } from 'prosemirror-state'\nimport type { Decoration } from 'prosemirror-view'\n\n/**\n * Represents a cache of doc positions to the node and decorations at that position\n */\nexport class DecorationCache {\n  private cache: Map<number, [node: ProseMirrorNode, decorations: Decoration[]]>\n\n  constructor(\n    cache?: Map<number, [node: ProseMirrorNode, decorations: Decoration[]]>,\n  ) {\n    this.cache = new Map(cache)\n  }\n\n  /**\n   * Gets the cache entry at the given doc position, or null if it doesn't exist\n   * @param pos The doc position of the node you want the cache for\n   */\n  get(pos: number) {\n    return this.cache.get(pos)\n  }\n\n  /**\n   * Sets the cache entry at the given position with the give node/decoration\n   * values\n   * @param pos The doc position of the node to set the cache for\n   * @param node The node to place in cache\n   * @param decorations The decorations to place in cache\n   */\n  set(pos: number, node: ProseMirrorNode, decorations: Decoration[]): void {\n    if (pos < 0) {\n      return\n    }\n\n    this.cache.set(pos, [node, decorations])\n  }\n\n  /**\n   * Removes the value at the oldPos (if it exists) and sets the new position to\n   * the given values\n   * @param oldPos The old node position to overwrite\n   * @param newPos The new node position to set the cache for\n   * @param node The new node to place in cache\n   * @param decorations The new decorations to place in cache\n   */\n  private replace(\n    oldPos: number,\n    newPos: number,\n    node: ProseMirrorNode,\n    decorations: Decoration[],\n  ): void {\n    this.remove(oldPos)\n    this.set(newPos, node, decorations)\n  }\n\n  /**\n   * Removes the cache entry at the given position\n   * @param pos The doc position to remove from cache\n   */\n  remove(pos: number): void {\n    this.cache.delete(pos)\n  }\n\n  /**\n   * Invalidates the cache by removing all decoration entries on nodes that have\n   * changed, updating the positions of the nodes that haven't and removing all\n   * the entries that have been deleted; NOTE: this does not affect the current\n   * cache, but returns an entirely new one\n   * @param tr A transaction to map the current cache to\n   */\n  invalidate(tr: Transaction): DecorationCache {\n    const returnCache = new DecorationCache(this.cache)\n    const mapping = tr.mapping\n\n    this.cache.forEach(([node, decorations], pos) => {\n      if (pos < 0) {\n        return\n      }\n\n      const result = mapping.mapResult(pos)\n      const mappedNode = tr.doc.nodeAt(result.pos)\n\n      if (result.deleted || !mappedNode?.eq(node)) {\n        returnCache.remove(pos)\n      } else if (pos !== result.pos) {\n        // update the decorations' from/to values to match the new node position\n        const updatedDecorations = decorations\n          .map((d): Decoration | null => {\n            // @ts-expect-error: internal api\n            // eslint-disable-next-line @typescript-eslint/no-unsafe-call\n            return d.map(mapping, 0, 0) as Decoration | null\n          })\n          .filter((d): d is Decoration => d != null)\n        returnCache.replace(pos, result.pos, mappedNode, updatedDecorations)\n      }\n    })\n\n    return returnCache\n  }\n}\n","import type { Node as ProseMirrorNode } from 'prosemirror-model'\nimport { Plugin, PluginKey } from 'prosemirror-state'\nimport { type Decoration, DecorationSet } from 'prosemirror-view'\n\nimport { DecorationCache } from './cache'\nimport type { LanguageExtractor, Parser } from './types'\n\n/**\n * Describes the current state of the highlightPlugin\n */\nexport interface HighlightPluginState {\n  cache: DecorationCache\n  decorations: DecorationSet | undefined\n  promises: Promise<void>[]\n}\n\n/**\n * Creates a plugin that highlights the contents of all nodes (via Decorations)\n * with a type passed in blockTypes\n */\nexport function createHighlightPlugin({\n  parser,\n  nodeTypes = ['code_block', 'codeBlock'],\n  languageExtractor = (node) => node.attrs.language as string | undefined,\n}: {\n  /**\n   * A function that returns an array of decorations for the given node text\n   * content, language, and position.\n   */\n  parser: Parser\n\n  /**\n   * An array containing all the node type name to target for highlighting.\n   *\n   * @default ['code_block', 'codeBlock']\n   */\n  nodeTypes?: string[]\n\n  /**\n   * A function that returns the language string to use when highlighting that\n   * node. By default, it returns `node.attrs.language`.\n   */\n  languageExtractor?: LanguageExtractor\n}): Plugin<HighlightPluginState> {\n  const key = new PluginKey<HighlightPluginState>('prosemirror-highlight')\n\n  return new Plugin<HighlightPluginState>({\n    key,\n    state: {\n      init(_, instance) {\n        const cache = new DecorationCache()\n        const [decorations, promises] = calculateDecoration(\n          instance.doc,\n          parser,\n          nodeTypes,\n          languageExtractor,\n          cache,\n        )\n\n        return { cache, decorations, promises }\n      },\n      apply: (tr, data) => {\n        const cache = data.cache.invalidate(tr)\n        const refresh = !!tr.getMeta('prosemirror-highlight-refresh')\n\n        if (!tr.docChanged && !refresh) {\n          const decorations = data.decorations?.map(tr.mapping, tr.doc)\n          const promises = data.promises\n          return { cache, decorations, promises }\n        }\n\n        const [decorations, promises] = calculateDecoration(\n          tr.doc,\n          parser,\n          nodeTypes,\n          languageExtractor,\n          cache,\n        )\n        return { cache, decorations, promises }\n      },\n    },\n    view: (view) => {\n      const promises = new Set<Promise<void>>()\n\n      // Refresh the decorations when all promises resolve\n      const refresh = () => {\n        if (promises.size > 0) {\n          return\n        }\n        const tr = view.state.tr.setMeta('prosemirror-highlight-refresh', true)\n        view.dispatch(tr)\n      }\n\n      const check = () => {\n        const state = key.getState(view.state)\n\n        for (const promise of state?.promises ?? []) {\n          promises.add(promise)\n          promise\n            .then(() => {\n              promises.delete(promise)\n              refresh()\n            })\n            .catch((error) => {\n              console.error(\n                '[prosemirror-highlight] Error resolving parser:',\n                error,\n              )\n              promises.delete(promise)\n            })\n        }\n      }\n\n      check()\n\n      return {\n        update: () => {\n          check()\n        },\n      }\n    },\n    props: {\n      decorations(this, state) {\n        return this.getState(state)?.decorations\n      },\n    },\n  })\n}\n\nfunction calculateDecoration(\n  doc: ProseMirrorNode,\n  parser: Parser,\n  nodeTypes: string[],\n  languageExtractor: LanguageExtractor,\n  cache: DecorationCache,\n): [DecorationSet | undefined, Promise<void>[]] {\n  const allDecorations: Decoration[][] = []\n  const promises: Promise<void>[] = []\n  const nodes = collectCodeBlocks(doc, nodeTypes)\n\n  try {\n    for (const [node, pos] of nodes) {\n      const language = languageExtractor(node)\n      const cached = cache.get(pos)\n\n      if (cached) {\n        const [_, decorations] = cached\n        if (decorations.length > 0) {\n          allDecorations.push(decorations)\n        }\n      } else {\n        const parsed = parser({\n          content: node.textContent,\n          language: language || undefined,\n          pos,\n          size: node.nodeSize,\n        })\n        if (parsed && Array.isArray(parsed)) {\n          cache.set(pos, node, parsed)\n          if (parsed.length > 0) {\n            allDecorations.push(parsed)\n          }\n        } else if (parsed instanceof Promise) {\n          cache.remove(pos)\n          promises.push(parsed)\n        } else {\n          console.error(\n            `[prosemirror-highlight] Invalid parser result:`,\n            parsed,\n          )\n        }\n      }\n    }\n  } catch (error) {\n    console.error(`[prosemirror-highlight] Error parsing code blocks:`, error)\n  }\n\n  const decorationSet =\n    allDecorations.length > 0\n      ? DecorationSet.create(doc, allDecorations.flat())\n      : undefined\n  return [decorationSet, promises]\n}\n\nfunction collectCodeBlocks(\n  doc: ProseMirrorNode,\n  nodeTypes: string[],\n): Array<[node: ProseMirrorNode, pos: number]> {\n  const nodes: Array<[node: ProseMirrorNode, pos: number]> = []\n  doc.descendants((node, pos) => {\n    if (node.type.isTextblock && nodeTypes.includes(node.type.name)) {\n      nodes.push([node, pos])\n      return false\n    }\n  })\n  return nodes\n}\n","import { Decoration } from 'prosemirror-view'\n\nimport type { Parser, ParserOptions } from './types'\n\n/**\n * Returns a new parser that adds line numbers to the parsed decorations.\n *\n * Line numbers are added as `<span class=\"line-number\">` elements with\n * the line number as the text content.\n */\nexport function withLineNumbers(parser: Parser): Parser {\n  return function parserWithLineNumbers(options: ParserOptions) {\n    const parsed = parser(options)\n    if (parsed && Array.isArray(parsed)) {\n      const { pos, content } = options\n      const start = pos + 1\n      const lineStarts = [start]\n      for (const match of content.matchAll(/(?:\\r?\\n)/g)) {\n        lineStarts.push(start + match.index + match[0].length)\n      }\n      const decorations: Decoration[] = []\n      for (const [index, lineStart] of lineStarts.entries()) {\n        decorations.push(createLineStartWidget(lineStart, index + 1))\n      }\n      return [...decorations, ...parsed]\n    }\n    return parsed\n  }\n}\n\nexport function createLineStartWidget(\n  from: number,\n  lineNumber: number,\n): Decoration {\n  return Decoration.widget(from, () => createLineStartSpan(lineNumber), {\n    key: 'line-number',\n    side: -20,\n  })\n}\n\nfunction createLineStartSpan(lineNumber: number) {\n  const span = document.createElement('span')\n  span.className = 'line-number'\n  span.textContent = lineNumber.toString()\n  return span\n}\n"],"mappings":";;;;;;;AAOA,IAAa,kBAAb,MAAa,gBAAgB;CAG3B,YACE,OACA;AACA,OAAK,QAAQ,IAAI,IAAI,MAAM;;;;;;CAO7B,IAAI,KAAa;AACf,SAAO,KAAK,MAAM,IAAI,IAAI;;;;;;;;;CAU5B,IAAI,KAAa,MAAuB,aAAiC;AACvE,MAAI,MAAM,EACR;AAGF,OAAK,MAAM,IAAI,KAAK,CAAC,MAAM,YAAY,CAAC;;;;;;;;;;CAW1C,AAAQ,QACN,QACA,QACA,MACA,aACM;AACN,OAAK,OAAO,OAAO;AACnB,OAAK,IAAI,QAAQ,MAAM,YAAY;;;;;;CAOrC,OAAO,KAAmB;AACxB,OAAK,MAAM,OAAO,IAAI;;;;;;;;;CAUxB,WAAW,IAAkC;EAC3C,MAAM,cAAc,IAAI,gBAAgB,KAAK,MAAM;EACnD,MAAM,UAAU,GAAG;AAEnB,OAAK,MAAM,SAAS,CAAC,MAAM,cAAc,QAAQ;AAC/C,OAAI,MAAM,EACR;GAGF,MAAM,SAAS,QAAQ,UAAU,IAAI;GACrC,MAAM,aAAa,GAAG,IAAI,OAAO,OAAO,IAAI;AAE5C,OAAI,OAAO,WAAW,CAAC,YAAY,GAAG,KAAK,CACzC,aAAY,OAAO,IAAI;YACd,QAAQ,OAAO,KAAK;IAE7B,MAAM,qBAAqB,YACxB,KAAK,MAAyB;AAG7B,YAAO,EAAE,IAAI,SAAS,GAAG,EAAE;MAC3B,CACD,QAAQ,MAAuB,KAAK,KAAK;AAC5C,gBAAY,QAAQ,KAAK,OAAO,KAAK,YAAY,mBAAmB;;IAEtE;AAEF,SAAO;;;;;;;;;;AC/EX,SAAgB,sBAAsB,EACpC,QACA,YAAY,CAAC,cAAc,YAAY,EACvC,qBAAqB,SAAS,KAAK,MAAM,YAoBV;CAC/B,MAAM,MAAM,IAAI,UAAgC,wBAAwB;AAExE,QAAO,IAAI,OAA6B;EACtC;EACA,OAAO;GACL,KAAK,GAAG,UAAU;IAChB,MAAM,QAAQ,IAAI,iBAAiB;IACnC,MAAM,CAAC,aAAa,YAAY,oBAC9B,SAAS,KACT,QACA,WACA,mBACA,MACD;AAED,WAAO;KAAE;KAAO;KAAa;KAAU;;GAEzC,QAAQ,IAAI,SAAS;IACnB,MAAM,QAAQ,KAAK,MAAM,WAAW,GAAG;IACvC,MAAM,UAAU,CAAC,CAAC,GAAG,QAAQ,gCAAgC;AAE7D,QAAI,CAAC,GAAG,cAAc,CAAC,QAGrB,QAAO;KAAE;KAAO,aAFI,KAAK,aAAa,IAAI,GAAG,SAAS,GAAG,IAAI;KAEhC,UADZ,KAAK;KACiB;IAGzC,MAAM,CAAC,aAAa,YAAY,oBAC9B,GAAG,KACH,QACA,WACA,mBACA,MACD;AACD,WAAO;KAAE;KAAO;KAAa;KAAU;;GAE1C;EACD,OAAO,SAAS;GACd,MAAM,2BAAW,IAAI,KAAoB;GAGzC,MAAM,gBAAgB;AACpB,QAAI,SAAS,OAAO,EAClB;IAEF,MAAM,KAAK,KAAK,MAAM,GAAG,QAAQ,iCAAiC,KAAK;AACvE,SAAK,SAAS,GAAG;;GAGnB,MAAM,cAAc;IAClB,MAAM,QAAQ,IAAI,SAAS,KAAK,MAAM;AAEtC,SAAK,MAAM,WAAW,OAAO,YAAY,EAAE,EAAE;AAC3C,cAAS,IAAI,QAAQ;AACrB,aACG,WAAW;AACV,eAAS,OAAO,QAAQ;AACxB,eAAS;OACT,CACD,OAAO,UAAU;AAChB,cAAQ,MACN,mDACA,MACD;AACD,eAAS,OAAO,QAAQ;OACxB;;;AAIR,UAAO;AAEP,UAAO,EACL,cAAc;AACZ,WAAO;MAEV;;EAEH,OAAO,EACL,YAAkB,OAAO;AACvB,UAAO,KAAK,SAAS,MAAM,EAAE;KAEhC;EACF,CAAC;;AAGJ,SAAS,oBACP,KACA,QACA,WACA,mBACA,OAC8C;CAC9C,MAAM,iBAAiC,EAAE;CACzC,MAAM,WAA4B,EAAE;CACpC,MAAM,QAAQ,kBAAkB,KAAK,UAAU;AAE/C,KAAI;AACF,OAAK,MAAM,CAAC,MAAM,QAAQ,OAAO;GAC/B,MAAM,WAAW,kBAAkB,KAAK;GACxC,MAAM,SAAS,MAAM,IAAI,IAAI;AAE7B,OAAI,QAAQ;IACV,MAAM,CAAC,GAAG,eAAe;AACzB,QAAI,YAAY,SAAS,EACvB,gBAAe,KAAK,YAAY;UAE7B;IACL,MAAM,SAAS,OAAO;KACpB,SAAS,KAAK;KACd,UAAU,YAAY;KACtB;KACA,MAAM,KAAK;KACZ,CAAC;AACF,QAAI,UAAU,MAAM,QAAQ,OAAO,EAAE;AACnC,WAAM,IAAI,KAAK,MAAM,OAAO;AAC5B,SAAI,OAAO,SAAS,EAClB,gBAAe,KAAK,OAAO;eAEpB,kBAAkB,SAAS;AACpC,WAAM,OAAO,IAAI;AACjB,cAAS,KAAK,OAAO;UAErB,SAAQ,MACN,kDACA,OACD;;;UAIA,OAAO;AACd,UAAQ,MAAM,sDAAsD,MAAM;;AAO5E,QAAO,CAHL,eAAe,SAAS,IACpB,cAAc,OAAO,KAAK,eAAe,MAAM,CAAC,GAChD,QACiB,SAAS;;AAGlC,SAAS,kBACP,KACA,WAC6C;CAC7C,MAAM,QAAqD,EAAE;AAC7D,KAAI,aAAa,MAAM,QAAQ;AAC7B,MAAI,KAAK,KAAK,eAAe,UAAU,SAAS,KAAK,KAAK,KAAK,EAAE;AAC/D,SAAM,KAAK,CAAC,MAAM,IAAI,CAAC;AACvB,UAAO;;GAET;AACF,QAAO;;;;;;;;;;;ACzLT,SAAgB,gBAAgB,QAAwB;AACtD,QAAO,SAAS,sBAAsB,SAAwB;EAC5D,MAAM,SAAS,OAAO,QAAQ;AAC9B,MAAI,UAAU,MAAM,QAAQ,OAAO,EAAE;GACnC,MAAM,EAAE,KAAK,YAAY;GACzB,MAAM,QAAQ,MAAM;GACpB,MAAM,aAAa,CAAC,MAAM;AAC1B,QAAK,MAAM,SAAS,QAAQ,SAAS,aAAa,CAChD,YAAW,KAAK,QAAQ,MAAM,QAAQ,MAAM,GAAG,OAAO;GAExD,MAAM,cAA4B,EAAE;AACpC,QAAK,MAAM,CAAC,OAAO,cAAc,WAAW,SAAS,CACnD,aAAY,KAAK,sBAAsB,WAAW,QAAQ,EAAE,CAAC;AAE/D,UAAO,CAAC,GAAG,aAAa,GAAG,OAAO;;AAEpC,SAAO;;;AAIX,SAAgB,sBACd,MACA,YACY;AACZ,QAAO,WAAW,OAAO,YAAY,oBAAoB,WAAW,EAAE;EACpE,KAAK;EACL,MAAM;EACP,CAAC;;AAGJ,SAAS,oBAAoB,YAAoB;CAC/C,MAAM,OAAO,SAAS,cAAc,OAAO;AAC3C,MAAK,YAAY;AACjB,MAAK,cAAc,WAAW,UAAU;AACxC,QAAO"}

package/dist/lezer.d.ts

@@ -0,0 +1,15 @@
+import { n as Parser, r as ParserOptions } from "./types-jXJYKXGQ.js";
+import { Highlighter } from "@lezer/highlight";
+import { Tree } from "@lezer/common";
+
+//#region src/lezer.d.ts
+declare function createParser({
+  parse,
+  highlighter
+}: {
+  parse: (options: ParserOptions) => Tree | undefined;
+  highlighter: Highlighter | readonly Highlighter[];
+}): Parser;
+//#endregion
+export { type Parser, type ParserOptions, createParser };
+//# sourceMappingURL=lezer.d.ts.map

package/dist/lezer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"lezer.d.ts","names":[],"sources":["../src/lezer.ts"],"mappings":";;;;;iBASgB,YAAA,CAAA;EACd,KAAA;EACA;AAAA;EAEA,KAAA,GAAQ,OAAA,EAAS,aAAA,KAAkB,IAAA;EACnC,WAAA,EAAa,WAAA,YAAuB,WAAA;AAAA,IAClC,MAAA"}

package/dist/lezer.js

@@ -0,0 +1,20 @@
+import { Decoration } from "prosemirror-view";
+import { highlightTree } from "@lezer/highlight";
+
+//#region src/lezer.ts
+function createParser({ parse, highlighter }) {
+	return function lezerParser(options) {
+		const tree = parse(options);
+		if (!tree) return [];
+		const decorations = [];
+		const offset = options.pos + 1;
+		highlightTree(tree, highlighter, (from, to, classes) => {
+			if (classes && from < to) decorations.push(Decoration.inline(offset + from, offset + to, { class: classes }));
+		});
+		return decorations;
+	};
+}
+
+//#endregion
+export { createParser };
+//# sourceMappingURL=lezer.js.map

package/dist/lezer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lezer.js","names":[],"sources":["../src/lezer.ts"],"sourcesContent":["import type { Tree } from '@lezer/common'\nimport type { Highlighter } from '@lezer/highlight'\nimport { highlightTree } from '@lezer/highlight'\nimport { Decoration } from 'prosemirror-view'\n\nimport type { Parser, ParserOptions } from './types'\n\nexport type { Parser, ParserOptions }\n\nexport function createParser({\n  parse,\n  highlighter,\n}: {\n  parse: (options: ParserOptions) => Tree | undefined\n  highlighter: Highlighter | readonly Highlighter[]\n}): Parser {\n  return function lezerParser(options) {\n    const tree = parse(options)\n\n    if (!tree) {\n      return []\n    }\n\n    const decorations: Decoration[] = []\n    const offset = options.pos + 1\n    highlightTree(tree, highlighter, (from, to, classes) => {\n      if (classes && from < to) {\n        decorations.push(\n          Decoration.inline(offset + from, offset + to, { class: classes }),\n        )\n      }\n    })\n\n    return decorations\n  }\n}\n"],"mappings":";;;;AASA,SAAgB,aAAa,EAC3B,OACA,eAIS;AACT,QAAO,SAAS,YAAY,SAAS;EACnC,MAAM,OAAO,MAAM,QAAQ;AAE3B,MAAI,CAAC,KACH,QAAO,EAAE;EAGX,MAAM,cAA4B,EAAE;EACpC,MAAM,SAAS,QAAQ,MAAM;AAC7B,gBAAc,MAAM,cAAc,MAAM,IAAI,YAAY;AACtD,OAAI,WAAW,OAAO,GACpB,aAAY,KACV,WAAW,OAAO,SAAS,MAAM,SAAS,IAAI,EAAE,OAAO,SAAS,CAAC,CAClE;IAEH;AAEF,SAAO"}

package/dist/lowlight.d.ts

@@ -1,4 +1,4 @@
-import { n as Parser } from "./types-DCbyVqHc.js";
+import { n as Parser } from "./types-jXJYKXGQ.js";
 import { Root } from "hast";
 
 //#region src/lowlight.d.ts

package/dist/refractor.d.ts

@@ -1,4 +1,4 @@
-import { n as Parser } from "./types-DCbyVqHc.js";
+import { n as Parser } from "./types-jXJYKXGQ.js";
 import { Refractor } from "refractor/core";
 
 //#region src/refractor.d.ts

package/dist/shiki.d.ts

@@ -1,4 +1,4 @@
-import { n as Parser } from "./types-DCbyVqHc.js";
+import { n as Parser } from "./types-jXJYKXGQ.js";
 import { CodeToTokensOptions, HighlighterGeneric } from "@shikijs/types";
 
 //#region src/shiki.d.ts

package/dist/shiki.js

@@ -30,7 +30,7 @@ function createParser(highlighter, options) {
 	};
 }
 /**
-* Copied from https://github.com/shikijs/shiki/blob/f76a371dbc2752cba341023df00ebfe9b66cb3f6/packages/core/src/utils.ts#L213
+* Copied from https://github.com/shikijs/shiki/blob/v3.22.0/packages/core/src/utils/tokens.ts#L151
 *
 * Copy instead of import it from `shiki` to avoid importing the `shiki` package in this file.
 */

package/dist/shiki.js.map

@@ -1 +1 @@
-{"version":3,"file":"shiki.js","names":[],"sources":["../src/shiki.ts"],"sourcesContent":["import type { CodeToTokensOptions, HighlighterGeneric } from '@shikijs/types'\nimport { Decoration } from 'prosemirror-view'\n\nimport type { Parser } from './types'\n\nexport type { Parser }\n\nexport function createParser<\n  Language extends string = string,\n  Theme extends string = string,\n>(\n  highlighter: HighlighterGeneric<Language, Theme>,\n  options?: CodeToTokensOptions<Language, Theme>,\n): Parser {\n  return function parser({ content, language, pos, size }) {\n    const decorations: Decoration[] = []\n\n    const { tokens, fg, bg, rootStyle } = highlighter.codeToTokens(content, {\n      lang: language as Language | undefined,\n\n      // Use provided options for themes or just use first loaded theme\n      ...(options ?? {\n        theme: highlighter.getLoadedThemes()[0],\n      }),\n    })\n\n    const style =\n      rootStyle ||\n      (fg && bg\n        ? `--prosemirror-highlight:${fg};--prosemirror-highlight-bg:${bg}`\n        : '')\n\n    if (style) {\n      const decoration = Decoration.node(pos, pos + size, { style })\n      decorations.push(decoration)\n    }\n\n    let from = pos + 1\n\n    for (const line of tokens) {\n      for (const token of line) {\n        const to = from + token.content.length\n\n        const decoration = Decoration.inline(from, to, {\n          // When using `options.themes` the `htmlStyle` field will be set, otherwise `color` will be set\n          style: stringifyTokenStyle(\n            token.htmlStyle ?? `color: ${token.color}`,\n          ),\n          class: 'shiki',\n        })\n\n        decorations.push(decoration)\n\n        from = to\n      }\n\n      from += 1\n    }\n\n    return decorations\n  }\n}\n\n/**\n * Copied from https://github.com/shikijs/shiki/blob/f76a371dbc2752cba341023df00ebfe9b66cb3f6/packages/core/src/utils.ts#L213\n *\n * Copy instead of import it from `shiki` to avoid importing the `shiki` package in this file.\n */\nfunction stringifyTokenStyle(token: string | Record<string, string>): string {\n  if (typeof token === 'string') return token\n  return Object.entries(token)\n    .map(([key, value]) => `${key}:${value}`)\n    .join(';')\n}\n"],"mappings":";;;AAOA,SAAgB,aAId,aACA,SACQ;AACR,QAAO,SAAS,OAAO,EAAE,SAAS,UAAU,KAAK,QAAQ;EACvD,MAAM,cAA4B,EAAE;EAEpC,MAAM,EAAE,QAAQ,IAAI,IAAI,cAAc,YAAY,aAAa,SAAS;GACtE,MAAM;GAGN,GAAI,WAAW,EACb,OAAO,YAAY,iBAAiB,CAAC,IACtC;GACF,CAAC;EAEF,MAAM,QACJ,cACC,MAAM,KACH,2BAA2B,GAAG,8BAA8B,OAC5D;AAEN,MAAI,OAAO;GACT,MAAM,aAAa,WAAW,KAAK,KAAK,MAAM,MAAM,EAAE,OAAO,CAAC;AAC9D,eAAY,KAAK,WAAW;;EAG9B,IAAI,OAAO,MAAM;AAEjB,OAAK,MAAM,QAAQ,QAAQ;AACzB,QAAK,MAAM,SAAS,MAAM;IACxB,MAAM,KAAK,OAAO,MAAM,QAAQ;IAEhC,MAAM,aAAa,WAAW,OAAO,MAAM,IAAI;KAE7C,OAAO,oBACL,MAAM,aAAa,UAAU,MAAM,QACpC;KACD,OAAO;KACR,CAAC;AAEF,gBAAY,KAAK,WAAW;AAE5B,WAAO;;AAGT,WAAQ;;AAGV,SAAO;;;;;;;;AASX,SAAS,oBAAoB,OAAgD;AAC3E,KAAI,OAAO,UAAU,SAAU,QAAO;AACtC,QAAO,OAAO,QAAQ,MAAM,CACzB,KAAK,CAAC,KAAK,WAAW,GAAG,IAAI,GAAG,QAAQ,CACxC,KAAK,IAAI"}
+{"version":3,"file":"shiki.js","names":[],"sources":["../src/shiki.ts"],"sourcesContent":["import type { CodeToTokensOptions, HighlighterGeneric } from '@shikijs/types'\nimport { Decoration } from 'prosemirror-view'\n\nimport type { Parser } from './types'\n\nexport type { Parser }\n\nexport function createParser<\n  Language extends string = string,\n  Theme extends string = string,\n>(\n  highlighter: HighlighterGeneric<Language, Theme>,\n  options?: CodeToTokensOptions<Language, Theme>,\n): Parser {\n  return function parser({ content, language, pos, size }) {\n    const decorations: Decoration[] = []\n\n    const { tokens, fg, bg, rootStyle } = highlighter.codeToTokens(content, {\n      lang: language as Language | undefined,\n\n      // Use provided options for themes or just use first loaded theme\n      ...(options ?? {\n        theme: highlighter.getLoadedThemes()[0],\n      }),\n    })\n\n    const style =\n      rootStyle ||\n      (fg && bg\n        ? `--prosemirror-highlight:${fg};--prosemirror-highlight-bg:${bg}`\n        : '')\n\n    if (style) {\n      const decoration = Decoration.node(pos, pos + size, { style })\n      decorations.push(decoration)\n    }\n\n    let from = pos + 1\n\n    for (const line of tokens) {\n      for (const token of line) {\n        const to = from + token.content.length\n\n        const decoration = Decoration.inline(from, to, {\n          // When using `options.themes` the `htmlStyle` field will be set, otherwise `color` will be set\n          style: stringifyTokenStyle(\n            token.htmlStyle ?? `color: ${token.color}`,\n          ),\n          class: 'shiki',\n        })\n\n        decorations.push(decoration)\n\n        from = to\n      }\n\n      from += 1\n    }\n\n    return decorations\n  }\n}\n\n/**\n * Copied from https://github.com/shikijs/shiki/blob/v3.22.0/packages/core/src/utils/tokens.ts#L151\n *\n * Copy instead of import it from `shiki` to avoid importing the `shiki` package in this file.\n */\nfunction stringifyTokenStyle(token: string | Record<string, string>): string {\n  if (typeof token === 'string') return token\n  return Object.entries(token)\n    .map(([key, value]) => `${key}:${value}`)\n    .join(';')\n}\n"],"mappings":";;;AAOA,SAAgB,aAId,aACA,SACQ;AACR,QAAO,SAAS,OAAO,EAAE,SAAS,UAAU,KAAK,QAAQ;EACvD,MAAM,cAA4B,EAAE;EAEpC,MAAM,EAAE,QAAQ,IAAI,IAAI,cAAc,YAAY,aAAa,SAAS;GACtE,MAAM;GAGN,GAAI,WAAW,EACb,OAAO,YAAY,iBAAiB,CAAC,IACtC;GACF,CAAC;EAEF,MAAM,QACJ,cACC,MAAM,KACH,2BAA2B,GAAG,8BAA8B,OAC5D;AAEN,MAAI,OAAO;GACT,MAAM,aAAa,WAAW,KAAK,KAAK,MAAM,MAAM,EAAE,OAAO,CAAC;AAC9D,eAAY,KAAK,WAAW;;EAG9B,IAAI,OAAO,MAAM;AAEjB,OAAK,MAAM,QAAQ,QAAQ;AACzB,QAAK,MAAM,SAAS,MAAM;IACxB,MAAM,KAAK,OAAO,MAAM,QAAQ;IAEhC,MAAM,aAAa,WAAW,OAAO,MAAM,IAAI;KAE7C,OAAO,oBACL,MAAM,aAAa,UAAU,MAAM,QACpC;KACD,OAAO;KACR,CAAC;AAEF,gBAAY,KAAK,WAAW;AAE5B,WAAO;;AAGT,WAAQ;;AAGV,SAAO;;;;;;;;AASX,SAAS,oBAAoB,OAAgD;AAC3E,KAAI,OAAO,UAAU,SAAU,QAAO;AACtC,QAAO,OAAO,QAAQ,MAAM,CACzB,KAAK,CAAC,KAAK,WAAW,GAAG,IAAI,GAAG,QAAQ,CACxC,KAAK,IAAI"}

package/dist/sugar-high.d.ts

@@ -1,4 +1,4 @@
-import { n as Parser } from "./types-DCbyVqHc.js";
+import { n as Parser } from "./types-jXJYKXGQ.js";
 
 //#region src/sugar-high.d.ts
 declare function createParser(): Parser;

package/dist/{types-DCbyVqHc.d.ts → types-jXJYKXGQ.d.ts}

@@ -2,13 +2,7 @@ import { Decoration } from "prosemirror-view";
 import { Node } from "prosemirror-model";
 
 //#region src/types.d.ts
-/**
- * A function that parses the text content of a code block node and returns an
- * array of ProseMirror decorations. If the underlying syntax highlighter is
- * still loading, you can return a promise that will be resolved when the
- * highlighter is ready.
- */
-type Parser = (options: {
+interface ParserOptions {
   /**
    * The text content of the code block node.
    */
@@ -25,11 +19,18 @@ type Parser = (options: {
    * The size of the code block node.
    */
   size: number;
-}) => Decoration[] | Promise<void>;
+}
+/**
+ * A function that parses the text content of a code block node and returns an
+ * array of ProseMirror decorations. If the underlying syntax highlighter is
+ * still loading, you can return a promise that will be resolved when the
+ * highlighter is ready.
+ */
+type Parser = (options: ParserOptions) => Decoration[] | Promise<void>;
 /**
  * A function that extracts the language of a code block node.
  */
 type LanguageExtractor = (node: Node) => string | undefined;
 //#endregion
-export { Parser as n, LanguageExtractor as t };
-//# sourceMappingURL=types-DCbyVqHc.d.ts.map
+export { Parser as n, ParserOptions as r, LanguageExtractor as t };
+//# sourceMappingURL=types-jXJYKXGQ.d.ts.map

package/dist/types-jXJYKXGQ.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types-jXJYKXGQ.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;;UAGiB,aAAA;;AAAjB;;EAIE,OAAA;EAJ4B;;;EAS5B,GAAA;EAUA;;;EALA,QAAA;EAcgB;;;EAThB,IAAA;AAAA;;;;;;;KASU,MAAA,IAAU,OAAA,EAAS,aAAA,KAAkB,UAAA,KAAe,OAAA;;AAKhE;;KAAY,iBAAA,IAAqB,IAAA,EAAM,IAAA"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "prosemirror-highlight",
   "type": "module",
-  "version": "0.14.0",
+  "version": "0.15.1",
   "description": "A ProseMirror plugin to highlight code blocks",
   "author": "ocavue <ocavue@gmail.com>",
   "license": "MIT",
@@ -19,7 +19,8 @@
     "shiki",
     "refractor",
     "lowlight",
-    "prism"
+    "prism",
+    "lezer"
   ],
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -30,6 +31,10 @@
       "types": "./dist/index.d.ts",
       "default": "./dist/index.js"
     },
+    "./lezer": {
+      "types": "./dist/lezer.d.ts",
+      "default": "./dist/lezer.js"
+    },
     "./lowlight": {
       "types": "./dist/lowlight.d.ts",
       "default": "./dist/lowlight.js"
@@ -51,7 +56,9 @@
     "dist"
   ],
   "peerDependencies": {
-    "@shikijs/types": "^1.29.2 || ^2.0.0 || ^3.0.0",
+    "@lezer/common": "^1.0.0",
+    "@lezer/highlight": "^1.0.0",
+    "@shikijs/types": "^1.29.2 || ^2.0.0 || ^3.0.0 || ^4.0.0",
     "@types/hast": "^3.0.0",
     "highlight.js": "^11.9.0",
     "lowlight": "^3.1.0",
@@ -60,9 +67,15 @@
     "prosemirror-transform": "^1.8.0",
     "prosemirror-view": "^1.32.4",
     "refractor": "^5.0.0",
-    "sugar-high": "^0.6.1 || ^0.7.0 || ^0.8.0 || ^0.9.0"
+    "sugar-high": "^0.6.1 || ^0.7.0 || ^0.8.0 || ^0.9.0 || ^1.0.0"
   },
   "peerDependenciesMeta": {
+    "@lezer/common": {
+      "optional": true
+    },
+    "@lezer/highlight": {
+      "optional": true
+    },
     "@shikijs/types": {
       "optional": true
     },
@@ -95,14 +108,19 @@
     }
   },
   "devDependencies": {
-    "@ocavue/eslint-config": "^3.11.2",
+    "@lezer/common": "^1.5.1",
+    "@lezer/css": "^1.3.1",
+    "@lezer/highlight": "^1.2.3",
+    "@lezer/javascript": "^1.5.4",
+    "@ocavue/eslint-config": "^4.2.0",
     "@ocavue/tsconfig": "^0.6.3",
-    "@shikijs/types": "^3.22.0",
+    "@shikijs/types": "^4.0.0",
     "@types/hast": "^3.0.4",
     "@types/node": "^24.0.0",
-    "eslint": "^9.39.2",
+    "diffable-html-snapshot": "^0.2.0",
+    "eslint": "^10.0.2",
     "highlight.js": "^11.11.1",
-    "jsdom": "^28.0.0",
+    "jsdom": "^28.1.0",
     "lowlight": "^3.3.0",
     "pkg-pr-new": "^0.0.63",
     "prettier": "^3.8.1",
@@ -113,7 +131,7 @@
     "prosemirror-transform": "^1.11.0",
     "prosemirror-view": "^1.41.6",
     "refractor": "^5.0.0",
-    "shiki": "^3.22.0",
+    "shiki": "^4.0.0",
     "sugar-high": "^0.9.5",
     "tsdown": "^0.20.3",
     "typescript": "^5.9.3",

package/dist/types-DCbyVqHc.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"types-DCbyVqHc.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;;;;AASA;;;;KAAY,MAAA,IAAU,OAAA;EASpB;;;EALA,OAAA;EAgBI;;;EAXJ,GAAA;EAgBU;;;EAXV,QAAA;EAWoD;;;EANpD,IAAA;AAAA,MACI,UAAA,KAAe,OAAA;;;;KAKT,iBAAA,IAAqB,IAAA,EAAM,IAAA"}