/**************************************************************************
* Tint2 : Icon theme handling
*
* Copyright (C) 2015       (mrovi9000@gmail.com)
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License version 2
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
* GNU General Public License for more details.
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
**************************************************************************/

/* http://standards.freedesktop.org/icon-theme-spec/ */

#include "icon-theme-common.h"

#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "apps-common.h"
#include "common.h"
#include "cache.h"

gboolean debug_icons = FALSE;

#define ICON_DIR_TYPE_SCALABLE 0
#define ICON_DIR_TYPE_FIXED 1
#define ICON_DIR_TYPE_THRESHOLD 2
typedef struct IconThemeDir {
    char *name;
    int size;
    int type;
    int max_size;
    int min_size;
    int threshold;
} IconThemeDir;

int parse_theme_line(char *line, char **key, char **value)
{
    return parse_dektop_line(line, key, value);
}

static GSList *icon_locations = NULL;
// Do not free the result.
const GSList *get_icon_locations()
{
    if (icon_locations)
        return icon_locations;

    icon_locations = load_locations_from_env(icon_locations, "XDG_DATA_HOME", ".icons", NULL);

    icon_locations = g_slist_append(icon_locations, g_build_filename(g_get_home_dir(), ".icons", NULL));
    icon_locations = g_slist_append(icon_locations, g_build_filename(g_get_home_dir(), ".local/share/icons", NULL));

    icon_locations = load_locations_from_env(icon_locations, "XDG_DATA_DIRS", ".icons", ".pixmaps", NULL);

    icon_locations = g_slist_append(icon_locations, g_strdup("/usr/local/share/icons"));
    icon_locations = g_slist_append(icon_locations, g_strdup("/usr/local/share/pixmaps"));
    icon_locations = g_slist_append(icon_locations, g_strdup("/usr/share/icons"));
    icon_locations = g_slist_append(icon_locations, g_strdup("/usr/share/pixmaps"));
    icon_locations = g_slist_append(icon_locations, g_strdup("/opt/share/icons"));
    icon_locations = g_slist_append(icon_locations, g_strdup("/opt/share/pixmaps"));

    icon_locations = slist_remove_duplicates(icon_locations, g_str_equal, g_free);

    return icon_locations;
}

static GSList *icon_extensions = NULL;
const GSList *get_icon_extensions()
{
    if (icon_extensions)
        return icon_extensions;

    icon_extensions = g_slist_append(icon_extensions, ".png");
    icon_extensions = g_slist_append(icon_extensions, ".xpm");
#ifdef HAVE_RSVG
    icon_extensions = g_slist_append(icon_extensions, ".svg");
#endif
    icon_extensions = g_slist_append(icon_extensions, "");
    return icon_extensions;
}

IconTheme *make_theme(const char *name)
{
    IconTheme *theme = calloc(1, sizeof(IconTheme));
    theme->name = strdup(name);
    theme->list_inherits = NULL;
    theme->list_directories = NULL;
    return theme;
}

// TODO Use UTF8 when parsing the file
IconTheme *load_theme_from_index(const char *file_name, const char *name)
{
    IconTheme *theme;
    FILE *f;
    char *line = NULL;
    size_t line_size;

    if ((f = fopen(file_name, "rt")) == NULL) {
        fprintf(stderr, "tint2: Could not open theme '%s'\n", file_name);
        return NULL;
    }

    theme = make_theme(name);

    IconThemeDir *current_dir = NULL;
    int inside_header = 1;
    while (getline(&line, &line_size, f) >= 0) {
        char *key, *value;

        int line_len = strlen(line);
        if (line_len >= 1) {
            if (line[line_len - 1] == '\n') {
                line[line_len - 1] = '\0';
                line_len--;
            }
        }

        if (line_len == 0)
            continue;

        if (inside_header) {
            if (parse_theme_line(line, &key, &value)) {
                if (strcmp(key, "Inherits") == 0) {
                    // value is like oxygen,wood,default
                    char *token;
                    token = strtok(value, ",\n");
                    while (token != NULL) {
                        theme->list_inherits = g_slist_append(theme->list_inherits, strdup(token));
                        token = strtok(NULL, ",\n");
                    }
                } else if (strcmp(key, "Directories") == 0) {
                    // value is like 48x48/apps,48x48/mimetypes,32x32/apps,scalable/apps,scalable/mimetypes
                    char *token;
                    token = strtok(value, ",\n");
                    while (token != NULL) {
                        IconThemeDir *dir = calloc(1, sizeof(IconThemeDir));
                        dir->name = strdup(token);
                        dir->max_size = dir->min_size = dir->size = -1;
                        dir->type = ICON_DIR_TYPE_THRESHOLD;
                        dir->threshold = 2;
                        theme->list_directories = g_slist_append(theme->list_directories, dir);
                        token = strtok(NULL, ",\n");
                    }
                }
            }
        } else if (current_dir != NULL) {
            if (parse_theme_line(line, &key, &value)) {
                if (strcmp(key, "Size") == 0) {
                    // value is like 24
                    sscanf(value, "%d", &current_dir->size);
                    if (current_dir->max_size == -1)
                        current_dir->max_size = current_dir->size;
                    if (current_dir->min_size == -1)
                        current_dir->min_size = current_dir->size;
                } else if (strcmp(key, "MaxSize") == 0) {
                    // value is like 24
                    sscanf(value, "%d", &current_dir->max_size);
                } else if (strcmp(key, "MinSize") == 0) {
                    // value is like 24
                    sscanf(value, "%d", &current_dir->min_size);
                } else if (strcmp(key, "Threshold") == 0) {
                    // value is like 2
                    sscanf(value, "%d", &current_dir->threshold);
                } else if (strcmp(key, "Type") == 0) {
                    // value is Fixed, Scalable or Threshold : default to scalable for unknown Type.
                    if (strcmp(value, "Fixed") == 0) {
                        current_dir->type = ICON_DIR_TYPE_FIXED;
                    } else if (strcmp(value, "Threshold") == 0) {
                        current_dir->type = ICON_DIR_TYPE_THRESHOLD;
                    } else {
                        current_dir->type = ICON_DIR_TYPE_SCALABLE;
                    }
                }
            }
        }

        if (line[0] == '[' && line[line_len - 1] == ']' && strcmp(line, "[Icon Theme]") != 0) {
            inside_header = 0;
            current_dir = NULL;
            line[line_len - 1] = '\0';
            char *dir_name = line + 1;
            GSList *dir_item = theme->list_directories;
            while (dir_item != NULL) {
                IconThemeDir *dir = dir_item->data;
                if (strcmp(dir->name, dir_name) == 0) {
                    current_dir = dir;
                    break;
                }
                dir_item = g_slist_next(dir_item);
            }
        }
    }
    fclose(f);

    free(line);
    return theme;
}

void load_theme_from_fs_dir(IconTheme *theme, const char *dir_name)
{
    gchar *file_name = g_build_filename(dir_name, "index.theme", NULL);
    if (g_file_test(file_name, G_FILE_TEST_EXISTS)) {
        g_free(file_name);
        return;
    }

    GDir *d = g_dir_open(dir_name, 0, NULL);
    if (d) {
        const gchar *size_name;
        while ((size_name = g_dir_read_name(d))) {
            gchar *full_size_name = g_build_filename(dir_name, size_name, NULL);
            if (g_file_test(file_name, G_FILE_TEST_IS_DIR)) {
                int size, size2;
                if ((sscanf(size_name, "%dx%d", &size, &size2) == 2 && size == size2) ||
                    (sscanf(size_name, "%d", &size) == 1)) {
                    GDir *dSize = g_dir_open(full_size_name, 0, NULL);
                    if (dSize) {
                        const gchar *subdir_name;
                        while ((subdir_name = g_dir_read_name(dSize))) {
                            IconThemeDir *dir = calloc(1, sizeof(IconThemeDir));
                            // value is like 48x48/apps
                            gchar *value = g_build_filename(size_name, subdir_name, NULL);
                            dir->name = strdup(value);
                            g_free(value);
                            dir->max_size = dir->min_size = dir->size = size;
                            dir->type = ICON_DIR_TYPE_FIXED;
                            theme->list_directories = g_slist_append(theme->list_directories, dir);
                        }
                        g_dir_close(dSize);
                    }
                }
            }
            g_free(full_size_name);
        }
        g_dir_close(d);
    }
}

