Code

screen_artist: check connection!=NULL in load_song_list()
[ncmpc.git] / src / screen_artist.c
1 /* ncmpc (Ncurses MPD Client)
2  * (c) 2004-2009 The Music Player Daemon Project
3  * Project homepage: http://musicpd.org
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.
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.
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 */
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"
31 #include <assert.h>
32 #include <string.h>
33 #include <glib.h>
35 #define BUFSIZE 1024
37 typedef enum { LIST_ARTISTS, LIST_ALBUMS, LIST_SONGS } artist_mode_t;
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;
44 static struct screen_browser browser;
46 static gint
47 compare_utf8(gconstpointer s1, gconstpointer s2)
48 {
49         const char *const*t1 = s1, *const*t2 = s2;
50         char *key1, *key2;
51         int n;
53         key1 = g_utf8_collate_key(*t1,-1);
54         key2 = g_utf8_collate_key(*t2,-1);
55         n = strcmp(key1,key2);
56         g_free(key1);
57         g_free(key2);
58         return n;
59 }
61 /* list_window callback */
62 static const char *
63 screen_artist_lw_callback(unsigned idx, void *data)
64 {
65         GPtrArray *list = data;
66         static char buf[BUFSIZE];
67         char *str, *str_utf8;
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                 }
79                 --idx;
80         }
82         assert(idx < list->len);
84         str_utf8 = g_ptr_array_index(list, idx);
85         assert(str_utf8 != NULL);
87         str = utf8_to_locale(str_utf8);
88         g_snprintf(buf, BUFSIZE, "[%s]", str);
89         g_free(str);
91         return buf;
92 }
94 static void
95 screen_artist_paint(void);
97 static void
98 artist_repaint(void)
99 {
100         screen_artist_paint();
101         wrefresh(browser.lw->w);
104 static void
105 string_array_free(GPtrArray *array)
107         unsigned i;
109         for (i = 0; i < array->len; ++i) {
110                 char *value = g_ptr_array_index(array, i);
111                 g_free(value);
112         }
114         g_ptr_array_free(array, TRUE);
117 static void
118 free_lists(void)
120         if (artist_list != NULL) {
121                 string_array_free(artist_list);
122                 artist_list = NULL;
123         }
125         if (album_list != NULL) {
126                 string_array_free(album_list);
127                 album_list = NULL;
128         }
130         if (browser.filelist) {
131                 filelist_free(browser.filelist);
132                 browser.filelist = NULL;
133         }
136 static void
137 recv_tag_values(struct mpd_connection *connection, enum mpd_tag_type tag,
138                 GPtrArray *list)
140         struct mpd_pair *pair;
142         while ((pair = mpd_recv_pair_tag(connection, tag)) != NULL) {
143                 g_ptr_array_add(list, g_strdup(pair->value));
144                 mpd_return_pair(connection, pair);
145         }
148 static void
149 load_artist_list(struct mpdclient *c)
151         struct mpd_connection *connection = mpdclient_get_connection(c);
153         assert(mode == LIST_ARTISTS);
154         assert(artist == NULL);
155         assert(album == NULL);
156         assert(artist_list == NULL);
157         assert(album_list == NULL);
158         assert(browser.filelist == NULL);
160         artist_list = g_ptr_array_new();
162         if (connection != NULL) {
163                 mpd_search_db_tags(connection, MPD_TAG_ARTIST);
164                 mpd_search_commit(connection);
165                 recv_tag_values(connection, MPD_TAG_ARTIST, artist_list);
167                 if (!mpd_response_finish(connection))
168                         mpdclient_handle_error(c);
169         }
171         /* sort list */
172         g_ptr_array_sort(artist_list, compare_utf8);
173         list_window_set_length(browser.lw, artist_list->len);
176 static void
177 load_album_list(struct mpdclient *c)
179         struct mpd_connection *connection = mpdclient_get_connection(c);
181         assert(mode == LIST_ALBUMS);
182         assert(artist != NULL);
183         assert(album == NULL);
184         assert(album_list == NULL);
185         assert(browser.filelist == NULL);
187         album_list = g_ptr_array_new();
189         if (connection != NULL) {
190                 mpd_search_db_tags(connection, MPD_TAG_ALBUM);
191                 mpd_search_add_tag_constraint(connection,
192                                               MPD_OPERATOR_DEFAULT,
193                                               MPD_TAG_ARTIST, artist);
194                 mpd_search_commit(connection);
196                 recv_tag_values(connection, MPD_TAG_ALBUM, album_list);
198                 if (!mpd_response_finish(connection))
199                         mpdclient_handle_error(c);
200         }
202         /* sort list */
203         g_ptr_array_sort(album_list, compare_utf8);
205         list_window_set_length(browser.lw, album_list->len + 2);
208 static void
209 load_song_list(struct mpdclient *c)
211         struct mpd_connection *connection = mpdclient_get_connection(c);
213         assert(mode == LIST_SONGS);
214         assert(artist != NULL);
215         assert(album != NULL);
216         assert(browser.filelist == NULL);
218         browser.filelist = filelist_new();
219         /* add a dummy entry for ".." */
220         filelist_append(browser.filelist, NULL);
222         if (connection != NULL) {
223                 mpd_search_db_songs(connection, true);
224                 mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
225                                               MPD_TAG_ARTIST, artist);
226                 if (album[0] != 0)
227                         mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
228                                                       MPD_TAG_ALBUM, album);
229                 mpd_search_commit(connection);
231                 filelist_recv(browser.filelist, connection);
233                 if (!mpd_response_finish(connection))
234                         mpdclient_handle_error(c);
235         }
237 #ifndef NCMPC_MINI
238         /* fix highlights */
239         screen_browser_sync_highlights(browser.filelist, &c->playlist);
240 #endif
241         list_window_set_length(browser.lw, filelist_length(browser.filelist));
244 static void
245 free_state(void)
247         g_free(artist);
248         g_free(album);
249         artist = NULL;
250         album = NULL;
252         free_lists();
255 static void
256 open_artist_list(struct mpdclient *c)
258         free_state();
260         mode = LIST_ARTISTS;
261         load_artist_list(c);
264 static void
265 open_album_list(struct mpdclient *c, char *_artist)
267         assert(_artist != NULL);
269         free_state();
271         mode = LIST_ALBUMS;
272         artist = _artist;
273         load_album_list(c);
276 static void
277 open_song_list(struct mpdclient *c, char *_artist, char *_album)
279         assert(_artist != NULL);
280         assert(_album != NULL);
282         free_state();
284         mode = LIST_SONGS;
285         artist = _artist;
286         album = _album;
287         load_song_list(c);
290 static void
291 reload_lists(struct mpdclient *c)
293         free_lists();
295         switch (mode) {
296         case LIST_ARTISTS:
297                 load_artist_list(c);
298                 break;
300         case LIST_ALBUMS:
301                 load_album_list(c);
302                 break;
304         case LIST_SONGS:
305                 load_song_list(c);
306                 break;
307         }
310 static void
311 screen_artist_init(WINDOW *w, int cols, int rows)
313         browser.lw = list_window_init(w, cols, rows);
314         artist = NULL;
315         album = NULL;
318 static void
319 screen_artist_quit(void)
321         free_state();
322         list_window_free(browser.lw);
325 static void
326 screen_artist_open(struct mpdclient *c)
328         if (artist_list == NULL && album_list == NULL &&
329             browser.filelist == NULL)
330                 reload_lists(c);
333 static void
334 screen_artist_resize(int cols, int rows)
336         list_window_resize(browser.lw, cols, rows);
339 static void
340 screen_artist_paint(void)
342         if (browser.filelist) {
343                 screen_browser_paint(&browser);
344         } else if (album_list != NULL)
345                 list_window_paint(browser.lw, screen_artist_lw_callback,
346                                   album_list);
347         else if (artist_list != NULL)
348                 list_window_paint(browser.lw, screen_artist_lw_callback,
349                                   artist_list);
350         else {
351                 wmove(browser.lw->w, 0, 0);
352                 wclrtobot(browser.lw->w);
353         }
356 static const char *
357 screen_artist_get_title(char *str, size_t size)
359         char *s1, *s2;
361         switch(mode) {
362         case LIST_ARTISTS:
363                 g_snprintf(str, size, _("All artists"));
364                 break;
366         case LIST_ALBUMS:
367                 s1 = utf8_to_locale(artist);
368                 g_snprintf(str, size, _("Albums of artist: %s"), s1);
369                 g_free(s1);
370                 break;
372         case LIST_SONGS:
373                 s1 = utf8_to_locale(artist);
374                 if (*album != 0) {
375                         s2 = utf8_to_locale(album);
376                         g_snprintf(str, size,
377                                    _("Album: %s - %s"), s1, s2);
378                         g_free(s2);
379                 } else
380                         g_snprintf(str, size,
381                                    _("All tracks of artist: %s"), s1);
382                 g_free(s1);
383                 break;
384         }
386         return str;
389 static void
390 screen_artist_update(struct mpdclient *c)
392         if (browser.filelist == NULL)
393                 return;
395         if (c->events & MPD_IDLE_DATABASE)
396                 /* the db has changed -> update the list */
397                 reload_lists(c);
399 #ifndef NCMPC_MINI
400         if (c->events & (MPD_IDLE_DATABASE | MPD_IDLE_PLAYLIST))
401                 screen_browser_sync_highlights(browser.filelist, &c->playlist);
402 #endif
404         if (c->events & (MPD_IDLE_DATABASE
405 #ifndef NCMPC_MINI
406                          | MPD_IDLE_PLAYLIST
407 #endif
408                          ))
409                 artist_repaint();
412 static void
413 add_query(struct mpdclient *c, enum mpd_tag_type table, char *_filter)
415         struct mpd_connection *connection = mpdclient_get_connection(c);
416         char *str;
417         struct filelist *addlist;
419         assert(filter != NULL);
421         str = utf8_to_locale(_filter);
422         if (table == MPD_TAG_ALBUM)
423                 screen_status_printf("Adding album %s...", str);
424         else
425                 screen_status_printf("Adding %s...", str);
426         g_free(str);
428         mpd_search_db_songs(connection, true);
429         mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
430                                       table, _filter);
431         mpd_search_commit(connection);
433         addlist = filelist_new_recv(connection);
435         if (mpd_response_finish(connection))
436                 mpdclient_filelist_add_all(c, addlist);
437         else
438                 mpdclient_handle_error(c);
440         filelist_free(addlist);
443 static int
444 screen_artist_lw_cmd(struct mpdclient *c, command_t cmd)
446         switch (mode) {
447         case LIST_ARTISTS:
448         case LIST_ALBUMS:
449                 return list_window_cmd(browser.lw, cmd);
451         case LIST_SONGS:
452                 return browser_cmd(&browser, c, cmd);
453         }
455         assert(0);
456         return 0;
459 static int
460 string_array_find(GPtrArray *array, const char *value)
462         guint i;
464         for (i = 0; i < array->len; ++i)
465                 if (strcmp((const char*)g_ptr_array_index(array, i),
466                            value) == 0)
467                         return i;
469         return -1;
472 static bool
473 screen_artist_cmd(struct mpdclient *c, command_t cmd)
475         struct list_window_range range;
476         char *selected;
477         char *old;
478         int idx;
480         switch(cmd) {
481         case CMD_PLAY:
482                 switch (mode) {
483                 case LIST_ARTISTS:
484                         if (browser.lw->selected >= artist_list->len)
485                                 return true;
487                         selected = g_ptr_array_index(artist_list,
488                                                      browser.lw->selected);
489                         open_album_list(c, g_strdup(selected));
490                         list_window_reset(browser.lw);
492                         artist_repaint();
493                         return true;
495                 case LIST_ALBUMS:
496                         if (browser.lw->selected == 0) {
497                                 /* handle ".." */
498                                 old = g_strdup(artist);
500                                 open_artist_list(c);
501                                 list_window_reset(browser.lw);
502                                 /* restore previous list window state */
503                                 idx = string_array_find(artist_list, old);
504                                 g_free(old);
506                                 if (idx >= 0) {
507                                         list_window_set_cursor(browser.lw, idx);
508                                         list_window_center(browser.lw, idx);
509                                 }
510                         } else if (browser.lw->selected == album_list->len + 1) {
511                                 /* handle "show all" */
512                                 open_song_list(c, g_strdup(artist), g_strdup("\0"));
513                                 list_window_reset(browser.lw);
514                         } else {
515                                 /* select album */
516                                 selected = g_ptr_array_index(album_list,
517                                                              browser.lw->selected - 1);
518                                 open_song_list(c, g_strdup(artist), g_strdup(selected));
519                                 list_window_reset(browser.lw);
520                         }
522                         artist_repaint();
523                         return true;
525                 case LIST_SONGS:
526                         if (browser.lw->selected == 0) {
527                                 /* handle ".." */
528                                 old = g_strdup(album);
530                                 open_album_list(c, g_strdup(artist));
531                                 list_window_reset(browser.lw);
532                                 /* restore previous list window state */
533                                 idx = *old == 0
534                                         ? (int)album_list->len
535                                         : string_array_find(album_list, old);
536                                 g_free(old);
538                                 if (idx >= 0) {
539                                         ++idx;
540                                         list_window_set_cursor(browser.lw, idx);
541                                         list_window_center(browser.lw, idx);
542                                 }
544                                 artist_repaint();
545                                 return true;
546                         }
547                         break;
548                 }
549                 break;
551                 /* FIXME? CMD_GO_* handling duplicates code from CMD_PLAY */
553         case CMD_GO_PARENT_DIRECTORY:
554                 switch (mode) {
555                 case LIST_ARTISTS:
556                         break;
558                 case LIST_ALBUMS:
559                         old = g_strdup(artist);
561                         open_artist_list(c);
562                         list_window_reset(browser.lw);
563                         /* restore previous list window state */
564                         idx = string_array_find(artist_list, old);
565                         g_free(old);
567                         if (idx >= 0) {
568                                 list_window_set_cursor(browser.lw, idx);
569                                 list_window_center(browser.lw, idx);
570                         }
571                         break;
573                 case LIST_SONGS:
574                         old = g_strdup(album);
576                         open_album_list(c, g_strdup(artist));
577                         list_window_reset(browser.lw);
578                         /* restore previous list window state */
579                         idx = *old == 0
580                                 ? (int)album_list->len
581                                 : string_array_find(album_list, old);
582                         g_free(old);
584                         if (idx >= 0) {
585                                 ++idx;
586                                 list_window_set_cursor(browser.lw, idx);
587                                 list_window_center(browser.lw, idx);
588                         }
589                         break;
590                 }
592                 artist_repaint();
593                 break;
595         case CMD_GO_ROOT_DIRECTORY:
596                 switch (mode) {
597                 case LIST_ARTISTS:
598                         break;
600                 case LIST_ALBUMS:
601                 case LIST_SONGS:
602                         open_artist_list(c);
603                         list_window_reset(browser.lw);
604                         /* restore first list window state (pop while returning true) */
605                         /* XXX */
606                         break;
607                 }
609                 artist_repaint();
610                 break;
612         case CMD_SELECT:
613         case CMD_ADD:
614                 switch(mode) {
615                 case LIST_ARTISTS:
616                         if (browser.lw->selected >= artist_list->len)
617                                 return true;
619                         list_window_get_range(browser.lw, &range);
620                         for (unsigned i = range.start; i < range.end; ++i) {
621                                 selected = g_ptr_array_index(artist_list, i);
622                                 add_query(c, MPD_TAG_ARTIST, selected);
623                                 cmd = CMD_LIST_NEXT; /* continue and select next item... */
624                         }
625                         break;
627                 case LIST_ALBUMS:
628                         list_window_get_range(browser.lw, &range);
629                         for (unsigned i = range.start; i < range.end; ++i) {
630                                 if(i == album_list->len + 1)
631                                         add_query(c, MPD_TAG_ARTIST, artist);
632                                 else if (i > 0)
633                                 {
634                                         selected = g_ptr_array_index(album_list,
635                                                                      browser.lw->selected - 1);
636                                         add_query(c, MPD_TAG_ALBUM, selected);
637                                         cmd = CMD_LIST_NEXT; /* continue and select next item... */
638                                 }
639                         }
640                         break;
642                 case LIST_SONGS:
643                         /* handled by browser_cmd() */
644                         break;
645                 }
646                 break;
648                 /* continue and update... */
649         case CMD_SCREEN_UPDATE:
650                 reload_lists(c);
651                 return false;
653         case CMD_LIST_FIND:
654         case CMD_LIST_RFIND:
655         case CMD_LIST_FIND_NEXT:
656         case CMD_LIST_RFIND_NEXT:
657                 switch (mode) {
658                 case LIST_ARTISTS:
659                         screen_find(browser.lw, cmd,
660                                     screen_artist_lw_callback, artist_list);
661                         artist_repaint();
662                         return true;
664                 case LIST_ALBUMS:
665                         screen_find(browser.lw, cmd,
666                                     screen_artist_lw_callback, album_list);
667                         artist_repaint();
668                         return true;
670                 case LIST_SONGS:
671                         /* handled by browser_cmd() */
672                         break;
673                 }
674         case CMD_LIST_JUMP:
675                 switch (mode) {
676                 case LIST_ARTISTS:
677                         screen_jump(browser.lw, screen_artist_lw_callback,
678                                     NULL, artist_list);
679                         artist_repaint();
680                         return true;
682                 case LIST_ALBUMS:
683                         screen_jump(browser.lw, screen_artist_lw_callback,
684                                     NULL, album_list);
685                         artist_repaint();
686                         return true;
688                 case LIST_SONGS:
689                         /* handled by browser_cmd() */
690                         break;
691                 }
693                 break;
695         default:
696                 break;
697         }
699         if (screen_artist_lw_cmd(c, cmd)) {
700                 if (screen_is_visible(&screen_artist))
701                         artist_repaint();
702                 return true;
703         }
705         return false;
708 const struct screen_functions screen_artist = {
709         .init = screen_artist_init,
710         .exit = screen_artist_quit,
711         .open = screen_artist_open,
712         .resize = screen_artist_resize,
713         .paint = screen_artist_paint,
714         .update = screen_artist_update,
715         .cmd = screen_artist_cmd,
716         .get_title = screen_artist_get_title,
717 };