@unifast/shiki 0.0.4

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 kenzwada
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/index.cjs

@@ -0,0 +1,146 @@
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+let _unifast_core = require("@unifast/core");
+let shiki = require("shiki");
+//#region src/transformer.ts
+function resolveThemeConfig(opts) {
+	const { themes } = opts;
+	if (themes && typeof themes === "object" && !Array.isArray(themes) && "light" in themes) return {
+		kind: "dual",
+		light: themes.light,
+		dark: themes.dark,
+		allThemes: [themes.light, themes.dark],
+		defaultColor: opts.defaultColor ?? false
+	};
+	if (Array.isArray(themes)) return {
+		kind: "single",
+		theme: opts.defaultTheme ?? themes[0] ?? "github-dark",
+		allThemes: themes
+	};
+	if (typeof themes === "string") return {
+		kind: "single",
+		theme: themes,
+		allThemes: [themes]
+	};
+	const fallback = opts.defaultTheme ?? "github-dark";
+	return {
+		kind: "single",
+		theme: fallback,
+		allThemes: [fallback]
+	};
+}
+async function createShikiTransformer(options = {}) {
+	const themeConfig = resolveThemeConfig(options);
+	const langs = options.langs ?? [];
+	const highlighter = await (0, shiki.createHighlighter)({
+		themes: themeConfig.allThemes,
+		langs
+	});
+	const loadedLangs = new Set(highlighter.getLoadedLanguages());
+	function highlight(code, lang) {
+		if (themeConfig.kind === "dual") return highlighter.codeToHtml(code, {
+			lang,
+			themes: {
+				light: themeConfig.light,
+				dark: themeConfig.dark
+			},
+			defaultColor: themeConfig.defaultColor
+		});
+		return highlighter.codeToHtml(code, {
+			lang,
+			theme: themeConfig.theme
+		});
+	}
+	function highlightToHast(code, lang) {
+		if (themeConfig.kind === "dual") return highlighter.codeToHast(code, {
+			lang,
+			themes: {
+				light: themeConfig.light,
+				dark: themeConfig.dark
+			},
+			defaultColor: themeConfig.defaultColor
+		});
+		return highlighter.codeToHast(code, {
+			lang,
+			theme: themeConfig.theme
+		});
+	}
+	function extractLang(element) {
+		const code = (0, _unifast_core.findCodeChild)(element);
+		if (!code) return null;
+		return (0, _unifast_core.extractLang)(code);
+	}
+	function transformNode(node) {
+		if (node.type === "element") {
+			if (node.tagName === "pre") {
+				const lang = extractLang(node);
+				if (lang && loadedLangs.has(lang)) {
+					const pre = highlightToHast((0, _unifast_core.extractText)(node), lang).children[0];
+					if (pre && typeof pre === "object" && "type" in pre) return pre;
+				}
+			}
+			return {
+				...node,
+				children: node.children.map(transformNode)
+			};
+		}
+		if (node.type === "root") return {
+			...node,
+			children: node.children.map(transformNode)
+		};
+		return node;
+	}
+	function transformMdxJs(js) {
+		const pattern = /_jsx\("pre",\s*\{\s*children:\s*_jsx\("code",\s*\{(?:\s*children:\s*("(?:[^"\\]|\\.)*"),\s*className:\s*"language-([\w+-]+)"|\s*className:\s*"language-([\w+-]+)",\s*children:\s*("(?:[^"\\]|\\.)*"))\s*\}\)\s*\}\)/g;
+		const replacements = [];
+		let m;
+		while ((m = pattern.exec(js)) !== null) {
+			const lang = m[2] ?? m[3];
+			const codeStr = JSON.parse(m[1] ?? m[4]);
+			if (!loadedLangs.has(lang)) continue;
+			try {
+				const highlighted = highlight(codeStr, lang);
+				const replacement = `_jsx("div", { dangerouslySetInnerHTML: { __html: ${JSON.stringify(highlighted)} } })`;
+				replacements.push({
+					start: m.index,
+					end: m.index + m[0].length,
+					replacement
+				});
+			} catch {}
+		}
+		let result = js;
+		for (let i = replacements.length - 1; i >= 0; i--) {
+			const r = replacements[i];
+			result = result.slice(0, r.start) + r.replacement + result.slice(r.end);
+		}
+		return result;
+	}
+	return {
+		transform(hast) {
+			return {
+				...hast,
+				children: hast.children.map(transformNode)
+			};
+		},
+		transformMdxJs
+	};
+}
+//#endregion
+//#region src/plugin.ts
+async function createShikiPlugin(options) {
+	const transformer = await createShikiTransformer(options);
+	return {
+		name: "shiki",
+		options: { highlight: { enabled: false } },
+		hastTransform: (hast) => transformer.transform(hast),
+		mdxJsTransform: (js) => transformer.transformMdxJs(js)
+	};
+}
+//#endregion
+exports.createShikiPlugin = createShikiPlugin;
+exports.createShikiTransformer = createShikiTransformer;
+Object.defineProperty(exports, "hastToHtml", {
+	enumerable: true,
+	get: function() {
+		return _unifast_core.hastToHtml;
+	}
+});

