Update content for md

1 files changed, 32 insertions, 0 deletions

diff --git a/gemfeed/aitest/cmd/aitest/main.go b/gemfeed/aitest/cmd/aitest/main.go
new file mode 100644
index 00000000..130cfe0d
--- /dev/null
+++ b/gemfeed/aitest/cmd/aitest/main.go
@@ -0,0 +1,32 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"os"
+
+	"aitest/internal"
+)
+
+func main() {
+	var versionFlag bool
+	flag.BoolVar(&versionFlag, "v", false, "print version")
+	dir := flag.String("dir", "", "directory to count files in")
+	flag.Parse()
+
+	if versionFlag {
+		fmt.Println(internal.GetVersion())
+		return
+	}
+
+	if *dir != "" {
+		fileCount, err := internal.CountFiles(*dir)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "Error counting files: %v\n", err)
+			os.Exit(1)
+		}
+		fmt.Printf("Number of files in directory %s: %d\n", *dir, fileCount)
+	} else {
+		fmt.Println("No directory specified. No count given.")
+	}
+}