about summary refs log tree commit homepage
path: root/script/public-inbox-edit
diff options
context:
space:
mode:
Diffstat (limited to 'script/public-inbox-edit')
-rwxr-xr-xscript/public-inbox-edit73
1 files changed, 43 insertions, 30 deletions
diff --git a/script/public-inbox-edit b/script/public-inbox-edit
index ae5d8289..88115d7c 100755
--- a/script/public-inbox-edit
+++ b/script/public-inbox-edit
@@ -1,5 +1,5 @@
 #!/usr/bin/perl -w
-# Copyright (C) 2019-2020 all contributors <meta@public-inbox.org>
+# Copyright (C) all contributors <meta@public-inbox.org>
 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
 #
 # Used for editing messages in a public-inbox.
@@ -9,20 +9,36 @@ use warnings;
 use Getopt::Long qw(:config gnu_getopt no_ignore_case auto_abbrev);
 use PublicInbox::AdminEdit;
 use File::Temp 0.19 (); # 0.19 for TMPDIR
-use PublicInbox::ContentId qw(content_id);
+use PublicInbox::ContentHash qw(content_hash);
 use PublicInbox::MID qw(mid_clean mids);
 PublicInbox::Admin::check_require('-index');
-use PublicInbox::MIME;
-use PublicInbox::InboxWritable;
+use PublicInbox::Eml;
+use PublicInbox::InboxWritable qw(eml_from_path);
 use PublicInbox::Import;
 
-my $usage = "$0 -m MESSAGE_ID [--all] [INBOX_DIRS]";
+my $help = <<'EOF';
+usage: public-inbox-edit -m MESSAGE-ID [--all] [INBOX_DIRS]
+
+  destructively edit messages in a public inbox
+
+options:
+
+  --all               edit all configured inboxes
+  -m MESSAGE-ID       edit the message with a given Message-ID
+  -F FILE             edit the message matching the contents of FILE
+  --force             forcibly edit even if Message-ID is ambiguous
+  --raw               do not perform "From " line escaping
+
+See public-inbox-edit(1) man page for full documentation.
+EOF
+
 my $opt = { verbose => 1, all => 0, -min_inbox_version => 2, raw => 0 };
-my @opt = qw(mid|m=s file|F=s raw);
-GetOptions($opt, @PublicInbox::AdminEdit::OPT, @opt) or
-        die "bad command-line args\n$usage\n";
+my @opt = qw(mid|m=s file|F=s raw C=s@);
+GetOptions($opt, @PublicInbox::AdminEdit::OPT, @opt) or die $help;
+if ($opt->{help}) { print $help; exit 0 };
+PublicInbox::Admin::do_chdir(delete $opt->{C});
 
-my $cfg = eval { PublicInbox::Config->new };
+my $cfg = PublicInbox::Config->new;
 my $editor = $ENV{MAIL_EDITOR}; # e.g. "mutt -f"
 unless (defined $editor) {
         my $k = 'publicinbox.mailEditor';
@@ -43,7 +59,7 @@ if (defined $mid && defined $file) {
 my @ibxs = PublicInbox::Admin::resolve_inboxes(\@ARGV, $opt, $cfg);
 PublicInbox::AdminEdit::check_editable(\@ibxs);
 
-my $found = {}; # cid => [ [ibx, smsg] [, [ibx, smsg] ] ]
+my $found = {}; # chash => [ [ibx, smsg] [, [ibx, smsg] ] ]
 
 sub find_mid ($$$) {
         my ($found, $mid, $ibxs) = @_;
@@ -52,10 +68,10 @@ sub find_mid ($$$) {
                 my ($id, $prev);
                 while (my $smsg = $over->next_by_mid($mid, \$id, \$prev)) {
                         my $ref = $ibx->msg_by_smsg($smsg);
-                        my $mime = PublicInbox::MIME->new($ref);
-                        my $cid = content_id($mime);
+                        my $mime = PublicInbox::Eml->new($ref);
+                        my $chash = content_hash($mime);
                         my $tuple = [ $ibx, $smsg ];
-                        push @{$found->{$cid} ||= []}, $tuple
+                        push @{$found->{$chash} ||= []}, $tuple
                 }
                 PublicInbox::InboxWritable::cleanup($ibx);
         }
@@ -92,13 +108,11 @@ Multiple messages with different content found matching
                 warn "Will edit all of them\n";
         }
 } else {
-        open my $fh, '<', $file or die "open($file) failed: $!";
-        my $orig = do { local $/; <$fh> };
-        my $mime = PublicInbox::MIME->new(\$orig);
-        my $mids = mids($mime->header_obj);
+        my $eml = eml_from_path($file) or die "open($file) failed: $!";
+        my $mids = mids($eml);
         find_mid($found, $_, \@ibxs) for (@$mids); # populates $found
-        my $cid = content_id($mime);
-        my $to_edit = $found->{$cid};
+        my $chash = content_hash($eml);
+        my $to_edit = $found->{$chash};
         unless ($to_edit) {
                 my $nr = scalar(keys %$found);
                 if ($nr > 0) {
@@ -116,11 +130,11 @@ $mids
                 }
                 exit 1;
         }
-        $found = { $cid => $to_edit };
+        $found = { $chash => $to_edit };
 }
 
 my %tmpopt = (
-        TEMPLATE => 'public-inbox-edit-XXXXXX',
+        TEMPLATE => 'public-inbox-edit-XXXX',
         TMPDIR => 1,
         SUFFIX => $opt->{raw} ? '.eml' : '.mbox',
 );
@@ -170,11 +184,10 @@ retry_edit:
         # rename/relink $edit_fn
         open my $new_fh, '<', $edit_fn or
                 die "can't read edited file ($edit_fn): $!\n";
-        my $new_raw = do { local $/; <$new_fh> };
+        my $new_raw = PublicInbox::IO::read_all $new_fh;
 
         if (!$opt->{raw}) {
-                # get rid of the From we added
-                $new_raw =~ s/\A[\r\n]*From [^\r\n]*\r?\n//s;
+                PublicInbox::Eml::strip_from($new_raw);
 
                 # check if user forgot to purge (in mutt) after editing
                 if ($new_raw =~ /^From /sm) {
@@ -206,8 +219,8 @@ W: possible message boundary splitting error
                 $new_raw =~ s/^>(>*From )/$1/gm;
         }
 
-        my $new_mime = PublicInbox::MIME->new(\$new_raw);
-        my $old_mime = PublicInbox::MIME->new($old_raw);
+        my $new_mime = PublicInbox::Eml->new(\$new_raw);
+        my $old_mime = PublicInbox::Eml->new($old_raw);
 
         # make sure we don't compare unwanted headers, since mutt adds
         # Content-Length, Status, and Lines headers:
@@ -216,10 +229,10 @@ W: possible message boundary splitting error
 
         # allow changing Received: and maybe other headers which can
         # contain sensitive info.
-        my $nhdr = $new_mime->header_obj;
-        my $ohdr = $old_mime->header_obj;
-        if (($nhdr->as_string eq $ohdr->as_string) &&
-            (content_id($new_mime) eq content_id($old_mime))) {
+        my $nhdr = $new_mime->header_obj->as_string;
+        my $ohdr = $old_mime->header_obj->as_string;
+        if (($nhdr eq $ohdr) &&
+            (content_hash($new_mime) eq content_hash($old_mime))) {
                 warn "No change detected to:\n", show_cmd($ibx, $smsg);
 
                 next unless $opt->{verbose};