Skip to content

Commit

Permalink
rework attribute selector matching to not use regexes (sveltejs#1710)
Browse files Browse the repository at this point in the history
  • Loading branch information
Conduitry committed Sep 11, 2019
1 parent 14a46a1 commit ae8b45f
Show file tree
Hide file tree
Showing 3 changed files with 36 additions and 23 deletions.
46 changes: 23 additions & 23 deletions src/compiler/compile/css/Selector.ts
Original file line number Diff line number Diff line change
Expand Up @@ -138,8 +138,9 @@ function apply_selector(stylesheet: Stylesheet, blocks: Block[], node: Node, sta

while (i--) {
const selector = block.selectors[i];
const name = typeof selector.name === 'string' && selector.name.replace(/\\(.)/g, '$1');

if (selector.type === 'PseudoClassSelector' && selector.name === 'global') {
if (selector.type === 'PseudoClassSelector' && name === 'global') {
// TODO shouldn't see this here... maybe we should enforce that :global(...)
// cannot be sandwiched between non-global selectors?
return false;
Expand All @@ -150,20 +151,19 @@ function apply_selector(stylesheet: Stylesheet, blocks: Block[], node: Node, sta
}

if (selector.type === 'ClassSelector') {
if (!attribute_matches(node, 'class', selector.name, '~=', false) && !class_matches(node, selector.name)) return false;
if (!attribute_matches(node, 'class', name, '~=', false) && !node.classes.some(c => c.name === name)) return false;
}

else if (selector.type === 'IdSelector') {
if (!attribute_matches(node, 'id', selector.name, '=', false)) return false;
if (!attribute_matches(node, 'id', name, '=', false)) return false;
}

else if (selector.type === 'AttributeSelector') {
if (!attribute_matches(node, selector.name.name, selector.value && unquote(selector.value), selector.matcher, selector.flags)) return false;
}

else if (selector.type === 'TypeSelector') {
// remove toLowerCase() in v2, when uppercase elements will be forbidden
if (node.name.toLowerCase() !== selector.name.toLowerCase() && selector.name !== '*') return false;
if (node.name.toLowerCase() !== name.toLowerCase() && name !== '*') return false;
}

else {
Expand Down Expand Up @@ -206,14 +206,21 @@ function apply_selector(stylesheet: Stylesheet, blocks: Block[], node: Node, sta
return true;
}

const operators = {
'=' : (value: string, flags: string) => new RegExp(`^${value}$`, flags),
'~=': (value: string, flags: string) => new RegExp(`\\b${value}\\b`, flags),
'|=': (value: string, flags: string) => new RegExp(`^${value}(-.+)?$`, flags),
'^=': (value: string, flags: string) => new RegExp(`^${value}`, flags),
'$=': (value: string, flags: string) => new RegExp(`${value}$`, flags),
'*=': (value: string, flags: string) => new RegExp(value, flags)
};
function test_attribute(operator, expected_value, case_insensitive, value) {
if (case_insensitive) {
expected_value = expected_value.toLowerCase();
value = value.toLowerCase();
}
switch (operator) {
case '=': return value === expected_value;
case '~=': return ` ${value} `.includes(` ${expected_value} `);
case '|=': return `${value}-`.startsWith(`${expected_value}-`);
case '^=': return value.startsWith(expected_value);
case '$=': return value.endsWith(expected_value);
case '*=': return value.includes(expected_value);
default: throw new Error(`this shouldn't happen`);
}
}

function attribute_matches(node: Node, name: string, expected_value: string, operator: string, case_insensitive: boolean) {
const spread = node.attributes.find(attr => attr.type === 'Spread');
Expand All @@ -227,29 +234,22 @@ function attribute_matches(node: Node, name: string, expected_value: string, ope
if (attr.chunks.length > 1) return true;
if (!expected_value) return true;

const pattern = operators[operator](expected_value, case_insensitive ? 'i' : '');
const value = attr.chunks[0];

if (!value) return false;
if (value.type === 'Text') return pattern.test(value.data);
if (value.type === 'Text') return test_attribute(operator, expected_value, case_insensitive, value.data);

const possible_values = new Set();
gather_possible_values(value.node, possible_values);
if (possible_values.has(UNKNOWN)) return true;

for (const x of Array.from(possible_values)) { // TypeScript for-of is slightly unlike JS
if (pattern.test(x)) return true;
for (const value of possible_values) {
if (test_attribute(operator, expected_value, case_insensitive, value)) return true;
}

return false;
}

function class_matches(node, name: string) {
return node.classes.some((class_directive) => {
return new RegExp(`\\b${name}\\b`).test(class_directive.name);
});
}

function unquote(value: Node) {
if (value.type === 'Identifier') return value.name;
const str = value.value;
Expand Down
1 change: 1 addition & 0 deletions test/css/samples/weird-selectors/expected.css
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
.-foo.svelte-xyz{color:red}[title='['].svelte-xyz{color:blue}
12 changes: 12 additions & 0 deletions test/css/samples/weird-selectors/input.svelte
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
<div class='-foo'>foo</div>

<div title='['>bar</div>

<style>
.-foo {
color: red;
}
[title='['] {
color: blue;
}
</style>

0 comments on commit ae8b45f

Please sign in to comment.