169 lines
		
	
	
		
			5.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			169 lines
		
	
	
		
			5.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| "use strict";
 | |
| 
 | |
| Object.defineProperty(exports, "__esModule", {
 | |
|   value: true
 | |
| });
 | |
| exports.default = void 0;
 | |
| var _core = require("@babel/core");
 | |
| var _helperPluginUtils = require("@babel/helper-plugin-utils");
 | |
| var _template = require("@babel/template");
 | |
| {
 | |
|   var DefineAccessorHelper = _template.default.expression.ast`
 | |
|     function (type, obj, key, fn) {
 | |
|       var desc = { configurable: true, enumerable: true };
 | |
|       desc[type] = fn;
 | |
|       return Object.defineProperty(obj, key, desc);
 | |
|     }
 | |
|   `;
 | |
|   DefineAccessorHelper._compact = true;
 | |
| }
 | |
| var _default = exports.default = (0, _helperPluginUtils.declare)((api, options) => {
 | |
|   var _api$assumption;
 | |
|   api.assertVersion(7);
 | |
|   const setComputedProperties = (_api$assumption = api.assumption("setComputedProperties")) != null ? _api$assumption : options.loose;
 | |
|   const pushComputedProps = setComputedProperties ? pushComputedPropsLoose : pushComputedPropsSpec;
 | |
|   function buildDefineAccessor(state, obj, prop) {
 | |
|     const type = prop.kind;
 | |
|     const key = !prop.computed && _core.types.isIdentifier(prop.key) ? _core.types.stringLiteral(prop.key.name) : prop.key;
 | |
|     const fn = getValue(prop);
 | |
|     {
 | |
|       let helper;
 | |
|       if (state.availableHelper("defineAccessor")) {
 | |
|         helper = state.addHelper("defineAccessor");
 | |
|       } else {
 | |
|         const file = state.file;
 | |
|         helper = file.get("fallbackDefineAccessorHelper");
 | |
|         if (!helper) {
 | |
|           const id = file.scope.generateUidIdentifier("defineAccessor");
 | |
|           file.scope.push({
 | |
|             id,
 | |
|             init: DefineAccessorHelper
 | |
|           });
 | |
|           file.set("fallbackDefineAccessorHelper", helper = id);
 | |
|         }
 | |
|         helper = _core.types.cloneNode(helper);
 | |
|       }
 | |
|       return _core.types.callExpression(helper, [_core.types.stringLiteral(type), obj, key, fn]);
 | |
|     }
 | |
|   }
 | |
|   function getValue(prop) {
 | |
|     if (_core.types.isObjectProperty(prop)) {
 | |
|       return prop.value;
 | |
|     } else if (_core.types.isObjectMethod(prop)) {
 | |
|       return _core.types.functionExpression(null, prop.params, prop.body, prop.generator, prop.async);
 | |
|     }
 | |
|   }
 | |
|   function pushAssign(objId, prop, body) {
 | |
|     body.push(_core.types.expressionStatement(_core.types.assignmentExpression("=", _core.types.memberExpression(_core.types.cloneNode(objId), prop.key, prop.computed || _core.types.isLiteral(prop.key)), getValue(prop))));
 | |
|   }
 | |
|   function pushComputedPropsLoose(info) {
 | |
|     const {
 | |
|       computedProps,
 | |
|       state,
 | |
|       initPropExpression,
 | |
|       objId,
 | |
|       body
 | |
|     } = info;
 | |
|     for (const prop of computedProps) {
 | |
|       if (_core.types.isObjectMethod(prop) && (prop.kind === "get" || prop.kind === "set")) {
 | |
|         if (computedProps.length === 1) {
 | |
|           return buildDefineAccessor(state, initPropExpression, prop);
 | |
|         } else {
 | |
|           body.push(_core.types.expressionStatement(buildDefineAccessor(state, _core.types.cloneNode(objId), prop)));
 | |
|         }
 | |
|       } else {
 | |
|         pushAssign(_core.types.cloneNode(objId), prop, body);
 | |
|       }
 | |
|     }
 | |
|   }
 | |
|   function pushComputedPropsSpec(info) {
 | |
|     const {
 | |
|       objId,
 | |
|       body,
 | |
|       computedProps,
 | |
|       state
 | |
|     } = info;
 | |
|     const CHUNK_LENGTH_CAP = 10;
 | |
|     let currentChunk = null;
 | |
|     const computedPropsChunks = [];
 | |
|     for (const prop of computedProps) {
 | |
|       if (!currentChunk || currentChunk.length === CHUNK_LENGTH_CAP) {
 | |
|         currentChunk = [];
 | |
|         computedPropsChunks.push(currentChunk);
 | |
|       }
 | |
|       currentChunk.push(prop);
 | |
|     }
 | |
|     for (const chunk of computedPropsChunks) {
 | |
|       const single = computedPropsChunks.length === 1;
 | |
|       let node = single ? info.initPropExpression : _core.types.cloneNode(objId);
 | |
|       for (const prop of chunk) {
 | |
|         if (_core.types.isObjectMethod(prop) && (prop.kind === "get" || prop.kind === "set")) {
 | |
|           node = buildDefineAccessor(info.state, node, prop);
 | |
|         } else {
 | |
|           node = _core.types.callExpression(state.addHelper("defineProperty"), [node, _core.types.toComputedKey(prop), getValue(prop)]);
 | |
|         }
 | |
|       }
 | |
|       if (single) return node;
 | |
|       body.push(_core.types.expressionStatement(node));
 | |
|     }
 | |
|   }
 | |
|   return {
 | |
|     name: "transform-computed-properties",
 | |
|     visitor: {
 | |
|       ObjectExpression: {
 | |
|         exit(path, state) {
 | |
|           const {
 | |
|             node,
 | |
|             parent,
 | |
|             scope
 | |
|           } = path;
 | |
|           let hasComputed = false;
 | |
|           for (const prop of node.properties) {
 | |
|             hasComputed = prop.computed === true;
 | |
|             if (hasComputed) break;
 | |
|           }
 | |
|           if (!hasComputed) return;
 | |
|           const initProps = [];
 | |
|           const computedProps = [];
 | |
|           let foundComputed = false;
 | |
|           for (const prop of node.properties) {
 | |
|             if (_core.types.isSpreadElement(prop)) {
 | |
|               continue;
 | |
|             }
 | |
|             if (prop.computed) {
 | |
|               foundComputed = true;
 | |
|             }
 | |
|             if (foundComputed) {
 | |
|               computedProps.push(prop);
 | |
|             } else {
 | |
|               initProps.push(prop);
 | |
|             }
 | |
|           }
 | |
|           const objId = scope.generateUidIdentifierBasedOnNode(parent);
 | |
|           const initPropExpression = _core.types.objectExpression(initProps);
 | |
|           const body = [];
 | |
|           body.push(_core.types.variableDeclaration("var", [_core.types.variableDeclarator(objId, initPropExpression)]));
 | |
|           const single = pushComputedProps({
 | |
|             scope,
 | |
|             objId,
 | |
|             body,
 | |
|             computedProps,
 | |
|             initPropExpression,
 | |
|             state
 | |
|           });
 | |
|           if (single) {
 | |
|             path.replaceWith(single);
 | |
|           } else {
 | |
|             if (setComputedProperties) {
 | |
|               body.push(_core.types.expressionStatement(_core.types.cloneNode(objId)));
 | |
|             }
 | |
|             path.replaceWithMultiple(body);
 | |
|           }
 | |
|         }
 | |
|       }
 | |
|     }
 | |
|   };
 | |
| });
 | |
| 
 | |
| //# sourceMappingURL=index.js.map
 |