Implement swaylock configuration file parsing

1 files changed, 184 insertions, 57 deletions

diff --git a/swaylock/main.c b/swaylock/main.c
index faebc757..3d57eb75 100644
--- a/swaylock/main.c
+++ b/swaylock/main.c
@@ -21,6 +21,7 @@
 #include "pool-buffer.h"
 #include "cairo.h"
 #include "log.h"
+#include "readline.h"
 #include "stringop.h"
 #include "util.h"
 #include "wlr-input-inhibitor-unstable-v1-client-protocol.h"
@@ -412,15 +413,14 @@ static void set_default_colors(struct swaylock_colors *colors) {
 	};
 }
 
-static struct swaylock_state state;
-
-int main(int argc, char **argv) {
-	enum line_mode {
-		LM_LINE,
-		LM_INSIDE,
-		LM_RING,
-	};
+enum line_mode {
+	LM_LINE,
+	LM_INSIDE,
+	LM_RING,
+};
 
+static int parse_options(int argc, char **argv, struct swaylock_state *state,
+		enum line_mode *line_mode) {
 	enum long_option_codes {
 		LO_BS_HL_COLOR = 256,
 		LO_FONT,
@@ -447,6 +447,7 @@ int main(int argc, char **argv) {
 	};
 
 	static struct option long_options[] = {
+		{"config", required_argument, NULL, 'C'},
 		{"color", required_argument, NULL, 'c'},
 		{"ignore-empty-password", no_argument, NULL, 'e'},
 		{"daemonize", no_argument, NULL, 'f'},
@@ -487,6 +488,8 @@ int main(int argc, char **argv) {
 	const char usage[] =
 		"Usage: swaylock [options...]\n"
 		"\n"
+		"  -C, --config <config_file>     "
+			"Path to the config file.\n"
 		"  -c, --color <color>            "
 			"Turn the screen into the given color instead of white.\n"
 		"  -e, --ignore-empty-password    "
@@ -559,58 +562,48 @@ int main(int argc, char **argv) {
 		"\n"
 		"All <color> options are of the form <rrggbb[aa]>.\n";
 
-	enum line_mode line_mode = LM_LINE;
-	state.args = (struct swaylock_args){
-		.mode = BACKGROUND_MODE_SOLID_COLOR,
-		.font = strdup("sans-serif"),
-		.radius = 50,
-		.thickness = 10,
-		.ignore_empty = false,
-		.show_indicator = true,
-	};
-	wl_list_init(&state.images);
-	set_default_colors(&state.args.colors);
-
-	wlr_log_init(WLR_DEBUG, NULL);
-
 	int c;
+	optind = 1;
 	while (1) {
 		int opt_idx = 0;
-		c = getopt_long(argc, argv, "c:efhi:nrs:tuv", long_options, &opt_idx);
+		c = getopt_long(argc, argv, "c:efhi:nrs:tuvC:", long_options, &opt_idx);
 		if (c == -1) {
 			break;
 		}
 		switch (c) {
+		case 'C':
+			// Config file. This will have already been handled so just ignore.
+			break;
 		case 'c':
-			state.args.colors.background = parse_color(optarg);
-			state.args.mode = BACKGROUND_MODE_SOLID_COLOR;
+			state->args.colors.background = parse_color(optarg);
+			state->args.mode = BACKGROUND_MODE_SOLID_COLOR;
 			break;
 		case 'e':
-			state.args.ignore_empty = true;
+			state->args.ignore_empty = true;
 			break;
 		case 'f':
-			state.args.daemonize = true;
+			state->args.daemonize = true;
 			break;
 		case 'i':
-			load_image(optarg, &state);
+			load_image(optarg, state);
 			break;
 		case 'n':
-			line_mode = LM_INSIDE;
+			*line_mode = LM_INSIDE;
 			break;
 		case 'r':
-			line_mode = LM_RING;
+			*line_mode = LM_RING;
 			break;
 		case 's':
-			state.args.mode = parse_background_mode(optarg);
-			if (state.args.mode == BACKGROUND_MODE_INVALID) {
+			state->args.mode = parse_background_mode(optarg);
+			if (state->args.mode == BACKGROUND_MODE_INVALID) {
 				return 1;
 			}
 			break;
 		case 't':
-			state.args.mode = BACKGROUND_MODE_TILE;
+			state->args.mode = BACKGROUND_MODE_TILE;
 			break;
 		case 'u':
-			state.args.show_indicator = false;
+			state->args.show_indicator = false;
 			break;
 		case 'v':
 #if defined SWAY_GIT_VERSION && defined SWAY_GIT_BRANCH && defined SWAY_VERSION_DATE
@@ -621,71 +614,71 @@ int main(int argc, char **argv) {
 #endif
 			return 0;
 		case LO_BS_HL_COLOR:
-			state.args.colors.bs_highlight = parse_color(optarg);
+			state->args.colors.bs_highlight = parse_color(optarg);
 			break;
 		case LO_FONT:
-			free(state.args.font);
-			state.args.font = strdup(optarg);
+			free(state->args.font);
+			state->args.font = strdup(optarg);
 			break;
 		case LO_IND_RADIUS:
-			state.args.radius = strtol(optarg, NULL, 0);
+			state->args.radius = strtol(optarg, NULL, 0);
 			break;
 		case LO_IND_THICKNESS:
-			state.args.thickness = strtol(optarg, NULL, 0);
+			state->args.thickness = strtol(optarg, NULL, 0);
 			break;
 		case LO_INSIDE_COLOR:
-			state.args.colors.inside.input = parse_color(optarg);
+			state->args.colors.inside.input = parse_color(optarg);
 			break;
 		case LO_INSIDE_CLEAR_COLOR:
-			state.args.colors.inside.cleared = parse_color(optarg);
+			state->args.colors.inside.cleared = parse_color(optarg);
 			break;
 		case LO_INSIDE_VER_COLOR:
-			state.args.colors.inside.verifying = parse_color(optarg);
+			state->args.colors.inside.verifying = parse_color(optarg);
 			break;
 		case LO_INSIDE_WRONG_COLOR:
-			state.args.colors.inside.wrong = parse_color(optarg);
+			state->args.colors.inside.wrong = parse_color(optarg);
 			break;
 		case LO_KEY_HL_COLOR:
-			state.args.colors.key_highlight = parse_color(optarg);
+			state->args.colors.key_highlight = parse_color(optarg);
 			break;
 		case LO_LINE_COLOR:
-			state.args.colors.line.input = parse_color(optarg);
+			state->args.colors.line.input = parse_color(optarg);
 			break;
 		case LO_LINE_CLEAR_COLOR:
-			state.args.colors.line.cleared = parse_color(optarg);
+			state->args.colors.line.cleared = parse_color(optarg);
 			break;
 		case LO_LINE_VER_COLOR:
-			state.args.colors.line.verifying = parse_color(optarg);
+			state->args.colors.line.verifying = parse_color(optarg);
 			break;
 		case LO_LINE_WRONG_COLOR:
-			state.args.colors.line.wrong = parse_color(optarg);
+			state->args.colors.line.wrong = parse_color(optarg);
 			break;
 		case LO_RING_COLOR:
-			state.args.colors.ring.input = parse_color(optarg);
+			state->args.colors.ring.input = parse_color(optarg);
 			break;
 		case LO_RING_CLEAR_COLOR:
-			state.args.colors.ring.cleared = parse_color(optarg);
+			state->args.colors.ring.cleared = parse_color(optarg);
 			break;
 		case LO_RING_VER_COLOR:
-			state.args.colors.ring.verifying = parse_color(optarg);
+			state->args.colors.ring.verifying = parse_color(optarg);
 			break;
 		case LO_RING_WRONG_COLOR:
-			state.args.colors.ring.wrong = parse_color(optarg);
+			state->args.colors.ring.wrong = parse_color(optarg);
 			break;
 		case LO_SEP_COLOR:
-			state.args.colors.separator = parse_color(optarg);
+			state->args.colors.separator = parse_color(optarg);
 			break;
 		case LO_TEXT_COLOR:
-			state.args.colors.text.input = parse_color(optarg);
+			state->args.colors.text.input = parse_color(optarg);
 			break;
 		case LO_TEXT_CLEAR_COLOR:
-			state.args.colors.text.cleared = parse_color(optarg);
+			state->args.colors.text.cleared = parse_color(optarg);
 			break;
 		case LO_TEXT_VER_COLOR:
-			state.args.colors.text.verifying = parse_color(optarg);
+			state->args.colors.text.verifying = parse_color(optarg);
 			break;
 		case LO_TEXT_WRONG_COLOR:
-			state.args.colors.text.wrong = parse_color(optarg);
+			state->args.colors.text.wrong = parse_color(optarg);
 			break;
 		default:
 			fprintf(stderr, "%s", usage);
@@ -693,6 +686,140 @@ int main(int argc, char **argv) {
 		}
 	}
 
+	return 0;
+}
+
+static bool file_exists(const char *path) {
+	return path && access(path, R_OK) != -1;
+}
+
+static char *get_config_path(void) {
+	static const char *config_paths[] = {
+		"$HOME/.swaylock/config",
+		"$XDG_CONFIG_HOME/swaylock/config",
+		SYSCONFDIR "/swaylock/config",
+	};
+
+	if (!getenv("XDG_CONFIG_HOME")) {
+		char *home = getenv("HOME");
+		char *config_home = malloc(strlen(home) + strlen("/.config") + 1);
+		if (!config_home) {
+			wlr_log(WLR_ERROR, "Unable to allocate $HOME/.config");
+		} else {
+			strcpy(config_home, home);
+			strcat(config_home, "/.config");
+			setenv("XDG_CONFIG_HOME", config_home, 1);
+			wlr_log(WLR_DEBUG, "Set XDG_CONFIG_HOME to %s", config_home);
+			free(config_home);
+		}
+	}
+
+	wordexp_t p;
+	char *path;
+	for (int i = 0; i < (int)(sizeof(config_paths) / sizeof(char *)); ++i) {
+		if (wordexp(config_paths[i], &p, 0) == 0) {
+			path = strdup(p.we_wordv[0]);
+			wordfree(&p);
+			if (file_exists(path)) {
+				return path;
+			}
+			free(path);
+		}
+	}
+
+	return NULL;
+}
+
+static int load_config(char *path, struct swaylock_state *state,
+		enum line_mode *line_mode) {
+	FILE *config = fopen(path, "r");
+	if (!config) {
+		wlr_log(WLR_ERROR, "Failed to read config. Running without it.");
+		return 0;
+	}
+	char *line;
+	int line_number = 0;
+	while (!feof(config)) {
+		line = read_line(config);
+		if (!line) {
+			continue;
+		}
+
+		line_number++;
+		if (line[0] == '#') {
+			free(line);
+			continue;
+		}
+		if (strlen(line) == 0) {
+			free(line);
+			continue;
+		}
+
+		wlr_log(WLR_DEBUG, "Config Line #%d: %s", line_number, line);
+		char flag[strlen(line) + 3];
+		sprintf(flag, "--%s", line);
+		char *argv[] = {"swaylock", flag};
+		int result = parse_options(2, argv, state, line_mode);
+		if (result != 0) {
+			free(line);
+			fclose(config);
+			return result;
+		}
+		free(line);
+	}
+	fclose(config);
+	return 0;
+}
+
+static struct swaylock_state state;
+
+int main(int argc, char **argv) {
+	enum line_mode line_mode = LM_LINE;
+	state.args = (struct swaylock_args){
+		.mode = BACKGROUND_MODE_SOLID_COLOR,
+		.font = strdup("sans-serif"),
+		.radius = 50,
+		.thickness = 10,
+		.ignore_empty = false,
+		.show_indicator = true,
+	};
+	wl_list_init(&state.images);
+	set_default_colors(&state.args.colors);
+
+	wlr_log_init(WLR_DEBUG, NULL);
+
+	char *config_path = NULL;
+	for (int i = 0; i < argc; i++) {
+		if (strcmp(argv[i], "-C") == 0 || strcmp(argv[i], "--config") == 0) {
+			if (i + 1 == argc) {
+				wlr_log(WLR_ERROR, "Config file path is missing");
+				return 1;
+			}
+			config_path = strdup(argv[i + 1]);
+			break;
+		}
+	}
+	if (!config_path) {
+		config_path = get_config_path();
+	}
+
+	if (config_path) {
+		wlr_log(WLR_DEBUG, "Found config at %s", config_path);
+		int config_status = load_config(config_path, &state, &line_mode);
+		free(config_path);
+		if (config_status != 0) {
+			return config_status;
+		}
+	}
+
+	if (argc > 1) {
+		wlr_log(WLR_DEBUG, "Parsing CLI Args");
+		int result = parse_options(argc, argv, &state, &line_mode);
+		if (result != 0) {
+			return result;
+		}
+	}
+
 	if (line_mode == LM_INSIDE) {
 		state.args.colors.line = state.args.colors.inside;
 	} else if (line_mode == LM_RING) {