convince/sway/config.c

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <unistd.h>
#include "readline.h"
#include "stringop.h"
#include "list.h"
#include "log.h"
#include "commands.h"
#include "config.h"
#include "layout.h"
#include "input_state.h"

struct sway_config *config = NULL;

static bool file_exists(const char *path) {
	return access(path, R_OK) != -1;
}

static void config_defaults(struct sway_config *config) {
	config->symbols = create_list();
	config->modes = create_list();
	config->workspace_outputs = create_list();
	config->output_configs = create_list();

	config->cmd_queue = create_list();

	config->current_mode = malloc(sizeof(struct sway_mode));
	config->current_mode->name = NULL;
	config->current_mode->bindings = create_list();
	list_add(config->modes, config->current_mode);

	config->floating_mod = 0;
	config->default_layout = L_NONE;
	config->default_orientation = L_NONE;
	// Flags
	config->focus_follows_mouse = true;
	config->mouse_warping = true;
	config->reloading = false;
	config->active = false;
	config->failed = false;
	config->auto_back_and_forth = false;

	config->gaps_inner = 0;
	config->gaps_outer = 0;
}

void free_mode(struct sway_mode *mode) {
	free(mode->name);
	int i;
	for (i = 0; i < mode->bindings->length; ++i) {
		struct sway_binding *bind = mode->bindings->items[i];
		list_free(bind->keys);
		free(bind->command);
		free(bind);
	}
	list_free(mode->bindings);
}

void free_config(struct sway_config *config) {
	int i;
	for (i = 0; i < config->symbols->length; ++i) {
		struct sway_variable *var = config->symbols->items[i];
		free(var->name);
		free(var->value);
		free(var);
	}
	list_free(config->symbols);

	for (i = 0; i < config->modes->length; ++i) {
		free_mode(config->modes->items[i]);
	}
	list_free(config->modes);

	for (i = 0; i < config->cmd_queue->length; ++i) {
		free(config->cmd_queue->items[i]);
	}
	list_free(config->cmd_queue);

	for (i = 0; i < config->workspace_outputs->length; ++i) {
		struct workspace_output *wso = config->workspace_outputs->items[i];
		free(wso->output);
		free(wso->workspace);
		free(wso);
	}
	list_free(config->workspace_outputs);

	for (i = 0; i < config->output_configs->length; ++i) {
		struct output_config *oc = config->output_configs->items[i];
		free(oc->name);
		free(oc);
	}
	list_free(config->output_configs);
	free(config);
}

static char *get_config_path(void) {
	char *config_path = NULL;
	char *paths[3] = {getenv("HOME"), getenv("XDG_CONFIG_HOME"), ""};
	int pathlen[3] = {0, 0, 0};
	int i;
#define home paths[0]
#define conf paths[1]
	// Get home and config directories
	home = home ? strdup(home) : NULL;
	if (conf) {
		conf = strdup(conf);
	} else if (home) {
		const char *def = "/.config";
		conf = malloc(strlen(home) + strlen(def) + 1);
		strcpy(conf, home);
		strcat(conf, def);
	} else {
		home = strdup("");
		conf = strdup("");
	}
	pathlen[0] = strlen(home);
	pathlen[1] = strlen(conf);
#undef home
#undef conf
	// Search for config file from search paths
	static const char *search_paths[] = {
		"/.sway/config", // Prepend with $home
		"/sway/config", // Prepend with $config
		"/etc/sway/config",
		"/.i3/config", // $home
		"/.i3/config", // $config
		"/etc/i3/config"
	};
	for (i = 0; i < (int)(sizeof(search_paths) / sizeof(char *)); ++i) {
		char *test = malloc(pathlen[i%3] + strlen(search_paths[i]) + 1);
		strcpy(test, paths[i%3]);
		strcat(test, search_paths[i]);
		sway_log(L_DEBUG, "Checking for config at %s", test);
		if (file_exists(test)) {
			config_path = test;
			goto cleanup;
		}
		free(test);
	}

	sway_log(L_DEBUG, "Trying to find config in XDG_CONFIG_DIRS");
	char *xdg_config_dirs = getenv("XDG_CONFIG_DIRS");
	if (xdg_config_dirs) {
		list_t *paths = split_string(xdg_config_dirs, ":");
		const char *name = "/sway/config";
		for (i = 0; i < paths->length; i++ ) {
			char *test = malloc(strlen(paths->items[i]) + strlen(name) + 1);
			strcpy(test, paths->items[i]);
			strcat(test, name);
			if (file_exists(test)) {
				config_path = test;
				break;
			}
			free(test);
		}
		free_flat_list(paths);
	}

cleanup:
	free(paths[0]);
	free(paths[1]);
	return config_path;
}

