mws/trunk/MWS.pm

#!/usr/bin/perl -w

package MWS;

use strict;
use warnings;
use Carp;

use Mail::Box::Manager;
use Config::IniFiles;
use POSIX qw(strftime);
#use MWS_plucene;
use MWS_swish;

require Exporter;

our @ISA = qw(Exporter);

our %EXPORT_TAGS = ();
our @EXPORT_OK;
our @EXPORT;

our $VERSION = '1.00';


my $folder;     # placeholder for folders

my $debug = 2;

sub new {
        my $class = shift;
        my $self = {};
        bless($self, $class);

        my $config_file = shift || die "need index file";

        $self->{config} = new Config::IniFiles( -file => $config_file );

        my $index_file = $self->{config}->val('global', 'index') || croak "can't find [index] section in config file with path of index";

        $self->{mgr} = Mail::Box::Manager->new(access => 'r');
        $self->{index_file} = $index_file;

        # placeholder for opened folders
        $self->{folder} = {};

        return $self;
}

sub open_folder {
        my $self = shift;

        my $mbox = shift || croak "open_folder needs mbox name";

        if (! $self->{folder}->{$mbox}) {
                my $mbox_path = $self->{config}->val('folders', $mbox) || croak "comeone removed folder $mbox from config?";
                print STDERR "about to open_folder($mbox)\n" if ($debug == 2);
                $self->{folder}->{$mbox} = $self->{mgr}->open($mbox_path) || croak "can't open folder $mbox at '$mbox_path': $!";
                print STDERR "open_folder($mbox)\n" if ($debug);
        }

        return $self->{folder}->{$mbox};

}

sub fetch_message {
        my $self = shift;

        my $mbox_id = shift || die "need mbox_id!";
        my ($mbox,$id) = split(/ /,$mbox_id);

        # return message with ID
        print STDERR "fetch $id from $mbox\n" if ($debug);
        return $self->open_folder($mbox)->find($id) ||
                print STDERR "can't find message $id in $mbox. Time to re-index?\n";
}


sub search {
        my $self = shift;

        my $s = shift || carp "search called without argument!";

        print STDERR "search_index($s)\n" if ($debug == 2);
        my @index_ids = $self->search_index($s);

        $self->{'index_ids'} = \@index_ids;

        my $results = $#index_ids + 1;
        $self->{'results'} = $results;

        $self->{'curr_result'} = 0;

        print STDERR "$results results\n" if ($debug == 2);

        return $results || 'error';
}

