# Copyright (C) 2015-2018 all contributors
# License: AGPL-3.0+
# based on notmuch, but with no concept of folders, files or flags
#
# Wraps a document inside our Xapian search index.
package PublicInbox::SearchMsg;
use strict;
use warnings;
use PublicInbox::MID qw/mid_clean mid_mime/;
use PublicInbox::Address;
use PublicInbox::MsgTime qw(msg_timestamp msg_datestamp);
use Time::Local qw(timegm);
sub new {
my ($class, $mime) = @_;
my $doc = Search::Xapian::Document->new;
bless { doc => $doc, mime => $mime }, $class;
}
sub wrap {
my ($class, $doc, $mid) = @_;
bless { doc => $doc, mime => undef, mid => $mid }, $class;
}
sub get {
my ($class, $head, $db, $mid) = @_;
my $doc_id = $head->get_docid;
my $doc = $db->get_document($doc_id);
load_expand(wrap($class, $doc, $mid))
}
sub get_val ($$) {
my ($doc, $col) = @_;
Search::Xapian::sortable_unserialise($doc->get_value($col));
}
sub to_doc_data {
my ($self, $oid, $mid0) = @_;
$oid = '' unless defined $oid;
join("\n",
$self->subject,
$self->from,
$self->references,
$self->to,
$self->cc,
$oid,
$mid0,
$self->{bytes},
$self->{lines}
);
}
sub load_from_data ($$) {
my ($self) = $_[0]; # data = $_[1]
(
$self->{subject},
$self->{from},
$self->{references},
# To: and Cc: are stored to optimize HDR/XHDR in NNTP since
# some NNTP clients will use that for message displays.
$self->{to},
$self->{cc},
$self->{blob},
$self->{mid},
$self->{bytes},
$self->{lines}
) = split(/\n/, $_[1]);
}
sub load_expand {
my ($self) = @_;
my $doc = $self->{doc};
my $data = $doc->get_data or return;
$self->{ts} = get_val($doc, PublicInbox::Search::TS());
my $dt = get_val($doc, PublicInbox::Search::DT());
my ($yyyy, $mon, $dd, $hh, $mm, $ss) = unpack('A4A2A2A2A2A2', $dt);
$self->{ds} = timegm($ss, $mm, $hh, $dd, $mon - 1, $yyyy);
utf8::decode($data);
load_from_data($self, $data);
$self;
}
sub load_doc {
my ($class, $doc) = @_;
my $self = bless { doc => $doc }, $class;
$self->load_expand;
}
# :bytes and :lines metadata in RFC 3977
sub bytes ($) { $_[0]->{bytes} }
sub lines ($) { $_[0]->{lines} }
sub num ($) { $_[0]->{num} ||= _get_term_val($_[0], 'XNUM', qr/\AXNUM/) }
sub __hdr ($$) {
my ($self, $field) = @_;
my $val = $self->{$field};
return $val if defined $val;
my $mime = $self->{mime} or return;
$val = $mime->header($field);
$val = '' unless defined $val;
$val =~ tr/\n/ /;
$val =~ tr/\r//d;
$self->{$field} = $val;
}
sub subject ($) { __hdr($_[0], 'subject') }
sub to ($) { __hdr($_[0], 'to') }
sub cc ($) { __hdr($_[0], 'cc') }
# no strftime, that is locale-dependent and not for RFC822
my @DoW = qw(Sun Mon Tue Wed Thu Fri Sat);
my @MoY = qw(Jan Feb Mar Apr May Jun Jul Aug Sep Oct Nov Dec);
sub date ($) {
my ($self) = @_;
my $ds = $self->{ds};
return unless defined $ds;
my ($sec, $min, $hour, $mday, $mon, $year, $wday) = gmtime($ds);
"$DoW[$wday], " . sprintf("%02d $MoY[$mon] %04d %02d:%02d:%02d +0000",
$mday, $year+1900, $hour, $min, $sec);
}
sub from ($) {
my ($self) = @_;
my $from = __hdr($self, 'from');
if (defined $from && !defined $self->{from_name}) {
my @n = PublicInbox::Address::names($from);
$self->{from_name} = join(', ', @n);
}
$from;
}
sub from_name {
my ($self) = @_;
my $from_name = $self->{from_name};
return $from_name if defined $from_name;
$self->from;
$self->{from_name};
}
sub ts {
my ($self) = @_;
$self->{ts} ||= eval { msg_timestamp($self->{mime}->header_obj) } || 0;
}
sub ds {
my ($self) = @_;
$self->{ds} ||= eval { msg_datestamp($self->{mime}->header_obj); } || 0;
}
sub references {
my ($self) = @_;
my $x = $self->{references};
defined $x ? $x : '';
}
sub _get_term_val ($$$) {
my ($self, $pfx, $re) = @_;
my $doc = $self->{doc};
my $end = $doc->termlist_end;
my $i = $doc->termlist_begin;
$i->skip_to($pfx);
if ($i != $end) {
my $val = $i->get_termname;
$val =~ s/$re// and return $val;
}
undef;
}
sub mid ($;$) {
my ($self, $mid) = @_;
if (defined $mid) {
$self->{mid} = $mid;
} elsif (my $rv = $self->{mid}) {
$rv;
} elsif ($self->{doc}) {
$self->{mid} = _get_term_val($self, 'Q', qr/\AQ/);
} else {
$self->_extract_mid; # v1 w/o Xapian
}
}
sub _extract_mid { mid_clean(mid_mime($_[0]->{mime})) }
1;