chore: add back global

Author: ricardonunez-io

packages/mdx/src/plugins/rehype/rehypeSyntaxHighlighting.ts

@@ -23,6 +23,7 @@ import {
   getTwoslashOptions,
   parseLineComment,
   getCdnTwoslashTransformer,
+  cdnTwoslash,
 } from './twoslash/config.js';
 import { getLanguage } from './utils.js';
 
@@ -147,8 +148,10 @@ async function traverseNode({
 
     code = splitCode.join('\n');
 
+    await cdnTwoslash.init();
+    await cdnTwoslash.prepareTypes(code);
     const twoslashOptions = getTwoslashOptions({ linkMap });
-    const cdnTwoslashTransformer = await getCdnTwoslashTransformer(code, twoslashOptions);
+    const cdnTwoslashTransformer = getCdnTwoslashTransformer(twoslashOptions);
 
     const hast = highlighter.codeToHast(code, {
       lang: lang ?? DEFAULT_LANG,

packages/mdx/src/plugins/rehype/twoslash/config.ts

@@ -5,10 +5,12 @@ import {
 } from '@shikijs/twoslash';
 import type { ElementContent } from 'hast';
 import type { ShikiTransformer } from 'shiki/types';
-import { type TwoslashInstance } from 'twoslash';
+import { createTwoslasher, type TwoslashInstance } from 'twoslash';
 import { createTwoslashFromCDN, TwoslashCdnReturn } from 'twoslash-cdn';
 import * as ts from 'typescript';
 
+type TransformerFactory = (options?: TransformerTwoslashOptions) => ShikiTransformer;
+
 const twoslashCompilerOptions: ts.CompilerOptions = {
   target: ts.ScriptTarget.ESNext,
   lib: ['ESNext', 'DOM', 'esnext', 'dom', 'es2020'],
@@ -18,15 +20,29 @@ const twoslashCompilerOptions: ts.CompilerOptions = {
   alwaysStrict: false,
 };
 
+const fsMap: Map<string, string> = new Map();
+const twoslashStorageMap = new Map();
+
+export const cdnTwoslash = createTwoslashFromCDN({
+  compilerOptions: twoslashCompilerOptions,
+  fsMap,
+  storage: {
+    getItemRaw(key) {
+      return twoslashStorageMap.get(key);
+    },
+    setItemRaw(key, value) {
+      twoslashStorageMap.set(key, value);
+    },
+  },
+});
+
 let cachedInstance: TwoslashCdnReturn | undefined;
 
-export async function getCdnTwoslashTransformer(
-  code: string,
-  options: TransformerTwoslashOptions
-): Promise<ShikiTransformer> {
-  const fsMap: Map<string, string> = new Map();
-  const twoslashStorageMap = new Map();
+export const cdnTwoslashTransformer: TransformerFactory = createTransformerFactory(
+  cdnTwoslash.runSync
+);
 
+export function getCdnTwoslashTransformer(options: TransformerTwoslashOptions): ShikiTransformer {
   function getInstance() {
     cachedInstance ??= createTwoslashFromCDN({
       compilerOptions: twoslashCompilerOptions,
@@ -43,10 +59,6 @@ export async function getCdnTwoslashTransformer(
     return cachedInstance;
   }
 
-  const instance = getInstance();
-  await instance.init();
-  await instance.prepareTypes(code);
-
   return createTransformerFactory(
     // lazy load Twoslash instance so it works on serverless platforms
     ((...args) => getInstance().runSync(...args)) as TwoslashInstance