UNPKG

microcms_sdk_generator

Version:
144 lines (143 loc) 5.1 kB
// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license. // Documentation and interface for walk were adapted from Go // https://golang.org/pkg/path/filepath/#Walk // Copyright 2009 The Go Authors. All rights reserved. BSD license. import * as dntShim from "../../../../_dnt.shims.js"; import { assert } from "../_util/asserts.js"; import { join, normalize } from "../path/mod.js"; import { createWalkEntry, createWalkEntrySync, toPathString, } from "./_util.js"; export class WalkError extends Error { cause; name = "WalkError"; path; constructor(cause, path) { super(`${cause instanceof Error ? cause.message : cause} for path "${path}"`); this.path = path; this.cause = cause; } } function include(path, exts, match, skip) { if (exts && !exts.some((ext) => path.endsWith(ext))) { return false; } if (match && !match.some((pattern) => !!path.match(pattern))) { return false; } if (skip && skip.some((pattern) => !!path.match(pattern))) { return false; } return true; } function wrapErrorWithPath(err, root) { if (err instanceof WalkError) return err; return new WalkError(err, root); } /** * Walks the file tree rooted at root, yielding each file or directory in the * tree filtered according to the given options. * * @example * ```ts * import { walk } from "https://deno.land/std@$STD_VERSION/fs/walk.ts"; * import { assert } from "https://deno.land/std@$STD_VERSION/testing/asserts.ts"; * * for await (const entry of walk(".")) { * console.log(entry.path); * assert(entry.isFile); * } * ``` */ export async function* walk(root, { maxDepth = Infinity, includeFiles = true, includeDirs = true, followSymlinks = false, exts = undefined, match = undefined, skip = undefined, } = {}) { if (maxDepth < 0) { return; } root = toPathString(root); if (includeDirs && include(root, exts, match, skip)) { yield await createWalkEntry(root); } if (maxDepth < 1 || !include(root, undefined, undefined, skip)) { return; } try { for await (const entry of dntShim.Deno.readDir(root)) { assert(entry.name != null); let path = join(root, entry.name); let { isSymlink, isDirectory } = entry; if (isSymlink) { if (!followSymlinks) continue; path = await dntShim.Deno.realPath(path); // Caveat emptor: don't assume |path| is not a symlink. realpath() // resolves symlinks but another process can replace the file system // entity with a different type of entity before we call lstat(). ({ isSymlink, isDirectory } = await dntShim.Deno.lstat(path)); } if (isSymlink || isDirectory) { yield* walk(path, { maxDepth: maxDepth - 1, includeFiles, includeDirs, followSymlinks, exts, match, skip, }); } else if (includeFiles && include(path, exts, match, skip)) { yield { path, ...entry }; } } } catch (err) { throw wrapErrorWithPath(err, normalize(root)); } } /** Same as walk() but uses synchronous ops */ export function* walkSync(root, { maxDepth = Infinity, includeFiles = true, includeDirs = true, followSymlinks = false, exts = undefined, match = undefined, skip = undefined, } = {}) { root = toPathString(root); if (maxDepth < 0) { return; } if (includeDirs && include(root, exts, match, skip)) { yield createWalkEntrySync(root); } if (maxDepth < 1 || !include(root, undefined, undefined, skip)) { return; } let entries; try { entries = dntShim.Deno.readDirSync(root); } catch (err) { throw wrapErrorWithPath(err, normalize(root)); } for (const entry of entries) { assert(entry.name != null); let path = join(root, entry.name); let { isSymlink, isDirectory } = entry; if (isSymlink) { if (!followSymlinks) continue; path = dntShim.Deno.realPathSync(path); // Caveat emptor: don't assume |path| is not a symlink. realpath() // resolves symlinks but another process can replace the file system // entity with a different type of entity before we call lstat(). ({ isSymlink, isDirectory } = dntShim.Deno.lstatSync(path)); } if (isSymlink || isDirectory) { yield* walkSync(path, { maxDepth: maxDepth - 1, includeFiles, includeDirs, followSymlinks, exts, match, skip, }); } else if (includeFiles && include(path, exts, match, skip)) { yield { path, ...entry }; } } }