dir_idle: treat IN_MOVED_FROM as a gone event

[public-inbox.git] / lib / PublicInbox / DirIdle.pm

diff --git a/lib/PublicInbox/DirIdle.pm b/lib/PublicInbox/DirIdle.pm
index e53fd9d1e5fe0db5b1a2cac1b053e6d641a276d1..9206da9cb710186462ffefd10cb4af0764f7e615 100644 (file)
--- a/lib/PublicInbox/DirIdle.pm
+++ b/lib/PublicInbox/DirIdle.pm
@@ -5,48 +5,69 @@
 package PublicInbox::DirIdle;
 use strict;
 use parent 'PublicInbox::DS';
-use PublicInbox::Syscall qw(EPOLLIN EPOLLET);
+use PublicInbox::Syscall qw(EPOLLIN);
 use PublicInbox::In2Tie;
 
 my ($MAIL_IN, $MAIL_GONE, $ino_cls);
 if ($^O eq 'linux' && eval { require Linux::Inotify2; 1 }) {
        $MAIL_IN = Linux::Inotify2::IN_MOVED_TO() |
                Linux::Inotify2::IN_CREATE();
-       $MAIL_GONE = Linux::Inotify2::IN_DELETE();
+       $MAIL_GONE = Linux::Inotify2::IN_DELETE() |
+                       Linux::Inotify2::IN_DELETE_SELF() |
+                       Linux::Inotify2::IN_MOVE_SELF() |
+                       Linux::Inotify2::IN_MOVED_FROM();
        $ino_cls = 'Linux::Inotify2';
 # Perl 5.22+ is needed for fileno(DIRHANDLE) support:
 } elsif ($^V ge v5.22 && eval { require PublicInbox::KQNotify }) {
        $MAIL_IN = PublicInbox::KQNotify::MOVED_TO_OR_CREATE();
-       $MAIL_GONE = PublicInbox::KQNotify::NOTE_DELETE();
+       $MAIL_GONE = PublicInbox::KQNotify::NOTE_DELETE() |
+               PublicInbox::KQNotify::NOTE_REVOKE() |
+               PublicInbox::KQNotify::NOTE_RENAME();
        $ino_cls = 'PublicInbox::KQNotify';
 } else {
        require PublicInbox::FakeInotify;
        $MAIL_IN = PublicInbox::FakeInotify::MOVED_TO_OR_CREATE();
-       $MAIL_GONE = PublicInbox::FakeInotify::IN_DELETE();
+       $MAIL_GONE = PublicInbox::FakeInotify::IN_DELETE() |
+                       PublicInbox::FakeInotify::IN_DELETE_SELF() |
+                       PublicInbox::FakeInotify::IN_MOVE_SELF();
 }
 
 sub new {
-       my ($class, $dirs, $cb, $gone) = @_;
+       my ($class, $cb) = @_;
        my $self = bless { cb => $cb }, $class;
        my $inot;
        if ($ino_cls) {
                $inot = $ino_cls->new or die "E: $ino_cls->new: $!";
                my $io = PublicInbox::In2Tie::io($inot);
-               $self->SUPER::new($io, EPOLLIN | EPOLLET);
+               $self->SUPER::new($io, EPOLLIN);
        } else {
                require PublicInbox::FakeInotify;
                $inot = PublicInbox::FakeInotify->new; # starts timer
        }
-
-       # Linux::Inotify2->watch or similar
-       my $fl = $MAIL_IN;
-       $fl |= $MAIL_GONE if $gone;
-       $inot->watch($_, $fl) for @$dirs;
        $self->{inot} = $inot;
-       PublicInbox::FakeInotify::poll_once($self) if !$ino_cls;
        $self;
 }
 
+sub add_watches {
+       my ($self, $dirs, $gone) = @_;
+       my $fl = $MAIL_IN | ($gone ? $MAIL_GONE : 0);
+       my @ret;
+       for my $d (@$dirs) {
+               my $w = $self->{inot}->watch($d, $fl) or next;
+               push @ret, $w;
+       }
+       PublicInbox::FakeInotify::poll_once($self) if !$ino_cls;
+       @ret
+}
+
+sub rm_watches {
+       my ($self, $dir) = @_;
+       my $inot = $self->{inot};
+       if (my $cb = $inot->can('rm_watches')) { # TODO for fake watchers
+               $cb->($inot, $dir);
+       }
+}
+
 sub event_step {
        my ($self) = @_;
        my $cb = $self->{cb};
@@ -58,4 +79,15 @@ sub event_step {
        warn "$self->{inot}->read err: $@\n" if $@;
 }
 
+sub force_close {
+       my ($self) = @_;
+       my $inot = delete $self->{inot} // return;
+       if ($inot->can('fh')) { # Linux::Inotify2 2.3+
+               close($inot->fh) or warn "CLOSE ERROR: $!";
+       } elsif ($inot->isa('Linux::Inotify2')) {
+               require PublicInbox::LI2Wrap;
+               PublicInbox::LI2Wrap::wrapclose($inot);
+       }
+}
+
 1;