| 1 |
#!/usr/bin/perl -w |
| 2 |
|
| 3 |
package MWS; |
| 4 |
|
| 5 |
use strict; |
| 6 |
use warnings; |
| 7 |
use Carp; |
| 8 |
|
| 9 |
use Mail::Box::Manager; |
| 10 |
use Config::IniFiles; |
| 11 |
use POSIX qw(strftime); |
| 12 |
use Text::Autoformat; |
| 13 |
use Text::Iconv; |
| 14 |
|
| 15 |
#use MWS_plucene; |
| 16 |
use MWS_swish; |
| 17 |
|
| 18 |
require Exporter; |
| 19 |
|
| 20 |
our @ISA = qw(Exporter); |
| 21 |
|
| 22 |
our %EXPORT_TAGS = (); |
| 23 |
our @EXPORT_OK; |
| 24 |
our @EXPORT; |
| 25 |
|
| 26 |
our $VERSION = '1.00'; |
| 27 |
|
| 28 |
my $folder; # placeholder for folders |
| 29 |
|
| 30 |
my $debug = 2; |
| 31 |
|
| 32 |
sub new { |
| 33 |
my $class = shift; |
| 34 |
my $self = {}; |
| 35 |
bless($self, $class); |
| 36 |
|
| 37 |
my $config_file = shift || die "need index file"; |
| 38 |
|
| 39 |
$self->{config} = new Config::IniFiles( -file => $config_file ); |
| 40 |
|
| 41 |
my $index_file = $self->{config}->val('global', 'index') || croak "can't find [index] section in config file with path of index"; |
| 42 |
|
| 43 |
$self->{mgr} = Mail::Box::Manager->new(access => 'r'); |
| 44 |
$self->{index_file} = $index_file; |
| 45 |
|
| 46 |
# placeholder for opened folders |
| 47 |
$self->{folder} = {}; |
| 48 |
|
| 49 |
$self->{wrap_margin} = $self->{config}->val('global', 'wrap_margin'); |
| 50 |
$self->{max_results} = $self->{config}->val('global', 'max_results') || 100; |
| 51 |
|
| 52 |
return $self; |
| 53 |
} |
| 54 |
|
| 55 |
sub mbox_name2path { |
| 56 |
my $self = shift; |
| 57 |
|
| 58 |
my $mbox = shift || croak "folder_name2path needs mbox name"; |
| 59 |
|
| 60 |
return $self->{config}->val('folders', $mbox) || croak "comeone removed folder $mbox from config?"; |
| 61 |
} |
| 62 |
|
| 63 |
sub open_folder { |
| 64 |
my $self = shift; |
| 65 |
|
| 66 |
my $mbox = shift || croak "open_folder needs mbox name"; |
| 67 |
|
| 68 |
if (! $self->{folder}->{$mbox}) { |
| 69 |
my $mbox_path = $self->mbox_name2path($mbox); |
| 70 |
|
| 71 |
print STDERR "about to open_folder($mbox)\n" if ($debug == 2); |
| 72 |
|
| 73 |
$self->{folder}->{$mbox} = $self->{mgr}->open($mbox_path) || croak "can't open folder $mbox at '$mbox_path': $!"; |
| 74 |
|
| 75 |
print STDERR "open_folder($mbox) ok\n" if ($debug); |
| 76 |
} |
| 77 |
|
| 78 |
$self->{fetch_count} = 0; |
| 79 |
|
| 80 |
return $self->{folder}->{$mbox}; |
| 81 |
|
| 82 |
} |
| 83 |
|
| 84 |
sub close_folder { |
| 85 |
my $self = shift; |
| 86 |
|
| 87 |
my $mbox = shift || croak "open_folder needs mbox name"; |
| 88 |
|
| 89 |
$self->{folder}->{$mbox}->close(write => 'NEVER') || croak "can't close folder $mbox"; |
| 90 |
|
| 91 |
# XXX this is rather agressive!!! |
| 92 |
$self->{folder} = {}; |
| 93 |
return |
| 94 |
} |
| 95 |
|
| 96 |
sub fetch_message { |
| 97 |
my $self = shift; |
| 98 |
|
| 99 |
my $mbox_id = shift || die "need mbox_id!"; |
| 100 |
my ($mbox,$id) = split(/ /,$mbox_id); |
| 101 |
|
| 102 |
# return message with ID |
| 103 |
print STDERR "fetch $id from $mbox\n" if ($debug); |
| 104 |
|
| 105 |
if ($self->{fetch_count}++ > 100) { |
| 106 |
$self->close_folder($mbox); |
| 107 |
print STDERR "close_folder($mbox) forced on ",$self->{fetch_count},"iteration\n"; |
| 108 |
} |
| 109 |
|
| 110 |
my $msg = $self->open_folder($mbox)->find($id); |
| 111 |
if ($msg) { |
| 112 |
return $msg; |
| 113 |
} else { |
| 114 |
print STDERR "can't find message $id in $mbox. Time to re-index?\n"; |
| 115 |
return; |
| 116 |
} |
| 117 |
} |
| 118 |
|
| 119 |
|
| 120 |
sub search { |
| 121 |
my $self = shift; |
| 122 |
|
| 123 |
my $s = shift || carp "search called without argument!"; |
| 124 |
|
| 125 |
print STDERR "search_index($s)\n" if ($debug == 2); |
| 126 |
my @index_ids = $self->search_index($s); |
| 127 |
|
| 128 |
$self->{'index_ids'} = \@index_ids; |
| 129 |
|
| 130 |
#my $results = $#index_ids + 1; |
| 131 |
#$self->{'results'} = $results; |
| 132 |
|
| 133 |
my $results = $self->{'total_hits'} || ($#index_ids + 1); |
| 134 |
|
| 135 |
$self->{'curr_result'} = 0; |
| 136 |
|
| 137 |
print STDERR "$results results\n" if ($debug == 2); |
| 138 |
|
| 139 |
return $results || 'error'; |
| 140 |
} |
| 141 |
|
| 142 |
sub decode_qp($) { |
| 143 |
my $self = shift; |
| 144 |
|
| 145 |
my $tmp = shift || return; |
| 146 |
|
| 147 |
sub decode($$) { |
| 148 |
my ($cp,$qp) = @_; |
| 149 |
my $iconv = Text::Iconv->new($cp,'ISO-8859-2'); |
| 150 |
print STDERR "decode($cp,$qp) -> " if ($debug == 2); |
| 151 |
$qp =~ s/=([a-f0-9][a-f0-9])/chr(hex($1))/ieg; |
| 152 |
$qp =~ s/_/ /g; |
| 153 |
print STDERR "$qp\n" if ($debug == 2); |
| 154 |
return $iconv->convert($qp) || $qp; |
| 155 |
} |
| 156 |
|
| 157 |
$tmp =~ s/=\?([^\?]+)\?Q\?(.+?)\?=/decode($1,$2)/ex; |
| 158 |
$tmp =~ s/^\s*["']+(.*?)["']+\s*$/$1/g; |
| 159 |
return $tmp; |
| 160 |
} |
| 161 |
|
| 162 |
sub unroll($$$) { |
| 163 |
my $self = shift; |
| 164 |
|
| 165 |
my ($message,$part,$sub) = @_; |
| 166 |
|
| 167 |
my @arr; |
| 168 |
|
| 169 |
return if (! $message->$part); |
| 170 |
|
| 171 |
foreach my $from ($message->$part) { |
| 172 |
my $tmp = $from->$sub || next; |
| 173 |
|
| 174 |
$tmp = $self->decode_qp($tmp); |
| 175 |
push @arr, $tmp; |
| 176 |
} |
| 177 |
|
| 178 |
return @arr; |
| 179 |
} |
| 180 |
|
| 181 |
sub fetch_all_results { |
| 182 |
my $self = shift; |
| 183 |
|
| 184 |
croak "results called before search!" if (! $self->{'index_ids'}); |
| 185 |
|
| 186 |
print STDERR "fetch_all_results [",scalar @{$self->{'index_ids'}},"]\n" if ($debug == 2); |
| 187 |
|
| 188 |
my @arr; |
| 189 |
|
| 190 |
foreach my $id (@{$self->{'index_ids'}}) { |
| 191 |
push @arr, $self->fetch_result_by_id($id); |
| 192 |
} |
| 193 |
|
| 194 |
return @arr; |
| 195 |
} |
| 196 |
|
| 197 |
sub fetch_result { |
| 198 |
my $self = shift; |
| 199 |
|
| 200 |
my $args = {@_}; |
| 201 |
|
| 202 |
croak "results called before search!" if (! $self->{'index_ids'}); |
| 203 |
|
| 204 |
my $curr = $self->{'curr_result'}++; |
| 205 |
|
| 206 |
my $id = $self->{'index_ids'}->[$curr]; |
| 207 |
|
| 208 |
print STDERR "fetch_result: $curr = $id\n" if ($debug == 2); |
| 209 |
|
| 210 |
return $self->fetch_result_by_id($id); |
| 211 |
} |
| 212 |
|
| 213 |
sub plain_text_body { |
| 214 |
my $self = shift; |
| 215 |
my $message = shift || croak "plain_text_body needs message!"; |
| 216 |
|
| 217 |
my $body; |
| 218 |
|
| 219 |
if (! $message->isMultipart) { |
| 220 |
$body = $message->decoded->string; |
| 221 |
} else { |
| 222 |
foreach my $part ($message->parts) { |
| 223 |
if ($part->body->mimeType eq 'text/plain') { |
| 224 |
$body = $part->decoded->string; |
| 225 |
last; |
| 226 |
} |
| 227 |
} |
| 228 |
} |
| 229 |
|
| 230 |
# reformat with Text::Autoformat |
| 231 |
my $wrap = $self->{wrap_margin}; |
| 232 |
if ($wrap && $body && $body =~ m/^.{$wrap}..*$/m) { |
| 233 |
$body =~ s/[\r\n]/\n/gs; |
| 234 |
$body = autoformat($body, {right=>$wrap}); |
| 235 |
$body .="\n[reformated using autoformat, margin at $wrap]" if ($debug == 2); |
| 236 |
} |
| 237 |
|
| 238 |
return $body; |
| 239 |
} |
| 240 |
|
| 241 |
|
| 242 |
sub fetch_result_by_id { |
| 243 |
my $self = shift; |
| 244 |
|
| 245 |
my $id = shift || return; |
| 246 |
|
| 247 |
my $row = $self->{cache}->{$id}; |
| 248 |
|
| 249 |
if (! $row) { |
| 250 |
|
| 251 |
print STDERR "fetch_result_by_id($id) not in cache, hitting disk\n" if ($debug == 2); |
| 252 |
|
| 253 |
my $message = $self->fetch_message($id) || return; |
| 254 |
|
| 255 |
$row->{'id'} = $id; |
| 256 |
@{$row->{'from'}} = $self->unroll($message,'from','phrase'); |
| 257 |
@{$row->{'to'}} = $self->unroll($message,'to','phrase'); |
| 258 |
@{$row->{'cc'}} = $self->unroll($message,'cc','phrase'); |
| 259 |
$row->{'subject'} = $self->decode_qp($message->subject); |
| 260 |
$row->{'body'} = $self->plain_text_body($message); |
| 261 |
$row->{'date'} = $message->date; |
| 262 |
|
| 263 |
# XXX store in cache? |
| 264 |
$self->{cache}->{$id} = $row; |
| 265 |
print STDERR "$id stored in cache\n" if ($debug == 2); |
| 266 |
} else { |
| 267 |
print STDERR "fetch_result_by_id($id) in cache\n" if ($debug == 2); |
| 268 |
} |
| 269 |
|
| 270 |
return $row; |
| 271 |
|
| 272 |
} |
Parent Directory
| 
Revision Log