src/screen_artist.c

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