IconTheme *load_theme_from_fs(const char *name, IconTheme *theme)
{
    gchar *dir_name = NULL;
    for (const GSList *location = get_icon_locations(); location; location = g_slist_next(location)) {
        gchar *path = (gchar *)location->data;
        dir_name = g_build_filename(path, name, NULL);
        if (g_file_test(dir_name, G_FILE_TEST_IS_DIR)) {
            if (!theme) {
                theme = make_theme(name);
            }
            load_theme_from_fs_dir(theme, dir_name);
        }
        g_free(dir_name);
        dir_name = NULL;
    }

    return theme;
}

IconTheme *load_theme(const char *name)
{
    // Look for name/index.theme in $HOME/.icons, /usr/share/icons, /usr/share/pixmaps (stop at the first found)
    // Parse index.theme -> list of IconThemeDir with attributes
    // Return IconTheme*

    if (name == NULL)
        return NULL;

    gchar *file_name = NULL;
    for (const GSList *location = get_icon_locations(); location; location = g_slist_next(location)) {
        gchar *path = (gchar *)location->data;
        file_name = g_build_filename(path, name, "index.theme", NULL);
        if (!g_file_test(file_name, G_FILE_TEST_EXISTS)) {
            g_free(file_name);
            file_name = NULL;
        }
        if (file_name)
            break;
    }

    IconTheme *theme = NULL;
    if (file_name) {
        theme = load_theme_from_index(file_name, name);
        g_free(file_name);
    }

    return load_theme_from_fs(name, theme);
}

void free_icon_theme(IconTheme *theme)
{
    if (!theme)
        return;
    free(theme->name);
    theme->name = NULL;
    free(theme->description);
    theme->description = NULL;
    for (GSList *l_inherits = theme->list_inherits; l_inherits; l_inherits = l_inherits->next) {
        free(l_inherits->data);
    }
    g_slist_free(theme->list_inherits);
    theme->list_inherits = NULL;
    for (GSList *l_dir = theme->list_directories; l_dir; l_dir = l_dir->next) {
        IconThemeDir *dir = (IconThemeDir *)l_dir->data;
        free(dir->name);
        free(l_dir->data);
    }
    g_slist_free(theme->list_directories);
    theme->list_directories = NULL;
}

void free_themes(IconThemeWrapper *wrapper)
{
    if (!wrapper)
        return;
    free(wrapper->icon_theme_name);
    for (GSList *l = wrapper->themes; l; l = l->next) {
        IconTheme *theme = (IconTheme *)l->data;
        free_icon_theme(theme);
        free(theme);
    }
    g_slist_free(wrapper->themes);
    for (GSList *l = wrapper->themes_fallback; l; l = l->next) {
        IconTheme *theme = (IconTheme *)l->data;
        free_icon_theme(theme);
        free(theme);
    }
    g_slist_free(wrapper->themes_fallback);
    g_slist_free_full(wrapper->_queued, free);
    free_cache(&wrapper->_cache);
    free(wrapper);
}

void test_launcher_read_theme_file()
{
    fprintf(stdout, YELLOW);
    IconTheme *theme = load_theme("oxygen");
    if (!theme) {
        fprintf(stderr, "tint2: Could not load theme\n");
        return;
    }
    fprintf(stderr, "tint2: Loaded theme: %s\n", theme->name);
    GSList *item = theme->list_inherits;
    while (item != NULL) {
        fprintf(stderr, "tint2: Inherits:%s\n", (char *)item->data);
        item = g_slist_next(item);
    }
    item = theme->list_directories;
    while (item != NULL) {
        IconThemeDir *dir = item->data;
        fprintf(stderr, "tint2: Dir:%s Size=%d MinSize=%d MaxSize=%d Threshold=%d Type=%s\n",
               dir->name,
               dir->size,
               dir->min_size,
               dir->max_size,
               dir->threshold,
               dir->type == ICON_DIR_TYPE_FIXED ? "Fixed" : dir->type == ICON_DIR_TYPE_SCALABLE
                                                                ? "Scalable"
                                                                : dir->type == ICON_DIR_TYPE_THRESHOLD ? "Threshold"
                                                                                                       : "?????");
        item = g_slist_next(item);
    }
    fprintf(stdout, RESET);
}