bool load_config(const char *file) {
	sway_log(L_INFO, "Loading config");

	input_init();

	char *path;
	if (file != NULL) {
		path = strdup(file);
	} else {
		path = get_config_path();
	}

	if (path == NULL) {
		sway_log(L_ERROR, "Unable to find a config file!");
		return false;
	}

	FILE *f = fopen(path, "r");
	if (!f) {
		fprintf(stderr, "Unable to open %s for reading", path);
		free(path);
		return false;
	}
	free(path);

	bool config_load_success;
	if (config) {
		config_load_success = read_config(f, true);
	} else {
		config_load_success = read_config(f, false);
	}
	fclose(f);

	return config_load_success;
}

bool read_config(FILE *file, bool is_active) {
	struct sway_config *temp_config = malloc(sizeof(struct sway_config));
	config_defaults(temp_config);
	if (is_active) {
		sway_log(L_DEBUG, "Performing configuration file reload");
		temp_config->reloading = true;
		temp_config->active = true;
	}

	bool success = true;

	int temp_depth = 0; // Temporary: skip all config sections with depth

	while (!feof(file)) {
		int _;
		char *line = read_line(file);
		line = strip_whitespace(line, &_);
		line = strip_comments(line);
		if (!line[0]) {
			goto _continue;
		}
		if (temp_depth && line[0] == '}') {
			temp_depth--;
			goto _continue;
		}

		// Any command which would require wlc to be initialized
		// should be queued for later execution
		list_t *args = split_string(line, " ");
		struct cmd_handler *handler;
		if ((handler = find_handler(args->items[0]))) {
			if (handler->config_type == CMD_KEYBIND) {
				sway_log(L_ERROR, "Invalid command during config ``%s''", line);
			} else if (handler->config_type == CMD_COMPOSITOR_READY && !is_active) {
				sway_log(L_DEBUG, "Deferring command ``%s''", line);
				char *cmd = malloc(strlen(line) + 1);
				strcpy(cmd, line);
				list_add(temp_config->cmd_queue, cmd);
			} else if (!temp_depth && !handle_command(temp_config, line)) {
				sway_log(L_DEBUG, "Config load failed for line ``%s''", line);
				success = false;
				temp_config->failed = true;
			}
		} else {
			sway_log(L_ERROR, "Invalid command ``%s''", line);
		}
		free_flat_list(args);

_continue:
		if (line && line[strlen(line) - 1] == '{') {
			temp_depth++;
		}
		free(line);
	}

	if (is_active) {
		temp_config->reloading = false;
		arrange_windows(&root_container, -1, -1);
	}
	if (config) {
		free_config(config);
	}
	config = temp_config;

	return success;
}

char *do_var_replacement(struct sway_config *config, char *str) {
	// TODO: Handle escaping $ and using $ in string literals
	int i;
	for (i = 0; str[i]; ++i) {
		if (str[i] == '$') {
			// Try for match (note: this could be faster)
			int j;
			for (j = 0; j < config->symbols->length; ++j) {
				struct sway_variable *var = config->symbols->items[j];
				if (strstr(str + i, var->name) == str + i) {
					// Match, do replacement
					char *new_string = malloc(
						strlen(str) -
						strlen(var->name) +
						strlen(var->value) + 1);
					strncpy(new_string, str, i);
					new_string[i] = 0;
					strcat(new_string, var->value);
					strcat(new_string, str + i + strlen(var->name));
					free(str);
					str = new_string;
				}
			}
		}
	}
	return str;
}