]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/SearchView.pm
www_stream: add response wrapper sub
[public-inbox.git] / lib / PublicInbox / SearchView.pm
index 8283c1aa96f8df604f44b9116fb2ae50eac092d1..30a310c9f58ceede52857d3b07d23668eaedcf6b 100644 (file)
@@ -8,14 +8,17 @@ use warnings;
 use PublicInbox::SearchMsg;
 use PublicInbox::Hval qw/ascii_html/;
 use PublicInbox::View;
-use PublicInbox::MID qw(mid2path mid_clean mid_mime);
+use PublicInbox::MID qw(mid2path mid_mime mid_clean);
 use Email::MIME;
 require PublicInbox::Git;
+require PublicInbox::Thread;
 our $LIM = 50;
 
+sub noop {}
+
 sub sres_top_html {
        my ($ctx) = @_;
-       my $q = PublicInbox::SearchQuery->new($ctx->{cgi});
+       my $q = PublicInbox::SearchQuery->new($ctx->{qp});
        my $code = 200;
 
        # double the limit for expanded views:
@@ -26,44 +29,44 @@ sub sres_top_html {
                relevance => $q->{r},
        };
        my ($mset, $total);
-
        eval {
-               $mset = $ctx->{srch}->query($q->{q}, $opts);
+               $mset = $ctx->{srch}->query($q->{'q'}, $opts);
                $total = $mset->get_matches_estimated;
        };
        my $err = $@;
-       my $res = html_start($q, $ctx) . '<pre>';
+       ctx_prepare($q, $ctx);
+       my $cb;
        if ($err) {
                $code = 400;
-               $res .= err_txt($ctx, $err) . "</pre><hr /><pre>" . foot($ctx);
+               $ctx->{-html_tip} = '<pre>'.err_txt($ctx, $err).'</pre><hr />';
+               $cb = *noop;
        } elsif ($total == 0) {
                $code = 404;
-               $res .= "\n\n[No results found]</pre><hr /><pre>".foot($ctx);
+               $ctx->{-html_tip} = "<pre>\n[No results found]</pre><hr />";
+               $cb = *noop;
        } else {
                my $x = $q->{x};
                return sub { adump($_[0], $mset, $q, $ctx) } if ($x eq 'A');
 
-               $res .= search_nav_top($mset, $q) . "\n\n";
+               $ctx->{-html_tip} = search_nav_top($mset, $q) . "\n\n";
                if ($x eq 't') {
-                       return sub { tdump($_[0], $res, $mset, $q, $ctx) };
+                       $cb = mset_thread($ctx, $mset, $q);
+               } else {
+                       $cb = mset_summary($ctx, $mset, $q);
                }
-               dump_mset(\$res, $mset);
-               $res .= '</pre>' . search_nav_bot($mset, $q) .
-                       "\n\n" . foot($ctx);
        }
-
-       $res .= "</pre></body></html>";
-       [$code, ['Content-Type'=>'text/html; charset=UTF-8'], [$res]];
+       PublicInbox::WwwStream->response($ctx, $code, $cb);
 }
 
 # display non-threaded search results similar to what users expect from
 # regular WWW search engines:
