introduce PublicInbox::MIME wrapper class

[public-inbox.git] / lib / PublicInbox / SearchView.pm

diff --git a/lib/PublicInbox/SearchView.pm b/lib/PublicInbox/SearchView.pm
index 50a2c01c4749f05ab54dc73a5d05ee04747d4884..ccc53abf23e4f5a0a24d4d07b56fc648b857932a 100644 (file)
--- a/lib/PublicInbox/SearchView.pm
+++ b/lib/PublicInbox/SearchView.pm
@@ -10,7 +10,7 @@ use PublicInbox::Hval qw/ascii_html/;
 use PublicInbox::View;
 use PublicInbox::WwwAtomStream;
 use PublicInbox::MID qw(mid2path mid_mime mid_clean mid_escape);
-use Email::MIME;
+use PublicInbox::MIME;
 require PublicInbox::Git;
 require PublicInbox::SearchThread;
 our $LIM = 50;
@@ -161,6 +161,15 @@ sub search_nav_bot {
        $rv .= '</pre>';
 }
 
+sub sort_relevance {
+       my ($pct) = @_;
+       sub {
+               [ sort { (eval { $pct->{$b->topmost->{id}} } || 0)
+                               <=>
+                       (eval { $pct->{$a->topmost->{id}} } || 0)
+       } @{$_[0]} ] };
+}
+
 sub mset_thread {
        my ($ctx, $mset, $q) = @_;
        my %pct;
@@ -171,18 +180,8 @@ sub mset_thread {
                $smsg;
        } ($mset->items) ]});
 
-       my $th = PublicInbox::SearchThread->new($msgs);
-       $th->thread;
-       if ($q->{r}) { # order by relevance
-               $th->order(sub {
-                       [ sort { (eval { $pct{$b->topmost->{id}} } || 0)
-                                       <=>
-                               (eval { $pct{$a->topmost->{id}} } || 0)
-                       } @{$_[0]} ];
-               });
-       } else { # order by time (default for threaded view)
-               $th->order(*PublicInbox::View::sort_ts);
-       }
+       my $rootset = PublicInbox::SearchThread::thread($msgs,
+               $q->{r} ? sort_relevance(\%pct) : *PublicInbox::View::sort_ts);
        my $skel = search_nav_bot($mset, $q). "<pre>";
        my $inbox = $ctx->{-inbox};
        $ctx->{-upfx} = '';
@@ -196,7 +195,7 @@ sub mset_thread {
        $ctx->{seen} = {};
        $ctx->{s_nr} = scalar(@$msgs).'+ results';
 
-       PublicInbox::View::walk_thread($th, $ctx,
+       PublicInbox::View::walk_thread($rootset, $ctx,
                *PublicInbox::View::pre_thread);
 
        my $mime;
@@ -206,7 +205,7 @@ sub mset_thread {
                        $mime = $inbox->msg_by_smsg($mime) and last;
                }
                if ($mime) {
-                       $mime = Email::MIME->new($mime);
+                       $mime = PublicInbox::MIME->new($mime);
                        return PublicInbox::View::index_entry($mime, $ctx,
                                scalar @$msgs);
                }
@@ -240,7 +239,7 @@ sub adump {
                while (my $x = shift @items) {
                        $x = load_doc_retry($srch, $x);
                        $x = $ibx->msg_by_smsg($x) and
-                                       return Email::MIME->new($x);
+                                       return PublicInbox::MIME->new($x);
                }
                return undef;
        });