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