]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/Import.pm
remove Email::Address dependency
[public-inbox.git] / lib / PublicInbox / Import.pm
index e2156f1040f22f17e20e686e0f2ca57a6fec8b4b..e3d65f4aa73b8eb28fa3e9024361c057e2284b9c 100644 (file)
@@ -8,7 +8,6 @@ package PublicInbox::Import;
 use strict;
 use warnings;
 use Fcntl qw(:flock :DEFAULT);
-use Email::Address;
 use PublicInbox::Spawn qw(spawn);
 use PublicInbox::MID qw(mid_mime mid2path);
 
@@ -45,12 +44,13 @@ sub gfi_start {
                        --quiet --done --date-format=rfc2822));
        my $rdr = { 0 => fileno($out_r), 1 => fileno($in_w) };
        my $pid = spawn(\@cmd, undef, $rdr);
-       die "spawn failed: $!" unless defined $pid;
+       die "spawn fast-import failed: $!" unless defined $pid;
        $out_w->autoflush(1);
        $self->{in} = $in_r;
        $self->{out} = $out_w;
        $self->{lockfh} = $lockfh;
        $self->{pid} = $pid;
+       $self->{nchg} = 0;
        ($in_r, $out_w);
 }
 
@@ -65,8 +65,15 @@ sub now2822 () {
                $day, $t[3], $mon, $t[5] + 1900, $t[2], $t[1], $t[0]);
 }
 
+sub norm_body ($) {
+       my ($mime) = @_;
+       my $b = $mime->body_raw;
+       $b =~ s/(\r?\n)+\z//s;
+       $b
+}
+
 # returns undef on non-existent