gboolean str_list_contains(const GSList *list, const char *value)
{
    const GSList *item = list;
    while (item != NULL) {
        if (g_str_equal(item->data, value)) {
            return TRUE;
        }
        item = g_slist_next(item);
    }
    return FALSE;
}

void load_themes_helper(const char *name, GSList **themes, GSList **queued)
{
    if (str_list_contains(*queued, name))
        return;
    GSList *queue = g_slist_append(NULL, strdup(name));
    *queued = g_slist_append(*queued, strdup(name));

    // Load wrapper->themes
    while (queue) {
        char *queued_name = queue->data;
        queue = g_slist_remove(queue, queued_name);

        fprintf(stderr, "tint2:  '%s',", queued_name);
        IconTheme *theme = load_theme(queued_name);
        if (theme != NULL) {
            *themes = g_slist_append(*themes, theme);

            GSList *item = theme->list_inherits;
            int pos = 0;
            while (item != NULL) {
                char *parent = item->data;
                if (!str_list_contains(*queued, parent)) {
                    queue = g_slist_insert(queue, strdup(parent), pos);
                    pos++;
                    *queued = g_slist_append(*queued, strdup(parent));
                }
                item = g_slist_next(item);
            }
        }

        free(queued_name);
    }
    fprintf(stderr, "tint2: \n");

    // Free the queue
    GSList *l;
    for (l = queue; l; l = l->next)
        free(l->data);
    g_slist_free(queue);
}

void load_default_theme(IconThemeWrapper *wrapper)
{
    if (wrapper->_themes_loaded)
        return;

    fprintf(stderr, GREEN "tint2: Loading icon theme %s:" RESET "\n", wrapper->icon_theme_name);

    load_themes_helper(wrapper->icon_theme_name, &wrapper->themes, &wrapper->_queued);
    load_themes_helper("hicolor", &wrapper->themes, &wrapper->_queued);

    wrapper->_themes_loaded = TRUE;
}

void load_fallbacks(IconThemeWrapper *wrapper)
{
    if (wrapper->_fallback_loaded)
        return;

    fprintf(stderr, RED "tint2: Loading additional icon themes (this means your icon theme is incomplete)..." RESET "\n");

    // Load wrapper->themes_fallback
    const GSList *location;
    for (location = get_icon_locations(); location; location = g_slist_next(location)) {
        gchar *path = (gchar *)location->data;
        GDir *d = g_dir_open(path, 0, NULL);
        if (d) {
            const gchar *name;
            while ((name = g_dir_read_name(d))) {
                gchar *file_name = g_build_filename(path, name, "index.theme", NULL);
                if (g_file_test(file_name, G_FILE_TEST_EXISTS) && !g_file_test(file_name, G_FILE_TEST_IS_DIR)) {
                    load_themes_helper(name, &wrapper->themes_fallback, &wrapper->_queued);
                }
                g_free(file_name);
            }
            g_dir_close(d);
        }
    }

    wrapper->_fallback_loaded = TRUE;
}

gchar *get_icon_cache_path()
{
    return g_build_filename(g_get_user_cache_dir(), "tint2", "icon.cache", NULL);
}

void load_icon_cache(IconThemeWrapper *wrapper)
{
    if (wrapper->_cache.loaded)
        return;

    fprintf(stderr, GREEN "tint2: Loading icon theme cache..." RESET "\n");

    gchar *cache_path = get_icon_cache_path();
    load_cache(&wrapper->_cache, cache_path);
    g_free(cache_path);
}

void save_icon_cache(IconThemeWrapper *wrapper)
{
    if (!wrapper || !wrapper->_cache.dirty)
        return;

    fprintf(stderr, GREEN "tint2: Saving icon theme cache..." RESET "\n");
    gchar *cache_path = get_icon_cache_path();
    save_cache(&wrapper->_cache, cache_path);
    g_free(cache_path);
}

