#include "maturita.h"

/* ------------------------------------------------------------------ */
/* Progress                                                           */
/* ------------------------------------------------------------------ */

void unit_save_progress(UnitCtx *u) {
    GKeyFile *kf = g_key_file_new();
    for (int i = 1; i <= u->n_ex; i++) {
        gchar key[8];
        g_snprintf(key, sizeof(key), "%d", i);
        g_key_file_set_boolean(kf, "done", key, u->done[i]);
    }

    if (g_mkdir_with_parents(PROGRESS_DIR, 0755) != 0) {
        g_key_file_free(kf);
        return;
    }

    gchar *data = g_key_file_to_data(kf, NULL, NULL);
    g_key_file_free(kf);

    if (data) {
        GError *err = NULL;
        g_file_set_contents(u->progress_file, data, -1, &err);
        if (err)
            g_error_free(err);
        g_free(data);
    }
}

void unit_load_progress(UnitCtx *u) {
    GKeyFile *kf = g_key_file_new();
    GError *err = NULL;

    if (!g_key_file_load_from_file(kf, u->progress_file, G_KEY_FILE_NONE, &err)) {
        if (err)
            g_error_free(err);
        g_key_file_free(kf);
        return;
    }

    for (int i = 1; i <= u->n_ex; i++) {
        gchar key[8];
        g_snprintf(key, sizeof(key), "%d", i);
        u->done[i] = g_key_file_get_boolean(kf, "done", key, NULL);
    }

    g_key_file_free(kf);
}

void load_progress(void) {
    for (int i = 0; i < NUM_UNLOCKED; i++)
        unit_load_progress(&units[i]);
}

/* Redraw the exercise rails of every unit that already has a page. */
void all_rails_redraw(void) {
    for (int i = 0; i < NUM_UNLOCKED; i++) {
        if (units[i].ex_rail)
            gtk_widget_queue_draw(units[i].ex_rail);
    }
}

void refresh_completion_ui(void) {
    for (int uidx = 0; uidx < NUM_UNLOCKED; uidx++) {
        UnitCtx *u = &units[uidx];
        gboolean all = TRUE;

        for (int i = 1; i <= u->n_ex; i++) {
            if (!u->done[i]) {
                all = FALSE;
                break;
            }
        }

        /* per-bubble markers */
        if (u->ex_cells[0] != NULL) {
            for (int i = 1; i <= u->n_ex; i++) {
                GtkWidget *btn = u->ex_cells[i - 1];
                if (!btn)
                    continue;
                if (u->done[i]) {
                    gtk_widget_add_css_class(btn, "done");
                    if (u->ex_icons[i])
                        gtk_widget_set_visible(u->ex_icons[i], TRUE);
                } else {
                    gtk_widget_remove_css_class(btn, "done");
                    if (u->ex_icons[i])
                        gtk_widget_set_visible(u->ex_icons[i], FALSE);
                }
            }
        }

        /* roadmap node marker */
        if (u->node) {
            if (all && u->n_ex > 0) {
                gtk_widget_remove_css_class(u->node, "current");
                gtk_widget_add_css_class(u->node, "done");
                if (u->node_done_icon)
                    gtk_widget_set_visible(u->node_done_icon, TRUE);
            } else {
                gtk_widget_remove_css_class(u->node, "done");
                gtk_widget_add_css_class(u->node, "current");
                if (u->node_done_icon)
                    gtk_widget_set_visible(u->node_done_icon, FALSE);
            }
        }
    }

    all_rails_redraw();
    refresh_stats_ui();
}

void mark_done(UnitCtx *u, int n) {
    if (!u || n < 1 || n > u->n_ex)
        return;
    u->done[n] = TRUE;
    refresh_completion_ui();
    unit_save_progress(u);
}