package/dist/index.d.cts

@@ -0,0 +1,23 @@
+import { HastComment, HastDoctype, HastElement, HastNode, HastRaw, HastRoot, HastRoot as HastRoot$1, HastText, UnifastPlugin, hastToHtml } from "@unifast/core";
+import { BundledLanguage, BundledTheme } from "shiki";
+
+//#region src/transformer.d.ts
+type ShikiTransformerOptions = {
+  themes?: BundledTheme | BundledTheme[] | {
+    light: BundledTheme;
+    dark: BundledTheme;
+  };
+  defaultTheme?: BundledTheme;
+  defaultColor?: string | false;
+  langs?: BundledLanguage[];
+};
+type ShikiTransformer = {
+  transform(hast: HastRoot$1): HastRoot$1;
+  transformMdxJs(js: string): string;
+};
+declare function createShikiTransformer(options?: ShikiTransformerOptions): Promise<ShikiTransformer>;
+//#endregion
+//#region src/plugin.d.ts
+declare function createShikiPlugin(options?: ShikiTransformerOptions): Promise<UnifastPlugin>;
+//#endregion
+export { type HastComment, type HastDoctype, type HastElement, type HastNode, type HastRaw, type HastRoot, type HastText, type ShikiTransformer, type ShikiTransformerOptions, createShikiPlugin, createShikiTransformer, hastToHtml };

package/dist/index.d.mts

@@ -0,0 +1,23 @@
+import { HastComment, HastDoctype, HastElement, HastNode, HastRaw, HastRoot, HastRoot as HastRoot$1, HastText, UnifastPlugin, hastToHtml } from "@unifast/core";
+import { BundledLanguage, BundledTheme } from "shiki";
+
+//#region src/transformer.d.ts
+type ShikiTransformerOptions = {
+  themes?: BundledTheme | BundledTheme[] | {
+    light: BundledTheme;
+    dark: BundledTheme;
+  };
+  defaultTheme?: BundledTheme;
+  defaultColor?: string | false;
+  langs?: BundledLanguage[];
+};
+type ShikiTransformer = {
+  transform(hast: HastRoot$1): HastRoot$1;
+  transformMdxJs(js: string): string;
+};
+declare function createShikiTransformer(options?: ShikiTransformerOptions): Promise<ShikiTransformer>;
+//#endregion
+//#region src/plugin.d.ts
+declare function createShikiPlugin(options?: ShikiTransformerOptions): Promise<UnifastPlugin>;
+//#endregion
+export { type HastComment, type HastDoctype, type HastElement, type HastNode, type HastRaw, type HastRoot, type HastText, type ShikiTransformer, type ShikiTransformerOptions, createShikiPlugin, createShikiTransformer, hastToHtml };

package/dist/index.mjs

