fix: multiple lines fail case

lvjiaxuan · Sep 21, 2022 · c521151 · c521151
1 parent b8e2c9f
commit c521151
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/extension.ts b/src/extension.ts
@@ -114,12 +114,18 @@ const disposes = [
         )
       } else {
         // Wrap lines. Press `ctrl+d `to edit rules at between lines.
+
+        const ruleIDSet = new Set<string>()
+        for (const line in lineRuleIdsMap) {
+          lineRuleIdsMap[line].forEach(item => ruleIDSet.add(item))
+        }
+
         void activeTextEditor.insertSnippet(
-          new SnippetString(`/* eslint-disable \${1|${ lineRuleIdsMap[selection.start.line + 1].join('\\, ') }|} */\n`),
+          new SnippetString(`/* eslint-disable \${1|${ [ ...ruleIDSet ].join('\\, ') }|} */\n`),
           new Position(selection.start.line, insertIndex),
         )
         void activeTextEditor.insertSnippet(
-          new SnippetString(`/* eslint-enable \${1|${ lineRuleIdsMap[selection.start.line + 1].join('\\, ') }|} */\n`),
+          new SnippetString(`/* eslint-enable \${1|${ [ ...ruleIDSet ].join('\\, ') }|} */\n`),
           new Position(selection.end.line + 2, insertIndex),
         )
       }