IconThemeWrapper *load_themes(const char *icon_theme_name)
{
    IconThemeWrapper *wrapper = calloc(1, sizeof(IconThemeWrapper));

    if (!icon_theme_name) {
        fprintf(stderr, "tint2: Missing icon_theme_name theme, default to 'hicolor'.\n");
        icon_theme_name = "hicolor";
    }

    wrapper->icon_theme_name = strdup(icon_theme_name);

    return wrapper;
}

int directory_matches_size(IconThemeDir *dir, int size)
{
    if (dir->type == ICON_DIR_TYPE_FIXED) {
        return dir->size == size;
    } else if (dir->type == ICON_DIR_TYPE_SCALABLE) {
        return dir->min_size <= size && size <= dir->max_size;
    } else /*if (dir->type == ICON_DIR_TYPE_THRESHOLD)*/ {
        return dir->size - dir->threshold <= size && size <= dir->size + dir->threshold;
    }
}

int directory_size_distance(IconThemeDir *dir, int size)
{
    if (dir->type == ICON_DIR_TYPE_FIXED) {
        return abs(dir->size - size);
    } else if (dir->type == ICON_DIR_TYPE_SCALABLE) {
        if (size < dir->min_size) {
            return dir->min_size - size;
        } else if (size > dir->max_size) {
            return size - dir->max_size;
        } else {
            return 0;
        }
    } else /*if (dir->type == ICON_DIR_TYPE_THRESHOLD)*/ {
        if (size < dir->size - dir->threshold) {
            return dir->min_size - size;
        } else if (size > dir->size + dir->threshold) {
            return size - dir->max_size;
        } else {
            return 0;
        }
    }
}

gint compare_theme_directories(gconstpointer a, gconstpointer b, gpointer size_query)
{
    int size = GPOINTER_TO_INT(size_query);
    const IconThemeDir *da = (const IconThemeDir *)a;
    const IconThemeDir *db = (const IconThemeDir *)b;
    return abs(da->size - size) - abs(db->size - size);
}

Bool is_full_path(const char *s)
{
    if (!s)
        return FALSE;
    return s[0] == '/';
}

Bool file_exists(const char *path)
{
    return g_file_test(path, G_FILE_TEST_EXISTS);
}

char *icon_path_from_full_path(const char *s)
{
    if (is_full_path(s) && file_exists(s))
        return strdup(s);
    char *expanded = expand_tilde(s);
    if (is_full_path(expanded) && file_exists(expanded))
        return expanded;
    free(expanded);
    return NULL;
}

