#!/usr/bin/ruby

# Copyright (c) 2006 Peter Palfrader <peter@palfrader.org>
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

SPAMIN="mail/spam-learn"
SPAMOUT="mail/spam-learned"
HAMIN="mail/ham-learn"
HAMOUT="mail/ham-learned"

#PREFIX = "sudo -H -u amavis "
PREFIX = ""

$verbose = 0
if ARGV[0] == "-v"
	$verbose = 1
end

def check_maildir(d)
	throw "#{d} is not a maildir" unless FileTest.directory?(d)
	throw "#{d} is not a maildir" unless FileTest.directory?(d+"/new")
	throw "#{d} is not a maildir" unless FileTest.directory?(d+"/cur")
	throw "#{d} is not a maildir" unless FileTest.directory?(d+"/tmp")
	true
end

def move(filename, to)
	basename = File.basename(filename)
	component = File.basename( File.dirname(filename) )
	throw "Bad component '#{component}' in filename '#{filename}'." unless %w{cur new tmp}.include?(component)
	File.link(filename, to+"/"+component+"/"+basename)
	File.unlink(filename)
end

check_maildir SPAMIN
check_maildir SPAMOUT
check_maildir HAMIN
check_maildir HAMOUT

Dir[SPAMIN+"/new/*"].each do |filename|
	puts "learning #{filename}" if $verbose > 0
	if system("#{PREFIX}sa-learn --spam < #{filename} > /dev/null")
		if system("#{PREFIX}bogofilter --register-spam < #{filename} > /dev/null")
			puts "moving #{filename}" if $verbose > 0
			move(filename, SPAMOUT)
		else
			STDERR.puts "Learning of #{filename} failed (bogofilter)."
		end
	else
		STDERR.puts "Learning of #{filename} failed (SA)."
	end
end
Dir[HAMIN+"/new/*"].each do |filename|
	puts "learning #{filename}" if $verbose > 0
	if system("#{PREFIX}sa-learn --ham < #{filename} > /dev/null")
		if system("#{PREFIX}bogofilter --register-ham < #{filename} > /dev/null")
			puts "moving #{filename}" if $verbose > 0
			move(filename, HAMOUT)
		else
			STDERR.puts "Learning of #{filename} failed (bogofilter)."
		end
	else
		STDERR.puts "Learning of #{filename} failed (SA)."
	end
end