50 lines
1.3 KiB
TypeScript
50 lines
1.3 KiB
TypeScript
import { defineConfig } from "vite";
|
|
import blog from "@bspeice/vite-plugin-blog";
|
|
import mdx from "@mdx-js/rollup";
|
|
import react from "@vitejs/plugin-react-swc";
|
|
import rehypeHighlight from "rehype-highlight";
|
|
|
|
import { Root, Element } from "hast";
|
|
import { Plugin } from "unified";
|
|
import { visit } from "unist-util-visit";
|
|
|
|
// Add the `hljs` class to `pre` so it picks up the highlight background color
|
|
const highlightPre: Plugin<[], Root> = () => {
|
|
return (tree) => {
|
|
visit(tree, (node, _, parent) => {
|
|
if (
|
|
!parent ||
|
|
(parent as Element).tagName !== "pre" ||
|
|
(node as Element).tagName !== "code"
|
|
) {
|
|
return;
|
|
}
|
|
|
|
const parentElement = parent as Element;
|
|
const parentProperties = parentElement.properties;
|
|
|
|
if (!parentProperties) {
|
|
return;
|
|
}
|
|
|
|
if (!Array.isArray(parentProperties["className"])) {
|
|
parentProperties["className"] = ["hljs"];
|
|
} else {
|
|
parentProperties["className"].unshift("hljs");
|
|
}
|
|
});
|
|
};
|
|
};
|
|
|
|
export default defineConfig({
|
|
plugins: [
|
|
blog({
|
|
"/": "/pages/index.tsx",
|
|
"/about": "/pages/about.mdx",
|
|
"/2019/02/the-whole-world": "/posts/2019/02/the-whole-world.mdx",
|
|
}),
|
|
mdx({ rehypePlugins: [rehypeHighlight, highlightPre] }),
|
|
react(),
|
|
],
|
|
});
|