packager: buck assets: wrap in __d

Reviewed By: davidaurelio

Differential Revision: D5003464

fbshipit-source-id: 59800dded389d133394b60530eb97642d67a5c89

packager/src/ModuleGraph/worker/JsFileWrapping.js

@@ -0,0 +1,52 @@
+/**
+ * Copyright (c) 2016-present, Facebook, Inc.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree. An additional grant
+ * of patent rights can be found in the PATENTS file in the same directory.
+ *
+ * @flow
+ */
+
+'use strict';
+
+const babel = require('babel-core');
+
+const MODULE_FACTORY_PARAMETERS = ['global', 'require', 'module', 'exports'];
+const POLYFILL_FACTORY_PARAMETERS = ['global'];
+
+function wrapModule(fileAst: Object, dependencyMapName: string): Object {
+  const t = babel.types;
+  const params = MODULE_FACTORY_PARAMETERS.concat(dependencyMapName);
+  const factory = functionFromProgram(fileAst.program, params);
+  const def = t.callExpression(t.identifier('__d'), [factory]);
+  return t.file(t.program([t.expressionStatement(def)]));
+}
+
+function wrapPolyfill(fileAst: Object): Object {
+  const t = babel.types;
+  const factory = functionFromProgram(fileAst.program, POLYFILL_FACTORY_PARAMETERS);
+  const iife = t.callExpression(factory, [t.identifier('this')]);
+  return t.file(t.program([t.expressionStatement(iife)]));
+}
+
+function functionFromProgram(program: Object, parameters: Array<string>): Object {
+  const t = babel.types;
+  return t.functionExpression(
+    t.identifier(''),
+    parameters.map(makeIdentifier),
+    t.blockStatement(program.body, program.directives),
+  );
+}
+
+function makeIdentifier(name: string): Object {
+  return babel.types.identifier(name);
+}
+
+module.exports = {
+  MODULE_FACTORY_PARAMETERS,
+  POLYFILL_FACTORY_PARAMETERS,
+  wrapModule,
+  wrapPolyfill,
+};

packager/src/ModuleGraph/worker/transform-module.js

@@ -10,7 +10,8 @@
  */
 'use strict';
 
-const babel = require('babel-core');
+const JsFileWrapping = require('./JsFileWrapping');
+
 const collectDependencies = require('./collect-dependencies');
 const defaults = require('../../../defaults');
 const docblock = require('../../node-haste/DependencyGraph/docblock');
@@ -38,8 +39,6 @@ export type TransformOptions = {|
 |};
 
 const defaultVariants = {default: {}};
-const moduleFactoryParameters = ['global', 'require', 'module', 'exports'];
-const polyfillFactoryParameters = ['global'];
 
 const ASSET_EXTENSIONS = new Set(defaults.assetExts);
 
@@ -109,7 +108,7 @@ function transformJSON(json, options, callback) {
   const value = JSON.parse(json);
   const {filename} = options;
   const code =
-    `__d(function(${moduleFactoryParameters.join(', ')}) { module.exports = \n${
+    `__d(function(${JsFileWrapping.MODULE_FACTORY_PARAMETERS.join(', ')}) { module.exports = \n${
       json
     }\n});`;
 
@@ -162,42 +161,14 @@ function makeResult(ast, filename, sourceCode, isPolyfill = false) {
   let dependencies, dependencyMapName, file;
   if (isPolyfill) {
     dependencies = [];
-    file = wrapPolyfill(ast);
+    file = JsFileWrapping.wrapPolyfill(ast);
   } else {
     ({dependencies, dependencyMapName} = collectDependencies(ast));
-    file = wrapModule(ast, dependencyMapName);
+    file = JsFileWrapping.wrapModule(ast, dependencyMapName);
   }
 
   const gen = generate(file, filename, sourceCode);
   return {code: gen.code, map: gen.map, dependencies, dependencyMapName};
 }
 
-function wrapModule(file, dependencyMapName) {
-  const t = babel.types;
-  const params = moduleFactoryParameters.concat(dependencyMapName);
-  const factory = functionFromProgram(file.program, params);
-  const def = t.callExpression(t.identifier('__d'), [factory]);
-  return t.file(t.program([t.expressionStatement(def)]));
-}
-
-function wrapPolyfill(file) {
-  const t = babel.types;
-  const factory = functionFromProgram(file.program, polyfillFactoryParameters);
-  const iife = t.callExpression(factory, [t.identifier('this')]);
-  return t.file(t.program([t.expressionStatement(iife)]));
-}
-
-function functionFromProgram(program, parameters) {
-  const t = babel.types;
-  return t.functionExpression(
-    t.identifier(''),
-    parameters.map(makeIdentifier),
-    t.blockStatement(program.body, program.directives),
-  );
-}
-
-function makeIdentifier(name) {
-  return babel.types.identifier(name);
-}
-
 module.exports = transformModule;