feature: @putout/plugin-montag: add-newline-before-text: multiline

Author: coderaiser

packages/plugin-montag/lib/add-newline-before-text/fixture/indent-fix.js

@@ -0,0 +1,11 @@
+test('putout: operator: json: isDocker', ({ok}) => {
+    const source = montag`
+        __putout_processor_docker([
+            ['MAINTAINER', 'docker@moby.com']
+        ]);\n
+    `;
+    
+    const is = isDocker(source);
+    
+    ok(is);
+});

packages/plugin-montag/lib/add-newline-before-text/fixture/indent.js

@@ -0,0 +1,10 @@
+test('putout: operator: json: isDocker', ({ok}) => {
+    const source = montag`__putout_processor_docker([
+        ['MAINTAINER', 'docker@moby.com']
+    ]);\n
+    `;
+    
+    const is = isDocker(source);
+    
+    ok(is);
+});

packages/plugin-montag/lib/add-newline-before-text/index.js

@@ -1,3 +1,7 @@
+import {types} from 'putout';
+
+const {isVariableDeclarator} = types;
+
 export const report = () => `Add '\\n' after montag`;
 
 export const match = () => ({
@@ -14,12 +18,50 @@ export const match = () => ({
 export const replace = () => ({
     'montag`__a`': ({__a}, path) => {
         const {raw} = __a.value;
-        const {column} = path.node.loc.start;
-        const indentBefore = Array(column + 4).join(' ');
-        const indentAfter = Array(column).join(' ');
+        const column = getColumn(path);
 
-        __a.value.raw = `\n${indentBefore}${raw}\n${indentAfter}`;
+        if (raw.includes('\n'))
+            __a.value.raw = createMultilineValue({
+                raw,
+                column,
+            });
+        else
+            __a.value.raw = createOnelineValue({
+                raw,
+                column,
+            });
 
         return path;
     },
 });
+
+function createOnelineValue({raw, column}) {
+    const indentAfter = Array(column).join(' ');
+    const indentBefore = Array(column + 4).join(' ');
+    
+    return `\n${indentBefore}${raw}\n${indentAfter}`;
+}
+
+function createMultilineValue({raw, column}) {
+    const rawLines = raw.split('\n');
+    const indentAfter = Array(column).join(' ');
+    const indentBefore = Array(column).join(' ');
+    const count = rawLines.length - 1;
+    const lines = [];
+    
+    for (const [index, line] of rawLines.entries()) {
+        if (index === count)
+            break;
+        
+        lines.push(`${indentBefore} ${line}`);
+    }
+    
+    return `\n    ${lines.join('\n')}\n ${indentAfter}`;
+}
+
+function getColumn(path) {
+    if (isVariableDeclarator(path.parentPath))
+        return path.parentPath.parentPath.node.loc.start.column;
+    
+    return path.node.loc.start.column;
+}

packages/plugin-montag/lib/add-newline-before-text/index.spec.js

@@ -21,3 +21,8 @@ test('montag: add-newline-before-text: no report: space', (t) => {
     t.noReport('space');
     t.end();
 });
+
+test('montag: add-newline-before-text: transform: indent', (t) => {
+    t.transform('indent');
+    t.end();
+});