src/screen_artist.c

   1 /* ncmpc (Ncurses MPD Client)
   2  * (c) 2004-2017 The Music Player Daemon Project
   3  * Project homepage: http://musicpd.org
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18  */
  19
  20 #include "screen_artist.h"
  21 #include "screen_interface.h"
  22 #include "screen_status.h"
  23 #include "screen_find.h"
  24 #include "screen_browser.h"
  25 #include "screen.h"
  26 #include "i18n.h"
  27 #include "charset.h"
  28 #include "mpdclient.h"
  29 #include "screen_browser.h"
  30 #include "filelist.h"
  31 #include "options.h"
  32
  33 #include <assert.h>
  34 #include <string.h>
  35 #include <glib.h>
  36
  37 #define BUFSIZE 1024
  38
  39 typedef enum { LIST_ARTISTS, LIST_ALBUMS, LIST_SONGS } artist_mode_t;
  40
  41 static artist_mode_t mode = LIST_ARTISTS;
  42 static GPtrArray *artist_list, *album_list;
  43 static char *artist = NULL;
  44 static char *album  = NULL;
  45 static char ALL_TRACKS[] = "";
  46
  47 static struct screen_browser browser;
  48
  49 static gint
  50 compare_utf8(gconstpointer s1, gconstpointer s2)
  51 {
  52         const char *const*t1 = s1, *const*t2 = s2;
  53
  54         char *key1 = g_utf8_collate_key(*t1,-1);
  55         char *key2 = g_utf8_collate_key(*t2,-1);
  56         int n = strcmp(key1,key2);
  57         g_free(key1);
  58         g_free(key2);
  59         return n;
  60 }
  61
  62 /* list_window callback */
  63 static const char *
  64 screen_artist_lw_callback(unsigned idx, void *data)
  65 {
  66         GPtrArray *list = data;
  67
  68         if (mode == LIST_ALBUMS) {
  69                 if (idx == 0)
  70                         return "..";
  71                 else if (idx == list->len + 1)
  72                         return _("All tracks");
  73
  74                 --idx;
  75         }
  76
  77         assert(idx < list->len);
  78
  79         char *str_utf8 = g_ptr_array_index(list, idx);
  80         assert(str_utf8 != NULL);
  81
  82         char *str = utf8_to_locale(str_utf8);
  83
  84         static char buf[BUFSIZE];
  85         g_strlcpy(buf, str, sizeof(buf));
  86         g_free(str);
  87
  88         return buf;
  89 }
  90
  91 static void
  92 string_array_free(GPtrArray *array)
  93 {
  94         for (unsigned i = 0; i < array->len; ++i) {
  95                 char *value = g_ptr_array_index(array, i);
  96                 g_free(value);
  97         }
  98
  99         g_ptr_array_free(array, TRUE);
 100 }
 101
 102 static void
 103 free_lists(void)
 104 {
 105         if (artist_list != NULL) {
 106                 string_array_free(artist_list);
 107                 artist_list = NULL;
 108         }
 109
 110         if (album_list != NULL) {
 111                 string_array_free(album_list);
 112                 album_list = NULL;
 113         }
 114
 115         if (browser.filelist) {
 116                 filelist_free(browser.filelist);
 117                 browser.filelist = NULL;
 118         }
 119 }
 120
 121 static void
 122 recv_tag_values(struct mpd_connection *connection, enum mpd_tag_type tag,
 123                 GPtrArray *list)
 124 {
 125         struct mpd_pair *pair;
 126
 127         while ((pair = mpd_recv_pair_tag(connection, tag)) != NULL) {
 128                 g_ptr_array_add(list, g_strdup(pair->value));
 129                 mpd_return_pair(connection, pair);
 130         }
 131 }
 132
 133 static void
 134 load_artist_list(struct mpdclient *c)
 135 {
 136         struct mpd_connection *connection = mpdclient_get_connection(c);
 137
 138         assert(mode == LIST_ARTISTS);
 139         assert(artist == NULL);
 140         assert(album == NULL);
 141         assert(artist_list == NULL);
 142         assert(album_list == NULL);
 143         assert(browser.filelist == NULL);
 144
 145         artist_list = g_ptr_array_new();
 146
 147         if (connection != NULL) {
 148                 mpd_search_db_tags(connection, MPD_TAG_ARTIST);
 149                 mpd_search_commit(connection);
 150                 recv_tag_values(connection, MPD_TAG_ARTIST, artist_list);
 151
 152                 mpdclient_finish_command(c);
 153         }
 154
 155         /* sort list */
 156         g_ptr_array_sort(artist_list, compare_utf8);
 157         list_window_set_length(browser.lw, artist_list->len);
 158 }
 159
 160 static void
 161 load_album_list(struct mpdclient *c)
 162 {
 163         struct mpd_connection *connection = mpdclient_get_connection(c);
 164
 165         assert(mode == LIST_ALBUMS);
 166         assert(artist != NULL);
 167         assert(album == NULL);
 168         assert(album_list == NULL);
 169         assert(browser.filelist == NULL);
 170
 171         album_list = g_ptr_array_new();
 172
 173         if (connection != NULL) {
 174                 mpd_search_db_tags(connection, MPD_TAG_ALBUM);
 175                 mpd_search_add_tag_constraint(connection,
 176                                               MPD_OPERATOR_DEFAULT,
 177                                               MPD_TAG_ARTIST, artist);
 178                 mpd_search_commit(connection);
 179
 180                 recv_tag_values(connection, MPD_TAG_ALBUM, album_list);
 181
 182                 mpdclient_finish_command(c);
 183         }
 184
 185         /* sort list */
 186         g_ptr_array_sort(album_list, compare_utf8);
 187
 188         list_window_set_length(browser.lw, album_list->len + 2);
 189 }
 190
 191 static void
 192 load_song_list(struct mpdclient *c)
 193 {
 194         struct mpd_connection *connection = mpdclient_get_connection(c);
 195
 196         assert(mode == LIST_SONGS);
 197         assert(artist != NULL);
 198         assert(album != NULL);
 199         assert(browser.filelist == NULL);
 200
 201         browser.filelist = filelist_new();
 202         /* add a dummy entry for ".." */
 203         filelist_append(browser.filelist, NULL);
 204
 205         if (connection != NULL) {
 206                 mpd_search_db_songs(connection, true);
 207                 mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
 208                                               MPD_TAG_ARTIST, artist);
 209                 if (album != ALL_TRACKS)
 210                         mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
 211                                                       MPD_TAG_ALBUM, album);
 212                 mpd_search_commit(connection);
 213
 214                 filelist_recv(browser.filelist, connection);
 215
 216                 mpdclient_finish_command(c);
 217         }
 218
 219         /* fix highlights */
 220         screen_browser_sync_highlights(browser.filelist, &c->playlist);
 221         list_window_set_length(browser.lw, filelist_length(browser.filelist));
 222 }
 223
 224 static void
 225 free_state(void)
 226 {
 227         g_free(artist);
 228         if (album != ALL_TRACKS)
 229                 g_free(album);
 230         artist = NULL;
 231         album = NULL;
 232
 233         free_lists();
 234 }
 235
 236 static void
 237 open_artist_list(struct mpdclient *c)
 238 {
 239         free_state();
 240
 241         mode = LIST_ARTISTS;
 242         load_artist_list(c);
 243 }
 244
 245 static void
 246 open_album_list(struct mpdclient *c, char *_artist)
 247 {
 248         assert(_artist != NULL);
 249
 250         free_state();
 251
 252         mode = LIST_ALBUMS;
 253         artist = _artist;
 254         load_album_list(c);
 255 }
 256
 257 static void
 258 open_song_list(struct mpdclient *c, char *_artist, char *_album)
 259 {
 260         assert(_artist != NULL);
 261         assert(_album != NULL);
 262
 263         free_state();
 264
 265         mode = LIST_SONGS;
 266         artist = _artist;
 267         album = _album;
 268         load_song_list(c);
 269 }
 270
 271 static void
 272 reload_lists(struct mpdclient *c)
 273 {
 274         free_lists();
 275
 276         switch (mode) {
 277         case LIST_ARTISTS:
 278                 load_artist_list(c);
 279                 break;
 280
 281         case LIST_ALBUMS:
 282                 load_album_list(c);
 283                 break;
 284
 285         case LIST_SONGS:
 286                 load_song_list(c);
 287                 break;
 288         }
 289 }
 290
 291 static void
 292 screen_artist_init(WINDOW *w, int cols, int rows)
 293 {
 294         browser.lw = list_window_init(w, cols, rows);
 295         browser.song_format = options.list_format;
 296         artist = NULL;
 297         album = NULL;
 298 }
 299
 300 static void
 301 screen_artist_quit(void)
 302 {
 303         free_state();
 304         list_window_free(browser.lw);
 305 }
 306
 307 static void
 308 screen_artist_open(struct mpdclient *c)
 309 {
 310         if (artist_list == NULL && album_list == NULL &&
 311             browser.filelist == NULL)
 312                 reload_lists(c);
 313 }
 314
 315 static void
 316 screen_artist_resize(int cols, int rows)
 317 {
 318         list_window_resize(browser.lw, cols, rows);
 319 }
 320
 321 /**
 322  * Paint one item in the artist list.
 323  */
 324 static void
 325 paint_artist_callback(WINDOW *w, unsigned i,
 326                       gcc_unused unsigned y, unsigned width,
 327                       bool selected, const void *data)
 328 {
 329         const GPtrArray *list = data;
 330         char *p = utf8_to_locale(g_ptr_array_index(list, i));
 331
 332         screen_browser_paint_directory(w, width, selected, p);
 333         g_free(p);
 334 }
 335
 336 /**
 337  * Paint one item in the album list.  There are two virtual items
 338  * inserted: at the beginning, there's the special item ".." to go to
 339  * the parent directory, and at the end, there's the item "All tracks"
 340  * to view the tracks of all albums.
 341  */
 342 static void
 343 paint_album_callback(WINDOW *w, unsigned i,
 344                      gcc_unused unsigned y, unsigned width,
 345                      bool selected, const void *data)
 346 {
 347         const GPtrArray *list = data;
 348         const char *p;
 349         char *q = NULL;
 350
 351         if (i == 0)
 352                 p = "..";
 353         else if (i == list->len + 1)
 354                 p = _("All tracks");
 355         else
 356                 p = q = utf8_to_locale(g_ptr_array_index(list, i - 1));
 357
 358         screen_browser_paint_directory(w, width, selected, p);
 359         g_free(q);
 360 }
 361
 362 static void
 363 screen_artist_paint(void)
 364 {
 365         if (browser.filelist) {
 366                 screen_browser_paint(&browser);
 367         } else if (album_list != NULL)
 368                 list_window_paint2(browser.lw,
 369                                    paint_album_callback, album_list);
 370         else if (artist_list != NULL)
 371                 list_window_paint2(browser.lw,
 372                                    paint_artist_callback, artist_list);
 373         else {
 374                 wmove(browser.lw->w, 0, 0);
 375                 wclrtobot(browser.lw->w);
 376         }
 377 }
 378
 379 static const char *
 380 screen_artist_get_title(char *str, size_t size)
 381 {
 382         switch(mode) {
 383                 char *s1, *s2;
 384
 385         case LIST_ARTISTS:
 386                 g_snprintf(str, size, _("All artists"));
 387                 break;
 388
 389         case LIST_ALBUMS:
 390                 s1 = utf8_to_locale(artist);
 391                 g_snprintf(str, size, _("Albums of artist: %s"), s1);
 392                 g_free(s1);
 393                 break;
 394
 395         case LIST_SONGS:
 396                 s1 = utf8_to_locale(artist);
 397
 398                 if (album == ALL_TRACKS)
 399                         g_snprintf(str, size,
 400                                    _("All tracks of artist: %s"), s1);
 401                 else if (*album != 0) {
 402                         s2 = utf8_to_locale(album);
 403                         g_snprintf(str, size, _("Album: %s - %s"), s1, s2);
 404                         g_free(s2);
 405                 } else
 406                         g_snprintf(str, size,
 407                                    _("Tracks of no album of artist: %s"), s1);
 408                 g_free(s1);
 409                 break;
 410         }
 411
 412         return str;
 413 }
 414
 415 static void
 416 screen_artist_update(struct mpdclient *c)
 417 {
 418         if (browser.filelist == NULL)
 419                 return;
 420
 421         if (c->events & MPD_IDLE_DATABASE)
 422                 /* the db has changed -> update the list */
 423                 reload_lists(c);
 424
 425         if (c->events & (MPD_IDLE_DATABASE | MPD_IDLE_QUEUE))
 426                 screen_browser_sync_highlights(browser.filelist, &c->playlist);
 427
 428         if (c->events & (MPD_IDLE_DATABASE
 429 #ifndef NCMPC_MINI
 430                          | MPD_IDLE_QUEUE
 431 #endif
 432                          ))
 433                 screen_artist_paint();
 434 }
 435
 436 /* add_query - Add all songs satisfying specified criteria.
 437    _artist is actually only used in the ALBUM case to distinguish albums with
 438    the same name from different artists. */
 439 static void
 440 add_query(struct mpdclient *c, enum mpd_tag_type table, const char *_filter,
 441           const char *_artist)
 442 {
 443         struct mpd_connection *connection = mpdclient_get_connection(c);
 444
 445         assert(_filter != NULL);
 446
 447         if (connection == NULL)
 448                 return;
 449
 450         char *str = utf8_to_locale(_filter);
 451         if (table == MPD_TAG_ALBUM)
 452                 screen_status_printf(_("Adding album %s..."), str);
 453         else
 454                 screen_status_printf(_("Adding %s..."), str);
 455         g_free(str);
 456
 457         mpd_search_db_songs(connection, true);
 458         mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
 459                                       table, _filter);
 460         if (table == MPD_TAG_ALBUM)
 461                 mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
 462                                               MPD_TAG_ARTIST, _artist);
 463         mpd_search_commit(connection);
 464
 465         struct filelist *addlist = filelist_new_recv(connection);
 466
 467         if (mpdclient_finish_command(c))
 468                 mpdclient_filelist_add_all(c, addlist);
 469
 470         filelist_free(addlist);
 471 }
 472
 473 static int
 474 screen_artist_lw_cmd(struct mpdclient *c, command_t cmd)
 475 {
 476         switch (mode) {
 477         case LIST_ARTISTS:
 478         case LIST_ALBUMS:
 479                 return list_window_cmd(browser.lw, cmd);
 480
 481         case LIST_SONGS:
 482                 return browser_cmd(&browser, c, cmd);
 483         }
 484
 485         assert(0);
 486         return 0;
 487 }
 488
 489 static int
 490 string_array_find(GPtrArray *array, const char *value)
 491 {
 492         guint i;
 493
 494         for (i = 0; i < array->len; ++i)
 495                 if (strcmp((const char*)g_ptr_array_index(array, i),
 496                            value) == 0)
 497                         return i;
 498
 499         return -1;
 500 }
 501
 502 static bool
 503 screen_artist_cmd(struct mpdclient *c, command_t cmd)
 504 {
 505         switch(cmd) {
 506                 struct list_window_range range;
 507                 char *selected;
 508                 char *old;
 509                 char *old_ptr;
 510                 int idx;
 511
 512         case CMD_PLAY:
 513                 switch (mode) {
 514                 case LIST_ARTISTS:
 515                         if (browser.lw->selected >= artist_list->len)
 516                                 return true;
 517
 518                         selected = g_ptr_array_index(artist_list,
 519                                                      browser.lw->selected);
 520                         open_album_list(c, g_strdup(selected));
 521                         list_window_reset(browser.lw);
 522
 523                         screen_artist_paint();
 524                         return true;
 525
 526                 case LIST_ALBUMS:
 527                         if (browser.lw->selected == 0) {
 528                                 /* handle ".." */
 529                                 old = g_strdup(artist);
 530
 531                                 open_artist_list(c);
 532                                 list_window_reset(browser.lw);
 533                                 /* restore previous list window state */
 534                                 idx = string_array_find(artist_list, old);
 535                                 g_free(old);
 536
 537                                 if (idx >= 0) {
 538                                         list_window_set_cursor(browser.lw, idx);
 539                                         list_window_center(browser.lw, idx);
 540                                 }
 541                         } else if (browser.lw->selected == album_list->len + 1) {
 542                                 /* handle "show all" */
 543                                 open_song_list(c, g_strdup(artist), ALL_TRACKS);
 544                                 list_window_reset(browser.lw);
 545                         } else {
 546                                 /* select album */
 547                                 selected = g_ptr_array_index(album_list,
 548                                                              browser.lw->selected - 1);
 549                                 open_song_list(c, g_strdup(artist), g_strdup(selected));
 550                                 list_window_reset(browser.lw);
 551                         }
 552
 553                         screen_artist_paint();
 554                         return true;
 555
 556                 case LIST_SONGS:
 557                         if (browser.lw->selected == 0) {
 558                                 /* handle ".." */
 559                                 old = g_strdup(album);
 560                                 old_ptr = album;
 561
 562                                 open_album_list(c, g_strdup(artist));
 563                                 list_window_reset(browser.lw);
 564                                 /* restore previous list window state */
 565                                 idx = old_ptr == ALL_TRACKS
 566                                         ? (int)album_list->len
 567                                         : string_array_find(album_list, old);
 568                                 g_free(old);
 569
 570                                 if (idx >= 0) {
 571                                         ++idx;
 572                                         list_window_set_cursor(browser.lw, idx);
 573                                         list_window_center(browser.lw, idx);
 574                                 }
 575
 576                                 screen_artist_paint();
 577                                 return true;
 578                         }
 579                         break;
 580                 }
 581                 break;
 582
 583                 /* FIXME? CMD_GO_* handling duplicates code from CMD_PLAY */
 584
 585         case CMD_GO_PARENT_DIRECTORY:
 586                 switch (mode) {
 587                 case LIST_ARTISTS:
 588                         break;
 589
 590                 case LIST_ALBUMS:
 591                         old = g_strdup(artist);
 592
 593                         open_artist_list(c);
 594                         list_window_reset(browser.lw);
 595                         /* restore previous list window state */
 596                         idx = string_array_find(artist_list, old);
 597                         g_free(old);
 598
 599                         if (idx >= 0) {
 600                                 list_window_set_cursor(browser.lw, idx);
 601                                 list_window_center(browser.lw, idx);
 602                         }
 603                         break;
 604
 605                 case LIST_SONGS:
 606                         old = g_strdup(album);
 607                         old_ptr = album;
 608
 609                         open_album_list(c, g_strdup(artist));
 610                         list_window_reset(browser.lw);
 611                         /* restore previous list window state */
 612                         idx = old_ptr == ALL_TRACKS
 613                                 ? (int)album_list->len
 614                                 : string_array_find(album_list, old);
 615                         g_free(old);
 616
 617                         if (idx >= 0) {
 618                                 ++idx;
 619                                 list_window_set_cursor(browser.lw, idx);
 620                                 list_window_center(browser.lw, idx);
 621                         }
 622                         break;
 623                 }
 624
 625                 screen_artist_paint();
 626                 break;
 627
 628         case CMD_GO_ROOT_DIRECTORY:
 629                 switch (mode) {
 630                 case LIST_ARTISTS:
 631                         break;
 632
 633                 case LIST_ALBUMS:
 634                 case LIST_SONGS:
 635                         open_artist_list(c);
 636                         list_window_reset(browser.lw);
 637                         /* restore first list window state (pop while returning true) */
 638                         /* XXX */
 639                         break;
 640                 }
 641
 642                 screen_artist_paint();
 643                 break;
 644
 645         case CMD_SELECT:
 646         case CMD_ADD:
 647                 switch(mode) {
 648                 case LIST_ARTISTS:
 649                         if (browser.lw->selected >= artist_list->len)
 650                                 return true;
 651
 652                         list_window_get_range(browser.lw, &range);
 653                         for (unsigned i = range.start; i < range.end; ++i) {
 654                                 selected = g_ptr_array_index(artist_list, i);
 655                                 add_query(c, MPD_TAG_ARTIST, selected, NULL);
 656                                 cmd = CMD_LIST_NEXT; /* continue and select next item... */
 657                         }
 658                         break;
 659
 660                 case LIST_ALBUMS:
 661                         list_window_get_range(browser.lw, &range);
 662                         for (unsigned i = range.start; i < range.end; ++i) {
 663                                 if(i == album_list->len + 1)
 664                                         add_query(c, MPD_TAG_ARTIST, artist, NULL);
 665                                 else if (i > 0)
 666                                 {
 667                                         selected = g_ptr_array_index(album_list,
 668                                                                      browser.lw->selected - 1);
 669                                         add_query(c, MPD_TAG_ALBUM, selected, artist);
 670                                         cmd = CMD_LIST_NEXT; /* continue and select next item... */
 671                                 }
 672                         }
 673                         break;
 674
 675                 case LIST_SONGS:
 676                         /* handled by browser_cmd() */
 677                         break;
 678                 }
 679                 break;
 680
 681                 /* continue and update... */
 682         case CMD_SCREEN_UPDATE:
 683                 reload_lists(c);
 684                 return false;
 685
 686         case CMD_LIST_FIND:
 687         case CMD_LIST_RFIND:
 688         case CMD_LIST_FIND_NEXT:
 689         case CMD_LIST_RFIND_NEXT:
 690                 switch (mode) {
 691                 case LIST_ARTISTS:
 692                         screen_find(browser.lw, cmd,
 693                                     screen_artist_lw_callback, artist_list);
 694                         screen_artist_paint();
 695                         return true;
 696
 697                 case LIST_ALBUMS:
 698                         screen_find(browser.lw, cmd,
 699                                     screen_artist_lw_callback, album_list);
 700                         screen_artist_paint();
 701                         return true;
 702
 703                 case LIST_SONGS:
 704                         /* handled by browser_cmd() */
 705                         break;
 706                 }
 707         case CMD_LIST_JUMP:
 708                 switch (mode) {
 709                 case LIST_ARTISTS:
 710                         screen_jump(browser.lw,
 711                                     screen_artist_lw_callback, artist_list,
 712                                     paint_artist_callback, artist_list);
 713                         screen_artist_paint();
 714                         return true;
 715
 716                 case LIST_ALBUMS:
 717                         screen_jump(browser.lw,
 718                                     screen_artist_lw_callback, album_list,
 719                                     paint_album_callback, album_list);
 720                         screen_artist_paint();
 721                         return true;
 722
 723                 case LIST_SONGS:
 724                         /* handled by browser_cmd() */
 725                         break;
 726                 }
 727
 728                 break;
 729
 730         default:
 731                 break;
 732         }
 733
 734         if (screen_artist_lw_cmd(c, cmd)) {
 735                 if (screen_is_visible(&screen_artist))
 736                         screen_artist_paint();
 737                 return true;
 738         }
 739
 740         return false;
 741 }
 742
 743 const struct screen_functions screen_artist = {
 744         .init = screen_artist_init,
 745         .exit = screen_artist_quit,
 746         .open = screen_artist_open,
 747         .resize = screen_artist_resize,
 748         .paint = screen_artist_paint,
 749         .update = screen_artist_update,
 750         .cmd = screen_artist_cmd,
 751         .get_title = screen_artist_get_title,
 752 };