Add the files

1 files changed, 340 insertions, 0 deletions

diff --git a/src/detection/wmtheme/wmtheme_linux.c b/src/detection/wmtheme/wmtheme_linux.c
new file mode 100644
index 0000000..e912302
--- /dev/null
+++ b/src/detection/wmtheme/wmtheme_linux.c
@@ -0,0 +1,340 @@
+#include "wmtheme.h"
+#include "common/io.h"
+#include "common/properties.h"
+#include "common/parsing.h"
+#include "common/settings.h"
+#include "common/strutil.h"
+#include "common/mallocHelper.h"
+#include "detection/gtk_qt/gtk_qt.h"
+#include "detection/displayserver/displayserver.h"
+
+static bool detectWMThemeFromConfigFile(const char* configFile, const char* themeRegex, const char* defaultValue, FFstrbuf* themeOrError) {
+    if (!ffParsePropFileConfig(configFile, themeRegex, themeOrError)) {
+        ffStrbufAppendF(themeOrError, "Config file %s doesn't exist", configFile);
+        return false;
+    }
+
+    if (themeOrError->length == 0) {
+        if (defaultValue == NULL) {
+            ffStrbufAppendF(themeOrError, "Couldn't find WM theme in %s", configFile);
+            return false;
+        }
+
+        ffStrbufAppendS(themeOrError, defaultValue);
+        return true;
+    }
+
+    // Remove Plasma-generated prefixes
+    uint32_t idx = 0;
+
+    idx = ffStrbufFirstIndexS(themeOrError, "qml_");
+    if (idx != themeOrError->length) {
+        ffStrbufSubstrAfter(themeOrError, idx + 3);
+    }
+
+    idx = ffStrbufFirstIndexS(themeOrError, "svg__");
+    if (idx != themeOrError->length) {
+        ffStrbufSubstrAfter(themeOrError, idx + 4);
+    }
+
+    return true;
+}
+
+static bool detectWMThemeFromSettings(const char* dconfKey, const char* gsettingsSchemaName, const char* gsettingsPath, const char* gsettingsKey, FFstrbuf* themeOrError) {
+    const char* theme = ffSettingsGetGnome(dconfKey, gsettingsSchemaName, gsettingsPath, gsettingsKey, FF_VARIANT_TYPE_STRING).strValue;
+
+    if (!ffStrSet(theme)) {
+        ffStrbufAppendS(themeOrError, "Couldn't find WM theme in DConf or GSettings");
+        return false;
+    }
+
+    ffStrbufAppendS(themeOrError, theme);
+    return true;
+}
+
+static bool detectGTKThemeAsWMTheme(FFstrbuf* themeOrError) {
+    const FFGTKResult* gtk = ffDetectGTK4();
+    if (gtk->theme.length > 0) {
+        goto ok;
+    }
+
+    gtk = ffDetectGTK3();
+    if (gtk->theme.length > 0) {
+        goto ok;
+    }
+
+    gtk = ffDetectGTK2();
+    if (gtk->theme.length > 0) {
+        goto ok;
+    }
+
+    ffStrbufAppendS(themeOrError, "Couldn't detect GTK4/3/2 theme");
+    return false;
+
+ok:
+    ffStrbufAppend(themeOrError, &gtk->theme);
+    return true;
+}
+
+static bool detectMutter(FFstrbuf* themeOrError) {
+    const char* theme = ffSettingsGetGnome("/org/gnome/shell/extensions/user-theme/name", "org.gnome.shell.extensions.user-theme", NULL, "name", FF_VARIANT_TYPE_STRING).strValue;
+    if (ffStrSet(theme)) {
+        ffStrbufAppendS(themeOrError, theme);
+        return true;
+    }
+
+    return detectGTKThemeAsWMTheme(themeOrError);
+}
+
+static bool detectMuffin(FFstrbuf* themeOrError) {
+    FF_AUTO_FREE const char* name = ffSettingsGetGnome("/org/cinnamon/theme/name", "org.cinnamon.theme", NULL, "name", FF_VARIANT_TYPE_STRING).strValue;
+    FF_AUTO_FREE const char* theme = ffSettingsGetGnome("/org/cinnamon/desktop/wm/preferences/theme", "org.cinnamon.desktop.wm.preferences", NULL, "theme", FF_VARIANT_TYPE_STRING).strValue;
+
+    if (name == NULL && theme == NULL) {
+        ffStrbufAppendS(themeOrError, "Couldn't find muffin theme in GSettings / DConf");
+        return false;
+    }
+
+    if (name == NULL) {
+        ffStrbufAppendS(themeOrError, theme);
+        return true;
+    }
+
+    if (theme == NULL) {
+        ffStrbufAppendS(themeOrError, name);
+        return true;
+    }
+
+    ffStrbufAppendF(themeOrError, "%s (%s)", name, theme);
+    return true;
+}
+
+static bool detectXFWM4(FFstrbuf* themeOrError) {
+    const char* theme = ffSettingsGetXFConf("xfwm4", "/general/theme", FF_VARIANT_TYPE_STRING).strValue;
+
+    if (theme == NULL) {
+        ffStrbufAppendS(themeOrError, "Couldn't find xfwm4::/general/theme in XFConf");
+        return false;
+    }
+
+    ffStrbufAppendS(themeOrError, theme);
+    return true;
+}
+
+static bool detectOpenbox(const FFstrbuf* dePrettyName, FFstrbuf* themeOrError) {
+    FF_STRBUF_AUTO_DESTROY absolutePath = ffStrbufCreateA(64);
+    const char* configFileSubpath = "openbox/rc.xml";
+    if (ffStrbufIgnCaseEqualS(dePrettyName, "LXQt")) {
+        configFileSubpath = "openbox/lxqt-rc.xml";
+    } else if (ffStrbufIgnCaseEqualS(dePrettyName, "LXDE")) {
+        configFileSubpath = "openbox/lxde-rc.xml";
+    }
+
+    if (!ffSearchUserConfigFile(&instance.state.platform.configDirs, configFileSubpath, &absolutePath)) {
+        ffStrbufAppendF(themeOrError, "Couldn't find config file \"%s\"", configFileSubpath);
+        return false;
+    }
+
+    FF_STRBUF_AUTO_DESTROY content = ffStrbufCreate();
+    if (!ffReadFileBuffer(absolutePath.chars, &content)) {
+        ffStrbufAppendF(themeOrError, "Couldn't read \"%s\"", absolutePath.chars);
+        return false;
+    }
+
+    const char* themeStart = strstr(content.chars, "<theme>");
+    if (themeStart == NULL) {
+        goto theme_not_found;
+    }
+
+    const char* themeEnd = strstr(themeStart, "</theme>");
+    if (__builtin_expect(themeEnd == NULL, false)) { // very rare case
+        goto theme_not_found;
+    }
+
+    const char* nameStart = strstr(themeStart, "<name>");
+    if (nameStart == NULL) {
+        goto name_not_found;
+    }
+
+    const char* nameEnd = strstr(nameStart, "</name>");
+    if (nameEnd == NULL || nameEnd > themeEnd) { // (nameEnd > themeEnd) means name is not a theme's child
+        goto name_not_found;
+    }
+
+    nameStart += strlen("<name>");
+    ffStrbufAppendNS(themeOrError, (uint32_t) (nameEnd - nameStart), nameStart);
+    ffStrbufTrim(themeOrError, ' ');
+
+    if (themeOrError->length == 0) {
+        goto name_not_found;
+    }
+
+    return true;
+
+theme_not_found:
+    ffStrbufAppendF(themeOrError, "Couldn't find theme node in \"%s\"", absolutePath.chars);
+    return false;
+
+name_not_found:
+    ffStrbufAppendF(themeOrError, "Couldn't find theme name in \"%s\"", absolutePath.chars);
+    return false;
+}
+
+static bool detectCosmicComp(FFstrbuf* themeOrError) {
+    FF_STRBUF_AUTO_DESTROY path = ffStrbufCreateCopy(FF_LIST_FIRST(FFstrbuf, instance.state.platform.configDirs));
+    ffStrbufAppendS(&path, "cosmic/");
+    uint32_t basePathLength = path.length;
+
+    const char* variant;
+    {
+        char isDarkC;
+        ffStrbufAppendS(&path, "com.system76.CosmicTheme.Mode/v1/is_dark");
+        if (ffReadFileData(path.chars, 1, &isDarkC) == 1) {
+            variant = isDarkC == 't' || isDarkC == '1' ? "Dark" : "Light";
+        } else {
+            ffStrbufAppendF(themeOrError, "Couldn't read cosmic theme mode from %s", path.chars);
+            return false;
+        }
+        ffStrbufSubstrBefore(&path, basePathLength);
+    }
+
+    FF_STRBUF_AUTO_DESTROY name = ffStrbufCreate();
+    ffStrbufAppendF(&path, "com.system76.CosmicTheme.%s/v1/name", variant);
+    if (ffReadFileBuffer(path.chars, &name)) {
+        ffStrbufTrimSpace(&name);
+        ffStrbufTrim(&name, '"');
+    }
+    ffStrbufSubstrBefore(&path, basePathLength);
+
+    if (name.length > 0) {
+        ffStrbufAppend(themeOrError, &name);
+    } else {
+        ffStrbufAppendS(themeOrError, variant);
+    }
+
+    FF_STRBUF_AUTO_DESTROY accent = ffStrbufCreate();
+    ffStrbufAppendF(&path, "com.system76.CosmicTheme.%s/v1/accent", variant);
+    if (ffReadFileBuffer(path.chars, &accent)) {
+        const char* baseStart = strstr(accent.chars, "base:");
+        if (baseStart != NULL) {
+            const char* contentStart = strchr(baseStart, '(');
+            if (contentStart != NULL) {
+                int depth = 1;
+                const char* contentEnd = contentStart + 1;
+                while (*contentEnd != '\0' && depth > 0) {
+                    if (*contentEnd == '(') {
+                        ++depth;
+                    } else if (*contentEnd == ')') {
+                        --depth;
+                    }
+                    ++contentEnd;
+                }
+
+                if (depth == 0) {
+                    double red = 0.0, green = 0.0, blue = 0.0, alpha = 1.0;
+                    bool ok = true;
+                    for (uint32_t i = 0; i < 4; ++i) {
+                        const char* key =
+                            i == 0 ? "red:" : (i == 1 ? "green:" : (i == 2 ? "blue:" : "alpha:"));
+                        const char* componentStart = strstr(contentStart, key);
+                        if (componentStart == NULL || componentStart >= contentEnd) {
+                            ok = false;
+                            break;
+                        }
+
+                        componentStart += strlen(key);
+                        char* componentEnd = NULL;
+                        double value = strtod(componentStart, &componentEnd);
+                        if (componentEnd == componentStart) {
+                            ok = false;
+                            break;
+                        }
+
+                        if (i == 0) {
+                            red = value;
+                        } else if (i == 1) {
+                            green = value;
+                        } else if (i == 2) {
+                            blue = value;
+                        } else {
+                            alpha = value;
+                        }
+                    }
+
+                    if (ok) {
+                        uint32_t r = red <= 0.0 ? 0 : (red >= 1.0 ? 255 : (uint32_t) (red * 255.0 + 0.5));
+                        uint32_t g = green <= 0.0 ? 0 : (green >= 1.0 ? 255 : (uint32_t) (green * 255.0 + 0.5));
+                        uint32_t b = blue <= 0.0 ? 0 : (blue >= 1.0 ? 255 : (uint32_t) (blue * 255.0 + 0.5));
+                        uint32_t a = alpha <= 0.0 ? 0 : (alpha >= 1.0 ? 255 : (uint32_t) (alpha * 255.0 + 0.5));
+                        uint32_t rgb = (r << 16) | (g << 8) | b;
+                        if (a == 255) {
+                            ffStrbufAppendF(themeOrError, " - #%06X", rgb);
+                        } else {
+                            ffStrbufAppendF(themeOrError, " - #%06X%02X", rgb, a);
+                        }
+                    }
+                }
+            }
+        }
+    }
+    ffStrbufSubstrBefore(&path, basePathLength);
+
+    if (name.length > 0) {
+        ffStrbufAppendF(themeOrError, " (%s)", variant);
+    }
+
+    return true;
+}
+
+bool ffDetectWmTheme(FFstrbuf* themeOrError) {
+    const FFDisplayServerResult* wm = ffConnectDisplayServer();
+
+    if (wm->wmPrettyName.length == 0) {
+        ffStrbufAppendS(themeOrError, "WM Theme needs successful WM detection");
+        return false;
+    }
+
+    if (ffStrbufIgnCaseEqualS(&wm->wmPrettyName, FF_WM_PRETTY_KWIN)) {
+        return detectWMThemeFromConfigFile("kwinrc", "theme =", "Breeze", themeOrError);
+    }
+
+    if (ffStrbufIgnCaseEqualS(&wm->wmPrettyName, FF_WM_PRETTY_COSMIC_COMP)) {
+        return detectCosmicComp(themeOrError);
+    }
+
+    if (
+        ffStrbufIgnCaseEqualS(&wm->wmPrettyName, FF_WM_PRETTY_XFWM4) ||
+        (ffStrbufIgnCaseEqualS(&wm->wmPrettyName, "labwc") && ffStrbufIgnCaseEqualS(&wm->dePrettyName, FF_DE_PRETTY_XFCE4))) {
+        return detectXFWM4(themeOrError);
+    }
+
+    if (ffStrbufIgnCaseEqualS(&wm->wmPrettyName, FF_WM_PRETTY_MUTTER)) {
+        if (
+            ffStrbufIgnCaseEqualS(&wm->dePrettyName, FF_DE_PRETTY_GNOME) ||
+            ffStrbufIgnCaseEqualS(&wm->dePrettyName, FF_DE_PRETTY_GNOME_CLASSIC)) {
+            return detectMutter(themeOrError);
+        } else {
+            return detectGTKThemeAsWMTheme(themeOrError);
+        }
+    }
+
+    if (ffStrbufIgnCaseEqualS(&wm->wmPrettyName, FF_WM_PRETTY_MUFFIN)) {
+        return detectMuffin(themeOrError);
+    }
+
+    if (ffStrbufIgnCaseEqualS(&wm->wmPrettyName, FF_WM_PRETTY_MARCO)) {
+        return detectWMThemeFromSettings("/org/mate/Marco/general/theme", "org.mate.Marco.general", NULL, "theme", themeOrError);
+    }
+
+    if (ffStrbufIgnCaseEqualS(&wm->wmPrettyName, FF_WM_PRETTY_OPENBOX)) {
+        return detectOpenbox(&wm->dePrettyName, themeOrError);
+    }
+
+    if (ffStrbufIgnCaseEqualS(&wm->wmPrettyName, FF_WM_PRETTY_ENLIGHTENMENT)) {
+        return detectGTKThemeAsWMTheme(themeOrError);
+    }
+
+    ffStrbufAppendS(themeOrError, "Unknown WM: ");
+    ffStrbufAppend(themeOrError, &wm->wmPrettyName);
+    return false;
+}