fix(eslint-plugin): [typedef] allow array/object destructuring in for/of

typescript-eslint · Feb 5, 2020 · 0f39133 · 0f39133
1 parent 10d86b1
commit 0f39133
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 2 deletions.
diff --git a/packages/eslint-plugin/docs/rules/typedef.md b/packages/eslint-plugin/docs/rules/typedef.md
@@ -76,6 +76,9 @@ Examples of **correct** code with `{ "arrayDestructuring": true }`:
 const [a]: number[] = [1];
 const [b]: [number] = [2];
 const [c, d]: [boolean, string] = [true, 'text'];
+
+for (const [key, val] of new Map([['key', 1]])) {
+}
 ```
 
 ### `arrowParameter`
@@ -144,6 +147,9 @@ Examples of **correct** code with `{ "objectDestructuring": true }`:
 ```ts
 const { length }: { length: number } = 'text';
 const [b, c]: [number, number] = Math.random() ? [1, 2] : [3, 4];
+
+for (const { key, val } of [{ key: 'key', val: 1 }]) {
+}
 ```
 
 ### `parameter`

diff --git a/packages/eslint-plugin/src/rules/typedef.ts b/packages/eslint-plugin/src/rules/typedef.ts
@@ -69,6 +69,26 @@ export default util.createRule<[Options], MessageIds>({
       return node.type === AST_NODE_TYPES.Identifier ? node.name : undefined;
     }
 
+    function isForOfStatementContext(
+      node: TSESTree.ArrayPattern | TSESTree.ObjectPattern,
+    ): boolean {
+      let current: TSESTree.Node | undefined = node.parent;
+      while (current) {
+        if (
+          current.type === AST_NODE_TYPES.VariableDeclarator ||
+          current.type === AST_NODE_TYPES.VariableDeclaration
+        ) {
+          current = current.parent;
+        } else if (current.type === AST_NODE_TYPES.ForOfStatement) {
+          return true;
+        } else {
+          break;
+        }
+      }
+
+      return false;
+    }
+
     function checkParameters(params: TSESTree.Parameter[]): void {
       for (const param of params) {
         let annotationNode: TSESTree.Node | undefined;
@@ -102,7 +122,11 @@ export default util.createRule<[Options], MessageIds>({
 
     return {
       ArrayPattern(node): void {
-        if (options[OptionKeys.ArrayDestructuring] && !node.typeAnnotation) {
+        if (
+          options[OptionKeys.ArrayDestructuring] &&
+          !node.typeAnnotation &&
+          !isForOfStatementContext(node)
+        ) {
           report(node);
         }
       },
@@ -132,7 +156,11 @@ export default util.createRule<[Options], MessageIds>({
         }
       },
       ObjectPattern(node): void {
-        if (options[OptionKeys.ObjectDestructuring] && !node.typeAnnotation) {
+        if (
+          options[OptionKeys.ObjectDestructuring] &&
+          !node.typeAnnotation &&
+          !isForOfStatementContext(node)
+        ) {
           report(node);
         }
       },

diff --git a/packages/eslint-plugin/tests/rules/typedef.test.ts b/packages/eslint-plugin/tests/rules/typedef.test.ts
@@ -38,6 +38,14 @@ ruleTester.run('typedef', rule, {
         },
       ],
     },
+    {
+      code: `for (const [key, val] of new Map([['key', 1]])) {}`,
+      options: [
+        {
+          arrayDestructuring: true,
+        },
+      ],
+    },
     `let a: number;
     [a] = [1];`,
     // Arrow parameters
@@ -101,6 +109,14 @@ ruleTester.run('typedef', rule, {
         },
       ],
     },
+    {
+      code: `for (const { key, val } of [{ key: 'key', val: 1 }]) {}`,
+      options: [
+        {
+          objectDestructuring: true,
+        },
+      ],
+    },
     // Function parameters
     `function receivesNumber(a: number): void { }`,
     `function receivesStrings(a: string, b: string): void { }`,