Code

check the return value of mpdclient_get_connection()
[ncmpc.git] / src / screen_browser.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_browser.h"
21 #include "screen_file.h"
22 #include "screen_song.h"
23 #include "screen_lyrics.h"
24 #include "screen_message.h"
25 #include "screen_find.h"
26 #include "screen.h"
27 #include "i18n.h"
28 #include "options.h"
29 #include "charset.h"
30 #include "strfsong.h"
31 #include "mpdclient.h"
32 #include "filelist.h"
33 #include "colors.h"
34 #include "paint.h"
35 #include "song_paint.h"
37 #include <mpd/client.h>
39 #include <string.h>
41 #define BUFSIZE 1024
43 #ifndef NCMPC_MINI
44 #define HIGHLIGHT  (0x01)
45 #endif
47 static const char playlist_format[] = "*%s*";
49 #ifndef NCMPC_MINI
51 /* sync highlight flags with playlist */
52 void
53 screen_browser_sync_highlights(struct filelist *fl,
54                                const struct mpdclient_playlist *playlist)
55 {
56         guint i;
58         for (i = 0; i < filelist_length(fl); ++i) {
59                 struct filelist_entry *entry = filelist_get(fl, i);
60                 const struct mpd_entity *entity = entry->entity;
62                 if (entity != NULL && mpd_entity_get_type(entity) == MPD_ENTITY_TYPE_SONG) {
63                         const struct mpd_song *song =
64                                 mpd_entity_get_song(entity);
66                         if (playlist_get_index_from_same_song(playlist,
67                                                               song) >= 0)
68                                 entry->flags |= HIGHLIGHT;
69                         else
70                                 entry->flags &= ~HIGHLIGHT;
71                 }
72         }
73 }
75 #endif
77 /* list_window callback */
78 static const char *
79 browser_lw_callback(unsigned idx, void *data)
80 {
81         const struct filelist *fl = (const struct filelist *) data;
82         static char buf[BUFSIZE];
83         const struct filelist_entry *entry;
84         const struct mpd_entity *entity;
86         assert(fl != NULL);
87         assert(idx < filelist_length(fl));
89         entry = filelist_get(fl, idx);
90         assert(entry != NULL);
92         entity = entry->entity;
94         if( entity == NULL )
95                 return "..";
97         if (mpd_entity_get_type(entity) == MPD_ENTITY_TYPE_DIRECTORY) {
98                 const struct mpd_directory *dir =
99                         mpd_entity_get_directory(entity);
100                 char *directory = utf8_to_locale(g_basename(mpd_directory_get_path(dir)));
101                 g_strlcpy(buf, directory, sizeof(buf));
102                 return buf;
103         } else if (mpd_entity_get_type(entity) == MPD_ENTITY_TYPE_SONG) {
104                 const struct mpd_song *song = mpd_entity_get_song(entity);
106                 strfsong(buf, BUFSIZE, options.list_format, song);
107                 return buf;
108         } else if (mpd_entity_get_type(entity) == MPD_ENTITY_TYPE_PLAYLIST) {
109                 const struct mpd_playlist *playlist =
110                         mpd_entity_get_playlist(entity);
111                 char *filename = utf8_to_locale(g_basename(mpd_playlist_get_path(playlist)));
113                 g_snprintf(buf, BUFSIZE, playlist_format, filename);
114                 g_free(filename);
115                 return buf;
116         }
118         return "Error: Unknown entry!";
121 static bool
122 load_playlist(struct mpdclient *c, const struct mpd_playlist *playlist)
124         struct mpd_connection *connection = mpdclient_get_connection(c);
126         if (connection == NULL)
127                 return false;
129         if (mpd_run_load(connection, mpd_playlist_get_path(playlist))) {
130                 char *filename = utf8_to_locale(mpd_playlist_get_path(playlist));
131                 screen_status_printf(_("Loading playlist %s..."),
132                                      g_basename(filename));
133                 g_free(filename);
135                 c->events |= MPD_IDLE_QUEUE;
136         } else
137                 mpdclient_handle_error(c);
139         return true;
142 static bool
143 enqueue_and_play(struct mpdclient *c, struct filelist_entry *entry)
145         struct mpd_connection *connection = mpdclient_get_connection(c);
146         const struct mpd_song *song = mpd_entity_get_song(entry->entity);
147         int id;
149 #ifndef NCMPC_MINI
150         if (!(entry->flags & HIGHLIGHT))
151                 id = -1;
152         else
153 #endif
154                 id = playlist_get_id_from_same_song(&c->playlist, song);
156         if (id < 0) {
157                 char buf[BUFSIZE];
159                 id = mpd_run_add_id(connection, mpd_song_get_uri(song));
160                 if (id < 0) {
161                         mpdclient_handle_error(c);
162                         return false;
163                 }
165 #ifndef NCMPC_MINI
166                 entry->flags |= HIGHLIGHT;
167 #endif
168                 strfsong(buf, BUFSIZE, options.list_format, song);
169                 screen_status_printf(_("Adding \'%s\' to playlist"), buf);
170         }
172         if (!mpd_run_play_id(connection, id)) {
173                 mpdclient_handle_error(c);
174                 return false;
175         }
177         return true;
180 struct filelist_entry *
181 browser_get_selected_entry(const struct screen_browser *browser)
183         struct list_window_range range;
185         list_window_get_range(browser->lw, &range);
187         if (browser->filelist == NULL ||
188             range.end <= range.start ||
189             range.end > range.start + 1 ||
190             range.start >= filelist_length(browser->filelist))
191                 return NULL;
193         return filelist_get(browser->filelist, range.start);
196 static const struct mpd_entity *
197 browser_get_selected_entity(const struct screen_browser *browser)
199         const struct filelist_entry *entry = browser_get_selected_entry(browser);
201         return entry != NULL
202                 ? entry->entity
203                 : NULL;
206 static const struct mpd_song *
207 browser_get_selected_song(const struct screen_browser *browser)
209         const struct mpd_entity *entity = browser_get_selected_entity(browser);
211         return entity != NULL &&
212                 mpd_entity_get_type(entity) == MPD_ENTITY_TYPE_SONG
213                 ? mpd_entity_get_song(entity)
214                 : NULL;
217 static struct filelist_entry *
218 browser_get_index(const struct screen_browser *browser, unsigned i)
220         if (browser->filelist == NULL ||
221             i >= filelist_length(browser->filelist))
222                 return NULL;
224         return filelist_get(browser->filelist, i);
227 static bool
228 browser_handle_enter(struct screen_browser *browser, struct mpdclient *c)
230         struct filelist_entry *entry = browser_get_selected_entry(browser);
231         struct mpd_entity *entity;
233         if (entry == NULL)
234                 return false;
236         entity = entry->entity;
237         if (entity == NULL)
238                 return false;
240         if (mpd_entity_get_type(entity) == MPD_ENTITY_TYPE_PLAYLIST)
241                 return load_playlist(c, mpd_entity_get_playlist(entity));
242         else if (mpd_entity_get_type(entity) == MPD_ENTITY_TYPE_SONG)
243                 return enqueue_and_play(c, entry);
244         return false;
247 static bool
248 browser_select_entry(struct mpdclient *c, struct filelist_entry *entry,
249                      G_GNUC_UNUSED gboolean toggle)
251         assert(entry != NULL);
252         assert(entry->entity != NULL);
254         if (mpd_entity_get_type(entry->entity) == MPD_ENTITY_TYPE_PLAYLIST)
255                 return load_playlist(c, mpd_entity_get_playlist(entry->entity));
257         if (mpd_entity_get_type(entry->entity) == MPD_ENTITY_TYPE_DIRECTORY) {
258                 const struct mpd_directory *dir =
259                         mpd_entity_get_directory(entry->entity);
261                 if (mpdclient_cmd_add_path(c, mpd_directory_get_path(dir))) {
262                         char *tmp = utf8_to_locale(mpd_directory_get_path(dir));
264                         screen_status_printf(_("Adding \'%s\' to playlist"), tmp);
265                         g_free(tmp);
266                 }
268                 return true;
269         }
271         if (mpd_entity_get_type(entry->entity) != MPD_ENTITY_TYPE_SONG)
272                 return false;
274 #ifndef NCMPC_MINI
275         if (!toggle || (entry->flags & HIGHLIGHT) == 0)
276 #endif
277         {
278                 const struct mpd_song *song =
279                         mpd_entity_get_song(entry->entity);
281 #ifndef NCMPC_MINI
282                 entry->flags |= HIGHLIGHT;
283 #endif
285                 if (mpdclient_cmd_add(c, song)) {
286                         char buf[BUFSIZE];
288                         strfsong(buf, BUFSIZE, options.list_format, song);
289                         screen_status_printf(_("Adding \'%s\' to playlist"), buf);
290                 }
291 #ifndef NCMPC_MINI
292         } else {
293                 /* remove song from playlist */
294                 const struct mpd_song *song =
295                         mpd_entity_get_song(entry->entity);
296                 int idx;
298                 entry->flags &= ~HIGHLIGHT;
300                 while ((idx = playlist_get_index_from_same_song(&c->playlist,
301                                                                 song)) >= 0)
302                         mpdclient_cmd_delete(c, idx);
303 #endif
304         }
306         return true;
309 static bool
310 browser_handle_select(struct screen_browser *browser, struct mpdclient *c)
312         struct list_window_range range;
313         struct filelist_entry *entry;
314         bool success;
316         list_window_get_range(browser->lw, &range);
317         for (unsigned i = range.start; i < range.end; ++i) {
318                 entry = browser_get_index(browser, i);
320                 if (entry != NULL && entry->entity != NULL)
321                         success = browser_select_entry(c, entry, TRUE);
322         }
324         return range.end == range.start + 1 && success;
327 static bool
328 browser_handle_add(struct screen_browser *browser, struct mpdclient *c)
330         struct list_window_range range;
331         struct filelist_entry *entry;
332         bool success;
334         list_window_get_range(browser->lw, &range);
335         for (unsigned i = range.start; i < range.end; ++i) {
336                 entry = browser_get_index(browser, i);
338                 if (entry != NULL && entry->entity != NULL)
339                         success = browser_select_entry(c, entry, FALSE);
340         }
342         return range.end == range.start + 1 && success;
345 static void
346 browser_handle_select_all(struct screen_browser *browser, struct mpdclient *c)
348         guint i;
350         if (browser->filelist == NULL)
351                 return;
353         for (i = 0; i < filelist_length(browser->filelist); ++i) {
354                 struct filelist_entry *entry = filelist_get(browser->filelist, i);
356                 if (entry != NULL && entry->entity != NULL)
357                         browser_select_entry(c, entry, FALSE);
358         }
361 #ifdef HAVE_GETMOUSE
362 static int
363 browser_handle_mouse_event(struct screen_browser *browser, struct mpdclient *c)
365         int row;
366         unsigned prev_selected = browser->lw->selected;
367         unsigned long bstate;
369         if (screen_get_mouse_event(c, &bstate, &row) ||
370             list_window_mouse(browser->lw, bstate, row))
371                 return 1;
373         list_window_set_cursor(browser->lw, browser->lw->start + row);
375         if( bstate & BUTTON1_CLICKED ) {
376                 if (prev_selected == browser->lw->selected)
377                         browser_handle_enter(browser, c);
378         } else if (bstate & BUTTON3_CLICKED) {
379                 if (prev_selected == browser->lw->selected)
380                         browser_handle_select(browser, c);
381         }
383         return 1;
385 #endif
387 static void
388 screen_browser_paint_callback(WINDOW *w, unsigned i, unsigned y,
389                               unsigned width, bool selected, void *data);
391 bool
392 browser_cmd(struct screen_browser *browser,
393             struct mpdclient *c, command_t cmd)
395         const struct mpd_song *song;
397         if (browser->filelist == NULL)
398                 return false;
400         if (list_window_cmd(browser->lw, cmd))
401                 return true;
403         switch (cmd) {
404         case CMD_LIST_FIND:
405         case CMD_LIST_RFIND:
406         case CMD_LIST_FIND_NEXT:
407         case CMD_LIST_RFIND_NEXT:
408                 screen_find(browser->lw, cmd, browser_lw_callback,
409                             browser->filelist);
410                 return true;
411         case CMD_LIST_JUMP:
412                 screen_jump(browser->lw, browser_lw_callback,
413                             screen_browser_paint_callback, browser->filelist);
414                 return true;
416 #ifdef HAVE_GETMOUSE
417         case CMD_MOUSE_EVENT:
418                 browser_handle_mouse_event(browser, c);
419                 return true;
420 #endif
422 #ifdef ENABLE_SONG_SCREEN
423         case CMD_SCREEN_SONG:
424                 song = browser_get_selected_song(browser);
425                 if (song == NULL)
426                         return false;
428                 screen_song_switch(c, song);
429                 return true;
430 #endif
432 #ifdef ENABLE_LYRICS_SCREEN
433         case CMD_SCREEN_LYRICS:
434                 song = browser_get_selected_song(browser);
435                 if (song == NULL)
436                         return false;
438                 screen_lyrics_switch(c, song, false);
439                 return true;
440 #endif
441         case CMD_SCREEN_SWAP:
442                 screen_swap(c, browser_get_selected_song(browser));
443                 return true;
445         default:
446                 break;
447         }
449         if (!mpdclient_is_connected(c))
450                 return false;
452         switch (cmd) {
453         case CMD_PLAY:
454                 browser_handle_enter(browser, c);
455                 return true;
457         case CMD_SELECT:
458                 if (browser_handle_select(browser, c))
459                         list_window_cmd(browser->lw, CMD_LIST_NEXT);
460                 return true;
462         case CMD_ADD:
463                 if (browser_handle_add(browser, c))
464                         list_window_cmd(browser->lw, CMD_LIST_NEXT);
465                 return true;
467         case CMD_SELECT_ALL:
468                 browser_handle_select_all(browser, c);
469                 return true;
471         case CMD_LOCATE:
472                 song = browser_get_selected_song(browser);
473                 if (song == NULL)
474                         return false;
476                 screen_file_goto_song(c, song);
477                 return true;
479         default:
480                 break;
481         }
483         return false;
486 static void
487 screen_browser_paint_directory(WINDOW *w, unsigned width,
488                                bool selected, const char *name)
490         row_color(w, COLOR_DIRECTORY, selected);
492         waddch(w, '[');
493         waddstr(w, name);
494         waddch(w, ']');
496         /* erase the unused space after the text */
497         row_clear_to_eol(w, width, selected);
500 static void
501 screen_browser_paint_playlist(WINDOW *w, unsigned width,
502                               bool selected, const char *name)
504         row_paint_text(w, width, COLOR_PLAYLIST, selected, name);
507 static void
508 screen_browser_paint_callback(WINDOW *w, unsigned i,
509                               unsigned y, unsigned width,
510                               bool selected, void *data)
512         const struct filelist *fl = (const struct filelist *) data;
513         const struct filelist_entry *entry;
514         const struct mpd_entity *entity;
515         bool highlight;
516         const struct mpd_directory *directory;
517         const struct mpd_playlist *playlist;
518         char *p;
520         assert(fl != NULL);
521         assert(i < filelist_length(fl));
523         entry = filelist_get(fl, i);
524         assert(entry != NULL);
526         entity = entry->entity;
527         if (entity == NULL) {
528                 screen_browser_paint_directory(w, width, selected, "..");
529                 return;
530         }
532 #ifndef NCMPC_MINI
533         highlight = (entry->flags & HIGHLIGHT) != 0;
534 #else
535         highlight = false;
536 #endif
538         if (highlight)
539                 colors_use(w, COLOR_LIST_BOLD);
540         else
541                 colors_use(w, COLOR_LIST);
543         switch (mpd_entity_get_type(entity)) {
544         case MPD_ENTITY_TYPE_DIRECTORY:
545                 directory = mpd_entity_get_directory(entity);
546                 p = utf8_to_locale(g_basename(mpd_directory_get_path(directory)));
547                 screen_browser_paint_directory(w, width, selected, p);
548                 g_free(p);
549                 break;
551         case MPD_ENTITY_TYPE_SONG:
552                 paint_song_row(w, y, width, selected, highlight,
553                                mpd_entity_get_song(entity));
554                 break;
556         case MPD_ENTITY_TYPE_PLAYLIST:
557                 playlist = mpd_entity_get_playlist(entity);
558                 p = utf8_to_locale(g_basename(mpd_playlist_get_path(playlist)));
559                 screen_browser_paint_playlist(w, width, selected, p);
560                 g_free(p);
561                 break;
563         default:
564                 waddstr(w, "<unknown>");
565         }
567         whline(w, ' ', width);
569         if (selected)
570                 wattroff(w, A_REVERSE);
573 void
574 screen_browser_paint(const struct screen_browser *browser)
576         list_window_paint2(browser->lw, screen_browser_paint_callback,
577                            browser->filelist);