Code

screen_browser: added wrapper function screen_browser_paint()
[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, G_GNUC_UNUSED bool *highlight,
64                           G_GNUC_UNUSED char** sc, G_GNUC_UNUSED void *data)
65 {
66         GPtrArray *list = data;
67         static char buf[BUFSIZE];
68         char *str, *str_utf8;
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                 }
80                 --idx;
81         }
83         assert(idx < list->len);
85         str_utf8 = g_ptr_array_index(list, idx);
86         assert(str_utf8 != NULL);
88         str = utf8_to_locale(str_utf8);
89         g_snprintf(buf, BUFSIZE, "[%s]", str);
90         g_free(str);
92         return buf;
93 }
95 static void
96 screen_artist_paint(void);
98 static void
99 artist_repaint(void)
101         screen_artist_paint();
102         wrefresh(browser.lw->w);
105 static void
106 string_array_free(GPtrArray *array)
108         unsigned i;
110         for (i = 0; i < array->len; ++i) {
111                 char *value = g_ptr_array_index(array, i);
112                 g_free(value);
113         }
115         g_ptr_array_free(array, TRUE);
118 static void
119 free_lists(void)
121         if (artist_list != NULL) {
122                 string_array_free(artist_list);
123                 artist_list = NULL;
124         }
126         if (album_list != NULL) {
127                 string_array_free(album_list);
128                 album_list = NULL;
129         }
131         if (browser.filelist) {
132                 filelist_free(browser.filelist);
133                 browser.filelist = NULL;
134         }
137 static void
138 recv_tag_values(struct mpd_connection *connection, enum mpd_tag_type tag,
139                 GPtrArray *list)
141         struct mpd_pair *pair;
143         while ((pair = mpd_recv_pair_tag(connection, tag)) != NULL) {
144                 g_ptr_array_add(list, g_strdup(pair->value));
145                 mpd_return_pair(connection, pair);
146         }
149 static void
150 load_artist_list(struct mpdclient *c)
152         struct mpd_connection *connection = mpdclient_get_connection(c);
154         assert(mode == LIST_ARTISTS);
155         assert(artist == NULL);
156         assert(album == NULL);
157         assert(artist_list == NULL);
158         assert(album_list == NULL);
159         assert(browser.filelist == NULL);
161         artist_list = g_ptr_array_new();
163         if (connection != NULL) {
164                 mpd_search_db_tags(connection, MPD_TAG_ARTIST);
165                 mpd_search_commit(connection);
166                 recv_tag_values(connection, MPD_TAG_ARTIST, artist_list);
168                 if (!mpd_response_finish(connection))
169                         mpdclient_handle_error(c);
170         }
172         /* sort list */
173         g_ptr_array_sort(artist_list, compare_utf8);
174         list_window_set_length(browser.lw, artist_list->len);
177 static void
178 load_album_list(struct mpdclient *c)
180         struct mpd_connection *connection = mpdclient_get_connection(c);
182         assert(mode == LIST_ALBUMS);
183         assert(artist != NULL);
184         assert(album == NULL);
185         assert(album_list == NULL);
186         assert(browser.filelist == NULL);
188         album_list = g_ptr_array_new();
190         if (connection != NULL) {
191                 mpd_search_db_tags(connection, MPD_TAG_ALBUM);
192                 mpd_search_add_tag_constraint(connection,
193                                               MPD_OPERATOR_DEFAULT,
194                                               MPD_TAG_ARTIST, artist);
195                 mpd_search_commit(connection);
197                 recv_tag_values(connection, MPD_TAG_ALBUM, album_list);
199                 if (!mpd_response_finish(connection))
200                         mpdclient_handle_error(c);
201         }
203         /* sort list */
204         g_ptr_array_sort(album_list, compare_utf8);
206         list_window_set_length(browser.lw, album_list->len + 2);
209 static void
210 load_song_list(struct mpdclient *c)
212         struct mpd_connection *connection = mpdclient_get_connection(c);
214         assert(mode == LIST_SONGS);
215         assert(artist != NULL);
216         assert(album != NULL);
217         assert(browser.filelist == NULL);
219         mpd_search_db_songs(connection, true);
220         mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
221                                       MPD_TAG_ARTIST, artist);
222         if (album[0] != 0)
223                 mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
224                                               MPD_TAG_ALBUM, album);
225         mpd_search_commit(connection);
227         browser.filelist = filelist_new();
228         /* add a dummy entry for ".." */
229         filelist_append(browser.filelist, NULL);
231         filelist_recv(browser.filelist, connection);
233         if (!mpd_response_finish(connection))
234                 mpdclient_handle_error(c);
236 #ifndef NCMPC_MINI
237         /* fix highlights */
238         screen_browser_sync_highlights(browser.filelist, &c->playlist);
239 #endif
240         list_window_set_length(browser.lw, filelist_length(browser.filelist));
243 static void
244 free_state(void)
246         g_free(artist);
247         g_free(album);
248         artist = NULL;
249         album = NULL;
251         free_lists();
254 static void
255 open_artist_list(struct mpdclient *c)
257         free_state();
259         mode = LIST_ARTISTS;
260         load_artist_list(c);
263 static void
264 open_album_list(struct mpdclient *c, char *_artist)
266         assert(_artist != NULL);
268         free_state();
270         mode = LIST_ALBUMS;
271         artist = _artist;
272         load_album_list(c);
275 static void
276 open_song_list(struct mpdclient *c, char *_artist, char *_album)
278         assert(_artist != NULL);
279         assert(_album != NULL);
281         free_state();
283         mode = LIST_SONGS;
284         artist = _artist;
285         album = _album;
286         load_song_list(c);
289 static void
290 reload_lists(struct mpdclient *c)
292         free_lists();
294         switch (mode) {
295         case LIST_ARTISTS:
296                 load_artist_list(c);
297                 break;
299         case LIST_ALBUMS:
300                 load_album_list(c);
301                 break;
303         case LIST_SONGS:
304                 load_song_list(c);
305                 break;
306         }
309 static void
310 screen_artist_init(WINDOW *w, int cols, int rows)
312         browser.lw = list_window_init(w, cols, rows);
313         artist = NULL;
314         album = NULL;
317 static void
318 screen_artist_quit(void)
320         free_state();
321         list_window_free(browser.lw);
324 static void
325 screen_artist_open(struct mpdclient *c)
327         if (artist_list == NULL && album_list == NULL &&
328             browser.filelist == NULL)
329                 reload_lists(c);
332 static void
333 screen_artist_resize(int cols, int rows)
335         list_window_resize(browser.lw, cols, rows);
338 static void
339 screen_artist_paint(void)
341         if (browser.filelist) {
342                 screen_browser_paint(&browser);
343         } else if (album_list != NULL)
344                 list_window_paint(browser.lw, screen_artist_lw_callback,
345                                   album_list);
346         else if (artist_list != NULL)
347                 list_window_paint(browser.lw, screen_artist_lw_callback,
348                                   artist_list);
349         else {
350                 wmove(browser.lw->w, 0, 0);
351                 wclrtobot(browser.lw->w);
352         }
355 static const char *
356 screen_artist_get_title(char *str, size_t size)
358         char *s1, *s2;
360         switch(mode) {
361         case LIST_ARTISTS:
362                 g_snprintf(str, size, _("All artists"));
363                 break;
365         case LIST_ALBUMS:
366                 s1 = utf8_to_locale(artist);
367                 g_snprintf(str, size, _("Albums of artist: %s"), s1);
368                 g_free(s1);
369                 break;
371         case LIST_SONGS:
372                 s1 = utf8_to_locale(artist);
373                 if (*album != 0) {
374                         s2 = utf8_to_locale(album);
375                         g_snprintf(str, size,
376                                    _("Album: %s - %s"), s1, s2);
377                         g_free(s2);
378                 } else
379                         g_snprintf(str, size,
380                                    _("All tracks of artist: %s"), s1);
381                 g_free(s1);
382                 break;
383         }
385         return str;
388 static void
389 screen_artist_update(struct mpdclient *c)
391         if (browser.filelist == NULL)
392                 return;
394         if (c->events & MPD_IDLE_DATABASE)
395                 /* the db has changed -> update the list */
396                 reload_lists(c);
398 #ifndef NCMPC_MINI
399         if (c->events & (MPD_IDLE_DATABASE | MPD_IDLE_PLAYLIST))
400                 screen_browser_sync_highlights(browser.filelist, &c->playlist);
401 #endif
403         if (c->events & (MPD_IDLE_DATABASE
404 #ifndef NCMPC_MINI
405                          | MPD_IDLE_PLAYLIST
406 #endif
407                          ))
408                 artist_repaint();
411 static void
412 add_query(struct mpdclient *c, enum mpd_tag_type table, char *_filter)
414         struct mpd_connection *connection = mpdclient_get_connection(c);
415         char *str;
416         struct filelist *addlist;
418         assert(filter != NULL);
420         str = utf8_to_locale(_filter);
421         if (table == MPD_TAG_ALBUM)
422                 screen_status_printf("Adding album %s...", str);
423         else
424                 screen_status_printf("Adding %s...", str);
425         g_free(str);
427         mpd_search_db_songs(connection, true);
428         mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
429                                       table, _filter);
430         mpd_search_commit(connection);
432         addlist = filelist_new_recv(connection);
434         if (mpd_response_finish(connection))
435                 mpdclient_filelist_add_all(c, addlist);
436         else
437                 mpdclient_handle_error(c);
439         filelist_free(addlist);
442 static int
443 screen_artist_lw_cmd(struct mpdclient *c, command_t cmd)
445         switch (mode) {
446         case LIST_ARTISTS:
447         case LIST_ALBUMS:
448                 return list_window_cmd(browser.lw, cmd);
450         case LIST_SONGS:
451                 return browser_cmd(&browser, c, cmd);
452         }
454         assert(0);
455         return 0;
458 static int
459 string_array_find(GPtrArray *array, const char *value)
461         guint i;
463         for (i = 0; i < array->len; ++i)
464                 if (strcmp((const char*)g_ptr_array_index(array, i),
465                            value) == 0)
466                         return i;
468         return -1;
471 static bool
472 screen_artist_cmd(struct mpdclient *c, command_t cmd)
474         struct list_window_range range;
475         char *selected;
476         char *old;
477         int idx;
479         switch(cmd) {
480         case CMD_PLAY:
481                 switch (mode) {
482                 case LIST_ARTISTS:
483                         if (browser.lw->selected >= artist_list->len)
484                                 return true;
486                         selected = g_ptr_array_index(artist_list,
487                                                      browser.lw->selected);
488                         open_album_list(c, g_strdup(selected));
489                         list_window_reset(browser.lw);
491                         artist_repaint();
492                         return true;
494                 case LIST_ALBUMS:
495                         if (browser.lw->selected == 0) {
496                                 /* handle ".." */
497                                 old = g_strdup(artist);
499                                 open_artist_list(c);
500                                 list_window_reset(browser.lw);
501                                 /* restore previous list window state */
502                                 idx = string_array_find(artist_list, old);
503                                 g_free(old);
505                                 if (idx >= 0) {
506                                         list_window_set_cursor(browser.lw, idx);
507                                         list_window_center(browser.lw, idx);
508                                 }
509                         } else if (browser.lw->selected == album_list->len + 1) {
510                                 /* handle "show all" */
511                                 open_song_list(c, g_strdup(artist), g_strdup("\0"));
512                                 list_window_reset(browser.lw);
513                         } else {
514                                 /* select album */
515                                 selected = g_ptr_array_index(album_list,
516                                                              browser.lw->selected - 1);
517                                 open_song_list(c, g_strdup(artist), g_strdup(selected));
518                                 list_window_reset(browser.lw);
519                         }
521                         artist_repaint();
522                         return true;
524                 case LIST_SONGS:
525                         if (browser.lw->selected == 0) {
526                                 /* handle ".." */
527                                 old = g_strdup(album);
529                                 open_album_list(c, g_strdup(artist));
530                                 list_window_reset(browser.lw);
531                                 /* restore previous list window state */
532                                 idx = *old == 0
533                                         ? (int)album_list->len
534                                         : string_array_find(album_list, old);
535                                 g_free(old);
537                                 if (idx >= 0) {
538                                         ++idx;
539                                         list_window_set_cursor(browser.lw, idx);
540                                         list_window_center(browser.lw, idx);
541                                 }
543                                 artist_repaint();
544                                 return true;
545                         }
546                         break;
547                 }
548                 break;
550                 /* FIXME? CMD_GO_* handling duplicates code from CMD_PLAY */
552         case CMD_GO_PARENT_DIRECTORY:
553                 switch (mode) {
554                 case LIST_ARTISTS:
555                         break;
557                 case LIST_ALBUMS:
558                         old = g_strdup(artist);
560                         open_artist_list(c);
561                         list_window_reset(browser.lw);
562                         /* restore previous list window state */
563                         idx = string_array_find(artist_list, old);
564                         g_free(old);
566                         if (idx >= 0) {
567                                 list_window_set_cursor(browser.lw, idx);
568                                 list_window_center(browser.lw, idx);
569                         }
570                         break;
572                 case LIST_SONGS:
573                         old = g_strdup(album);
575                         open_album_list(c, g_strdup(artist));
576                         list_window_reset(browser.lw);
577                         /* restore previous list window state */
578                         idx = *old == 0
579                                 ? (int)album_list->len
580                                 : string_array_find(album_list, old);
581                         g_free(old);
583                         if (idx >= 0) {
584                                 ++idx;
585                                 list_window_set_cursor(browser.lw, idx);
586                                 list_window_center(browser.lw, idx);
587                         }
588                         break;
589                 }
591                 artist_repaint();
592                 break;
594         case CMD_GO_ROOT_DIRECTORY:
595                 switch (mode) {
596                 case LIST_ARTISTS:
597                         break;
599                 case LIST_ALBUMS:
600                 case LIST_SONGS:
601                         open_artist_list(c);
602                         list_window_reset(browser.lw);
603                         /* restore first list window state (pop while returning true) */
604                         /* XXX */
605                         break;
606                 }
608                 artist_repaint();
609                 break;
611         case CMD_SELECT:
612         case CMD_ADD:
613                 switch(mode) {
614                 case LIST_ARTISTS:
615                         if (browser.lw->selected >= artist_list->len)
616                                 return true;
618                         list_window_get_range(browser.lw, &range);
619                         for (unsigned i = range.start; i < range.end; ++i) {
620                                 selected = g_ptr_array_index(artist_list, i);
621                                 add_query(c, MPD_TAG_ARTIST, selected);
622                                 cmd = CMD_LIST_NEXT; /* continue and select next item... */
623                         }
624                         break;
626                 case LIST_ALBUMS:
627                         list_window_get_range(browser.lw, &range);
628                         for (unsigned i = range.start; i < range.end; ++i) {
629                                 if(i == album_list->len + 1)
630                                         add_query(c, MPD_TAG_ARTIST, artist);
631                                 else if (i > 0)
632                                 {
633                                         selected = g_ptr_array_index(album_list,
634                                                                      browser.lw->selected - 1);
635                                         add_query(c, MPD_TAG_ALBUM, selected);
636                                         cmd = CMD_LIST_NEXT; /* continue and select next item... */
637                                 }
638                         }
639                         break;
641                 case LIST_SONGS:
642                         /* handled by browser_cmd() */
643                         break;
644                 }
645                 break;
647                 /* continue and update... */
648         case CMD_SCREEN_UPDATE:
649                 reload_lists(c);
650                 return false;
652         case CMD_LIST_FIND:
653         case CMD_LIST_RFIND:
654         case CMD_LIST_FIND_NEXT:
655         case CMD_LIST_RFIND_NEXT:
656                 switch (mode) {
657                 case LIST_ARTISTS:
658                         screen_find(browser.lw, cmd,
659                                     screen_artist_lw_callback, artist_list);
660                         artist_repaint();
661                         return true;
663                 case LIST_ALBUMS:
664                         screen_find(browser.lw, cmd,
665                                     screen_artist_lw_callback, album_list);
666                         artist_repaint();
667                         return true;
669                 case LIST_SONGS:
670                         /* handled by browser_cmd() */
671                         break;
672                 }
673         case CMD_LIST_JUMP:
674                 switch (mode) {
675                 case LIST_ARTISTS:
676                         screen_jump(browser.lw, screen_artist_lw_callback,
677                                     artist_list);
678                         artist_repaint();
679                         return true;
681                 case LIST_ALBUMS:
682                         screen_jump(browser.lw, screen_artist_lw_callback,
683                                     album_list);
684                         artist_repaint();
685                         return true;
687                 case LIST_SONGS:
688                         /* handled by browser_cmd() */
689                         break;
690                 }
692                 break;
694         default:
695                 break;
696         }
698         if (screen_artist_lw_cmd(c, cmd)) {
699                 if (screen_is_visible(&screen_artist))
700                         artist_repaint();
701                 return true;
702         }
704         return false;
707 const struct screen_functions screen_artist = {
708         .init = screen_artist_init,
709         .exit = screen_artist_quit,
710         .open = screen_artist_open,
711         .resize = screen_artist_resize,
712         .paint = screen_artist_paint,
713         .update = screen_artist_update,
714         .cmd = screen_artist_cmd,
715         .get_title = screen_artist_get_title,
716 };