| /** | 
|  * Copyright (c) 2014-present, Facebook, Inc. | 
|  * | 
|  * This source code is licensed under the MIT license found in the | 
|  * LICENSE file in the root directory of this source tree. | 
|  */ | 
|   | 
| import * as util from "./util"; | 
|   | 
| // this function converts a shorthand object generator method into a normal | 
| // (non-shorthand) object property which is a generator function expression. for | 
| // example, this: | 
| // | 
| //  var foo = { | 
| //    *bar(baz) { return 5; } | 
| //  } | 
| // | 
| // should be replaced with: | 
| // | 
| //  var foo = { | 
| //    bar: function*(baz) { return 5; } | 
| //  } | 
| // | 
| // to do this, it clones the parameter array and the body of the object generator | 
| // method into a new FunctionExpression. | 
| // | 
| // this method can be passed any Function AST node path, and it will return | 
| // either: | 
| //   a) the path that was passed in (iff the path did not need to be replaced) or | 
| //   b) the path of the new FunctionExpression that was created as a replacement | 
| //     (iff the path did need to be replaced) | 
| // | 
| // In either case, though, the caller can count on the fact that the return value | 
| // is a Function AST node path. | 
| // | 
| // If this function is called with an AST node path that is not a Function (or with an | 
| // argument that isn't an AST node path), it will throw an error. | 
| export default function replaceShorthandObjectMethod(path) { | 
|   const t = util.getTypes(); | 
|   | 
|   if (!path.node || !t.isFunction(path.node)) { | 
|     throw new Error("replaceShorthandObjectMethod can only be called on Function AST node paths."); | 
|   } | 
|   | 
|   // this function only replaces shorthand object methods (called ObjectMethod | 
|   // in Babel-speak). | 
|   if (!t.isObjectMethod(path.node)) { | 
|     return path; | 
|   } | 
|   | 
|   // this function only replaces generators. | 
|   if (!path.node.generator) { | 
|     return path; | 
|   } | 
|   | 
|   const parameters = path.node.params.map(function (param) { | 
|     return t.cloneDeep(param); | 
|   }) | 
|   | 
|   const functionExpression = t.functionExpression( | 
|     null, // id | 
|     parameters, // params | 
|     t.cloneDeep(path.node.body), // body | 
|     path.node.generator, | 
|     path.node.async | 
|   ); | 
|   | 
|   util.replaceWithOrRemove(path, | 
|     t.objectProperty( | 
|       t.cloneDeep(path.node.key), // key | 
|       functionExpression, //value | 
|       path.node.computed, // computed | 
|       false // shorthand | 
|     ) | 
|   ); | 
|   | 
|   // path now refers to the ObjectProperty AST node path, but we want to return a | 
|   // Function AST node path for the function expression we created. we know that | 
|   // the FunctionExpression we just created is the value of the ObjectProperty, | 
|   // so return the "value" path off of this path. | 
|   return path.get("value"); | 
| } |