feature: goldstein: convert: add-array

coderaiser · Oct 19, 2023 · 2346464 · 2346464
1 parent 070d081
commit 2346464
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/packages/convert/add-array/index.js b/packages/convert/add-array/index.js
@@ -0,0 +1,4 @@
+export const report = () => `Use 'add-array'`;
+export const replace = () => ({
+    '__a.push(...__array)': '__a += __array',
+});
diff --git a/packages/convert/index.js b/packages/convert/index.js
@@ -3,6 +3,7 @@ import {transform} from 'putout';
 import {print} from '../printer/index.js';
 import * as removeImportTry from './remove-import-try/index.js';
 import * as applyTry from './apply-try/index.js';
+import * as addArray from './add-array/index.js';
 import {
     fixEmpty,
     parse,
@@ -13,6 +14,7 @@ export const convert = (source) => {
 
     transform(ast, source, {
         plugins: [
+            ['add-array', addArray],
             ['apply-try', applyTry],
             ['remove-import-try', removeImportTry],
         ],

diff --git a/packages/convert/index.spec.js b/packages/convert/index.spec.js
@@ -29,3 +29,18 @@ test('goldstein: convert: tryCatch: import', (t) => {
     t.equal(result, expected);
     t.end();
 });
+
+test('goldstein: convert: add-array', (t) => {
+    const source = `
+        a.push(...[2, 3]);
+    `;
+
+    const result = convert(source);
+
+    const expected = montag`
+        a += [2, 3];\n
+    `;
+
+    t.equal(result, expected);
+    t.end();
+});