apps/playlist_catalog.c

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  * $Id$
   9  *
  10  * Copyright (C) 2006 Sebastian Henriksen, Hardeep Sidhu
  11  *
  12  * This program is free software; you can redistribute it and/or
  13  * modify it under the terms of the GNU General Public License
  14  * as published by the Free Software Foundation; either version 2
  15  * of the License, or (at your option) any later version.
  16  *
  17  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  18  * KIND, either express or implied.
  19  *
  20  ****************************************************************************/
  21
  22 #include <stdbool.h>
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #include "action.h"
  27 #include "dir.h"
  28 #include "file.h"
  29 #include "filetree.h"
  30 #include "kernel.h"
  31 #include "keyboard.h"
  32 #include "lang.h"
  33 #include "list.h"
  34 #include "misc.h"
  35 #include "onplay.h"
  36 #include "playlist.h"
  37 #include "settings.h"
  38 #include "splash.h"
  39 #include "sprintf.h"
  40 #include "tree.h"
  41 #include "yesno.h"
  42 #include "filetypes.h"
  43 #include "debug.h"
  44 #include "playlist_catalog.h"
  45 #include "talk.h"
  46
  47 #define MAX_PLAYLISTS 400
  48
  49 /* Use for recursive directory search */
  50 struct add_track_context {
  51     int fd;
  52     int count;
  53 };
  54
  55 /* keep track of most recently used playlist */
  56 static char most_recent_playlist[MAX_PATH];
  57
  58 /* directory where our playlists our stored */
  59 static char playlist_dir[MAX_PATH];
  60 static int  playlist_dir_length;
  61 static bool playlist_dir_exists = false;
  62
  63 /* Retrieve playlist directory from config file and verify it exists */
  64 static int initialize_catalog(void)
  65 {
  66     static bool initialized = false;
  67
  68     if (!initialized)
  69     {
  70         bool default_dir = true;
  71
  72         /* directory config is of the format: "dir: /path/to/dir" */
  73         if (global_settings.playlist_catalog_dir[0])
  74         {
  75             strcpy(playlist_dir, global_settings.playlist_catalog_dir);
  76             default_dir = false;
  77         }
  78
  79         /* fall back to default directory if no or invalid config */
  80         if (default_dir)
  81             strlcpy(playlist_dir, PLAYLIST_CATALOG_DEFAULT_DIR,
  82                 sizeof(playlist_dir));
  83
  84         playlist_dir_length = strlen(playlist_dir);
  85
  86         if (dir_exists(playlist_dir))
  87         {
  88             playlist_dir_exists = true;
  89             memset(most_recent_playlist, 0, sizeof(most_recent_playlist));
  90             initialized = true;
  91         }
  92     }
  93
  94     if (!playlist_dir_exists)
  95     {
  96         if (mkdir(playlist_dir) < 0) {
  97             splashf(HZ*2, ID2P(LANG_CATALOG_NO_DIRECTORY), playlist_dir);
  98             return -1;
  99         }
 100         else {
 101             playlist_dir_exists = true;
 102             memset(most_recent_playlist, 0, sizeof(most_recent_playlist));
 103             initialized = true;
 104         }
 105     }
 106
 107     return 0;
 108 }
 109 /* Use the filetree functions to retrieve the list of playlists in the
 110    directory */
 111 static int create_playlist_list(char** playlists, int num_items,
 112                                 int* num_playlists)
 113 {
 114     int result = -1;
 115     int num_files = 0;
 116     int index = 0;
 117     int i;
 118     bool most_recent = false;
 119     struct entry *files;
 120     struct tree_context* tc = tree_get_context();
 121     int dirfilter = *(tc->dirfilter);
 122
 123     *num_playlists = 0;
 124
 125     /* use the tree browser dircache to load only playlists */
 126     *(tc->dirfilter) = SHOW_PLAYLIST;
 127
 128     if (ft_load(tc, playlist_dir) < 0)
 129     {
 130         splashf(HZ*2, ID2P(LANG_CATALOG_NO_DIRECTORY), playlist_dir);
 131         goto exit;
 132     }
 133
 134     files = (struct entry*) tc->dircache;
 135     num_files = tc->filesindir;
 136
 137     /* we've overwritten the dircache so tree browser will need to be
 138        reloaded */
 139     reload_directory();
 140
 141     /* if it exists, most recent playlist will always be index 0 */
 142     if (most_recent_playlist[0] != '\0')
 143     {
 144         index = 1;
 145         most_recent = true;
 146     }
 147
 148     for (i=0; i<num_files && index<num_items; i++)
 149     {
 150         if (files[i].attr & FILE_ATTR_M3U)
 151         {
 152             if (most_recent && !strncmp(files[i].name, most_recent_playlist,
 153                 sizeof(most_recent_playlist)))
 154             {
 155                 playlists[0] = files[i].name;
 156                 most_recent = false;
 157             }
 158             else
 159             {
 160                 playlists[index] = files[i].name;
 161                 index++;
 162             }
 163         }
 164     }
 165
 166     *num_playlists = index;
 167
 168     /* we couldn't find the most recent playlist, shift all playlists up */
 169     if (most_recent)
 170     {
 171         for (i=0; i<index-1; i++)
 172             playlists[i] = playlists[i+1];
 173
 174         (*num_playlists)--;
 175
 176         most_recent_playlist[0] = '\0';
 177     }
 178
 179     result = 0;
 180
 181 exit:
 182     *(tc->dirfilter) = dirfilter;
 183     return result;
 184 }
 185
 186 /* Callback for gui_synclist */
 187 static const char* playlist_callback_name(int selected_item, void* data,
 188                                           char* buffer, size_t buffer_len)
 189 {
 190     char** playlists = (char**) data;
 191
 192     strlcpy(buffer, playlists[selected_item], buffer_len);
 193
 194     if (buffer[0] != '.' && !(global_settings.show_filename_ext == 1
 195         || (global_settings.show_filename_ext == 3
 196             && global_settings.dirfilter == 0)))
 197     {
 198         char* dot = strrchr(buffer, '.');
 199
 200         if (dot != NULL)
 201         {
 202             *dot = '\0';
 203         }
 204     }
 205
 206     return buffer;
 207 }
 208
 209 static int playlist_callback_voice(int selected_item, void* data)
 210 {
 211     char** playlists = (char**) data;
 212     talk_file_or_spell(playlist_dir, playlists[selected_item], NULL, false);
 213     return 0;
 214 }
 215
 216 /* Display all playlists in catalog.  Selected "playlist" is returned.
 217    If "view" mode is set then we're not adding anything into playlist. */
 218 static int display_playlists(char* playlist, bool view)
 219 {
 220     int result = -1;
 221     int num_playlists = 0;
 222     bool exit = false;
 223     char temp_buf[MAX_PATH];
 224     char* playlists[MAX_PLAYLISTS];
 225     struct gui_synclist playlist_lists;
 226
 227     if (create_playlist_list(playlists, sizeof(playlists),
 228             &num_playlists) != 0)
 229         return -1;
 230
 231     if (num_playlists <= 0)
 232     {
 233         splash(HZ*2, ID2P(LANG_CATALOG_NO_PLAYLISTS));
 234         return -1;
 235     }
 236
 237     if (!playlist)
 238         playlist = temp_buf;
 239
 240     gui_synclist_init(&playlist_lists, playlist_callback_name, playlists,
 241                        false, 1, NULL);
 242     if(global_settings.talk_menu)
 243         gui_synclist_set_voice_callback(&playlist_lists,
 244                                         playlist_callback_voice);
 245     gui_synclist_set_nb_items(&playlist_lists, num_playlists);
 246     gui_synclist_draw(&playlist_lists);
 247     gui_synclist_speak_item(&playlist_lists);
 248
 249     while (!exit)
 250     {
 251         int button;
 252         char* sel_file;
 253         list_do_action(CONTEXT_LIST,HZ/2,
 254                        &playlist_lists, &button,LIST_WRAP_UNLESS_HELD);
 255         sel_file = playlists[gui_synclist_get_sel_pos(&playlist_lists)];
 256
 257         switch (button)
 258         {
 259             case ACTION_STD_CANCEL:
 260                 exit = true;
 261                 break;
 262
 263             case ACTION_STD_OK:
 264                 snprintf(playlist, MAX_PATH, "%s/%s", playlist_dir, sel_file);
 265
 266                 if (view)
 267                 {
 268                     /* In view mode, selecting a playlist starts playback */
 269                     ft_play_playlist(playlist, playlist_dir, sel_file);
 270                 }
 271
 272                 result = 0;
 273                 exit = true;
 274                 break;
 275
 276             case ACTION_STD_CONTEXT:
 277                 /* context menu only available in view mode */
 278                 if (view)
 279                 {
 280                     snprintf(playlist, MAX_PATH, "%s/%s", playlist_dir,
 281                         sel_file);
 282
 283                     if (onplay(playlist, FILE_ATTR_M3U,
 284                             CONTEXT_TREE) != ONPLAY_OK)
 285                     {
 286                         result = 0;
 287                         exit = true;
 288                     }
 289                     else
 290                     {
 291                         gui_synclist_draw(&playlist_lists);
 292                         gui_synclist_speak_item(&playlist_lists);
 293                     }
 294                 }
 295                 break;
 296
 297             default:
 298                 if(default_event_handler(button) == SYS_USB_CONNECTED)
 299                 {
 300                     result = -1;
 301                     exit = true;
 302                 }
 303                 break;
 304         }
 305     }
 306     return result;
 307 }
 308
 309 /* display number of tracks inserted into playlists.  Used for directory
 310    insert */
 311 static void display_insert_count(int count)
 312 {
 313     static long talked_tick = 0;
 314     if(global_settings.talk_menu && count &&
 315         (talked_tick == 0 || TIME_AFTER(current_tick, talked_tick+5*HZ)))
 316     {
 317         talked_tick = current_tick;
 318         talk_number(count, false);
 319         talk_id(LANG_PLAYLIST_INSERT_COUNT, true);
 320     }
 321
 322     splashf(0, str(LANG_PLAYLIST_INSERT_COUNT), count, str(LANG_OFF_ABORT));
 323 }
 324
 325 /* Add specified track into playlist.  Callback from directory insert */
 326 static int add_track_to_playlist(char* filename, void* context)
 327 {
 328     struct add_track_context* c = (struct add_track_context*) context;
 329
 330     if (fdprintf(c->fd, "%s\n", filename) <= 0)
 331         return -1;
 332
 333     (c->count)++;
 334
 335     if (((c->count)%PLAYLIST_DISPLAY_COUNT) == 0)
 336         display_insert_count(c->count);
 337
 338     return 0;
 339 }
 340
 341 /* Add "sel" file into specified "playlist".  How to insert depends on type
 342    of file */
 343 static int add_to_playlist(const char* playlist, bool new_playlist,
 344                            const char* sel, int sel_attr)
 345 {
 346     int fd;
 347     int result = -1;
 348     int flags = O_CREAT|O_WRONLY;
 349
 350     if (new_playlist)
 351         flags |= O_TRUNC;
 352     else
 353         flags |= O_APPEND;
 354
 355     fd = open(playlist, flags);
 356     if(fd < 0)
 357         return result;
 358
 359     /* In case we're in the playlist directory */
 360     reload_directory();
 361
 362     if ((sel_attr & FILE_ATTR_MASK) == FILE_ATTR_AUDIO)
 363     {
 364         /* append the selected file */
 365         if (fdprintf(fd, "%s\n", sel) > 0)
 366             result = 0;
 367     }
 368     else if ((sel_attr & FILE_ATTR_MASK) == FILE_ATTR_M3U)
 369     {
 370         /* append playlist */
 371         int f, fs, i;
 372         char buf[1024];
 373
 374         if(strcasecmp(playlist, sel) == 0)
 375             goto exit;
 376
 377         f = open(sel, O_RDONLY);
 378         if (f < 0)
 379             goto exit;
 380
 381         fs = filesize(f);
 382
 383         for (i=0; i<fs;)
 384         {
 385             int n;
 386
 387             n = read(f, buf, sizeof(buf));
 388             if (n < 0)
 389                 break;
 390
 391             if (write(fd, buf, n) < 0)
 392                 break;
 393
 394             i += n;
 395         }
 396
 397         if (i >= fs)
 398             result = 0;
 399
 400         close(f);
 401     }
 402     else if (sel_attr & ATTR_DIRECTORY)
 403     {
 404         /* search directory for tracks and append to playlist */
 405         bool recurse = false;
 406         const char *lines[] = {
 407             ID2P(LANG_RECURSE_DIRECTORY_QUESTION), sel};
 408         const struct text_message message={lines, 2};
 409         struct add_track_context context;
 410
 411         if (global_settings.recursive_dir_insert != RECURSE_ASK)
 412             recurse = (bool)global_settings.recursive_dir_insert;
 413         else
 414         {
 415             /* Ask if user wants to recurse directory */
 416             recurse = (gui_syncyesno_run(&message, NULL, NULL)==YESNO_YES);
 417         }
 418
 419         context.fd = fd;
 420         context.count = 0;
 421
 422         display_insert_count(0);
 423
 424         result = playlist_directory_tracksearch(sel, recurse,
 425             add_track_to_playlist, &context);
 426
 427         display_insert_count(context.count);
 428     }
 429
 430 exit:
 431     close(fd);
 432     return result;
 433 }
 434
 435 bool catalog_view_playlists(void)
 436 {
 437     if (initialize_catalog() == -1)
 438         return false;
 439
 440     if (display_playlists(NULL, true) == -1)
 441         return false;
 442
 443     return true;
 444 }
 445
 446 bool catalog_add_to_a_playlist(const char* sel, int sel_attr,
 447                                bool new_playlist, char *m3u8name)
 448 {
 449     char playlist[MAX_PATH];
 450
 451     if (initialize_catalog() == -1)
 452         return false;
 453
 454     if (new_playlist)
 455     {
 456         size_t len;
 457         if (m3u8name == NULL)
 458         {
 459             snprintf(playlist, MAX_PATH, "%s/", playlist_dir);
 460             if (kbd_input(playlist, MAX_PATH))
 461                 return false;
 462         }
 463         else
 464             strcpy(playlist, m3u8name);
 465
 466         len = strlen(playlist);
 467
 468         if(len > 4 && !strcasecmp(&playlist[len-4], ".m3u"))
 469             strcat(playlist, "8");
 470         else if(len <= 5 || strcasecmp(&playlist[len-5], ".m3u8"))
 471             strcat(playlist, ".m3u8");
 472     }
 473     else
 474     {
 475         if (display_playlists(playlist, false) == -1)
 476             return false;
 477     }
 478
 479     if (add_to_playlist(playlist, new_playlist, sel, sel_attr) == 0)
 480     {
 481         strlcpy(most_recent_playlist, playlist+playlist_dir_length+1,
 482             sizeof(most_recent_playlist));
 483         return true;
 484     }
 485     else
 486         return false;
 487 }