diff --git a/fs/walk.ts b/fs/walk.ts index cce3bad64d5861..d986ef47f5a486 100644 --- a/fs/walk.ts +++ b/fs/walk.ts @@ -1,13 +1,14 @@ // 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. -const { readDir, readDirSync } = Deno; +const { readDir, readDirSync, stat, statSync } = Deno; type FileInfo = Deno.FileInfo; import { unimplemented } from "../testing/asserts.ts"; import { join } from "./path/mod.ts"; export interface WalkOptions { maxDepth?: number; + includeDirs?: boolean; exts?: string[]; match?: RegExp[]; skip?: RegExp[]; @@ -47,13 +48,14 @@ export interface WalkInfo { info: FileInfo; } -/** Walks the file tree rooted at root, calling walkFn for each file or - * directory in the tree, including root. The files are walked in lexical +/** Walks the file tree rooted at root, yielding each file or directory in the + * tree filtered according to the given options. The files are walked in lexical * order, which makes the output deterministic but means that for very large * directories walk() can be inefficient. * * Options: * - maxDepth?: number; + * - includeDirs?: boolean; * - exts?: string[]; * - match?: RegExp[]; * - skip?: RegExp[]; @@ -70,6 +72,10 @@ export async function* walk( options: WalkOptions = {} ): AsyncIterableIterator { options.maxDepth! -= 1; + if (options.includeDirs && include(root, options)) { + const rootInfo = await stat(root); + yield { filename: root, info: rootInfo }; + } let ls: FileInfo[] = []; try { ls = await readDir(root); @@ -108,6 +114,10 @@ export function* walkSync( options: WalkOptions = {} ): IterableIterator { options.maxDepth! -= 1; + if (options.includeDirs && include(root, options)) { + const rootInfo = statSync(root); + yield { filename: root, info: rootInfo }; + } let ls: FileInfo[] = []; try { ls = readDirSync(root); diff --git a/fs/walk_test.ts b/fs/walk_test.ts index 1111f9f00f632f..034e327e2e0bb9 100644 --- a/fs/walk_test.ts +++ b/fs/walk_test.ts @@ -118,6 +118,20 @@ testWalk( } ); +testWalk( + async (d: string): Promise => { + await touch(d + "/a"); + await mkdir(d + "/b"); + await touch(d + "/b/c"); + }, + async function includeDirs(): Promise { + assertReady(2); + const arr = await walkArray(".", { includeDirs: true }); + assertEquals(arr.length, 4); + assertEquals(arr, [".", "a", "b", "b/c"]); + } +); + testWalk( async (d: string): Promise => { await touch(d + "/x.ts");