119 lines
		
	
	
	
		
			4.4 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			119 lines
		
	
	
	
		
			4.4 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
"use strict";
 | 
						|
var __importDefault = (this && this.__importDefault) || function (mod) {
 | 
						|
    return (mod && mod.__esModule) ? mod : { "default": mod };
 | 
						|
};
 | 
						|
Object.defineProperty(exports, "__esModule", { value: true });
 | 
						|
exports.compileToken = exports.compileUnsafe = exports.compile = void 0;
 | 
						|
var css_what_1 = require("css-what");
 | 
						|
var boolbase_1 = require("boolbase");
 | 
						|
var sort_1 = __importDefault(require("./sort"));
 | 
						|
var procedure_1 = require("./procedure");
 | 
						|
var general_1 = require("./general");
 | 
						|
var subselects_1 = require("./pseudo-selectors/subselects");
 | 
						|
/**
 | 
						|
 * Compiles a selector to an executable function.
 | 
						|
 *
 | 
						|
 * @param selector Selector to compile.
 | 
						|
 * @param options Compilation options.
 | 
						|
 * @param context Optional context for the selector.
 | 
						|
 */
 | 
						|
function compile(selector, options, context) {
 | 
						|
    var next = compileUnsafe(selector, options, context);
 | 
						|
    return (0, subselects_1.ensureIsTag)(next, options.adapter);
 | 
						|
}
 | 
						|
exports.compile = compile;
 | 
						|
function compileUnsafe(selector, options, context) {
 | 
						|
    var token = typeof selector === "string" ? (0, css_what_1.parse)(selector) : selector;
 | 
						|
    return compileToken(token, options, context);
 | 
						|
}
 | 
						|
exports.compileUnsafe = compileUnsafe;
 | 
						|
function includesScopePseudo(t) {
 | 
						|
    return (t.type === "pseudo" &&
 | 
						|
        (t.name === "scope" ||
 | 
						|
            (Array.isArray(t.data) &&
 | 
						|
                t.data.some(function (data) { return data.some(includesScopePseudo); }))));
 | 
						|
}
 | 
						|
var DESCENDANT_TOKEN = { type: css_what_1.SelectorType.Descendant };
 | 
						|
var FLEXIBLE_DESCENDANT_TOKEN = {
 | 
						|
    type: "_flexibleDescendant",
 | 
						|
};
 | 
						|
var SCOPE_TOKEN = {
 | 
						|
    type: css_what_1.SelectorType.Pseudo,
 | 
						|
    name: "scope",
 | 
						|
    data: null,
 | 
						|
};
 | 
						|
/*
 | 
						|
 * CSS 4 Spec (Draft): 3.3.1. Absolutizing a Scope-relative Selector
 | 
						|
 * http://www.w3.org/TR/selectors4/#absolutizing
 | 
						|
 */
 | 
						|
function absolutize(token, _a, context) {
 | 
						|
    var adapter = _a.adapter;
 | 
						|
    // TODO Use better check if the context is a document
 | 
						|
    var hasContext = !!(context === null || context === void 0 ? void 0 : context.every(function (e) {
 | 
						|
        var parent = adapter.isTag(e) && adapter.getParent(e);
 | 
						|
        return e === subselects_1.PLACEHOLDER_ELEMENT || (parent && adapter.isTag(parent));
 | 
						|
    }));
 | 
						|
    for (var _i = 0, token_1 = token; _i < token_1.length; _i++) {
 | 
						|
        var t = token_1[_i];
 | 
						|
        if (t.length > 0 && (0, procedure_1.isTraversal)(t[0]) && t[0].type !== "descendant") {
 | 
						|
            // Don't continue in else branch
 | 
						|
        }
 | 
						|
        else if (hasContext && !t.some(includesScopePseudo)) {
 | 
						|
            t.unshift(DESCENDANT_TOKEN);
 | 
						|
        }
 | 
						|
        else {
 | 
						|
            continue;
 | 
						|
        }
 | 
						|
        t.unshift(SCOPE_TOKEN);
 | 
						|
    }
 | 
						|
}
 | 
						|
function compileToken(token, options, context) {
 | 
						|
    var _a;
 | 
						|
    token = token.filter(function (t) { return t.length > 0; });
 | 
						|
    token.forEach(sort_1.default);
 | 
						|
    context = (_a = options.context) !== null && _a !== void 0 ? _a : context;
 | 
						|
    var isArrayContext = Array.isArray(context);
 | 
						|
    var finalContext = context && (Array.isArray(context) ? context : [context]);
 | 
						|
    absolutize(token, options, finalContext);
 | 
						|
    var shouldTestNextSiblings = false;
 | 
						|
    var query = token
 | 
						|
        .map(function (rules) {
 | 
						|
        if (rules.length >= 2) {
 | 
						|
            var first = rules[0], second = rules[1];
 | 
						|
            if (first.type !== "pseudo" || first.name !== "scope") {
 | 
						|
                // Ignore
 | 
						|
            }
 | 
						|
            else if (isArrayContext && second.type === "descendant") {
 | 
						|
                rules[1] = FLEXIBLE_DESCENDANT_TOKEN;
 | 
						|
            }
 | 
						|
            else if (second.type === "adjacent" ||
 | 
						|
                second.type === "sibling") {
 | 
						|
                shouldTestNextSiblings = true;
 | 
						|
            }
 | 
						|
        }
 | 
						|
        return compileRules(rules, options, finalContext);
 | 
						|
    })
 | 
						|
        .reduce(reduceRules, boolbase_1.falseFunc);
 | 
						|
    query.shouldTestNextSiblings = shouldTestNextSiblings;
 | 
						|
    return query;
 | 
						|
}
 | 
						|
exports.compileToken = compileToken;
 | 
						|
function compileRules(rules, options, context) {
 | 
						|
    var _a;
 | 
						|
    return rules.reduce(function (previous, rule) {
 | 
						|
        return previous === boolbase_1.falseFunc
 | 
						|
            ? boolbase_1.falseFunc
 | 
						|
            : (0, general_1.compileGeneralSelector)(previous, rule, options, context, compileToken);
 | 
						|
    }, (_a = options.rootFunc) !== null && _a !== void 0 ? _a : boolbase_1.trueFunc);
 | 
						|
}
 | 
						|
function reduceRules(a, b) {
 | 
						|
    if (b === boolbase_1.falseFunc || a === boolbase_1.trueFunc) {
 | 
						|
        return a;
 | 
						|
    }
 | 
						|
    if (a === boolbase_1.falseFunc || b === boolbase_1.trueFunc) {
 | 
						|
        return b;
 | 
						|
    }
 | 
						|
    return function combine(elem) {
 | 
						|
        return a(elem) || b(elem);
 | 
						|
    };
 | 
						|
}
 |