Code

screen_artist: check cursor position in cmd() method
[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         char *key1, *key2;
50         int n;
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 }
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;
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         if (idx >= list->len)
83                 return NULL;
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 GPtrArray *
106 g_list_to_ptr_array(GList *in)
108         GPtrArray *out = g_ptr_array_sized_new(g_list_length(in));
109         GList *head = in;
111         while (in != NULL) {
112                 g_ptr_array_add(out, in->data);
113                 in = g_list_next(in);
114         }
116         g_list_free(head);
117         return out;
120 static void
121 string_array_free(GPtrArray *array)
123         unsigned i;
125         for (i = 0; i < array->len; ++i) {
126                 char *value = g_ptr_array_index(array, i);
127                 g_free(value);
128         }
130         g_ptr_array_free(array, TRUE);
133 static void
134 free_lists(G_GNUC_UNUSED struct mpdclient *c)
136         if (artist_list != NULL) {
137                 string_array_free(artist_list);
138                 artist_list = NULL;
139         }
141         if (album_list != NULL) {
142                 string_array_free(album_list);
143                 album_list = NULL;
144         }
146         if (browser.filelist) {
147                 filelist_free(browser.filelist);
148                 browser.filelist = NULL;
149         }
152 static void
153 load_artist_list(struct mpdclient *c)
155         GList *list;
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);
164         list = mpdclient_get_artists(c);
165         /* sort list */
166         list = g_list_sort(list, compare_utf8);
168         artist_list = g_list_to_ptr_array(list);
171 static void
172 load_album_list(struct mpdclient *c)
174         GList *list;
176         assert(mode == LIST_ALBUMS);
177         assert(artist != NULL);
178         assert(album == NULL);
179         assert(album_list == NULL);
180         assert(browser.filelist == NULL);
182         list = mpdclient_get_albums(c, artist);
183         /* sort list */
184         list = g_list_sort(list, compare_utf8);
186         album_list = g_list_to_ptr_array(list);
189 static void
190 load_song_list(struct mpdclient *c)
192         struct mpd_connection *connection = c->connection;
194         assert(mode == LIST_SONGS);
195         assert(artist != NULL);
196         assert(album != NULL);
197         assert(browser.filelist == NULL);
199         mpd_search_db_songs(connection, true);
200         mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
201                                       MPD_TAG_ARTIST, artist);
202         if (album[0] != 0)
203                 mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
204                                               MPD_TAG_ALBUM, album);
205         mpd_search_commit(connection);
207         browser.filelist = filelist_new_recv(connection);
209         if (!mpd_response_finish(connection))
210                 mpdclient_handle_error(c);
212         /* add a dummy entry for ".." */
213         filelist_prepend(browser.filelist, NULL);
215 #ifndef NCMPC_MINI
216         /* fix highlights */
217         screen_browser_sync_highlights(browser.filelist, &c->playlist);
218 #endif
221 static void
222 free_state(struct mpdclient *c)
224         g_free(artist);
225         g_free(album);
226         artist = NULL;
227         album = NULL;
229         free_lists(c);
232 static void
233 open_artist_list(struct mpdclient *c)
235         free_state(c);
237         mode = LIST_ARTISTS;
238         load_artist_list(c);
241 static void
242 open_album_list(struct mpdclient *c, char *_artist)
244         assert(_artist != NULL);
246         free_state(c);
248         mode = LIST_ALBUMS;
249         artist = _artist;
250         load_album_list(c);
253 static void
254 open_song_list(struct mpdclient *c, char *_artist, char *_album)
256         assert(_artist != NULL);
257         assert(_album != NULL);
259         free_state(c);
261         mode = LIST_SONGS;
262         artist = _artist;
263         album = _album;
264         load_song_list(c);
267 static void
268 reload_lists(struct mpdclient *c)
270         free_lists(c);
272         switch (mode) {
273         case LIST_ARTISTS:
274                 load_artist_list(c);
275                 break;
277         case LIST_ALBUMS:
278                 load_album_list(c);
279                 break;
281         case LIST_SONGS:
282                 load_song_list(c);
283                 break;
284         }
287 static void
288 screen_artist_init(WINDOW *w, int cols, int rows)
290         browser.lw = list_window_init(w, cols, rows);
291         artist = NULL;
292         album = NULL;
295 static void
296 screen_artist_quit(void)
298         free_state(NULL);
299         list_window_free(browser.lw);
302 static void
303 screen_artist_open(struct mpdclient *c)
305         if (artist_list == NULL && album_list == NULL &&
306             browser.filelist == NULL)
307                 reload_lists(c);
310 static void
311 screen_artist_resize(int cols, int rows)
313         browser.lw->cols = cols;
314         browser.lw->rows = rows;
317 static void
318 screen_artist_paint(void)
320         if (browser.filelist) {
321                 list_window_paint(browser.lw, browser_lw_callback,
322                                   browser.filelist);
323         } else if (album_list != NULL)
324                 list_window_paint(browser.lw, screen_artist_lw_callback,
325                                   album_list);
326         else if (artist_list != NULL)
327                 list_window_paint(browser.lw, screen_artist_lw_callback,
328                                   artist_list);
329         else {
330                 wmove(browser.lw->w, 0, 0);
331                 wclrtobot(browser.lw->w);
332         }
335 static const char *
336 screen_artist_get_title(char *str, size_t size)
338         char *s1, *s2;
340         switch(mode) {
341         case LIST_ARTISTS:
342                 g_snprintf(str, size, _("All artists"));
343                 break;
345         case LIST_ALBUMS:
346                 s1 = utf8_to_locale(artist);
347                 g_snprintf(str, size, _("Albums of artist: %s"), s1);
348                 g_free(s1);
349                 break;
351         case LIST_SONGS:
352                 s1 = utf8_to_locale(artist);
353                 if (*album != 0) {
354                         s2 = utf8_to_locale(album);
355                         g_snprintf(str, size,
356                                    _("Album: %s - %s"), s1, s2);
357                         g_free(s2);
358                 } else
359                         g_snprintf(str, size,
360                                    _("All tracks of artist: %s"), s1);
361                 g_free(s1);
362                 break;
363         }
365         return str;
368 static void
369 screen_artist_update(struct mpdclient *c)
371         if (browser.filelist == NULL)
372                 return;
374         if (c->events & MPD_IDLE_DATABASE)
375                 /* the db has changed -> update the list */
376                 reload_lists(c);
378 #ifndef NCMPC_MINI
379         if (c->events & (MPD_IDLE_DATABASE | MPD_IDLE_PLAYLIST))
380                 screen_browser_sync_highlights(browser.filelist, &c->playlist);
381 #endif
383         if (c->events & (MPD_IDLE_DATABASE
384 #ifndef NCMPC_MINI
385                          | MPD_IDLE_PLAYLIST
386 #endif
387                          ))
388                 artist_repaint();
391 static void
392 add_query(struct mpdclient *c, enum mpd_tag_type table, char *_filter)
394         struct mpd_connection *connection = c->connection;
395         char *str;
396         struct filelist *addlist;
398         assert(filter != NULL);
400         str = utf8_to_locale(_filter);
401         if (table == MPD_TAG_ALBUM)
402                 screen_status_printf("Adding album %s...", str);
403         else
404                 screen_status_printf("Adding %s...", str);
405         g_free(str);
407         mpd_search_db_songs(connection, true);
408         mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
409                                       table, _filter);
410         mpd_search_commit(connection);
412         addlist = filelist_new_recv(connection);
414         if (mpd_response_finish(connection))
415                 mpdclient_filelist_add_all(c, addlist);
416         else
417                 mpdclient_handle_error(c);
419         filelist_free(addlist);
422 static unsigned
423 metalist_length(void)
425         assert(mode != LIST_ARTISTS || artist_list != NULL);
426         assert(mode != LIST_ALBUMS || album_list != NULL);
428         return mode == LIST_ALBUMS
429                 ? album_list->len + 2
430                 : artist_list->len;
433 static int
434 screen_artist_lw_cmd(struct mpdclient *c, command_t cmd)
436         switch (mode) {
437         case LIST_ARTISTS:
438         case LIST_ALBUMS:
439                 return list_window_cmd(browser.lw, metalist_length(), cmd);
441         case LIST_SONGS:
442                 return browser_cmd(&browser, c, cmd);
443         }
445         assert(0);
446         return 0;
449 static int
450 string_array_find(GPtrArray *array, const char *value)
452         guint i;
454         for (i = 0; i < array->len; ++i)
455                 if (strcmp((const char*)g_ptr_array_index(array, i),
456                            value) == 0)
457                         return i;
459         return -1;
462 static bool
463 screen_artist_cmd(struct mpdclient *c, command_t cmd)
465         char *selected;
466         char *old;
467         int idx;
469         switch(cmd) {
470         case CMD_PLAY:
471                 switch (mode) {
472                 case LIST_ARTISTS:
473                         if (browser.lw->selected >= artist_list->len)
474                                 return true;
476                         selected = g_ptr_array_index(artist_list,
477                                                      browser.lw->selected);
478                         open_album_list(c, g_strdup(selected));
479                         list_window_reset(browser.lw);
481                         artist_repaint();
482                         return true;
484                 case LIST_ALBUMS:
485                         if (browser.lw->selected == 0) {
486                                 /* handle ".." */
487                                 old = g_strdup(artist);
489                                 open_artist_list(c);
490                                 list_window_reset(browser.lw);
491                                 /* restore previous list window state */
492                                 idx = string_array_find(artist_list, old);
493                                 g_free(old);
495                                 if (idx >= 0) {
496                                         list_window_set_selected(browser.lw, idx);
497                                         list_window_center(browser.lw,
498                                                            artist_list->len, idx);
499                                 }
500                         } else if (browser.lw->selected == album_list->len + 1) {
501                                 /* handle "show all" */
502                                 open_song_list(c, g_strdup(artist), g_strdup("\0"));
503                                 list_window_reset(browser.lw);
504                         } else {
505                                 /* select album */
506                                 selected = g_ptr_array_index(album_list,
507                                                              browser.lw->selected - 1);
508                                 open_song_list(c, g_strdup(artist), g_strdup(selected));
509                                 list_window_reset(browser.lw);
510                         }
512                         artist_repaint();
513                         return true;
515                 case LIST_SONGS:
516                         if (browser.lw->selected == 0) {
517                                 /* handle ".." */
518                                 old = g_strdup(album);
520                                 open_album_list(c, g_strdup(artist));
521                                 list_window_reset(browser.lw);
522                                 /* restore previous list window state */
523                                 idx = *old == 0
524                                         ? (int)album_list->len
525                                         : string_array_find(album_list, old);
526                                 g_free(old);
528                                 if (idx >= 0) {
529                                         ++idx;
530                                         list_window_set_selected(browser.lw, idx);
531                                         list_window_center(browser.lw,
532                                                            album_list->len, idx);
533                                 }
535                                 artist_repaint();
536                                 return true;
537                         }
538                         break;
539                 }
540                 break;
542                 /* FIXME? CMD_GO_* handling duplicates code from CMD_PLAY */
544         case CMD_GO_PARENT_DIRECTORY:
545                 switch (mode) {
546                 case LIST_ARTISTS:
547                         break;
549                 case LIST_ALBUMS:
550                         old = g_strdup(artist);
552                         open_artist_list(c);
553                         list_window_reset(browser.lw);
554                         /* restore previous list window state */
555                         idx = string_array_find(artist_list, old);
556                         g_free(old);
558                         if (idx >= 0) {
559                                 list_window_set_selected(browser.lw, idx);
560                                 list_window_center(browser.lw,
561                                                    artist_list->len, idx);
562                         }
563                         break;
565                 case LIST_SONGS:
566                         old = g_strdup(album);
568                         open_album_list(c, g_strdup(artist));
569                         list_window_reset(browser.lw);
570                         /* restore previous list window state */
571                         idx = *old == 0
572                                 ? (int)album_list->len
573                                 : string_array_find(album_list, old);
574                         g_free(old);
576                         if (idx >= 0) {
577                                 ++idx;
578                                 list_window_set_selected(browser.lw, idx);
579                                 list_window_center(browser.lw,
580                                                    album_list->len, idx);
581                         }
582                         break;
583                 }
585                 artist_repaint();
586                 break;
588         case CMD_GO_ROOT_DIRECTORY:
589                 switch (mode) {
590                 case LIST_ARTISTS:
591                         break;
593                 case LIST_ALBUMS:
594                 case LIST_SONGS:
595                         open_artist_list(c);
596                         list_window_reset(browser.lw);
597                         /* restore first list window state (pop while returning true) */
598                         /* XXX */
599                         break;
600                 }
602                 artist_repaint();
603                 break;
605         case CMD_SELECT:
606         case CMD_ADD:
607                 switch(mode) {
608                 case LIST_ARTISTS:
609                         if (browser.lw->selected >= artist_list->len)
610                                 return true;
612                         for (unsigned i = browser.lw->selected_start;
613                              i <= browser.lw->selected_end; ++i) {
614                                 selected = g_ptr_array_index(artist_list, i);
615                                 add_query(c, MPD_TAG_ARTIST, selected);
616                                 cmd = CMD_LIST_NEXT; /* continue and select next item... */
617                         }
618                         break;
620                 case LIST_ALBUMS:
621                         for (unsigned i = browser.lw->selected_start;
622                              i <= browser.lw->selected_end; ++i) {
623                                 if(i == album_list->len + 1)
624                                         add_query(c, MPD_TAG_ARTIST, artist);
625                                 else if (i > 0)
626                                 {
627                                         selected = g_ptr_array_index(album_list,
628                                                                      browser.lw->selected - 1);
629                                         add_query(c, MPD_TAG_ALBUM, selected);
630                                         cmd = CMD_LIST_NEXT; /* continue and select next item... */
631                                 }
632                         }
633                         break;
635                 case LIST_SONGS:
636                         /* handled by browser_cmd() */
637                         break;
638                 }
639                 break;
641                 /* continue and update... */
642         case CMD_SCREEN_UPDATE:
643                 reload_lists(c);
644                 return false;
646         case CMD_LIST_FIND:
647         case CMD_LIST_RFIND:
648         case CMD_LIST_FIND_NEXT:
649         case CMD_LIST_RFIND_NEXT:
650                 switch (mode) {
651                 case LIST_ARTISTS:
652                         screen_find(browser.lw, artist_list->len,
653                                     cmd, screen_artist_lw_callback,
654                                     artist_list);
655                         artist_repaint();
656                         return true;
658                 case LIST_ALBUMS:
659                         screen_find(browser.lw, album_list->len + 2,
660                                     cmd, screen_artist_lw_callback,
661                                     album_list);
662                         artist_repaint();
663                         return true;
665                 case LIST_SONGS:
666                         /* handled by browser_cmd() */
667                         break;
668                 }
669         case CMD_LIST_JUMP:
670                 switch (mode) {
671                 case LIST_ARTISTS:
672                         screen_jump(browser.lw, screen_artist_lw_callback,
673                                     artist_list);
674                         artist_repaint();
675                         return true;
677                 case LIST_ALBUMS:
678                         screen_jump(browser.lw, screen_artist_lw_callback,
679                                     album_list);
680                         artist_repaint();
681                         return true;
683                 case LIST_SONGS:
684                         /* handled by browser_cmd() */
685                         break;
686                 }
688                 break;
690         default:
691                 break;
692         }
694         if (screen_artist_lw_cmd(c, cmd)) {
695                 if (screen_is_visible(&screen_artist))
696                         artist_repaint();
697                 return true;
698         }
700         return false;
703 const struct screen_functions screen_artist = {
704         .init = screen_artist_init,
705         .exit = screen_artist_quit,
706         .open = screen_artist_open,
707         .resize = screen_artist_resize,
708         .paint = screen_artist_paint,
709         .update = screen_artist_update,
710         .cmd = screen_artist_cmd,
711         .get_title = screen_artist_get_title,
712 };