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 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);
102 }
104 static GPtrArray *
105 g_list_to_ptr_array(GList *in)
106 {
107 GPtrArray *out = g_ptr_array_sized_new(g_list_length(in));
108 GList *head = in;
110 while (in != NULL) {
111 g_ptr_array_add(out, in->data);
112 in = g_list_next(in);
113 }
115 g_list_free(head);
116 return out;
117 }
119 static void
120 string_array_free(GPtrArray *array)
121 {
122 unsigned i;
124 for (i = 0; i < array->len; ++i) {
125 char *value = g_ptr_array_index(array, i);
126 g_free(value);
127 }
129 g_ptr_array_free(array, TRUE);
130 }
132 static void
133 free_lists(G_GNUC_UNUSED struct mpdclient *c)
134 {
135 if (artist_list != NULL) {
136 string_array_free(artist_list);
137 artist_list = NULL;
138 }
140 if (album_list != NULL) {
141 string_array_free(album_list);
142 album_list = NULL;
143 }
145 if (browser.filelist) {
146 filelist_free(browser.filelist);
147 browser.filelist = NULL;
148 }
149 }
151 static void
152 load_artist_list(struct mpdclient *c)
153 {
154 GList *list;
156 assert(mode == LIST_ARTISTS);
157 assert(artist == NULL);
158 assert(album == NULL);
159 assert(artist_list == NULL);
160 assert(album_list == NULL);
161 assert(browser.filelist == NULL);
163 list = mpdclient_get_artists(c);
164 /* sort list */
165 list = g_list_sort(list, compare_utf8);
167 artist_list = g_list_to_ptr_array(list);
168 list_window_set_length(browser.lw, artist_list->len);
169 }
171 static void
172 load_album_list(struct mpdclient *c)
173 {
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);
187 list_window_set_length(browser.lw, album_list->len + 2);
188 }
190 static void
191 load_song_list(struct mpdclient *c)
192 {
193 struct mpd_connection *connection = mpdclient_get_connection(c);
195 assert(mode == LIST_SONGS);
196 assert(artist != NULL);
197 assert(album != NULL);
198 assert(browser.filelist == NULL);
200 mpd_search_db_songs(connection, true);
201 mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
202 MPD_TAG_ARTIST, artist);
203 if (album[0] != 0)
204 mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
205 MPD_TAG_ALBUM, album);
206 mpd_search_commit(connection);
208 browser.filelist = filelist_new();
209 /* add a dummy entry for ".." */
210 filelist_append(browser.filelist, NULL);
212 filelist_recv(browser.filelist, connection);
214 if (!mpd_response_finish(connection))
215 mpdclient_handle_error(c);
217 #ifndef NCMPC_MINI
218 /* fix highlights */
219 screen_browser_sync_highlights(browser.filelist, &c->playlist);
220 #endif
221 list_window_set_length(browser.lw, filelist_length(browser.filelist));
222 }
224 static void
225 free_state(struct mpdclient *c)
226 {
227 g_free(artist);
228 g_free(album);
229 artist = NULL;
230 album = NULL;
232 free_lists(c);
233 }
235 static void
236 open_artist_list(struct mpdclient *c)
237 {
238 free_state(c);
240 mode = LIST_ARTISTS;
241 load_artist_list(c);
242 }
244 static void
245 open_album_list(struct mpdclient *c, char *_artist)
246 {
247 assert(_artist != NULL);
249 free_state(c);
251 mode = LIST_ALBUMS;
252 artist = _artist;
253 load_album_list(c);
254 }
256 static void
257 open_song_list(struct mpdclient *c, char *_artist, char *_album)
258 {
259 assert(_artist != NULL);
260 assert(_album != NULL);
262 free_state(c);
264 mode = LIST_SONGS;
265 artist = _artist;
266 album = _album;
267 load_song_list(c);
268 }
270 static void
271 reload_lists(struct mpdclient *c)
272 {
273 free_lists(c);
275 switch (mode) {
276 case LIST_ARTISTS:
277 load_artist_list(c);
278 break;
280 case LIST_ALBUMS:
281 load_album_list(c);
282 break;
284 case LIST_SONGS:
285 load_song_list(c);
286 break;
287 }
288 }
290 static void
291 screen_artist_init(WINDOW *w, int cols, int rows)
292 {
293 browser.lw = list_window_init(w, cols, rows);
294 artist = NULL;
295 album = NULL;
296 }
298 static void
299 screen_artist_quit(void)
300 {
301 free_state(NULL);
302 list_window_free(browser.lw);
303 }
305 static void
306 screen_artist_open(struct mpdclient *c)
307 {
308 if (artist_list == NULL && album_list == NULL &&
309 browser.filelist == NULL)
310 reload_lists(c);
311 }
313 static void
314 screen_artist_resize(int cols, int rows)
315 {
316 browser.lw->cols = cols;
317 browser.lw->rows = rows;
318 }
320 static void
321 screen_artist_paint(void)
322 {
323 if (browser.filelist) {
324 list_window_paint(browser.lw, browser_lw_callback,
325 browser.filelist);
326 } else if (album_list != NULL)
327 list_window_paint(browser.lw, screen_artist_lw_callback,
328 album_list);
329 else if (artist_list != NULL)
330 list_window_paint(browser.lw, screen_artist_lw_callback,
331 artist_list);
332 else {
333 wmove(browser.lw->w, 0, 0);
334 wclrtobot(browser.lw->w);
335 }
336 }
338 static const char *
339 screen_artist_get_title(char *str, size_t size)
340 {
341 char *s1, *s2;
343 switch(mode) {
344 case LIST_ARTISTS:
345 g_snprintf(str, size, _("All artists"));
346 break;
348 case LIST_ALBUMS:
349 s1 = utf8_to_locale(artist);
350 g_snprintf(str, size, _("Albums of artist: %s"), s1);
351 g_free(s1);
352 break;
354 case LIST_SONGS:
355 s1 = utf8_to_locale(artist);
356 if (*album != 0) {
357 s2 = utf8_to_locale(album);
358 g_snprintf(str, size,
359 _("Album: %s - %s"), s1, s2);
360 g_free(s2);
361 } else
362 g_snprintf(str, size,
363 _("All tracks of artist: %s"), s1);
364 g_free(s1);
365 break;
366 }
368 return str;
369 }
371 static void
372 screen_artist_update(struct mpdclient *c)
373 {
374 if (browser.filelist == NULL)
375 return;
377 if (c->events & MPD_IDLE_DATABASE)
378 /* the db has changed -> update the list */
379 reload_lists(c);
381 #ifndef NCMPC_MINI
382 if (c->events & (MPD_IDLE_DATABASE | MPD_IDLE_PLAYLIST))
383 screen_browser_sync_highlights(browser.filelist, &c->playlist);
384 #endif
386 if (c->events & (MPD_IDLE_DATABASE
387 #ifndef NCMPC_MINI
388 | MPD_IDLE_PLAYLIST
389 #endif
390 ))
391 artist_repaint();
392 }
394 static void
395 add_query(struct mpdclient *c, enum mpd_tag_type table, char *_filter)
396 {
397 struct mpd_connection *connection = mpdclient_get_connection(c);
398 char *str;
399 struct filelist *addlist;
401 assert(filter != NULL);
403 str = utf8_to_locale(_filter);
404 if (table == MPD_TAG_ALBUM)
405 screen_status_printf("Adding album %s...", str);
406 else
407 screen_status_printf("Adding %s...", str);
408 g_free(str);
410 mpd_search_db_songs(connection, true);
411 mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
412 table, _filter);
413 mpd_search_commit(connection);
415 addlist = filelist_new_recv(connection);
417 if (mpd_response_finish(connection))
418 mpdclient_filelist_add_all(c, addlist);
419 else
420 mpdclient_handle_error(c);
422 filelist_free(addlist);
423 }
425 static int
426 screen_artist_lw_cmd(struct mpdclient *c, command_t cmd)
427 {
428 switch (mode) {
429 case LIST_ARTISTS:
430 case LIST_ALBUMS:
431 return list_window_cmd(browser.lw, cmd);
433 case LIST_SONGS:
434 return browser_cmd(&browser, c, cmd);
435 }
437 assert(0);
438 return 0;
439 }
441 static int
442 string_array_find(GPtrArray *array, const char *value)
443 {
444 guint i;
446 for (i = 0; i < array->len; ++i)
447 if (strcmp((const char*)g_ptr_array_index(array, i),
448 value) == 0)
449 return i;
451 return -1;
452 }
454 static bool
455 screen_artist_cmd(struct mpdclient *c, command_t cmd)
456 {
457 char *selected;
458 char *old;
459 int idx;
461 switch(cmd) {
462 case CMD_PLAY:
463 switch (mode) {
464 case LIST_ARTISTS:
465 if (browser.lw->selected >= artist_list->len)
466 return true;
468 selected = g_ptr_array_index(artist_list,
469 browser.lw->selected);
470 open_album_list(c, g_strdup(selected));
471 list_window_reset(browser.lw);
473 artist_repaint();
474 return true;
476 case LIST_ALBUMS:
477 if (browser.lw->selected == 0) {
478 /* handle ".." */
479 old = g_strdup(artist);
481 open_artist_list(c);
482 list_window_reset(browser.lw);
483 /* restore previous list window state */
484 idx = string_array_find(artist_list, old);
485 g_free(old);
487 if (idx >= 0) {
488 list_window_set_cursor(browser.lw, idx);
489 list_window_center(browser.lw, idx);
490 }
491 } else if (browser.lw->selected == album_list->len + 1) {
492 /* handle "show all" */
493 open_song_list(c, g_strdup(artist), g_strdup("\0"));
494 list_window_reset(browser.lw);
495 } else {
496 /* select album */
497 selected = g_ptr_array_index(album_list,
498 browser.lw->selected - 1);
499 open_song_list(c, g_strdup(artist), g_strdup(selected));
500 list_window_reset(browser.lw);
501 }
503 artist_repaint();
504 return true;
506 case LIST_SONGS:
507 if (browser.lw->selected == 0) {
508 /* handle ".." */
509 old = g_strdup(album);
511 open_album_list(c, g_strdup(artist));
512 list_window_reset(browser.lw);
513 /* restore previous list window state */
514 idx = *old == 0
515 ? (int)album_list->len
516 : string_array_find(album_list, old);
517 g_free(old);
519 if (idx >= 0) {
520 ++idx;
521 list_window_set_cursor(browser.lw, idx);
522 list_window_center(browser.lw, idx);
523 }
525 artist_repaint();
526 return true;
527 }
528 break;
529 }
530 break;
532 /* FIXME? CMD_GO_* handling duplicates code from CMD_PLAY */
534 case CMD_GO_PARENT_DIRECTORY:
535 switch (mode) {
536 case LIST_ARTISTS:
537 break;
539 case LIST_ALBUMS:
540 old = g_strdup(artist);
542 open_artist_list(c);
543 list_window_reset(browser.lw);
544 /* restore previous list window state */
545 idx = string_array_find(artist_list, old);
546 g_free(old);
548 if (idx >= 0) {
549 list_window_set_cursor(browser.lw, idx);
550 list_window_center(browser.lw, idx);
551 }
552 break;
554 case LIST_SONGS:
555 old = g_strdup(album);
557 open_album_list(c, g_strdup(artist));
558 list_window_reset(browser.lw);
559 /* restore previous list window state */
560 idx = *old == 0
561 ? (int)album_list->len
562 : string_array_find(album_list, old);
563 g_free(old);
565 if (idx >= 0) {
566 ++idx;
567 list_window_set_cursor(browser.lw, idx);
568 list_window_center(browser.lw, idx);
569 }
570 break;
571 }
573 artist_repaint();
574 break;
576 case CMD_GO_ROOT_DIRECTORY:
577 switch (mode) {
578 case LIST_ARTISTS:
579 break;
581 case LIST_ALBUMS:
582 case LIST_SONGS:
583 open_artist_list(c);
584 list_window_reset(browser.lw);
585 /* restore first list window state (pop while returning true) */
586 /* XXX */
587 break;
588 }
590 artist_repaint();
591 break;
593 case CMD_SELECT:
594 case CMD_ADD:
595 switch(mode) {
596 case LIST_ARTISTS:
597 if (browser.lw->selected >= artist_list->len)
598 return true;
600 for (unsigned i = browser.lw->selected_start;
601 i <= browser.lw->selected_end; ++i) {
602 selected = g_ptr_array_index(artist_list, i);
603 add_query(c, MPD_TAG_ARTIST, selected);
604 cmd = CMD_LIST_NEXT; /* continue and select next item... */
605 }
606 break;
608 case LIST_ALBUMS:
609 for (unsigned i = browser.lw->selected_start;
610 i <= browser.lw->selected_end; ++i) {
611 if(i == album_list->len + 1)
612 add_query(c, MPD_TAG_ARTIST, artist);
613 else if (i > 0)
614 {
615 selected = g_ptr_array_index(album_list,
616 browser.lw->selected - 1);
617 add_query(c, MPD_TAG_ALBUM, selected);
618 cmd = CMD_LIST_NEXT; /* continue and select next item... */
619 }
620 }
621 break;
623 case LIST_SONGS:
624 /* handled by browser_cmd() */
625 break;
626 }
627 break;
629 /* continue and update... */
630 case CMD_SCREEN_UPDATE:
631 reload_lists(c);
632 return false;
634 case CMD_LIST_FIND:
635 case CMD_LIST_RFIND:
636 case CMD_LIST_FIND_NEXT:
637 case CMD_LIST_RFIND_NEXT:
638 switch (mode) {
639 case LIST_ARTISTS:
640 screen_find(browser.lw, cmd,
641 screen_artist_lw_callback, artist_list);
642 artist_repaint();
643 return true;
645 case LIST_ALBUMS:
646 screen_find(browser.lw, cmd,
647 screen_artist_lw_callback, album_list);
648 artist_repaint();
649 return true;
651 case LIST_SONGS:
652 /* handled by browser_cmd() */
653 break;
654 }
655 case CMD_LIST_JUMP:
656 switch (mode) {
657 case LIST_ARTISTS:
658 screen_jump(browser.lw, screen_artist_lw_callback,
659 artist_list);
660 artist_repaint();
661 return true;
663 case LIST_ALBUMS:
664 screen_jump(browser.lw, screen_artist_lw_callback,
665 album_list);
666 artist_repaint();
667 return true;
669 case LIST_SONGS:
670 /* handled by browser_cmd() */
671 break;
672 }
674 break;
676 default:
677 break;
678 }
680 if (screen_artist_lw_cmd(c, cmd)) {
681 if (screen_is_visible(&screen_artist))
682 artist_repaint();
683 return true;
684 }
686 return false;
687 }
689 const struct screen_functions screen_artist = {
690 .init = screen_artist_init,
691 .exit = screen_artist_quit,
692 .open = screen_artist_open,
693 .resize = screen_artist_resize,
694 .paint = screen_artist_paint,
695 .update = screen_artist_update,
696 .cmd = screen_artist_cmd,
697 .get_title = screen_artist_get_title,
698 };