mws/trunk/MWS.pm

#!/usr/bin/perl -w

package MWS;

use strict;
use warnings;
use Carp;

use Plucene::Simple;
use Mail::Box::Manager;
use Config::IniFiles;

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 = 1;

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->{index} = Plucene::Simple->open($index_file) || croak "can't open index '$index_file': $!";

        $self->{mgr} = Mail::Box::Manager->new;

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

        return $self;
}


sub fetch_message {
        my $self = shift;

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

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

        my $message = $self->{folder}->{$mbox}->find($id) ||
                print STDERR "can't find message $id in $mbox. Time to re-index?\n";
        
        return $message;
}


sub search {
        my $self = shift;

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

        my @index_ids = $self->{index}->search($s);

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

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

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

        return $results || 'error';
}

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

        my @arr;

        foreach my $from ($message->$part) {
                push @arr, $from->$sub;
        }
        return \@arr;
}

sub fetch_all_results {
        my $self = shift;

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

        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 fetch_result_by_id {
        my $self = shift;

        my $id = shift || return;

        my $message = $self->fetch_message($id);

        my $row;

        $row->{'from'} = unroll($message,'from','phrase');
        $row->{'subject'} = $message->get('Subject');
        if (! $message->isMultipart) {
                $row->{'body'} = $message->decoded->string;
        } else {
                foreach my $part ($message->parts) {
                        if ($part->body->mimeType eq 'text/plain') {
                                $row->{'body'} = $part->decoded->string;
                                last;
                        }
                }
        }

        return $row;
        
}
1	#!/usr/bin/perl -w
2
3	package MWS;
4
5	use strict;
6	use warnings;
7	use Carp;
8
9	use Plucene::Simple;
10	use Mail::Box::Manager;
11	use Config::IniFiles;
12
13	require Exporter;
14
15	our @ISA = qw(Exporter);
16
17	our %EXPORT_TAGS = ();
18	our @EXPORT_OK;
19	our @EXPORT;
20
21	our $VERSION = '1.00';
22
23
24	my $folder; # placeholder for folders
25
26	my $debug = 1;
27
28	sub new {
29	my $class = shift;
30	my $self = {};
31	bless($self, $class);
32
33	my $config_file = shift \|\| die "need index file";
34
35	$self->{config} = new Config::IniFiles( -file => $config_file );
36
37	my $index_file = $self->{config}->val('global', 'index') \|\| croak "can't find [index] section in config file with path of index";
38
39	$self->{index} = Plucene::Simple->open($index_file) \|\| croak "can't open index '$index_file': $!";
40
41	$self->{mgr} = Mail::Box::Manager->new;
42
43	# placeholder for opened folders
44	$self->{folder} = {};
45
46	return $self;
47	}
48
49
50	sub fetch_message {
51	my $self = shift;
52
53	my $mbox_id = shift \|\| die "need mbox_id!";
54	my ($mbox,$id) = split(/ /,$mbox_id);
55
56	if (! $self->{folder}->{$mbox}) {
57	my $mbox_path = $self->{config}->val('folders', $mbox) \|\| croak "comeone removed folder $mbox from config?";
58	$self->{folder}->{$mbox} = $self->{mgr}->open($mbox_path) \|\| croak "can't open folder $mbox at '$mbox_path': $!";
59	print STDERR "## open($mbox)\n" if ($debug);
60	}
61
62	my $message = $self->{folder}->{$mbox}->find($id) \|\|
63	print STDERR "can't find message $id in $mbox. Time to re-index?\n";
64
65	return $message;
66	}
67
68
69	sub search {
70	my $self = shift;
71
72	my $s = shift \|\| carp "search called without argument!";
73
74	my @index_ids = $self->{index}->search($s);
75
76	$self->{'index_ids'} = \@index_ids;
77
78	my $results = $#index_ids + 1;
79	$self->{'results'} = $results;
80
81	$self->{'curr_result'} = 0;
82
83	return $results \|\| 'error';
84	}
85
86	sub unroll($$$) {
87	my ($message,$part,$sub) = @_;
88
89	my @arr;
90
91	foreach my $from ($message->$part) {
92	push @arr, $from->$sub;
93	}
94	return \@arr;
95	}
96
97	sub fetch_all_results {
98	my $self = shift;
99
100	croak "results called before search!" if (! $self->{'index_ids'});
101
102	my @arr;
103
104	foreach my $id (@{$self->{'index_ids'}}) {
105	push @arr, $self->fetch_result_by_id($id);
106	}
107
108	return @arr;
109	}
110
111	sub fetch_result {
112	my $self = shift;
113
114	my $args = {@_};
115
116	croak "results called before search!" if (! $self->{'index_ids'});
117
118	my $curr = $self->{'curr_result'}++;
119
120	my $id = $self->{'index_ids'}->[$curr];
121
122	return $self->fetch_result_by_id($id);
123	}
124
125	sub fetch_result_by_id {
126	my $self = shift;
127
128	my $id = shift \|\| return;
129
130	my $message = $self->fetch_message($id);
131
132	my $row;
133
134	$row->{'from'} = unroll($message,'from','phrase');
135	$row->{'subject'} = $message->get('Subject');
136	if (! $message->isMultipart) {
137	$row->{'body'} = $message->decoded->string;
138	} else {
139	foreach my $part ($message->parts) {
140	if ($part->body->mimeType eq 'text/plain') {
141	$row->{'body'} = $part->decoded->string;
142	last;
143	}
144	}
145	}
146
147	return $row;
148
149	}