You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					175 lines
				
				4.8 KiB
			
		
		
			
		
	
	
					175 lines
				
				4.8 KiB
			| 
											3 years ago
										 | "use strict"; | ||
|  | 
 | ||
|  | Object.defineProperty(exports, "__esModule", { | ||
|  |   value: true | ||
|  | }); | ||
|  | exports.default = transformWithoutHelper; | ||
|  | 
 | ||
|  | var _core = require("@babel/core"); | ||
|  | 
 | ||
|  | function transformWithoutHelper(loose, path, state) { | ||
|  |   const pushComputedProps = loose ? pushComputedPropsLoose : pushComputedPropsSpec; | ||
|  |   const { | ||
|  |     node | ||
|  |   } = path; | ||
|  |   const build = pushComputedProps(path, state); | ||
|  |   const declar = build.declar; | ||
|  |   const loop = build.loop; | ||
|  |   const block = loop.body; | ||
|  |   path.ensureBlock(); | ||
|  | 
 | ||
|  |   if (declar) { | ||
|  |     block.body.push(declar); | ||
|  |   } | ||
|  | 
 | ||
|  |   block.body.push(...node.body.body); | ||
|  | 
 | ||
|  |   _core.types.inherits(loop, node); | ||
|  | 
 | ||
|  |   _core.types.inherits(loop.body, node.body); | ||
|  | 
 | ||
|  |   if (build.replaceParent) { | ||
|  |     path.parentPath.replaceWithMultiple(build.node); | ||
|  |     path.remove(); | ||
|  |   } else { | ||
|  |     path.replaceWithMultiple(build.node); | ||
|  |   } | ||
|  | } | ||
|  | 
 | ||
|  | const buildForOfLoose = _core.template.statement(`
 | ||
|  |   for (var LOOP_OBJECT = OBJECT, | ||
|  |           IS_ARRAY = Array.isArray(LOOP_OBJECT), | ||
|  |           INDEX = 0, | ||
|  |           LOOP_OBJECT = IS_ARRAY ? LOOP_OBJECT : LOOP_OBJECT[Symbol.iterator]();;) { | ||
|  |     INTERMEDIATE; | ||
|  |     if (IS_ARRAY) { | ||
|  |       if (INDEX >= LOOP_OBJECT.length) break; | ||
|  |       ID = LOOP_OBJECT[INDEX++]; | ||
|  |     } else { | ||
|  |       INDEX = LOOP_OBJECT.next(); | ||
|  |       if (INDEX.done) break; | ||
|  |       ID = INDEX.value; | ||
|  |     } | ||
|  |   } | ||
|  | `);
 | ||
|  | 
 | ||
|  | const buildForOf = _core.template.statements(`
 | ||
|  |   var ITERATOR_COMPLETION = true; | ||
|  |   var ITERATOR_HAD_ERROR_KEY = false; | ||
|  |   var ITERATOR_ERROR_KEY = undefined; | ||
|  |   try { | ||
|  |     for ( | ||
|  |       var ITERATOR_KEY = OBJECT[Symbol.iterator](), STEP_KEY; | ||
|  |       !(ITERATOR_COMPLETION = (STEP_KEY = ITERATOR_KEY.next()).done); | ||
|  |       ITERATOR_COMPLETION = true | ||
|  |     ) {} | ||
|  |   } catch (err) { | ||
|  |     ITERATOR_HAD_ERROR_KEY = true; | ||
|  |     ITERATOR_ERROR_KEY = err; | ||
|  |   } finally { | ||
|  |     try { | ||
|  |       if (!ITERATOR_COMPLETION && ITERATOR_KEY.return != null) { | ||
|  |         ITERATOR_KEY.return(); | ||
|  |       } | ||
|  |     } finally { | ||
|  |       if (ITERATOR_HAD_ERROR_KEY) { | ||
|  |         throw ITERATOR_ERROR_KEY; | ||
|  |       } | ||
|  |     } | ||
|  |   } | ||
|  | `);
 | ||
|  | 
 | ||
