const axios = require("axios");
const fs = require("fs-extra");
const path = require("path");

const mahmud = async () => {
        const base = await axios.get("https://raw.githubusercontent.com/mahmudx7/HINATA/main/baseApiUrl.json");
        return base.data.mahmud;
};

module.exports = {
        config: {
                name: "edit",
                aliases: ["imgedit"],
                version: "1.7",
                author: "ajmaul", // credit Change dile Tor mare cudmu .
                countDown: 10,
                role: 0,
                description: {
                        bn: "এআই এর মাধ্যমে আপনার ছবি এডিট করুন",
                        en: "Edit your image using AI prompt",
                        vi: "Chỉnh sửa hình ảnh của bạn bằng lời nhắc AI"
                },
                category: "image",
                guide: {
                        bn: '   {pn} <প্রম্পট>: ছবির রিপ্লাই দিয়ে এডিট প্রম্পট লিখুন'
                                + '\n   উদাহরণ: {pn} change hair color to red',
                        en: '   {pn} <prompt>: Reply to an image with edit instructions'
                                + '\n   Example: {pn} add sunglasses to face',
                        vi: '   {pn} <lời nhắc>: Phản hồi ảnh kèm hướng dẫn chỉnh sửa'
                                + '\n   Ví dụ: {pn} đổi màu tóc thành đỏ'
                }
        },

        langs: {
                bn: {
                        noInput: "× বেবি, একটি ছবিতে রিপ্লাই দিয়ে বলো কি এডিট করতে হবে! 🪄",
                        wait: "🔄 | তোমার ছবি এডিট করা হচ্ছে, একটু অপেক্ষা করো বেবি...",
                        success: "✅ | তোমার এডিট করা ছবি তৈরি: \"%1\"",
                        error: "× এডিট করতে সমস্যা হয়েছে: %1। প্রয়োজনে Contact Aizen।"
                },
                en: {
                        noInput: "× Baby, please reply to a photo with your prompt to edit it! 🪄",
                        wait: "🔄 | Editing your image, please wait...",
                        success: "✅ Here's your Edited image\nPrompt: %1",
                        error: "× Failed to edit: %1. Contact Aizen for help."
                },
                vi: {
                        noInput: "× Cưng ơi, vui lòng phản hồi ảnh kèm lời nhắc chỉnh sửa! 🪄",
                        wait: "🔄 | Đang chỉnh sửa ảnh, vui lòng chờ chút nhé...",
                        success: "✅ | Ảnh đã chỉnh sửa cho: \"%1\"",
                        error: "× Lỗi chỉnh sửa: %1. Liên hệ Aizen để hỗ trợ."
                }
        },

        onStart: async function ({ api, event, args, message, getLang }) {

                const prompt = args.join(" ");
                const repliedImage = event.messageReply?.attachments?.[0];

                if (!prompt || !repliedImage || repliedImage.type !== "photo") {
                        return message.reply(getLang("noInput"));
                }

                const cacheDir = path.join(__dirname, "cache");
                const imgPath = path.join(cacheDir, `${Date.now()}_edit.jpg`);
                await fs.ensureDir(cacheDir);

                const waitMsg = await message.reply(getLang("wait"));

                try {
                        const baseURL = await mahmud();
                        const res = await axios.post(
                                `${baseURL}/api/edit`,
                                { prompt, imageUrl: repliedImage.url },
                                { responseType: "arraybuffer" }
                        );

                        await fs.writeFile(imgPath, Buffer.from(res.data, "binary"));

                        await message.reply({
                                body: getLang("success", prompt),
                                attachment: fs.createReadStream(imgPath)
                        });

                } catch (err) {
                        console.error("Edit Command Error:", err);
                        return message.reply(getLang("error", err.message));
                } finally {
                        if (waitMsg?.messageID) api.unsendMessage(waitMsg.messageID);
                        setTimeout(() => {
                                if (fs.existsSync(imgPath)) fs.unlinkSync(imgPath);
                        }, 10000);
                }
        }
};