From 030cff6e37a76b2494009ce95616ce795a935f02 Mon Sep 17 00:00:00 2001
From: Arya Tabaie <arya.pourtabatabaie@gmail.com>
Date: Tue, 20 Sep 2022 17:46:28 -0400
Subject: [PATCH] feat: directory as input

---
 main.go | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index 55e9e70..87b8deb 100644
--- a/main.go
+++ b/main.go
@@ -15,8 +15,29 @@ func main() {
 
 	flag.Parse()
 
-	for _, filename := range flag.Args() {
-		process(filename)
+	for _, path := range flag.Args() {
+		processPath(path)
+	}
+}
+
+func processPath(path string) {
+	fileInfo, err := os.Stat(path)
+	if err != nil {
+		log.Fatal(err)
+	}
+	if fileInfo.IsDir() {
+		entries, err := os.ReadDir(path)
+
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		for _, entry := range entries {
+			processPath(entry.Name())
+		}
+
+	} else {
+		process(path)
 	}
 }