feat: support unwasm condition for core (#97)

Author: antfu

packages/shikiji-core/src/index.ts

@@ -4,7 +4,7 @@ export * from './utils'
 export * from './types'
 
 export { loadWasm } from './oniguruma'
-export { getShikiInternal } from './internal'
+export { getShikiInternal, setDefaultWasmLoader } from './internal'
 export { codeToThemedTokens } from './tokenizer'
 export { tokenizeAnsiWithTheme } from './tokenizer-ansi'
 export { codeToHast } from './renderer-hast'

packages/shikiji-core/src/internal.ts

@@ -1,9 +1,19 @@
 import type { HighlighterCoreOptions, LanguageInput, MaybeGetter, ShikiInternal, ThemeInput, ThemeRegistrationResolved } from './types'
+import type { LoadWasmOptions } from './oniguruma'
 import { createOnigScanner, createOnigString, loadWasm } from './oniguruma'
 import { Registry } from './registry'
 import { Resolver } from './resolver'
 import { normalizeTheme } from './normalize'
 
+let _defaultWasmLoader: LoadWasmOptions | undefined
+/**
+ * Set the default wasm loader for `loadWasm`.
+ * @internal
+ */
+export function setDefaultWasmLoader(_loader: LoadWasmOptions) {
+  _defaultWasmLoader = _loader
+}
+
 /**
  * Get the minimal shiki context for rendering.
  */
@@ -18,13 +28,15 @@ export async function getShikiInternal(options: HighlighterCoreOptions = {}): Pr
     )).flat()))
   }
 
+  const wasmLoader = options.loadWasm || _defaultWasmLoader
+
   const [
     themes,
     langs,
   ] = await Promise.all([
     Promise.all((options.themes || []).map(normalizeGetter)).then(r => r.map(normalizeTheme)),
     resolveLangs(options.langs || []),
-    options.loadWasm ? loadWasm(options.loadWasm) : undefined,
+    wasmLoader ? loadWasm(wasmLoader) : undefined,
   ] as const)
 
   const resolver = new Resolver(

packages/shikiji/package.json

@@ -23,6 +23,7 @@
     },
     "./core": {
       "types": "./dist/core.d.mts",
+      "unwasm": "./dist/core-unwasm.mjs",
       "default": "./dist/core.mjs"
     },
     "./wasm": {

packages/shikiji/rollup.config.mjs

@@ -13,6 +13,7 @@ import fg from 'fast-glob'
 const entries = [
   'src/index.ts',
   'src/core.ts',
+  'src/core-unwasm.ts',
   'src/types.ts',
   'src/themes.ts',
   'src/langs.ts',

packages/shikiji/src/core-unwasm.ts

@@ -0,0 +1,10 @@
+/**
+ * In environments where WebAssembly can be treated as native ESM and https://github.com/unjs/unwasm,
+ * We add the wasm file as the dependency so users don't need to call `loadWasm` manually.
+ */
+
+import { setDefaultWasmLoader } from 'shikiji-core'
+
+setDefaultWasmLoader(() => import('shikiji/wasm'))
+
+export * from 'shikiji-core'