blob: 6c218eed65262f2fb890d71eecd2fbe3dd8c1c0b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
#!/usr/bin/ruby
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
|
