storybook-addon-customize-antd-theme/dist/esm/components/LessModify.js

a Storybook addon help you visually customize an ant design theme

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; }

/* eslint-disable import/no-webpack-loader-syntax */
import React, { useEffect, useRef } from 'react';
import { Helmet } from 'react-helmet';
import { addons } from '@storybook/addons';
import lessScript from '!!file-loader?modules!../assets/js/less.min.js';
import antdLess from '!!file-loader?modules!../assets/less/antd.less';
import { EVENT_CHANGE_LESS, EVENT_RESET_LESS, EVENT_EXPORT_LESS, TRIGGER_EXPORT_LESS, EVENT_EXPORT_JS, TRIGGER_EXPORT_JS, EVENT_LESS_LOADED } from '../constants';
var modifies = {};
var scriptLoaded = false;
export default function LessModify(_ref) {
  var active = _ref.active;
  var activeRef = useRef(active);

  if (activeRef.current !== active) {
    activeRef.current = active;
  }

  useEffect(function () {
    if (!active) {
      var sheetId = "less:".concat(location.pathname.replace(/[^/]+$/, '').replace(/^\//, '').replace(/\//g, '-')).concat(antdLess.replace(/\.less$/, ''));
      var dom = document.getElementById(sheetId);
      dom && dom.remove();
    } else if (scriptLoaded) {
      window.less.modifyVars(modifies);
    }
  }, [active]);
  useEffect(function () {
    var bus = addons.getChannel();
    bus.on(EVENT_CHANGE_LESS, function (args) {
      var vars = {};

      for (var _i = 0, _Object$entries = Object.entries(args); _i < _Object$entries.length; _i++) {
        var _Object$entries$_i = _slicedToArray(_Object$entries[_i], 2),
            _key = _Object$entries$_i[0],
            value = _Object$entries$_i[1];

        vars["@".concat(_key)] = value;
      }

      Object.assign(modifies, vars);
      activeRef.current && window.less.modifyVars(modifies);
    });
    bus.on(EVENT_RESET_LESS, function (args) {
      var vars = {};
      var allKeys = Object.keys(modifies);

      for (var _i2 = 0, _Object$entries2 = Object.entries(args); _i2 < _Object$entries2.length; _i2++) {
        var _Object$entries2$_i = _slicedToArray(_Object$entries2[_i2], 2),
            _key2 = _Object$entries2$_i[0],
            value = _Object$entries2$_i[1];

        if (allKeys.includes("@".concat(_key2))) {
          vars["@".concat(_key2)] = value;
        }
      }

      Object.assign(modifies, vars);
      activeRef.current && window.less.modifyVars(modifies);
    });
    bus.on(TRIGGER_EXPORT_JS, function () {
      var prefix = /^@/;
      bus.emit(EVENT_EXPORT_JS, Object.fromEntries(Object.entries(modifies).map(function (_ref2) {
        var _ref3 = _slicedToArray(_ref2, 2),
            key = _ref3[0],
            val = _ref3[1];

        return [key.replace(prefix, ''), val];
      })));
    });
    bus.on(TRIGGER_EXPORT_LESS, function () {
      bus.emit(EVENT_EXPORT_LESS, modifies);
    });

    window.onLessScriptLoaded = function () {
      var start;

      function retry(timestamp) {
        if (start === undefined) start = timestamp;
        var elapsed = timestamp - start;

        if (typeof window.less.modifyVars === 'function') {
          bus.emit(EVENT_LESS_LOADED);
          return;
        }

        if (elapsed < 20000) {
          // Stop the animation after 20 seconds
          window.requestAnimationFrame(retry);
        }
      }

      window.requestAnimationFrame(retry);
    };

    scriptLoaded = true;
  }, []); // console.log('LessModify', modifies, antdLess);
  // 实现less自动挂载和检测是否挂载，及手动挂载

  if (scriptLoaded) return null;
  return /*#__PURE__*/React.createElement(Helmet, null, /*#__PURE__*/React.createElement("script", {
    type: "text/javascript"
  }, "\n          less = {\n            env: \"development\",\n            async: false,\n            fileAsync: false,\n            poll: 1000,\n            javascriptEnabled: true,\n            modifyVars: ".concat(JSON.stringify(modifies), ",\n          };\n        ")), /*#__PURE__*/React.createElement("script", {
    src: lessScript,
    type: "text/javascript"
  }), /*#__PURE__*/React.createElement("link", {
    rel: "stylesheet/less",
    type: "text/css",
    href: antdLess
  }), /*#__PURE__*/React.createElement("script", {
    type: "text/javascript"
  }, "\n          onLessScriptLoaded();\n        "));
}