7 use Log::Any::Adapter Stderr => log_level => 'error';
9 use Syntax::Keyword::Try;
12 use Log::Any qw($log);
13 use Time::Duration qw(duration_exact);
14 use Time::Duration::Parse qw(parse_duration);
15 has $log_level :reader = 'warn';
16 has $target_queue_length :reader = 10;
17 has $mpd_host :reader = undef;
18 has $mpd_port :reader = undef;
19 has $db_path :reader = 'mpd-feeder';
20 has $db_user :reader = undef;
21 has $db_password :reader = undef;
22 has $min_album_interval :reader = parse_duration('5h');
23 has $min_song_interval :reader = parse_duration('13d');
24 has $min_artist_interval :reader = parse_duration('1h 15m');
25 has $skip_db_update :reader = 0;
27 method parse_command_line {
28 Getopt::Long::GetOptions(
29 'log-level=s' => \$log_level,
30 'skip-db-update!' => \$skip_db_update,
31 'tql|target-queue-length=n' => sub {
32 $target_queue_length = parse_integer(pop);
34 'mpd-host=s' => \$mpd_host,
35 'mpd-port=s' => \$mpd_port,
36 'db-path=s' => \$db_path,
37 'db-user=s' => \$db_user,
38 'min-album-interval=s' => sub {
39 $min_album_interval = parse_duration(pop);
41 'min-sing-interval=s' => sub {
42 $min_song_interval = parse_duration(pop);
44 'min-artist-interval=s' => sub {
45 $min_artist_interval = parse_duration(pop);
50 sub handle_config_option( $ini, $section, $option, $target_ref,
53 return undef unless exists $ini->{$section}{$option};
55 my $value = $ini->{$section}{$option};
57 $value = $converter->($value) if $converter;
59 $$target_ref = $value;
61 $log->trace("Option $section.$option = $value");
66 say "log_level = $log_level";
69 say "host = " . ( $mpd_host // '' );
70 say "port = " . ( $mpd_port // '' );
73 say "target-length = $target_queue_length";
74 say "min-song-interval = " . duration_exact($min_song_interval);
75 say "min-album-interval = " . duration_exact($min_album_interval);
76 say "min-artist-interval = " . duration_exact($min_artist_interval);
79 say "path = " . ( $db_path // '' );
80 say "user = " . ( $db_user // '' );
81 say "password = " . ( $db_password // '' );
84 sub parse_integer($input) {
85 die "Invalid integer value '$input'" unless $input =~ /^\+?\d{1,18}$/;
89 method parse_config_file($path) {
90 $log->trace("Parsing configuration file $path");
92 use Config::INI::Reader;
93 my $ini = Config::INI::Reader->read_file($path);
95 handle_config_option( $ini => mpd => host => \$mpd_host );
96 handle_config_option( $ini => mpd => port => \$mpd_port );
98 handle_config_option( $ini => 'mpd-feeder' => log_level => \$log_level );
100 handle_config_option(
101 $ini => queue => 'target-length' => \$target_queue_length,
104 handle_config_option(
105 $ini => queue => 'min-song-interval' => \$min_song_interval,
108 handle_config_option(
109 $ini => queue => 'min-album-interval' => \$min_album_interval,
112 handle_config_option(
113 $ini => queue => 'min-artist-interval' => \$min_artist_interval,
117 handle_config_option( $ini => db => path => \$db_path );
118 handle_config_option( $ini => db => user => \$db_user );
119 handle_config_option( $ini => db => password => \$db_password );
121 # FIXME: complain about unknown sections/parameters
126 has $cfg_file :reader;
130 has $db_needs_update :writer = 1;
133 use constant DEFAULT_CONFIG_FILE => '/etc/mpd-feeder/mpd-feeder.conf';
137 use Log::Any qw($log);
138 use IO::Async::Signal;
142 Getopt::Long::Configure('pass_through');
143 Getopt::Long::GetOptions('cfg|config=s' => \$cfg_file);
144 Getopt::Long::Configure('no_pass_through');
146 $cfg_file //= DEFAULT_CONFIG_FILE if -e DEFAULT_CONFIG_FILE;
150 $db_needs_update = 0 if $opt->skip_db_update;
154 my $new_opt = Options->new;
156 $new_opt->parse_config_file($cfg_file) if $cfg_file;
158 $new_opt->parse_command_line;
160 Log::Any::Adapter->set( Stderr => log_level => $new_opt->log_level );
168 my %conn = ( auto_connect => 1 );
169 $conn{host} = $opt->mpd_host if $opt->mpd_host;
170 $conn{port} = $opt->mpd_port if $opt->mpd_port;
172 $mpd = Net::Async::MPD->new(%conn);
175 IO::Async::Signal->new(
178 $log->debug("SIGHUP received. Stopping loop");
179 $mpd->loop->stop('reload');
185 IO::Async::Signal->new(
188 $log->debug("SIGUSR1 received. Dumping configuration to STDERR");
189 my $old = select \*STDERR;
204 $db = DBI->connect( "dbi:Pg:dbname=" . $opt->db_path,
205 $opt->db_user, $opt->db_password,
206 { RaiseError => 1, PrintError => 0, AutoCommit => 1 } );
208 $log->info( "Connected to database " . $opt->db_path );
209 $db_generation = $self->db_get_option('generation');
210 $log->debug("DB generation is $db_generation");
214 method db_get_option($name) {
215 my $sth = $db->prepare_cached("select $name from options");
217 my @result = $sth->fetchrow_array;
222 method db_set_option( $name, $value ) {
223 my $sth = $db->prepare_cached("update options set $name = ?");
224 $sth->execute($value);
227 method db_store_song($song, $artist, $album) {
228 return unless length($song) and length($artist) and length($album);
231 <<'SQL')->execute( $song, $artist, $album, $db_generation );
232 INSERT INTO songs(path, artist, album, generation)
233 VALUES($1, $2, $3, $4)
234 ON CONFLICT ON CONSTRAINT songs_pkey DO
235 UPDATE SET artist = $2
239 $db->prepare_cached(<<'SQL')->execute( $artist, $album, $db_generation );
240 INSERT INTO albums(artist, album, generation)
242 ON CONFLICT ON CONSTRAINT albums_pkey DO
243 UPDATE SET generation = $3
245 $db->prepare_cached(<<'SQL')->execute( $artist, $db_generation );
246 INSERT INTO artists(artist, generation)
248 ON CONFLICT ON CONSTRAINT artists_pkey DO
249 UPDATE SET generation = $2
253 method db_remove_stale_entries {
255 $db->prepare_cached('DELETE FROM songs WHERE generation <> ?');
256 $sth->execute($db_generation);
257 $log->debug( sprintf( "Deleted %d stale songs", $sth->rows ) );
259 $sth = $db->prepare_cached('DELETE FROM albums WHERE generation <> ?');
260 $sth->execute($db_generation);
261 $log->debug( sprintf( "Deleted %d stale albums", $sth->rows ) );
264 $db->prepare_cached('DELETE FROM artists WHERE generation <> ?');
265 $sth->execute($db_generation);
266 $log->debug( sprintf( "Deleted %d stale artists", $sth->rows ) );
269 method db_note_song_qeued($item) {
271 'UPDATE songs SET last_queued=current_timestamp WHERE path=?')
272 ->execute( $item->{song} );
274 'UPDATE artists SET last_queued=CURRENT_TIMESTAMP WHERE artist=?')
275 ->execute( $item->{artist} );
277 'UPDATE albums SET last_queued=CURRENT_TIMESTAMP WHERE artist=? AND album=?'
278 )->execute( $item->{artist}, $item->{album} );
281 method update_db($force = undef) {
282 if (!$db_needs_update and !$force) {
283 $log->debug("Skipping DB update");
287 $log->info('Updating song database');
291 my $rows = $mpd->send('listallinfo')->get;
299 foreach my $entry (@$rows) {
300 next unless exists $entry->{file};
301 $self->db_store_song( $entry->{file},
302 $entry->{AlbumArtist} // $entry->{Artist},
307 $log->info("Updated data about $song_count songs");
309 $self->db_remove_stale_entries;
311 $self->db_set_option( generation => $db_generation );
315 $db_needs_update = 0;
328 method db_find_suitable_songs($num) {
334 SELECT s.path, s.artist, s.album
336 JOIN artists ar ON ar.artist=s.artist
337 JOIN albums al ON al.album=s.album AND al.artist=s.artist
338 WHERE (s.last_queued IS NULL OR s.last_queued < CURRENT_TIMESTAMP - (? || ' seconds')::interval)
339 AND (ar.last_queued IS NULL OR ar.last_queued < CURRENT_TIMESTAMP - (? || ' seconds')::interval)
340 AND (al.last_queued IS NULL OR al.last_queued < CURRENT_TIMESTAMP - (? || ' seconds')::interval)
341 AND NOT EXISTS (SELECT 1 FROM unwanted_artists uar WHERE uar.artist = s.artist)
342 AND NOT EXISTS (SELECT 1 FROM unwanted_albums ual WHERE ual.album = s.album)
347 $opt->min_song_interval, $opt->min_artist_interval,
348 $opt->min_album_interval, $num,
350 my $sth = $db->prepare_cached($sql);
351 $sth->execute(@params);
352 while ( my @row = $sth->fetchrow_array ) {
354 { song => $row[0], artist => $row[1], album => $row[2] };
360 method db_add_unwanted_artist($artist) {
366 INSERT INTO unwanted_artists(artist, generation)
369 undef, $artist, $db_generation
376 $log->debug("PostgreSQL error: $err");
377 $log->debug( "SQLSTATE = " . $db->state );
378 return 0 if $db->state eq '23505';
384 method db_del_unwanted_artist($artist) {
389 DELETE FROM unwanted_artists
396 method queue_songs($num = undef, $callback = undef) {
399 $mpd->send('playlist')->on_done(
401 my $present = scalar @{ $_[0] };
403 $log->notice( "Playlist contains $present songs. Wanted: "
404 . $opt->target_queue_length );
405 if ( $present < $opt->target_queue_length ) {
407 $opt->target_queue_length - $present, $callback );
410 $callback->() if $callback;
418 my @list = $self->db_find_suitable_songs($num);
420 die "Found no suitable songs" unless @list;
422 if ( @list < $num ) {
425 'Found only %d suitable songs instead of %d',
431 $log->info("About to add $num songs to the playlist");
434 for my $song (@list) {
435 my $path = $song->{song};
440 $log->debug( "Adding " . join( ', ', map {"«$_»"} @paths ) );
443 push @commands, [ add => "\"$_\"" ];
446 my $f = $mpd->send( \@commands );
447 $f->on_fail( sub { die @_ } );
450 $self->db_note_song_qeued($_) for @list;
451 $callback->(@_) if $callback;
456 method prepare_to_wait_idle {
457 $log->trace('declaring idle mode');
458 $mpd->send('idle database playlist')->on_done(
462 if ( $result->{changed} eq 'database' ) {
463 $db_needs_update = 1;
464 $self->prepare_to_wait_idle;
466 elsif ( $result->{changed} eq 'playlist' ) {
467 $self->queue_songs( undef,
468 sub { $self->prepare_to_wait_idle } );
473 "Unknown result from idle: " . to_json($result) );
474 $self->prepare_to_wait_idle;
483 die "Connection to MPD lost";
487 $self->prepare_to_wait_idle;
494 if ($db->{ActiveKids}) {
495 $log->warn("$db->{ActiveKids} active DB statements");
496 for my $st ( @{ $db->{ChildHandles} } ) {
497 next unless $st->{Active};
498 while(my($k,$v) = each %$st) {
499 $log->debug("$k = ".($v//'<NULL>'));
510 my $feeder = Feeder->new();
513 my $cmd = shift @ARGV;
515 if ($cmd eq 'dump-config') {
516 die "dump-config command accepts no arguments\n" if @ARGV;
522 if ( $cmd eq 'add-unwanted-artist' ) {
523 die "Missing command arguments\n" unless @ARGV;
524 $feeder->set_db_needs_update(0);
525 for my $artist (@ARGV) {
526 if ( $feeder->db_add_unwanted_artist($artist) ) {
527 $log->info("Artist '$artist' added to the unwanted list\n");
530 $log->warn("Artist '$artist' already in the unwanted list\n");
536 if ( $cmd eq 'del-unwanted-artist' ) {
537 die "Missing command arguments\n" unless @ARGV;
538 $feeder->set_db_needs_update(0);
539 for my $artist (@ARGV) {
540 if ( $feeder->db_del_unwanted_artist($artist) ) {
541 $log->info("Artist '$artist' deleted from the unwanted list\n");
544 $log->warn("Artist '$artist' is not in the unwanted list\n");
550 if ( $cmd eq 'add-unwanted-album' ) {
551 die "NOT IMPLEMENTED\n";
554 if ( $cmd eq 'one-shot' ) {
555 die "one-shot command accepts no arguments\n" if @ARGV;
557 $feeder->queue_songs(undef, sub { exit });
558 $feeder->mpd->loop->run;
560 elsif ( $cmd eq 'single' ) {
561 die "single command accepts no arguments\n" if @ARGV;
563 $feeder->queue_songs(1, sub { exit });
564 $feeder->mpd->loop->run;
567 die "Unknown command '$cmd'";
572 $feeder->queue_songs( undef, sub { $feeder->run } );
574 $log->debug("Entering event loop. PID=$$");
576 my $result = $feeder->mpd->loop->run;
577 $log->trace( "Got loop result of " . ( $result // 'undef' ) );
579 if ('reload' eq $result) {
580 $log->notice("disconnecting");
583 exec( "$0", '--config', $feeder->cfg_file, '--skip-db-update' );