Refactor registerExtensions to reduce conditional branches

TypeStrong · May 29, 2019 · 28f22c9 · 28f22c9
1 parent 56fe604
commit 28f22c9
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -436,21 +436,21 @@ function registerExtensions (
   register: Register,
   originalJsHandler: (m: NodeModule, filename: string) => any
 ) {
-  if (!opts.preferTsExts) {
-    return extensions.forEach(ext => registerExtension(ext, ignore, register, originalJsHandler))
+  if (opts.preferTsExts) {
+    extensions.unshift(
+      '.ts',
+      '.tsx',
+      ...Object.keys(require.extensions), // tslint:disable-line
+    )
   }
 
   // @todo a better way with options
-  Array.from(new Set([
-    '.ts',
-    '.tsx',
-    ...Object.keys(require.extensions), // tslint:disable-line
-    ...extensions
-  ])).forEach(ext => {
-    registerExtension(ext, ignore, register, originalJsHandler)
-
-    if (ext in require.extensions) refreshRequireExtension(ext) // tslint:disable-line
-  })
+  Array.from(new Set(extensions))
+       .forEach(ext => {
+         registerExtension(ext, ignore, register, originalJsHandler)
+
+         if (ext in require.extensions) refreshRequireExtension(ext) // tslint:disable-line
+       })
 }
 
 /**