]> git.ktnx.net Git - mpd-feeder.git/blobdiff - lib/App/MPD/Feeder.pm
working re-connection on disconnection
[mpd-feeder.git] / lib / App / MPD / Feeder.pm
index 3808c91f8195b36fc8983999e800d3f655999ce1..c951c70e8dc6134a7cbdda903042445e43f17d9f 100644 (file)
-package App::MPD::Feeder;
-
-use strict;
-use warnings;
+use v5.28;
 use utf8;
-use feature 'state';
+use Object::Pad;
+class App::MPD::Feeder;
 
-use App::MPD::Feeder::Options;
 use App::MPD::Feeder::DB;
+use App::MPD::Feeder::Options;
+use App::MPD::Feeder::WorkQueue;
 use DBD::Pg;
 use DBI;
 use Getopt::Long;
 use IO::Async::Signal;
+use IO::Async::Timer::Countdown;
+use IO::Async::Timer::Periodic;
 use Log::Any qw($log);
 use Net::Async::MPD;
 use Object::Pad;
 use Syntax::Keyword::Try;
+use Time::Duration qw(duration_exact);
+
+has $cfg_file :reader;
+has $opt :reader;
+has $db :reader;
+has $db_needs_update :writer = 1;
+has $mpd :reader;
+has $mpd_connected = 0;
+has $playlist_needs_filling = 1;
+has $quit_requested = 0;
+has $reload_requested = 0;
+has $idler;
+has $last_mpd_comm;
+has $reconnect_delay = 5;
 
+use constant DEFAULT_CONFIG_FILE => '/etc/mpd-feeder/mpd-feeder.conf';
 
