src/screen_lyrics.c

   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  */
  19
  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"
  32
  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>
  40
  41 static list_window_t *lw = NULL;
  42
  43 static struct {
  44         const struct mpd_song *song;
  45
  46         char *artist, *title;
  47
  48         struct lyrics_loader *loader;
  49
  50         GPtrArray *lines;
  51 } current;
  52
  53 static void
  54 screen_lyrics_abort(void)
  55 {
  56         if (current.loader != NULL) {
  57                 lyrics_free(current.loader);
  58                 current.loader = NULL;
  59         }
  60
  61         if (current.artist != NULL) {
  62                 g_free(current.artist);
  63                 current.artist = NULL;
  64         }
  65
  66         if (current.title != NULL) {
  67                 g_free(current.title);
  68                 current.artist = NULL;
  69         }
  70
  71         current.song = NULL;
  72 }
  73
  74 static void
  75 screen_lyrics_clear(void)
  76 {
  77         guint i;
  78
  79         assert(current.loader == NULL ||
  80                lyrics_result(current.loader) == LYRICS_SUCCESS);
  81
  82         for (i = 0; i < current.lines->len; ++i)
  83                 g_free(g_ptr_array_index(current.lines, i));
  84
  85         g_ptr_array_set_size(current.lines, 0);
  86 }
  87
  88 static void
  89 screen_lyrics_set(const GString *str)
  90 {
  91         const char *p, *eol, *next;
  92
  93         screen_lyrics_clear();
  94
  95         p = str->str;
  96         while ((eol = strchr(p, '\n')) != NULL) {
  97                 char *line;
  98
  99                 next = eol + 1;
 100
 101                 /* strip whitespace at end */
 102
 103                 while (eol > p && (unsigned char)eol[-1] <= 0x20)
 104                         --eol;
 105
 106                 /* create copy and append it to current.lines*/
 107
 108                 line = g_malloc(eol - p + 1);
 109                 memcpy(line, p, eol - p);
 110                 line[eol - p] = 0;
 111
 112                 g_ptr_array_add(current.lines, line);
 113
 114                 /* reset control characters */
 115
 116                 for (eol = line + (eol - p); line < eol; ++line)
 117                         if ((unsigned char)*line < 0x20)
 118                                 *line = ' ';
 119
 120                 p = next;
 121         }
 122
 123         if (*p != 0)
 124                 g_ptr_array_add(current.lines, g_strdup(p));
 125 }
 126
 127 static int
 128 screen_lyrics_poll(void)
 129 {
 130         assert(current.loader != NULL);
 131
 132         switch (lyrics_result(current.loader)) {
 133         case LYRICS_BUSY:
 134                 return 0;
 135
 136         case LYRICS_SUCCESS:
 137                 screen_lyrics_set(lyrics_get(current.loader));
 138                 lyrics_free(current.loader);
 139                 current.loader = NULL;
 140                 return 1;
 141
 142         case LYRICS_FAILED:
 143                 lyrics_free(current.loader);
 144                 current.loader = NULL;
 145                 screen_status_message (_("No lyrics"));
 146                 return -1;
 147         }
 148
 149         assert(0);
 150         return -1;
 151 }
 152
 153 static void
 154 screen_lyrics_load(struct mpd_song *song)
 155 {
 156         char buffer[MAX_SONGNAME_LENGTH];
 157
 158         assert(song != NULL);
 159
 160         screen_lyrics_abort();
 161         screen_lyrics_clear();
 162
 163         current.song = song;
 164
 165         strfsong(buffer, sizeof(buffer), "%artist%", song);
 166         current.artist = g_strdup(buffer);
 167
 168         strfsong(buffer, sizeof(buffer), "%title%", song);
 169         current.title = g_strdup(buffer);
 170
 171         current.loader = lyrics_load(current.artist, current.title);
 172 }
 173
 174 static void lyrics_paint(screen_t *screen, mpdclient_t *c);
 175
 176 static FILE *create_lyr_file(const char *artist, const char *title)
 177 {
 178         char path[1024];
 179
 180         snprintf(path, 1024, "%s/.lyrics",
 181                  getenv("HOME"));
 182         mkdir(path, S_IRWXU);
 183
 184         snprintf(path, 1024, "%s/.lyrics/%s - %s.txt",
 185                  getenv("HOME"), artist, title);
 186
 187         return fopen(path, "w");
 188 }
 189
 190 static int store_lyr_hd(void)
 191 {
 192         FILE *lyr_file;
 193         unsigned i;
 194
 195         lyr_file = create_lyr_file(current.artist, current.title);
 196         if (lyr_file == NULL)
 197                 return -1;
 198
 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));
 202
 203         fclose(lyr_file);
 204         return 0;
 205 }
 206
 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;
 212
 213         return g_ptr_array_index(current.lines, idx);
 214 }
 215
 216
 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 }
 224
 225 static void
 226 lyrics_resize(int cols, int rows)
 227 {
 228         lw->cols = cols;
 229         lw->rows = rows;
 230 }
 231
 232 static void
 233 lyrics_exit(void)
 234 {
 235         list_window_free(lw);
 236
 237         screen_lyrics_abort();
 238         screen_lyrics_clear();
 239
 240         g_ptr_array_free(current.lines, TRUE);
 241         current.lines = NULL;
 242 }
 243
 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 }
 252
 253
 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 }
 267
 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 }
 275
 276
 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 }
 286
 287
 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;
 293
 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         }
 323
 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         }
 332
 333         return 0;
 334 }
 335
 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 };
 347
 348 #endif /* ENABLE_LYRICS_SCREEN */