mirror of
https://github.com/withastro/astro.git
synced 2024-12-23 21:53:55 -05:00
82 lines
2.2 KiB
TypeScript
82 lines
2.2 KiB
TypeScript
|
import type { AstroInstance } from 'astro';
|
||
|
import type { RenderableTreeNode } from '@markdoc/markdoc';
|
||
|
import { createComponent, renderComponent, render } from 'astro/runtime/server/index.js';
|
||
|
import Markdoc from '@markdoc/markdoc';
|
||
|
import { MarkdocError, isCapitalized } from '../dist/utils.js';
|
||
|
|
||
|
export type TreeNode =
|
||
|
| {
|
||
|
type: 'text';
|
||
|
content: string;
|
||
|
}
|
||
|
| {
|
||
|
type: 'component';
|
||
|
component: AstroInstance['default'];
|
||
|
props: Record<string, any>;
|
||
|
children: TreeNode[];
|
||
|
}
|
||
|
| {
|
||
|
type: 'element';
|
||
|
tag: string;
|
||
|
attributes: Record<string, any>;
|
||
|
children: TreeNode[];
|
||
|
};
|
||
|
|
||
|
export const ComponentNode = createComponent({
|
||
|
factory(result: any, { treeNode }: { treeNode: TreeNode }) {
|
||
|
if (treeNode.type === 'text') return render`${treeNode.content}`;
|
||
|
const slots = {
|
||
|
default: () => render`${treeNode.children.map((child) =>
|
||
|
renderComponent(result, 'ComponentNode', ComponentNode, { treeNode: child })
|
||
|
)}`,
|
||
|
};
|
||
|
if (treeNode.type === 'component') {
|
||
|
return renderComponent(
|
||
|
result,
|
||
|
treeNode.component.name,
|
||
|
treeNode.component,
|
||
|
treeNode.props,
|
||
|
slots
|
||
|
);
|
||
|
}
|
||
|
return renderComponent(result, treeNode.tag, treeNode.tag, treeNode.attributes, slots);
|
||
|
},
|
||
|
propagation: 'none',
|
||
|
});
|
||
|
|
||
|
export function createTreeNode(
|
||
|
node: RenderableTreeNode,
|
||
|
components: Record<string, AstroInstance['default']> = {}
|
||
|
): TreeNode {
|
||
|
if (typeof node === 'string' || typeof node === 'number') {
|
||
|
return { type: 'text', content: String(node) };
|
||
|
} else if (node === null || typeof node !== 'object' || !Markdoc.Tag.isTag(node)) {
|
||
|
return { type: 'text', content: '' };
|
||
|
}
|
||
|
|
||
|
if (node.name in components) {
|
||
|
const component = components[node.name];
|
||
|
const props = node.attributes;
|
||
|
const children = node.children.map((child) => createTreeNode(child, components));
|
||
|
|
||
|
return {
|
||
|
type: 'component',
|
||
|
component,
|
||
|
props,
|
||
|
children,
|
||
|
};
|
||
|
} else if (isCapitalized(node.name)) {
|
||
|
throw new MarkdocError({
|
||
|
message: `Unable to render ${JSON.stringify(node.name)}.`,
|
||
|
hint: 'Did you add this to the "components" prop on your <Content /> component?',
|
||
|
});
|
||
|
} else {
|
||
|
return {
|
||
|
type: 'element',
|
||
|
tag: node.name,
|
||
|
attributes: node.attributes,
|
||
|
children: node.children.map((child) => createTreeNode(child, components)),
|
||
|
};
|
||
|
}
|
||
|
}
|