Add a maximum for mails to keep

inspired by (stolen from) sj26#375

lib/mail_catcher.rb

@@ -89,6 +89,7 @@ def log_exception(message, context, exception)
     :http_ip => "127.0.0.1",
     :http_port => "1080",
     :http_path => "/",
+    :max_mail_to_keep => -1,
     :verbose => false,
     :daemon => !windows?,
     :browse => false,
@@ -129,6 +130,10 @@ def parse! arguments=ARGV, defaults=@defaults
           options[:http_port] = port
         end
 
+        parser.on("--max-mail-to-keep NUM", Integer, "Set the maximum number of mails to keep") do |num|
+          options[:max_mail_to_keep] = num
+        end
+
         parser.on("--http-path PATH", String, "Add a prefix to all HTTP paths") do |path|
           clean_path = Rack::Utils.clean_path_info("/#{path}")
 

lib/mail_catcher/mail.rb

@@ -53,6 +53,7 @@ def add_message(message)
       cid = part.cid if part.respond_to? :cid
       add_message_part(message_id, cid, part.mime_type || "text/plain", part.attachment? ? 1 : 0, part.filename, part.charset, body, body.length)
     end
+    delete_older_messages!
 
     EventMachine.next_tick do
       message = MailCatcher::Mail.message message_id
@@ -165,4 +166,13 @@ def delete_message!(message_id)
     @delete_messages_query.execute(message_id) and
     @delete_message_parts_query.execute(message_id)
   end
+
+  def delete_older_messages!(max_mail_to_keep = MailCatcher::options[:max_mail_to_keep])
+    return if max_mail_to_keep <= 0
+    @delete_older_messages_query ||= db.prepare "DELETE FROM message WHERE id NOT IN (SELECT id FROM message ORDER BY created_at DESC LIMIT ?)"
+    @delete_older_message_parts_query ||= db.prepare "DELETE FROM message_part WHERE message_id NOT IN (SELECT id FROM message ORDER BY created_at DESC LIMIT ?)"
+
+    @delete_older_message_parts_query.execute(max_mail_to_keep) and
+    @delete_older_messages_query.execute(max_mail_to_keep)
+  end
 end