#include "controller/callbacks.h"
#include "controller/callbacks_internal.h"
#include "core/app_context.h"
#include "core/import_export.h"
#include "core/lua_config.h"
#include "utils/string_ext.h"
#include "utils/safe_string.h"
#include "utils/error_code.h"
#include "utils/os_env.h"
#include "utils/logger.h"
#include "utils/ui_constants.h"
#include "ui/ui_utils.h"
#include <string.h>
#include <stdio.h>
#include <windows.h>

// 按钮回调：导入
int btn_import_cb(Ihandle *self)
{
    Ihandle *dlg = IupGetDialog(self);
    AppContext *ctx = get_app_context_from_dlg(dlg);
    if (!ctx)
        return IUP_DEFAULT;

    if (!check_admin())
    {
        IupMessage(_("Error"), _("Administrator privileges are required to import PATH!"));
        return IUP_DEFAULT;
    }

    Ihandle *filedlg = IupFileDlg();
    IupSetAttribute(filedlg, "DIALOGTYPE", "OPEN");
    IupSetAttribute(filedlg, "TITLE", lua_config_get_string("label", "import_title"));
    IupSetAttribute(filedlg, "FILTER", "json");
    IupSetAttribute(filedlg, "EXTFILTER", "JSON 文件 (*.json)|*.json|文本文件 (*.txt)|*.txt|所有文件 (*.*)|*.*");

    IupPopup(filedlg, IUP_CENTER, IUP_CENTER);

    if (IupGetInt(filedlg, "STATUS") != -1)
    {
        char *filepath = IupGetAttribute(filedlg, "VALUE");
        if (filepath)
        {
            ExportData imported;
            init_string_list(&imported.system);
            init_string_list(&imported.user);
            ErrorCode import_result = import_paths_from_file(filepath, &imported);
            if (import_result == ERR_OK)
            {
                int has_system = imported.system.count > 0;
                int has_user = imported.user.count > 0;

                if (!has_system && !has_user)
                {
                    IupMessage(_("Error"), _("No valid paths found in file!"));
                    clear_string_list(&imported.system);
                    clear_string_list(&imported.user);
                    IupDestroy(filedlg);
                    return IUP_DEFAULT;
                }

                int choice = 0;
                if (has_system && has_user)
                {
                    choice = IupAlarm(_("Import Options"), _("Please select import target:"),
                                      _("System Variables Only"), _("User Variables Only"), _("Import All"));
                }
                else if (has_system)
                {
                    // TXT file import: let user choose target (system or user)
                    choice = IupAlarm(_("Import Options"), _("Please select import target:"),
                                      _("Import to System"), _("Import to User"), NULL);
                    // IupAlarm 返回 1 或 2，转换为 1(系统) 或 2(用户)
                }
                else
                {
                    choice = 2;
                }

                int total_imported = 0;

                if (choice == 1 || choice == 3)
                {
                    clear_string_list(&ctx->sys_paths);
                    for (int i = 0; i < imported.system.count; i++)
                    {
                        add_string_list(&ctx->sys_paths, imported.system.items[i]);
                    }
                    Ihandle *list_sys = IupGetDialogChild(dlg, CTRL_LIST_SYS);
                    sync_string_list_to_ui(list_sys, &ctx->sys_paths);
                    total_imported += imported.system.count;
                }

                if (choice == 2 || choice == 3)
                {
                    clear_string_list(&ctx->user_paths);
                    for (int i = 0; i < imported.user.count; i++)
                    {
                        add_string_list(&ctx->user_paths, imported.user.items[i]);
                    }
                    Ihandle *list_user = IupGetDialogChild(dlg, CTRL_LIST_USER);
                    sync_string_list_to_ui(list_user, &ctx->user_paths);
                    total_imported += imported.user.count;
                }

                // 释放导入数据
                clear_string_list(&imported.system);
                clear_string_list(&imported.user);

                char msg[256];
                snprintf(msg, sizeof(msg), _("Successfully imported %d paths!"), total_imported);
                IupMessage(_("Import Success"), msg);

                Ihandle *lbl_status = IupGetDialogChild(dlg, CTRL_LBL_STATUS);
                if (lbl_status)
                    IupSetAttribute(lbl_status, "TITLE", lua_config_get_string("status", "loaded"));
            }
            else
            {
                log_error("Import failed: error code %d", import_result);
                IupMessage(_("Error"), _("Import failed, please check if the file format is correct!"));
            }
        }
    }
    IupDestroy(filedlg);
    return IUP_DEFAULT;
}

// 按钮回调：导出
int btn_export_cb(Ihandle *self)
{
    Ihandle *dlg = IupGetDialog(self);
    AppContext *ctx = get_app_context_from_dlg(dlg);
    if (!ctx)
        return IUP_DEFAULT;

    ExportData data;
    data.system = ctx->sys_paths;
    data.user = ctx->user_paths;

    Ihandle *filedlg = IupFileDlg();
    IupSetAttribute(filedlg, "DIALOGTYPE", "SAVE");
    IupSetAttribute(filedlg, "TITLE", lua_config_get_string("label", "export_title"));
    IupSetAttribute(filedlg, "FILTER", "json");
    IupSetAttribute(filedlg, "EXTFILTER", "JSON 文件 (*.json)|*.json|CSV 文件 (*.csv)|*.csv");
    IupSetAttribute(filedlg, "DEFAULTEXT", "json");

    char default_name[64];
    snprintf(default_name, sizeof(default_name), "path_all.json");
    IupSetAttribute(filedlg, "VALUE", default_name);

    IupPopup(filedlg, IUP_CENTER, IUP_CENTER);

    if (IupGetInt(filedlg, "STATUS") != -1)
    {
        char *filepath = IupGetAttribute(filedlg, "VALUE");
        if (filepath)
        {
            char final_path[MAX_PATH];
            if (strchr(filepath, '.') == NULL)
            {
                snprintf(final_path, sizeof(final_path), "%s.json", filepath);
            }
            else
            {
                safe_strcpy(final_path, sizeof(final_path), filepath);
            }
            filepath = final_path;

            ErrorCode export_result = export_paths_to_file(&data, filepath);
            if (export_result == ERR_OK)
            {
                char msg[512];
                snprintf(msg, sizeof(msg), _("Export successful!\nSystem variables: %d\nUser variables: %d\n\nSave location: %s"),
                         data.system.count, data.user.count, filepath);
                IupMessage(_("Export Success"), msg);
            }
            else
            {
                log_error("Export failed: error code %d", export_result);
                IupMessage(_("Error"), _("Export failed!"));
            }
        }
    }
    IupDestroy(filedlg);
    return IUP_DEFAULT;
}