X-Git-Url: http://gitweb.michael.orlitzky.com/?p=mailshears.git;a=blobdiff_plain;f=bin%2Fmailshears;h=d9d16253cee734bba0428d64c0cbf07d036de669;hp=215bd6ba1f7bf046a71f0df9bbfaa7b34f384101;hb=cc120c717cc6edb0291cc0cfbb8f2404daa57dbb;hpb=e7c89a22bd64fe2f1023b87728315b97cac715b8

diff --git a/bin/mailshears b/bin/mailshears
index 215bd6b..d9d1625 100755
--- a/bin/mailshears
+++ b/bin/mailshears
@@ -1,27 +1,19 @@
-#!/usr/bin/ruby -wKU
+#!/usr/bin/ruby -wU
 #
-# mailshears, to prune unused mail directories.
-#
-# Mail users for virtual hosts are stored in SQL, and managed by
-# Postfixadmin. However, the physical directories are handled by
-# Postfix/Dovecot and are left untouched by Postfixadmin. This is good
-# for security, but comes at a cost: Postfixadmin can't remove a
-# user's mail directory when his or her account is deleted.
-#
-# This program compares the list of filesystem users with the ones
-# in the database. It outputs any users that exist in the
-# filesystem, but not the database.
+# mailshears, to mangle your mail garden
 #
 
+# Load all of our lib/ code.
+require 'mailshears'
+
 # Define a usage string using the program name.
-exe = File.basename($PROGRAM_NAME)
-usage = "#{exe} [prune | rm <target> | mv <src> <dst>]"
+program_name = File.basename($PROGRAM_NAME)
 
 # Defaults
 mode_name = 'prune'
 mode = :prune
 
-# Now, if a mode was supplied, it should be in ARGV[0].
+# If a mode was supplied, it should be in ARGV[0].
 if ARGV.length() > 0
   mode_names = ['prune', 'rm', 'mv']
   if mode_names.include?(ARGV.first().downcase()) then
@@ -30,29 +22,39 @@ if ARGV.length() > 0
   end
 end
 
+# Determine the mode from its name.
 if mode_name == 'rm' then
   mode = :rm
 elsif mode_name == 'mv' then
   mode = :mv
 end
 
-# Need this before referencing ExitCodes.
-require 'mailshears'
+# Since we removed the mode name (if it existed) from ARGV, what
+# remains should be the required arguments. Figure out if we have the
+# wrong number of arguments, and store the associated error message in
+# args_error_message if necessary.
+args_error_message = nil
+
+if mode == :prune and ARGV.length() != 0 then
+  args_error_message = "ERROR: prune mode takes no additional arguments."
+elsif mode == :rm and ARGV.length() < 1 then
+  args_error_message = "ERROR: rm mode takes two or more user arguments."
+elsif mode == :mv and ARGV.length() != 2 then
+  args_error_message = "ERROR: mv mode takes exactly two user arguments."
+end
 
-# Since we removed both the executable name and the mode name (if it
-# existed) from ARGV, what remains should be the required
-# arguments.
-if (mode == :prune and (ARGV.length() != 0)) or
-    (mode == :rm and (ARGV.length() < 1))   or
-    (mode == :mv and (ARGV.length() != 2))   then
-  puts "ERROR: missing (or extra) command-line arguments."
-  puts "Usage: #{usage}"
+# If we got the wrong number of arguments, we'll have an error message
+# here. Report it and exit with a failure code.
+if not args_error_message.nil? then
+  STDERR.puts args_error_message
+  puts "Usage: #{UserInterface.usage(program_name)}"
   Kernel.exit(ExitCodes::BAD_COMMAND_LINE)
 end
 
-cfg = Configuration.new()
 
 # Load each of the plugins that we'll need.
+cfg = Configuration.new()
+
 cfg.plugins.each do |plugin_file|
   require "#{mode_name}/plugins/#{plugin_file}"
 end
@@ -61,28 +63,7 @@ end
 require "#{mode_name}/#{mode_name}_runner"
 require "#{mode_name}/#{mode_name}_dummy_runner"
 
-def make_header(plugin_name)
-  # The header that we output before the list of domains/users.
-  # Just the path of this script, the current time, and the plugin name.
-  exe = File.basename($PROGRAM_NAME)
-  header = "#{exe}, "
-
-  current_time = Time.now()
-  if current_time.respond_to?(:iso8601)
-    # Somehow this method is missing on some machines.
-    header += current_time.iso8601.to_s
-  else
-    # Fall back to whatever this looks like.
-    header += current_time.to_s
-  end
-
-  header += ' (Plugin: ' + plugin_name + ")\n"
-  header += '-' * header.size # Underline the header.
-
-  return header
-end
-
-
+# Now we figure out which plugin module to use based on our mode.
 plugin_module = nil
 
 if mode == :rm then
@@ -94,6 +75,25 @@ else
   plugin_module = PrunePlugin
 end
 
+# Parse the remaining arguments as User/Domain objects. If we get some
+# other argument that isn't one of those, it's an error.
+parsed_args = []
+
+ARGV.each do |arg|
+  begin
+    u = User.new(arg)
+    parsed_args << u
+  rescue InvalidUserError
+    begin
+      d = Domain.new(arg)
+      parsed_args << d
+    rescue InvalidDomainError
+      STDERR.puts "ERROR: invalid user/domain argument #{arg}"
+      Kernel.exit(ExitCodes::BAD_COMMAND_LINE)
+    end
+  end
+end
+
 
 # Buffer the output so that we can avoid printing the informational
 # header when no plugins produce output.
@@ -101,26 +101,24 @@ require 'stringio'
 output_buffer = StringIO.new()
 $stdout = output_buffer
 
-plugin_module.includers.each do |plugin_module_includer|
-  plugin = plugin_module_includer.new(cfg)
-
-  if cfg.i_mean_business then
-    runner = plugin.runner().new()
-  else
-    runner = plugin.dummy_runner().new()
+begin
+  plugin_module.run(cfg, *parsed_args)
+ensure
+  # Now restore stdout, and print the header plus whatever the plugins
+  # produced (if they produced anything). If they didn't produce any
+  # output, then we avoid printing the header.
+  #
+  # This gets wrapped in an "ensure" block because otherwise, if
+  # plugin_module.run() crashes, the traceback will get stored in
+  # output_buffer and never get printed.
+  $stdout = STDOUT
+
+  if output_buffer.size > 0 then
+    puts UserInterface.make_header(program_name, plugin_module.to_s())
+    puts output_buffer.string()
   end
-
-  # The splat passes the correct (we hope) number of arguments to the
-  # appropriate runner. The Rm(Dummy)Runner have splats on their
-  # *target arguments as well, to turn ARGV back into an array.
-  runner.run(plugin, *ARGV)
 end
 
-# Restore stdout, and print the header plus whatever the plugins
-# produced if they produced anything. If they didn't, we avoid
-# printing the header.
-$stdout = STDOUT
-if output_buffer.size > 0 then
-  puts make_header(plugin_module.to_s())
-  puts output_buffer.string()
-end
+# If we made it here without crashing, well that sounds pretty
+# successful to me.
+Kernel.exit(ExitCodes::SUCCESS)