From 1898c9a76e218ecdd61fb7bba0e4fc8136d746e7 Mon Sep 17 00:00:00 2001
From: TJ Holowaychuk <tj@vision-media.ca>
Date: Sat, 18 Nov 2017 17:39:42 -0800
Subject: [PATCH] add WithDirectories() option

---
 prune.go | 32 +++++++++++++++++++++++++++++++-
 1 file changed, 31 insertions(+), 1 deletion(-)

diff --git a/prune.go b/prune.go
index b4d3cc4..90fbc43 100644
--- a/prune.go
+++ b/prune.go
@@ -9,7 +9,28 @@ import (
 	"github.com/pkg/errors"
 )
 
+// DefaultDirectories pruned.
+//
+// Copied from yarn (mostly).
+var DefaultDirectories = []string{
+	"__tests__",
+	"test",
+	"tests",
+	"powered-test",
+	"docs",
+	"doc",
+	"website",
+	"images",
+	"assets",
+	"example",
+	"examples",
+	"coverage",
+	".nyc_output",
+}
+
 // DefaultExtensions pruned.
+//
+// Copied from yarn (mostly).
 var DefaultExtensions = []string{
 	".md",
 	".ts",
@@ -26,6 +47,7 @@ type Stats struct {
 type Pruner struct {
 	dir  string
 	log  log.Interface
+	dirs map[string]struct{}
 	exts map[string]struct{}
 }
 
@@ -61,6 +83,13 @@ func WithExtensions(s []string) Option {
 	}
 }
 
+// WithDirectories option.
+func WithDirectories(s []string) Option {
+	return func(v *Pruner) {
+		v.dirs = toMap(s)
+	}
+}
+
 // Prune performs the pruning.
 func (p Pruner) Prune() (*Stats, error) {
 	var stats Stats
@@ -97,7 +126,8 @@ func (p Pruner) Prune() (*Stats, error) {
 // prune returns true if the file or dir should be pruned.
 func (p Pruner) prune(path string, info os.FileInfo) bool {
 	if info.IsDir() {
-		return false
+		_, ok := p.dirs[info.Name()]
+		return ok
 	}
 
 	ext := filepath.Ext(path)