feature: @putout/operator-filesystem: findFile: exclude: function

Author: coderaiser

packages/operator-filesystem/README.md

@@ -37,7 +37,7 @@ const [dirPath] = findFile(ast, 'hello');
 const newDirectoryPath = createDirectory(dirPath, 'world');
 ```
 
-### `findFile(directoryPath: DirectoryPath, name: string | string[], exclude?: string[]): (FilePath | DirectoryPath)[]`
+### `findFile(directoryPath: DirectoryPath, name: string | string[], exclude?: string[] | (FilePath) => boolean): (FilePath | DirectoryPath)[]`
 
 ```js
 const {operator} = require('putout');
@@ -60,8 +60,14 @@ Or `exclude` some files:
 ```js
 import {operator} from 'putout';
 
-const {findFile} = operator;
-const coupleFiles = findFile(ast, '*.ts', ['*.d.ts']);
+const {findFile, getFilename} = operator;
+
+const coupleFilesWithExcludeArray = findFile(ast, '*.ts', ['*.d.ts']);
+
+const coupleFilesWithExcludeFn = findFile(ast, '*.ts', (filePath) => {
+    const name = getFilename(filePath);
+    return !name.endsWith('*.d.ts');
+});
 ```
 
 And even search for a directory:

packages/operator-filesystem/lib/filesystem.js

@@ -11,6 +11,7 @@ const {
 } = require('@putout/operate');
 
 const maybeFS = require('./maybe-fs');
+const isFn = (a) => typeof a === 'function';
 const isString = (a) => typeof a === 'string';
 const {isArray} = Array;
 const maybeArray = (a) => isArray(a) ? a : [a];
@@ -64,7 +65,10 @@ module.exports.getParentDirectory = (filePath) => {
 
 module.exports.findFile = findFile;
 
-function isExcluded(name, base, exclude) {
+function isExcluded({path, name, base, exclude}) {
+    if (isFn(exclude))
+        return exclude(path);
+    
     for (const currentExclude of exclude) {
         if (name === currentExclude || getRegExp(currentExclude).test(base))
             return true;
@@ -85,10 +89,18 @@ function findFile(node, name, exclude = []) {
 
         for (const name of names) {
             if (value === name || getRegExp(name).test(base)) {
-                if (isExcluded(name, base, exclude))
+                const path = filenamePath.parentPath;
+                const excluded = isExcluded({
+                    path,
+                    name,
+                    base,
+                    exclude,
+                });
+                
+                if (excluded)
                     continue;
 
-                filePaths.push(filenamePath.parentPath);
+                filePaths.push(path);
             }
         }
     }

packages/operator-filesystem/lib/filesystem.spec.js

@@ -179,6 +179,39 @@ test('putout: operator: filesystem: findFile: exclude', (t) => {
     t.end();
 });
 
+test('putout: operator: filesystem: findFile: exclude: function', (t) => {
+    const ast = parseFilesystem([
+        '/',
+        '/hello/',
+        ['/hello/world.js', 'const a = 5'],
+        ['/hello/hello.js', 'const a = <A></A>'],
+    ]);
+    
+    const exclude = (path) => {
+        const source = readFileContent(path);
+        return source.includes('<');
+    };
+    
+    findFile(ast, '*.js', exclude).map(removeFile);
+    
+    const expected = {
+        type: 'directory',
+        filename: '/',
+        files: [{
+            type: 'directory',
+            filename: '/hello',
+            files: [{
+                type: 'file',
+                filename: '/hello/hello.js',
+                content: 'const a = <A></A>',
+            }],
+        }],
+    };
+    
+    t.equalFilesystems(ast, expected);
+    t.end();
+});
+
 test('putout: operator: filesystem: findFile: no names', (t) => {
     const ast = parse(montag`
         ${FS}({