import parser, { Node } from './parser'; import * as A from '../prelude/array'; import * as S from '../prelude/string'; export default (source: string): Node[] => { if (source == null || source == '') { return null; } let nodes: Node[] = parser.root.tryParse(source); const combineText = (es: Node[]): Node => ({ name: 'text', props: { text: S.concat(es.map(e => e.props.text)) } }); const concatText = (nodes: Node[]): Node[] => A.concat(A.groupOn(x => x.name, nodes).map(es => es[0].name === 'text' ? [combineText(es)] : es )); const concatTextRecursive = (es: Node[]): void => es.filter(x => x.children).forEach(x => { x.children = concatText(x.children); concatTextRecursive(x.children); }); nodes = concatText(nodes); concatTextRecursive(nodes); const removeEmptyTextNodes = (nodes: Node[]) => { nodes.forEach(n => { if (n.children) { n.children = removeEmptyTextNodes(n.children); } }); return nodes.filter(n => !(n.name == 'text' && n.props.text == '')); }; nodes = removeEmptyTextNodes(nodes); return nodes; };