1
import type { AstroIntegration } from "astro";
2
import rehypePrettyCode from "rehype-pretty-code";
3
import type { BuiltinTheme } from "shiki";
4
import { expressiveCodeConfig } from "../../config/";
5

6
type ConfigSetupHookArgs = Parameters<
7
  NonNullable<AstroIntegration["hooks"]["astro:config:setup"]>
8
>[0];
9

10
export default function astroRehypePretty() {
11
  return {
12
    name: "astro-rehype-pretty",
13
    hooks: {
14
      "astro:config:setup": async (args: unknown) => {
15
        const { config: astroConfig } = args as ConfigSetupHookArgs;
16

17
        const markdownProcessorOptions = astroConfig.markdown.processor
18
          .options as { rehypePlugins: unknown[] };
19
        markdownProcessorOptions.rehypePlugins.push(() =>
20
          rehypePrettyCode({
21
            theme: {
22
              light: expressiveCodeConfig.lightTheme as BuiltinTheme,
23
              dark: expressiveCodeConfig.darkTheme as BuiltinTheme,
24
            },
25
          }),
26
        );
27
      },
28
    },
29
  } satisfies AstroIntegration;
30
}

1
import type { Root } from "hast";
2
import { toString as hastToString } from "hast-util-to-string";
3
import { visit } from "unist-util-visit";
4

5
export default function rehypeEmCJK() {
6
  return (tree: Root) => {
7
    visit(tree, "element", (node) => {
8
      if (["em"].includes(node.tagName)) {
9
        const textContent = hastToString(node);
10

11
        if (
12
          textContent.match(
13
            /[\p{Script=Hani}\p{Script=Hira}\p{Script=Kana}\p{Script=Hang}\p{Script=Bopo}]+/gu,
14
          )
15
        ) {
16
          node.properties.className = node.properties.className ?? [];
17
          node.properties.className.push("em-cjk");
18
        }
19
      }
20
    });
21
  };
22
}