storybook-addon-pseudo-states 1.14.0 → 1.15.1

package/dist/cjs/withPseudoState.js

@@ -11,102 +11,74 @@ var _coreEvents = require("@storybook/core-events");
 
 var _constants = require("./constants");
 
-var _splitSelectors = require("./splitSelectors");
+var _rewriteStyleSheet = require("./rewriteStyleSheet");
 
-function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }
+/* eslint-env browser */
+const channel = _addons.addons.getChannel();
 
-function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { _defineProperty(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }
-
-function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
-
-function _createForOfIteratorHelper(o, allowArrayLike) { var it = typeof Symbol !== "undefined" && o[Symbol.iterator] || o["@@iterator"]; if (!it) { if (Array.isArray(o) || (it = _unsupportedIterableToArray(o)) || allowArrayLike && o && typeof o.length === "number") { if (it) o = it; var i = 0; var F = function F() {}; return { s: F, n: function n() { if (i >= o.length) return { done: true }; return { done: false, value: o[i++] }; }, e: function e(_e2) { throw _e2; }, f: F }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } var normalCompletion = true, didErr = false, err; return { s: function s() { it = it.call(o); }, n: function n() { var step = it.next(); normalCompletion = step.done; return step; }, e: function e(_e3) { didErr = true; err = _e3; }, f: function f() { try { if (!normalCompletion && it.return != null) it.return(); } finally { if (didErr) throw err; } } }; }
-
-function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
-
-function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
-
-function _iterableToArrayLimit(arr, i) { var _i = arr == null ? null : typeof Symbol !== "undefined" && arr[Symbol.iterator] || arr["@@iterator"]; if (_i == null) return; var _arr = []; var _n = true; var _d = false; var _s, _e; try { for (_i = _i.call(arr); !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"] != null) _i["return"](); } finally { if (_d) throw _e; } } return _arr; }
-
-function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
-
-function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }
-
-function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
-
-function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
-
-function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }
-
-function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }
-
-function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) { arr2[i] = arr[i]; } return arr2; }
-
-var pseudoStates = Object.values(_constants.PSEUDO_STATES);
-var matchOne = new RegExp(":(".concat(pseudoStates.join("|"), ")"));
-var matchAll = new RegExp(":(".concat(pseudoStates.join("|"), ")"), "g"); // Drops any existing pseudo state classnames that carried over from a previously viewed story
+const shadowHosts = new Set(); // Drops any existing pseudo state classnames that carried over from a previously viewed story
 // before adding the new classnames. We do this the old-fashioned way, for IE compatibility.
 
