From fa9cf9d1c54a6eaefa9334f9ca2bc5d35a7737f0 Mon Sep 17 00:00:00 2001
From: Joscha <joscha@plugh.de>
Date: Thu, 17 Aug 2023 19:18:25 +0200
Subject: [PATCH] Avoid excessive amount of data in large repos

---
 src/worker/run/internal.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/worker/run/internal.rs b/src/worker/run/internal.rs
index 5536d32..ee8ab51 100644
--- a/src/worker/run/internal.rs
+++ b/src/worker/run/internal.rs
@@ -96,6 +96,17 @@ fn count(path: &Path) -> somehow::Result<Counts> {
         }
     }
 
+    // Avoid excessive amounts of data in very large repos
+    if counts.files_by_dir.len() > 1000 {
+        counts.files_by_dir.retain(|name, _| !name.contains('/'));
+    }
+    if counts.lines_by_dir.len() > 1000 {
+        counts.lines_by_dir.retain(|name, _| !name.contains('/'));
+    }
+    if counts.todos_by_dir.len() > 1000 {
+        counts.todos_by_dir.retain(|name, _| !name.contains('/'));
+    }
+
     Ok(counts)
 }