/* $OpenBSD$ */

/*
* Copyright (c) 2008 Nicholas Marriott <[email protected]>
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
* IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
* OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/

#include <sys/types.h>

#include <ctype.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "tmux.h"

struct client            *cfg_client;
int                       cfg_finished;
static char             **cfg_causes;
static u_int              cfg_ncauses;
static struct cmdq_item  *cfg_item;

int                       cfg_quiet = 1;
char                    **cfg_files;
u_int                     cfg_nfiles;

static enum cmd_retval
cfg_client_done(__unused struct cmdq_item *item, __unused void *data)
{
       if (!cfg_finished)
               return (CMD_RETURN_WAIT);
       return (CMD_RETURN_NORMAL);
}

static enum cmd_retval
cfg_done(__unused struct cmdq_item *item, __unused void *data)
{
       if (cfg_finished)
               return (CMD_RETURN_NORMAL);
       cfg_finished = 1;

       cfg_show_causes(NULL);

       if (cfg_item != NULL)
               cmdq_continue(cfg_item);

       status_prompt_load_history();

       return (CMD_RETURN_NORMAL);
}

void
start_cfg(void)
{
       struct client    *c;
       u_int             i;
       int               flags = 0;

       /*
        * Configuration files are loaded without a client, so commands are run
        * in the global queue with item->client NULL.
        *
        * However, we must block the initial client (but just the initial
        * client) so that its command runs after the configuration is loaded.
        * Because start_cfg() is called so early, we can be sure the client's
        * command queue is currently empty and our callback will be at the
        * front - we need to get in before MSG_COMMAND.
        */
       cfg_client = c = TAILQ_FIRST(&clients);
       if (c != NULL) {
               cfg_item = cmdq_get_callback(cfg_client_done, NULL);
               cmdq_append(c, cfg_item);
       }

       if (cfg_quiet)
               flags = CMD_PARSE_QUIET;
       for (i = 0; i < cfg_nfiles; i++)
               load_cfg(cfg_files[i], c, NULL, NULL, flags, NULL);

       cmdq_append(NULL, cmdq_get_callback(cfg_done, NULL));
}

int
load_cfg(const char *path, struct client *c, struct cmdq_item *item,
   struct cmd_find_state *current, int flags, struct cmdq_item **new_item)
{
       FILE                    *f;
       struct cmd_parse_input   pi;
       struct cmd_parse_result *pr;
       struct cmdq_item        *new_item0;
       struct cmdq_state       *state;

       if (new_item != NULL)
               *new_item = NULL;

       log_debug("loading %s", path);
       if ((f = fopen(path, "rb")) == NULL) {
               if (errno == ENOENT && (flags & CMD_PARSE_QUIET))
                       return (0);
               cfg_add_cause("%s: %s", path, strerror(errno));
               return (-1);
       }

       memset(&pi, 0, sizeof pi);
       pi.flags = flags;
       pi.file = path;
       pi.line = 1;
       pi.item = item;
       pi.c = c;

       pr = cmd_parse_from_file(f, &pi);
       fclose(f);
       if (pr->status == CMD_PARSE_ERROR) {
               cfg_add_cause("%s", pr->error);
               free(pr->error);
               return (-1);
       }
       if (flags & CMD_PARSE_PARSEONLY) {
               cmd_list_free(pr->cmdlist);
               return (0);
       }

       if (item != NULL)
               state = cmdq_copy_state(cmdq_get_state(item), current);
       else
               state = cmdq_new_state(NULL, NULL, 0);
       cmdq_add_format(state, "current_file", "%s", pi.file);

       new_item0 = cmdq_get_command(pr->cmdlist, state);
       if (item != NULL)
               new_item0 = cmdq_insert_after(item, new_item0);
       else
               new_item0 = cmdq_append(NULL, new_item0);
       cmd_list_free(pr->cmdlist);
       cmdq_free_state(state);

       if (new_item != NULL)
               *new_item = new_item0;
       return (0);
}

int
load_cfg_from_buffer(const void *buf, size_t len, const char *path,
   struct client *c, struct cmdq_item *item, struct cmd_find_state *current,
   int flags, struct cmdq_item **new_item)
{
       struct cmd_parse_input   pi;
       struct cmd_parse_result *pr;
       struct cmdq_item        *new_item0;
       struct cmdq_state       *state;

       if (new_item != NULL)
               *new_item = NULL;

       log_debug("loading %s", path);

       memset(&pi, 0, sizeof pi);
       pi.flags = flags;
       pi.file = path;
       pi.line = 1;
       pi.item = item;
       pi.c = c;

       pr = cmd_parse_from_buffer(buf, len, &pi);
       if (pr->status == CMD_PARSE_ERROR) {
               cfg_add_cause("%s", pr->error);
               free(pr->error);
               return (-1);
       }
       if (flags & CMD_PARSE_PARSEONLY) {
               cmd_list_free(pr->cmdlist);
               return (0);
       }

       if (item != NULL)
               state = cmdq_copy_state(cmdq_get_state(item), current);
       else
               state = cmdq_new_state(NULL, NULL, 0);
       cmdq_add_format(state, "current_file", "%s", pi.file);

       new_item0 = cmdq_get_command(pr->cmdlist, state);
       if (item != NULL)
               new_item0 = cmdq_insert_after(item, new_item0);
       else
               new_item0 = cmdq_append(NULL, new_item0);
       cmd_list_free(pr->cmdlist);
       cmdq_free_state(state);

       if (new_item != NULL)
               *new_item = new_item0;
       return (0);
}

void
cfg_add_cause(const char *fmt, ...)
{
       va_list  ap;
       char    *msg;

       va_start(ap, fmt);
       xvasprintf(&msg, fmt, ap);
       va_end(ap);

       cfg_ncauses++;
       cfg_causes = xreallocarray(cfg_causes, cfg_ncauses, sizeof *cfg_causes);
       cfg_causes[cfg_ncauses - 1] = msg;
}

void
cfg_print_causes(struct cmdq_item *item)
{
       u_int    i;

       for (i = 0; i < cfg_ncauses; i++) {
               cmdq_print(item, "%s", cfg_causes[i]);
               free(cfg_causes[i]);
       }

       free(cfg_causes);
       cfg_causes = NULL;
       cfg_ncauses = 0;
}

void
cfg_show_causes(struct session *s)
{
       struct client                   *c = TAILQ_FIRST(&clients);
       struct window_pane              *wp;
       struct window_mode_entry        *wme;
       u_int                            i;

       if (cfg_ncauses == 0)
               return;

       if (c != NULL && (c->flags & CLIENT_CONTROL)) {
               for (i = 0; i < cfg_ncauses; i++) {
                       control_write(c, "%%config-error %s", cfg_causes[i]);
                       free(cfg_causes[i]);
               }
               goto out;
       }

       if (s == NULL) {
               if (c != NULL && c->session != NULL)
                       s = c->session;
               else
                       s = RB_MIN(sessions, &sessions);
       }
       if (s == NULL || s->attached == 0) /* wait for an attached session */
               return;
       wp = s->curw->window->active;

       wme = TAILQ_FIRST(&wp->modes);
       if (wme == NULL || wme->mode != &window_view_mode)
               window_pane_set_mode(wp, NULL, &window_view_mode, NULL, NULL);
       for (i = 0; i < cfg_ncauses; i++) {
               window_copy_add(wp, 0, "%s", cfg_causes[i]);
               free(cfg_causes[i]);
       }

out:
       free(cfg_causes);
       cfg_causes = NULL;
       cfg_ncauses = 0;
}