-var applyClasses = function applyClasses(element, classnames) {
-  var _element$className$sp;
-
-  element.className = (_element$className$sp = element.className.split(" ").filter(function (classname) {
-    return classname && classname.indexOf("pseudo-") !== 0;
-  })).concat.apply(_element$className$sp, _toConsumableArray(classnames)).join(" ");
+const applyClasses = (element, classnames) => {
+  element.className = element.className.split(" ").filter(classname => classname && classname.indexOf("pseudo-") !== 0).concat(...classnames).join(" ");
 };
 
-var applyParameter = function applyParameter(element, parameter) {
-  return applyClasses(element, Object.entries(parameter || {}).filter(function (_ref) {
-    var _ref2 = _slicedToArray(_ref, 2),
-        _ = _ref2[0],
-        value = _ref2[1];
+const applyParameter = (rootElement, parameter) => {
+  const map = new Map([[rootElement, new Set()]]);
+
+  const add = (target, state) => map.set(target, new Set([...(map.get(target) || []), state]));
 
-    return value;
-  }).map(function (_ref3) {
-    var _ref4 = _slicedToArray(_ref3, 1),
-        key = _ref4[0];
+  Object.entries(parameter || {}).forEach(_ref => {
+    let [state, value] = _ref;
 
-    return "pseudo-".concat(_constants.PSEUDO_STATES[key]);
-  }));
+    if (typeof value === "boolean") {
+      // default API - applying pseudo class to root element.
+      add(rootElement, value && state);
+    } else if (typeof value === "string") {
+      // explicit selectors API - applying pseudo class to a specific element
+      rootElement.querySelectorAll(value).forEach(el => add(el, state));
+    } else if (Array.isArray(value)) {
+      // explicit selectors API - we have an array (of strings) recursively handle each one
+      value.forEach(sel => rootElement.querySelectorAll(sel).forEach(el => add(el, state)));
+    }
+  });
+  map.forEach((states, target) => {
+    const classnames = [];
+    states.forEach(key => _constants.PSEUDO_STATES[key] && classnames.push("pseudo-".concat(_constants.PSEUDO_STATES[key])));
+    applyClasses(target, classnames);
+  });
 }; // Traverses ancestry to collect relevant pseudo classnames, and applies them to the shadow host.
 // Shadow DOM can only access classes on its host. Traversing is needed to mimic the CSS cascade.
 
 
-var updateShadowHost = function updateShadowHost(shadowHost) {
-  var classnames = new Set();
+const updateShadowHost = shadowHost => {
+  const classnames = new Set();
 
-  for (var element = shadowHost.parentElement; element; element = element.parentElement) {
+  for (let element = shadowHost.parentElement; element; element = element.parentElement) {
     if (!element.className) continue;
-    element.className.split(" ").filter(function (classname) {
-      return classname.indexOf("pseudo-") === 0;
-    }).forEach(function (classname) {
-      return classnames.add(classname);
-    });
+    element.className.split(" ").filter(classname => classname.indexOf("pseudo-") === 0).forEach(classname => classnames.add(classname));
   }
 
   applyClasses(shadowHost, classnames);
-}; // Keep track of attached shadow host elements for the current story
-
-
-var shadowHosts = new Set();
-
-_addons.addons.getChannel().on(_coreEvents.STORY_CHANGED, function () {
-  return shadowHosts.clear();
-}); // Global decorator that rewrites stylesheets and applies classnames to render pseudo styles
-
-
-var withPseudoState = function withPseudoState(StoryFn, _ref5) {
-  var viewMode = _ref5.viewMode,
-      parameters = _ref5.parameters,
-      id = _ref5.id,
-      globalsArgs = _ref5.globals;
-  var parameter = parameters.pseudo;
-  var globals = globalsArgs.pseudo;
-
-  var channel = _addons.addons.getChannel(); // Sync parameter to globals, used by the toolbar (only in canvas as this
+}; // Global decorator that rewrites stylesheets and applies classnames to render pseudo styles
+
+
+const withPseudoState = (StoryFn, _ref2) => {
+  let {
+    viewMode,
+    parameters,
+    id,
+    globals: globalsArgs
+  } = _ref2;
+  const {
+    pseudo: parameter
+  } = parameters;
+  const {
+    pseudo: globals
+  } = globalsArgs; // Sync parameter to globals, used by the toolbar (only in canvas as this
   // doesn't make sense for docs because many stories are displayed at once)
 
-
-  (0, _addons.useEffect)(function () {
+  (0, _addons.useEffect)(() => {
     if (parameter !== globals && viewMode === "story") {
       channel.emit(_coreEvents.UPDATE_GLOBALS, {
         globals: {
@@ -117,135 +89,34 @@ var withPseudoState = function withPseudoState(StoryFn, _ref5) {
   }, [parameter, viewMode]); // Convert selected states to classnames and apply them to the story root element.
   // Then update each shadow host to redetermine its own pseudo classnames.
 
-  (0, _addons.useEffect)(function () {
-    var timeout = setTimeout(function () {
-      var element = document.getElementById(viewMode === "docs" ? "story--".concat(id) : "root");
-      applyParameter(element, parameter);
+  (0, _addons.useEffect)(() => {
+    const timeout = setTimeout(() => {
+      const element = document.getElementById(viewMode === "docs" ? "story--".concat(id) : "root");
+      applyParameter(element, globals || parameter);
       shadowHosts.forEach(updateShadowHost);
     }, 0);
-    return function () {
-      return clearTimeout(timeout);
-    };
-  }, [parameter, viewMode]);
+    return () => clearTimeout(timeout);
+  }, [globals, parameter, viewMode]);
   return StoryFn();
-};
-
-exports.withPseudoState = withPseudoState;
-var warnings = new Set();
-
-var warnOnce = function warnOnce(message) {
-  if (warnings.has(message)) return; // eslint-disable-next-line no-console
-
-  console.warn(message);
-  warnings.add(message);
 }; // Rewrite CSS rules for pseudo-states on all stylesheets to add an alternative selector
 
 
-function rewriteStyleSheets(shadowRoot) {
+exports.withPseudoState = withPseudoState;
+
+const rewriteStyleSheets = shadowRoot => {
   var _shadowRoot$adoptedSt;
 
-  var styleSheets = shadowRoot ? shadowRoot.styleSheets : document.styleSheets;
+  let styleSheets = shadowRoot ? shadowRoot.styleSheets : document.styleSheets;
   if (shadowRoot !== null && shadowRoot !== void 0 && (_shadowRoot$adoptedSt = shadowRoot.adoptedStyleSheets) !== null && _shadowRoot$adoptedSt !== void 0 && _shadowRoot$adoptedSt.length) styleSheets = shadowRoot.adoptedStyleSheets;
+  Array.from(styleSheets).forEach(sheet => (0, _rewriteStyleSheet.rewriteStyleSheet)(sheet, shadowRoot, shadowHosts));
+}; // Only track shadow hosts for the current story
 
-  var _iterator = _createForOfIteratorHelper(styleSheets),
-      _step;
-
-  try {
-    for (_iterator.s(); !(_step = _iterator.n()).done;) {
-      var sheet = _step.value;
-
-      if (sheet._pseudoStatesRewritten) {
-        continue;
-      } else {
-        sheet._pseudoStatesRewritten = true;
-      }
-
-      try {
-        var index = 0;
-
-        var _iterator2 = _createForOfIteratorHelper(sheet.cssRules),
-            _step2;
-
-        try {
-          for (_iterator2.s(); !(_step2 = _iterator2.n()).done;) {
-            var _step2$value = _step2.value,
-                cssText = _step2$value.cssText,
-                selectorText = _step2$value.selectorText;
-
-            if (matchOne.test(selectorText)) {
-              var selectors = (0, _splitSelectors.splitSelectors)(selectorText);
-              var newRule = cssText.replace(selectorText, selectors.flatMap(function (selector) {
-                if (selector.includes(".pseudo-")) return [];
-                var states = [];
-                var plainSelector = selector.replace(matchAll, function (_, state) {
-                  states.push(state);
-                  return "";
-                });
-                var stateSelector;
-
-                if (!matchOne.test(selector)) {
-                  return [selector];
-                }
-
-                if (selector.startsWith(":host(") || selector.startsWith("::slotted(")) {
-                  stateSelector = states.reduce(function (acc, state) {
-                    return acc.replaceAll(":".concat(state), ".pseudo-".concat(state));
-                  }, selector);
-                } else if (shadowRoot) {
-                  stateSelector = ":host(".concat(states.map(function (s) {
-                    return ".pseudo-".concat(s);
-                  }).join(""), ") ").concat(plainSelector);
-                } else {
-                  stateSelector = "".concat(states.map(function (s) {
-                    return ".pseudo-".concat(s);
-                  }).join(""), " ").concat(plainSelector);
-                }
-
-                return [selector, stateSelector];
-              }).join(", "));
-              sheet.deleteRule(index);
-              sheet.insertRule(newRule, index);
-              if (shadowRoot) shadowHosts.add(shadowRoot.host);
-            }
-
-            index++;
-
-            if (index > 1000) {
-              warnOnce("Reached maximum of 1000 pseudo selectors per sheet, skipping the rest.");
-              break;
-            }
-          }
-        } catch (err) {
-          _iterator2.e(err);
-        } finally {
-          _iterator2.f();
-        }
-      } catch (e) {
-        if (e.toString().includes("cssRules")) {
-          warnOnce("Can't access cssRules, likely due to CORS restrictions: ".concat(sheet.href));
-        } else {
-          // eslint-disable-next-line no-console
-          console.error(e, sheet.href);
-        }
-      }
-    }
-  } catch (err) {
-    _iterator.e(err);
-  } finally {
-    _iterator.f();
-  }
-} // Reinitialize CSS enhancements every time the story changes
-
-
-_addons.addons.getChannel().on(_coreEvents.STORY_RENDERED, function () {
-  return rewriteStyleSheets();
-}); // Reinitialize CSS enhancements every time a docs page is rendered
 
+channel.on(_coreEvents.STORY_CHANGED, () => shadowHosts.clear()); // Reinitialize CSS enhancements every time the story changes
 
-_addons.addons.getChannel().on(_coreEvents.DOCS_RENDERED, function () {
-  return rewriteStyleSheets();
-}); // IE doesn't support shadow DOM
+channel.on(_coreEvents.STORY_RENDERED, () => rewriteStyleSheets()); // Reinitialize CSS enhancements every time a docs page is rendered
 
+channel.on(_coreEvents.DOCS_RENDERED, () => rewriteStyleSheets()); // IE doesn't support shadow DOM
 
 if (Element.prototype.attachShadow) {
   // Monkeypatch the attachShadow method so we can handle pseudo styles inside shadow DOM
@@ -253,12 +124,12 @@ if (Element.prototype.attachShadow) {
 
   Element.prototype.attachShadow = function attachShadow(init) {
     // Force "open" mode, so we can access the shadowRoot
-    var shadowRoot = this._attachShadow(_objectSpread(_objectSpread({}, init), {}, {
+    const shadowRoot = this._attachShadow({ ...init,
       mode: "open"
-    })); // Wait for it to render and apply its styles before rewriting them
+    }); // Wait for it to render and apply its styles before rewriting them
 
 
-    requestAnimationFrame(function () {
+    requestAnimationFrame(() => {
       rewriteStyleSheets(shadowRoot);
       updateShadowHost(shadowRoot.host);
     });

package/dist/esm/PseudoStateTool.js

@@ -1,34 +1,20 @@
-function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }
-
-function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { _defineProperty(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }
-
-function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
-
-function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
-
-function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
-
-function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
-
-function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) { arr2[i] = arr[i]; } return arr2; }
-
-function _iterableToArrayLimit(arr, i) { var _i = arr == null ? null : typeof Symbol !== "undefined" && arr[Symbol.iterator] || arr["@@iterator"]; if (_i == null) return; var _arr = []; var _n = true; var _d = false; var _s, _e; try { for (_i = _i.call(arr); !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"] != null) _i["return"](); } finally { if (_d) throw _e; } } return _arr; }
-
-function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
-
 import React, { useCallback, useMemo } from "react";
 import { useGlobals } from "@storybook/api";
 import { Icons, IconButton, WithTooltip, TooltipLinkList } from "@storybook/components";
 import { styled, color } from "@storybook/theming";
 import { PSEUDO_STATES } from "./constants";
-var LinkTitle = styled.span(function (_ref) {
-  var active = _ref.active;
+const LinkTitle = styled.span(_ref => {
+  let {
+    active
+  } = _ref;
   return {
     color: active ? color.secondary : "inherit"
   };
 });
-var LinkIcon = styled(Icons)(function (_ref2) {
-  var active = _ref2.active;
+const LinkIcon = styled(Icons)(_ref2 => {
+  let {
+    active
+  } = _ref2;
   return {
     opacity: active ? 1 : 0,
     path: {
@@ -36,53 +22,40 @@ var LinkIcon = styled(Icons)(function (_ref2) {
     }
   };
 });
-var options = Object.keys(PSEUDO_STATES).sort();
-export var PseudoStateTool = function PseudoStateTool() {
-  var _useGlobals = useGlobals(),
-      _useGlobals2 = _slicedToArray(_useGlobals, 2),
-      pseudo = _useGlobals2[0].pseudo,
-      updateGlobals = _useGlobals2[1];
-
-  var hasSelection = useMemo(function () {
-    return !!pseudo && Object.values(pseudo).includes(true);
-  }, [pseudo]);
-  var getValue = useCallback(function (option) {
-    return pseudo ? pseudo[option] : false;
-  }, [pseudo]);
-  var toggleOption = useCallback(function (option) {
-    return function () {
-      return updateGlobals({
-        pseudo: _objectSpread(_objectSpread({}, pseudo), {}, _defineProperty({}, option, !getValue(option)))
-      });
-    };
-  }, [pseudo, getValue]);
+const options = Object.keys(PSEUDO_STATES).sort();
+export const PseudoStateTool = () => {
+  const [{
+    pseudo
+  }, updateGlobals] = useGlobals();
+  const isActive = useCallback(option => (pseudo === null || pseudo === void 0 ? void 0 : pseudo[option]) === true, [pseudo]);
+  const toggleOption = useCallback(option => () => updateGlobals({
+    pseudo: { ...pseudo,
+      [option]: !isActive(option)
+    }
+  }), [pseudo]);
   return /*#__PURE__*/React.createElement(WithTooltip, {
     placement: "top",
     trigger: "click",
-    tooltip: function tooltip() {
-      return /*#__PURE__*/React.createElement(TooltipLinkList, {
-        links: options.map(function (option) {
-          return {
-            id: option,
-            title: /*#__PURE__*/React.createElement(LinkTitle, {
-              active: getValue(option)
-            }, ":", PSEUDO_STATES[option]),
-            right: /*#__PURE__*/React.createElement(LinkIcon, {
-              icon: "check",
-              width: 12,
-              height: 12,
-              active: getValue(option)
-            }),
-            onClick: toggleOption(option),
-            active: getValue(option)
-          };
-        })
-      });
-    }
+    tooltip: () => /*#__PURE__*/React.createElement(TooltipLinkList, {
+      links: options.map(option => ({
+        id: option,
+        title: /*#__PURE__*/React.createElement(LinkTitle, {
+          active: isActive(option)
+        }, ":", PSEUDO_STATES[option]),
+        right: /*#__PURE__*/React.createElement(LinkIcon, {
+          icon: "check",
+          width: 12,
+          height: 12,
+          active: isActive(option)
+        }),
+        onClick: toggleOption(option),
+        active: isActive(option)
+      }))
+    })
   }, /*#__PURE__*/React.createElement(IconButton, {
     key: "pseudo-state",
     title: "Select CSS pseudo states",
-    active: hasSelection
+    active: options.some(isActive)
   }, /*#__PURE__*/React.createElement(Icons, {
     icon: "button"
   })));

package/dist/esm/constants.js

@@ -1,8 +1,8 @@
-export var ADDON_ID = "storybook/pseudo-states";
-export var TOOL_ID = "".concat(ADDON_ID, "/tool"); // Dynamic pseudo-classes
+export const ADDON_ID = "storybook/pseudo-states";
+export const TOOL_ID = "".concat(ADDON_ID, "/tool"); // Dynamic pseudo-classes
 // @see https://www.w3.org/TR/2018/REC-selectors-3-20181106/#dynamic-pseudos
 
-export var PSEUDO_STATES = {
+export const PSEUDO_STATES = {
   hover: "hover",
   active: "active",
   focusVisible: "focus-visible",

package/dist/esm/preset/manager.js

@@ -1,12 +1,14 @@
 import { addons, types } from "@storybook/addons";
 import { ADDON_ID, TOOL_ID } from "../constants";
 import { PseudoStateTool } from "../PseudoStateTool";
-addons.register(ADDON_ID, function () {
+addons.register(ADDON_ID, () => {
   addons.add(TOOL_ID, {
     type: types.TOOL,
     title: "CSS pseudo states",
-    match: function match(_ref) {
-      var viewMode = _ref.viewMode;
+    match: _ref => {
+      let {
+        viewMode
+      } = _ref;
       return viewMode === "story";
     },
     render: PseudoStateTool

package/dist/esm/preset/preview.js

@@ -1,2 +1,2 @@
 import { withPseudoState } from "../withPseudoState";
-export var decorators = [withPseudoState];
+export const decorators = [withPseudoState];

package/dist/esm/rewriteStyleSheet.js

@@ -0,0 +1,76 @@
+import { PSEUDO_STATES } from "./constants";
+import { splitSelectors } from "./splitSelectors";
+const pseudoStates = Object.values(PSEUDO_STATES);
+const matchOne = new RegExp(":(".concat(pseudoStates.join("|"), ")"));
+const matchAll = new RegExp(":(".concat(pseudoStates.join("|"), ")"), "g");
+const warnings = new Set();
+
+const warnOnce = message => {
+  if (warnings.has(message)) return; // eslint-disable-next-line no-console
+
+  console.warn(message);
+  warnings.add(message);
+};
+
+const rewriteRule = (cssText, selectorText, shadowRoot) => {
+  return cssText.replace(selectorText, splitSelectors(selectorText).flatMap(selector => {
+    if (selector.includes(".pseudo-")) {
+      return [];
+    }
+
+    if (!matchOne.test(selector)) {
+      return [selector];
+    }
+
+    const states = [];
+    const plainSelector = selector.replace(matchAll, (_, state) => {
+      states.push(state);
+      return "";
+    });
+    const classSelector = states.reduce((acc, state) => acc.replace(new RegExp(":".concat(state), "g"), ".pseudo-".concat(state)), selector);
+
+    if (selector.startsWith(":host(") || selector.startsWith("::slotted(")) {
+      return [selector, classSelector];
+    }
+
+    const ancestorSelector = shadowRoot ? ":host(".concat(states.map(s => ".pseudo-".concat(s)).join(""), ") ").concat(plainSelector) : "".concat(states.map(s => ".pseudo-".concat(s)).join(""), " ").concat(plainSelector);
+    return [selector, classSelector, ancestorSelector].filter(selector => !selector.includes(":not()"));
+  }).join(", "));
+}; // Rewrites the style sheet to add alternative selectors for any rule that targets a pseudo state.
+// A sheet can only be rewritten once, and may carry over between stories.
+
+
+export const rewriteStyleSheet = (sheet, shadowRoot, shadowHosts) => {
+  if (sheet.__pseudoStatesRewritten) return;
+  sheet.__pseudoStatesRewritten = true;
+
+  try {
+    let index = 0;
+
+    for (const {
+      cssText,
+      selectorText
+    } of sheet.cssRules) {
+      if (matchOne.test(selectorText)) {
+        const newRule = rewriteRule(cssText, selectorText, shadowRoot);
+        sheet.deleteRule(index);
+        sheet.insertRule(newRule, index);
+        if (shadowRoot) shadowHosts.add(shadowRoot.host);
+      }
+
+      index++;
+
+      if (index > 1000) {
+        warnOnce("Reached maximum of 1000 pseudo selectors per sheet, skipping the rest.");
+        break;
+      }
+    }
+  } catch (e) {
+    if (e.toString().includes("cssRules")) {
+      warnOnce("Can't access cssRules, likely due to CORS restrictions: ".concat(sheet.href));
+    } else {
+      // eslint-disable-next-line no-console
+      console.error(e, sheet.href);
+    }
+  }
+};

package/dist/esm/rewriteStyleSheet.test.js

@@ -0,0 +1,68 @@
+import { rewriteStyleSheet } from "./rewriteStyleSheet";
+
+class Sheet {
+  constructor() {
+    this.__pseudoStatesRewritten = false;
+
+    for (var _len = arguments.length, rules = new Array(_len), _key = 0; _key < _len; _key++) {
+      rules[_key] = arguments[_key];
+    }
+
+    this.cssRules = rules.map(cssText => ({
+      cssText,
+      selectorText: cssText.slice(0, cssText.indexOf(" {"))
+    }));
+  }
+
+  deleteRule(index) {
+    this.cssRules.splice(index, 1);
+  }
+
+  insertRule(cssText, index) {
+    this.cssRules.splice(index, 0, cssText);
+  }
+
+}
+
+describe("rewriteStyleSheet", () => {
+  it("adds alternative selector targeting the element directly", () => {
+    const sheet = new Sheet("a:hover { color: red }");
+    rewriteStyleSheet(sheet);
+    expect(sheet.cssRules[0]).toContain("a.pseudo-hover");
+  });
+  it("adds alternative selector targeting an ancestor", () => {
+    const sheet = new Sheet("a:hover { color: red }");
+    rewriteStyleSheet(sheet);
+    expect(sheet.cssRules[0]).toContain(".pseudo-hover a");
+  });
+  it("adds alternative selector for each pseudo selector", () => {
+    const sheet = new Sheet("a:hover, a:focus { color: red }");
+    rewriteStyleSheet(sheet);
+    expect(sheet.cssRules[0]).toContain("a.pseudo-hover");
+    expect(sheet.cssRules[0]).toContain("a.pseudo-focus");
+    expect(sheet.cssRules[0]).toContain(".pseudo-hover a");
+    expect(sheet.cssRules[0]).toContain(".pseudo-focus a");
+  });
+  it("keeps non-pseudo selectors as-is", () => {
+    const sheet = new Sheet("a.class, a:hover, a:focus, a#id { color: red }");
+    rewriteStyleSheet(sheet);
+    expect(sheet.cssRules[0]).toContain("a.class");
+    expect(sheet.cssRules[0]).toContain("a#id");
+  });
+  it("supports combined pseudo selectors", () => {
+    const sheet = new Sheet("a:hover:focus { color: red }");
+    rewriteStyleSheet(sheet);
+    expect(sheet.cssRules[0]).toContain("a.pseudo-hover.pseudo-focus");
+    expect(sheet.cssRules[0]).toContain(".pseudo-hover.pseudo-focus a");
+  });
+  it('supports ":host"', () => {
+    const sheet = new Sheet(":host(:hover) { color: red }");
+    rewriteStyleSheet(sheet);
+    expect(sheet.cssRules[0]).toEqual(":host(:hover), :host(.pseudo-hover) { color: red }");
+  });
+  it('supports ":not"', () => {
+    const sheet = new Sheet(":not(:hover) { color: red }");
+    rewriteStyleSheet(sheet);
+    expect(sheet.cssRules[0]).toEqual(":not(:hover), :not(.pseudo-hover) { color: red }");
+  });
+});

package/dist/esm/splitSelectors.js

@@ -1,16 +1,14 @@
-var isAtRule = function isAtRule(selector) {
-  return selector.indexOf("@") === 0;
-};
+const isAtRule = selector => selector.indexOf("@") === 0;
 
-export var splitSelectors = function splitSelectors(selectors) {
+export const splitSelectors = selectors => {
   if (isAtRule(selectors)) return [selectors];
-  var result = [];
-  var parentheses = 0;
-  var brackets = 0;
-  var selector = "";
+  let result = [];
+  let parentheses = 0;
+  let brackets = 0;
+  let selector = "";
 
-  for (var i = 0, len = selectors.length; i < len; i++) {
-    var char = selectors[i];
+  for (let i = 0, len = selectors.length; i < len; i++) {
+    const char = selectors[i];
 
     if (char === "(") {
       parentheses += 1;