fix(@angular-devkit/build-optimizer): further improvements to enum va…

…lidation
angular · May 9, 2018 · 0eab26d · 0eab26d
1 parent 5426382
commit 0eab26d
Show file tree

Hide file tree

Showing 5 changed files with 75 additions and 59 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/packages/angular_devkit/build_optimizer/src/build-optimizer/build-optimizer.ts b/packages/angular_devkit/build_optimizer/src/build-optimizer/build-optimizer.ts
@@ -16,7 +16,7 @@ import { getImportTslibTransformer, testImportTslib } from '../transforms/import
 import { getPrefixClassesTransformer, testPrefixClasses } from '../transforms/prefix-classes';
 import { getPrefixFunctionsTransformer } from '../transforms/prefix-functions';
 import { getScrubFileTransformer, testScrubFile } from '../transforms/scrub-file';
-import { getWrapEnumsTransformer, testWrapEnums } from '../transforms/wrap-enums';
+import { getWrapEnumsTransformer } from '../transforms/wrap-enums';
 
 
 // Angular packages are known to have no side effects.
@@ -140,9 +140,7 @@ export function buildOptimizer(options: BuildOptimizerOptions): TransformJavascr
     getTransforms.unshift(getImportTslibTransformer);
   }
 
-  if (testWrapEnums(content)) {
-    getTransforms.unshift(getWrapEnumsTransformer);
-  }
+  getTransforms.unshift(getWrapEnumsTransformer);
 
   const transformJavascriptOpts: TransformJavascriptOptions = {
     content: content,

diff --git a/packages/angular_devkit/build_optimizer/src/index.ts b/packages/angular_devkit/build_optimizer/src/index.ts
@@ -18,4 +18,11 @@ export { getImportTslibTransformer, testImportTslib } from './transforms/import-
 export { getPrefixClassesTransformer, testPrefixClasses } from './transforms/prefix-classes';
 export { getPrefixFunctionsTransformer } from './transforms/prefix-functions';
 export { getScrubFileTransformer, testScrubFile } from './transforms/scrub-file';
-export { getWrapEnumsTransformer, testWrapEnums } from './transforms/wrap-enums';
+export { getWrapEnumsTransformer } from './transforms/wrap-enums';
+
+/**
+ * @deprecated since version 6.0
+ */
+export function testWrapEnums(_content: string) {
+  return true;
+}
diff --git a/packages/angular_devkit/build_optimizer/src/transforms/wrap-enums.ts b/packages/angular_devkit/build_optimizer/src/transforms/wrap-enums.ts
@@ -6,13 +6,6 @@
  * found in the LICENSE file at https://angular.io/license
  */
 import * as ts from 'typescript';
-import { drilldownNodes } from '../helpers/ast-utils';
-
-
-export function testWrapEnums(content: string) {
-  // TODO: remove this method, it's not doing anything anymore.
-  return true;
-}
 
 function isBlockLike(node: ts.Node): node is ts.BlockLike {
   return node.kind === ts.SyntaxKind.Block
@@ -166,73 +159,96 @@ function visitBlockStatements(
 
 // TS 2.3 enums have statements that are inside a IIFE.
 function findTs2_3EnumIife(name: string, statement: ts.Statement): ts.CallExpression | null {
-  if (!ts.isExpressionStatement(statement) || !ts.isCallExpression(statement.expression)) {
+  if (!ts.isExpressionStatement(statement)) {
+    return null;
+  }
+
+  let expression = statement.expression;
+  while (ts.isParenthesizedExpression(expression)) {
+    expression = expression.expression;
+  }
+
+  if (!expression || !ts.isCallExpression(expression) || expression.arguments.length !== 1) {
+    return null;
+  }
+
+  const callExpression = expression;
+
+  const argument = expression.arguments[0];
+  if (!ts.isBinaryExpression(argument)
+      || argument.operatorToken.kind !== ts.SyntaxKind.BarBarToken) {
+    return null;
+  }
+
+  if (!ts.isIdentifier(argument.left) || argument.left.text !== name) {
+    return null;
+  }
+
+  expression = expression.expression;
+  while (ts.isParenthesizedExpression(expression)) {
+    expression = expression.expression;
+  }
+
+  if (!expression || !ts.isFunctionExpression(expression) || expression.parameters.length !== 1) {
     return null;
   }
 
-  const funcExpr = drilldownNodes<ts.FunctionExpression>(statement,
-    [
-      { prop: null, kind: ts.SyntaxKind.ExpressionStatement },
-      { prop: 'expression', kind: ts.SyntaxKind.CallExpression },
-      { prop: 'expression', kind: ts.SyntaxKind.ParenthesizedExpression },
-      { prop: 'expression', kind: ts.SyntaxKind.FunctionExpression },
-    ]);
-
-  if (funcExpr === null) { return null; }
-
-  if (!(
-    funcExpr.parameters.length === 1
-    && funcExpr.parameters[0].name.kind === ts.SyntaxKind.Identifier
-    && (funcExpr.parameters[0].name as ts.Identifier).text === name
-  )) {
+  const parameter = expression.parameters[0];
+  if (!ts.isIdentifier(parameter.name) || parameter.name.text !== name) {
     return null;
   }
 
   // In TS 2.3 enums, the IIFE contains only expressions with a certain format.
   // If we find any that is different, we ignore the whole thing.
-  for (const innerStmt of funcExpr.body.statements) {
+  for (let bodyIndex = 0; bodyIndex < expression.body.statements.length; ++bodyIndex) {
+    const bodyStatement = expression.body.statements[bodyIndex];
 
-    const innerBinExpr = drilldownNodes<ts.BinaryExpression>(innerStmt,
-      [
-        { prop: null, kind: ts.SyntaxKind.ExpressionStatement },
-        { prop: 'expression', kind: ts.SyntaxKind.BinaryExpression },
-      ]);
+    if (!ts.isExpressionStatement(bodyStatement) || !bodyStatement.expression) {
+      return null;
+    }
 
-    if (innerBinExpr === null) { return null; }
+    if (!ts.isBinaryExpression(bodyStatement.expression)
+        || bodyStatement.expression.operatorToken.kind !== ts.SyntaxKind.FirstAssignment) {
+      return null;
+    }
 
-    if (!(innerBinExpr.operatorToken.kind === ts.SyntaxKind.FirstAssignment
-        && innerBinExpr.left.kind === ts.SyntaxKind.ElementAccessExpression)) {
+    const assignment = bodyStatement.expression.left;
+    const value = bodyStatement.expression.right;
+    if (!ts.isElementAccessExpression(assignment) || !ts.isStringLiteral(value)) {
       return null;
     }
 
-    const innerElemAcc = innerBinExpr.left as ts.ElementAccessExpression;
+    if (!ts.isIdentifier(assignment.expression) || assignment.expression.text !== name) {
+      return null;
+    }
 
-    if (!(
-      innerElemAcc.expression.kind === ts.SyntaxKind.Identifier
-      && (innerElemAcc.expression as ts.Identifier).text === name
-      && innerElemAcc.argumentExpression
-      && innerElemAcc.argumentExpression.kind === ts.SyntaxKind.BinaryExpression
-    )) {
+    const memberArgument = assignment.argumentExpression;
+    if (!memberArgument || !ts.isBinaryExpression(memberArgument)
+        || memberArgument.operatorToken.kind !== ts.SyntaxKind.FirstAssignment) {
       return null;
     }
 
-    const innerArgBinExpr = innerElemAcc.argumentExpression as ts.BinaryExpression;
 
-    if (innerArgBinExpr.left.kind !== ts.SyntaxKind.ElementAccessExpression) {
+    if (!ts.isElementAccessExpression(memberArgument.left)) {
+      return null;
+    }
+
+    if (!ts.isIdentifier(memberArgument.left.expression)
+        || memberArgument.left.expression.text !== name) {
       return null;
     }
 
-    const innerArgElemAcc = innerArgBinExpr.left as ts.ElementAccessExpression;
+    if (!memberArgument.left.argumentExpression
+        || !ts.isStringLiteral(memberArgument.left.argumentExpression)) {
+      return null;
+    }
 
-    if (!(
-      innerArgElemAcc.expression.kind === ts.SyntaxKind.Identifier
-      && (innerArgElemAcc.expression as ts.Identifier).text === name
-    )) {
+    if (memberArgument.left.argumentExpression.text !== value.text) {
       return null;
     }
   }
 
-  return statement.expression;
+  return callExpression;
 }
 
 // TS 2.2 enums have statements after the variable declaration, with index statements followed

diff --git a/packages/angular_devkit/build_optimizer/src/transforms/wrap-enums_spec.ts b/packages/angular_devkit/build_optimizer/src/transforms/wrap-enums_spec.ts
@@ -8,7 +8,7 @@
 // tslint:disable-next-line:no-implicit-dependencies
 import { tags } from '@angular-devkit/core';
 import { transformJavascript } from '../helpers/transform-javascript';
-import { getWrapEnumsTransformer, testWrapEnums } from './wrap-enums';
+import { getWrapEnumsTransformer } from './wrap-enums';
 
 
 const transform = (content: string) => transformJavascript(
@@ -34,7 +34,6 @@ describe('wrap-enums', () => {
       }());
     `;
 
-    expect(testWrapEnums(input)).toBeTruthy();
     expect(tags.oneLine`${transform(input)}`).toEqual(tags.oneLine`${output}`);
   });
 
@@ -54,7 +53,6 @@ describe('wrap-enums', () => {
       })({});
     `;
 
-    expect(testWrapEnums(input)).toBeTruthy();
     expect(tags.oneLine`${transform(input)}`).toEqual(tags.oneLine`${output}`);
   });
 
@@ -78,7 +76,6 @@ describe('wrap-enums', () => {
       })({});
     `;
 
-    expect(testWrapEnums(input)).toBeTruthy();
     expect(tags.oneLine`${transform(input)}`).toEqual(tags.oneLine`${output}`);
   });
 
@@ -112,7 +109,6 @@ describe('wrap-enums', () => {
       }());
     `;
 
-    expect(testWrapEnums(input)).toBeTruthy();
     expect(tags.oneLine`${transform(input)}`).toEqual(tags.oneLine`${output}`);
   });
 
@@ -162,7 +158,6 @@ describe('wrap-enums', () => {
       })({});
     `;
 
-    expect(testWrapEnums(input)).toBeTruthy();
     expect(tags.oneLine`${transform(input)}`).toEqual(tags.oneLine`${output}`);
   });
 });