diff options
Diffstat (limited to 'swaylock/main.c')
| -rw-r--r-- | swaylock/main.c | 244 |
1 files changed, 187 insertions, 57 deletions
diff --git a/swaylock/main.c b/swaylock/main.c index faebc757..ae5b86b9 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,143 @@ 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 (size_t i = 0; i < 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; + static struct option long_options[] = { + {"config", required_argument, NULL, 'C'}, + {0, 0, 0, 0}, + }; + while (1) { + int c = getopt_long(argc, argv, "C:", long_options, NULL); + if (c == -1) { + break; + } else if (c == 'C') { + config_path = strdup(optarg); + 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) { |