[proxy] facilitate manually pruning orphaned packages (including installed ones)

1 files changed, 27 insertions, 1 deletions

diff --git a/src/hydrilla/proxy/state_impl/concrete_state.py b/src/hydrilla/proxy/state_impl/concrete_state.py
index 04bb67f..d8706e8 100644
--- a/src/hydrilla/proxy/state_impl/concrete_state.py
+++ b/src/hydrilla/proxy/state_impl/concrete_state.py
@@ -146,7 +146,33 @@ class ConcreteHaketiloState(base.HaketiloStateWithFields):
 
             self.rebuild_structures(rules=False)
 
-    def prune_orphans(self) -> None:
+    def count_orphan_items(self) -> st.OrphanItemsStats:
+        with self.cursor() as cursor:
+            cursor.execute(
+                '''
+                SELECT
+                        COALESCE(SUM(i.type = 'M'), 0),
+                        COALESCE(SUM(i.type = 'R'), 0)
+                FROM
+                             item_versions     AS iv
+                        JOIN items             AS i  USING (item_id)
+                        JOIN orphan_iterations AS oi USING (repo_iteration_id)
+                WHERE
+                        iv.active != 'R';
+                '''
+            )
+
+            (orphan_mappings, orphan_resources), = cursor.fetchall()
+
+        return st.OrphanItemsStats(orphan_mappings, orphan_resources)
+
+    def prune_orphan_items(self) -> None:
+        with self.cursor(transaction=True) as cursor:
+            _operations.prune_orphans(cursor, aggressive=True)
+
+            self.recompute_dependencies()
+
+    def soft_prune_orphan_items(self) -> None:
         with self.cursor() as cursor:
             assert self.connection.in_transaction