sub unroll($$$) {
        my $self = shift;

        my ($message,$part,$sub) = @_;

        my @arr;

        foreach my $from ($message->$part) {
                my $tmp = $from->$sub;
                $tmp =~ s/^\s*["'](.*)["']\s*$/$1/;
                push @arr, $tmp;
        }
        return \@arr;
}

sub fetch_all_results {
        my $self = shift;

        croak "results called before search!" if (! $self->{'index_ids'});

        print STDERR "fetch_all_results_results\n" if ($debug == 2);

        my @arr;

        foreach my $id (@{$self->{'index_ids'}}) {
                push @arr, $self->fetch_result_by_id($id);
        }

        return @arr;
}

sub fetch_result {
        my $self = shift;

        my $args = {@_};

        croak "results called before search!" if (! $self->{'index_ids'});

        my $curr = $self->{'curr_result'}++;

        my $id = $self->{'index_ids'}->[$curr];

        return $self->fetch_result_by_id($id);
}

sub plain_text_body {
        my $self = shift;
        my $message = shift || croak "plain_text_body needs message!";

        if (! $message->isMultipart) {
                return $message->decoded->string;
        } else {
                foreach my $part ($message->parts) {
                        if ($part->body->mimeType eq 'text/plain') {
                                return $part->decoded->string;
                        }
                }
        }
}


sub fetch_result_by_id {
        my $self = shift;

        my $id = shift || return;

        my $row = $self->{cache}->{$id};

        if (! $row) {

                print STDERR "fetch_result_by_id($id) not in cache, hitting disk\n" if ($debug == 2);

                my $message = $self->fetch_message($id) || print STDERR "can't fetch message '$id'";

                $row->{'id'} = $id;
                $row->{'from'} = $self->unroll($message,'from','phrase');
                $row->{'to'} = $self->unroll($message,'to','phrase');
                $row->{'cc'} = $self->unroll($message,'cc','phrase');
                $row->{'subject'} = $message->subject;
                $row->{'body'} = $self->plain_text_body($message);
                $row->{'date'} = $message->date;

                # XXX store in cache?
                $self->{cache}->{$id} = $row;
                print STDERR "$id stored in cache\n" if ($debug == 2);
        } else {
                print STDERR "fetch_result_by_id($id) in cache\n" if ($debug == 2);
        }

        return $row;
        
}
1	dpavlin	2	#!/usr/bin/perl -w
2
3			package MWS;
4
5			use strict;
6			use warnings;
7			use Carp;
8
9	dpavlin	3	use Mail::Box::Manager;
10	dpavlin	4	use Config::IniFiles;
11	dpavlin	13	use POSIX qw(strftime);
12	dpavlin	12	#use MWS_plucene;
13			use MWS_swish;
14	dpavlin	3
15	dpavlin	2	require Exporter;
16
17			our @ISA = qw(Exporter);
18
19			our %EXPORT_TAGS = ();
20			our @EXPORT_OK;
21			our @EXPORT;
22
23			our $VERSION = '1.00';
24
25
26			my $folder; # placeholder for folders
27
28	dpavlin	13	my $debug = 2;
29	dpavlin	2
30			sub new {
31			my $class = shift;
32			my $self = {};
33			bless($self, $class);
34
35	dpavlin	4	my $config_file = shift \|\| die "need index file";
36	dpavlin	2
37	dpavlin	4	$self->{config} = new Config::IniFiles( -file => $config_file );
38	dpavlin	2
39	dpavlin	4	my $index_file = $self->{config}->val('global', 'index') \|\| croak "can't find [index] section in config file with path of index";
40
41	dpavlin	12	$self->{mgr} = Mail::Box::Manager->new(access => 'r');
42			$self->{index_file} = $index_file;
43	dpavlin	4
44	dpavlin	2	# placeholder for opened folders
45			$self->{folder} = {};
46
47			return $self;
48			}
49
50	dpavlin	7	sub open_folder {
51	dpavlin	2	my $self = shift;
52
53	dpavlin	7	my $mbox = shift \|\| croak "open_folder needs mbox name";
54	dpavlin	2
55			if (! $self->{folder}->{$mbox}) {
56	dpavlin	4	my $mbox_path = $self->{config}->val('folders', $mbox) \|\| croak "comeone removed folder $mbox from config?";
57	dpavlin	13	print STDERR "about to open_folder($mbox)\n" if ($debug == 2);
58	dpavlin	4	$self->{folder}->{$mbox} = $self->{mgr}->open($mbox_path) \|\| croak "can't open folder $mbox at '$mbox_path': $!";
59	dpavlin	13	print STDERR "open_folder($mbox)\n" if ($debug);
60	dpavlin	2	}
61
62	dpavlin	7	return $self->{folder}->{$mbox};
63
64			}
65
66			sub fetch_message {
67			my $self = shift;
68
69			my $mbox_id = shift \|\| die "need mbox_id!";
70			my ($mbox,$id) = split(/ /,$mbox_id);
71
72			# return message with ID
73	dpavlin	13	print STDERR "fetch $id from $mbox\n" if ($debug);
74	dpavlin	7	return $self->open_folder($mbox)->find($id) \|\|
75	dpavlin	2	print STDERR "can't find message $id in $mbox. Time to re-index?\n";
76			}
77
78
79			sub search {
80			my $self = shift;
81
82			my $s = shift \|\| carp "search called without argument!";
83
84	dpavlin	13	print STDERR "search_index($s)\n" if ($debug == 2);
85	dpavlin	9	my @index_ids = $self->search_index($s);
86	dpavlin	2
87			$self->{'index_ids'} = \@index_ids;
88
89			my $results = $#index_ids + 1;
90			$self->{'results'} = $results;
91
92			$self->{'curr_result'} = 0;
93
94	dpavlin	13	print STDERR "$results results\n" if ($debug == 2);
95
96	dpavlin	2	return $results \|\| 'error';
97			}
98
99			sub unroll($$$) {
100	dpavlin	13	my $self = shift;
101
102	dpavlin	2	my ($message,$part,$sub) = @_;
103
104			my @arr;
105
106			foreach my $from ($message->$part) {
107	dpavlin	7	my $tmp = $from->$sub;
108			$tmp =~ s/^\s["'](.)["']\s*$/$1/;
109			push @arr, $tmp;
110	dpavlin	2	}
111			return \@arr;
112			}
113	dpavlin	4
114			sub fetch_all_results {
115			my $self = shift;
116
117			croak "results called before search!" if (! $self->{'index_ids'});
118
119	dpavlin	13	print STDERR "fetch_all_results_results\n" if ($debug == 2);
120
121	dpavlin	4	my @arr;
122
123			foreach my $id (@{$self->{'index_ids'}}) {
124			push @arr, $self->fetch_result_by_id($id);
125			}
126
127			return @arr;
128			}
129
130	dpavlin	2	sub fetch_result {
131			my $self = shift;
132
133	dpavlin	3	my $args = {@_};
134
135	dpavlin	2	croak "results called before search!" if (! $self->{'index_ids'});
136
137			my $curr = $self->{'curr_result'}++;
138
139			my $id = $self->{'index_ids'}->[$curr];
140
141	dpavlin	4	return $self->fetch_result_by_id($id);
142			}
143	dpavlin	2
144	dpavlin	7	sub plain_text_body {
145			my $self = shift;
146			my $message = shift \|\| croak "plain_text_body needs message!";
147
148			if (! $message->isMultipart) {
149			return $message->decoded->string;
150			} else {
151			foreach my $part ($message->parts) {
152			if ($part->body->mimeType eq 'text/plain') {
153			return $part->decoded->string;
154			}
155			}
156			}
157			}
158
159
160	dpavlin	4	sub fetch_result_by_id {
161			my $self = shift;
162
163			my $id = shift \|\| return;
164
165	dpavlin	13	my $row = $self->{cache}->{$id};
166	dpavlin	2
167	dpavlin	13	if (! $row) {
168	dpavlin	2
169	dpavlin	13	print STDERR "fetch_result_by_id($id) not in cache, hitting disk\n" if ($debug == 2);
170	dpavlin	2
171	dpavlin	13	my $message = $self->fetch_message($id) \|\| print STDERR "can't fetch message '$id'";
172
173			$row->{'id'} = $id;
174			$row->{'from'} = $self->unroll($message,'from','phrase');
175			$row->{'to'} = $self->unroll($message,'to','phrase');
176			$row->{'cc'} = $self->unroll($message,'cc','phrase');
177			$row->{'subject'} = $message->subject;
178			$row->{'body'} = $self->plain_text_body($message);
179			$row->{'date'} = $message->date;
180
181			# XXX store in cache?
182			$self->{cache}->{$id} = $row;
183			print STDERR "$id stored in cache\n" if ($debug == 2);
184			} else {
185			print STDERR "fetch_result_by_id($id) in cache\n" if ($debug == 2);
186			}
187
188	dpavlin	2	return $row;
189
190			}