play.c

   1 /* Copyright (C) 2012 Andre Noll <maan@tuebingen.mpg.de>, see file COPYING. */
   2
   3 /** \file play.c Paraslash's standalone player. */
   4
   5 #include <regex.h>
   6 #include <signal.h>
   7 #include <lopsub.h>
   8
   9 #include "recv_cmd.lsg.h"
  10 #include "play_cmd.lsg.h"
  11 #include "write_cmd.lsg.h"
  12 #include "play.lsg.h"
  13 #include "para.h"
  14 #include "lsu.h"
  15 #include "list.h"
  16 #include "error.h"
  17 #include "buffer_tree.h"
  18 #include "version.h"
  19 #include "string.h"
  20 #include "sched.h"
  21 #include "filter.h"
  22 #include "afh.h"
  23 #include "recv.h"
  24 #include "write.h"
  25 #include "fd.h"
  26
  27 /**
  28  * Besides playback tasks which correspond to the receiver/filter/writer nodes,
  29  * para_play creates two further tasks: The play task and the i9e task. It is
  30  * important whether a function can be called in the context of para_play or
  31  * i9e or both. As a rule, all command handlers are called only in i9e context via
  32  * the line handler (input mode) or the key handler (command mode) below.
  33  *
  34  * Playlist handling is done exclusively in play context.
  35  */
  36
  37 /** Array of error strings. */
  38 DEFINE_PARA_ERRLIST;
  39
  40 static struct lls_parse_result *play_lpr;
  41
  42 #define CMD_PTR (lls_cmd(0, play_suite))
  43 #define OPT_RESULT(_name) \
  44         (lls_opt_result(LSG_PLAY_PARA_PLAY_OPT_ ## _name, play_lpr))
  45 #define OPT_GIVEN(_name) (lls_opt_given(OPT_RESULT(_name)))
  46 #define OPT_UINT32_VAL(_name) (lls_uint32_val(0, OPT_RESULT(_name)))
  47 #define OPT_STRING_VAL(_name) (lls_string_val(0, OPT_RESULT(_name)))
  48
  49 /**
  50  * Describes a request to change the state of para_play.
  51  *
  52  * There is only one variable of this type: \a rq of the global play task
  53  * structure. Command handlers only set this variable and the post_select()
  54  * function of the play task investigates its value during each iteration of
  55  * the scheduler run and performs the actual work.
  56  */
  57 enum state_change_request_type {
  58         /** Everybody is happy. */
  59         CRT_NONE,
  60         /** Stream must be repositioned (com_jmp(), com_ff()). */
  61         CRT_REPOS,
  62         /** New file should be loaded (com_next()). */
  63         CRT_FILE_CHANGE,
  64         /** Someone wants us for dead (com_quit()). */
  65         CRT_TERM_RQ
  66 };
  67
  68 struct play_task {
  69         struct task *task;
  70         /* A bit array of invalid files (those will be skipped). */
  71         bool *invalid;
  72         /* The file which is currently open. */
  73         unsigned current_file;
  74         /* When to update the status again. */
  75         struct timeval next_update;
  76
  77         /* Root of the buffer tree for command and status output. */
  78         struct btr_node *btrn;
  79
  80         /* The decoding machinery.  */
  81         struct receiver_node rn;
  82         struct filter_node fn;
  83         struct writer_node wn;
  84
  85         /* See comment to enum \ref state_change_request_type above. */
  86         enum state_change_request_type rq;
  87         /* only relevant if rq == CRT_FILE_CHANGE */
  88         unsigned next_file;
  89         /*
  90                 bg: read lines at prompt, fg: display status and wait
  91                 for keystroke.
  92         */
  93         bool background;
  94
  95         /* We have the *intention* to play. Set by com_play(). */
  96         bool playing;
  97
  98         /* as returned by afh_recv->open() */
  99         int audio_format_num;
 100
 101         /* retrieved via the btr exec mechanism */
 102         long unsigned start_chunk;
 103         long unsigned seconds;
 104         long unsigned num_chunks;
 105         char *afhi_txt;
 106 };
 107
 108 typedef int (*play_cmd_handler_t)(struct lls_parse_result *lpr);
 109 struct play_command_info {
 110         play_cmd_handler_t handler;
 111 };
 112 #define EXPORT_PLAY_CMD_HANDLER(_cmd) \
 113         const struct play_command_info lsg_play_cmd_com_ ## _cmd ## _user_data = { \
 114                 .handler = com_ ## _cmd \
 115         };
 116
 117 static int loglevel = LL_WARNING;
 118
 119 /** The log function which writes log messages to stderr. */
 120 INIT_STDERR_LOGGING(loglevel);
 121
 122 char *stat_item_values[NUM_STAT_ITEMS] = {NULL};
 123
 124 static struct sched sched = {.max_fileno = 0};
 125 static struct play_task play_task, *pt = &play_task;
 126
 127 #define AFH_RECV_CMD (lls_cmd(LSG_RECV_CMD_CMD_AFH, recv_cmd_suite))
 128 #define AFH_RECV ((struct receiver *)lls_user_data(AFH_RECV_CMD))
 129
 130 static unsigned *shuffle_map;
 131
 132 static const char *get_playlist_file(unsigned idx)
 133 {
 134         return lls_input(shuffle_map[idx], play_lpr);
 135 }
 136
 137 static void handle_help_flags(void)
 138 {
 139         char *help;
 140
 141         if (OPT_GIVEN(DETAILED_HELP))
 142                 help = lls_long_help(CMD_PTR);
 143         else if (OPT_GIVEN(HELP) || lls_num_inputs(play_lpr) == 0)
 144                 help = lls_short_help(CMD_PTR);
 145         else
 146                 return;
 147         printf("%s\n", help);
 148         free(help);
 149         exit(EXIT_SUCCESS);
 150 }
 151
 152 static void parse_config_or_die(int argc, char *argv[])
 153 {
 154         int i, ret;
 155         unsigned num_kmas;
 156         char *errctx;
 157
 158         ret = lls(lls_parse(argc, argv, CMD_PTR, &play_lpr, &errctx));
 159         if (ret < 0) {
 160                 if (errctx)
 161                         PARA_EMERG_LOG("%s\n", errctx);
 162                 free(errctx);
 163                 PARA_EMERG_LOG("failed to parse command line options: %s\n",
 164                         para_strerror(-ret));
 165                 exit(EXIT_FAILURE);
 166         }
 167         loglevel = OPT_UINT32_VAL(LOGLEVEL);
 168         version_handle_flag("play", OPT_GIVEN(VERSION));
 169         handle_help_flags(); /* also handles the zero-arg case */
 170         ret = lsu_merge_config_file_options(OPT_STRING_VAL(CONFIG_FILE),
 171                 "play.conf", &play_lpr, CMD_PTR, play_suite, 0 /* flags */);
 172         if (ret < 0) {
 173                 PARA_EMERG_LOG("failed to parse config file: %s\n",
 174                         para_strerror(-ret));
 175                 exit(EXIT_FAILURE);
 176         }
 177         loglevel = OPT_UINT32_VAL(LOGLEVEL);
 178         num_kmas = OPT_GIVEN(KEY_MAP);
 179         for (i = 0; i < num_kmas; i++) {
 180                 const char *kma = lls_string_val(i, OPT_RESULT(KEY_MAP));
 181                 if (*kma && strchr(kma + 1, ':'))
 182                         continue;
 183                 PARA_EMERG_LOG("invalid key map arg: %s\n", kma);
 184                 exit(EXIT_FAILURE);
 185         }
 186 }
 187
 188 static char get_playback_state(void)
 189 {
 190         switch (pt->rq) {
 191         case CRT_NONE: return pt->playing? 'P' : 'U';
 192         case CRT_REPOS: return 'R';
 193         case CRT_FILE_CHANGE: return 'F';
 194         case CRT_TERM_RQ: return 'X';
 195         }
 196         assert(false);
 197 };
 198
 199 static long unsigned get_play_time(void)
 200 {
 201         char state = get_playback_state();
 202         long unsigned result;
 203
 204         if (state != 'P' && state != 'U')
 205                 return 0;
 206         if (pt->num_chunks == 0 || pt->seconds == 0)
 207                 return 0;
 208         /* where the stream started (in seconds) */
 209         result = pt->start_chunk * pt->seconds / pt->num_chunks;
 210         if (pt->wn.btrn) { /* Add the uptime of the writer node */
 211                 struct timeval diff = {.tv_sec = 0}, wstime;
 212                 btr_get_node_start(pt->wn.btrn, &wstime);
 213                 if (wstime.tv_sec > 0)
 214                         tv_diff(now, &wstime, &diff);
 215                 result += diff.tv_sec;
 216         }
 217         result = PARA_MIN(result, pt->seconds);
 218         result = PARA_MAX(result, 0UL);
 219         return result;
 220 }
 221
 222
 223 static void wipe_receiver_node(void)
 224 {
 225         PARA_NOTICE_LOG("cleaning up receiver node\n");
 226         btr_remove_node(&pt->rn.btrn);
 227         AFH_RECV->close(&pt->rn);
 228         lls_free_parse_result(pt->rn.lpr, AFH_RECV_CMD);
 229         memset(&pt->rn, 0, sizeof(struct receiver_node));
 230 }
 231
 232 /* returns: 0 not eof, 1: eof, < 0: fatal error.  */
 233 static int get_playback_error(void)
 234 {
 235         int err;
 236
 237         if (!pt->wn.task)
 238                 return 0;
 239         err = task_status(pt->wn.task);
 240         if (err >= 0)
 241                 return 0;
 242         if (task_status(pt->fn.task) >= 0)
 243                 return 0;
 244         if (task_status(pt->rn.task) >= 0)
 245                 return 0;
 246         if (err == -E_BTR_EOF || err == -E_RECV_EOF || err == -E_EOF
 247                         || err == -E_WRITE_COMMON_EOF)
 248                 return 1;
 249         return err;
 250 }
 251
 252 static int eof_cleanup(void)
 253 {
 254         const struct filter *decoder;
 255         const struct writer *w = writer_get(-1); /* default writer */
 256         int ret;
 257
 258         ret = get_playback_error();
 259         if (ret == 0)
 260                 return ret;
 261         PARA_NOTICE_LOG("cleaning up wn/fn nodes\n");
 262         task_reap(&pt->wn.task);
 263         w->close(&pt->wn);
 264         btr_remove_node(&pt->wn.btrn);
 265         lls_free_parse_result(pt->wn.lpr, WRITE_CMD(pt->wn.wid));
 266         memset(&pt->wn, 0, sizeof(struct writer_node));
 267
 268         decoder = filter_get(pt->fn.filter_num);
 269         task_reap(&pt->fn.task);
 270         if (decoder->close)
 271                 decoder->close(&pt->fn);
 272         btr_remove_node(&pt->fn.btrn);
 273         free(pt->fn.conf);
 274         memset(&pt->fn, 0, sizeof(struct filter_node));
 275
 276         task_reap(&pt->rn.task);
 277         btr_remove_node(&pt->rn.btrn);
 278         /*
 279          * On eof (ret > 0), we do not wipe the receiver node struct until a
 280          * new file is loaded because we still need it for jumping around when
 281          * paused.
 282          */
 283         if (ret < 0)
 284                 wipe_receiver_node();
 285         return ret;
 286 }
 287
 288 static int shuffle_compare(__a_unused const void *a, __a_unused const void *b)
 289 {
 290         return para_random(100) - 50;
 291 }
 292
 293 static void init_shuffle_map(void)
 294 {
 295         unsigned n, num_inputs = lls_num_inputs(play_lpr);
 296         shuffle_map = para_malloc(num_inputs * sizeof(unsigned));
 297         for (n = 0; n < num_inputs; n++)
 298                 shuffle_map[n] = n;
 299         if (!OPT_GIVEN(RANDOMIZE))
 300                 return;
 301         srandom(time(NULL));
 302         qsort(shuffle_map, num_inputs, sizeof(unsigned), shuffle_compare);
 303 }
 304
 305 static struct btr_node *new_recv_btrn(struct receiver_node *rn)
 306 {
 307         return btr_new_node(&(struct btr_node_description)
 308                 EMBRACE(.name = lls_command_name(AFH_RECV_CMD), .context = rn,
 309                         .handler = AFH_RECV->execute));
 310 }
 311
 312 static int open_new_file(void)
 313 {
 314         int ret;
 315         const char *path = get_playlist_file(pt->next_file);
 316         char *tmp = para_strdup(path), *errctx;
 317         char *argv[] = {"play", "-f", tmp, "-b", "0", NULL};
 318
 319         PARA_NOTICE_LOG("next file: %s\n", path);
 320         wipe_receiver_node();
 321         pt->start_chunk = 0;
 322         pt->rn.btrn = new_recv_btrn(&pt->rn);
 323         ret = lls(lls_parse(ARRAY_SIZE(argv) - 1, argv, AFH_RECV_CMD,
 324                 &pt->rn.lpr, &errctx));
 325         free(tmp);
 326         assert(ret >= 0);
 327         pt->rn.receiver = AFH_RECV;
 328         ret = AFH_RECV->open(&pt->rn);
 329         if (ret < 0) {
 330                 PARA_ERROR_LOG("could not open %s\n", path);
 331                 goto fail;
 332         }
 333         pt->audio_format_num = ret;
 334         free(pt->afhi_txt);
 335         ret = btr_exec_up(pt->rn.btrn, "afhi", &pt->afhi_txt);
 336         if (ret < 0)
 337                 pt->afhi_txt = make_message("[afhi command failed]\n");
 338         ret = btr_exec_up(pt->rn.btrn, "seconds_total", &tmp);
 339         if (ret < 0)
 340                 pt->seconds = 1;
 341         else {
 342                 int32_t x;
 343                 ret = para_atoi32(tmp, &x);
 344                 pt->seconds = ret < 0? 1 : x;
 345                 free(tmp);
 346                 tmp = NULL;
 347         }
 348         ret = btr_exec_up(pt->rn.btrn, "chunks_total", &tmp);
 349         if (ret < 0)
 350                 pt->num_chunks = 1;
 351         else {
 352                 int32_t x;
 353                 ret = para_atoi32(tmp, &x);
 354                 pt->num_chunks = ret < 0? 1 : x;
 355                 free(tmp);
 356                 tmp = NULL;
 357         }
 358         return 1;
 359 fail:
 360         wipe_receiver_node();
 361         return ret;
 362 }
 363
 364 static int load_file(void)
 365 {
 366         const char *af;
 367         char *tmp, buf[20];
 368         int ret;
 369         const struct filter *decoder;
 370         static struct lls_parse_result *filter_lpr, *writer_lpr;
 371
 372         btr_remove_node(&pt->rn.btrn);
 373         if (!pt->rn.receiver || pt->next_file != pt->current_file) {
 374                 ret = open_new_file();
 375                 if (ret < 0)
 376                         return ret;
 377         } else {
 378                 pt->rn.btrn = new_recv_btrn(&pt->rn);
 379                 sprintf(buf, "repos %lu", pt->start_chunk);
 380                 ret = btr_exec_up(pt->rn.btrn, buf, &tmp);
 381                 if (ret < 0)
 382                         PARA_CRIT_LOG("repos failed: %s\n", para_strerror(-ret));
 383                 freep(&tmp);
 384         }
 385         if (!pt->playing)
 386                 return 0;
 387         /* set up decoding filter */
 388         af = audio_format_name(pt->audio_format_num);
 389         tmp = make_message("%sdec", af);
 390         ret = filter_setup(tmp, &pt->fn.conf, &filter_lpr);
 391         freep(&tmp);
 392         if (ret < 0)
 393                 goto fail;
 394         pt->fn.filter_num = ret;
 395         pt->fn.lpr = filter_lpr;
 396         decoder = filter_get(ret);
 397         pt->fn.btrn = btr_new_node(&(struct btr_node_description)
 398                 EMBRACE(.name = filter_name(pt->fn.filter_num),
 399                         .parent = pt->rn.btrn, .handler = decoder->execute,
 400                         .context = &pt->fn));
 401         if (decoder->open)
 402                 decoder->open(&pt->fn);
 403         PARA_INFO_LOG("buffer tree:\n");
 404         btr_log_tree(pt->rn.btrn, LL_INFO);
 405
 406         /* setup default writer */
 407         pt->wn.wid = check_writer_arg_or_die(NULL, &writer_lpr);
 408         pt->wn.lpr = writer_lpr;
 409         /* success, register tasks */
 410         pt->rn.task = task_register(
 411                 &(struct task_info) {
 412                         .name = lls_command_name(AFH_RECV_CMD),
 413                         .pre_select = AFH_RECV->pre_select,
 414                         .post_select = AFH_RECV->post_select,
 415                         .context = &pt->rn
 416                 }, &sched);
 417         sprintf(buf, "%s decoder", af);
 418         pt->fn.task = task_register(
 419                 &(struct task_info) {
 420                         .name = buf,
 421                         .pre_select = decoder->pre_select,
 422                         .post_select = decoder->post_select,
 423                         .context = &pt->fn
 424                 }, &sched);
 425         register_writer_node(&pt->wn, pt->fn.btrn, &sched);
 426         return 1;
 427 fail:
 428         wipe_receiver_node();
 429         return ret;
 430 }
 431
 432 static int next_valid_file(void)
 433 {
 434         int i, j = pt->current_file;
 435         unsigned num_inputs = lls_num_inputs(play_lpr);
 436
 437         for (i = 0; i < num_inputs; i++) {
 438                 j = (j + 1) % num_inputs;
 439                 if (!pt->invalid[j])
 440                         return j;
 441         }
 442         return -E_NO_VALID_FILES;
 443 }
 444
 445 static int load_next_file(void)
 446 {
 447         int ret;
 448
 449 again:
 450         if (pt->rq == CRT_NONE) {
 451                 pt->start_chunk = 0;
 452                 ret = next_valid_file();
 453                 if (ret < 0)
 454                         return ret;
 455                 pt->next_file = ret;
 456         } else if (pt->rq == CRT_REPOS)
 457                 pt->next_file = pt->current_file;
 458         ret = load_file();
 459         if (ret < 0) {
 460                 PARA_ERROR_LOG("%s: marking file as invalid\n",
 461                         para_strerror(-ret));
 462                 pt->invalid[pt->next_file] = true;
 463                 pt->rq = CRT_NONE;
 464                 goto again;
 465         }
 466         pt->current_file = pt->next_file;
 467         pt->rq = CRT_NONE;
 468         return ret;
 469 }
 470
 471 static void kill_stream(void)
 472 {
 473         if (pt->wn.task)
 474                 task_notify(pt->wn.task, E_EOF);
 475 }
 476
 477 #ifdef HAVE_READLINE
 478
 479 /* only called from com_prev(), nec. only if we have readline */
 480 static int previous_valid_file(void)
 481 {
 482         int i, j = pt->current_file;
 483         unsigned num_inputs = lls_num_inputs(play_lpr);
 484
 485         for (i = 0; i < num_inputs; i++) {
 486                 j--;
 487                 if (j < 0)
 488                         j = num_inputs - 1;
 489                 if (!pt->invalid[j])
 490                         return j;
 491         }
 492         return -E_NO_VALID_FILES;
 493 }
 494
 495 #include "interactive.h"
 496
 497 /*
 498  * Define the default (internal) key mappings and helper functions to get the
 499  * key sequence or the command from a key id, which is what we obtain from
 500  * i9e/readline when the key is pressed.
 501  *
 502  * In some of these helper functions we could return pointers to the constant
 503  * arrays defined below. However, for others we can not, so let's better be
 504  * consistent and allocate all returned strings on the heap.
 505  */
 506
 507 #define INTERNAL_KEYMAP_ENTRIES \
 508         KEYMAP_ENTRY("^", "jmp 0"), \
 509         KEYMAP_ENTRY("1", "jmp 10"), \
 510         KEYMAP_ENTRY("2", "jmp 21"), \
 511         KEYMAP_ENTRY("3", "jmp 32"), \
 512         KEYMAP_ENTRY("4", "jmp 43"), \
 513         KEYMAP_ENTRY("5", "jmp 54"), \
 514         KEYMAP_ENTRY("6", "jmp 65"), \
 515         KEYMAP_ENTRY("7", "jmp 76"), \
 516         KEYMAP_ENTRY("8", "jmp 87"), \
 517         KEYMAP_ENTRY("9", "jmp 98"), \
 518         KEYMAP_ENTRY("+", "next"), \
 519         KEYMAP_ENTRY("-", "prev"), \
 520         KEYMAP_ENTRY(":", "bg"), \
 521         KEYMAP_ENTRY("i", "info"), \
 522         KEYMAP_ENTRY("l", "ls"), \
 523         KEYMAP_ENTRY("s", "play"), \
 524         KEYMAP_ENTRY("p", "pause"), \
 525         KEYMAP_ENTRY("q", "quit"), \
 526         KEYMAP_ENTRY("?", "help"), \
 527         KEYMAP_ENTRY("\033[D", "ff -10"), \
 528         KEYMAP_ENTRY("\033[C", "ff 10"), \
 529         KEYMAP_ENTRY("\033[A", "ff 60"), \
 530         KEYMAP_ENTRY("\033[B", "ff -60"), \
 531
 532 #define KEYMAP_ENTRY(a, b) a
 533 static const char *default_keyseqs[] = {INTERNAL_KEYMAP_ENTRIES};
 534 #undef KEYMAP_ENTRY
 535 #define KEYMAP_ENTRY(a, b) b
 536 static const char *default_commands[] = {INTERNAL_KEYMAP_ENTRIES};
 537 #undef KEYMAP_ENTRY
 538 #define NUM_INTERNALLY_MAPPED_KEYS ARRAY_SIZE(default_commands)
 539 #define NUM_MAPPED_KEYS (NUM_INTERNALLY_MAPPED_KEYS + OPT_GIVEN(KEY_MAP))
 540 #define FOR_EACH_MAPPED_KEY(i) for (i = 0; i < NUM_MAPPED_KEYS; i++)
 541
 542 static inline bool is_internal_key(int key)
 543 {
 544         return key < NUM_INTERNALLY_MAPPED_KEYS;
 545 }
 546
 547 /* for internal keys, the key id is just the array index. */
 548 static inline int get_internal_key_map_idx(int key)
 549 {
 550         assert(is_internal_key(key));
 551         return key;
 552 }
 553
 554 /*
 555  * For user-defined keys, we have to subtract NUM_INTERNALLY_MAPPED_KEYS. The
 556  * difference is the index to the array of user defined key maps.
 557  */
 558 static inline int get_user_key_map_idx(int key)
 559 {
 560         assert(!is_internal_key(key));
 561         return key - NUM_INTERNALLY_MAPPED_KEYS;
 562 }
 563
 564 static inline int get_key_map_idx(int key)
 565 {
 566         return is_internal_key(key)?
 567                 get_internal_key_map_idx(key) : get_user_key_map_idx(key);
 568 }
 569
 570 static inline const char *get_user_key_map_arg(int key)
 571 {
 572         return lls_string_val(get_user_key_map_idx(key), OPT_RESULT(KEY_MAP));
 573 }
 574
 575 static inline char *get_internal_key_map_seq(int key)
 576 {
 577         return para_strdup(default_keyseqs[get_internal_key_map_idx(key)]);
 578 }
 579
 580 static char *get_user_key_map_seq(int key)
 581 {
 582         const char *kma = get_user_key_map_arg(key);
 583         const char *p = strchr(kma + 1, ':');
 584         char *result;
 585         int len;
 586
 587         if (!p)
 588                 return NULL;
 589         len = p - kma;
 590         result = para_malloc(len + 1);
 591         memcpy(result, kma, len);
 592         result[len] = '\0';
 593         return result;
 594 }
 595
 596 static char *get_key_map_seq(int key)
 597 {
 598         return is_internal_key(key)?
 599                 get_internal_key_map_seq(key) : get_user_key_map_seq(key);
 600 }
 601
 602 static char *get_key_map_seq_safe(int key)
 603 {
 604         const char hex[] = "0123456789abcdef";
 605         char *seq = get_key_map_seq(key), *sseq;
 606         size_t n, len = strlen(seq);
 607
 608         if (len == 1 && isprint(*seq))
 609                 return seq;
 610         sseq = para_malloc(2 + 2 * len + 1);
 611         sseq[0] = '0';
 612         sseq[1] = 'x';
 613         for (n = 0; n < len; n++) {
 614                 uint8_t val = (seq[n] & 0xf0) >> 4;
 615                 sseq[2 + 2 * n] = hex[val];
 616                 val = seq[n] & 0xf;
 617                 sseq[2 + 2 * n + 1] = hex[val];
 618         }
 619         free(seq);
 620         sseq[2 + 2 * n] = '\0';
 621         return sseq;
 622 }
 623
 624 static inline char *get_internal_key_map_cmd(int key)
 625 {
 626         return para_strdup(default_commands[get_internal_key_map_idx(key)]);
 627 }
 628
 629 static char *get_user_key_map_cmd(int key)
 630 {
 631         const char *kma = get_user_key_map_arg(key);
 632         const char *p = strchr(kma + 1, ':');
 633
 634         if (!p)
 635                 return NULL;
 636         return para_strdup(p + 1);
 637 }
 638
 639 static char *get_key_map_cmd(int key)
 640 {
 641         return is_internal_key(key)?
 642                 get_internal_key_map_cmd(key) : get_user_key_map_cmd(key);
 643 }
 644
 645 static char **get_mapped_keyseqs(void)
 646 {
 647         char **result;
 648         int i;
 649
 650         result = para_malloc((NUM_MAPPED_KEYS + 1) * sizeof(char *));
 651         FOR_EACH_MAPPED_KEY(i) {
 652                 char *seq = get_key_map_seq(i);
 653                 result[i] = seq;
 654         }
 655         result[i] = NULL;
 656         return result;
 657 }
 658
 659 static struct i9e_completer pp_completers[];
 660
 661 I9E_DUMMY_COMPLETER(jmp);
 662 I9E_DUMMY_COMPLETER(next);
 663 I9E_DUMMY_COMPLETER(prev);
 664 I9E_DUMMY_COMPLETER(fg);
 665 I9E_DUMMY_COMPLETER(bg);
 666 I9E_DUMMY_COMPLETER(ls);
 667 I9E_DUMMY_COMPLETER(info);
 668 I9E_DUMMY_COMPLETER(play);
 669 I9E_DUMMY_COMPLETER(pause);
 670 I9E_DUMMY_COMPLETER(tasks);
 671 I9E_DUMMY_COMPLETER(quit);
 672 I9E_DUMMY_COMPLETER(ff);
 673
 674 static void help_completer(struct i9e_completion_info *ci,
 675                 struct i9e_completion_result *cr)
 676 {
 677         char *opts[] = {LSG_PLAY_CMD_HELP_OPTS, NULL};
 678
 679         if (ci->word[0] == '-') {
 680                 i9e_complete_option(opts, ci, cr);
 681                 return;
 682         }
 683         cr->matches = i9e_complete_commands(ci->word, pp_completers);
 684 }
 685
 686 static struct i9e_completer pp_completers[] = {
 687 #define LSG_PLAY_CMD_CMD(_name) {.name = #_name, \
 688         .completer = _name ## _completer}
 689         LSG_PLAY_CMD_SUBCOMMANDS
 690 #undef LSG_PLAY_CMD_CMD
 691         {.name = NULL}
 692 };
 693
 694 static void attach_stdout(const char *name)
 695 {
 696         if (pt->btrn)
 697                 return;
 698         pt->btrn = btr_new_node(&(struct btr_node_description)
 699                 EMBRACE(.name = name));
 700         i9e_attach_to_stdout(pt->btrn);
 701 }
 702
 703 static void detach_stdout(void)
 704 {
 705         btr_remove_node(&pt->btrn);
 706 }
 707
 708 static int com_quit(__a_unused struct lls_parse_result *lpr)
 709 {
 710         pt->rq = CRT_TERM_RQ;
 711         return 0;
 712 }
 713 EXPORT_PLAY_CMD_HANDLER(quit);
 714
 715 static int com_help(struct lls_parse_result *lpr)
 716 {
 717         int i;
 718         char *buf;
 719         size_t sz;
 720         unsigned n;
 721         const struct lls_opt_result *r =
 722                 lls_opt_result(LSG_PLAY_CMD_HELP_OPT_LONG, lpr);
 723         bool long_help = lls_opt_given(r);
 724
 725         if (!pt->background) {
 726                 FOR_EACH_MAPPED_KEY(i) {
 727                         bool internal = is_internal_key(i);
 728                         int idx = get_key_map_idx(i);
 729                         char *seq = get_key_map_seq_safe(i);
 730                         char *kmc = get_key_map_cmd(i);
 731                         sz = xasprintf(&buf, "%s key #%d: %s -> %s\n",
 732                                 internal? "internal" : "user-defined",
 733                                 idx, seq, kmc);
 734                         btr_add_output(buf, sz, pt->btrn);
 735                         free(seq);
 736                         free(kmc);
 737                 }
 738                 return 0;
 739         }
 740         lsu_com_help(long_help, lpr, play_cmd_suite, NULL, &buf, &n);
 741         btr_add_output(buf, n, pt->btrn);
 742         return 0;
 743 }
 744 EXPORT_PLAY_CMD_HANDLER(help);
 745
 746 static int com_info(__a_unused struct lls_parse_result *lpr)
 747 {
 748         char *buf;
 749         size_t sz;
 750         static char dflt[] = "[no information available]";
 751
 752         sz = xasprintf(&buf, "playlist_pos: %u\npath: %s\n",
 753                 pt->current_file, get_playlist_file(pt->current_file));
 754         btr_add_output(buf, sz, pt->btrn);
 755         buf = pt->afhi_txt? pt->afhi_txt : dflt;
 756         btr_add_output_dont_free(buf, strlen(buf), pt->btrn);
 757         return 0;
 758 }
 759 EXPORT_PLAY_CMD_HANDLER(info);
 760
 761 static void list_file(int num)
 762 {
 763         char *buf;
 764         size_t sz;
 765
 766         sz = xasprintf(&buf, "%s %4d %s\n", num == pt->current_file?
 767                 "*" : " ", num, get_playlist_file(num));
 768         btr_add_output(buf, sz, pt->btrn);
 769 }
 770
 771 static int com_tasks(__a_unused struct lls_parse_result *lpr)
 772 {
 773         static char state;
 774         char *buf;
 775         size_t sz;
 776
 777         buf = get_task_list(&sched);
 778         btr_add_output(buf, strlen(buf), pt->btrn);
 779         state = get_playback_state();
 780         sz = xasprintf(&buf, "state: %c\n", state);
 781         btr_add_output(buf, sz, pt->btrn);
 782         return 0;
 783 }
 784 EXPORT_PLAY_CMD_HANDLER(tasks);
 785
 786 static int com_ls(__a_unused struct lls_parse_result *lpr)
 787 {
 788         int i;
 789         unsigned num_inputs = lls_num_inputs(play_lpr);
 790
 791         for (i = 0; i < num_inputs; i++)
 792                 list_file(i);
 793         return 0;
 794 }
 795 EXPORT_PLAY_CMD_HANDLER(ls);
 796
 797 static int com_play(struct lls_parse_result *lpr)
 798 {
 799         int32_t x;
 800         int ret;
 801         char state, *errctx;
 802
 803         ret = lls(lls_check_arg_count(lpr, 0, 1, &errctx));
 804         if (ret < 0) {
 805                 if (errctx)
 806                         PARA_ERROR_LOG("%s\n", errctx);
 807                 free(errctx);
 808                 return ret;
 809         }
 810         state = get_playback_state();
 811         if (lls_num_inputs(lpr) == 0) {
 812                 if (state == 'P')
 813                         return 0;
 814                 pt->next_file = pt->current_file;
 815                 pt->rq = CRT_REPOS;
 816                 pt->playing = true;
 817                 return 0;
 818         }
 819         ret = para_atoi32(lls_input(0, lpr), &x);
 820         if (ret < 0)
 821                 return ret;
 822         if (x < 0 || x >= lls_num_inputs(play_lpr))
 823                 return -ERRNO_TO_PARA_ERROR(EINVAL);
 824         kill_stream();
 825         pt->next_file = x;
 826         pt->rq = CRT_FILE_CHANGE;
 827         return 0;
 828 }
 829 EXPORT_PLAY_CMD_HANDLER(play);
 830
 831 static int com_pause(__a_unused struct lls_parse_result *lpr)
 832 {
 833         char state;
 834         long unsigned seconds, ss;
 835
 836         state = get_playback_state();
 837         pt->playing = false;
 838         if (state != 'P')
 839                 return 0;
 840         seconds = get_play_time();
 841         pt->playing = false;
 842         ss = 0;
 843         if (pt->seconds > 0)
 844                 ss = seconds * pt->num_chunks / pt->seconds + 1;
 845         ss = PARA_MAX(ss, 0UL);
 846         ss = PARA_MIN(ss, pt->num_chunks);
 847         pt->start_chunk = ss;
 848         kill_stream();
 849         return 0;
 850 }
 851 EXPORT_PLAY_CMD_HANDLER(pause);
 852
 853 static int com_prev(__a_unused struct lls_parse_result *lpr)
 854 {
 855         int ret;
 856
 857         ret = previous_valid_file();
 858         if (ret < 0)
 859                 return ret;
 860         kill_stream();
 861         pt->next_file = ret;
 862         pt->rq = CRT_FILE_CHANGE;
 863         pt->start_chunk = 0;
 864         return 0;
 865 }
 866 EXPORT_PLAY_CMD_HANDLER(prev);
 867
 868 static int com_next(__a_unused struct lls_parse_result *lpr)
 869 {
 870         int ret;
 871
 872         ret = next_valid_file();
 873         if (ret < 0)
 874                 return ret;
 875         kill_stream();
 876         pt->next_file = ret;
 877         pt->rq = CRT_FILE_CHANGE;
 878         pt->start_chunk = 0;
 879         return 0;
 880 }
 881 EXPORT_PLAY_CMD_HANDLER(next);
 882
 883 static int com_fg(__a_unused struct lls_parse_result *lpr)
 884 {
 885         pt->background = false;
 886         return 0;
 887 }
 888 EXPORT_PLAY_CMD_HANDLER(fg);
 889
 890 static int com_bg(__a_unused struct lls_parse_result *lpr)
 891 {
 892         pt->background = true;
 893         return 0;
 894 }
 895 EXPORT_PLAY_CMD_HANDLER(bg);
 896
 897 static int com_jmp(struct lls_parse_result *lpr)
 898 {
 899         int32_t percent;
 900         int ret;
 901         char *errctx;
 902
 903         ret = lls(lls_check_arg_count(lpr, 1, 1, &errctx));
 904         if (ret < 0) {
 905                 if (errctx)
 906                         PARA_ERROR_LOG("%s\n", errctx);
 907                 free(errctx);
 908                 return ret;
 909         }
 910         ret = para_atoi32(lls_input(0, lpr), &percent);
 911         if (ret < 0)
 912                 return ret;
 913         if (percent < 0 || percent > 100)
 914                 return -ERRNO_TO_PARA_ERROR(EINVAL);
 915         if (percent == 100)
 916                 return com_next(NULL);
 917         if (pt->playing && !pt->fn.btrn)
 918                 return 0;
 919         pt->start_chunk = percent * pt->num_chunks / 100;
 920         if (!pt->playing)
 921                 return 0;
 922         pt->rq = CRT_REPOS;
 923         kill_stream();
 924         return 0;
 925 }
 926 EXPORT_PLAY_CMD_HANDLER(jmp);
 927
 928 static int com_ff(struct lls_parse_result *lpr)
 929 {
 930         int32_t seconds;
 931         char *errctx;
 932         int ret;
 933
 934         ret = lls(lls_check_arg_count(lpr, 1, 1, &errctx));
 935         if (ret < 0) {
 936                 if (errctx)
 937                         PARA_ERROR_LOG("%s\n", errctx);
 938                 free(errctx);
 939                 return ret;
 940         }
 941         ret = para_atoi32(lls_input(0, lpr), &seconds);
 942         if (ret < 0)
 943                 return ret;
 944         if (pt->playing && !pt->fn.btrn)
 945                 return 0;
 946         seconds += get_play_time();
 947         seconds = PARA_MIN(seconds, (typeof(seconds))pt->seconds - 4);
 948         seconds = PARA_MAX(seconds, 0);
 949         pt->start_chunk = pt->num_chunks * seconds / pt->seconds;
 950         pt->start_chunk = PARA_MIN(pt->start_chunk, pt->num_chunks - 1);
 951         pt->start_chunk = PARA_MAX(pt->start_chunk, 0UL);
 952         if (!pt->playing)
 953                 return 0;
 954         pt->rq = CRT_REPOS;
 955         kill_stream();
 956         return 0;
 957 }
 958 EXPORT_PLAY_CMD_HANDLER(ff);
 959
 960 static int run_command(char *line)
 961 {
 962         int ret, argc;
 963         char **argv = NULL;
 964         char *errctx = NULL;
 965         const struct play_command_info *pci;
 966         struct lls_parse_result *lpr;
 967         const struct lls_command *cmd;
 968
 969         attach_stdout(__FUNCTION__);
 970         ret = create_argv(line, " ", &argv);
 971         if (ret < 0)
 972                 goto out;
 973         if (ret == 0)
 974                 goto out;
 975         argc = ret;
 976         ret = lls(lls_lookup_subcmd(argv[0], play_cmd_suite, &errctx));
 977         if (ret < 0)
 978                 goto out;
 979         cmd = lls_cmd(ret, play_cmd_suite);
 980         ret = lls(lls_parse(argc, argv, cmd, &lpr, &errctx));
 981         if (ret < 0)
 982                 goto out;
 983         pci = lls_user_data(cmd);
 984         ret = pci->handler(lpr);
 985         lls_free_parse_result(lpr, cmd);
 986 out:
 987         if (errctx)
 988                 PARA_ERROR_LOG("%s\n", errctx);
 989         free(errctx);
 990         free_argv(argv);
 991         return ret;
 992 }
 993
 994 static int play_i9e_line_handler(char *line)
 995 {
 996         return run_command(line);
 997 }
 998
 999 static int play_i9e_key_handler(int key)
1000 {
1001         int idx = get_key_map_idx(key);
1002         char *seq = get_key_map_seq(key);
1003         char *cmd = get_key_map_cmd(key);
1004         bool internal = is_internal_key(key);
1005
1006         PARA_NOTICE_LOG("pressed %d: %s key #%d (%s -> %s)\n",
1007                 key, internal? "internal" : "user-defined",
1008                 idx, seq, cmd);
1009         run_command(cmd);
1010         free(seq);
1011         free(cmd);
1012         pt->next_update = *now;
1013         return 0;
1014 }
1015
1016 static struct i9e_client_info ici = {
1017         .fds = {0, 1, 2},
1018         .prompt = "para_play> ",
1019         .line_handler = play_i9e_line_handler,
1020         .key_handler = play_i9e_key_handler,
1021         .completers = pp_completers,
1022 };
1023
1024 static void sigint_handler(int sig)
1025 {
1026         pt->background = true;
1027         i9e_signal_dispatch(sig);
1028 }
1029
1030 /*
1031  * We start with para_log() set to the standard log function which writes to
1032  * stderr. Once the i9e subsystem has been initialized, we switch to the i9e
1033  * log facility.
1034  */
1035 static void session_open(void)
1036 {
1037         int ret;
1038         char *history_file;
1039         struct sigaction act;
1040
1041         PARA_NOTICE_LOG("\n%s\n", version_text("play"));
1042         if (OPT_GIVEN(HISTORY_FILE))
1043                 history_file = para_strdup(OPT_STRING_VAL(HISTORY_FILE));
1044         else {
1045                 char *home = para_homedir();
1046                 history_file = make_message("%s/.paraslash/play.history",
1047                         home);
1048                 free(home);
1049         }
1050         ici.history_file = history_file;
1051         ici.loglevel = loglevel;
1052
1053         act.sa_handler = sigint_handler;
1054         sigemptyset(&act.sa_mask);
1055         act.sa_flags = 0;
1056         sigaction(SIGINT, &act, NULL);
1057         act.sa_handler = i9e_signal_dispatch;
1058         sigemptyset(&act.sa_mask);
1059         act.sa_flags = 0;
1060         sigaction(SIGWINCH, &act, NULL);
1061         sched.select_function = i9e_select;
1062
1063         ici.bound_keyseqs = get_mapped_keyseqs();
1064         pt->btrn = ici.producer = btr_new_node(&(struct btr_node_description)
1065                 EMBRACE(.name = __FUNCTION__));
1066         ret = i9e_open(&ici, &sched);
1067         if (ret < 0)
1068                 goto out;
1069         para_log = i9e_log;
1070         return;
1071 out:
1072         free(history_file);
1073         if (ret >= 0)
1074                 return;
1075         PARA_EMERG_LOG("fatal: %s\n", para_strerror(-ret));
1076         exit(EXIT_FAILURE);
1077 }
1078
1079 static void session_update_time_string(char *str, unsigned len)
1080 {
1081         if (pt->background)
1082                 return;
1083         if (pt->btrn) {
1084                 if (btr_get_output_queue_size(pt->btrn) > 0)
1085                         return;
1086                 if (btr_get_input_queue_size(pt->btrn) > 0)
1087                         return;
1088         }
1089         ie9_print_status_bar(str, len);
1090 }
1091
1092 /*
1093  * If we are about to die we must call i9e_close() to reset the terminal.
1094  * However, i9e_close() must be called in *this* context, i.e. from
1095  * play_task.post_select() rather than from i9e_post_select(), because
1096  * otherwise i9e would access freed memory upon return. So the play task must
1097  * stay alive until the i9e task terminates.
1098  *
1099  * We achieve this by sending a fake SIGTERM signal via i9e_signal_dispatch()
1100  * and reschedule. In the next iteration, i9e->post_select returns an error and
1101  * terminates. Subsequent calls to i9e_get_error() then return negative and we
1102  * are allowed to call i9e_close() and terminate as well.
1103  */
1104 static int session_post_select(__a_unused struct sched *s)
1105 {
1106         int ret;
1107
1108         if (pt->background)
1109                 detach_stdout();
1110         else
1111                 attach_stdout(__FUNCTION__);
1112         ret = i9e_get_error();
1113         if (ret < 0) {
1114                 kill_stream();
1115                 i9e_close();
1116                 para_log = stderr_log;
1117                 free(ici.history_file);
1118                 return ret;
1119         }
1120         if (get_playback_state() == 'X')
1121                 i9e_signal_dispatch(SIGTERM);
1122         return 0;
1123 }
1124
1125 #else /* HAVE_READLINE */
1126
1127 static int session_post_select(struct sched *s)
1128 {
1129         char c;
1130
1131         if (!FD_ISSET(STDIN_FILENO, &s->rfds))
1132                 return 0;
1133         if (read(STDIN_FILENO, &c, 1))
1134                 do_nothing;
1135         kill_stream();
1136         return 1;
1137 }
1138
1139 static void session_open(void)
1140 {
1141 }
1142
1143 static void session_update_time_string(char *str, __a_unused unsigned len)
1144 {
1145         printf("\r%s     ", str);
1146         fflush(stdout);
1147 }
1148 #endif /* HAVE_READLINE */
1149
1150 static void play_pre_select(struct sched *s, __a_unused void *context)
1151 {
1152         char state;
1153
1154         para_fd_set(STDIN_FILENO, &s->rfds, &s->max_fileno);
1155         state = get_playback_state();
1156         if (state == 'R' || state == 'F' || state == 'X')
1157                 return sched_min_delay(s);
1158         sched_request_barrier_or_min_delay(&pt->next_update, s);
1159 }
1160
1161 static unsigned get_time_string(char **result)
1162 {
1163         int seconds, length;
1164         char state = get_playback_state();
1165
1166         /* do not return anything if things are about to change */
1167         if (state != 'P' && state != 'U') {
1168                 *result = NULL;
1169                 return 0;
1170         }
1171         length = pt->seconds;
1172         if (length == 0)
1173                 return xasprintf(result, "0:00 [0:00] (0%%/0:00)");
1174         seconds = get_play_time();
1175         return xasprintf(result, "#%u: %d:%02d [%d:%02d] (%d%%/%d:%02d) %s",
1176                 pt->current_file,
1177                 seconds / 60,
1178                 seconds % 60,
1179                 (length - seconds) / 60,
1180                 (length - seconds) % 60,
1181                 length? (seconds * 100 + length / 2) / length : 0,
1182                 length / 60,
1183                 length % 60,
1184                 get_playlist_file(pt->current_file)
1185         );
1186 }
1187
1188 static int play_post_select(struct sched *s, __a_unused void *context)
1189 {
1190         int ret;
1191
1192         ret = eof_cleanup();
1193         if (ret < 0) {
1194                 pt->rq = CRT_TERM_RQ;
1195                 return 0;
1196         }
1197         ret = session_post_select(s);
1198         if (ret < 0)
1199                 goto out;
1200         if (!pt->wn.btrn && !pt->fn.btrn) {
1201                 char state = get_playback_state();
1202                 if (state == 'P' || state == 'R' || state == 'F') {
1203                         PARA_NOTICE_LOG("state: %c\n", state);
1204                         ret = load_next_file();
1205                         if (ret < 0) {
1206                                 PARA_ERROR_LOG("%s\n", para_strerror(-ret));
1207                                 pt->rq = CRT_TERM_RQ;
1208                                 ret = 1;
1209                                 goto out;
1210                         }
1211                         pt->next_update = *now;
1212                 }
1213         }
1214         if (tv_diff(now, &pt->next_update, NULL) >= 0) {
1215                 char *str;
1216                 unsigned len = get_time_string(&str);
1217                 struct timeval delay = {.tv_sec = 0, .tv_usec = 100 * 1000};
1218                 if (str && len > 0)
1219                         session_update_time_string(str, len);
1220                 free(str);
1221                 tv_add(now, &delay, &pt->next_update);
1222         }
1223         ret = 1;
1224 out:
1225         return ret;
1226 }
1227
1228 /**
1229  * The main function of para_play.
1230  *
1231  * \param argc Standard.
1232  * \param argv Standard.
1233  *
1234  * \return \p EXIT_FAILURE or \p EXIT_SUCCESS.
1235  */
1236 int main(int argc, char *argv[])
1237 {
1238         int ret;
1239         unsigned num_inputs;
1240
1241         /* needed this early to make help work */
1242         recv_init();
1243
1244         sched.default_timeout.tv_sec = 5;
1245         parse_config_or_die(argc, argv);
1246         AFH_RECV->init();
1247         session_open();
1248         num_inputs = lls_num_inputs(play_lpr);
1249         init_shuffle_map();
1250         pt->invalid = para_calloc(sizeof(*pt->invalid) * num_inputs);
1251         pt->rq = CRT_FILE_CHANGE;
1252         pt->current_file = num_inputs - 1;
1253         pt->playing = true;
1254         pt->task = task_register(&(struct task_info){
1255                 .name = "play",
1256                 .pre_select = play_pre_select,
1257                 .post_select = play_post_select,
1258                 .context = pt,
1259         }, &sched);
1260         ret = schedule(&sched);
1261         sched_shutdown(&sched);
1262         if (ret < 0)
1263                 PARA_ERROR_LOG("%s\n", para_strerror(-ret));
1264         return ret < 0? EXIT_FAILURE : EXIT_SUCCESS;
1265 }