@@ -0,0 +1,138 @@
+import { extractLang, extractText, findCodeChild, hastToHtml } from "@unifast/core";
+import { createHighlighter } from "shiki";
+//#region src/transformer.ts
+function resolveThemeConfig(opts) {
+	const { themes } = opts;
+	if (themes && typeof themes === "object" && !Array.isArray(themes) && "light" in themes) return {
+		kind: "dual",
+		light: themes.light,
+		dark: themes.dark,
+		allThemes: [themes.light, themes.dark],
+		defaultColor: opts.defaultColor ?? false
+	};
+	if (Array.isArray(themes)) return {
+		kind: "single",
+		theme: opts.defaultTheme ?? themes[0] ?? "github-dark",
+		allThemes: themes
+	};
+	if (typeof themes === "string") return {
+		kind: "single",
+		theme: themes,
+		allThemes: [themes]
+	};
+	const fallback = opts.defaultTheme ?? "github-dark";
+	return {
+		kind: "single",
+		theme: fallback,
+		allThemes: [fallback]
+	};
+}
+async function createShikiTransformer(options = {}) {
+	const themeConfig = resolveThemeConfig(options);
+	const langs = options.langs ?? [];
+	const highlighter = await createHighlighter({
+		themes: themeConfig.allThemes,
+		langs
+	});
+	const loadedLangs = new Set(highlighter.getLoadedLanguages());
+	function highlight(code, lang) {
+		if (themeConfig.kind === "dual") return highlighter.codeToHtml(code, {
+			lang,
+			themes: {
+				light: themeConfig.light,
+				dark: themeConfig.dark
+			},
+			defaultColor: themeConfig.defaultColor
+		});
+		return highlighter.codeToHtml(code, {
+			lang,
+			theme: themeConfig.theme
+		});
+	}
+	function highlightToHast(code, lang) {
+		if (themeConfig.kind === "dual") return highlighter.codeToHast(code, {
+			lang,
+			themes: {
+				light: themeConfig.light,
+				dark: themeConfig.dark
+			},
+			defaultColor: themeConfig.defaultColor
+		});
+		return highlighter.codeToHast(code, {
+			lang,
+			theme: themeConfig.theme
+		});
+	}
+	function extractLang$1(element) {
+		const code = findCodeChild(element);
+		if (!code) return null;
+		return extractLang(code);
+	}
+	function transformNode(node) {
+		if (node.type === "element") {
+			if (node.tagName === "pre") {
+				const lang = extractLang$1(node);
+				if (lang && loadedLangs.has(lang)) {
+					const pre = highlightToHast(extractText(node), lang).children[0];
+					if (pre && typeof pre === "object" && "type" in pre) return pre;
+				}
+			}
+			return {
+				...node,
+				children: node.children.map(transformNode)
+			};
+		}
+		if (node.type === "root") return {
+			...node,
+			children: node.children.map(transformNode)
+		};
+		return node;
+	}
+	function transformMdxJs(js) {
+		const pattern = /_jsx\("pre",\s*\{\s*children:\s*_jsx\("code",\s*\{(?:\s*children:\s*("(?:[^"\\]|\\.)*"),\s*className:\s*"language-([\w+-]+)"|\s*className:\s*"language-([\w+-]+)",\s*children:\s*("(?:[^"\\]|\\.)*"))\s*\}\)\s*\}\)/g;
+		const replacements = [];
+		let m;
+		while ((m = pattern.exec(js)) !== null) {
+			const lang = m[2] ?? m[3];
+			const codeStr = JSON.parse(m[1] ?? m[4]);
+			if (!loadedLangs.has(lang)) continue;
+			try {
+				const highlighted = highlight(codeStr, lang);
+				const replacement = `_jsx("div", { dangerouslySetInnerHTML: { __html: ${JSON.stringify(highlighted)} } })`;
+				replacements.push({
+					start: m.index,
+					end: m.index + m[0].length,
+					replacement
+				});
+			} catch {}
+		}
+		let result = js;
+		for (let i = replacements.length - 1; i >= 0; i--) {
+			const r = replacements[i];
+			result = result.slice(0, r.start) + r.replacement + result.slice(r.end);
+		}
+		return result;
+	}
+	return {
+		transform(hast) {
+			return {
+				...hast,
+				children: hast.children.map(transformNode)
+			};
+		},
+		transformMdxJs
+	};
+}
+//#endregion
+//#region src/plugin.ts
+async function createShikiPlugin(options) {
+	const transformer = await createShikiTransformer(options);
+	return {
+		name: "shiki",
+		options: { highlight: { enabled: false } },
+		hastTransform: (hast) => transformer.transform(hast),
+		mdxJsTransform: (js) => transformer.transformMdxJs(js)
+	};
+}
+//#endregion
+export { createShikiPlugin, createShikiTransformer, hastToHtml };

package/package.json

@@ -0,0 +1,39 @@
+{
+  "name": "@unifast/shiki",
+  "version": "0.0.4",
+  "description": "Shiki syntax highlighting plugin for unifast via HAST transforms",
+  "files": [
+    "dist"
+  ],
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.cts",
+  "exports": {
+    ".": {
+      "import": "./dist/index.mjs",
+      "require": "./dist/index.cjs"
+    },
+    "./package.json": "./package.json"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "shiki": "^3.23.0",
+    "@unifast/core": "0.0.4"
+  },
+  "devDependencies": {
+    "typescript": "^5.9.3",
+    "vitest": "^4.1.0"
+  },
+  "scripts": {
+    "build": "tsdown",
+    "lint:oxc": "oxlint -c ../../.oxlintrc.json .",
+    "lint:oxc:fix": "oxlint -c ../../.oxlintrc.json --fix .",
+    "typecheck": "tsc --noEmit",
+    "fmt:oxc": "oxfmt --write .",
+    "fmt:oxc:check": "oxfmt --check .",
+    "test": "vitest run",
+    "test:coverage": "vitest run --coverage --coverage.reporter=text --coverage.reporter=lcov"
+  }
+}