[WIP] fix body update

__tests__/lib/index.test.ts

@@ -9,7 +9,8 @@ describe('vue2-migration-helper', () => {
   test('should be able to extract js ast from vue sfc', () => {
     expect(
       vue2MigrationHelper({
-        path: './__tests__/data/text.vue'
+        source: './__tests__/data/text.vue',
+        dryRun: true
       })
     ).toBeDefined()
   })

package.json

@@ -30,8 +30,10 @@
     "@types/eslint": "^6.1.3",
     "@types/eslint-plugin-prettier": "^2.2.0",
     "@types/gh-pages": "^2.0.1",
+    "@types/glob": "^7.1.1",
     "@types/jest": "^24.0.23",
     "@types/prettier": "^1.19.0",
+    "@types/yargs": "^13.0.3",
     "@typescript-eslint/eslint-plugin": "^2.11.0",
     "@typescript-eslint/parser": "^2.11.0",
     "babel-eslint": "^10.0.3",
@@ -41,7 +43,7 @@
     "eslint-config-prettier": "^6.7.0",
     "eslint-plugin-import": "^2.19.1",
     "eslint-plugin-jest": "^23.1.1",
-    "eslint-plugin-prettier": "^3.1.1",
+    "eslint-plugin-prettier": "^3.1.2",
     "eslint-plugin-vue": "^6.0.1",
     "gh-pages": "^2.1.1",
     "jest": "^24.9.0",

src/cli/index.ts

@@ -1,30 +1,40 @@
 #!/usr/bin/env node
-import { green, red } from 'chalk'
+import { green, red, yellow } from 'chalk'
 import glob from 'glob'
 import { resolve } from 'path'
 import { scriptName } from 'yargs'
-// const fs = require('fs')
 import { vue2MigrationHelper } from '../index'
 
 const options = scriptName('vue2-migration-helper')
   .option('source', {
     alias: ['s'],
     describe: 'Source directory containing Vue single file components.',
-    demandOption: true
+    demandOption: true,
+    type: 'string'
   })
   .option('target', {
     alias: ['t'],
     describe: 'Target directory to save transformed components.',
-    demandOption: false
+    demandOption: false,
+    type: 'string'
   })
   .option('dry-run', {
     alias: ['d', 'dryRun'],
     describe: 'Target directory to save transformed components.',
     demandOption: false,
-    default: false
+    default: false,
+    type: 'boolean'
   })
   .help()
-  .example('vue2-migration-helper --dry-run --s="source" --t="target"').argv
+  .example('vue2-migration-helper --dry-run --s="source" --t="target"', '').argv
+
+if (options['dry-run']) {
+  console.log(
+    yellow(
+      'vue2-migration-helper in dry-run mode, no files will be written to disk.'
+    )
+  )
+}
 
 const sourceGlob = resolve(options.source, '**/*.vue').replace('\\', '/')
 
@@ -34,12 +44,12 @@ glob(sourceGlob, (err, sources) => {
   sources.forEach(source => {
     console.log(green(`Processing: ${source}`))
 
-    const content = vue2MigrationHelper({
-      path: source
+    // TODO: pass target file
+    vue2MigrationHelper({
+      source: source,
+      dryRun: options.dryRun === true ? true : false
     })
 
-    console.log(content)
-
-    // fs.writeFileSync(path, content)
+    console.log(green('Succesffuly processed file: ', source))
   })
 })

src/index.ts

@@ -1,4 +1,5 @@
 import { types } from '@babel/core'
+import fs from 'fs'
 import { getAst, getCode } from './lib/astUtilities'
 import { addBody } from './lib/generators/body'
 import { addImports } from './lib/generators/imports'
@@ -9,8 +10,8 @@ import { updateVueObjectReferences } from './lib/transformers/vueObjectReference
 
 type Options = {
   source: string
-  target: string
-  dryRun: boolean
+  target?: string
+  dryRun?: boolean
 }
 
 export function vue2MigrationHelper({
@@ -31,14 +32,23 @@ export function vue2MigrationHelper({
   // add body
   outputAst = addBody(outputAst)
 
+  console.log(getCode(outputAst))
+
   // update template refs
   outputAst = updateTemplateRefs(outputAst)
 
   // update vue object this references
   outputAst = updateVueObjectReferences(outputAst)
 
+  // get final code
+  const code = getOutputTemplate(originalTemplate, getCode(outputAst))
+
+  if (!dryRun && target) {
+    fs.writeFileSync(target, code)
+  }
+
   // return final code
-  return getOutputTemplate(originalTemplate, getCode(outputAst))
+  return code
 }
 
 export default {

src/lib/transformers/templateRefs.ts

@@ -22,6 +22,9 @@ export function updateTemplateRefs(ast: types.File) {
   // addd ref(null) for each template ref
   const setupMethodBody = getSetupMethod(ast).body.body
   const returnStatement = setupMethodBody.slice(-1)[0] as types.ReturnStatement
+
+  console.log(setupMethodBody, returnStatement)
+
   const argument = returnStatement.argument as types.ObjectExpression
 
   templateRefList.forEach(templateRef => {