71 lines
2 KiB
TypeScript
71 lines
2 KiB
TypeScript
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
import { promises as fs } from "fs";
|
|
import * as path from "path";
|
|
|
|
/**
|
|
* Recursively walk a directory, returning the files it finds.
|
|
*
|
|
* @param dirPath Path to the directory to walk.
|
|
*
|
|
* @return Generator that yields the files found while walking the directory.
|
|
*/
|
|
export const walkDir = async function* (
|
|
dirPath: string
|
|
): AsyncGenerator<string> {
|
|
const dirEnts = await fs.readdir(dirPath, { withFileTypes: true });
|
|
for (const dirEnt of dirEnts) {
|
|
if (dirEnt.isDirectory()) {
|
|
yield* walkDir(path.join(dirPath, dirEnt.name));
|
|
}
|
|
if (dirEnt.isFile()) {
|
|
yield path.join(dirPath, dirEnt.name);
|
|
}
|
|
}
|
|
};
|
|
|
|
/**
|
|
* Purge cached versions of a node module and all of its dependencies from the
|
|
* global require cache, ensuring that future imports reload the module from
|
|
* disk.
|
|
*
|
|
* @param modName Name of the module to purge from the require cache.
|
|
*/
|
|
export const purgeModuleAndDepsFromCache = (modName: string): void => {
|
|
const modPath = require.resolve(modName);
|
|
if (modPath == null) {
|
|
return;
|
|
}
|
|
const mod = require.cache[modPath];
|
|
if (mod == null) {
|
|
return;
|
|
}
|
|
for (const child of mod.children) {
|
|
purgeModuleAndDepsFromCache(child.id);
|
|
}
|
|
delete require.cache[modPath];
|
|
};
|
|
|
|
export type Flattenable<T> = Array<T | Flattenable<T>>;
|
|
|
|
/**
|
|
* Flatten an arbitrarily-deeply nested array into a flat array.
|
|
*
|
|
* @param arr Array to flatten.
|
|
*
|
|
* @return Flattened array.
|
|
*/
|
|
export const flatDeep = <T>(arr: Flattenable<T>): T[] => {
|
|
const flattenedArr: T[] = [];
|
|
for (const val of arr) {
|
|
if (Array.isArray(val)) {
|
|
flattenedArr.push(...flatDeep(val));
|
|
} else {
|
|
flattenedArr.push(val);
|
|
}
|
|
}
|
|
return flattenedArr;
|
|
};
|