123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218 |
- 'use strict';
- const { visitSkip, detachNodeFromParent } = require('../lib/xast.js');
- const { collectStylesheet, computeStyle } = require('../lib/style.js');
- const {
- elems,
- attrsGroups,
- elemsGroups,
- attrsGroupsDefaults,
- presentationNonInheritableGroupAttrs,
- } = require('./_collections');
- exports.type = 'visitor';
- exports.name = 'removeUnknownsAndDefaults';
- exports.active = true;
- exports.description =
- 'removes unknown elements content and attributes, removes attrs with default values';
- // resolve all groups references
- /**
- * @type {Map<string, Set<string>>}
- */
- const allowedChildrenPerElement = new Map();
- /**
- * @type {Map<string, Set<string>>}
- */
- const allowedAttributesPerElement = new Map();
- /**
- * @type {Map<string, Map<string, string>>}
- */
- const attributesDefaultsPerElement = new Map();
- for (const [name, config] of Object.entries(elems)) {
- /**
- * @type {Set<string>}
- */
- const allowedChildren = new Set();
- if (config.content) {
- for (const elementName of config.content) {
- allowedChildren.add(elementName);
- }
- }
- if (config.contentGroups) {
- for (const contentGroupName of config.contentGroups) {
- const elemsGroup = elemsGroups[contentGroupName];
- if (elemsGroup) {
- for (const elementName of elemsGroup) {
- allowedChildren.add(elementName);
- }
- }
- }
- }
- /**
- * @type {Set<string>}
- */
- const allowedAttributes = new Set();
- if (config.attrs) {
- for (const attrName of config.attrs) {
- allowedAttributes.add(attrName);
- }
- }
- /**
- * @type {Map<string, string>}
- */
- const attributesDefaults = new Map();
- if (config.defaults) {
- for (const [attrName, defaultValue] of Object.entries(config.defaults)) {
- attributesDefaults.set(attrName, defaultValue);
- }
- }
- for (const attrsGroupName of config.attrsGroups) {
- const attrsGroup = attrsGroups[attrsGroupName];
- if (attrsGroup) {
- for (const attrName of attrsGroup) {
- allowedAttributes.add(attrName);
- }
- }
- const groupDefaults = attrsGroupsDefaults[attrsGroupName];
- if (groupDefaults) {
- for (const [attrName, defaultValue] of Object.entries(groupDefaults)) {
- attributesDefaults.set(attrName, defaultValue);
- }
- }
- }
- allowedChildrenPerElement.set(name, allowedChildren);
- allowedAttributesPerElement.set(name, allowedAttributes);
- attributesDefaultsPerElement.set(name, attributesDefaults);
- }
- /**
- * Remove unknown elements content and attributes,
- * remove attributes with default values.
- *
- * @author Kir Belevich
- *
- * @type {import('../lib/types').Plugin<{
- * unknownContent?: boolean,
- * unknownAttrs?: boolean,
- * defaultAttrs?: boolean,
- * uselessOverrides?: boolean,
- * keepDataAttrs?: boolean,
- * keepAriaAttrs?: boolean,
- * keepRoleAttr?: boolean,
- * }>}
- */
- exports.fn = (root, params) => {
- const {
- unknownContent = true,
- unknownAttrs = true,
- defaultAttrs = true,
- uselessOverrides = true,
- keepDataAttrs = true,
- keepAriaAttrs = true,
- keepRoleAttr = false,
- } = params;
- const stylesheet = collectStylesheet(root);
- return {
- element: {
- enter: (node, parentNode) => {
- // skip namespaced elements
- if (node.name.includes(':')) {
- return;
- }
- // skip visiting foreignObject subtree
- if (node.name === 'foreignObject') {
- return visitSkip;
- }
- // remove unknown element's content
- if (unknownContent && parentNode.type === 'element') {
- const allowedChildren = allowedChildrenPerElement.get(
- parentNode.name
- );
- if (allowedChildren == null || allowedChildren.size === 0) {
- // remove unknown elements
- if (allowedChildrenPerElement.get(node.name) == null) {
- detachNodeFromParent(node, parentNode);
- return;
- }
- } else {
- // remove not allowed children
- if (allowedChildren.has(node.name) === false) {
- detachNodeFromParent(node, parentNode);
- return;
- }
- }
- }
- const allowedAttributes = allowedAttributesPerElement.get(node.name);
- const attributesDefaults = attributesDefaultsPerElement.get(node.name);
- const computedParentStyle =
- parentNode.type === 'element'
- ? computeStyle(stylesheet, parentNode)
- : null;
- // remove element's unknown attrs and attrs with default values
- for (const [name, value] of Object.entries(node.attributes)) {
- if (keepDataAttrs && name.startsWith('data-')) {
- continue;
- }
- if (keepAriaAttrs && name.startsWith('aria-')) {
- continue;
- }
- if (keepRoleAttr && name === 'role') {
- continue;
- }
- // skip xmlns attribute
- if (name === 'xmlns') {
- continue;
- }
- // skip namespaced attributes except xml:* and xlink:*
- if (name.includes(':')) {
- const [prefix] = name.split(':');
- if (prefix !== 'xml' && prefix !== 'xlink') {
- continue;
- }
- }
- if (
- unknownAttrs &&
- allowedAttributes &&
- allowedAttributes.has(name) === false
- ) {
- delete node.attributes[name];
- }
- if (
- defaultAttrs &&
- node.attributes.id == null &&
- attributesDefaults &&
- attributesDefaults.get(name) === value
- ) {
- // keep defaults if parent has own or inherited style
- if (
- computedParentStyle == null ||
- computedParentStyle[name] == null
- ) {
- delete node.attributes[name];
- }
- }
- if (uselessOverrides && node.attributes.id == null) {
- const style =
- computedParentStyle == null ? null : computedParentStyle[name];
- if (
- presentationNonInheritableGroupAttrs.includes(name) === false &&
- style != null &&
- style.type === 'static' &&
- style.value === value
- ) {
- delete node.attributes[name];
- }
- }
- }
- },
- },
- };
- };
|