char *get_icon_path_helper(GSList *themes, const char *icon_name, int size)
{
    if (!icon_name)
        return NULL;

    char *result = icon_path_from_full_path(icon_name);
    if (result)
        return result;

    const GSList *basenames = get_icon_locations();
    const GSList *extensions = get_icon_extensions();

    GSList *theme;

    // Best size match
    // Contrary to the freedesktop spec, we are not choosing the closest icon in size, but the next larger icon
    // otherwise the quality is worse when scaling up (for size 22, if you can choose 16 or 32, you're better with 32)
    // We do fallback to the closest size if we cannot find a larger or equal icon

    // These 3 variables are used for keeping the closest size match
    int minimal_size = INT_MAX;
    char *best_file_name = NULL;
    GSList *best_file_theme = NULL;

    // These 3 variables are used for keeping the next larger match
    int next_larger_size = -1;
    char *next_larger = NULL;
    GSList *next_larger_theme = NULL;

    size_t file_name_size = 4096;
    char *file_name = calloc(file_name_size, 1);

    for (theme = themes; theme; theme = g_slist_next(theme)) {
        if (debug_icons)
            fprintf(stderr, "tint2: Searching theme: %s\n", ((IconTheme *)theme->data)->name);
        ((IconTheme *)theme->data)->list_directories =
            g_slist_sort_with_data(((IconTheme *)theme->data)->list_directories,
                                   compare_theme_directories,
                                   GINT_TO_POINTER(size));
        GSList *dir;
        for (dir = ((IconTheme *)theme->data)->list_directories; dir; dir = g_slist_next(dir)) {
            // Closest match
            gboolean possible = directory_size_distance((IconThemeDir *)dir->data, size) < minimal_size &&
                                (!best_file_theme ? TRUE : theme == best_file_theme);
            // Next larger match
            possible = possible || (((IconThemeDir *)dir->data)->size >= size &&
                                    (next_larger_size == -1 || ((IconThemeDir *)dir->data)->size < next_larger_size) &&
                                    (!next_larger_theme ? 1 : theme == next_larger_theme));
            if (!possible)
                continue;
            if (debug_icons)
                fprintf(stderr, "tint2: Searching directory: %s\n", ((IconThemeDir *)dir->data)->name);
            const GSList *base;
            for (base = basenames; base; base = g_slist_next(base)) {
                for (const GSList *ext = extensions; ext; ext = g_slist_next(ext)) {
                    char *base_name = (char *)base->data;
                    char *theme_name = ((IconTheme *)theme->data)->name;
                    char *dir_name = ((IconThemeDir *)dir->data)->name;
                    char *extension = (char *)ext->data;
                    if (strlen(base_name) + strlen(theme_name) + strlen(dir_name) + strlen(icon_name) +
                            strlen(extension) + 100 >
                        file_name_size) {
                        file_name_size = strlen(base_name) + strlen(theme_name) + strlen(dir_name) + strlen(icon_name) +
                                         strlen(extension) + 100;
                        file_name = realloc(file_name, file_name_size);
                    }
                    file_name[0] = 0;
                    // filename = directory/$(themename)/subdirectory/iconname.extension
                    snprintf(file_name, (size_t)file_name_size, "%s/%s/%s/%s%s", base_name, theme_name, dir_name, icon_name, extension);
                    if (debug_icons)
                        fprintf(stderr, "tint2: Checking %s\n", file_name);
                    if (g_file_test(file_name, G_FILE_TEST_EXISTS)) {
                        if (debug_icons)
                            fprintf(stderr, "tint2: Found potential match: %s\n", file_name);
                        // Closest match
                        if (directory_size_distance((IconThemeDir *)dir->data, size) < minimal_size &&
                            (!best_file_theme ? 1 : theme == best_file_theme)) {
                            if (best_file_name) {
                                free(best_file_name);
                                best_file_name = NULL;
                            }
                            best_file_name = strdup(file_name);
                            minimal_size = directory_size_distance((IconThemeDir *)dir->data, size);
                            best_file_theme = theme;
                            if (debug_icons)
                                fprintf(stderr, "tint2: best_file_name = %s; minimal_size = %d\n", best_file_name, minimal_size);
                        }
                        // Next larger match
                        if (((IconThemeDir *)dir->data)->size >= size &&
                            (next_larger_size == -1 || ((IconThemeDir *)dir->data)->size < next_larger_size) &&
                            (!next_larger_theme ? 1 : theme == next_larger_theme)) {
                            if (next_larger) {
                                free(next_larger);
                                next_larger = NULL;
                            }
                            next_larger = strdup(file_name);
                            next_larger_size = ((IconThemeDir *)dir->data)->size;
                            next_larger_theme = theme;
                            if (debug_icons)
                                fprintf(stderr, "tint2: next_larger = %s; next_larger_size = %d\n", next_larger, next_larger_size);
                        }
                    }
                }
            }
        }
    }
    free(file_name);
    file_name = NULL;
    if (next_larger) {
        free(best_file_name);
        return next_larger;
    }
    if (best_file_name) {
        return best_file_name;
    }

    // Look in unthemed icons
    {
        if (debug_icons)
            fprintf(stderr, "tint2: Searching unthemed icons\n");
        for (const GSList *base = basenames; base; base = g_slist_next(base)) {
            for (const GSList *ext = extensions; ext; ext = g_slist_next(ext)) {
                char *base_name = (char *)base->data;
                char *extension = (char *)ext->data;
                size_t file_name_size2 = strlen(base_name) + strlen(icon_name) + strlen(extension) + 100;
                file_name = calloc(file_name_size2, 1);
                // filename = directory/iconname.extension
                snprintf(file_name, file_name_size2, "%s/%s%s", base_name, icon_name, extension);
                if (debug_icons)
                    fprintf(stderr, "tint2: Checking %s\n", file_name);
                if (g_file_test(file_name, G_FILE_TEST_EXISTS)) {
                    if (debug_icons)
                        fprintf(stderr, "tint2: Found %s\n", file_name);
                    return file_name;
                } else {
                    free(file_name);
                    file_name = NULL;
                }
            }
        }
    }

    return NULL;
}

