1 /*
2 * (c) 2006 by Kalle Wallin <kaw@linux.se>
3 * Copyright (C) 2008 Max Kellermann <max@duempel.org>
4 *
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.
9 *
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.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 *
18 */
20 #include "config.h"
21 #ifndef DISABLE_LYRICS_SCREEN
22 #include <sys/stat.h>
23 #include "ncmpc.h"
24 #include "options.h"
25 #include "mpdclient.h"
26 #include "command.h"
27 #include "screen.h"
28 #include "screen_utils.h"
29 #include "strfsong.h"
30 #include "lyrics.h"
31 #include "gcc.h"
33 #define _GNU_SOURCE
34 #include <stdlib.h>
35 #include <string.h>
36 #include <glib.h>
37 #include <ncurses.h>
38 #include <unistd.h>
39 #include <stdio.h>
41 static list_window_t *lw = NULL;
43 static struct {
44 const struct mpd_song *song;
46 char *artist, *title;
48 struct lyrics_loader *loader;
50 GPtrArray *lines;
51 } current;
53 static void
54 screen_lyrics_abort(void)
55 {
56 if (current.loader != NULL) {
57 lyrics_free(current.loader);
58 current.loader = NULL;
59 }
61 if (current.artist != NULL) {
62 g_free(current.artist);
63 current.artist = NULL;
64 }
66 if (current.title != NULL) {
67 g_free(current.title);
68 current.artist = NULL;
69 }
71 current.song = NULL;
72 }
74 static void
75 screen_lyrics_clear(void)
76 {
77 guint i;
79 assert(current.loader == NULL ||
80 lyrics_result(current.loader) == LYRICS_SUCCESS);
82 current.song = NULL;
84 for (i = 0; i < current.lines->len; ++i)
85 g_free(g_ptr_array_index(current.lines, i));
87 g_ptr_array_set_size(current.lines, 0);
88 }
90 static void
91 screen_lyrics_set(const GString *str)
92 {
93 const char *p, *eol, *next;
95 screen_lyrics_clear();
97 p = str->str;
98 while ((eol = strchr(p, '\n')) != NULL) {
99 char *line;
101 next = eol + 1;
103 /* strip whitespace at end */
105 while (eol > p && (unsigned char)eol[-1] <= 0x20)
106 --eol;
108 /* create copy and append it to current.lines*/
110 line = g_malloc(eol - p + 1);
111 memcpy(line, p, eol - p);
112 line[eol - p] = 0;
114 g_ptr_array_add(current.lines, line);
116 /* reset control characters */
118 for (eol = line + (eol - p); line < eol; ++line)
119 if ((unsigned char)*line < 0x20)
120 *line = ' ';
122 p = next;
123 }
125 if (*p != 0)
126 g_ptr_array_add(current.lines, g_strdup(p));
127 }
129 static int
130 screen_lyrics_poll(void)
131 {
132 assert(current.loader != NULL);
134 switch (lyrics_result(current.loader)) {
135 case LYRICS_BUSY:
136 return 0;
138 case LYRICS_SUCCESS:
139 screen_lyrics_set(lyrics_get(current.loader));
140 lyrics_free(current.loader);
141 current.loader = NULL;
142 return 1;
144 case LYRICS_FAILED:
145 lyrics_free(current.loader);
146 current.loader = NULL;
147 screen_status_message (_("No lyrics"));
148 return -1;
149 }
151 assert(0);
152 return -1;
153 }
155 static void
156 screen_lyrics_load(struct mpd_song *song)
157 {
158 char buffer[MAX_SONGNAME_LENGTH];
160 assert(song != NULL);
162 screen_lyrics_abort();
163 screen_lyrics_clear();
165 strfsong(buffer, sizeof(buffer), "%artist%", song);
166 current.artist = g_strdup(buffer);
168 strfsong(buffer, sizeof(buffer), "%title%", song);
169 current.title = g_strdup(buffer);
171 current.loader = lyrics_load(current.artist, current.title);
172 }
174 static void lyrics_paint(screen_t *screen, mpdclient_t *c);
176 static FILE *create_lyr_file(const char *artist, const char *title)
177 {
178 char path[1024];
180 snprintf(path, 1024, "%s/.lyrics",
181 getenv("HOME"));
182 mkdir(path, S_IRWXU);
184 snprintf(path, 1024, "%s/.lyrics/%s - %s.txt",
185 getenv("HOME"), artist, title);
187 return fopen(path, "w");
188 }
190 static int store_lyr_hd(void)
191 {
192 FILE *lyr_file;
193 unsigned i;
195 lyr_file = create_lyr_file(current.artist, current.title);
196 if (lyr_file == NULL)
197 return -1;
199 for (i = 0; i < current.lines->len; ++i)
200 fprintf(lyr_file, "%s\n",
201 (const char*)g_ptr_array_index(current.lines, i));
203 fclose(lyr_file);
204 return 0;
205 }
207 static const char *
208 list_callback(unsigned idx, mpd_unused int *highlight, mpd_unused void *data)
209 {
210 if (idx >= current.lines->len)
211 return NULL;
213 return g_ptr_array_index(current.lines, idx);
214 }
217 static void
218 lyrics_screen_init(WINDOW *w, int cols, int rows)
219 {
220 current.lines = g_ptr_array_new();
221 lw = list_window_init(w, cols, rows);
222 lw->flags = LW_HIDE_CURSOR;
223 }
225 static void
226 lyrics_resize(int cols, int rows)
227 {
228 lw->cols = cols;
229 lw->rows = rows;
230 }
232 static void
233 lyrics_exit(void)
234 {
235 list_window_free(lw);
237 screen_lyrics_abort();
238 screen_lyrics_clear();
240 g_ptr_array_free(current.lines, TRUE);
241 current.lines = NULL;
242 }
244 static void
245 lyrics_open(mpd_unused screen_t *screen, mpdclient_t *c)
246 {
247 if (c->song != NULL && c->song != current.song)
248 screen_lyrics_load(c->song);
249 else if (current.loader != NULL)
250 screen_lyrics_poll();
251 }
254 static const char *
255 lyrics_title(char *str, size_t size)
256 {
257 if (current.loader != NULL)
258 return "Lyrics (loading)";
259 else if (current.artist != NULL && current.title != NULL &&
260 current.lines->len > 0) {
261 snprintf(str, size, "Lyrics: %s - %s",
262 current.artist, current.title);
263 return str;
264 } else
265 return "Lyrics";
266 }
268 static void
269 lyrics_paint(mpd_unused screen_t *screen, mpd_unused mpdclient_t *c)
270 {
271 lw->clear = 1;
272 list_window_paint(lw, list_callback, NULL);
273 wrefresh(lw->w);
274 }
277 static void
278 lyrics_update(mpd_unused screen_t *screen, mpd_unused mpdclient_t *c)
279 {
280 if( lw->repaint ) {
281 list_window_paint(lw, list_callback, NULL);
282 wrefresh(lw->w);
283 lw->repaint = 0;
284 }
285 }
288 static int
289 lyrics_cmd(screen_t *screen, mpdclient_t *c, command_t cmd)
290 {
291 if (list_window_scroll_cmd(lw, current.lines->len, cmd))
292 return 1;
294 switch(cmd) {
295 case CMD_SELECT:
296 /* XXX */
297 if (current.loader != NULL) {
298 int ret = screen_lyrics_poll();
299 if (ret != 0)
300 lyrics_paint(NULL, NULL);
301 }
302 return 1;
303 case CMD_INTERRUPT:
304 if (current.loader != NULL) {
305 screen_lyrics_abort();
306 screen_lyrics_clear();
307 }
308 return 1;
309 case CMD_ADD:
310 if (current.loader == NULL && current.artist != NULL &&
311 current.title != NULL && store_lyr_hd() == 0)
312 screen_status_message (_("Lyrics saved!"));
313 return 1;
314 case CMD_LYRICS_UPDATE:
315 if (c->song != NULL) {
316 screen_lyrics_load(c->song);
317 lyrics_paint(NULL, NULL);
318 }
319 return 1;
320 default:
321 break;
322 }
324 lw->selected = lw->start+lw->rows;
325 if (screen_find(screen,
326 lw, current.lines->len,
327 cmd, list_callback, NULL)) {
328 /* center the row */
329 list_window_center(lw, current.lines->len, lw->selected);
330 return 1;
331 }
333 return 0;
334 }
336 const struct screen_functions screen_lyrics = {
337 .init = lyrics_screen_init,
338 .exit = lyrics_exit,
339 .open = lyrics_open,
340 .close = NULL,
341 .resize = lyrics_resize,
342 .paint = lyrics_paint,
343 .update = lyrics_update,
344 .cmd = lyrics_cmd,
345 .get_title = lyrics_title,
346 };
348 #endif /* ENABLE_LYRICS_SCREEN */