src/screen_artist.c

   1 /*
   2  * (c) 2005 by Kalle Wallin <kaw@linux.se>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  * You should have received a copy of the GNU General Public License
  14  * along with this program; if not, write to the Free Software
  15  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  16  *
  17  */
  18
  19 #include "ncmpc.h"
  20 #include "options.h"
  21 #include "support.h"
  22 #include "mpdclient.h"
  23 #include "utils.h"
  24 #include "strfsong.h"
  25 #include "command.h"
  26 #include "screen.h"
  27 #include "screen_utils.h"
  28 #include "screen_browser.h"
  29 #include "gcc.h"
  30
  31 #include <ctype.h>
  32 #include <stdlib.h>
  33 #include <string.h>
  34 #include <glib.h>
  35 #include <ncurses.h>
  36
  37 #define BUFSIZE 1024
  38
  39 typedef enum { LIST_ARTISTS, LIST_ALBUMS, LIST_SONGS } artist_mode_t;
  40
  41 static artist_mode_t mode = LIST_ARTISTS;
  42 static GPtrArray *artist_list, *album_list;
  43 static char *artist = NULL;
  44 static char *album  = NULL;
  45
  46 static struct screen_browser browser;
  47
  48 static gint
  49 compare_utf8(gconstpointer s1, gconstpointer s2)
  50 {
  51         char *key1, *key2;
  52         int n;
  53
  54         key1 = g_utf8_collate_key(s1,-1);
  55         key2 = g_utf8_collate_key(s2,-1);
  56         n = strcmp(key1,key2);
  57         g_free(key1);
  58         g_free(key2);
  59         return n;
  60 }
  61
  62 /* list_window callback */
  63 static const char *
  64 artist_lw_callback(unsigned idx, mpd_unused int *highlight, mpd_unused void *data)
  65 {
  66         GPtrArray *list = data;
  67         static char buf[BUFSIZE];
  68         char *str, *str_utf8;
  69
  70         if (mode == LIST_ALBUMS) {
  71                 if (idx == 0)
  72                         return "[..]";
  73                 else if (idx == list->len + 1) {
  74                         str = utf8_to_locale(_("All tracks"));
  75                         g_snprintf(buf, BUFSIZE, "[%s]", str);
  76                         g_free(str);
  77                         return buf;
  78                 }
  79
  80                 --idx;
  81         }
  82
  83         if (idx >= list->len)
  84                 return NULL;
  85
  86         str_utf8 = g_ptr_array_index(list, idx);
  87         assert(str_utf8 != NULL);
  88
  89         str = utf8_to_locale(str_utf8);
  90         g_snprintf(buf, BUFSIZE, "[%s]", str);
  91         g_free(str);
  92
  93         return buf;
  94 }
  95
  96 static void
  97 paint(mpdclient_t *c);
  98
  99 static void
 100 artist_repaint(void)
 101 {
 102         paint(NULL);
 103         wrefresh(browser.lw->w);
 104 }
 105
 106 static void
 107 artist_repaint_if_active(void)
 108 {
 109         if (screen_is_visible(&screen_artist))
 110                 artist_repaint();
 111 }
 112
 113 /* the playlist have been updated -> fix highlights */
 114 static void
 115 playlist_changed_callback(mpdclient_t *c, int event, gpointer data)
 116 {
 117         browser_playlist_changed(&browser, c, event, data);
 118
 119         artist_repaint_if_active();
 120 }
 121
 122 static GPtrArray *
 123 g_list_to_ptr_array(GList *in)
 124 {
 125         GPtrArray *out = g_ptr_array_sized_new(g_list_length(in));
 126         GList *head = in;
 127
 128         while (in != NULL) {
 129                 g_ptr_array_add(out, in->data);
 130                 in = g_list_next(in);
 131         }
 132
 133         g_list_free(head);
 134         return out;
 135 }
 136
 137 static void
 138 string_array_free(GPtrArray *array)
 139 {
 140         unsigned i;
 141
 142         for (i = 0; i < array->len; ++i) {
 143                 char *value = g_ptr_array_index(array, i);
 144                 free(value);
 145         }
 146
 147         g_ptr_array_free(array, TRUE);
 148 }
 149
 150 static void
 151 free_lists(struct mpdclient *c)
 152 {
 153         if (artist_list != NULL) {
 154                 string_array_free(artist_list);
 155                 artist_list = NULL;
 156         }
 157
 158         if (album_list != NULL) {
 159                 string_array_free(album_list);
 160                 album_list = NULL;
 161         }
 162
 163         if (browser.filelist) {
 164                 if (c != NULL)
 165                         mpdclient_remove_playlist_callback(c, playlist_changed_callback);
 166                 filelist_free(browser.filelist);
 167                 browser.filelist = NULL;
 168         }
 169 }
 170
 171 static void
 172 load_artist_list(struct mpdclient *c)
 173 {
 174         GList *list;
 175
 176         assert(mode == LIST_ARTISTS);
 177         assert(artist == NULL);
 178         assert(album == NULL);
 179         assert(artist_list == NULL);
 180         assert(album_list == NULL);
 181         assert(browser.filelist == NULL);
 182
 183         list = mpdclient_get_artists_utf8(c);
 184         /* sort list */
 185         list = g_list_sort(list, compare_utf8);
 186
 187         artist_list = g_list_to_ptr_array(list);
 188 }
 189
 190 static void
 191 load_album_list(struct mpdclient *c)
 192 {
 193         GList *list;
 194
 195         assert(mode == LIST_ALBUMS);
 196         assert(artist != NULL);
 197         assert(album == NULL);
 198         assert(album_list == NULL);
 199         assert(browser.filelist == NULL);
 200
 201         list = mpdclient_get_albums_utf8(c, artist);
 202         /* sort list */
 203         list = g_list_sort(list, compare_utf8);
 204
 205         album_list = g_list_to_ptr_array(list);
 206 }
 207
 208 static void
 209 load_song_list(struct mpdclient *c)
 210 {
 211         assert(mode == LIST_SONGS);
 212         assert(artist != NULL);
 213         assert(album != NULL);
 214         assert(browser.filelist == NULL);
 215
 216         if (album[0] == 0) {
 217                 album = g_strdup(_("All tracks"));
 218                 browser.filelist =
 219                         mpdclient_filelist_search_utf8(c, TRUE,
 220                                                        MPD_TABLE_ARTIST,
 221                                                        artist);
 222         } else
 223                 browser.filelist =
 224                         mpdclient_filelist_search_utf8(c, TRUE,
 225                                                        MPD_TABLE_ALBUM,
 226                                                        album);
 227         if (browser.filelist == NULL)
 228                 browser.filelist = filelist_new(NULL);
 229
 230         /* add a dummy entry for ".." */
 231         filelist_prepend(browser.filelist, NULL);
 232
 233         /* install playlist callback and fix highlights */
 234         sync_highlights(c, browser.filelist);
 235         mpdclient_install_playlist_callback(c, playlist_changed_callback);
 236 }
 237
 238 static void
 239 free_state(struct mpdclient *c)
 240 {
 241         g_free(artist);
 242         g_free(album);
 243         artist = NULL;
 244         album = NULL;
 245
 246         free_lists(c);
 247 }
 248
 249 static void
 250 open_artist_list(struct mpdclient *c)
 251 {
 252         free_state(c);
 253
 254         mode = LIST_ARTISTS;
 255         load_artist_list(c);
 256 }
 257
 258 static void
 259 open_album_list(struct mpdclient *c, char *_artist)
 260 {
 261         assert(_artist != NULL);
 262
 263         free_state(c);
 264
 265         mode = LIST_ALBUMS;
 266         artist = _artist;
 267         load_album_list(c);
 268 }
 269
 270 static void
 271 open_song_list(struct mpdclient *c, char *_artist, char *_album)
 272 {
 273         assert(_artist != NULL);
 274         assert(_album != NULL);
 275
 276         free_state(c);
 277
 278         mode = LIST_SONGS;
 279         artist = _artist;
 280         album = _album;
 281         load_song_list(c);
 282 }
 283
 284 static void
 285 reload_lists(struct mpdclient *c)
 286 {
 287         free_lists(c);
 288
 289         switch (mode) {
 290         case LIST_ARTISTS:
 291                 load_artist_list(c);
 292                 break;
 293
 294         case LIST_ALBUMS:
 295                 load_album_list(c);
 296                 break;
 297
 298         case LIST_SONGS:
 299                 load_song_list(c);
 300                 break;
 301         }
 302 }
 303
 304 /* db updated */
 305 static void
 306 browse_callback(mpdclient_t *c, int event, mpd_unused gpointer data)
 307 {
 308         switch(event) {
 309         case BROWSE_DB_UPDATED:
 310                 D("screen_artist.c> browse_callback() [BROWSE_DB_UPDATED]\n");
 311                 reload_lists(c);
 312                 break;
 313         default:
 314                 break;
 315         }
 316
 317         artist_repaint_if_active();
 318 }
 319
 320 static void
 321 init(WINDOW *w, int cols, int rows)
 322 {
 323         browser.lw = list_window_init(w, cols, rows);
 324         browser.lw_state = list_window_init_state();
 325         artist = NULL;
 326         album = NULL;
 327 }
 328
 329 static void
 330 quit(void)
 331 {
 332         free_state(NULL);
 333         list_window_free(browser.lw);
 334         list_window_free_state(browser.lw_state);
 335 }
 336
 337 static void
 338 open(mpd_unused screen_t *screen, mpdclient_t *c)
 339 {
 340         static gboolean callback_installed = FALSE;
 341
 342         if (artist_list == NULL && album_list == NULL &&
 343             browser.filelist == NULL)
 344                 reload_lists(c);
 345         if (!callback_installed) {
 346                 mpdclient_install_browse_callback(c, browse_callback);
 347                 callback_installed = TRUE;
 348         }
 349 }
 350
 351 static void
 352 resize(int cols, int rows)
 353 {
 354         browser.lw->cols = cols;
 355         browser.lw->rows = rows;
 356 }
 357
 358 static void
 359 paint(mpd_unused mpdclient_t *c)
 360 {
 361         if (browser.filelist) {
 362                 list_window_paint(browser.lw, browser_lw_callback,
 363                                   browser.filelist);
 364         } else if (album_list != NULL)
 365                 list_window_paint(browser.lw, artist_lw_callback, album_list);
 366         else if (artist_list != NULL)
 367                 list_window_paint(browser.lw, artist_lw_callback, artist_list);
 368         else {
 369                 wmove(browser.lw->w, 0, 0);
 370                 wclrtobot(browser.lw->w);
 371         }
 372 }
 373
 374 static const char *
 375 get_title(char *str, size_t size)
 376 {
 377         char *s1 = artist ? utf8_to_locale(artist) : NULL;
 378         char *s2 = album ? utf8_to_locale(album) : NULL;
 379
 380         switch(mode) {
 381         case LIST_ARTISTS:
 382                 g_snprintf(str, size,  _("Artist: [db browser - EXPERIMENTAL]"));
 383                 break;
 384         case LIST_ALBUMS:
 385                 g_snprintf(str, size,  _("Artist: %s"), s1);
 386                 break;
 387         case LIST_SONGS:
 388                 g_snprintf(str, size,  _("Artist: %s - %s"), s1, s2);
 389                 break;
 390         }
 391         g_free(s1);
 392         g_free(s2);
 393         return str;
 394 }
 395
 396 static void
 397 add_query(mpdclient_t *c, int table, char *_filter)
 398 {
 399         char *str;
 400         mpdclient_filelist_t *addlist;
 401
 402         assert(filter != NULL);
 403
 404         str = utf8_to_locale(_filter);
 405         if (table== MPD_TABLE_ALBUM)
 406                 screen_status_printf("Adding album %s...", str);
 407         else
 408                 screen_status_printf("Adding %s...", str);
 409         g_free(str);
 410
 411         addlist = mpdclient_filelist_search_utf8(c, TRUE, table, _filter);
 412         if (addlist) {
 413                 mpdclient_filelist_add_all(c, addlist);
 414                 filelist_free(addlist);
 415         }
 416 }
 417
 418 static unsigned
 419 metalist_length(void)
 420 {
 421         assert(mode != LIST_ARTISTS || artist_list != NULL);
 422         assert(mode != LIST_ALBUMS || album_list != NULL);
 423
 424         return mode == LIST_ALBUMS
 425                 ? album_list->len + 2
 426                 : artist_list->len;
 427 }
 428
 429 static int
 430 artist_lw_cmd(command_t cmd)
 431 {
 432         switch (mode) {
 433         case LIST_ARTISTS:
 434         case LIST_ALBUMS:
 435                 return list_window_cmd(browser.lw, metalist_length(), cmd);
 436
 437         case LIST_SONGS:
 438                 return list_window_cmd(browser.lw,
 439                                        filelist_length(browser.filelist),
 440                                        cmd);
 441         }
 442
 443         assert(0);
 444         return 0;
 445 }
 446
 447 static int
 448 artist_cmd(screen_t *screen, mpdclient_t *c, command_t cmd)
 449 {
 450         char *selected;
 451         int ret;
 452
 453         if (artist_lw_cmd(cmd)) {
 454                 artist_repaint();
 455                 return 1;
 456         }
 457
 458         switch(cmd) {
 459         case CMD_PLAY:
 460                 switch (mode) {
 461                 case LIST_ARTISTS:
 462                         selected = g_ptr_array_index(artist_list,
 463                                                      browser.lw->selected);
 464                         open_album_list(c, g_strdup(selected));
 465                         list_window_push_state(browser.lw_state, browser.lw);
 466
 467                         artist_repaint();
 468                         break;
 469
 470                 case LIST_ALBUMS:
 471                         if (browser.lw->selected == 0) {
 472                                 /* handle ".." */
 473
 474                                 open_artist_list(c);
 475                                 list_window_reset(browser.lw);
 476                                 /* restore previous list window state */
 477                                 list_window_pop_state(browser.lw_state, browser.lw);
 478                         } else if (browser.lw->selected == album_list->len + 1) {
 479                                 /* handle "show all" */
 480                                 open_song_list(c, g_strdup(artist), g_strdup("\0"));
 481                                 list_window_push_state(browser.lw_state, browser.lw);
 482                         } else {
 483                                 /* select album */
 484                                 selected = g_ptr_array_index(album_list,
 485                                                              browser.lw->selected - 1);
 486                                 open_song_list(c, g_strdup(artist), g_strdup(selected));
 487                                 list_window_push_state(browser.lw_state, browser.lw);
 488                         }
 489
 490                         artist_repaint();
 491                         break;
 492
 493                 case LIST_SONGS:
 494                         if (browser.lw->selected == 0) {
 495                                 /* handle ".." */
 496
 497                                 open_album_list(c, g_strdup(artist));
 498                                 list_window_reset(browser.lw);
 499                                 /* restore previous list window state */
 500                                 list_window_pop_state(browser.lw_state,
 501                                                       browser.lw);
 502
 503                                 artist_repaint();
 504                         } else
 505                                 browser_handle_enter(&browser, c);
 506                         break;
 507                 }
 508                 return 1;
 509
 510
 511                 /* FIXME? CMD_GO_* handling duplicates code from CMD_PLAY */
 512
 513         case CMD_GO_PARENT_DIRECTORY:
 514                 switch (mode) {
 515                 case LIST_ARTISTS:
 516                         break;
 517
 518                 case LIST_ALBUMS:
 519                         open_artist_list(c);
 520                         list_window_reset(browser.lw);
 521                         /* restore previous list window state */
 522                         list_window_pop_state(browser.lw_state, browser.lw);
 523                         break;
 524
 525                 case LIST_SONGS:
 526                         open_album_list(c, g_strdup(artist));
 527                         list_window_reset(browser.lw);
 528                         /* restore previous list window state */
 529                         list_window_pop_state(browser.lw_state, browser.lw);
 530                         break;
 531                 }
 532
 533                 artist_repaint();
 534                 break;
 535
 536         case CMD_GO_ROOT_DIRECTORY:
 537                 switch (mode) {
 538                 case LIST_ARTISTS:
 539                         break;
 540
 541                 case LIST_ALBUMS:
 542                 case LIST_SONGS:
 543                         open_artist_list(c);
 544                         list_window_reset(browser.lw);
 545                         /* restore first list window state (pop while returning true) */
 546                         while(list_window_pop_state(browser.lw_state, browser.lw));
 547                         break;
 548                 }
 549
 550                 artist_repaint();
 551                 break;
 552
 553         case CMD_SELECT:
 554         case CMD_ADD:
 555                 switch(mode) {
 556                 case LIST_ARTISTS:
 557                         selected = g_ptr_array_index(artist_list,
 558                                                      browser.lw->selected);
 559                         add_query(c, MPD_TABLE_ARTIST, selected);
 560                         cmd = CMD_LIST_NEXT; /* continue and select next item... */
 561                         break;
 562
 563                 case LIST_ALBUMS:
 564                         if (browser.lw->selected == album_list->len + 1)
 565                                 add_query(c, MPD_TABLE_ARTIST, artist);
 566                         else if (browser.lw->selected > 0) {
 567                                 selected = g_ptr_array_index(album_list,
 568                                                              browser.lw->selected - 1);
 569                                 add_query(c, MPD_TABLE_ALBUM, selected);
 570                                 cmd = CMD_LIST_NEXT; /* continue and select next item... */
 571                         }
 572                         break;
 573
 574                 case LIST_SONGS:
 575                         ret = cmd == CMD_SELECT
 576                                 ? browser_handle_select(&browser, c)
 577                                 : browser_handle_add(&browser, c);
 578                         if (ret == 0)
 579                                 /* continue and select next item... */
 580                                 cmd = CMD_LIST_NEXT;
 581                         break;
 582                 }
 583                 break;
 584
 585                 /* continue and update... */
 586         case CMD_SCREEN_UPDATE:
 587                 reload_lists(c);
 588                 screen_status_printf(_("Screen updated!"));
 589                 return 0;
 590
 591         case CMD_LIST_FIND:
 592         case CMD_LIST_RFIND:
 593         case CMD_LIST_FIND_NEXT:
 594         case CMD_LIST_RFIND_NEXT:
 595                 switch (mode) {
 596                 case LIST_ARTISTS:
 597                         screen_find(screen,
 598                                     browser.lw, artist_list->len,
 599                                     cmd, artist_lw_callback, artist_list);
 600                         break;
 601
 602                 case LIST_ALBUMS:
 603                         screen_find(screen,
 604                                     browser.lw, album_list->len + 2,
 605                                     cmd, artist_lw_callback, album_list);
 606                         break;
 607
 608                 case LIST_SONGS:
 609                         screen_find(screen,
 610                                     browser.lw,
 611                                     filelist_length(browser.filelist),
 612                                     cmd, browser_lw_callback,
 613                                     browser.filelist);
 614                         break;
 615                 }
 616
 617                 artist_repaint();
 618                 return 1;
 619
 620         case CMD_MOUSE_EVENT:
 621                 return browser_handle_mouse_event(&browser, c);
 622
 623         default:
 624                 break;
 625         }
 626
 627         return 0;
 628 }
 629
 630 const struct screen_functions screen_artist = {
 631         .init = init,
 632         .exit = quit,
 633         .open = open,
 634         .resize = resize,
 635         .paint = paint,
 636         .cmd = artist_cmd,
 637         .get_title = get_title,
 638 };