char *get_icon_path_from_cache(IconThemeWrapper *wrapper, const char *icon_name, int size)
{
    if (!wrapper || !icon_name || strlen(icon_name) == 0)
        return NULL;

    load_icon_cache(wrapper);

    gchar *key = g_strdup_printf("%s\t%s\t%d", wrapper->icon_theme_name, icon_name, size);
    const gchar *value = get_from_cache(&wrapper->_cache, key);
    g_free(key);

    if (!value) {
        fprintf(stderr,
                YELLOW "Icon path not found in cache: theme = %s, icon = %s, size = %d" RESET "\n",
                wrapper->icon_theme_name,
                icon_name,
                size);
        return NULL;
    }

    if (!g_file_test(value, G_FILE_TEST_EXISTS))
        return NULL;

    // fprintf(stderr, "tint2: Icon path found in cache: theme = %s, icon = %s, size = %d, path = %s\n",
    // wrapper->icon_theme_name, icon_name, size, value);

    return strdup(value);
}

void add_icon_path_to_cache(IconThemeWrapper *wrapper, const char *icon_name, int size, const char *path)
{
    if (!wrapper || !icon_name || strlen(icon_name) == 0 || !path || strlen(path) == 0)
        return;

    fprintf(stderr,
            "Adding icon path to cache: theme = %s, icon = %s, size = %d, path = %s\n",
            wrapper->icon_theme_name,
            icon_name,
            size,
            path);

    load_icon_cache(wrapper);

    gchar *key = g_strdup_printf("%s\t%s\t%d", wrapper->icon_theme_name, icon_name, size);
    add_to_cache(&wrapper->_cache, key, path);
    g_free(key);
}

char *get_icon_path(IconThemeWrapper *wrapper, const char *icon_name, int size, gboolean use_fallbacks)
{
    if (debug_icons)
        fprintf(stderr,
                "Searching for icon %s with size %d, fallbacks %sallowed\n",
                icon_name,
                size,
                use_fallbacks ? "" : "not ");
    if (!wrapper) {
        if (debug_icons)
            fprintf(stderr,
                    "Icon search aborted, themes not loaded\n");
        return NULL;
    }

    if (!icon_name || strlen(icon_name) == 0)
        goto notfound;

    char *path = get_icon_path_from_cache(wrapper, icon_name, size);
    if (path) {
        if (debug_icons)
            fprintf(stderr,
                    "Icon found in cache: %s\n", path);
        return path;
    }

    load_default_theme(wrapper);

    icon_name = icon_name ? icon_name : DEFAULT_ICON;
    path = get_icon_path_helper(wrapper->themes, icon_name, size);
    if (path) {
        if (debug_icons)
            fprintf(stderr, "tint2: Icon found: %s\n", path);
        add_icon_path_to_cache(wrapper, icon_name, size, path);
        return path;
    }

    if (!use_fallbacks)
        goto notfound;
    fprintf(stderr, YELLOW "tint2: Icon not found in default theme: %s" RESET "\n", icon_name);
    load_fallbacks(wrapper);

    path = get_icon_path_helper(wrapper->themes_fallback, icon_name, size);
    if (path) {
        add_icon_path_to_cache(wrapper, icon_name, size, path);
        return path;
    }

notfound:
    fprintf(stderr, RED "tint2: Could not find icon '%s', using default." RESET "\n", icon_name);
    path = get_icon_path_helper(wrapper->themes, DEFAULT_ICON, size);
    if (path)
        return path;
    path = get_icon_path_helper(wrapper->themes_fallback, DEFAULT_ICON, size);
    return path;
}
