wehub-resource-sync
48 lines
1.3 KiB
TypeScript
48 lines
1.3 KiB
TypeScript
import type { HTMLNode, CommentOrTextAST, ElementAST, AST } from './types';
|
|
|
|
export const splitHead = (str: string, sep: string) => {
|
|
const idx = str.indexOf(sep);
|
|
if (idx === -1) return [str];
|
|
return [str.slice(0, idx), str.slice(idx + sep.length)];
|
|
};
|
|
|
|
const unquote = (str: string) => {
|
|
const car = str.charAt(0);
|
|
const end = str.length - 1;
|
|
const isQuoteStart = car === '"' || car === "'";
|
|
if (isQuoteStart && car === str.charAt(end)) {
|
|
return str.slice(1, end);
|
|
}
|
|
return str;
|
|
};
|
|
|
|
const formatAttributes = (attributes: string[]) => {
|
|
return attributes.map((attribute) => {
|
|
const parts = splitHead(attribute.trim(), '=');
|
|
const key = parts[0];
|
|
const value = typeof parts[1] === 'string' ? unquote(parts[1]) : null;
|
|
return { key, value };
|
|
});
|
|
};
|
|
|
|
export const format = (nodes: HTMLNode[]): AST[] => {
|
|
return nodes.map((node) => {
|
|
if (node.type === 'element') {
|
|
const children = format(node.children);
|
|
const item: ElementAST = {
|
|
type: 'element',
|
|
tagName: node.tagName.toLowerCase(),
|
|
attributes: formatAttributes(node.attributes),
|
|
children,
|
|
};
|
|
return item;
|
|
}
|
|
|
|
const item: CommentOrTextAST = {
|
|
type: node.type,
|
|
content: node.content,
|
|
};
|
|
return item;
|
|
});
|
|
};
|