X-Git-Url: http://www.git.stargrave.org/?a=blobdiff_plain;f=script%2Fpublic-inbox-watch;h=af02d8f358f7b84392744b68df49a4328fc69636;hb=165efd91f75d64213eda4cebae11b310f8844300;hp=42ae55ae9b760567e4238738329a7457a59994b5;hpb=619b9d46b91cdf0f523b3804ec9a07b0f1ba4efe;p=public-inbox.git
diff --git a/script/public-inbox-watch b/script/public-inbox-watch
index 42ae55ae..af02d8f3 100755
--- a/script/public-inbox-watch
+++ b/script/public-inbox-watch
@@ -1,16 +1,58 @@
#!/usr/bin/perl -w
-# Copyright (C) 2016 all contributors
+# Copyright (C) 2016-2021 all contributors
# License: AGPL-3.0+
+my $help = <autoflush
+use PublicInbox::Watch;
use PublicInbox::Config;
-my $config = PublicInbox::Config->new;
-my $watch_md = PublicInbox::WatchMaildir->new($config);
-if ($watch_md) {
- my $scan = sub { $watch_md->scan };
- $SIG{USR1} = $scan;
- $SIG{ALRM} = sub { $SIG{ALRM} = 'DEFAULT'; $scan->() };
- alarm(1);
- $watch_md->watch;
+use PublicInbox::DS;
+my $do_scan = 1;
+GetOptions('scan!' => \$do_scan, # undocumented, testing only
+ 'help|h' => \(my $show_help)) or do { print STDERR $help; exit 1 };
+if ($show_help) { print $help; exit 0 };
+my $oldset = PublicInbox::DS::block_signals();
+STDOUT->autoflush(1);
+STDERR->autoflush(1);
+local $0 = $0; # local since this script may be eval-ed
+my $watch = PublicInbox::Watch->new(PublicInbox::Config->new);
+my $reload = sub {
+ my $prev = $watch or return; # SIGQUIT issued
+ $watch->quit;
+ $watch = PublicInbox::Watch->new(PublicInbox::Config->new);
+ if ($watch) {
+ warn("I: reloaded\n");
+ } else {
+ warn("E: reloading failed\n");
+ $watch = $prev;
+ }
+};
+
+if ($watch) {
+ my $scan = sub {
+ return if !$watch;
+ warn "I: scanning\n";
+ $watch->trigger_scan('full');
+ };
+ my $quit = sub {
+ $watch->quit if $watch;
+ $watch = undef;
+ $0 .= ' quitting';
+ };
+ my $sig = {
+ HUP => $reload,
+ USR1 => $scan,
+ CHLD => \&PublicInbox::DS::enqueue_reap,
+ };
+ $sig->{QUIT} = $sig->{TERM} = $sig->{INT} = $quit;
+
+ # --no-scan is only intended for testing atm, undocumented.
+ PublicInbox::DS::requeue($scan) if $do_scan;
+ $watch->watch($sig, $oldset) while ($watch);
}