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         list_window_set_length(browser.lw, artist_list->len);
 170 }
 171
 172 static void
 173 load_album_list(struct mpdclient *c)
 174 {
 175         GList *list;
 176
 177         assert(mode == LIST_ALBUMS);
 178         assert(artist != NULL);
 179         assert(album == NULL);
 180         assert(album_list == NULL);
 181         assert(browser.filelist == NULL);
 182
 183         list = mpdclient_get_albums(c, artist);
 184         /* sort list */
 185         list = g_list_sort(list, compare_utf8);
 186
 187         album_list = g_list_to_ptr_array(list);
 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         mpd_search_db_songs(connection, true);
 202         mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
 203                                       MPD_TAG_ARTIST, artist);
 204         if (album[0] != 0)
 205                 mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
 206                                               MPD_TAG_ALBUM, album);
 207         mpd_search_commit(connection);
 208
 209         browser.filelist = filelist_new();
 210         /* add a dummy entry for ".." */
 211         filelist_append(browser.filelist, NULL);
 212
 213         filelist_recv(browser.filelist, connection);
 214
 215         if (!mpd_response_finish(connection))
 216                 mpdclient_handle_error(c);
 217
 218 #ifndef NCMPC_MINI
 219         /* fix highlights */
 220         screen_browser_sync_highlights(browser.filelist, &c->playlist);
 221 #endif
 222         list_window_set_length(browser.lw, filelist_length(browser.filelist));
 223 }
 224
 225 static void
 226 free_state(struct mpdclient *c)
 227 {
 228         g_free(artist);
 229         g_free(album);
 230         artist = NULL;
 231         album = NULL;
 232
 233         free_lists(c);
 234 }
 235
 236 static void
 237 open_artist_list(struct mpdclient *c)
 238 {
 239         free_state(c);
 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(c);
 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(c);
 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(c);
 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         artist = NULL;
 296         album = NULL;
 297 }
 298
 299 static void
 300 screen_artist_quit(void)
 301 {
 302         free_state(NULL);
 303         list_window_free(browser.lw);
 304 }
 305
 306 static void
 307 screen_artist_open(struct mpdclient *c)
 308 {
 309         if (artist_list == NULL && album_list == NULL &&
 310             browser.filelist == NULL)
 311                 reload_lists(c);
 312 }
 313
 314 static void
 315 screen_artist_resize(int cols, int rows)
 316 {
 317         browser.lw->cols = cols;
 318         browser.lw->rows = rows;
 319 }
 320
 321 static void
 322 screen_artist_paint(void)
 323 {
 324         if (browser.filelist) {
 325                 list_window_paint(browser.lw, browser_lw_callback,
 326                                   browser.filelist);
 327         } else if (album_list != NULL)
 328                 list_window_paint(browser.lw, screen_artist_lw_callback,
 329                                   album_list);
 330         else if (artist_list != NULL)
 331                 list_window_paint(browser.lw, screen_artist_lw_callback,
 332                                   artist_list);
 333         else {
 334                 wmove(browser.lw->w, 0, 0);
 335                 wclrtobot(browser.lw->w);
 336         }
 337 }
 338
 339 static const char *
 340 screen_artist_get_title(char *str, size_t size)
 341 {
 342         char *s1, *s2;
 343
 344         switch(mode) {
 345         case LIST_ARTISTS:
 346                 g_snprintf(str, size, _("All artists"));
 347                 break;
 348
 349         case LIST_ALBUMS:
 350                 s1 = utf8_to_locale(artist);
 351                 g_snprintf(str, size, _("Albums of artist: %s"), s1);
 352                 g_free(s1);
 353                 break;
 354
 355         case LIST_SONGS:
 356                 s1 = utf8_to_locale(artist);
 357                 if (*album != 0) {
 358                         s2 = utf8_to_locale(album);
 359                         g_snprintf(str, size,
 360                                    _("Album: %s - %s"), s1, s2);
 361                         g_free(s2);
 362                 } else
 363                         g_snprintf(str, size,
 364                                    _("All tracks of artist: %s"), s1);
 365                 g_free(s1);
 366                 break;
 367         }
 368
 369         return str;
 370 }
 371
 372 static void
 373 screen_artist_update(struct mpdclient *c)
 374 {
 375         if (browser.filelist == NULL)
 376                 return;
 377
 378         if (c->events & MPD_IDLE_DATABASE)
 379                 /* the db has changed -> update the list */
 380                 reload_lists(c);
 381
 382 #ifndef NCMPC_MINI
 383         if (c->events & (MPD_IDLE_DATABASE | MPD_IDLE_PLAYLIST))
 384                 screen_browser_sync_highlights(browser.filelist, &c->playlist);
 385 #endif
 386
 387         if (c->events & (MPD_IDLE_DATABASE
 388 #ifndef NCMPC_MINI
 389                          | MPD_IDLE_PLAYLIST
 390 #endif
 391                          ))
 392                 artist_repaint();
 393 }
 394
 395 static void
 396 add_query(struct mpdclient *c, enum mpd_tag_type table, char *_filter)
 397 {
 398         struct mpd_connection *connection = mpdclient_get_connection(c);
 399         char *str;
 400         struct filelist *addlist;
 401
 402         assert(filter != NULL);
 403
 404         str = utf8_to_locale(_filter);
 405         if (table == MPD_TAG_ALBUM)
 406                 screen_status_printf("Adding album %s...", str);
 407         else
 408                 screen_status_printf("Adding %s...", str);
 409         g_free(str);
 410
 411         mpd_search_db_songs(connection, true);
 412         mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
 413                                       table, _filter);
 414         mpd_search_commit(connection);
 415
 416         addlist = filelist_new_recv(connection);
 417
 418         if (mpd_response_finish(connection))
 419                 mpdclient_filelist_add_all(c, addlist);
 420         else
 421                 mpdclient_handle_error(c);
 422
 423         filelist_free(addlist);
 424 }
 425
 426 static int
 427 screen_artist_lw_cmd(struct mpdclient *c, command_t cmd)
 428 {
 429         switch (mode) {
 430         case LIST_ARTISTS:
 431         case LIST_ALBUMS:
 432                 return list_window_cmd(browser.lw, cmd);
 433
 434         case LIST_SONGS:
 435                 return browser_cmd(&browser, c, cmd);
 436         }
 437
 438         assert(0);
 439         return 0;
 440 }
 441
 442 static int
 443 string_array_find(GPtrArray *array, const char *value)
 444 {
 445         guint i;
 446
 447         for (i = 0; i < array->len; ++i)
 448                 if (strcmp((const char*)g_ptr_array_index(array, i),
 449                            value) == 0)
 450                         return i;
 451
 452         return -1;
 453 }
 454
 455 static bool
 456 screen_artist_cmd(struct mpdclient *c, command_t cmd)
 457 {
 458         char *selected;
 459         char *old;
 460         int idx;
 461
 462         switch(cmd) {
 463         case CMD_PLAY:
 464                 switch (mode) {
 465                 case LIST_ARTISTS:
 466                         if (browser.lw->selected >= artist_list->len)
 467                                 return true;
 468
 469                         selected = g_ptr_array_index(artist_list,
 470                                                      browser.lw->selected);
 471                         open_album_list(c, g_strdup(selected));
 472                         list_window_reset(browser.lw);
 473
 474                         artist_repaint();
 475                         return true;
 476
 477                 case LIST_ALBUMS:
 478                         if (browser.lw->selected == 0) {
 479                                 /* handle ".." */
 480                                 old = g_strdup(artist);
 481
 482                                 open_artist_list(c);
 483                                 list_window_reset(browser.lw);
 484                                 /* restore previous list window state */
 485                                 idx = string_array_find(artist_list, old);
 486                                 g_free(old);
 487
 488                                 if (idx >= 0) {
 489                                         list_window_set_cursor(browser.lw, idx);
 490                                         list_window_center(browser.lw, idx);
 491                                 }
 492                         } else if (browser.lw->selected == album_list->len + 1) {
 493                                 /* handle "show all" */
 494                                 open_song_list(c, g_strdup(artist), g_strdup("\0"));
 495                                 list_window_reset(browser.lw);
 496                         } else {
 497                                 /* select album */
 498                                 selected = g_ptr_array_index(album_list,
 499                                                              browser.lw->selected - 1);
 500                                 open_song_list(c, g_strdup(artist), g_strdup(selected));
 501                                 list_window_reset(browser.lw);
 502                         }
 503
 504                         artist_repaint();
 505                         return true;
 506
 507                 case LIST_SONGS:
 508                         if (browser.lw->selected == 0) {
 509                                 /* handle ".." */
 510                                 old = g_strdup(album);
 511
 512                                 open_album_list(c, g_strdup(artist));
 513                                 list_window_reset(browser.lw);
 514                                 /* restore previous list window state */
 515                                 idx = *old == 0
 516                                         ? (int)album_list->len
 517                                         : string_array_find(album_list, old);
 518                                 g_free(old);
 519
 520                                 if (idx >= 0) {
 521                                         ++idx;
 522                                         list_window_set_cursor(browser.lw, idx);
 523                                         list_window_center(browser.lw, idx);
 524                                 }
 525
 526                                 artist_repaint();
 527                                 return true;
 528                         }
 529                         break;
 530                 }
 531                 break;
 532
 533                 /* FIXME? CMD_GO_* handling duplicates code from CMD_PLAY */
 534
 535         case CMD_GO_PARENT_DIRECTORY:
 536                 switch (mode) {
 537                 case LIST_ARTISTS:
 538                         break;
 539
 540                 case LIST_ALBUMS:
 541                         old = g_strdup(artist);
 542
 543                         open_artist_list(c);
 544                         list_window_reset(browser.lw);
 545                         /* restore previous list window state */
 546                         idx = string_array_find(artist_list, old);
 547                         g_free(old);
 548
 549                         if (idx >= 0) {
 550                                 list_window_set_cursor(browser.lw, idx);
 551                                 list_window_center(browser.lw, idx);
 552                         }
 553                         break;
 554
 555                 case LIST_SONGS:
 556                         old = g_strdup(album);
 557
 558                         open_album_list(c, g_strdup(artist));
 559                         list_window_reset(browser.lw);
 560                         /* restore previous list window state */
 561                         idx = *old == 0
 562                                 ? (int)album_list->len
 563                                 : string_array_find(album_list, old);
 564                         g_free(old);
 565
 566                         if (idx >= 0) {
 567                                 ++idx;
 568                                 list_window_set_cursor(browser.lw, idx);
 569                                 list_window_center(browser.lw, idx);
 570                         }
 571                         break;
 572                 }
 573
 574                 artist_repaint();
 575                 break;
 576
 577         case CMD_GO_ROOT_DIRECTORY:
 578                 switch (mode) {
 579                 case LIST_ARTISTS:
 580                         break;
 581
 582                 case LIST_ALBUMS:
 583                 case LIST_SONGS:
 584                         open_artist_list(c);
 585                         list_window_reset(browser.lw);
 586                         /* restore first list window state (pop while returning true) */
 587                         /* XXX */
 588                         break;
 589                 }
 590
 591                 artist_repaint();
 592                 break;
 593
 594         case CMD_SELECT:
 595         case CMD_ADD:
 596                 switch(mode) {
 597                 case LIST_ARTISTS:
 598                         if (browser.lw->selected >= artist_list->len)
 599                                 return true;
 600
 601                         for (unsigned i = browser.lw->selected_start;
 602                              i <= browser.lw->selected_end; ++i) {
 603                                 selected = g_ptr_array_index(artist_list, i);
 604                                 add_query(c, MPD_TAG_ARTIST, selected);
 605                                 cmd = CMD_LIST_NEXT; /* continue and select next item... */
 606                         }
 607                         break;
 608
 609                 case LIST_ALBUMS:
 610                         for (unsigned i = browser.lw->selected_start;
 611                              i <= browser.lw->selected_end; ++i) {
 612                                 if(i == album_list->len + 1)
 613                                         add_query(c, MPD_TAG_ARTIST, artist);
 614                                 else if (i > 0)
 615                                 {
 616                                         selected = g_ptr_array_index(album_list,
 617                                                                      browser.lw->selected - 1);
 618                                         add_query(c, MPD_TAG_ALBUM, selected);
 619                                         cmd = CMD_LIST_NEXT; /* continue and select next item... */
 620                                 }
 621                         }
 622                         break;
 623
 624                 case LIST_SONGS:
 625                         /* handled by browser_cmd() */
 626                         break;
 627                 }
 628                 break;
 629
 630                 /* continue and update... */
 631         case CMD_SCREEN_UPDATE:
 632                 reload_lists(c);
 633                 return false;
 634
 635         case CMD_LIST_FIND:
 636         case CMD_LIST_RFIND:
 637         case CMD_LIST_FIND_NEXT:
 638         case CMD_LIST_RFIND_NEXT:
 639                 switch (mode) {
 640                 case LIST_ARTISTS:
 641                         screen_find(browser.lw, cmd,
 642                                     screen_artist_lw_callback, artist_list);
 643                         artist_repaint();
 644                         return true;
 645
 646                 case LIST_ALBUMS:
 647                         screen_find(browser.lw, cmd,
 648                                     screen_artist_lw_callback, album_list);
 649                         artist_repaint();
 650                         return true;
 651
 652                 case LIST_SONGS:
 653                         /* handled by browser_cmd() */
 654                         break;
 655                 }
 656         case CMD_LIST_JUMP:
 657                 switch (mode) {
 658                 case LIST_ARTISTS:
 659                         screen_jump(browser.lw, screen_artist_lw_callback,
 660                                     artist_list);
 661                         artist_repaint();
 662                         return true;
 663
 664                 case LIST_ALBUMS:
 665                         screen_jump(browser.lw, screen_artist_lw_callback,
 666                                     album_list);
 667                         artist_repaint();
 668                         return true;
 669
 670                 case LIST_SONGS:
 671                         /* handled by browser_cmd() */
 672                         break;
 673                 }
 674
 675                 break;
 676
 677         default:
 678                 break;
 679         }
 680
 681         if (screen_artist_lw_cmd(c, cmd)) {
 682                 if (screen_is_visible(&screen_artist))
 683                         artist_repaint();
 684                 return true;
 685         }
 686
 687         return false;
 688 }
 689
 690 const struct screen_functions screen_artist = {
 691         .init = screen_artist_init,
 692         .exit = screen_artist_quit,
 693         .open = screen_artist_open,
 694         .resize = screen_artist_resize,
 695         .paint = screen_artist_paint,
 696         .update = screen_artist_update,
 697         .cmd = screen_artist_cmd,
 698         .get_title = screen_artist_get_title,
 699 };