Fix the inline prefix issue and added an option for default prefixing

Suchita Doshi · Suchita Doshi · commit 28576ce2c974 · 2019-12-23T15:47:08.000-08:00
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,3 @@
 node_modules
-*.log
+*.log
+.DS_Store
diff --git a/README.md b/README.md
@@ -20,7 +20,11 @@ A codemod for transforming your ember app code to start using `@tracked` propert
 npx ember-tracked-properties-codemod path/of/files/ or/some**/*glob.hbs
 ```
     
-    
+The codemod accepts the following options:
+
+|        Option         |  Value  |             Default             |                                                                     Details                                                                      |
+| --------------------- | ------- | ------------------------------- | ------------------------------------------------------------------------------------------------------------------------------------------------ |
+| `--always-prefix`      | boolean | `true`                          | Always prefix `@tracked` inline                                                                             |
 
 ## Contributing
 
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
   "name": "ember-tracked-properties-codemod",
   "description": "A codemod for transforming your ember app code to start using `@tracked` properties.",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "author": {
     "name": "Suchita Doshi",
     "email": "suchitadoshi1987@gmail.com",
@@ -14,10 +14,14 @@
   },
   "bin": "./bin/cli.js",
   "keywords": [
-    "codemod-cli"
+    "codemod-cli",
+    "codemods",
+    "ember",
+    "ember-codemod",
+    "ember-tracked-properties"
   ],
   "dependencies": {
-    "codemod-cli": "^0.2.7"
+    "codemod-cli": "^2.1.0"
   },
   "devDependencies": {
     "@babel/plugin-proposal-class-properties": "^7.7.4",
diff --git a/transforms/tracked-properties/__testfixtures__/basic-with-prefix-false.input.js b/transforms/tracked-properties/__testfixtures__/basic-with-prefix-false.input.js
@@ -0,0 +1,12 @@
+import Component from '@ember/component';
+import { computed, get } from '@ember/object';
+
+export default class Foo extends Component {
+  bar;
+  baz = 'barBaz';
+
+  @computed('baz')
+  get bazInfo() {
+    return `Name: ${get(this, 'baz')}`;
+  }
+}
diff --git a/transforms/tracked-properties/__testfixtures__/basic-with-prefix-false.options.json b/transforms/tracked-properties/__testfixtures__/basic-with-prefix-false.options.json
@@ -0,0 +1,3 @@
+{
+  "alwaysPrefix": "false"
+}
diff --git a/transforms/tracked-properties/__testfixtures__/basic-with-prefix-false.output.js b/transforms/tracked-properties/__testfixtures__/basic-with-prefix-false.output.js
@@ -0,0 +1,13 @@
+import { tracked } from '@glimmer/tracking';
+import Component from '@ember/component';
+import { computed, get } from '@ember/object';
+
+export default class Foo extends Component {
+  bar;
+  @tracked
+  baz = 'barBaz';
+
+  get bazInfo() {
+    return `Name: ${get(this, 'baz')}`;
+  }
+}
diff --git a/transforms/tracked-properties/__testfixtures__/basic.output.js b/transforms/tracked-properties/__testfixtures__/basic.output.js
@@ -4,8 +4,7 @@ import { computed, get } from '@ember/object';
 
 export default class Foo extends Component {
   bar;
-  @tracked
-  baz = 'barBaz';
+  @tracked baz = 'barBaz';
 
   get bazInfo() {
     return `Name: ${get(this, 'baz')}`;
diff --git a/transforms/tracked-properties/__testfixtures__/chained-computed.output.js b/transforms/tracked-properties/__testfixtures__/chained-computed.output.js
@@ -3,8 +3,7 @@ import Component from '@ember/component';
 import { computed, get } from '@ember/object';
 
 export default class Foo extends Component {
-  @tracked
-  foo = 'bar';
+  @tracked foo = 'bar';
   baz;
 
   get fooBar() {
diff --git a/transforms/tracked-properties/__testfixtures__/complex.output.js b/transforms/tracked-properties/__testfixtures__/complex.output.js
@@ -3,8 +3,7 @@ import Component from '@ember/component';
 import { computed, get } from '@ember/object';
 
 export default class Foo extends Component {
-  @tracked
-  firstName = 'Foo';
+  @tracked firstName = 'Foo';
   @tracked lastName;
   @tracked phone;
   zipcode;
diff --git a/transforms/tracked-properties/__testfixtures__/with-tracked.output.js b/transforms/tracked-properties/__testfixtures__/with-tracked.output.js
@@ -4,8 +4,7 @@ import { computed, get } from '@ember/object';
 
 export default class Foo extends Component {
   @tracked bar;
-  @tracked
-  baz = 'barBaz';
+  @tracked baz = 'barBaz';
 
   get bazInfo() {
     return `Name: ${get(this, 'baz')}`;
diff --git a/transforms/tracked-properties/index.js b/transforms/tracked-properties/index.js
@@ -3,9 +3,16 @@ const {
   addTrackedImport,
   getDependentKeys,
   buildTrackedDecorator,
+  reformatTrackedDecorators,
 } = require('./utils/helper');
 
+const { getOptions } = require('codemod-cli');
+const DEFAULT_OPTIONS = {
+  alwaysPrefix: 'true',
+};
+
 module.exports = function transformer(file, api) {
+  const configOptions = Object.assign({}, DEFAULT_OPTIONS, getOptions());
   const classProps = [];
   let computedProps = [];
   let computedPropsMap = {};
@@ -51,7 +58,7 @@ module.exports = function transformer(file, api) {
   // with @tracked. Also, set the `shouldImportBeAdded` to true which would help
   // determine if the import statement `@glimmer/tracking` needs to be added to
   // the file.
-  const trackedConvertedSource = j(file.source)
+  let trackedConvertedSource = j(file.source)
     .find(j.ClassProperty)
     .forEach(path => {
       if (computedProps.includes(path.node.key.name)) {
@@ -67,6 +74,10 @@ module.exports = function transformer(file, api) {
     })
     .toSource();
 
+  if (configOptions.alwaysPrefix === 'true') {
+    trackedConvertedSource = reformatTrackedDecorators(trackedConvertedSource);
+  }
+
   // Iterate on all the `computed` decorators and for each node, check if
   // all the arguments are a part of the `classProps` array, if so, then
   // remove the `@computed` decorator, else remove the arguments that are
diff --git a/transforms/tracked-properties/utils/helper.js b/transforms/tracked-properties/utils/helper.js
@@ -89,4 +89,35 @@ function buildTrackedDecorator(name, value, j) {
   return node;
 }
 
-module.exports = { addTrackedImport, getDependentKeys, buildTrackedDecorator };
+/**
+ * Returns the formatted @tracked signature. After adding the @tracked decorator
+ * to the source, if the class property has a value associated with it,
+ * the @tracked decorator is add above the property instead of prefixing it inline.
+ * This function will check if the length of characters including the @tracked addition
+ * doesn't add up to more than 50 characters, then reformat it to be prefixed instead
+ * of being on a separate line.
+ * @param {string} trackedConvertedSource
+ */
+function reformatTrackedDecorators(trackedConvertedSource) {
+  const matchedTracked = trackedConvertedSource.match(/@tracked\n(.*)\n/g);
+  if (matchedTracked) {
+    matchedTracked.forEach(matchedData => {
+      const convertedMatchedData = matchedData.replace(
+        /@tracked\n\s+/,
+        '@tracked '
+      );
+      trackedConvertedSource = trackedConvertedSource.replace(
+        matchedData,
+        convertedMatchedData
+      );
+    });
+  }
+  return trackedConvertedSource;
+}
+
+module.exports = {
+  addTrackedImport,
+  getDependentKeys,
+  buildTrackedDecorator,
+  reformatTrackedDecorators,
+};
diff --git a/yarn.lock b/yarn.lock

-Original file line number
+Diff line change
@@ @@ -1,2 +1,3 @@ @@
 node_modules
 -*.log
 +*.log
 +.DS_Store
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+{`
	`2`	`+ "alwaysPrefix": "false"`
	`3`	`+}`