Make pruning use the correct config and clean up *before* running tests, too.

[mailshears.git] / lib / prune / prune_runner.rb

diff --git a/lib/prune/prune_runner.rb b/lib/prune/prune_runner.rb
index 789ed573623ef3b72cd0e5d983c45e9f1bbc68c1..fbad8e8f11c4e5cb71f7dcc2ccbdc0caf7ad40b5 100644 (file)
--- a/lib/prune/prune_runner.rb
+++ b/lib/prune/prune_runner.rb
@@ -1,10 +1,24 @@
 require 'common/runner'
+require 'prune/plugins/postfixadmin'
+require 'rm/rm_runner'
 
 class PruneRunner
   include Runner
 
-  def run(plugin)
-    puts "Not implemented"
+  def run(cfg, plugin)
+    # We don't want to check the PostfixAdmin database against itself.
+    return if plugin.class == PostfixadminPrune
+
+    pfa = PostfixadminPrune.new(cfg)
+
+    db_users = pfa.list_users()
+    db_domains = pfa.list_domains()
+
+    leftovers  = plugin.get_leftover_users(db_users)
+    leftovers += plugin.get_leftover_domains(db_domains)
+
+    rm_runner = RmRunner.new()
+    rm_runner.run(cfg, plugin, *leftovers)
   end
 
 end