storybook-addon-customize-antd-theme
Version:
a Storybook addon help you visually customize an ant design theme
232 lines (205 loc) • 9.66 kB
JavaScript
function _typeof(obj) { "@babel/helpers - typeof"; if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }
function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); if (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 = arguments[i] != null ? arguments[i] : {}; if (i % 2) { ownKeys(Object(source), true).forEach(function (key) { _defineProperty(target, key, source[key]); }); } else if (Object.getOwnPropertyDescriptors) { Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)); } else { 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, { useState, useMemo, useEffect, useCallback, useRef } from 'react';
import { useParameter } from '@storybook/api';
import TrieSearch from 'trie-search';
import { AddonPanel, Button } from '@storybook/components';
import { addons } from '@storybook/addons';
import styled from '@emotion/styled';
import { pick } from 'lodash';
import Input from 'antd/lib/input';
import 'antd/lib/input/style/css';
import { EVENT_CHANGE_LESS, EVENT_RESET_LESS, EVENT_EXPORT_LESS, TRIGGER_EXPORT_LESS, EVENT_EXPORT_JS, TRIGGER_EXPORT_JS, EVENT_LESS_LOADED, PARAM_KEY } from '../constants';
import { LessArgGenerator } from '../lib/utils';
import antdLessValue from '../lib/antd-helper/antdLessValue';
import ArgsTable from './ArgsTable';
var Search = Input.Search;
var Tool = styled.div(function (props) {
return {
display: 'flex',
marginTop: '10px',
marginBottom: '-10px'
};
});
var InputFile = styled.label(function (props) {
return {
display: 'inline-block',
cursor: 'pointer',
'input[type="file"]': {
display: 'none'
}
};
});
var jsExtReg = /\.js$/;
var lineReg = /[\r\n]+/;
var jsVarReg = /^\s*['"@]*([^'"\s:]+)['"]?\s*:\s*['"](.+)['"]\s*,?\s*/;
var lessVarReg = /^\s*@([^\s:]+)\s*:\s*(\S.+)\s*;\s*/;
export default function Panel(props) {
var allLessArgs = useMemo(function () {
return new LessArgGenerator(antdLessValue).hints;
}, []);
var _useState = useState(allLessArgs),
_useState2 = _slicedToArray(_useState, 2),
lessArgs = _useState2[0],
setLessArgs = _useState2[1];
var _useState3 = useState(_objectSpread({}, antdLessValue)),
_useState4 = _slicedToArray(_useState3, 2),
argsValues = _useState4[0],
setArgs = _useState4[1];
var _useState5 = useState(function () {
return addons.getChannel();
}),
_useState6 = _slicedToArray(_useState5, 1),
bus = _useState6[0];
var paramsRef = useRef({});
var params = useParameter(PARAM_KEY);
paramsRef.current = params; // console.log('Panel', paramsRef.current, props, lessArgs, argsValues);
var handleUpdateArgs = useCallback(function () {
for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
args[_key] = arguments[_key];
}
// console.log('handleUpdateArgs', args);
setArgs(function (argsValues) {
return _objectSpread(_objectSpread({}, argsValues), args[0]);
});
bus.emit(EVENT_CHANGE_LESS, args[0]);
}, []);
useEffect(function () {
bus.on(EVENT_EXPORT_JS, function (vars) {
var a = document.createElement('a');
var blob = new Blob(['module.exports = ', JSON.stringify(vars, null, 2), ';']);
a.href = URL.createObjectURL(blob);
a.download = 'antd-theme.js';
a.click();
});
bus.on(EVENT_EXPORT_LESS, function (vars) {
var a = document.createElement('a');
var blob = new Blob(Object.keys(vars).map(function (key) {
return "".concat(key, ": ").concat(vars[key], ";\n");
}));
a.href = URL.createObjectURL(blob);
a.download = 'antd-theme.less';
a.click();
});
bus.on(EVENT_LESS_LOADED, function (vars) {
var _ref = paramsRef.current || {},
modifyVars = _ref.modifyVars;
if (_typeof(modifyVars) === 'object' && Object.keys(modifyVars).length) {
setArgs(function (argsValues) {
return _objectSpread(_objectSpread({}, argsValues), modifyVars);
});
bus.emit(EVENT_CHANGE_LESS, modifyVars);
} else if (typeof modifyVars === 'string') {
var lines = modifyVars.split(lineReg);
var args = {};
lines.forEach(function (line) {
var mm = line.match(lessVarReg);
if (mm) {
args[mm[1]] = mm[2];
}
});
setArgs(function (argsValues) {
return _objectSpread(_objectSpread({}, argsValues), args);
});
bus.emit(EVENT_CHANGE_LESS, args);
}
});
}, []); // 实现前缀树搜索
var handleSearch = function handleSearch(value) {
if (!value) {
return setLessArgs(allLessArgs);
}
var trie = new TrieSearch(['name', 'desc', 'category']);
trie.addAll(Object.values(allLessArgs));
setLessArgs(pick(allLessArgs, trie.search(value).map(function (arg) {
return arg.name;
})));
};
var handleReset = function handleReset() {
setArgs(_objectSpread({}, antdLessValue));
bus.emit(EVENT_RESET_LESS, _objectSpread({}, antdLessValue));
};
var onInputClick = function onInputClick(event) {
var element = event.target;
element.value = '';
};
var readFile = function readFile(event) {
var fr = new FileReader();
var file = event.target.files[0];
fr.onload = function () {
var lines = fr.result.split(lineReg);
var isJs = jsExtReg.test(file.name);
var args = {};
lines.forEach(function (line) {
var mm = line.match(isJs ? jsVarReg : lessVarReg);
if (mm) {
args[mm[1]] = mm[2];
}
});
setArgs(function (argsValues) {
return _objectSpread(_objectSpread({}, argsValues), args);
});
bus.emit(EVENT_CHANGE_LESS, args);
};
fr.readAsText(file);
};
return /*#__PURE__*/React.createElement(AddonPanel, props, /*#__PURE__*/React.createElement(Tool, null, /*#__PURE__*/React.createElement(Search, {
placeholder: "input search text",
onSearch: handleSearch,
style: {
width: 200,
marginLeft: 16
},
allowClear: true
}), /*#__PURE__*/React.createElement(Button, {
small: true,
secondary: true,
onClick: function onClick() {
bus.emit(TRIGGER_EXPORT_JS);
},
style: {
marginLeft: 'auto'
}
}, "export js"), /*#__PURE__*/React.createElement(Button, {
small: true,
secondary: true,
onClick: function onClick() {
bus.emit(TRIGGER_EXPORT_LESS);
},
style: {
marginLeft: '8px'
}
}, "export less"), /*#__PURE__*/React.createElement(Button, {
small: true,
secondary: true,
style: {
marginLeft: '8px'
}
}, /*#__PURE__*/React.createElement(InputFile, null, /*#__PURE__*/React.createElement("input", {
type: "file",
accept: ".js, .less",
onChange: readFile,
onClick: onInputClick
}), "import less")), /*#__PURE__*/React.createElement(Button, {
small: true,
gray: true,
style: {
marginLeft: '8px',
marginRight: '16px'
},
onClick: handleReset
}, "reset")), /*#__PURE__*/React.createElement(ArgsTable, {
key: "customize-antd-theme",
rows: lessArgs,
args: argsValues,
updateArgs: handleUpdateArgs
}));
}