wikiparser-node 1.3.4-b → 1.3.6

package/dist/lib/element.js

@@ -0,0 +1,625 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AstElement = void 0;
+const fs = require("fs");
+const path = require("path");
+const string_1 = require("../util/string");
+const constants_1 = require("../util/constants");
+const debug_1 = require("../util/debug");
+const selector_1 = require("../parser/selector");
+const ranges_1 = require("./ranges");
+const title_1 = require("./title");
+const Parser = require("../index");
+const node_1 = require("./node");
+/**
+ * optionally convert to lower cases
+ * @param val 属性值
+ * @param i 是否对大小写不敏感
+ */
+const toCase = (val, i) => i ? val.toLowerCase() : val;
+/**
+ * 检查某个下标是否符合表达式
+ * @param str 表达式
+ * @param i 待检查的下标
+ */
+const nth = (str, i) => new ranges_1.Ranges(str.split(',')).applyTo(i + 1).includes(i);
+/**
+ * 检测:lang()伪选择器
+ * @param node 节点
+ * @param node.attributes 节点属性
+ * @param regex 语言正则
+ */
+const matchesLang = ({ attributes }, regex) => {
+    const lang = attributes?.['lang'];
+    return typeof lang === 'string' && regex.test(lang);
+};
+/* NOT FOR BROWSER END */
+/** 类似HTMLElement */
+class AstElement extends node_1.AstNode {
+    /** 子节点总数 */
+    get length() {
+        return this.childNodes.length;
+    }
+    /* NOT FOR BROWSER */
+    /** 全部非文本子节点 */
+    get children() {
+        return this.childNodes.filter((child) => child.type !== 'text');
+    }
+    /** 首位非文本子节点 */
+    get firstElementChild() {
+        return this.childNodes.find((child) => child.type !== 'text');
+    }
+    /** 末位非文本子节点 */
+    get lastElementChild() {
+        return this.children.at(-1);
+    }
+    /** 非文本子节点总数 */
+    get childElementCount() {
+        return this.children.length;
+    }
+    /** 父节点 */
+    get parentElement() {
+        return this.parentNode;
+    }
+    /** AstElement.prototype.text()的getter写法 */
+    get outerText() {
+        return this.text();
+    }
+    /** 不可见 */
+    get hidden() {
+        return this.text() === '';
+    }
+    /** 后一个可见的兄弟节点 */
+    get nextVisibleSibling() {
+        let { nextSibling } = this;
+        while (nextSibling?.text() === '') {
+            ({ nextSibling } = nextSibling);
+        }
+        return nextSibling;
+    }
+    /** 前一个可见的兄弟节点 */
+    get previousVisibleSibling() {
+        let { previousSibling } = this;
+        while (previousSibling?.text() === '') {
+            ({ previousSibling } = previousSibling);
+        }
+        return previousSibling;
+    }
+    /** 内部高度 */
+    get clientHeight() {
+        const { innerText } = this;
+        return typeof innerText === 'string' ? innerText.split('\n').length : undefined;
+    }
+    /** 内部宽度 */
+    get clientWidth() {
+        const { innerText } = this;
+        return typeof innerText === 'string' ? innerText.split('\n').at(-1).length : undefined;
+    }
+    constructor() {
+        super();
+        this.seal('name');
+    }
+    /* NOT FOR BROWSER END */
+    /**
+     * 可见部分
+     * @param separator 子节点间的连接符
+     */
+    text(separator) {
+        return (0, string_1.text)(this.childNodes, separator);
+    }
+    /** 合并相邻的文本子节点 */
+    normalize() {
+        const childNodes = [...this.childNodes];
+        for (let i = childNodes.length - 1; i >= 0; i--) {
+            const { type, data } = childNodes[i], prev = childNodes[i - 1];
+            if (type !== 'text' || this.getGaps(i - 1)) {
+                //
+            }
+            else if (data === '') {
+                childNodes.splice(i, 1);
+            }
+            else if (prev?.type === 'text') {
+                prev.setAttribute('data', prev.data + data);
+                childNodes.splice(i, 1);
+            }
+        }
+        this.setAttribute('childNodes', childNodes);
+    }
+    /**
+     * 移除子节点
+     * @param i 移除位置
+     */
+    removeAt(i) {
+        this.verifyChild(i);
+        return (0, debug_1.setChildNodes)(this, i, 1)[0];
+    }
+    /**
+     * 插入子节点
+     * @param node 待插入的子节点
+     * @param i 插入位置
+     * @throws `RangeError` 不能插入祖先或子节点
+     */
+    insertAt(node, i = this.length) {
+        if (node.contains(this)) {
+            throw new RangeError('不能插入祖先节点！');
+        }
+        if (this.childNodes.includes(node)) {
+            throw new RangeError('不能插入子节点！');
+        }
+        this.verifyChild(i, 1);
+        node.parentNode?.removeChild(node);
+        (0, debug_1.setChildNodes)(this, i, 0, [node]);
+        return node;
+    }
+    /**
+     * 最近的祖先节点
+     * @param selector 选择器
+     */
+    closest(selector) {
+        let { parentNode } = this, condition;
+        if (/[^a-z\-,\s]/u.test(selector)) {
+            const stack = (0, selector_1.parseSelector)(selector);
+            condition = /** @implements */ (token) => token.#matchesStack(stack);
+        }
+        else {
+            const types = new Set(selector.split(',').map(str => str.trim()));
+            condition = /** @implements */ (token) => types.has(token.type);
+        }
+        while (parentNode) {
+            if (condition(parentNode)) {
+                return parentNode;
+            }
+            ({ parentNode } = parentNode);
+        }
+        return undefined;
+    }
+    /**
+     * 在末尾批量插入子节点
+     * @param elements 插入节点
+     */
+    append(...elements) {
+        for (const element of elements) {
+            this.insertAt(element);
+        }
+    }
+    /**
+     * 批量替换子节点
+     * @param elements 新的子节点
+     */
+    replaceChildren(...elements) {
+        for (let i = this.length - 1; i >= 0; i--) {
+            this.removeAt(i);
+        }
+        this.append(...elements);
+    }
+    /**
+     * 修改文本子节点
+     * @param str 新文本
+     * @param i 子节点位置
+     * @throws `RangeError` 对应位置的子节点不是文本节点
+     */
+    setText(str, i = 0) {
+        this.verifyChild(i);
+        const oldText = this.childNodes.at(i), { type, constructor: { name } } = oldText;
+        if (type === 'text') {
+            const { data } = oldText;
+            oldText.replaceData(str);
+            return data;
+        }
+        throw new RangeError(`第 ${i} 个子节点是 ${name}！`);
+    }
+    /** @private */
+    toString(omit, separator = '') {
+        return omit && this.matchesTypes(omit)
+            ? ''
+            : this.childNodes.map(child => child.toString(omit)).join(separator);
+    }
+    /**
+     * @override
+     * @param start
+     */
+    lint(start = this.getAbsoluteIndex()) {
+        const errors = [];
+        for (let i = 0, cur = start + this.getAttribute('padding'); i < this.length; i++) {
+            const child = this.childNodes[i];
+            errors.push(...child.lint(cur));
+            cur += String(child).length + this.getGaps(i);
+        }
+        return errors;
+    }
+    /**
+     * @override
+     * @param opt 选项
+     */
+    print(opt = {}) {
+        return String(this) ? `<span class="wpb-${opt.class ?? this.type}">${(0, string_1.print)(this.childNodes, opt)}</span>` : '';
+    }
+    /* NOT FOR BROWSER */
+    /** @private */
+    matchesTypes(types) {
+        return types.has(this.type);
+    }
+    /**
+     * 保存为JSON
+     * @param file 文件名
+     */
+    json(file) {
+        const json = {
+            ...this,
+            childNodes: this.childNodes.map(child => child.type === 'text' ? child.data : child.json()),
+        };
+        if (typeof file === 'string') {
+            fs.writeFileSync(path.join(__dirname.slice(0, -4), '..', 'printed', `${file}${file.endsWith('.json') ? '' : '.json'}`), JSON.stringify(json, null, 2));
+        }
+        return json;
+    }
+    /** 销毁 */
+    destroy() {
+        this.parentNode?.destroy();
+        for (const child of this.childNodes) {
+            child.setAttribute('parentNode', undefined);
+        }
+        Object.setPrototypeOf(this, null);
+    }
+    /** 是否受保护。保护条件来自Token，这里仅提前用于:required和:optional伪选择器。 */
+    #isProtected() {
+        const { parentNode } = this;
+        if (!parentNode) {
+            return undefined;
+        }
+        const { childNodes, fixed } = parentNode, protectedIndices = parentNode.getAttribute('protectedChildren').applyTo(childNodes);
+        return fixed || protectedIndices.includes(childNodes.indexOf(this));
+    }
+    /**
+     * 检查是否符合属性选择器
+     * @param key 属性键
+     * @param equal 比较符
+     * @param val 属性值
+     * @param i 是否对大小写不敏感
+     * @throws `RangeError` 复杂属性不能用于选择器
+     */
+    #matchesAttr(key, equal, val = '', i) {
+        const isAttr = typeof this.hasAttr === 'function' && typeof this.getAttr === 'function';
+        if (!(key in this) && (!isAttr || !this.hasAttr(key))) {
+            return equal === '!=';
+        }
+        else if (!equal) {
+            return true;
+        }
+        const v = toCase(val, i);
+        let thisVal = this.getAttribute(key);
+        if (isAttr) {
+            const attr = this.getAttr(key);
+            if (attr !== undefined) {
+                thisVal = attr === true ? '' : attr;
+            }
+        }
+        if (thisVal instanceof RegExp) {
+            thisVal = thisVal.source;
+        }
+        if (equal === '~=') {
+            const thisVals = typeof thisVal === 'string' ? thisVal.split(/\s/u) : thisVal;
+            return Boolean(thisVals?.[Symbol.iterator])
+                && [...thisVals].some(w => typeof w === 'string' && toCase(w, i) === v);
+        }
+        else if (typeof thisVal !== 'string') {
+            throw new RangeError(`复杂属性 ${key} 不能用于选择器！`);
+        }
+        const stringVal = toCase(thisVal, i);
+        switch (equal) {
+            case '|=':
+                return stringVal === v || stringVal.startsWith(`${v}-`);
+            case '^=':
+                return stringVal.startsWith(v);
+            case '$=':
+                return stringVal.endsWith(v);
+            case '*=':
+                return stringVal.includes(v);
+            case '!=':
+                return stringVal !== v;
+            default: // `=`
+                return stringVal === v;
+        }
+    }
+    /**
+     * 检查是否符合解析后的选择器，不含节点关系
+     * @param step 解析后的选择器
+     * @throws `SyntaxError` 错误的正则伪选择器
+     * @throws `SyntaxError` 未定义的伪选择器
+     */
+    #matches(step) {
+        const { parentNode, type, name, childNodes, link } = this, children = parentNode?.children, childrenOfType = children?.filter(({ type: t }) => t === type), siblingsCount = children?.length ?? 1, siblingsCountOfType = childrenOfType?.length ?? 1, index = (children?.indexOf(this) ?? 0) + 1, indexOfType = (childrenOfType?.indexOf(this) ?? 0) + 1, lastIndex = siblingsCount - index + 1, lastIndexOfType = siblingsCountOfType - indexOfType + 1;
+        return step.every(selector => {
+            if (typeof selector === 'string') {
+                switch (selector) { // 情形1：简单伪选择器、type和name
+                    case '*':
+                        return true;
+                    case ':root':
+                        return !parentNode;
+                    case ':first-child':
+                        return index === 1;
+                    case ':first-of-type':
+                        return indexOfType === 1;
+                    case ':last-child':
+                        return lastIndex === 1;
+                    case ':last-of-type':
+                        return lastIndexOfType === 1;
+                    case ':only-child':
+                        return siblingsCount === 1;
+                    case ':only-of-type':
+                        return siblingsCountOfType === 1;
+                    case ':empty':
+                        return !childNodes.some(({ type: t, data }) => t !== 'text' || data);
+                    case ':parent':
+                        return childNodes.some(({ type: t, data }) => t !== 'text' || data);
+                    case ':header':
+                        return type === 'heading';
+                    case ':hidden':
+                        return this.text() === '';
+                    case ':visible':
+                        return this.text() !== '';
+                    case ':only-whitespace':
+                        return this.text().trim() === '';
+                    case ':any-link':
+                        return type === 'link'
+                            || type === 'free-ext-link'
+                            || type === 'ext-link'
+                            || (type === 'file' || type === 'gallery-image' && link);
+                    case ':local-link':
+                        return (type === 'link' || type === 'file' || type === 'gallery-image')
+                            && link instanceof title_1.Title
+                            && link.title === '';
+                    case ':invalid':
+                        return type === 'table-inter' || type === 'image-parameter' && name === 'invalid';
+                    case ':required':
+                        return this.#isProtected() === true;
+                    case ':optional':
+                        return this.#isProtected() === false;
+                    default: {
+                        const [t, n] = selector.split('#');
+                        return (!t || t === type || Boolean(constants_1.typeAliases[type]?.includes(t))) && (!n || n === name);
+                    }
+                }
+            }
+            else if (selector.length === 4) { // 情形2：属性选择器
+                return this.#matchesAttr(...selector);
+            }
+            const [s, pseudo] = selector; // 情形3：复杂伪选择器
+            switch (pseudo) {
+                case 'is':
+                    return this.matches(s);
+                case 'not':
+                    return !this.matches(s);
+                case 'nth-child':
+                    return nth(s, index);
+                case 'nth-of-type':
+                    return nth(s, indexOfType);
+                case 'nth-last-child':
+                    return nth(s, lastIndex);
+                case 'nth-last-of-type':
+                    return nth(s, lastIndexOfType);
+                case 'contains':
+                    return this.text().includes(s);
+                case 'has':
+                    return Boolean(this.querySelector(s));
+                case 'lang': {
+                    const regex = new RegExp(`^${s}(?:-|$)`, 'u');
+                    return matchesLang(this, regex)
+                        || this.getAncestors().some(ancestor => matchesLang(ancestor, regex));
+                }
+                case 'regex': {
+                    const mt = /^([^,]+),\s*\/(.+)\/([a-z]*)$/u.exec(s);
+                    if (!mt) {
+                        throw new SyntaxError('错误的伪选择器用法。请使用形如 ":regex(\'attr, /re/i\')" 的格式。');
+                    }
+                    try {
+                        return new RegExp(mt[2], mt[3]).test(String(this.getAttribute(mt[1].trim())));
+                    }
+                    catch {
+                        throw new SyntaxError(`错误的正则表达式：/${mt[2]}/${mt[3]}`);
+                    }
+                }
+                default:
+                    throw new SyntaxError(`未定义的伪选择器：${pseudo}`);
+            }
+        });
+    }
+    /**
+     * 检查是否符合解析后的选择器
+     * @param copy 解析后的选择器
+     */
+    #matchesArray(copy) {
+        const condition = [...copy], step = condition.pop();
+        if (this.#matches(step)) {
+            const { parentNode, previousElementSibling } = this;
+            switch (condition.at(-1)?.relation) {
+                case undefined:
+                    return true;
+                case '>':
+                    return Boolean(parentNode && parentNode.#matchesArray(condition));
+                case '+':
+                    return Boolean(previousElementSibling && previousElementSibling.#matchesArray(condition));
+                case '~': {
+                    if (!parentNode) {
+                        return false;
+                    }
+                    const { children } = parentNode;
+                    return children.slice(0, children.indexOf(this))
+                        .some(child => child.#matchesArray(condition));
+                }
+                default: // ' '
+                    return this.getAncestors().some(ancestor => ancestor.#matchesArray(condition));
+            }
+        }
+        return false;
+    }
+    /**
+     * 检查是否符合解析后的组合选择器
+     * @param stack 解析后的一组选择器
+     */
+    #matchesStack(stack) {
+        return stack.some(condition => this.#matchesArray(condition));
+    }
+    /**
+     * 检查是否符合选择器
+     * @param selector 选择器
+     */
+    matches(selector) {
+        return selector === undefined || this.#matchesStack((0, selector_1.parseSelector)(selector));
+    }
+    /**
+     * 符合条件的第一个后代节点
+     * @param condition 条件
+     */
+    #getElementBy(condition) {
+        for (const child of this.children) {
+            if (condition(child)) {
+                return child;
+            }
+            const descendant = child.#getElementBy(condition);
+            if (descendant) {
+                return descendant;
+            }
+        }
+        return undefined;
+    }
+    /**
+     * 符合选择器的第一个后代节点
+     * @param selector 选择器
+     */
+    querySelector(selector) {
+        const stack = (0, selector_1.parseSelector)(selector);
+        return this.#getElementBy(token => token.#matchesStack(stack));
+    }
+    /**
+     * 类型选择器
+     * @param types
+     */
+    getElementByTypes(types) {
+        const typeSet = new Set(types.split(',').map(str => str.trim()));
+        return this.#getElementBy(({ type }) => typeSet.has(type));
+    }
+    /**
+     * id选择器
+     * @param id id名
+     */
+    getElementById(id) {
+        return this.#getElementBy(token => 'id' in token && token.id === id);
+    }
+    /**
+     * 符合条件的所有后代节点
+     * @param condition 条件
+     */
+    #getElementsBy(condition) {
+        const descendants = [];
+        for (const child of this.children) {
+            if (condition(child)) {
+                descendants.push(child);
+            }
+            descendants.push(...child.#getElementsBy(condition));
+        }
+        return descendants;
+    }
+    /**
+     * 符合选择器的所有后代节点
+     * @param selector 选择器
+     */
+    querySelectorAll(selector) {
+        const stack = (0, selector_1.parseSelector)(selector);
+        return this.#getElementsBy(token => token.#matchesStack(stack));
+    }
+    /**
+     * 类选择器
+     * @param className 类名之一
+     */
+    getElementsByClassName(className) {
+        return this.#getElementsBy(token => 'classList' in token && token.classList.has(className));
+    }
+    /**
+     * 标签名选择器
+     * @param tag 标签名
+     */
+    getElementsByTagName(tag) {
+        return this.#getElementsBy(({ type, name }) => name === tag && (type === 'html' || type === 'ext'));
+    }
+    /**
+     * 获取某一行的wikitext
+     * @param n 行号
+     */
+    getLine(n) {
+        return String(this).split('\n', n + 1)[n];
+    }
+    /**
+     * 在开头批量插入子节点
+     * @param elements 插入节点
+     */
+    prepend(...elements) {
+        for (let i = 0; i < elements.length; i++) {
+            this.insertAt(elements[i], i);
+        }
+    }
+    /**
+     * 获取子节点的位置
+     * @param node 子节点
+     * @throws `RangeError` 找不到子节点
+     */
+    #getChildIndex(node) {
+        const i = this.childNodes.indexOf(node);
+        if (i === -1) {
+            throw new RangeError('找不到子节点！');
+        }
+        return i;
+    }
+    /**
+     * 移除子节点
+     * @param node 子节点
+     */
+    removeChild(node) {
+        this.removeAt(this.#getChildIndex(node));
+        return node;
+    }
+    insertBefore(child, reference) {
+        return reference === undefined
+            ? this.insertAt(child)
+            : this.insertAt(child, this.#getChildIndex(reference));
+    }
+    /**
+     * 输出AST
+     * @param depth 当前深度
+     */
+    echo(depth = 0) {
+        const indent = '  '.repeat(depth), str = String(this), { childNodes, type, length } = this;
+        if (childNodes.every(child => child.type === 'text' || !String(child))) {
+            console.log(`${indent}\x1B[32m<%s>\x1B[0m${(0, string_1.noWrap)(str)}\x1B[32m</%s>\x1B[0m`, type, type);
+            return;
+        }
+        Parser.info(`${indent}<${type}>`);
+        let i = this.getAttribute('padding');
+        if (i) {
+            console.log(`${indent}  ${(0, string_1.noWrap)(str.slice(0, i))}`);
+        }
+        for (let j = 0; j < length; j++) {
+            const child = childNodes[j], childStr = String(child), gap = j === length - 1 ? 0 : this.getGaps(j);
+            if (!childStr) {
+                //
+            }
+            else if (child.type === 'text') {
+                console.log(`${indent}  ${(0, string_1.noWrap)(child.data)}`);
+            }
+            else {
+                child.echo(depth + 1);
+            }
+            i += childStr.length;
+            if (gap) {
+                console.log(`${indent}  ${(0, string_1.noWrap)(str.slice(i, i + gap))}`);
+                i += gap;
+            }
+        }
+        if (i < str.length) {
+            console.log(`${indent}  ${(0, string_1.noWrap)(str.slice(i))}`);
+        }
+        Parser.info(`${indent}</${type}>`);
+    }
+}
+exports.AstElement = AstElement;
+constants_1.classes['AstElement'] = __filename;