-
Notifications
You must be signed in to change notification settings - Fork 0
/
hastToPDF.tsx
73 lines (60 loc) · 1.98 KB
/
hastToPDF.tsx
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
/* eslint-disable no-use-before-define, @typescript-eslint/no-use-before-define */
/**
* Crude hast (HTML AST) to React-PDF format JSX transform
*/
import React from 'react'
import { uid } from 'react-uid'
import { View, Text } from '@react-pdf/renderer'
import { HastNodes } from 'mdast-util-to-hast/lib'
import { Root, Text as HastText, Element } from 'hast'
import PDFBulletList from './components/PDFBulletList'
const getBulletList = (element: Element) => {
return element.children
.filter(
(node): node is Element =>
node.type === 'element' && node.tagName === 'li'
)
.map((node) => node.children)
.map((children) => children.map((node) => hastToPDF(node)))
}
const handlers = {
root: (root: Root): React.ReactNode => {
const { children } = root
if (children.length) {
return children.map(hastToPDF)
}
// empty root has no react-PDF equiv
return undefined
},
text: (text: HastText): React.ReactNode => {
return <Text key={uid(text)}>{text.value}</Text>
},
element: (element: Element): React.ReactNode => {
const { tagName, children } = element
switch (tagName) {
case 'p':
return <View key={uid(element)}>{children.map(hastToPDF)}</View>
case 'ul':
return (
<PDFBulletList key={uid(element)} list={getBulletList(element)} />
)
default:
throw new Error(`hastToPDF: No transform for html element ${tagName}`)
}
},
}
function isHandledType(type: string): type is keyof typeof handlers {
return Object.keys(handlers).indexOf(type) !== -1
}
function hastToPDF(node: HastNodes | null | undefined): React.ReactNode {
const type = node && node.type
if (!type) {
throw new Error(`hastToPDF: Called with invalid root node`)
}
if (!isHandledType(type)) {
throw new Error(`hastToPDF: Cannot compile unknown node ${type}`)
}
// eslint-disable-next-line @typescript-eslint/no-explicit-any
return handlers[type](node as any)
}
export default hastToPDF