cfg.c - tmux - Rivoreo Source Code Repositories

 /* $OpenBSD$ */

 /*
  * Copyright (c) 2008 Nicholas Marriott <nicholas.marriott@gmail.com>
  *
  * 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;
 static char		 *cfg_file;
 int			  cfg_finished;
 static char		**cfg_causes;
 static u_int		  cfg_ncauses;
 static struct cmdq_item	 *cfg_item;

 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;

 	if (!RB_EMPTY(&sessions))
 		cfg_show_causes(RB_MIN(sessions, &sessions));

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

 	status_prompt_load_history();

 	return (CMD_RETURN_NORMAL);
 }

 void
 set_cfg_file(const char *path)
 {
 	free(cfg_file);
 	cfg_file = xstrdup(path);
 }

 static char *
 expand_cfg_file(const char *path, const char *home)
 {
 	char			*expanded, *name;
 	const char		*end;
 	struct environ_entry	*value;

 	if (strncmp(path, "~/", 2) == 0) {
 		if (home == NULL)
 			return (NULL);
 		xasprintf(&expanded, "%s%s", home, path + 1);
 		return (expanded);
 	}

 	if (*path == '$') {
 		end = strchr(path, '/');
 		if (end == NULL)
 			name = xstrdup(path + 1);
 		else
 			name = xstrndup(path + 1, end - path - 1);
 		value = environ_find(global_environ, name);
 		free(name);
 		if (value == NULL)
 			return (NULL);
 		if (end == NULL)
 			end = "";
 		xasprintf(&expanded, "%s%s", value->value, end);
 		return (expanded);
 	}

 	return (xstrdup(path));
 }

 void
 start_cfg(void)
 {
 	const char	*home = find_home();
 	struct client	*c;
 	char		*path, *copy, *next, *expanded;

 	/*
 	 * 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_file == NULL) {
 		path = copy = xstrdup(TMUX_CONF);
 		while ((next = strsep(&path, ":")) != NULL) {
 			expanded = expand_cfg_file(next, home);
 			if (expanded == NULL) {
 				log_debug("couldn't expand %s", next);
 				continue;
 			}
 			log_debug("expanded %s to %s", next, expanded);
 			load_cfg(expanded, c, NULL, CMD_PARSE_QUIET, NULL);
 			free(expanded);
 		}
 		free(copy);
 	} else
 		load_cfg(cfg_file, c, NULL, 0, NULL);

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

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

 	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_EMPTY)
 		return (0);
 	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);
 	}

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

 	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, int flags,
     struct cmdq_item **new_item)
 {
 	struct cmd_parse_input	 pi;
 	struct cmd_parse_result	*pr;
 	struct cmdq_item	*new_item0;

 	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_EMPTY)
 		return (0);
 	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);
 	}

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

 	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 window_pane		*wp;
 	struct window_mode_entry	*wme;
 	u_int				 i;

 	if (s == NULL || cfg_ncauses == 0)
 		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, "%s", cfg_causes[i]);
 		free(cfg_causes[i]);
 	}

 	free(cfg_causes);
 	cfg_causes = NULL;
 	cfg_ncauses = 0;
 }
	/* $OpenBSD$ */

	/*
	* Copyright (c) 2008 Nicholas Marriott <nicholas.marriott@gmail.com>
	*
	* 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;
	static char *cfg_file;
	int cfg_finished;
	static char **cfg_causes;
	static u_int cfg_ncauses;
	static struct cmdq_item *cfg_item;

	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;

	if (!RB_EMPTY(&sessions))
	cfg_show_causes(RB_MIN(sessions, &sessions));

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

	status_prompt_load_history();

	return (CMD_RETURN_NORMAL);
	}

	void
	set_cfg_file(const char *path)
	{
	free(cfg_file);
	cfg_file = xstrdup(path);
	}

	static char *
	expand_cfg_file(const char path, const char home)
	{
	char expanded, name;
	const char *end;
	struct environ_entry *value;

	if (strncmp(path, "~/", 2) == 0) {
	if (home == NULL)
	return (NULL);
	xasprintf(&expanded, "%s%s", home, path + 1);
	return (expanded);
	}

	if (*path == '$') {
	end = strchr(path, '/');
	if (end == NULL)
	name = xstrdup(path + 1);
	else
	name = xstrndup(path + 1, end - path - 1);
	value = environ_find(global_environ, name);
	free(name);
	if (value == NULL)
	return (NULL);
	if (end == NULL)
	end = "";
	xasprintf(&expanded, "%s%s", value->value, end);
	return (expanded);
	}

	return (xstrdup(path));
	}

	void
	start_cfg(void)
	{
	const char *home = find_home();
	struct client *c;
	char path, copy, next, expanded;

	/*
	* 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_file == NULL) {
	path = copy = xstrdup(TMUX_CONF);
	while ((next = strsep(&path, ":")) != NULL) {
	expanded = expand_cfg_file(next, home);
	if (expanded == NULL) {
	log_debug("couldn't expand %s", next);
	continue;
	}
	log_debug("expanded %s to %s", next, expanded);
	load_cfg(expanded, c, NULL, CMD_PARSE_QUIET, NULL);
	free(expanded);
	}
	free(copy);
	} else
	load_cfg(cfg_file, c, NULL, 0, NULL);

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

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

	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_EMPTY)
	return (0);
	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);
	}

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

	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, int flags,
	struct cmdq_item **new_item)
	{
	struct cmd_parse_input pi;
	struct cmd_parse_result *pr;
	struct cmdq_item *new_item0;

	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_EMPTY)
	return (0);
	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);
	}

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

	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 window_pane *wp;
	struct window_mode_entry *wme;
	u_int i;

	if (s == NULL \|\| cfg_ncauses == 0)
	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, "%s", cfg_causes[i]);
	free(cfg_causes[i]);
	}

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