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 */
19 #include "i18n.h"
20 #include "options.h"
21 #include "charset.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"
31 #include <ctype.h>
32 #include <stdlib.h>
33 #include <string.h>
34 #include <glib.h>
35 #include <ncurses.h>
37 #define BUFSIZE 1024
39 typedef enum { LIST_ARTISTS, LIST_ALBUMS, LIST_SONGS } artist_mode_t;
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;
46 static struct screen_browser browser;
48 static gint
49 compare_utf8(gconstpointer s1, gconstpointer s2)
50 {
51 char *key1, *key2;
52 int n;
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 }
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;
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 if (idx >= list->len)
84 return NULL;
86 str_utf8 = g_ptr_array_index(list, idx);
87 assert(str_utf8 != NULL);
89 str = utf8_to_locale(str_utf8);
90 g_snprintf(buf, BUFSIZE, "[%s]", str);
91 g_free(str);
93 return buf;
94 }
96 static void
97 paint(mpdclient_t *c);
99 static void
100 artist_repaint(void)
101 {
102 paint(NULL);
103 wrefresh(browser.lw->w);
104 }
106 static void
107 artist_repaint_if_active(void)
108 {
109 if (screen_is_visible(&screen_artist))
110 artist_repaint();
111 }
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);
119 artist_repaint_if_active();
120 }
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;
128 while (in != NULL) {
129 g_ptr_array_add(out, in->data);
130 in = g_list_next(in);
131 }
133 g_list_free(head);
134 return out;
135 }
137 static void
138 string_array_free(GPtrArray *array)
139 {
140 unsigned i;
142 for (i = 0; i < array->len; ++i) {
143 char *value = g_ptr_array_index(array, i);
144 free(value);
145 }
147 g_ptr_array_free(array, TRUE);
148 }
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 }
158 if (album_list != NULL) {
159 string_array_free(album_list);
160 album_list = NULL;
161 }
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 }
171 static void
172 load_artist_list(struct mpdclient *c)
173 {
174 GList *list;
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);
183 list = mpdclient_get_artists_utf8(c);
184 /* sort list */
185 list = g_list_sort(list, compare_utf8);
187 artist_list = g_list_to_ptr_array(list);
188 }
190 static void
191 load_album_list(struct mpdclient *c)
192 {
193 GList *list;
195 assert(mode == LIST_ALBUMS);
196 assert(artist != NULL);
197 assert(album == NULL);
198 assert(album_list == NULL);
199 assert(browser.filelist == NULL);
201 list = mpdclient_get_albums_utf8(c, artist);
202 /* sort list */
203 list = g_list_sort(list, compare_utf8);
205 album_list = g_list_to_ptr_array(list);
206 }
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);
216 if (album[0] == 0)
217 browser.filelist =
218 mpdclient_filelist_search_utf8(c, TRUE,
219 MPD_TABLE_ARTIST,
220 artist);
221 else
222 browser.filelist =
223 mpdclient_filelist_search_utf8(c, TRUE,
224 MPD_TABLE_ALBUM,
225 album);
226 if (browser.filelist == NULL)
227 browser.filelist = filelist_new(NULL);
229 /* add a dummy entry for ".." */
230 filelist_prepend(browser.filelist, NULL);
232 /* install playlist callback and fix highlights */
233 sync_highlights(c, browser.filelist);
234 mpdclient_install_playlist_callback(c, playlist_changed_callback);
235 }
237 static void
238 free_state(struct mpdclient *c)
239 {
240 g_free(artist);
241 g_free(album);
242 artist = NULL;
243 album = NULL;
245 free_lists(c);
246 }
248 static void
249 open_artist_list(struct mpdclient *c)
250 {
251 free_state(c);
253 mode = LIST_ARTISTS;
254 load_artist_list(c);
255 }
257 static void
258 open_album_list(struct mpdclient *c, char *_artist)
259 {
260 assert(_artist != NULL);
262 free_state(c);
264 mode = LIST_ALBUMS;
265 artist = _artist;
266 load_album_list(c);
267 }
269 static void
270 open_song_list(struct mpdclient *c, char *_artist, char *_album)
271 {
272 assert(_artist != NULL);
273 assert(_album != NULL);
275 free_state(c);
277 mode = LIST_SONGS;
278 artist = _artist;
279 album = _album;
280 load_song_list(c);
281 }
283 static void
284 reload_lists(struct mpdclient *c)
285 {
286 free_lists(c);
288 switch (mode) {
289 case LIST_ARTISTS:
290 load_artist_list(c);
291 break;
293 case LIST_ALBUMS:
294 load_album_list(c);
295 break;
297 case LIST_SONGS:
298 load_song_list(c);
299 break;
300 }
301 }
303 /* db updated */
304 static void
305 browse_callback(mpdclient_t *c, int event, mpd_unused gpointer data)
306 {
307 switch(event) {
308 case BROWSE_DB_UPDATED:
309 reload_lists(c);
310 break;
311 default:
312 break;
313 }
315 artist_repaint_if_active();
316 }
318 static void
319 init(WINDOW *w, int cols, int rows)
320 {
321 browser.lw = list_window_init(w, cols, rows);
322 artist = NULL;
323 album = NULL;
324 }
326 static void
327 quit(void)
328 {
329 free_state(NULL);
330 list_window_free(browser.lw);
331 }
333 static void
334 open(mpd_unused screen_t *screen, mpdclient_t *c)
335 {
336 static gboolean callback_installed = FALSE;
338 if (artist_list == NULL && album_list == NULL &&
339 browser.filelist == NULL)
340 reload_lists(c);
341 if (!callback_installed) {
342 mpdclient_install_browse_callback(c, browse_callback);
343 callback_installed = TRUE;
344 }
345 }
347 static void
348 resize(int cols, int rows)
349 {
350 browser.lw->cols = cols;
351 browser.lw->rows = rows;
352 }
354 static void
355 paint(mpd_unused mpdclient_t *c)
356 {
357 if (browser.filelist) {
358 list_window_paint(browser.lw, browser_lw_callback,
359 browser.filelist);
360 } else if (album_list != NULL)
361 list_window_paint(browser.lw, artist_lw_callback, album_list);
362 else if (artist_list != NULL)
363 list_window_paint(browser.lw, artist_lw_callback, artist_list);
364 else {
365 wmove(browser.lw->w, 0, 0);
366 wclrtobot(browser.lw->w);
367 }
368 }
370 static const char *
371 get_title(char *str, size_t size)
372 {
373 char *s1, *s2;
375 switch(mode) {
376 case LIST_ARTISTS:
377 g_snprintf(str, size, _("Artist: [db browser - EXPERIMENTAL]"));
378 break;
379 case LIST_ALBUMS:
380 s1 = utf8_to_locale(artist);
381 g_snprintf(str, size, _("Artist: %s"), s1);
382 g_free(s1);
383 break;
384 case LIST_SONGS:
385 s1 = utf8_to_locale(artist);
386 s2 = *album == 0
387 ? g_strdup(_("All tracks"))
388 : utf8_to_locale(album);
389 g_snprintf(str, size, _("Artist: %s - %s"), s1, s2);
390 g_free(s1);
391 g_free(s2);
392 break;
393 }
394 return str;
395 }
397 static void
398 add_query(mpdclient_t *c, int table, char *_filter)
399 {
400 char *str;
401 mpdclient_filelist_t *addlist;
403 assert(filter != NULL);
405 str = utf8_to_locale(_filter);
406 if (table== MPD_TABLE_ALBUM)
407 screen_status_printf("Adding album %s...", str);
408 else
409 screen_status_printf("Adding %s...", str);
410 g_free(str);
412 addlist = mpdclient_filelist_search_utf8(c, TRUE, table, _filter);
413 if (addlist) {
414 mpdclient_filelist_add_all(c, addlist);
415 filelist_free(addlist);
416 }
417 }
419 static unsigned
420 metalist_length(void)
421 {
422 assert(mode != LIST_ARTISTS || artist_list != NULL);
423 assert(mode != LIST_ALBUMS || album_list != NULL);
425 return mode == LIST_ALBUMS
426 ? album_list->len + 2
427 : artist_list->len;
428 }
430 static int
431 artist_lw_cmd(struct screen *screen, struct mpdclient *c, command_t cmd)
432 {
433 switch (mode) {
434 case LIST_ARTISTS:
435 case LIST_ALBUMS:
436 return list_window_cmd(browser.lw, metalist_length(), cmd);
438 case LIST_SONGS:
439 return browser_cmd(&browser, screen, c, cmd);
440 }
442 assert(0);
443 return 0;
444 }
446 static int
447 string_array_find(GPtrArray *array, const char *value)
448 {
449 guint i;
451 for (i = 0; i < array->len; ++i)
452 if (strcmp((const char*)g_ptr_array_index(array, i),
453 value) == 0)
454 return i;
456 return -1;
457 }
459 static int
460 artist_cmd(screen_t *screen, mpdclient_t *c, command_t cmd)
461 {
462 char *selected;
463 char *old;
464 int idx;
466 switch(cmd) {
467 case CMD_PLAY:
468 switch (mode) {
469 case LIST_ARTISTS:
470 selected = g_ptr_array_index(artist_list,
471 browser.lw->selected);
472 open_album_list(c, g_strdup(selected));
473 list_window_reset(browser.lw);
475 artist_repaint();
476 return true;
478 case LIST_ALBUMS:
479 if (browser.lw->selected == 0) {
480 /* handle ".." */
481 old = g_strdup(artist);
483 open_artist_list(c);
484 list_window_reset(browser.lw);
485 /* restore previous list window state */
486 idx = string_array_find(artist_list, old);
487 g_free(old);
489 if (idx >= 0) {
490 list_window_set_selected(browser.lw, idx);
491 list_window_center(browser.lw,
492 artist_list->len, idx);
493 }
494 } else if (browser.lw->selected == album_list->len + 1) {
495 /* handle "show all" */
496 open_song_list(c, g_strdup(artist), g_strdup("\0"));
497 list_window_reset(browser.lw);
498 } else {
499 /* select album */
500 selected = g_ptr_array_index(album_list,
501 browser.lw->selected - 1);
502 open_song_list(c, g_strdup(artist), g_strdup(selected));
503 list_window_reset(browser.lw);
504 }
506 artist_repaint();
507 return true;
509 case LIST_SONGS:
510 if (browser.lw->selected == 0) {
511 /* handle ".." */
512 old = g_strdup(album);
514 open_album_list(c, g_strdup(artist));
515 list_window_reset(browser.lw);
516 /* restore previous list window state */
517 idx = *old == 0
518 ? (int)album_list->len
519 : string_array_find(album_list, old);
520 g_free(old);
522 if (idx >= 0) {
523 ++idx;
524 list_window_set_selected(browser.lw, idx);
525 list_window_center(browser.lw,
526 album_list->len, idx);
527 }
529 artist_repaint();
530 return true;
531 }
532 break;
533 }
534 break;
536 /* FIXME? CMD_GO_* handling duplicates code from CMD_PLAY */
538 case CMD_GO_PARENT_DIRECTORY:
539 switch (mode) {
540 case LIST_ARTISTS:
541 break;
543 case LIST_ALBUMS:
544 old = g_strdup(artist);
546 open_artist_list(c);
547 list_window_reset(browser.lw);
548 /* restore previous list window state */
549 idx = string_array_find(artist_list, old);
550 g_free(old);
552 if (idx >= 0) {
553 list_window_set_selected(browser.lw, idx);
554 list_window_center(browser.lw,
555 artist_list->len, idx);
556 }
557 break;
559 case LIST_SONGS:
560 old = g_strdup(album);
562 open_album_list(c, g_strdup(artist));
563 list_window_reset(browser.lw);
564 /* restore previous list window state */
565 idx = *old == 0
566 ? (int)album_list->len
567 : string_array_find(album_list, old);
568 g_free(old);
570 if (idx >= 0) {
571 ++idx;
572 list_window_set_selected(browser.lw, idx);
573 list_window_center(browser.lw,
574 album_list->len, idx);
575 }
576 break;
577 }
579 artist_repaint();
580 break;
582 case CMD_GO_ROOT_DIRECTORY:
583 switch (mode) {
584 case LIST_ARTISTS:
585 break;
587 case LIST_ALBUMS:
588 case LIST_SONGS:
589 open_artist_list(c);
590 list_window_reset(browser.lw);
591 /* restore first list window state (pop while returning true) */
592 /* XXX */
593 break;
594 }
596 artist_repaint();
597 break;
599 case CMD_SELECT:
600 case CMD_ADD:
601 switch(mode) {
602 case LIST_ARTISTS:
603 selected = g_ptr_array_index(artist_list,
604 browser.lw->selected);
605 add_query(c, MPD_TABLE_ARTIST, selected);
606 cmd = CMD_LIST_NEXT; /* continue and select next item... */
607 break;
609 case LIST_ALBUMS:
610 if (browser.lw->selected == album_list->len + 1)
611 add_query(c, MPD_TABLE_ARTIST, artist);
612 else if (browser.lw->selected > 0) {
613 selected = g_ptr_array_index(album_list,
614 browser.lw->selected - 1);
615 add_query(c, MPD_TABLE_ALBUM, selected);
616 cmd = CMD_LIST_NEXT; /* continue and select next item... */
617 }
618 break;
620 case LIST_SONGS:
621 /* handled by browser_cmd() */
622 break;
623 }
624 break;
626 /* continue and update... */
627 case CMD_SCREEN_UPDATE:
628 reload_lists(c);
629 screen_status_printf(_("Screen updated!"));
630 return 0;
632 case CMD_LIST_FIND:
633 case CMD_LIST_RFIND:
634 case CMD_LIST_FIND_NEXT:
635 case CMD_LIST_RFIND_NEXT:
636 switch (mode) {
637 case LIST_ARTISTS:
638 screen_find(screen,
639 browser.lw, artist_list->len,
640 cmd, artist_lw_callback, artist_list);
641 artist_repaint();
642 return 1;
644 case LIST_ALBUMS:
645 screen_find(screen,
646 browser.lw, album_list->len + 2,
647 cmd, artist_lw_callback, album_list);
648 artist_repaint();
649 return 1;
651 case LIST_SONGS:
652 /* handled by browser_cmd() */
653 break;
654 }
656 break;
658 default:
659 break;
660 }
662 if (artist_lw_cmd(screen, c, cmd)) {
663 artist_repaint();
664 return 1;
665 }
667 return 0;
668 }
670 const struct screen_functions screen_artist = {
671 .init = init,
672 .exit = quit,
673 .open = open,
674 .resize = resize,
675 .paint = paint,
676 .cmd = artist_cmd,
677 .get_title = get_title,
678 };