-class App::MPD::Feeder {
-    has $cfg_file :reader;
-    has $opt :reader;
-    has $db :reader;
-    has $db_needs_update :writer = 1;
-    has $mpd :reader;
+ADJUST {
+    Getopt::Long::Configure('pass_through');
+    Getopt::Long::GetOptions( 'cfg|config=s' => \$cfg_file );
+    Getopt::Long::Configure('no_pass_through');
 
-use constant DEFAULT_CONFIG_FILE => '/etc/mpd-feeder/mpd-feeder.conf';
+    $cfg_file //= DEFAULT_CONFIG_FILE if -e DEFAULT_CONFIG_FILE;
 
-    ADJUST {
-        Getopt::Long::Configure('pass_through');
-        Getopt::Long::GetOptions('cfg|config=s' => \$cfg_file);
-        Getopt::Long::Configure('no_pass_through');
+    $self->configure;
 
-        $cfg_file //= DEFAULT_CONFIG_FILE if -e DEFAULT_CONFIG_FILE;
+    $db_needs_update = 0 if $opt->skip_db_update;
+}
 
-        $self->configure;
+method configure {
+    my $new_opt = App::MPD::Feeder::Options->new;
 
-        $db_needs_update = 0 if $opt->skip_db_update;
-    }
+    $new_opt->parse_config_file($cfg_file) if $cfg_file;
 
-    method configure {
-        my $new_opt = App::MPD::Feeder::Options->new;
+    $new_opt->parse_command_line;
 
-        $new_opt->parse_config_file($cfg_file) if $cfg_file;
+    Log::Any::Adapter->set( Stderr => log_level => $new_opt->log_level );
 
-        $new_opt->parse_command_line;
+    $opt = $new_opt;
 
-        Log::Any::Adapter->set( Stderr => log_level => $new_opt->log_level );
+    $reconnect_delay = $opt->initial_reconnect_delay;
 
-        $opt = $new_opt;
+    $db = App::MPD::Feeder::DB->new( opt => $opt );
+}
 
-        $db = App::MPD::Feeder::DB->new( opt => $opt );
-    }
+method init_mpd {
+    return if $mpd;
 
-    method connect_mpd {
-        return if $mpd;
+    my %conn = ( auto_connect => 0 );
+    $conn{host} = $opt->mpd_host if $opt->mpd_host;
+    $conn{port} = $opt->mpd_port if $opt->mpd_port;
 
-        my %conn = ( auto_connect => 1 );
-        $conn{host} = $opt->mpd_host if $opt->mpd_host;
-        $conn{port} = $opt->mpd_port if $opt->mpd_port;
+    $mpd = Net::Async::MPD->new(%conn);
 
-        $mpd = Net::Async::MPD->new(%conn);
+    $mpd->on(
+        close => sub {
+            $log->warn("Connection to MPD lost");
+            $mpd->loop->stop('disconnected');
+            $mpd_connected = 0;
+            $idler->cancel if $idler;
+            undef $mpd;
+            $self->init_mpd;
+        }
+    );
+    $mpd->on(
+        playlist => sub {
+            $playlist_needs_filling = 1;
+        }
+    );
+    $mpd->on(
+        database => sub {
+            $db_needs_update = 1;
+        }
+    );
 
-        $mpd->on(
-            close => sub {
-                die "Connection to MPD lost";
-            }
-        );
+    my $int_signal_handler = sub {
+        state $signal_count = 0;
+        $signal_count++;
 
-        my $int_signal_handler = sub {
-            state $signal_count = 0;
-            $signal_count++;
-            $log->debug("Signal received. Stopping loop");
-            $mpd->loop->stop('quit');
-
-            if ( $signal_count > 1 ) {
-                $log->warn("Another signal received (#$signal_count)");
-                $log->warn("Exiting abruptly");
-                exit 2;
-            }
-        };
-
-        for (qw(TERM INT)) {
-            $mpd->loop->add(
-                IO::Async::Signal->new(
-                    name       => $_,
-                    on_receipt => $int_signal_handler,
-                )
-            );
+        if ( $signal_count > 1 ) {
+            $log->warn("Another signal received (#$signal_count)");
+            $log->warn("Exiting abruptly");
+            exit 2;
         }
 
-        $mpd->loop->add(
-            IO::Async::Signal->new(
-                name       => 'HUP',
-                on_receipt => sub {
-                    $log->debug("SIGHUP received. Scheduling reload");
-                    $mpd->loop->stop('reload');
-                },
-            )
-        );
+        $log->debug("Signal received. Stopping loop");
+        $quit_requested = 1;
+        $mpd->loop->stop('quit');
+        $self->break_idle;
+    };
 
+    for (qw(TERM INT)) {
         $mpd->loop->add(
             IO::Async::Signal->new(
-                name       => 'USR1',
-                on_receipt => sub {
-                    $log->debug("SIGUSR1 received. Dumping configuration to STDERR");
-                    my $old = select \*STDERR;
-                    try {
-                        $opt->dump;
-                    }
-                    finally {
-                        select $old;
-                    }
-                },
+                name       => $_,
+                on_receipt => $int_signal_handler,
             )
         );
     }
 
-    method connect_db {
-        $db->connect($opt);
-        $self->update_db;
+    $mpd->loop->add(
+        IO::Async::Signal->new(
+            name       => 'HUP',
+            on_receipt => sub {
+                $log->debug("SIGHUP received. Scheduling reload");
+                $reload_requested = 1;
+                $mpd->loop->stop('reload');
+                $self->break_idle;
+            },
+        )
+    );
+
+    $mpd->loop->add(
+        IO::Async::Signal->new(
+            name       => 'USR1',
+            on_receipt => sub {
+                $log->debug(
+                    "SIGUSR1 received. Dumping configuration to STDERR");
+                my $old = select \*STDERR;
+                try {
+                    $opt->dump;
+                }
+                finally {
+                    select $old;
+                }
+            },
+        )
+    );
+}
+
+method connect_db {
+    $db->connect($opt);
+}
+
+method update_db( $force = undef ) {
+    if ( !$db_needs_update and !$force ) {
+        $log->debug("Skipping DB update");
+        return;
     }
 
-    method update_db($force = undef) {
-        if (!$db_needs_update and !$force) {
-            $log->debug("Skipping DB update");
-            return;
-        }
+    $log->info('Updating song database');
+
+    my $rows = $mpd->send('listallinfo')->get;
+
+    $log->trace('got all songs from MPD');
+
+    $db->start_update;
+    try {
+        my $song_count;
+
+        foreach my $entry (@$rows) {
+            next unless exists $entry->{file};
 
-        $log->info('Updating song database');
-        $self->connect_mpd;
+            $self->db->store_song( $entry->{file},
+                $entry->{AlbumArtist} // $entry->{Artist},
+                $entry->{Album} );
 
-        my $rows = $mpd->send('listallinfo')->get;
+            $song_count++;
+        }
 
-        $db->start_update;
-        try {
-            my $song_count;
+        my ($total_songs, $total_artists, $total_albums,
+            $new_songs,   $new_artists,   $new_albums
+        ) = $self->db->finish_update;
 
-            foreach my $entry (@$rows) {
-                next unless exists $entry->{file};
+        $log->info(
+            "Updated data about $song_count songs (including $new_songs new), "
+                . "$total_artists artists (including $new_artists new) "
 
-                $self->db->store_song( $entry->{file},
-                    $entry->{AlbumArtist} // $entry->{Artist},
-                    $entry->{Album} );
-                $song_count++;
-            }
+                . "and $total_albums albums (including $new_albums new)"
+        );
 
-            $log->info("Updated data about $song_count songs");
+        $db_needs_update = 0;
+    }
+    catch {
+        my $err = $@;
+        $self->db->cancel_update;
+        die $err;
+    }
+}
 
-            $self->db->remove_stale_entries;
+method queue_songs( $num = undef ) {
+    if ( !defined $num ) {
+        return unless $playlist_needs_filling;
 
-            $self->db->finish_update;
+        $log->trace("Requesting playlist");
+        my $present = $mpd->send('playlist')->get // [];
+        $present = scalar(@$present);
 
-            $db_needs_update = 0;
+        $log->notice( "Playlist contains $present songs. Wanted: "
+                . $opt->target_queue_length );
+        if ( $present < $opt->target_queue_length ) {
+            $self->queue_songs( $opt->target_queue_length - $present );
         }
-        catch {
-            my $err = $@;
-            $self->db->cancel_update;
-            die $err;
+        else {
+            $playlist_needs_filling = 0;
         }
+
+        return;
     }
 
-    method queue_songs($num = undef, $callback = undef) {
-        $self->connect_db;
-        if (!defined $num) {
-            $self->connect_mpd;
-            $mpd->send('playlist')->on_done(
-                sub {
-                    my $present = scalar @{ $_[0] };
-
-                    $log->notice( "Playlist contains $present songs. Wanted: "
-                            . $opt->target_queue_length );
-                    if ( $present < $opt->target_queue_length ) {
-                        $self->queue_songs(
-                            $opt->target_queue_length - $present, $callback );
-                    }
-                    else {
-                        $callback->() if $callback;
-                    }
-                }
-            );
+    my @list = $self->db->find_suitable_songs($num);
 
-            return;
-        }
+    die "Found no suitable songs" unless @list;
+
+    if ( @list < $num ) {
+        $log->warn(
+            sprintf(
+                'Found only %d suitable songs instead of %d',
+                scalar(@list), $num
+            )
+        );
+    }
 
-        my @list = $self->db->find_suitable_songs($num);
+    $log->info("About to add $num songs to the playlist");
 
-        die "Found no suitable songs" unless @list;
+    my @paths;
+    for my $song (@list) {
+        my $path = $song->{song};
+        $path =~ s/"/\\"/g;
+        push @paths, $path;
+    }
 
-        if ( @list < $num ) {
-            $log->warn(
-                sprintf(
-                    'Found only %d suitable songs instead of %d',
-                    scalar(@list), $num
-                )
-            );
+    $log->debug( "Adding " . join( ', ', map {"«$_»"} @paths ) );
+
+    # MPD needs raw bytes
+    utf8::encode($_) for @paths;
+    my @commands;
+    for (@paths) {
+        push @commands, [ add => "\"$_\"" ];
+    }
+    my $f = $mpd->send( \@commands );
+    $f->on_fail( sub { die @_ } );
+    $f->on_done(
+        sub {
+            $self->db->note_song_qeued($_) for @list;
+            $playlist_needs_filling = 0;
         }
+    );
+    $f->get;
+}
 
-        $log->info("About to add $num songs to the playlist");
+method reexec {
+    $log->notice("disconnecting and re-starting");
+    $db->disconnect;
+    undef $mpd;
 
-        my @paths;
-        for my $song (@list) {
-            my $path = $song->{song};
-            $path =~ s/"/\\"/g;
-            push @paths, $path;
+    my @exec = ( $0, '--config', $self->cfg_file, '--skip-db-update' );
+    if ( $log->is_trace ) {
+        $log->trace( 'exec ' . join( ' ', map { /\s/ ? "'$_'" : $_ } @exec ) );
+    }
+    exec(@exec);
+}
+
+method break_idle {
+    if ( $idler && !$idler->is_ready ) {
+        if ($mpd_connected) {
+            $log->trace("hand-sending 'noidle'");
+            undef $idler;
+            $mpd->{mpd_handle}->write("noidle\n");
+        }
+        else {
+            $log->trace("not connected to MPD: skipping 'noidle'");
         }
+    }
+    else {
+        $log->trace("no idler found");
+    }
+}
+
+method sleep_before_reconnection {
+    $log->debug( "Waiting for "
+            . duration_exact($reconnect_delay)
+            . " before re-connecting" );
+
+    $mpd->loop->add(
+        IO::Async::Timer::Countdown->new(
+            delay     => $reconnect_delay,
+            on_expire => sub { $mpd->loop->stop },
+        )->start
+    );
+
+    $reconnect_delay = $reconnect_delay * 1.5;
+    $reconnect_delay = 120 if $reconnect_delay > 120;
+    $mpd->loop->run;
+}
 
-        $log->debug( "Adding " . join( ', ', map {"«$_»"} @paths ) );
-        # MPD needs raw bytes
-        utf8::encode($_) for @paths;
-        my @commands;
-        for (@paths) {
-            push @commands, [ add => "\"$_\"" ];
+method pulse {
+    unless ($mpd_connected) {
+        $log->trace("Connecting to MPD...");
+        my $f = $mpd->connect->await;
+
+        if ( $f->is_done ) {
+            $mpd_connected          = 1;
+            $playlist_needs_filling = 1;
+            $reconnect_delay        = $opt->initial_reconnect_delay;
+            $mpd->loop->later( sub { $self->pulse } );
         }
-        $self->connect_mpd;
-        my $f = $mpd->send( \@commands );
-        $f->on_fail( sub { die @_ } );
-        $f->on_done(
-            sub {
-                $self->db->note_song_qeued($_) for @list;
-                $callback->(@_) if $callback;
-            }
-        );
+        elsif ( $f->is_failed ) {
+            $mpd->loop->stop('disconnected');
+            $log->warn($f->failure);
+            $self->sleep_before_reconnection;
+        }
+        else {
+            die "connect Future neither done nor failed!?";
+        }
+
+        return;
     }
 
-    method prepare_to_wait_idle {
-        $log->trace('declaring idle mode');
-        $mpd->send('idle database playlist')->on_done(
-            sub {
-                my $result = shift;
+    if ($db_needs_update) {
+        $self->update_db;
+        $mpd->loop->later( sub { $self->pulse } );
+        return;
+    }
 
-                if ( $result->{changed} eq 'database' ) {
-                    $db_needs_update = 1;
-                    $self->prepare_to_wait_idle;
-                }
-                elsif ( $result->{changed} eq 'playlist' ) {
-                    $self->queue_songs( undef,
-                        sub { $self->prepare_to_wait_idle } );
-                }
-                else {
-                    use JSON;
-                    $log->warn(
-                        "Unknown result from idle: " . to_json($result) );
-                    $self->prepare_to_wait_idle;
-                }
-            }
-        );
+    if ($playlist_needs_filling) {
+        $self->queue_songs;
+        $mpd->loop->later( sub { $self->pulse } );
+        return;
     }
 
-    method stop {
-        undef $mpd;
+    $log->debug("Waiting idle. PID=$$");
+    $last_mpd_comm = time;
+    $idler         = $mpd->send("idle database playlist");
+    $idler->await;
+
+    $log->trace('got out of idle');
 
-        $db->disconnect;
+    if ( $idler->is_done ) {
+        my $result = $idler->get;
+        undef $idler;
+        if ( ref $result and $result->{changed} ) {
+            my $changed = $result->{changed};
+            $changed = [$changed] unless ref $changed;
+
+            $mpd->emit($_) for @$changed;
+        }
+    }
+    elsif ( $idler->is_cancelled ) {
+        $log->trace("idle was cancelled");
+        undef $idler;
+    }
+    elsif ( $idler->is_failed ) {
+        $log->warn("idle failed: ".$idler->failure);
+        undef $idler;
     }
 
-    method run_loop {
-        $self->connect_mpd;
-        $self->connect_db;
+    $mpd->loop->stop;
+}
 
-        for ( ;; ) {
-            $self->queue_songs( undef, sub { $self->prepare_to_wait_idle } );
+method run_loop {
+    $self->connect_db;
 
-            $log->debug("Entering event loop. PID=$$");
+    $self->init_mpd;
 
-            my $result = $mpd->loop->run;
-            $log->trace( "Got loop result of " . ( $result // 'undef' ) );
+    $mpd->loop->add(
+        IO::Async::Timer::Periodic->new(
+            interval => 60,
+            on_tick  => sub {
+                if (!$mpd_connected) {
+                    $log->trace("Not connected to MPD. Skipping alive check.");
+                    return;
+                }
 
-            if ( 'reload' eq $result ) {
-                $log->notice("disconnecting");
-                $self->stop;
+                if ( time - $last_mpd_comm > 300 ) {
 
-                my @exec = ( $0, '--config', $self->cfg_file, '--skip-db-update' );
-                if ( $log->is_trace ) {
-                    $log->trace( 'exec '
-                            . join( ' ', map { /\s/ ? "'$_'" : $_ } @exec ) );
+                    $log->trace(
+                        "no active MPD communication for more that 5 minutes");
+                    $log->debug("forcing alive check");
+                    $self->break_idle;
+                }
+                else {
+                    $log->trace(
+                        "contacted MPD less than 5 minutes ago. skipping alive check"
+                    );
                 }
-                exec(@exec);
-            }
-
-            if ( 'quit' eq $result ) {
-                $log->trace("quitting because of 'quit' loop result");
-                $self->stop;
-                exit 0;
-            }
+            },
+        )->start
+    );
+
+    for ( ;; ) {
+        if ( $quit_requested ) {
+            $log->trace("about to quit");
+            undef $mpd;
+            $db->disconnect;
+            last;
+        }
+        elsif ( $reload_requested ) {
+            $self->reexec;
+            die "Not reached";
         }
+
+        $log->trace('About to run the loop');
+
+        $mpd->loop->later( sub { $self->pulse } );
+
+        $mpd->loop->run;
     }
 }
 
+1;