feature: @putout/engine-loader: async-loader: PUTOUT_LOAD_DIR

Author: coderaiser

packages/engine-loader/lib/load/async-loader.js

@@ -1,5 +1,8 @@
 'use strict';
 
+const process = require('node:process');
+const {join} = require('node:path');
+const once = require('once');
 const {nanomemoize} = require('nano-memoize');
 const tryToCatch = require('try-to-catch');
 const {simpleImport} = require('./simple-import');
@@ -14,10 +17,24 @@ module.exports.createAsyncLoader = (type) => nanomemoize(async (name, load) => {
     if (name.startsWith('import:')) {
         const shortName = name.replace('import:', '');
 
-        return await cleverLoad([require.resolve(shortName)], load);
+        return await cleverLoad([
+            require.resolve(shortName),
+        ], load);
     }
 
-    return await cleverLoad([`@putout/${type}-${name}`, `putout-${type}-${name}`], load);
+    const namesBase = [
+        `@putout/${type}-${name}`,
+        `putout-${type}-${name}`,
+    ];
+    
+    const namesFromPluginsDirs = namesBase.flatMap(buildPluginsDirs);
+    
+    const names = Array.from(new Set([
+        ...namesBase,
+        ...namesFromPluginsDirs,
+    ]));
+    
+    return await cleverLoad(names, load);
 });
 
 async function cleverLoad(names, load = simpleImport) {
@@ -46,3 +63,19 @@ async function cleverLoad(names, load = simpleImport) {
 
     throw e;
 }
+
+const getPutoutLoadDir = once(() => process.env.PUTOUT_LOAD_DIR);
+
+function buildPluginsDirs(name) {
+    const dir = getPutoutLoadDir();
+    
+    if (!dir)
+        return [name];
+    
+    const base = join(dir, name);
+    
+    return [
+        base,
+        `${base}.js`,
+    ];
+}

packages/engine-loader/lib/load/async-loader.spec.js

@@ -1,5 +1,7 @@
 'use strict';
 
+const process = require('node:process');
+const {join} = require('node:path');
 const tryToCatch = require('try-to-catch');
 
 const {test, stub} = require('supertape');
@@ -81,3 +83,23 @@ test('putout: loader: async-loader: calls load', async (t) => {
     t.deepEqual(error, Error('@putout/formatter-xxx: LOAD USED'));
     t.end();
 });
+
+test('putout: engine-loader: async-loader: PUTOUT_LOAD_DIR', async (t) => {
+    process.env.PUTOUT_LOAD_DIR = join(__dirname, 'fixture');
+    
+    const {createAsyncLoader} = reRequire('./async-loader');
+    const loadAsync = createAsyncLoader('plugin');
+    
+    const {report} = await loadAsync('hello');
+    
+    stopAll();
+    reRequire('./async-loader');
+    
+    delete process.env.PUTOUT_LOAD_DIR;
+    
+    const result = report();
+    const expected = 'hello';
+    
+    t.equal(result, expected);
+    t.end();
+});