-sub dump_mset {
-       my ($res, $mset) = @_;
+sub mset_summary {
+       my ($ctx, $mset, $q) = @_;
 
        my $total = $mset->get_matches_estimated;
        my $pad = length("$total");
        my $pfx = ' ' x $pad;
+       my $res = \($ctx->{-html_tip});
        foreach my $m ($mset->items) {
                my $rank = sprintf("%${pad}d", $m->get_rank + 1);
                my $pct = $m->get_percent;
@@ -76,6 +79,8 @@ sub dump_mset {
                        $s . "</a></b>\n";
                $$res .= "$pfx  - by $f @ $ts UTC [$pct%]\n\n";
        }
+       $$res .= search_nav_bot($mset, $q);
+       *noop;
 }
 
 sub err_txt {
@@ -84,14 +89,14 @@ sub err_txt {
        $u = PublicInbox::Hval::prurl($ctx->{cgi}->{env}, $u);
        $err =~ s/^\s*Exception:\s*//; # bad word to show users :P
        $err = ascii_html($err);
-       "\n\nBad query: <b>$err</b>\n" .
+       "\nBad query: <b>$err</b>\n" .
                qq{See <a\nhref="$u">$u</a> for Xapian query syntax};
 }
 
 sub search_nav_top {
        my ($mset, $q) = @_;
 
-       my $rv = "Search results ordered by [";
+       my $rv = "<pre>Search results ordered by [";
        if ($q->{r}) {
                my $d = $q->qs_html(r => 0);
                $rv .= qq{<a\nhref="?$d">date</a>|<b>relevance</b>};
@@ -121,26 +126,24 @@ sub search_nav_bot {
        my $o = $q->{o};
        my $end = $o + $nr;
        my $beg = $o + 1;
-       my $rv = "<hr /><pre>Results $beg-$end of $total";
+       my $rv = "</pre><hr /><pre>Results $beg-$end of $total";
        my $n = $o + $LIM;
 
        if ($n < $total) {
                my $qs = $q->qs_html(o => $n);
-               $rv .= qq{, <a\nhref="?$qs">next</a>}
+               $rv .= qq{, <a\nhref="?$qs"\nrel=next>next</a>}
        }
        if ($o > 0) {
                $rv .= $n < $total ? '/' : ',      ';
                my $p = $o - $LIM;
                my $qs = $q->qs_html(o => ($p > 0 ? $p : 0));
-               $rv .= qq{<a\nhref="?$qs">prev</a>};
+               $rv .= qq{<a\nhref="?$qs"\nrel=prev>prev</a>};
        }
-       $rv;
+       $rv .= '</pre>';
 }
 
-sub tdump {
-       my ($cb, $res, $mset, $q, $ctx) = @_;
-       my $fh = $cb->([200, ['Content-Type'=>'text/html; charset=UTF-8']]);
-       $fh->write($res .= '</pre>');
+sub mset_thread {
+       my ($ctx, $mset, $q) = @_;
        my %pct;
        my @m = map {
                my $i = $_;
@@ -150,99 +153,69 @@ sub tdump {
                $m;
        } ($mset->items);
 
-       require PublicInbox::Thread;
        my $th = PublicInbox::Thread->new(@m);
-       {
-               no warnings 'once';
-               $Mail::Thread::nosubject = 0;
-       }
        $th->thread;
-       if ($q->{r}) {
+       if ($q->{r}) { # order by relevance
                $th->order(sub {
                        sort { (eval { $pct{$b->topmost->messageid} } || 0)
                                        <=>
                                (eval { $pct{$a->topmost->messageid} } || 0)
                        } @_;
                });
-       } else {
-               no warnings 'once';
-               $th->order(*PublicInbox::View::rsort_ts);
+       } else { # order by time (default for threaded view)
+               $th->order(*PublicInbox::View::sort_ts);
        }
-
-       my $git = $ctx->{git} ||= PublicInbox::Git->new($ctx->{git_dir});
-       my $state = {
-               ctx => $ctx,
-               anchor_idx => 0,
-               pct => \%pct,
-               cur_level => 0
+       my $skel = search_nav_bot($mset, $q). "<pre>";
+       my $inbox = $ctx->{-inbox};
+       $ctx->{-upfx} = '';
+       $ctx->{anchor_idx} = 1;
+       $ctx->{cur_level} = 0;
+       $ctx->{dst} = \$skel;
+       $ctx->{mapping} = {};
+       $ctx->{pct} = \%pct;
+       $ctx->{prev_attr} = '';
+       $ctx->{prev_level} = 0;
+       $ctx->{seen} = {};
+
+       PublicInbox::View::walk_thread($th, $ctx,
+               *PublicInbox::View::pre_thread);
+
+       my $msgs = \@m;
+       my $mime;
+       sub {
+               return unless $msgs;
+               while ($mime = shift @$msgs) {
+                       my $mid = mid_clean(mid_mime($mime));
+                       $mime = $inbox->msg_by_mid($mid) and last;
+               }
+               if ($mime) {
+                       $mime = Email::MIME->new($mime);
+                       return PublicInbox::View::index_entry($mime, $ctx);
+               }
+               $msgs = undef;
+               $skel .= "\n</pre>";
        };
-       $ctx->{searchview} = 1;
-       tdump_ent($fh, $git, $state, $_, 0) for $th->rootset;
-       PublicInbox::View::thread_adj_level($fh, $state, 0);
-       Email::Address->purge_cache;
-
-       $fh->write(search_nav_bot($mset, $q). "\n\n" .
-                       foot($ctx). '</pre></body></html>');
-
-       $fh->close;
-}
-
-sub tdump_ent {
-       my ($fh, $git, $state, $node, $level) = @_;
-       return unless $node;
-       my $mime = $node->message;
-
-       if ($mime) {
-               # lazy load the full message from mini_mime:
-               my $mid = mid_mime($mime);
-               $mime = eval {
-                       my $path = mid2path(mid_clean($mid));
-                       Email::MIME->new($git->cat_file('HEAD:'.$path));
-               };
-       }
-       if ($mime) {
-               my $end =
-                 PublicInbox::View::thread_adj_level($fh, $state, $level);
-               PublicInbox::View::index_entry($fh, $mime, $level, $state);
-               $fh->write($end) if $end;
-       } else {
-               my $mid = $node->messageid;
-               PublicInbox::View::ghost_flush($fh, $state, '', $mid, $level);
-       }
-       tdump_ent($fh, $git, $state, $node->child, $level + 1);
-       tdump_ent($fh, $git, $state, $node->next, $level);
-}
-
-sub foot {
-       my ($ctx) = @_;
-       my $foot = $ctx->{footer} || '';
-       qq{Back to <a\nhref=".">index</a>.\n$foot};
 }
 
-sub html_start {
+sub ctx_prepare {
        my ($q, $ctx) = @_;
        my $qh = ascii_html($q->{'q'});
-       my $A = $q->qs_html(x => 'A', r => undef);
-       my $res = '<html><head>' . PublicInbox::Hval::STYLE .
-               "<title>$qh - search results</title>" .
-               qq{<link\nrel=alternate\ntitle="Atom feed"\n} .
-               qq!href="?$A"\ntype="application/atom+xml"/></head>! .
-               qq{<body><form\naction="">} .
-               qq{<input\nname=q\nvalue="$qh"\ntype=text />};
-
-       $res .= qq{<input\ntype=hidden\nname=r />} if $q->{r};
+       $ctx->{-q_value_html} = $qh;
+       $ctx->{-atom} = '?'.$q->qs_html(x => 'A', r => undef);
+       $ctx->{-title_html} = "$qh - search results";
+       my $extra = '';
+       $extra .= qq{<input\ntype=hidden\nname=r />} if $q->{r};
        if (my $x = $q->{x}) {
                $x = ascii_html($x);
-               $res .= qq{<input\ntype=hidden\nname=x\nvalue="$x" />};
+               $extra .= qq{<input\ntype=hidden\nname=x\nvalue="$x" />};
        }
-
-       $res .= qq{<input\ntype=submit\nvalue=search /></form>};
+       $ctx->{-extra_form_html} = $extra;
 }
 
 sub adump {
        my ($cb, $mset, $q, $ctx) = @_;
        my $fh = $cb->([ 200, ['Content-Type' => 'application/atom+xml']]);
-       my $git = $ctx->{git} ||= PublicInbox::Git->new($ctx->{git_dir});
+       my $ibx = $ctx->{-inbox};
        my $feed_opts = PublicInbox::Feed::get_feedopts($ctx);
        my $x = ascii_html($q->{'q'});
        $x = qq{$x - search results};
@@ -254,7 +227,8 @@ sub adump {
        for ($mset->items) {
                $x = PublicInbox::SearchMsg->load_doc($_->get_document)->mid;
                $x = mid2path($x);
-               PublicInbox::Feed::add_to_feed($feed_opts, $fh, $x, $git);
+               my $s = PublicInbox::Feed::feed_entry($feed_opts, $x, $ibx);
+               $fh->write($s) if defined $s;
        }
        PublicInbox::Feed::end_feed($fh);
 }
@@ -265,12 +239,13 @@ use warnings;
 use PublicInbox::Hval;
 
 sub new {
-       my ($class, $cgi) = @_;
-       my $r = $cgi->param('r');
+       my ($class, $qp) = @_;
+
+       my $r = $qp->{r};
        bless {
-               q => $cgi->param('q'),
-               x => $cgi->param('x') || '',
-               o => int($cgi->param('o') || 0) || 0,
+               q => $qp->{'q'},
+               x => $qp->{x} || '',
+               o => (($qp->{o} || '0') =~ /(\d+)/),
                r => (defined $r && $r ne '0'),
        }, $class;
 }