From 8e3a06c104ab96065933a4ca8c229b4a4e1d5fd5 Mon Sep 17 00:00:00 2001
From: Haoqun Jiang <haoqunjiang@gmail.com>
Date: Mon, 25 Nov 2019 21:21:18 +0800
Subject: [PATCH] fix: support `parser` option for codemods, and enable ts
 parsing by default

fixes #4861
---
 packages/@vue/cli/__tests__/Generator.spec.js | 18 +++++++++++
 packages/@vue/cli/lib/util/runCodemod.js      | 30 +++++++++++++++++--
 2 files changed, 45 insertions(+), 3 deletions(-)

diff --git a/packages/@vue/cli/__tests__/Generator.spec.js b/packages/@vue/cli/__tests__/Generator.spec.js
index 5fdc43b349..b2bca9c33a 100644
--- a/packages/@vue/cli/__tests__/Generator.spec.js
+++ b/packages/@vue/cli/__tests__/Generator.spec.js
@@ -23,6 +23,7 @@ new Vue({
 }).$mount('#app')
 `.trim())
 fs.writeFileSync(path.resolve(templateDir, 'empty-entry.js'), `;`)
+fs.writeFileSync(path.resolve(templateDir, 'main.ts'), `const a: string = 'hello';`)
 fs.writeFileSync(path.resolve(templateDir, 'hello.vue'), `
 <template>
   <p>Hello, {{ msg }}</p>
@@ -530,6 +531,23 @@ test('api: injectImports to empty file', async () => {
   expect(fs.readFileSync('/main.js', 'utf-8')).toMatch(/import foo from 'foo'\r?\nimport bar from 'bar'/)
 })
 
+test('api: injectImports to typescript file', async () => {
+  const generator = new Generator('/', { plugins: [
+    {
+      id: 'test',
+      apply: api => {
+        api.injectImports('main.ts', `import foo from 'foo'`)
+        api.render({
+          'main.ts': path.join(templateDir, 'main.ts')
+        })
+      }
+    }
+  ] })
+
+  await generator.generate()
+  expect(fs.readFileSync('/main.ts', 'utf-8')).toMatch(/import foo from 'foo'/)
+})
+
 test('api: addEntryDuplicateImport', async () => {
   const generator = new Generator('/', { plugins: [
     {
diff --git a/packages/@vue/cli/lib/util/runCodemod.js b/packages/@vue/cli/lib/util/runCodemod.js
index 22fbc82cf6..40ca92c28c 100644
--- a/packages/@vue/cli/lib/util/runCodemod.js
+++ b/packages/@vue/cli/lib/util/runCodemod.js
@@ -1,6 +1,30 @@
-const jscodeshift = require('jscodeshift')
 const adapt = require('vue-jscodeshift-adapter')
+let jscodeshift = require('jscodeshift')
 
-module.exports = function runCodemod (transform, fileInfo, options) {
-  return adapt(transform)(fileInfo, { jscodeshift }, options || {})
+module.exports = function runCodemod (transformModule, fileInfo, options = {}) {
+  const transform = typeof transformModule.default === 'function'
+    ? transformModule.default
+    : transformModule
+
+  let parser = transformModule.parser || options.parser
+  if (!parser) {
+    if (fileInfo.path.endsWith(('.ts'))) {
+      parser = 'ts'
+    } else if (fileInfo.path.endsWith('.tsx')) {
+      parser = 'tsx'
+    }
+  }
+
+  if (parser) {
+    jscodeshift = jscodeshift.withParser(parser)
+  }
+
+  const api = {
+    jscodeshift,
+    j: jscodeshift,
+    stats: () => {},
+    report: () => {}
+  }
+
+  return adapt(transform)(fileInfo, api, options)
 }