Added in config file loading from multiple sources

1 files changed, 77 insertions, 9 deletions

diff --git a/sway/config.c b/sway/config.c
index d96d23fc..eb08641a 100644
--- a/sway/config.c
+++ b/sway/config.c
@@ -1,6 +1,7 @@
 #include <stdio.h>
 #include <stdbool.h>
 #include <stdlib.h>
+#include <unistd.h>
 #include "readline.h"
 #include "stringop.h"
 #include "list.h"
@@ -10,21 +11,88 @@
 
 struct sway_config *config;
 
-bool load_config() {
-        sway_log(L_INFO, "Loading config");
-        // TODO: Allow use of more config file locations
-        const char *name = "/.sway/config";
+static bool exists(const char *path) {
+        return access(path, R_OK) != -1;
+}
+
+static char* get_config_path() {
+        char *name = "/.sway/config";
         const char *home = getenv("HOME");
+
+        sway_log(L_DEBUG, "Trying to find config in ~/.sway/config");
+        // Check home dir
         char *temp = malloc(strlen(home) + strlen(name) + 1);
         strcpy(temp, home);
         strcat(temp, name);
-        FILE *f = fopen(temp, "r");
+        if (exists(temp)) {
+                return temp;
+        }
+
+        // Check XDG_CONFIG_HOME with fallback to ~/.config/
+        sway_log(L_DEBUG, "Trying to find config in XDG_CONFIG_HOME/sway/config");
+        char *xdg_config_home = getenv("XDG_CONFIG_HOME");
+        if (xdg_config_home == NULL) {
+                sway_log(L_DEBUG, "Falling back to ~/.config/sway/config");
+                name = "/.config/sway/config";
+                temp = malloc(strlen(home) + strlen(name) + 1);
+                strcpy(temp, home);
+                strcat(temp, name);
+        } else {
+                name = "/sway/config";
+                temp = malloc(strlen(xdg_config_home) + strlen(name) + 1);
+                strcpy(temp, home);
+                strcat(temp, name);
+        }
+        if (exists(temp)) {
+                return temp;
+        }
+
+        // Check /etc/
+        sway_log(L_DEBUG, "Trying to find config in /etc/sway/config");
+        strcpy(temp, "/etc/sway/config");
+        if (exists(temp)) {
+                return temp;
+        }
+
+        // Check XDG_CONFIG_DIRS
+        sway_log(L_DEBUG, "Trying to find config in XDG_CONFIG_DIRS");
+        char *xdg_config_dirs = getenv("XDG_CONFIG_DIRS");
+        if (xdg_config_dirs != NULL) {
+                list_t *paths = split_string(xdg_config_dirs, ":");
+                name = "/sway/config";
+                int i;
+                for (i = 0; i < paths->length; i++ ) {
+                        temp = malloc(strlen(paths->items[i]) + strlen(name) + 1);
+                        strcpy(temp, paths->items[i]);
+                        strcat(temp, name);
+                        if (exists(temp)) {
+                                free_flat_list(paths);
+                                return temp;
+                        }
+                }
+                free_flat_list(paths);
+        }
+
+        return NULL;
+}
+
+bool load_config() {
+        sway_log(L_INFO, "Loading config");
+
+        char *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", temp);
-                free(temp);
+                fprintf(stderr, "Unable to open %s for reading", path);
+                free(path);
                 return false;
         }
-        free(temp);
+        free(path);
 
         bool config_load_success;
         if (config) {
@@ -95,7 +163,7 @@ bool read_config(FILE *file, bool is_active) {
                         sway_log(L_DEBUG, "Config load failed for line %s", line);
                         success = false;
                         temp_config->failed = true;
-                } 
+                }
                 list_free(args);
 
 _continue: