purge: start moving common options to AdminEdit module

[public-inbox.git] / script / public-inbox-purge

diff --git a/script/public-inbox-purge b/script/public-inbox-purge
index 264bcdefa2c88bb06b16457281c614acf51a1bef..d58a9baae096b66cb439c71385bbc2be7e7f9f87 100755 (executable)
--- a/script/public-inbox-purge
+++ b/script/public-inbox-purge
@@ -7,37 +7,32 @@
 use strict;
 use warnings;
 use Getopt::Long qw(:config gnu_getopt no_ignore_case auto_abbrev);
-use PublicInbox::Config;
-use PublicInbox::MIME;
-use PublicInbox::Admin qw(resolve_repo_dir);
-use PublicInbox::Filter::Base;
-*REJECT = *PublicInbox::Filter::Base::REJECT;
+use PublicInbox::AdminEdit;
+PublicInbox::Admin::check_require('-index');
+require PublicInbox::Filter::Base;
+require PublicInbox::Config;
+require PublicInbox::MIME;
+require PublicInbox::V2Writable;
 
-my $usage = "$0 [--all] [INBOX_DIRS] </path/to/message";
+{ no warnings 'once'; *REJECT = *PublicInbox::Filter::Base::REJECT }
 
-eval { require PublicInbox::V2Writable } or die
-       "DBI, DBD::SQLite and Search::Xapian required for purge\n";
+my $usage = "$0 [--all] [INBOX_DIRS] </path/to/message";
 my $config = eval { PublicInbox::Config->new };
 my $cfgfile = PublicInbox::Config::default_file();
-my ($all, $force);
-my $verbose = 1;
-my %opts = (
-       'all' => \$all,
-       'force|f' => \$force,
-       'verbose|v!' => \$verbose,
-);
-GetOptions(%opts) or die "bad command-line args\n", $usage, "\n";
+my $opt = { verbose => 1 };
+GetOptions($opt, @PublicInbox::AdminEdit::OPT) or
+       die "bad command-line args\n$usage\n";
 
 # TODO: clean this up and share code with -index via ::Admin
 my %dir2ibx; # ( path => Inbox object )
 my @inboxes;
 $config and $config->each_inbox(sub {
        my ($ibx) = @_;
-       push @inboxes, $ibx if $all && $ibx->{version} != 1;
+       push @inboxes, $ibx if $opt->{all} && $ibx->{version} != 1;
        $dir2ibx{$ibx->{mainrepo}} = $ibx;
 });
 
-if ($all) {
+if ($opt->{all}) {
        $config or die "--all specified, but $cfgfile not readable\n";
        @ARGV and die "--all specified, but directories specified\n";
 } else {
@@ -47,7 +42,7 @@ if ($all) {
 
        foreach my $dir (@dirs) {
                my $v;
-               my $dir = resolve_repo_dir($dir, \$v);
+               my $dir = PublicInbox::Admin::resolve_repo_dir($dir, \$v);
                if ($v == 1) {
                        push @err, $dir;
                        next;
@@ -63,11 +58,6 @@ if ($all) {
                                mainrepo => $dir,
                        });
                };
-
-               # somebody could "rm -r" all the Xapian directories;
-               # let them purge the overview, at least
-               $ibx->{indexlevel} ||= 'basic' unless $ibx->search;
-
                push @inboxes, $ibx;
        }
 
@@ -77,6 +67,40 @@ if ($all) {
        }
 }
 
+foreach my $ibx (@inboxes) {
+       my $lvl = $ibx->{indexlevel};
+       if (defined $lvl) {
+               PublicInbox::Admin::indexlevel_ok_or_die($lvl);
+               next;
+       }
+
+       # Undefined indexlevel, so `full'...
+       # Search::Xapian exists and the DB can be read, at least, fine
+       $ibx->search and next;
+
+       # it's possible for a Xapian directory to exist, but Search::Xapian
+       # to go missing/broken.  Make sure it's purged in that case:
+       $ibx->over or die "no over.sqlite3 in $ibx->{mainrepo}\n";
+
+       # $ibx->{search} is populated by $ibx->over call
+       my $xdir_ro = $ibx->{search}->xdir(1);
+       my $npart = 0;
+       foreach my $part (<$xdir_ro/*>) {
+               if (-d $part && $part =~ m!/[0-9]+\z!) {
+                       my $bytes = 0;
+                       $bytes += -s $_ foreach glob("$part/*");
+                       $npart++ if $bytes;
+               }
+       }
+       if ($npart) {
+               PublicInbox::Admin::require_or_die('-search');
+       } else {
+               # somebody could "rm -r" all the Xapian directories;
+               # let them purge the overview, at least
+               $ibx->{indexlevel} ||= 'basic';
+       }
+}
+
 my $data = do { local $/; scalar <STDIN> };
 $data =~ s/\A[\r\n]*From [^\r\n]*\r?\n//s;
 my $n_purged = 0;
@@ -98,7 +122,7 @@ foreach my $ibx (@inboxes) {
 
        $v2w->done;
 
-       if ($verbose) { # should we consider this machine-parseable?
+       if ($opt->{verbose}) { # should we consider this machine-parseable?
                print "$ibx->{mainrepo}:";
                if (scalar @$commits) {
                        print join("\n\t", '', @$commits), "\n";
@@ -110,7 +134,7 @@ foreach my $ibx (@inboxes) {
 }
 
 # behave like "rm -f"
-exit(0) if ($force || $n_purged);
+exit(0) if ($opt->{force} || $n_purged);
 
-warn "Not found\n" if $verbose;
+warn "Not found\n" if $opt->{verbose};
 exit(1);