]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/LeiForgetSearch.pm
imap+nntp: share COMPRESS implementation
[public-inbox.git] / lib / PublicInbox / LeiForgetSearch.pm
index 717fa5e9c2e1b68710706f97a9a0ec82628b63f3..dd358ae16a3a086c90b8b0dae8a6b00670149247 100644 (file)
@@ -5,28 +5,75 @@
 package PublicInbox::LeiForgetSearch;
 use strict;
 use v5.10.1;
+use parent qw(PublicInbox::LeiUp);
 use PublicInbox::LeiSavedSearch;
-use PublicInbox::LeiUp;
 use File::Path ();
 use SelectSaver;
 
-sub lei_forget_search {
-       my ($lei, $out) = @_;
-       my $d = PublicInbox::LeiSavedSearch::lss_dir_for($lei, \$out, 1);
-       if (-e $d) {
-               my $save;
-               my $opt = { safe => 1 };
-               if ($lei->{opt}->{verbose}) {
-                       $opt->{verbose} = 1;
-                       $save = SelectSaver->new($lei->{2});
+sub do_forget_search {
+       my ($lei, @outs) = @_;
+       my @dirs; # paths in ~/.local/share/lei/saved-search/
+       my $cwd;
+       for my $o (@outs) {
+               my $d = PublicInbox::LeiSavedSearch::lss_dir_for($lei, \$o, 1);
+               if (-e $d) {
+                       push @dirs, $d
+               } else { # keep going, like rm(1):
+                       $cwd //= $lei->rel2abs('.');
+                       warn "--save was not used with $o cwd=$cwd\n";
                }
-               File::Path::remove_tree($d, $opt);
-       } else {
-               $lei->fail("--save was not used with $out cwd=".
-                                       $lei->rel2abs('.'));
        }
+       my $save;
+       my $opt = { safe => 1 };
+       if ($lei->{opt}->{verbose}) {
+               $opt->{verbose} = 1;
+               $save = SelectSaver->new($lei->{2});
+       }
+       File::Path::remove_tree(@dirs, $opt);
+       $lei->child_error(0) if defined $cwd;
+}
+
+sub lei_forget_search {
+       my ($lei, @outs) = @_;
+       my $prune = $lei->{opt}->{prune};
+       $prune // return do_forget_search($lei, @outs);
+       return $lei->fail("--prune and @outs incompatible") if @outs;
+       my @tmp = PublicInbox::LeiSavedSearch::list($lei);
+       my $self = bless { -mail_sync => 1 }, __PACKAGE__;
+       $self->filter_lss($lei, $prune) // return
+                       $lei->fail("only --prune=$prune not understood");
+       if ($self->{o_remote}) { # setup lei->{auth}
+               $self->prepare_inputs($lei, $self->{o_remote}) or return;
+       }
+       $lei->wq1_start($self);
+}
+
+sub do_prune {
+       my ($self) = @_;
+       my $lei = $self->{lei};
+       for my $o (@{$self->{o_local} // []}) {
+               next if -e $o;
+               $lei->qerr("# pruning $o");
+               eval { do_forget_search($lei, $o) };
+               $lei->child_error(0, "E: $@") if $@;
+       }
+       for my $o (@{$self->{o_remote} // []}) {
+               my $uri = PublicInbox::URIimap->new($o);
+               next if $lei->{net}->mic_for_folder($uri);
+               $lei->qerr("# pruning $uri");
+               eval { do_forget_search($lei, $o) };
+               $lei->child_error(0, "E: $@") if $@;
+       }
+}
+
+# called in top-level lei-daemon when LeiAuth is done
+sub net_merge_all_done {
+       my ($self) = @_;
+       $self->wq_do('do_prune');
+       $self->wq_close;
 }
 
+*_wq_done_wait = \&PublicInbox::LEI::wq_done_wait;
 *_complete_forget_search = \&PublicInbox::LeiUp::_complete_up;
 
 1;