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