Support parsing typescript with babel

reactjs · Apr 22, 2019 · ec9e09c · ec9e09c
1 parent ff33fc8
commit ec9e09c
Showing 1 changed file with 36 additions and 26 deletions.
diff --git a/src/babelParser.js b/src/babelParser.js
@@ -8,32 +8,42 @@
  */
 
 const babel = require('@babel/core');
+const path = require('path');
 
-const defaultPlugins = [
-  'jsx',
-  'flow',
-  'asyncGenerators',
-  'bigInt',
-  'classProperties',
-  'classPrivateProperties',
-  'classPrivateMethods',
-  ['decorators', { decoratorsBeforeExport: false }],
-  'doExpressions',
-  'dynamicImport',
-  'exportDefaultFrom',
-  'exportNamespaceFrom',
-  'functionBind',
-  'functionSent',
-  'importMeta',
-  'logicalAssignment',
-  'nullishCoalescingOperator',
-  'numericSeparator',
-  'objectRestSpread',
-  'optionalCatchBinding',
-  'optionalChaining',
-  ['pipelineOperator', { proposal: 'minimal' }],
-  'throwExpressions',
-];
+const TYPESCRIPT_EXTS = {
+  '.ts': true,
+  '.tsx': true,
+};
+
+function getDefaultPlugins(options: BabelOptions) {
+  return [
+    'jsx',
+    TYPESCRIPT_EXTS[path.extname(options.filename || '')]
+      ? 'typescript'
+      : 'flow',
+    'asyncGenerators',
+    'bigInt',
+    'classProperties',
+    'classPrivateProperties',
+    'classPrivateMethods',
+    ['decorators', { decoratorsBeforeExport: false }],
+    'doExpressions',
+    'dynamicImport',
+    'exportDefaultFrom',
+    'exportNamespaceFrom',
+    'functionBind',
+    'functionSent',
+    'importMeta',
+    'logicalAssignment',
+    'nullishCoalescingOperator',
+    'numericSeparator',
+    'objectRestSpread',
+    'optionalCatchBinding',
+    'optionalChaining',
+    ['pipelineOperator', { proposal: 'minimal' }],
+    'throwExpressions',
+  ];
+}
 
 type ParserOptions = {
   plugins?: Array<string | [string, {}]>,
@@ -73,7 +83,7 @@ function buildOptions(
   const partialConfig = babel.loadPartialConfig(babelOptions);
 
   if (!partialConfig.hasFilesystemConfig() && parserOpts.plugins.length === 0) {
-    parserOpts.plugins = [...defaultPlugins];
+    parserOpts.plugins = getDefaultPlugins(babelOptions);
   }
 
   // Recast needs tokens to be in the tree