import { app } from "../../../scripts/app.js";
import { setupSizeNode } from "./image_size_base.js";

// SDXL分辨率选项
const sdxlResolutionOptions = {
    "Landscape": [
        "9:7 (1152x896)",   // 比例约 1.29
        "3:2 (1216x832)",   // 比例 1.50
        "7:4 (1344x768)",   // 比例 1.75
        "12:5 (1536x640)",  // 比例 2.40
    ],
    "Portrait": [
        "7:9 (896x1152)",   // 比例约 1.29
        "2:3 (832x1216)",   // 比例 1.50
        "4:7 (768x1344)",   // 比例 1.75
        "5:12 (640x1536)",  // 比例 2.40
    ],
    "Square": [
        "1:1 (1024x1024)",  // 比例 1.00
    ]
};

// Midjourney分辨率选项
const midjourneyResolutionOptions = {
    "Landscape": [
        "6:5 (1200x992)",   // 比例 1.21
        "4:3 (1232x928)",   // 比例 1.33
        "3:2 (1344x896)",   // 比例 1.50
        "16:9 (1456x816)",  // 比例 1.78
        "2:1 (1536x768)",   // 比例 2.00
    ],
    "Portrait": [
        "1:2 (768x1536)",   // 比例 0.50
        "9:16 (816x1456)",  // 比例 0.56
        "2:3 (896x1344)",   // 比例 0.67
        "3:4 (928x1232)",   // 比例 0.75
        "5:6 (992x1200)",   // 比例 0.83
    ],
    "Square": [
        "1:1 (1024x1024)",  // 比例 1.00
    ]
};

// 更新分辨率选项
function updatePlusResolutionOptions(node, mode, style, triggerUpdate = true) {
    console.log("[ImageSizeCreatorPlus] Updating resolution options for mode:", mode, "style:", style);
    
    // 找到分辨率widget
    const resolutionWidget = node.widgets?.find(w => w.name === "resolution");
    if (!resolutionWidget) {
        console.warn("[ImageSizeCreatorPlus] Resolution widget not found");
        return;
    }

    // 根据风格选择对应的分辨率选项
    const resolutionOptions = style === "Midjourney" ? midjourneyResolutionOptions : sdxlResolutionOptions;
    const options = resolutionOptions[mode] || [];
    
    // 如果选项没有变化，不进行更新
    if (JSON.stringify(resolutionWidget.options?.values) === JSON.stringify(options)) {
        return;
    }

    // 保存当前值
    const currentValue = resolutionWidget.value;

    // 更新选项
    resolutionWidget.options = { values: options };
    
    // 如果当前值在新选项中存在，保持不变
    if (options.includes(currentValue)) {
        resolutionWidget.value = currentValue;
    } else {
        resolutionWidget.value = options[0] || "";
    }

    // 触发widget的change事件
    resolutionWidget.callback?.(resolutionWidget.value);

    // 只在需要时触发更新
    if (triggerUpdate) {
        requestAnimationFrame(() => {
            app.graph.setDirtyCanvas(true);
        });
    }
}

function setupPlusSizeNode(nodeType, nodeData, app) {
    // 创建节点时的处理
    const onNodeCreated = nodeType.prototype.onNodeCreated;
    nodeType.prototype.onNodeCreated = function() {
        const result = onNodeCreated?.apply(this, arguments);

        // 初始化时更新分辨率选项
        const modeWidget = this.widgets?.find(w => w.name === "mode");
        const styleWidget = this.widgets?.find(w => w.name === "style");
        
        if (modeWidget && styleWidget) {
            // 初始更新，但不触发画布更新
            updatePlusResolutionOptions(this, modeWidget.value, styleWidget.value, false);

            // 监听模式变化
            const originalModeCallback = modeWidget.callback;
            modeWidget.callback = (value) => {
                console.log("[ImageSizeCreatorPlus] Mode changed to:", value);
                updatePlusResolutionOptions(this, value, styleWidget.value);
                originalModeCallback?.call(this, value);
            };

            // 监听风格变化
            const originalStyleCallback = styleWidget.callback;
            styleWidget.callback = (value) => {
                console.log("[ImageSizeCreatorPlus] Style changed to:", value);
                updatePlusResolutionOptions(this, modeWidget.value, value);
                originalStyleCallback?.call(this, value);
            };
        }

        return result;
    };

    // 加载配置时的处理
    const onConfigure = nodeType.prototype.onConfigure;
    nodeType.prototype.onConfigure = function(config) {
        const result = onConfigure?.apply(this, arguments);
        
        // 恢复时更新分辨率选项
        const modeWidget = this.widgets?.find(w => w.name === "mode");
        const styleWidget = this.widgets?.find(w => w.name === "style");
        
        if (modeWidget && styleWidget) {
            console.log("[ImageSizeCreatorPlus] Restoring configuration with mode:", modeWidget.value, "style:", styleWidget.value);
            // 延迟更新以确保连接稳定
            setTimeout(() => {
                updatePlusResolutionOptions(this, modeWidget.value, styleWidget.value, false);
            }, 0);
        }

        return result;
    };

    // 添加widget更改处理
    const onPropertyChanged = nodeType.prototype.onPropertyChanged;
    nodeType.prototype.onPropertyChanged = function(property, value) {
        if (property === "mode" || property === "style") {
            const modeWidget = this.widgets?.find(w => w.name === "mode");
            const styleWidget = this.widgets?.find(w => w.name === "style");
            if (modeWidget && styleWidget) {
                updatePlusResolutionOptions(this, modeWidget.value, styleWidget.value);
            }
        }
        return onPropertyChanged?.apply(this, arguments);
    };
}

// 注册扩展
app.registerExtension({
    name: "PaintingCoder.ImageSizeCreatorPlus",
    async beforeRegisterNodeDef(nodeType, nodeData, app) {
        if (nodeData.name === "PaintingCoder::ImageSizeCreatorPlus" || 
            nodeData.name === "PaintingCoder::ImageLatentCreatorPlus") {
            setupPlusSizeNode(nodeType, nodeData, app);
        }
    }
});