-# (-1, msg) on mismatch
+# ('MISMATCH', msg) on mismatch
 # (:MARK, msg) on success
 sub remove {
        my ($self, $mime) = @_; # mime = Email::MIME
@@ -76,13 +83,13 @@ sub remove {
 
        my ($r, $w) = $self->gfi_start;
        my $tip = $self->{tip};
-       return if $tip eq '';
+       return ('MISSING', undef) if $tip eq '';
 
        print $w "ls $tip $path\n" or wfail;
        local $/ = "\n";
        my $check = <$r>;
        defined $check or die "EOF from fast-import / ls: $!";
-       return if $check =~ /\Amissing /;
+       return ('MISSING', undef) if $check =~ /\Amissing /;
        $check =~ m!\A100644 blob ([a-f0-9]{40})\t!s or die "not blob: $check";
        my $blob = $1;
        print $w "cat-blob $blob\n" or wfail;
@@ -93,19 +100,21 @@ sub remove {
        my $left = $1;
        my $offset = 0;
        my $buf = '';
+       my $n;
        while ($left > 0) {
-               my $n = read($r, $buf, $left, $offset);
+               $n = read($r, $buf, $left, $offset);
                defined($n) or die "read cat-blob failed: $!";
                $n == 0 and die 'fast-export (cat-blob) died';
                $left -= $n;
                $offset += $n;
        }
-       read($r, my $lf, 1);
+       $n = read($r, my $lf, 1);
+       defined($n) or die "read final byte of cat-blob failed: $!";
        die "bad read on final byte: <$lf>" if $lf ne "\n";
        my $cur = Email::MIME->new($buf);
        if ($cur->header('Subject') ne $mime->header('Subject') ||
-                       $cur->body ne $mime->body) {
-               return (-1, $cur);
+                       norm_body($cur) ne norm_body($mime)) {
+               return ('MISMATCH', $cur);
        }
 
        my $ref = $self->{ref};
@@ -122,6 +131,7 @@ sub remove {
                "data 3\nrm\n\n",
                'from ', ($parent ? $parent : $tip), "\n" or wfail;
        print $w "D $path\n\n" or wfail;
+       $self->{nchg}++;
        (($self->{tip} = ":$commit"), $cur);
 }
 
@@ -130,9 +140,15 @@ sub add {
        my ($self, $mime) = @_; # mime = Email::MIME
 
        my $from = $mime->header('From');
-       my @from = Email::Address->parse($from);
-       my $name = $from[0]->name;
-       my $email = $from[0]->address;
+       my ($email) = ($from =~ /([^<\s]+\@[^>\s]+)/g);
+       my $name = $from;
+       $name =~ s/\s*\S+\@\S+\s*\z//;
+       # git gets confused with:
+       #  "'A U Thor <u@example.com>' via foo" <foo@example.com>
+       # ref:
+       # <CAD0k6qSUYANxbjjbE4jTW4EeVwOYgBD=bXkSu=akiYC_CB7Ffw@mail.gmail.com>
+       $name =~ tr/<>// and $name = $email;
+
        my $date = $mime->header('Date');
        my $subject = $mime->header('Subject');
        $subject = '(no subject)' unless defined $subject;
@@ -162,15 +178,21 @@ sub add {
        unless ($parent) {
                print $w "reset $ref\n" or wfail;
        }
+
+       # quiet down wide character warnings:
+       binmode $w, ':utf8' or die "binmode :utf8 failed: $!";
        print $w "commit $ref\nmark :$commit\n",
                "author $name <$email> $date\n",
                "committer $self->{ident} ", now2822(), "\n",
-               "data ", (length($subject) + 1), "\n",
+               "data ", (bytes::length($subject) + 1), "\n",
                $subject, "\n\n" or wfail;
+       binmode $w, ':raw' or die "binmode :raw failed: $!";
+
        if ($tip ne '') {
                print $w 'from ', ($parent ? $parent : $tip), "\n" or wfail;
        }
        print $w "M 100644 :$blob $path\n\n" or wfail;
+       $self->{nchg}++;
        $self->{tip} = ":$commit";
 }
 
@@ -182,9 +204,152 @@ sub done {
        my $pid = delete $self->{pid} or die 'BUG: missing {pid} when done';
        waitpid($pid, 0) == $pid or die 'fast-import did not finish';
        $? == 0 or die "fast-import failed: $?";
+       my $nchg = delete $self->{nchg};
+
+       # for compatibility with existing ssoma installations
+       # we can probably remove this entirely by 2020
+       my $git_dir = $self->{git}->{git_dir};
+       # XXX: change the following scope to: if (-e $index) # in 2018 or so..
+       my @cmd = ('git', "--git-dir=$git_dir");
+       if ($nchg && !$ENV{FAST}) {
+               my $index = "$git_dir/ssoma.index";
+               my $env = { GIT_INDEX_FILE => $index };
+               my @rt = (@cmd, qw(read-tree -m -v -i), $self->{ref});
+               $pid = spawn(\@rt, $env, undef);
+               defined $pid or die "spawn read-tree failed: $!";
+               waitpid($pid, 0) == $pid or die 'read-tree did not finish';
+               $? == 0 or die "failed to update $git_dir/ssoma.index: $?\n";
+       }
+       if ($nchg) {
+               $pid = spawn([@cmd, 'update-server-info'], undef, undef);
+               defined $pid or die "spawn update-server-info failed: $!\n";
+               waitpid($pid, 0) == $pid or
+                       die 'update-server-info did not finish';
+               $? == 0 or die "failed to update-server-info: $?\n";
+       }
+
        my $lockfh = delete $self->{lockfh} or die "BUG: not locked: $!";
        flock($lockfh, LOCK_UN) or die "unlock failed: $!";
        close $lockfh or die "close lock failed: $!";
 }
 
 1;
+__END__
+=pod
+
+=head1 NAME
+
+PublicInbox::Import - message importer for public-inbox
+
+=head1 VERSION
+
+version 1.0
+
+=head1 SYNOPSYS
+
+       use Email::MIME;
+       use PublicInbox::Git;
+       use PublicInbox::Import;
+
+       chomp(my $git_dir = `git rev-parse --git-dir`);
+       $git_dir or die "GIT_DIR= must be specified\n";
+       my $git = PublicInbox::Git->new($git_dir);
+       my @committer = ('inbox', 'inbox@example.org');
+       my $im = PublicInbox::Import->new($git, @committer);
+
+       # to add a message:
+       my $message = "From: <u\@example.org>\n".
+               "Subject: test message \n" .
+               "Date: Thu, 01 Jan 1970 00:00:00 +0000\n" .
+               "Message-ID: <m\@example.org>\n".
+               "\ntest message";
+       my $parsed = Email::MIME->new($message);
+       my $ret = $im->add($parsed);
+       if (!defined $ret) {
+               warn "duplicate: ",
+                       $parsed->header_obj->header_raw('Message-ID'), "\n";
+       } else {
+               print "imported at mark $ret\n";
+       }
+       $im->done;
+
+       # to remove a message
+       my $junk = Email::MIME->new($message);
+       my ($mark, $orig) = $im->remove($junk);
+       if ($mark eq 'MISSING') {
+               print "not found\n";
+       } elsif ($mark eq 'MISMATCH') {
+               print "Message exists but does not match\n\n",
+                       $orig->as_string, "\n",;
+       } else {
+               print "removed at mark $mark\n\n",
+                       $orig->as_string, "\n";
+       }
+       $im->done;
+
+=head1 DESCRIPTION
+
+An importer and remover for public-inboxes which takes L<Email::MIME>
+messages as input and stores them in a ssoma repository as
+documented in L<https://ssoma.public-inbox.org/ssoma_repository.txt>,
+except it does not allow duplicate Message-IDs.
+
+It requires L<git(1)> and L<git-fast-import(1)> to be installed.
+
+=head1 METHODS
+
+=cut
+
+=head2 new
+
+       my $im = PublicInbox::Import->new($git, @committer);
+
+Initialize a new PublicInbox::Import object.
+
+=head2 add
+
+       my $parsed = Email::MIME->new($message);
+       $im->add($parsed);
+
+Adds a message to to the git repository.  This will acquire
+C<$GIT_DIR/ssoma.lock> and start L<git-fast-import(1)> if necessary.
+
+Messages added will not be visible to other processes until L</done>
+is called, but L</remove> may be called on them.
+
+=head2 remove
+
+       my $junk = Email::MIME->new($message);
+       my ($code, $orig) = $im->remove($junk);
+
+Removes a message from the repository.  On success, it returns
+a ':'-prefixed numeric code representing the git-fast-import
+mark and the original messages as an Email::MIME object.
+If the message could not be found, the code is "MISSING"
+and the original message is undef.  If there is a mismatch where
+the "Message-ID" is matched but the subject and body do not match,
+the returned code is "MISMATCH" and the conflicting message
+is returned as orig.
+
+=head2 done
+
+Finalizes the L<git-fast-import(1)> and unlocks the repository.
+Calling this is required to finalize changes to a repository.
+
+=head1 SEE ALSO
+
+L<Email::MIME>
+
+=head1 CONTACT
+
+All feedback welcome via plain-text mail to L<mailto:meta@public-inbox.org>
+
+The mail archives are hosted at L<https://public-inbox.org/meta/>
+
+=head1 COPYRIGHT
+
+Copyright (C) 2016 all contributors L<mailto:meta@public-inbox.org>
+
+License: AGPL-3.0+ L<http://www.gnu.org/licenses/agpl-3.0.txt>
+
+=cut