@wyw-in-js/transform 0.2.3 → 0.4.0

package/types/transform/generators/createStylisPreprocessor.js

@@ -23,7 +23,8 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createStylisPreprocessor = exports.createStylisUrlReplacePlugin = exports.stylisGlobalPlugin = exports.transformUrl = void 0;
+exports.createStylisPreprocessor = exports.createKeyframeSuffixerPlugin = exports.createStylisUrlReplacePlugin = exports.stylisGlobalPlugin = exports.transformUrl = void 0;
+/* eslint-disable no-continue */
 const path = __importStar(require("path"));
 const stylis_1 = require("stylis");
 const POSIX_SEP = path.posix.sep;
@@ -38,11 +39,51 @@ function transformUrl(url, outputFilename, sourceFilename, platformPath = path)
     return relative.split(platformPath.sep).join(POSIX_SEP);
 }
 exports.transformUrl = transformUrl;
+const DEFINED_KEYFRAMES = Symbol('definedKeyframes');
+const ORIGINAL_VALUE_KEY = Symbol('originalValue');
+const IS_GLOBAL_KEYFRAMES = Symbol('isGlobalKeyframes');
+const getOriginalElementValue = (element) => {
+    return element ? element[ORIGINAL_VALUE_KEY] ?? element.value : '';
+};
+function throwIfNotProd(key, value, type) {
+    if (process.env.NODE_ENV !== 'production') {
+        throw new Error(`"element.${key}" has type "${type}" (${JSON.stringify(value, null, 2)}), it's not expected. Please report a bug if it happens.`);
+    }
+    return false;
+}
+function childrenIsString(children) {
+    return (typeof children === 'string' ||
+        throwIfNotProd('children', children, 'Element[]'));
+}
+function propsAreStrings(props) {
+    return Array.isArray(props) || throwIfNotProd('props', props, 'string');
+}
+function propsIsString(props) {
+    return (typeof props === 'string' || throwIfNotProd('props', props, 'string[]'));
+}
+const isDeclaration = (element) => {
+    return (element.type === stylis_1.DECLARATION &&
+        propsIsString(element.props) &&
+        childrenIsString(element.children));
+};
+const isKeyframes = (element) => {
+    return element.type === stylis_1.KEYFRAMES && propsAreStrings(element.props);
+};
+const isRuleset = (element) => {
+    return element.type === stylis_1.RULESET && propsAreStrings(element.props);
+};
 /**
  * Stylis plugin that mimics :global() selector behavior from Stylis v3.
  */
 const stylisGlobalPlugin = (element) => {
-    function getGlobalSelectorModifiers(value) {
+    function getGlobalSelectorModifiers(el) {
+        const { parent } = el;
+        const value = getOriginalElementValue(el);
+        const parentValue = getOriginalElementValue(parent);
+        if ((parent?.children.length === 0 && parentValue.includes(':global(')) ||
+            (parent && !value.includes(':global('))) {
+            return getGlobalSelectorModifiers(parent);
+        }
         const match = value.match(/(&\f( )?)?:global\(/);
         if (match === null) {
             throw new Error(`Failed to match :global() selector in "${value}". Please report a bug if it happens.`);
@@ -53,62 +94,53 @@ const stylisGlobalPlugin = (element) => {
             includeSpaceDelimiter: !!spaceDelimiter,
         };
     }
-    switch (element.type) {
-        case stylis_1.RULESET:
-            if (typeof element.props === 'string') {
-                if (process.env.NODE_ENV !== 'production') {
-                    throw new Error(`"element.props" has type "string" (${JSON.stringify(element.props, null, 2)}), it's not expected. Please report a bug if it happens.`);
-                }
-                return;
+    if (!isRuleset(element)) {
+        return;
+    }
+    Object.assign(element, {
+        props: element.props.map((cssSelector) => {
+            // The value can be changed by other middlewares, but we need an original one with `&`
+            Object.assign(element, { [ORIGINAL_VALUE_KEY]: element.value });
+            // Avoids calling tokenize() on every string
+            if (!cssSelector.includes(':global(')) {
+                return cssSelector;
             }
-            Object.assign(element, {
-                props: element.props.map((cssSelector) => {
-                    // Avoids calling tokenize() on every string
-                    if (!cssSelector.includes(':global(')) {
-                        return cssSelector;
-                    }
-                    if (element.children.length === 0) {
-                        Object.assign(element, {
-                            global: getGlobalSelectorModifiers(element.value),
-                        });
-                        return cssSelector;
-                    }
-                    const { includeBaseSelector, includeSpaceDelimiter } = element.parent?.global || getGlobalSelectorModifiers(element.value);
-                    const tokens = (0, stylis_1.tokenize)(cssSelector);
-                    let selector = '';
-                    for (let i = 0, len = tokens.length; i < len; i++) {
-                        const token = tokens[i];
-                        //
-                        // Match for ":global("
-                        if (token === ':' && tokens[i + 1] === 'global') {
-                            //
-                            // Match for ":global()"
-                            if (tokens[i + 2] === '()') {
-                                selector = [
-                                    ...tokens.slice(i + 4),
-                                    includeSpaceDelimiter ? ' ' : '',
-                                    ...(includeBaseSelector ? tokens.slice(0, i - 1) : []),
-                                    includeSpaceDelimiter ? '' : ' ',
-                                ].join('');
-                                break;
-                            }
-                            //
-                            // Match for ":global(selector)"
-                            selector = [
-                                tokens[i + 2].slice(1, -1),
-                                includeSpaceDelimiter ? ' ' : '',
-                                ...(includeBaseSelector ? tokens.slice(0, i - 1) : []),
-                                includeSpaceDelimiter ? '' : ' ',
-                            ].join('');
-                            break;
-                        }
+            if (element.children.length === 0) {
+                return cssSelector;
+            }
+            const { includeBaseSelector, includeSpaceDelimiter } = getGlobalSelectorModifiers(element);
+            const tokens = (0, stylis_1.tokenize)(cssSelector);
+            let selector = '';
+            for (let i = 0, len = tokens.length; i < len; i++) {
+                const token = tokens[i];
+                //
+                // Match for ":global("
+                if (token === ':' && tokens[i + 1] === 'global') {
+                    //
+                    // Match for ":global()"
+                    if (tokens[i + 2] === '()') {
+                        selector = [
+                            ...tokens.slice(i + 4),
+                            includeSpaceDelimiter ? ' ' : '',
+                            ...(includeBaseSelector ? tokens.slice(0, i - 1) : []),
+                            includeSpaceDelimiter ? '' : ' ',
+                        ].join('');
+                        break;
                     }
-                    return selector;
-                }),
-            });
-            break;
-        default:
-    }
+                    //
+                    // Match for ":global(selector)"
+                    selector = [
+                        tokens[i + 2].slice(1, -1),
+                        includeSpaceDelimiter ? ' ' : '',
+                        ...(includeBaseSelector ? tokens.slice(0, i - 1) : []),
+                        includeSpaceDelimiter ? '' : ' ',
+                    ].join('');
+                    break;
+                }
+            }
+            return selector;
+        }),
+    });
 };
 exports.stylisGlobalPlugin = stylisGlobalPlugin;
 function createStylisUrlReplacePlugin(filename, outputFilename) {
@@ -117,20 +149,116 @@ function createStylisUrlReplacePlugin(filename, outputFilename) {
             // When writing to a file, we need to adjust the relative paths inside url(..) expressions.
             // It'll allow css-loader to resolve an imported asset properly.
             // eslint-disable-next-line no-param-reassign
-            element.return = element.value.replace(/\b(url\((["']?))(\.[^)]+?)(\2\))/g, (match, p1, p2, p3, p4) => p1 + transformUrl(p3, outputFilename, filename) + p4);
+            element.return = element.value.replace(/\b(url\((["']?))(\.[^)]+?)(\2\))/g, (_match, p1, _p2, p3, p4) => p1 + transformUrl(p3, outputFilename, filename) + p4);
         }
     };
 }
 exports.createStylisUrlReplacePlugin = createStylisUrlReplacePlugin;
+function createKeyframeSuffixerPlugin() {
+    const prefixes = ['webkit', 'moz', 'ms', 'o', ''].map((i) => i ? `-${i}-` : '');
+    const getPrefixedProp = (prop) => prefixes.map((prefix) => `${prefix}${prop}`);
+    const buildPropsRegexp = (prop, isAtRule) => {
+        const [at, colon] = isAtRule ? ['@', ''] : ['', ':'];
+        return new RegExp(`^(${at}(?:${getPrefixedProp(prop).join('|')})${colon})\\s*`);
+    };
+    const animationNameRegexp = /:global\(([\w_-]+)\)|([\w_-]+)/;
+    const getReplacer = (startsWith, searchValue, replacer) => {
+        return (input) => {
+            const [fullMatch] = input.match(startsWith) ?? [];
+            if (fullMatch === undefined) {
+                return input;
+            }
+            const rest = input.slice(fullMatch.length);
+            return fullMatch + rest.replace(searchValue, replacer);
+        };
+    };
+    const elementToKeyframeSuffix = (el) => {
+        if (el.parent) {
+            return elementToKeyframeSuffix(el.parent);
+        }
+        return el.value.replaceAll(/[^a-zA-Z0-9_-]/g, '');
+    };
+    const animationPropsSet = new Set([
+        ...getPrefixedProp('animation'),
+        ...getPrefixedProp('animation-name'),
+    ]);
+    const getDefinedKeyframes = (element) => {
+        if (element[DEFINED_KEYFRAMES]) {
+            return element[DEFINED_KEYFRAMES];
+        }
+        if (element.parent) {
+            return getDefinedKeyframes(element.parent);
+        }
+        const keyframes = new Set();
+        for (const sibling of element.siblings ?? []) {
+            if (!isKeyframes(sibling) || sibling[IS_GLOBAL_KEYFRAMES] === true) {
+                continue;
+            }
+            keyframes.add(sibling.props[0]);
+        }
+        Object.assign(element, { [DEFINED_KEYFRAMES]: keyframes });
+        return keyframes;
+    };
+    return (element) => {
+        if (isKeyframes(element) && element.parent) {
+            const suffix = elementToKeyframeSuffix(element);
+            const replaceFn = (_match, globalMatch, scopedMatch) => globalMatch || `${scopedMatch}-${suffix}`;
+            Object.assign(element, {
+                [IS_GLOBAL_KEYFRAMES]: element.props[0]?.startsWith(':global(') ?? false,
+                props: element.props.map(getReplacer(/^\s*/, animationNameRegexp, replaceFn)),
+                value: getReplacer(buildPropsRegexp('keyframes', true), animationNameRegexp, replaceFn)(element.value),
+            });
+            return;
+        }
+        if (isDeclaration(element)) {
+            const suffix = elementToKeyframeSuffix(element);
+            const keys = [
+                'children',
+                'return',
+                'value',
+            ];
+            if (animationPropsSet.has(element.props)) {
+                const scopedKeyframes = getDefinedKeyframes(element);
+                const patch = Object.fromEntries(keys.map((key) => {
+                    const tokens = (0, stylis_1.tokenize)(element[key]);
+                    let result = '';
+                    for (let i = 0; i < tokens.length; i += 1) {
+                        if (tokens[i] === ':' &&
+                            tokens[i + 1] === 'global' &&
+                            tokens[i + 2].startsWith('(')) {
+                            const globalName = tokens[i + 2].substring(1, tokens[i + 2].length - 1);
+                            i += 2;
+                            result += globalName;
+                            if (tokens[i + 1] !== ';') {
+                                result += ' ';
+                            }
+                            continue;
+                        }
+                        if (scopedKeyframes.has(tokens[i])) {
+                            result += `${tokens[i]}-${suffix}`;
+                            continue;
+                        }
+                        result += tokens[i];
+                    }
+                    return [key, result];
+                }));
+                Object.assign(element, patch);
+            }
+        }
+    };
+}
+exports.createKeyframeSuffixerPlugin = createKeyframeSuffixerPlugin;
+const isMiddleware = (obj) => obj !== null;
 function createStylisPreprocessor(options) {
     function stylisPreprocess(selector, text) {
         const compiled = (0, stylis_1.compile)(`${selector} {${text}}\n`);
         return (0, stylis_1.serialize)(compiled, (0, stylis_1.middleware)([
             createStylisUrlReplacePlugin(options.filename, options.outputFilename),
             exports.stylisGlobalPlugin,
-            stylis_1.prefixer,
+            options.prefixer === false ? null : stylis_1.prefixer,
+            createKeyframeSuffixerPlugin(),
             stylis_1.stringify,
-        ]));
+        ].filter(isMiddleware)));
     }
     return stylisPreprocess;
 }