feature: @putout/apply-arrow: handle case when no loc found

Author: coderaiser

packages/plugin-apply-arrow/lib/apply-arrow.js

@@ -1,13 +1,12 @@
 'use strict';
 
+const {print} = require('putout');
+
 module.exports.report = () => `Use 'Arrow Function' instead of 'Function Declaration`;
 
 module.exports.match = () => ({
     'function __a(__args) {return __b}': ({__a, __b}, path) => {
-        const columnsCount = __b.loc.end.column - __b.loc.start.column;
-        const linesCount = __b.loc.end.line - __b.loc.start.line;
-        
-        if (columnsCount > 30 && !linesCount)
+        if (isToLong(__b))
             return false;
 
         const {name} = __a;
@@ -23,3 +22,25 @@ module.exports.match = () => ({
 module.exports.replace = () => ({
     'function __a(__args) {return __b}': 'const __a = (__args) => __b',
 });
+
+function isToLong(__b) {
+    let columnsCount = 0;
+    let linesCount = 0;
+    
+    if (__b.loc) {
+        columnsCount = __b.loc.end.column - __b.loc.start.column;
+        linesCount = __b.loc.end.line - __b.loc.start.line;
+    } else {
+        const code = print(__b);
+        const lines = code
+            .split('\n')
+            .filter(Boolean);
+        
+        const [first] = lines;
+        
+        linesCount = lines.length - 1;
+        columnsCount = first.length;
+    }
+    
+    return columnsCount > 30 && !linesCount;
+}

packages/plugin-apply-arrow/test/apply-arrow.js

@@ -1,7 +1,10 @@
 'use strict';
 
 const {createTest} = require('@putout/test');
+const {operator} = require('putout');
+
 const plugin = require('..');
+const {getTemplateValues} = operator;
 
 const test = createTest(__dirname, {
     plugins: [
@@ -23,3 +26,19 @@ test('putout: apply-arrow: no report: long', (t) => {
     t.noReport('long');
     t.end();
 });
+
+test('putout: apply-arrow: no report: no-loc', (t) => {
+    const FN = 'function __a(__args) {return __b}';
+    
+    t.transform('no-loc', {
+        noLoc: {
+            report: () => '',
+            fix: (path) => {
+                const {__b} = getTemplateValues(path, FN);
+                delete __b.loc;
+            },
+            include: () => [FN],
+        },
+    });
+    t.end();
+});

packages/plugin-apply-arrow/test/fixture/no-loc-fix.js

@@ -0,0 +1,3 @@
+export function equal(a, b) {
+    return a.type === b.type && a.value === b.value;
+}

packages/plugin-apply-arrow/test/fixture/no-loc.js

@@ -0,0 +1,4 @@
+export function equal(a, b) {
+    return a.type === b.type && a.value === b.value;
+}
+