-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
compile.ts
101 lines (96 loc) · 2.99 KB
/
compile.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
import { createProcessor, ProcessorOptions } from '@mdx-js/mdx'
import { Processor } from '@mdx-js/mdx/lib/core'
import remarkGfm from 'remark-gfm'
import rehypePrettyCode from 'rehype-pretty-code'
import { rehypeMdxTitle } from 'rehype-mdx-title'
import readingTime from 'remark-reading-time'
import {
remarkStaticImage,
remarkHeadings,
structurize,
parseMeta,
attachMeta
} from './mdx-plugins'
import { LoaderOptions, PageOpts, ReadingTime } from './types'
import theme from './theme.json'
import { truthy } from './utils'
const createCompiler = (mdxOptions: ProcessorOptions): Processor => {
const compiler = createProcessor(mdxOptions)
compiler.data('headingMeta', {
headings: []
})
return compiler
}
const rehypePrettyCodeOptions = {
theme,
onVisitLine(node: any) {
// Prevent lines from collapsing in `display: grid` mode, and
// allow empty lines to be copy/pasted
if (node.children.length === 0) {
node.children = [{ type: 'text', value: ' ' }]
}
},
onVisitHighlightedLine(node: any) {
node.properties.className.push('highlighted')
},
onVisitHighlightedWord(node: any) {
node.properties.className = ['highlighted']
}
}
export async function compileMdx(
source: string,
loaderOptions: Pick<
LoaderOptions,
'staticImage' | 'flexsearch' | 'defaultShowCopyCode' | 'readingTime'
> & {
mdxOptions?: LoaderOptions['mdxOptions'] &
Pick<ProcessorOptions, 'jsx' | 'outputFormat'>
} = {},
filePath = ''
) {
const structurizedData = Object.create(null)
const mdxOptions = loaderOptions.mdxOptions || {}
const compiler = createCompiler({
jsx: mdxOptions.jsx || false,
outputFormat: mdxOptions.outputFormat || 'function-body',
providerImportSource: '@mdx-js/react',
remarkPlugins: [
...(mdxOptions.remarkPlugins || []),
remarkGfm,
remarkHeadings,
loaderOptions.staticImage && ([remarkStaticImage, { filePath }] as any),
loaderOptions.flexsearch &&
structurize(structurizedData, loaderOptions.flexsearch),
loaderOptions.readingTime && readingTime
].filter(truthy),
rehypePlugins: [
...(mdxOptions.rehypePlugins || []),
parseMeta,
[
rehypePrettyCode,
{ ...rehypePrettyCodeOptions, ...mdxOptions.rehypePrettyCodeOptions }
],
[rehypeMdxTitle, { name: '__nextra_title__' }],
[attachMeta, { defaultShowCopyCode: loaderOptions.defaultShowCopyCode }]
]
})
try {
const vFile = await compiler.process(source)
const result = String(vFile)
.replace('export const __nextra_title__', 'const __nextra_title__')
.replace('export default MDXContent;', '')
const readingTime = vFile.data.readingTime as ReadingTime | undefined
return {
result,
...(compiler.data('headingMeta') as Pick<
PageOpts,
'headings' | 'hasJsxInH1'
>),
...(readingTime && { readingTime }),
structurizedData
}
} catch (err) {
console.error(`[nextra] Error compiling ${filePath}.`)
throw err
}
}