1 files changed, 21 insertions, 0 deletions
diff --git a/gemfeed/aitest/internal/count.go b/gemfeed/aitest/internal/count.go
new file mode 100644
index 00000000..544ace17
--- /dev/null
+++ b/gemfeed/aitest/internal/count.go
@@ -0,0 +1,21 @@
+package internal
+
+import (
+	"os"
+)
+
+func CountFiles(dir string) (int, error) {
+	files, err := os.ReadDir(dir)
+	if err != nil {
+		return 0, err
+	}
+
+	count := 0
+	for _, file := range files {
+		if !file.IsDir() {
+			count++
+		}
+	}
+
+	return count, nil
+}