]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/View.pm
switch read-only uses of walk_parts to msg_iter
[public-inbox.git] / lib / PublicInbox / View.pm
index d559cae4b9cbe34239f5c930dcf3abcca5929119..42601677922cd177a6571b8ab1a1fcea20e47696 100644 (file)
@@ -14,14 +14,13 @@ use Email::MIME::ContentType qw/parse_content_type/;
 use PublicInbox::Hval qw/ascii_html/;
 use PublicInbox::Linkify;
 use PublicInbox::MID qw/mid_clean id_compress mid2path mid_mime/;
+use PublicInbox::MsgIter;
 require POSIX;
 
 use constant INDENT => '  ';
 use constant TCHILD => '` ';
 sub th_pfx ($) { $_[0] == 0 ? '' : TCHILD };
 
-my $enc_utf8 = find_encoding('UTF-8');
-
 # public functions:
 sub msg_html {
        my ($ctx, $mime, $footer) = @_;
@@ -73,13 +72,6 @@ sub msg_reply {
        '</pre><hr /><pre>' . $footer .  '</pre></body></html>';
 }
 
-sub feed_entry {
-       my ($class, $mime) = @_;
-
-       # no <head> here for <style>...
-       PublicInbox::Hval::PRE . multipart_text_as_html($mime) . '</pre>';
-}
-
 sub in_reply_to {
        my ($hdr) = @_;
        my $irt = $hdr->header_raw('In-Reply-To');
@@ -99,9 +91,7 @@ sub index_entry {
        my $midx = $state->{anchor_idx}++;
        my $ctx = $state->{ctx};
        my $srch = $ctx->{srch};
-       my $part_nr = 0;
        my $hdr = $mime->header_obj;
-       my $enc = enc_for($hdr->header("Content-Type"));
        my $subj = $hdr->header('Subject');
 
        my $mid_raw = mid_clean(mid_mime($mime));
@@ -135,11 +125,8 @@ sub index_entry {
        my $mhref = "${path}$href/";
 
        # scan through all parts, looking for displayable text
-       $mime->walk_parts(sub {
-               index_walk($fh, $_[0], $enc, \$part_nr);
-       });
-       $mime->body_set('');
-       $rv = "\n" . html_footer($hdr, 0, $ctx, "$path$href/R");
+       msg_iter($mime, sub { index_walk($fh, $_[0]) });
+       $rv = "\n" . html_footer($hdr, 0, $ctx, "$path$href/R/");
 
        if (defined $irt) {
                unless (defined $parent_anchor) {
@@ -224,8 +211,8 @@ sub emit_thread_html {
 }
 
 sub index_walk {
-       my ($fh, $part, $enc, $part_nr) = @_;
-       my $s = add_text_body($enc, $part, $part_nr);
+       my ($fh, $p) = @_;
+       my $s = add_text_body($p);
 
        return if $s eq '';
 
@@ -234,49 +221,22 @@ sub index_walk {
        $fh->write($s);
 }
 
-sub enc_for {
-       my ($ct, $default) = @_;
-       $default ||= $enc_utf8;
-       defined $ct or return $default;
-       my $ct_parsed = parse_content_type($ct);
-       if ($ct_parsed) {
-               if (my $charset = $ct_parsed->{attributes}->{charset}) {
-                       my $enc = find_encoding($charset);
-                       return $enc if $enc;
-               }
-       }
-       $default;
-}
-
 sub multipart_text_as_html {
        my ($mime) = @_;
        my $rv = "";
-       my $part_nr = 0;
-       my $enc = enc_for($mime->header("Content-Type"));
 
        # scan through all parts, looking for displayable text
-       $mime->walk_parts(sub {
-               my ($part) = @_;
-               $part = add_text_body($enc, $part, \$part_nr);
-               $rv .= $part;
-               $rv .= "\n" if $part ne '';
+       msg_iter($mime, sub {
+               my ($p) = @_;
+               $p = add_text_body($p);
+               $rv .= $p;
+               $rv .= "\n" if $p ne '';
        });
-       $mime->body_set('');
        $rv;
 }
 
-sub add_filename_line {
-       my ($enc, $fn) = @_;
-       my $len = 72;
-       my $pad = "-";
-       $fn = $enc->decode($fn);
-       $len -= length($fn);
-       $pad x= ($len/2) if ($len > 0);
-       "$pad " . ascii_html($fn) . " $pad\n";
-}
-
 sub flush_quote {
-       my ($s, $l, $quot, $part_nr) = @_;
+       my ($s, $l, $quot) = @_;
 
        # show everything in the full version with anchor from
        # short version (see above)
@@ -289,35 +249,50 @@ sub flush_quote {
        $$s .= qq(<span\nclass="q">) . $rv . '</span>'
 }
 
-sub add_text_body {
-       my ($enc_msg, $part, $part_nr) = @_;
-       return '' if $part->subparts;
+sub attach_link ($$$) {
+       my ($ct, $p, $fn) = @_;
+       my ($part, $depth, @idx) = @$p;
+       my $nl = $idx[-1] > 1 ? "\n" : '';
+       my $idx = join('.', @idx);
+       my $size = bytes::length($part->body);
+       $ct ||= 'text/plain';
+       $ct =~ s/;.*//; # no attributes
+       $ct = ascii_html($ct);
+       my $desc = $part->header('Content-Description');
+       $desc = $fn unless defined $desc;
+       $desc = '' unless defined $desc;
+       $desc = ': '.$desc if $desc;
+       "$nl<b>[-- Attachment #$idx$desc --]\n" .
+       "[-- Type: $ct, Size: $size bytes --]</b>"
+}
 
+sub add_text_body {
+       my ($p) = @_; # from msg_iter: [ Email::MIME, depth, @idx ]
+       my ($part, $depth, @idx) = @$p;
        my $ct = $part->content_type;
-       # account for filter bugs...
+       my $fn = $part->filename;
+
        if (defined $ct && $ct =~ m!\btext/x?html\b!i) {
-               $part->body_set('');
-               return '';
+               return attach_link($ct, $p, $fn);
        }
-       my $enc = enc_for($ct, $enc_msg);
-       my $s = $part->body;
-       $part->body_set('');
-       $s = $enc->decode($s);
+
+       my $s = eval { $part->body_str };
+
+       # badly-encoded message? tell the world about it!
+       return attach_link($ct, $p, $fn) if $@;
+
        my @lines = split(/^/m, $s);
        $s = '';
-
-       if ($$part_nr > 0) {
-               my $fn = $part->filename;
-               defined($fn) or $fn = "part #" . ($$part_nr + 1);
-               $s .= add_filename_line($enc, $fn);
+       if (defined($fn) || $depth > 1 || $idx[0] > 1) {
+               $s .= attach_link($ct, $p, $fn);
+               $s .= "\n\n";
        }
-
        my @quot;
        my $l = PublicInbox::Linkify->new;
        while (defined(my $cur = shift @lines)) {
                if ($cur !~ /^>/) {
                        # show the previously buffered quote inline
-                       flush_quote(\$s, $l, \@quot, $$part_nr) if @quot;
+                       flush_quote(\$s, $l, \@quot) if @quot;
 
                        # regular line, OK
                        $cur = $l->linkify_1($cur);
@@ -328,9 +303,7 @@ sub add_text_body {
                }
        }
 
-       flush_quote(\$s, $l, \@quot, $$part_nr) if @quot;
-       ++$$part_nr;
-
+       flush_quote(\$s, $l, \@quot) if @quot;
        $s =~ s/[ \t]+$//sgm; # kill per-line trailing whitespace
        $s =~ s/\A\n+//s; # kill leading blank lines
        $s =~ s/\s+\z//s; # kill all trailing spaces (final "\n" added if ne '')
@@ -518,7 +491,8 @@ sub html_footer {
                        $irt = ' ' x length('parent ');
                }
                if ($next) {
-                       $irt .= "<a\nhref=\"$upfx$next/\"\nrel=next>next</a> ";
+                       my $n = PublicInbox::Hval->new_msgid($next)->as_href;
+                       $irt .= "<a\nhref=\"$upfx$n/\"\nrel=next>next</a> ";
                } else {
                        $irt .= ' ' x length('next ');
                }