Add requireArgs option in require-super-in-lifecycle-hooks rule

docs/rules/require-super-in-lifecycle-hooks.md

@@ -20,7 +20,7 @@ import Component from '@ember/component';
 export default Component.extend({
   init() {
     this.set('items', []);
-  }
+  },
 });
 ```
 
@@ -30,7 +30,7 @@ import Component from '@ember/component';
 export default Component.extend({
   didInsertElement() {
     // ...
-  }
+  },
 });
 ```
 
@@ -53,7 +53,7 @@ export default Component.extend({
   init(...args) {
     this._super(...args);
     this.set('items', []);
-  }
+  },
 });
 ```
 
@@ -64,7 +64,7 @@ export default Component.extend({
   didInsertElement(...args) {
     this._super(...args);
     // ...
-  }
+  },
 });
 ```
 
@@ -96,3 +96,4 @@ This rule takes an optional object containing:
 
 - `boolean` -- `checkInitOnly` -- whether the rule should only check the `init` lifecycle hook and not other lifecycle hooks (default `false`)
 - `boolean` -- `checkNativeClasses` -- whether the rule should check lifecycle hooks in native classes (in addition to classic classes) (default `true`)
+- `boolean` -- `requireArgs` -- whether the rule should check if super() in init hook is called with args (default `false`)

lib/rules/require-super-in-lifecycle-hooks.js

@@ -6,15 +6,20 @@ const estraverse = require('estraverse');
 
 function hasMatchingNode(node, matcher) {
   let foundMatch = false;
+  let foundNode = null;
   estraverse.traverse(node, {
     enter(child) {
       if (!foundMatch && matcher(child)) {
         foundMatch = true;
+        foundNode = child;
       }
     },
     fallback: 'iteration',
   });
-  return foundMatch;
+  return {
+    foundMatch,
+    foundNode,
+  };
 }
 
 /**
@@ -79,6 +84,10 @@ module.exports = {
             type: 'boolean',
             default: true,
           },
+          requireArgs: {
+            type: 'boolean',
+            default: false,
+          },
         },
         additionalProperties: false,
       },
@@ -88,8 +97,9 @@ module.exports = {
   create(context) {
     const checkInitOnly = context.options[0] && context.options[0].checkInitOnly;
     const checkNativeClasses = !context.options[0] || context.options[0].checkNativeClasses;
+    const requireArgs = context.options[0] && context.options[0].requireArgs;
 
-    function report(node, isNativeClass, lifecycleHookName) {
+    function report(node, isNativeClass, lifecycleHookName, hasSuper) {
       context.report({
         node,
         message: ERROR_MESSAGE,
@@ -100,12 +110,23 @@ module.exports = {
             : '...arguments';
 
           const replacement = isNativeClass
-            ? `super.${lifecycleHookName}(${replacementArgs});`
-            : `this._super(${replacementArgs});`;
+            ? `super.${lifecycleHookName}(${replacementArgs})`
+            : `this._super(${replacementArgs})`;
+
+          // If super is already called, replace it with the correct call
+          if (hasSuper) {
+            const nodeToBeReplaced = hasMatchingNode(node, (bodyChild) =>
+              isNativeClass
+                ? isNativeSuper(bodyChild, lifecycleHookName)
+                : isClassicSuper(bodyChild)
+            ).foundNode;
+            return fixer.replaceText(nodeToBeReplaced, replacement);
+          }
+
           // Insert right after function curly brace.
           const sourceCode = context.getSourceCode();
           const startOfBlockStatement = sourceCode.getFirstToken(node.value.body);
-          return fixer.insertTextAfter(startOfBlockStatement, `\n${replacement}`);
+          return fixer.insertTextAfter(startOfBlockStatement, `\n${replacement};`);
         },
       });
     }
@@ -157,9 +178,22 @@ module.exports = {
       const body = isNativeSuper ? node.value.body : node.body;
       const hasSuper = hasMatchingNode(body, (bodyChild) =>
         isNativeClass ? isNativeSuper(bodyChild, hookName) : isClassicSuper(bodyChild)
-      );
-      if (!hasSuper) {
-        report(node, isNativeClass, hookName);
+      ).foundMatch;
+
+      if (hookName === 'init' && hasSuper && requireArgs) {
+        const hasSuperWithArguments = hasMatchingNode(
+          body,
+          (bodyChild) =>
+            (isNativeSuper(bodyChild, hookName) || isClassicSuper(bodyChild)) &&
+            bodyChild.arguments &&
+            bodyChild.arguments.length > 0
+        ).foundMatch;
+
+        if (!hasSuperWithArguments) {
+          report(node, isNativeClass, hookName, hasSuper);
+        }
+      } else if (!hasSuper) {
+        report(node, isNativeClass, hookName, hasSuper);
       }
     }
 

tests/lib/rules/require-super-in-lifecycle-hooks.js

@@ -231,6 +231,19 @@ eslintTester.run('require-super-in-lifecycle-hooks', rule, {
 
     // Does not throw with node type (ClassProperty) not handled by estraverse.
     'Component.extend({init() { class Foo { @someDecorator() someProp }; return this._super(...arguments); } });',
+
+    // init hook should not be checked for arguments when requireArgs = false
+    `import Service from '@ember/service';
+      class Foo extends Service {
+        init() {
+          super.init();
+        }
+      }`,
+    `export default Component.extend({
+        init() {
+          this._super(...arguments);
+        },
+      });`,
   ],
   invalid: [
     {
@@ -858,5 +871,73 @@ super.didInsertElement(...arguments);}
       options: [{ checkNativeClasses: true, checkInitOnly: false }],
       errors: [{ message, line: 3 }],
     },
+
+    // init hooks shoudn't call super without ...arguments
+    {
+      code: `import Service from '@ember/service';
+      class Foo extends Service {
+        init() {
+          super.init();
+        }
+      }`,
+      output: `import Service from '@ember/service';
+      class Foo extends Service {
+        init() {
+          super.init(...arguments);
+        }
+      }`,
+      options: [{ requireArgs: true }],
+      errors: [{ message, line: 3 }],
+    },
+    {
+      code: `import Service from '@ember/service';
+      class Foo extends Service {
+        init() {
+          super.init();
+          console.log();
+        }
+      }`,
+      output: `import Service from '@ember/service';
+      class Foo extends Service {
+        init() {
+          super.init(...arguments);
+          console.log();
+        }
+      }`,
+      options: [{ requireArgs: true }],
+      errors: [{ message, line: 3 }],
+    },
+    {
+      code: `export default Component.extend({
+        init() {
+          this._super();
+        },
+      });`,
+      output: `export default Component.extend({
+        init() {
+          this._super(...arguments);
+        },
+      });`,
+      options: [{ requireArgs: true }],
+      errors: [{ message, line: 2 }],
+    },
+    {
+      code: `export default Component.extend({
+        init() {
+          this._super();
+          this.set('prop', 'value');
+          this.set('prop2', 'value2');
+        },
+      });`,
+      output: `export default Component.extend({
+        init() {
+          this._super(...arguments);
+          this.set('prop', 'value');
+          this.set('prop2', 'value2');
+        },
+      });`,
+      options: [{ requireArgs: true }],
+      errors: [{ message, line: 2 }],
+    },
   ],
 });