From fd44fe34d43f0522cfd069d7f13f7326d02d89bd Mon Sep 17 00:00:00 2001
From: Nayeem Rahman <muhammed.9939@gmail.com>
Date: Wed, 18 Sep 2019 16:37:37 +0100
Subject: Add includeDirs to WalkOptions (denoland/deno_std#601)

Original: https://github.com/denoland/deno_std/commit/de8d0ab4a100707b57f12cff5db635356221f324
---
 fs/walk.ts      | 16 +++++++++++++---
 fs/walk_test.ts | 14 ++++++++++++++
 2 files changed, 27 insertions(+), 3 deletions(-)

diff --git a/fs/walk.ts b/fs/walk.ts
index cce3bad64..d986ef47f 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<WalkInfo> {
   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<WalkInfo> {
   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 1111f9f00..034e327e2 100644
--- a/fs/walk_test.ts
+++ b/fs/walk_test.ts
@@ -118,6 +118,20 @@ testWalk(
   }
 );
 
+testWalk(
+  async (d: string): Promise<void> => {
+    await touch(d + "/a");
+    await mkdir(d + "/b");
+    await touch(d + "/b/c");
+  },
+  async function includeDirs(): Promise<void> {
+    assertReady(2);
+    const arr = await walkArray(".", { includeDirs: true });
+    assertEquals(arr.length, 4);
+    assertEquals(arr, [".", "a", "b", "b/c"]);
+  }
+);
+
 testWalk(
   async (d: string): Promise<void> => {
     await touch(d + "/x.ts");
-- 
cgit v1.2.3