tiptap/packages/core/src/utils/getSchema.ts

89 lines
2.7 KiB
TypeScript
Raw Normal View History

2020-10-22 03:13:38 +08:00
import { NodeSpec, MarkSpec, Schema } from 'prosemirror-model'
2020-09-04 17:59:09 +08:00
import { Extensions } from '../types'
2020-10-22 03:01:39 +08:00
import splitExtensions from './splitExtensions'
import getAttributesFromExtensions from './getAttributesFromExtensions'
import getRenderedAttributes from './getRenderedAttributes'
2020-10-23 15:20:26 +08:00
import isEmptyObject from './isEmptyObject'
2020-10-23 16:59:26 +08:00
function cleanUpSchemaItem<T>(data: T) {
2020-10-23 15:20:26 +08:00
return Object.fromEntries(Object.entries(data).filter(([key, value]) => {
if (key === 'attrs' && isEmptyObject(value)) {
return false
}
return value !== null && value !== undefined
2020-10-23 16:59:26 +08:00
})) as T
2020-10-23 15:20:26 +08:00
}
2020-09-04 17:59:09 +08:00
export default function getSchema(extensions: Extensions): Schema {
const allAttributes = getAttributesFromExtensions(extensions)
2020-10-22 03:13:38 +08:00
const { nodeExtensions, markExtensions } = splitExtensions(extensions)
2020-10-22 03:01:39 +08:00
const topNode = nodeExtensions.find(extension => extension.topNode)?.name
2020-10-12 16:32:54 +08:00
2020-10-21 21:17:05 +08:00
const nodes = Object.fromEntries(nodeExtensions.map(extension => {
2020-10-22 03:01:39 +08:00
const context = {
options: extension.options,
}
const attributes = allAttributes.filter(attribute => attribute.type === extension.name)
2020-10-22 03:01:39 +08:00
2020-10-23 15:20:26 +08:00
const schema: NodeSpec = cleanUpSchemaItem({
2020-10-12 16:32:54 +08:00
content: extension.content,
2020-10-22 03:01:39 +08:00
marks: extension.marks,
2020-10-12 16:32:54 +08:00
group: extension.group,
2020-10-22 03:01:39 +08:00
inline: extension.inline,
atom: extension.atom,
selectable: extension.selectable,
draggable: extension.draggable,
code: extension.code,
defining: extension.defining,
isolating: extension.isolating,
parseDOM: extension.parseHTML.bind(context)(),
toDOM: node => {
return extension.renderHTML.bind(context)({
node,
attributes: getRenderedAttributes(node, attributes),
})
},
attrs: Object.fromEntries(attributes.map(attribute => {
return [attribute.name, { default: attribute?.attribute?.default }]
})),
2020-10-23 15:20:26 +08:00
})
2020-10-12 16:32:54 +08:00
2020-10-21 21:17:05 +08:00
return [extension.name, schema]
2020-10-10 04:59:25 +08:00
}))
2020-10-22 03:13:38 +08:00
const marks = Object.fromEntries(markExtensions.map(extension => {
const context = {
options: extension.options,
}
2020-10-22 15:14:24 +08:00
const attributes = allAttributes.filter(attribute => attribute.type === extension.name)
2020-10-22 03:13:38 +08:00
2020-10-23 15:20:26 +08:00
const schema: MarkSpec = cleanUpSchemaItem({
2020-10-22 03:13:38 +08:00
inclusive: extension.inclusive,
excludes: extension.excludes,
group: extension.group,
spanning: extension.spanning,
parseDOM: extension.parseHTML.bind(context)(),
2020-10-22 15:14:24 +08:00
toDOM: mark => {
return extension.renderHTML.bind(context)({
mark,
attributes: getRenderedAttributes(mark, attributes),
})
},
attrs: Object.fromEntries(attributes.map(attribute => {
return [attribute.name, { default: attribute?.attribute?.default }]
})),
2020-10-23 15:20:26 +08:00
})
2020-10-22 03:13:38 +08:00
return [extension.name, schema]
}))
return new Schema({
2020-10-10 04:59:25 +08:00
topNode,
nodes,
2020-10-22 03:13:38 +08:00
marks,
})
}