|  | function pushComputedPropsLoose(path, state) { | ||
|  |   const { | ||
|  |     node, | ||
|  |     scope, | ||
|  |     parent | ||
|  |   } = path; | ||
|  |   const { | ||
|  |     left | ||
|  |   } = node; | ||
|  |   let declar, id, intermediate; | ||
|  | 
 | ||
|  |   if (_core.types.isIdentifier(left) || _core.types.isPattern(left) || _core.types.isMemberExpression(left)) { | ||
|  |     id = left; | ||
|  |     intermediate = null; | ||
|  |   } else if (_core.types.isVariableDeclaration(left)) { | ||
|  |     id = scope.generateUidIdentifier("ref"); | ||
|  |     declar = _core.types.variableDeclaration(left.kind, [_core.types.variableDeclarator(left.declarations[0].id, _core.types.identifier(id.name))]); | ||
|  |     intermediate = _core.types.variableDeclaration("var", [_core.types.variableDeclarator(_core.types.identifier(id.name))]); | ||
|  |   } else { | ||
|  |     throw state.buildCodeFrameError(left, `Unknown node type ${left.type} in ForStatement`); | ||
|  |   } | ||
|  | 
 | ||
|  |   const iteratorKey = scope.generateUidIdentifier("iterator"); | ||
|  |   const isArrayKey = scope.generateUidIdentifier("isArray"); | ||
|  |   const loop = buildForOfLoose({ | ||
|  |     LOOP_OBJECT: iteratorKey, | ||
|  |     IS_ARRAY: isArrayKey, | ||
|  |     OBJECT: node.right, | ||
|  |     INDEX: scope.generateUidIdentifier("i"), | ||
|  |     ID: id, | ||
|  |     INTERMEDIATE: intermediate | ||
|  |   }); | ||
|  | 
 | ||
|  |   const isLabeledParent = _core.types.isLabeledStatement(parent); | ||
|  | 
 | ||
|  |   let labeled; | ||
|  | 
 | ||
|  |   if (isLabeledParent) { | ||
|  |     labeled = _core.types.labeledStatement(parent.label, loop); | ||
|  |   } | ||
|  | 
 | ||
|  |   return { | ||
|  |     replaceParent: isLabeledParent, | ||
|  |     declar: declar, | ||
|  |     node: labeled || loop, | ||
|  |     loop: loop | ||
|  |   }; | ||
|  | } | ||
|  | 
 | ||
|  | function pushComputedPropsSpec(path, state) { | ||
|  |   const { | ||
|  |     node, | ||
|  |     scope, | ||
|  |     parent | ||
|  |   } = path; | ||
|  |   const left = node.left; | ||
|  |   let declar; | ||
|  |   const stepKey = scope.generateUid("step"); | ||
|  | 
 | ||
|  |   const stepValue = _core.types.memberExpression(_core.types.identifier(stepKey), _core.types.identifier("value")); | ||
|  | 
 | ||
|  |   if (_core.types.isIdentifier(left) || _core.types.isPattern(left) || _core.types.isMemberExpression(left)) { | ||
|  |     declar = _core.types.expressionStatement(_core.types.assignmentExpression("=", left, stepValue)); | ||
|  |   } else if (_core.types.isVariableDeclaration(left)) { | ||
|  |     declar = _core.types.variableDeclaration(left.kind, [_core.types.variableDeclarator(left.declarations[0].id, stepValue)]); | ||
|  |   } else { | ||
|  |     throw state.buildCodeFrameError(left, `Unknown node type ${left.type} in ForStatement`); | ||
|  |   } | ||
|  | 
 | ||
|  |   const template = buildForOf({ | ||
|  |     ITERATOR_HAD_ERROR_KEY: scope.generateUidIdentifier("didIteratorError"), | ||
|  |     ITERATOR_COMPLETION: scope.generateUidIdentifier("iteratorNormalCompletion"), | ||
|  |     ITERATOR_ERROR_KEY: scope.generateUidIdentifier("iteratorError"), | ||
|  |     ITERATOR_KEY: scope.generateUidIdentifier("iterator"), | ||
|  |     STEP_KEY: _core.types.identifier(stepKey), | ||
|  |     OBJECT: node.right | ||
|  |   }); | ||
|  | 
 | ||
|  |   const isLabeledParent = _core.types.isLabeledStatement(parent); | ||
|  | 
 | ||
|  |   const tryBody = template[3].block.body; | ||
|  |   const loop = tryBody[0]; | ||
|  | 
 | ||
|  |   if (isLabeledParent) { | ||
|  |     tryBody[0] = _core.types.labeledStatement(parent.label, loop); | ||
|  |   } | ||
|  | 
 | ||
|  |   return { | ||
|  |     replaceParent: isLabeledParent, | ||
|  |     declar: declar, | ||
|  |     loop: loop, | ||
|  |     node: template | ||
|  |   }; | ||
|  | } |