// Modules to control application life and create native browser window
const {
    app,
    Tray,
    Menu,
    clipboard,
    globalShortcut,
    BrowserWindow,
    ipcMain,
    dialog,
    Notification,
    shell,
    nativeImage,
    nativeTheme,
    BrowserView,
    screen,
    desktopCapturer,
} = require("electron");
const { Buffer } = require("buffer");
const robot = require("robotjs");
const Store = require("electron-store");
const path = require("path");
const run_path = path.resolve(__dirname, "");
const { exec } = require("child_process");
const fs = require("fs");
const os = require("os");
const ocr = require("./ocr/ocr");
const img_search = require("./lib/image_search");

var store = new Store();

var dev;
// 自动开启开发者模式
if (app.isPackaged || process.argv.includes("-n")) {
    dev = false;
} else {
    dev = true;
}

if (!store.get("硬件加速")) {
    app.disableHardwareAcceleration();
}

ipcMain.on("autostart", (event, m, v) => {
    if (m == "set") {
        if (process.platform == "linux") {
            if (v) {
                exec("mkdir ~/.config/autostart");
                exec(`cp ${run_path}/assets/e-search.desktop ~/.config/autostart/`);
            } else {
                exec("rm ~/.config/autostart/e-search.desktop");
            }
        } else {
            app.setLoginItemSettings({ openAtLogin: v });
        }
    } else {
        if (process.platform == "linux") {
            exec("test -e ~/.config/autostart/e-search.desktop", (error, stdout, stderr) => {
                error ? event.sender.send("开机启动状态", false) : event.sender.send("开机启动状态", true);
            });
        } else {
            event.sender.send("开机启动状态", app.getLoginItemSettings().openAtLogin);
        }
    }
});

function copy_text(callback) {
    var o_clipboard = clipboard.readText();
    if (process.platform == "darwin") {
        exec(
            `osascript -e 'tell application "System Events"' -e 'delay 0.1' -e 'key code 8 using command down' -e 'end tell'`
        );
    } else {
        robot.keyToggle("control", "down");
        setTimeout(() => {
            robot.keyTap("c");
        }, 100);
        setTimeout(() => {
            robot.keyToggle("control", "up");
        }, 200);
    }
    setTimeout(() => {
        let t = clipboard.readText();
        let v = false;
        if (o_clipboard != t) v = t;
        for (let i of store.get("主搜索功能.自动搜索排除")) {
            if (t.match(i)) {
                v = false;
                break;
            }
        }
        callback(v);
        clipboard.writeText(o_clipboard);
    }, 300);
}

// 自动判断选中搜索还是截屏搜索
function auto_open() {
    copy_text((t) => {
        if (t) {
            create_main_window("index.html", [t]);
        } else {
            full_screen();
        }
    });
}

function open_selection() {
    copy_text((t) => {
        if (t) create_main_window("index.html", [t]);
    });
}

function open_clip_board() {
    var t = clipboard.readText(
        process.platform == "linux" && store.get("主搜索功能.剪贴板选区搜索") ? "selection" : "clipboard"
    );
    create_main_window("index.html", [t]);
}

// cil参数重复启动;
var first_open = true;
const isFirstInstance = app.requestSingleInstanceLock();
if (!isFirstInstance) {
    first_open = false;
    app.quit();
} else {
    app.on("second-instance", (event, commanLine, workingDirectory) => {
        arg_run(commanLine);
    });
}
function arg_run(c) {
    if (c.includes("-n")) dev = false;
    switch (true) {
        case c.includes("-a"):
            auto_open();
            break;
        case c.includes("-c"):
            full_screen();
            break;
        case c.includes("-s"):
            open_selection();
            break;
        case c.includes("-b"):
            open_clip_board();
            break;
        case c.includes("-g"):
            create_main_window("index.html", [""]);
            break;
        case c.includes("-q"):
            quick_clip();
            break;
    }
}

var ocr_v = "2.5.0";
var file_o = { linux: ["Linux.tar.gz", 210], win32: ["Windows.zip", 91], darwin: ["macOS.zip", 340] };

async function download_ocr(callback) {
    const download = require("download");

    var download_path = app.getPath("userData");
    var url = `https://download.fastgit.org/xushengfeng/eSearch-OCR/releases/download/${ocr_v}/${
        file_o[process.platform][0]
    }`;

    let win = new BrowserWindow({ frame: false, transparent: true, width: 0, height: 0, icon: the_icon });
    new Notification({
        title: app.name,
        body: `${app.name} 服务正在下载中……`,
        icon: `${run_path}/assets/logo/64x64.png`,
    }).show();
    console.log("开始下载服务");
    await download(url, download_path, { extract: true })
        .on("response", (res) => {
            var download_len = 0;
            res.on("data", function (chunk) {
                download_len += chunk.length;
                var p = download_len / Number(res.headers["content-length"]);
                if (!win.isDestroyed()) {
                    win.setProgressBar(p);
                    win.setTitle(`${Math.round(p * 100)}%`);
                }
                if (p == 1) console.log("服务下载完成，解压中");
            });
            res.on("error", (err) => {
                callback(err);
            });
            win.on("close", () => {
                res.destroy();
                callback(true);
                new Notification({
                    title: app.name,
                    body: `${app.name} 服务下载已取消`,
                    icon: `${run_path}/assets/logo/64x64.png`,
                }).show();
            });
        })
        .on("error", (err) => {
            callback(err);
        });
    win.setProgressBar(-1.0);
    win.destroy();
    callback(null);
    new Notification({
        title: app.name,
        body: `${app.name} 服务已下载${process.platform == "linux" ? "" : "，正准备安装相关的依赖，请允许安装"}`,
        icon: `${run_path}/assets/logo/64x64.png`,
    }).show();
    if (process.platform == "win32") {
        exec(`cd ${download_path}\\ocr && .\\cp15e.exe`);
    } else if (process.platform == "darwin") {
        exec(`open -W "${download_path}/ocr/python-3.7.9-macosx10.9.pkg"`, (err) => {
            if (!err) fs.rm(`${download_path}/ocr/python-3.7.9-macosx10.9.pkg`, () => {});
        });
    }
}

async function rm_r(dir_path) {
    if (process.platform == "win32") {
        exec(`rd /s /q ${dir_path}`);
    } else {
        exec(`rm -r ${dir_path}`);
    }
}

var contextMenu, tray;

app.whenReady().then(() => {
    // 初始化设置
    Store.initRenderer();
    // 托盘
    tray =
        process.platform == "linux"
            ? new Tray(`${run_path}/assets/logo/32x32.png`)
            : new Tray(`${run_path}/assets/logo/16x16.png`);
    contextMenu = Menu.buildFromTemplate([
        {
            label: "自动搜索",
            click: () => {
                auto_open();
            },
        },
        {
            label: "截屏搜索",
            click: () => {
                setTimeout(() => {
                    full_screen();
                }, 500);
            },
        },
        {
            label: "选中搜索",
            click: () => {
                open_selection();
            },
        },
        {
            label: "剪贴板搜索",
            click: () => {
                open_clip_board();
            },
        },
        {
            type: "separator",
        },
        {
            label: "失焦关闭",
            submenu: [
                {
                    label: "主界面",
                    type: "checkbox",
                    checked: store.get("关闭窗口.失焦.主窗口"),
                    click: (i) => {
                        store.set("关闭窗口.失焦.主窗口", i.checked);
                    },
                },
                {
                    label: "搜索界面",
                    type: "checkbox",
                    checked: store.get("关闭窗口.失焦.搜索窗口"),
                    click: (i) => {
                        store.set("关闭窗口.失焦.搜索窗口", i.checked);
                    },
                },
            ],
        },
        {
            label: "浏览器打开",
            type: "checkbox",
            checked: store.get("浏览器中打开"),
            click: (i) => {
                store.set("浏览器中打开", i.checked);
            },
        },
        {
            type: "separator",
        },
        {
            label: "主页面",
            click: () => {
                create_main_window("index.html", [""]);
            },
        },
        {
            label: "设置",
            click: () => {
                Store.initRenderer();
                create_main_window("setting.html");
            },
        },
        {
            label: "教程帮助",
            click: () => {
                create_main_window("help.html");
            },
        },
        {
            type: "separator",
        },
        {
            label: "重启",
            click: () => {
                app.relaunch();
                app.exit(0);
            },
        },
        {
            label: "退出",
            click: () => {
                app.quit();
            },
        },
    ]);
    tray.setContextMenu(contextMenu);

    // 启动时提示
    if (first_open && store.get("启动提示"))
        new Notification({
            title: app.name,
            body: `${app.name} 已经在后台启动`,
            icon: `${run_path}/assets/logo/64x64.png`,
        }).show();

    if (store.get("首次运行") === undefined) set_default_setting();
    fix_setting_tree();

    /**
     * 检查ocr文件是否下载，否，则下载
     */
    async function check_ocr() {
        var ocr_path = path.join(app.getPath("userData"), "ocr"),
            old_ocr_path = path.join(app.getPath("userData"), "ocr.old");
        if (!store.get("OCR.检查OCR") || store.get("OCR.类型") != "离线") return;
        // 已经下载
        let download = fs.existsSync(ocr_path),
            // 需要升级
            update = store.get("OCR.版本") != ocr_v;

        if (!download)
            var resolve = await dialog.showMessageBox({
                title: "服务未下载",
                message: `${app.name} 离线OCR 服务未安装\n需要下载才能使用\n或前往 设置 配置 在线OCR`,
                checkboxLabel: "不再提示",
                buttons: [`下载(约${file_o[process.platform][1]}MB+)`, "前往 设置", "取消"],
                defaultId: 0,
                cancelId: 2,
            });
        if (download && update)
            var resolve = await dialog.showMessageBox({
                title: "服务需要升级",
                message: `${app.name} 离线OCR 服务版本较旧\n需要下载升级才能使用\n或前往 设置 配置 在线OCR`,
                buttons: [`下载(约${file_o[process.platform][1]}MB+)`, "前往 设置", "取消"],
                defaultId: 0,
                cancelId: 2,
            });
        if (resolve?.checkboxChecked) store.set("OCR.检查OCR", false);
        if (resolve?.response == 0) {
            if (download) fs.renameSync(ocr_path, old_ocr_path);
            download_ocr(async (err) => {
                if (err) {
                    if (download) fs.renameSync(old_ocr_path, ocr_path);
                    var resolve = await dialog.showMessageBox({
                        title: "服务下载失败",
                        message: `${
                            app.name
                        } 离线OCR 服务版本下载失败\n请前往网站手动下载\n解压并移动 ocr 文件夹到 ${app.getPath(
                            "userData"
                        )}`,
                        buttons: [`前往网站`, "取消"],
                        defaultId: 0,
                        cancelId: 1,
                    });
                    if (resolve?.response == 0) {
                        shell.openExternal("https://github.com/xushengfeng/eSearch-OCR/releases/tag/2.5.0");
                    }
                } else {
                    store.set("OCR.版本", ocr_v);
                    if (download) rm_r(old_ocr_path);
                }
            });
        } else if (resolve?.response == 1) create_main_window("setting.html");
    }

    // 快捷键
    var 快捷键函数 = {
        自动识别: { f: "auto_open()" },
        截屏搜索: { f: "full_screen()" },
        选中搜索: { f: "open_selection()" },
        剪贴板搜索: { f: "open_clip_board()" },
        快速截屏: { f: "quick_clip()" },
        主页面: { f: "create_main_window('index.html', [''])" },
    };
    ipcMain.on("快捷键", (event, arg) => {
        var [name, key] = arg;
        try {
            try {
                globalShortcut.unregister(store.get(`快捷键.${name}.key`));
            } catch {}
            if (key) {
                eval(`${arg[0]} = globalShortcut.register("${key}", () => {
                ${快捷键函数[arg[0]].f};});`);
            }
            // key为空或成功注册时保存，否则存为空
            store.set(`快捷键.${name}.key`, key === "" || eval(arg[0]) ? key : "");
            event.sender.send("状态", name, key ? eval(arg[0]) : true);
        } catch (error) {
            event.sender.send("状态", name, false);
            store.set(`快捷键.${name}.key`, "");
        }
    });

    var /**@type {Object} */ 快捷键 = store.get("快捷键");
    for (let k in 快捷键) {
        var m = 快捷键[k];
        try {
            if (m.key)
                eval(`globalShortcut.register("${m.key}", () => {
            ${快捷键函数[k].f};
        });`);
        } catch (error) {
            delete 快捷键[k].key;
            store.set(`快捷键`, 快捷键);
        }
    }

    // tmp目录
    if (!fs.existsSync(os.tmpdir() + "/eSearch")) fs.mkdir(os.tmpdir() + "/eSearch", () => {});
    create_clip_window();

    nativeTheme.themeSource = store.get("全局.深色模式");

    check_ocr();
});

app.on("will-quit", () => {
    // Unregister all shortcuts.
    globalShortcut.unregisterAll();

    // 删除临时文件夹
    rm_r(path.join(os.tmpdir(), "eSearch"));
});

var the_icon = null;
if (process.platform == "win32") {
    the_icon = path.join(run_path, "assets/logo/icon.ico");
} else {
    the_icon = path.join(run_path, "assets/logo/1024x1024.png");
}

// 截屏窗口
/**
 * @type BrowserWindow
 */
var clip_window = null;
function create_clip_window() {
    clip_window = new BrowserWindow({
        icon: the_icon,
        width: screen.getPrimaryDisplay().workAreaSize.width,
        height: screen.getPrimaryDisplay().workAreaSize.height,
        show: false,
        alwaysOnTop: true,
        fullscreenable: true,
        transparent: true,
        frame: false,
        skipTaskbar: true,
        autoHideMenuBar: true,
        movable: false,
        enableLargerThanScreen: true, // mac
        hasShadow: false,
        webPreferences: {
            nodeIntegration: true,
            contextIsolation: false,
        },
    });

    clip_window.loadFile("capture.html");
    clip_window.webContents.on("did-finish-load", () => {
        clip_window.webContents.setZoomFactor(store.get("全局.缩放") || 1.0);
        return new Promise(() => {});
    });

    if (dev) clip_window.webContents.openDevTools();

    // * 监听截屏奇奇怪怪的事件
    ipcMain.on("clip_main_b", (event, type, arg) => {
        switch (type) {
            case "window-close":
                n_full_screen();
                break;
            case "ocr":
                the_ocr(event, arg);
                break;
            case "search":
                image_search(event, arg);
                break;
            case "QR":
                if (arg != "nothing") {
                    create_main_window("index.html", [arg]);
                } else {
                    dialog.showMessageBox({
                        title: "警告",
                        message: "无法识别二维码\n请尝试重新识别",
                        icon: `${run_path}/assets/logo/warning.png`,
                    });
                }
                break;
            case "open":
                dialog
                    .showOpenDialog({
                        title: "选择要打开应用的位置",
                    })
                    .then((x) => {
                        console.log(x);
                        event.sender.send("open_path", x.filePaths[0]);
                    });
                break;
            case "save":
                var saved_path = store.get("保存路径") || "";
                n_full_screen();
                dialog
                    .showSaveDialog({
                        title: "选择要保存的位置",
                        defaultPath: `${saved_path}${get_file_name()}.${arg}`,
                        filters: [{ name: "图像", extensions: [arg] }],
                    })
                    .then((x) => {
                        event.sender.send("save_path", x.filePath);
                        if (x.filePath) {
                        } else {
                            new Notification({
                                title: `${app.name} 保存图像失败`,
                                body: `用户已取消保存`,
                                icon: `${run_path}/assets/logo/64x64.png`,
                            }).show();
                            clip_window.show();
                            clip_window.setSimpleFullScreen(true);
                        }
                    });
                break;
            case "ding":
                create_ding_window(arg[0], arg[1], arg[2], arg[3], arg[4]);
                break;
            case "mac_app":
                n_full_screen();
                dialog
                    .showOpenDialog({
                        defaultPath: "/Applications",
                    })
                    .then((x) => {
                        if (x.canceled) {
                            clip_window.show();
                            clip_window.setSimpleFullScreen(true);
                        }
                        event.sender.send("mac_app_path", x.canceled, x.filePaths);
                    });
                break;
            case "ok_save":
                noti(arg);
                var save_path = arg.split("/");
                save_path.pop();
                store.set("保存路径", save_path.join("/") + "/");
                break;
            case "record":
                var saved_path = store.get("保存路径") || "";
                n_full_screen();
                dialog
                    .showSaveDialog({
                        title: "选择要保存的位置",
                        defaultPath: `${saved_path}${get_file_name()}.${"webm"}`,
                        filters: [{ name: "视频", extensions: ["webm"] }],
                    })
                    .then((x) => {
                        if (x.filePath) {
                            create_recorder_window(x.filePath, arg);
                            event.sender.send("close");
                        } else {
                            new Notification({
                                title: `${app.name} 保存视频失败`,
                                body: `用户已取消保存`,
                                icon: `${run_path}/assets/logo/64x64.png`,
                            }).show();
                            clip_window.show();
                            clip_window.setSimpleFullScreen(true);
                        }
                    });
                break;
        }
    });

    // 移动光标
    ipcMain.on("move_mouse", (event, arrow, d) => {
        var mouse = robot.getMousePos();
        switch (arrow) {
            case "up":
                robot.moveMouse(mouse.x, mouse.y - 1 * d);
                break;
            case "right":
                robot.moveMouse(mouse.x + 1 * d, mouse.y);
                break;
            case "down":
                robot.moveMouse(mouse.x, mouse.y + 1 * d);
                break;
            case "left":
                robot.moveMouse(mouse.x - 1 * d, mouse.y);
                break;
        }
    });

    var x = robot.screen.capture();
    clip_window.webContents.send("reflash", x.image, x.width, x.height);
    x = null;

    // cil参数启动;
    if (first_open) arg_run(process.argv);
}

async function full_screen() {
    if (clip_window == null) {
        await create_clip_window();
    }
    let x = robot.screen.capture();
    clip_window.webContents.send("reflash", x.image, x.width, x.height);
    clip_window.show();
    clip_window.setSimpleFullScreen(true);
    let img_store = new Store({ name: "img_history" });
    let o_l = img_store.get("截屏记录") || {};
    if (store.get("记录截屏.限定保留")) {
        let s_l = Object.keys(o_l).sort();
        s_l = s_l.splice(-store.get("记录截屏.保留次数") + 1);
        var oo_l = {};
        for (let i of s_l) {
            oo_l[i] = o_l[i];
        }
        o_l = oo_l;
        oo_l = null;
    }
    img_store.set("截屏记录", o_l);
    let img_his_name = new Date().getTime();
    if (store.get("记录截屏.记录")) {
        let w = x.width,
            h = x.height;
        let image = nativeImage.createFromBuffer(Buffer.from(x.image), { width: x.width, height: x.height });
        let img_l = [{ src: image.toDataURL(), w, h }];
        img_store.set(`截屏记录.${img_his_name}`, img_l);
        img_l = null;
    }
    x = null;
}

function n_full_screen() {
    clip_window.setSimpleFullScreen(false);
    clip_window.hide();
}

function the_ocr(event, arg) {
    ocr(arg[0], arg[1], (err, result) => {
        if (err) {
            event.sender.send("ocr_back", "else");
            dialog.showMessageBox({
                title: "错误",
                message: `${err}`,
                buttons: ["确定"],
                icon: `${run_path}/assets/logo/warning.png`,
            });
        } else {
            event.sender.send("ocr_back", "ok");
            create_main_window("index.html", [result]);
        }
    });
}

function image_search(event, arg) {
    img_search(arg[0], arg[1], (err, url) => {
        if (err) {
            event.sender.send("search_back", "else");
            dialog.showMessageBox({
                title: "错误",
                message: `${err}`,
                buttons: ["确定"],
                icon: `${run_path}/assets/logo/warning.png`,
            });
        } else {
            if (store.get("浏览器中打开")) {
                shell.openExternal(url);
            } else {
                create_browser(null, url);
            }
            event.sender.send("search_back", "ok");
        }
    });
}

var /** @type {BrowserWindow}*/ recorder;
var mouse_ps = {};
var record_start = false;
var record_path = "";
function create_recorder_window(save_path, rect) {
    recorder = new BrowserWindow({
        icon: the_icon,
        width: 168,
        height: 24,
        alwaysOnTop: true,
        transparent: true,
        frame: false,
        autoHideMenuBar: true,
        resizable: false,
        titleBarStyle: "hiddenInset",
        webPreferences: {
            nodeIntegration: true,
            contextIsolation: false,
        },
    });
    recorder.loadFile("recorder.html");
    if (dev) recorder.webContents.openDevTools();

    mouse_ps = {};
    record_start = false;
    record_path = save_path;

    recorder.webContents.on("did-finish-load", () => {
        desktopCapturer.getSources({ types: ["window", "screen"] }).then(async (sources) => {
            recorder.webContents.send("record", "init", save_path, sources[0].id);
        });
    });
    record_start = true;
    mouse_ps = { rect };

    globalShortcut.register("Super+R", () => {
        recorder.webContents.send("record", "stop");
    });
}

ipcMain.on("record", (event, t) => {
    switch (t) {
        case "stop":
            record_start = false;
            let t = JSON.stringify(mouse_ps);
            fs.writeFile(record_path.replace("webm", "json"), t, () => {});
            break;
        case "start":
            function record_mouse() {
                if (record_start) {
                    let n_xy = screen.getCursorScreenPoint();
                    let s = screen.getAllDisplays()[0];
                    let t = new Date().getTime();
                    mouse_ps[t] = { ...n_xy, r: s.scaleFactor };
                    setTimeout(record_mouse, 10);
                }
            }
            record_mouse();
            break;
        case "close":
            recorder.close();
            break;
        case "min":
            recorder.minimize();
            break;
    }
});

ipcMain.on("setting", async (event, arg, arg1) => {
    switch (arg) {
        case "save_err":
            console.log("保存设置失败");
            break;
        case "reload_main":
            if (clip_window && !clip_window.isDestroyed() && !clip_window.isVisible()) clip_window.reload();
            contextMenu.items[5].submenu.items[0].checked = store.get("关闭窗口.失焦.主窗口");
            contextMenu.items[5].submenu.items[1].checked = store.get("关闭窗口.失焦.搜索窗口");
            contextMenu.items[6].checked = store.get("浏览器中打开");
            tray.setContextMenu(contextMenu);
            break;
        case "下载离线OCR":
            download_ocr();
            break;
        case "删除离线OCR":
            rm_r(path.join(app.getPath("userData"), "ocr"));
            break;
        case "set_default_setting":
            store.clear();
            set_default_setting();
            var resolve = await dialog.showMessageBox({
                title: "重启",
                message: `已恢复默认设置，部分设置需要重启 ${app.name} 生效`,
                buttons: ["重启", "稍后"],
                defaultId: 0,
                cancelId: 1,
            });
            if (resolve.response == 0) {
                app.relaunch();
                app.exit(0);
            }
            break;
        case "find":
            if (arg1.o?.start) {
                let o = arg1.o;
                delete arg1.o.start;
                event.sender.findInPage(arg1.t, o || {});
            } else {
                event.sender.stopFindInPage("clearSelection");
            }
            break;
    }
});

// 菜单栏设置(截屏没必要)
const isMac = process.platform === "darwin";
const template = [
    // { role: 'appMenu' }
    ...(isMac
        ? [
              {
                  label: app.name,
                  submenu: [
                      { label: `关于 ${app.name}`, role: "about" },
                      { type: "separator" },
                      {
                          label: "设置",
                          click: () => {
                              create_main_window("setting.html");
                          },
                          accelerator: "CmdOrCtrl+,",
                      },
                      { type: "separator" },
                      { label: "服务", role: "services" },
                      { type: "separator" },
                      { label: `隐藏 ${app.name}`, role: "hide" },
                      { label: "隐藏其他 ", role: "hideOthers" },
                      { label: "全部显示", role: "unhide" },
                      { type: "separator" },
                      { label: `退出 ${app.name}`, role: "quit" },
                  ],
              },
          ]
        : []),
    // { role: 'fileMenu' }
    {
        label: "文件",
        submenu: [
            {
                label: "保存到历史记录",
                click: (i, w) => {
                    main_edit(w, "save");
                },
                accelerator: "CmdOrCtrl+S",
            },
            { type: "separator" },
            ...(isMac
                ? []
                : [
                      {
                          label: "设置",
                          click: () => {
                              create_main_window("setting.html");
                          },
                          accelerator: "CmdOrCtrl+,",
                      },
                      { type: "separator" },
                  ]),
            {
                label: "其他编辑器打开",
                click: (i, w) => {
                    main_edit(w, "edit_on_other");
                },
            },
            {
                label: "打开方式...",
                click: (i, w) => {
                    main_edit(w, "choose_editer");
                },
            },
            { type: "separator" },
            { label: "关闭", role: "close" },
        ],
    },
    // { role: 'editMenu' }
    {
        label: "编辑",
        submenu: [
            {
                label: "打开链接",
                click: (i, w) => {
                    main_edit(w, "link");
                },
                accelerator: "CmdOrCtrl+Shift+L",
            },
            {
                label: "搜索",
                click: (i, w) => {
                    main_edit(w, "search");
                },
                accelerator: "CmdOrCtrl+Shift+S",
            },
            {
                label: "翻译",
                click: (i, w) => {
                    main_edit(w, "translate");
                },
                accelerator: "CmdOrCtrl+Shift+T",
            },
            { type: "separator" },
            {
                label: "撤销",
                click: (i, w) => {
                    main_edit(w, "undo");
                },
                accelerator: "CmdOrCtrl+Z",
            },
            {
                label: "重做",
                click: (i, w) => {
                    main_edit(w, "redo");
                },
                accelerator: isMac ? "Cmd+Shift+Z" : "Ctrl+Y",
            },
            { type: "separator" },
            {
                label: "剪切",
                click: (i, w) => {
                    main_edit(w, "cut");
                },
                accelerator: "CmdOrCtrl+X",
            },
            {
                label: "复制",
                click: (i, w) => {
                    main_edit(w, "copy");
                },
                accelerator: "CmdOrCtrl+C",
            },
            {
                label: "粘贴",
                click: (i, w) => {
                    main_edit(w, "paste");
                },
                accelerator: "CmdOrCtrl+V",
            },
            {
                label: "删除",
                click: (i, w) => {
                    main_edit(w, "delete");
                },
            },
            {
                label: "全选",
                click: (i, w) => {
                    main_edit(w, "select_all");
                },
                accelerator: "CmdOrCtrl+A",
            },
            {
                label: "自动删除换行",
                click: (i, w) => {
                    main_edit(w, "delete_enter");
                },
            },
            { type: "separator" },
            {
                label: "查找",
                click: (i, w) => {
                    main_edit(w, "show_find");
                },
                accelerator: "CmdOrCtrl+F",
            },
            {
                label: "替换",
                click: (i, w) => {
                    main_edit(w, "show_find");
                },
                accelerator: isMac ? "CmdOrCtrl+Option+F" : "CmdOrCtrl+H",
            },
            { type: "separator" },
            {
                label: "自动换行",
                click: (i, w) => {
                    main_edit(w, "wrap");
                },
            },
            {
                label: "拼写检查",
                click: (i, w) => {
                    main_edit(w, "spellcheck");
                },
            },
            { type: "separator" },
            ...(isMac
                ? [
                      {
                          label: "Speech",
                          submenu: [
                              { label: "开始朗读", role: "startSpeaking" },
                              { label: "停止朗读", role: "stopSpeaking" },
                          ],
                      },
                  ]
                : []),
        ],
    },
    {
        label: "浏览器",
        submenu: [
            {
                label: "后退",
                click: (i, w) => {
                    view_events(w, "back");
                },
                accelerator: isMac ? "Command+[" : "Alt+Left",
            },
            {
                label: "前进",
                click: (i, w) => {
                    view_events(w, "forward");
                },
                accelerator: isMac ? "Command+]" : "Alt+Right",
            },
            {
                label: "刷新",
                click: (i, w) => {
                    view_events(w, "reload");
                },
                accelerator: "F5",
            },
            {
                label: "停止加载",
                click: (i, w) => {
                    view_events(w, "stop");
                },
                accelerator: "Esc",
            },
            {
                label: "浏览器打开",
                click: (i, w) => {
                    view_events(w, "browser");
                },
            },
            {
                label: "保存到历史记录",
                click: (i, w) => {
                    view_events(w, "add_history");
                },
                accelerator: "CmdOrCtrl+D",
            },
        ],
    },
    // { role: 'viewMenu' }
    {
        label: "视图",
        submenu: [
            { label: "重新加载", role: "reload" },
            { label: "强制重载", role: "forceReload" },
            { label: "开发者工具", role: "toggleDevTools" },
            { type: "separator" },
            {
                label: "历史记录",
                click: (i, w) => {
                    main_edit(w, "show_history");
                },
                accelerator: "CmdOrCtrl+Shift+H",
            },
            { type: "separator" },
            { label: "实际大小", role: "resetZoom", accelerator: "" },
            { label: "放大", role: "zoomIn" },
            { label: "缩小", role: "zoomOut" },
            { type: "separator" },
            { label: "全屏", role: "togglefullscreen" },
        ],
    },
    // { role: 'windowMenu' }
    {
        label: "窗口",
        submenu: [
            { label: "最小化", role: "minimize" },
            { label: "关闭", role: "close" },
            ...(isMac
                ? [
                      { type: "separator" },
                      { label: "置于最前面", role: "front" },
                      { type: "separator" },
                      { label: "窗口", role: "window" },
                  ]
                : []),
        ],
    },
    {
        label: "帮助",
        role: "help",
        submenu: [
            {
                label: "教程帮助",
                click: () => {
                    create_main_window("help.html");
                },
            },
            { type: "separator" },
            {
                label: "关于",
                click: () => {
                    create_main_window("setting.html", true);
                },
            },
        ],
    },
];
const menu = Menu.buildFromTemplate(template);
Menu.setApplicationMenu(menu);

// ding窗口
var ding_windows_l = { dock: [0, 0, 10, 50] };
var ding_window;
function create_ding_window(x, y, w, h, img) {
    if (Object.keys(ding_windows_l).length == 1) {
        ding_window = new BrowserWindow({
            icon: the_icon,
            simpleFullscreen: true,
            fullscreen: true,
            transparent: true,
            frame: false,
            alwaysOnTop: true,
            skipTaskbar: true,
            autoHideMenuBar: true,
            enableLargerThanScreen: true, // mac
            hasShadow: false,
            webPreferences: {
                nodeIntegration: true,
                contextIsolation: false,
            },
        });

        ding_window.loadFile("ding.html");
        if (dev) ding_window.webContents.openDevTools();
        ding_window.webContents.on("did-finish-load", () => {
            ding_window.webContents.setZoomFactor(store.get("全局.缩放") || 1.0);
            var id = new Date().getTime();
            ding_window.webContents.send("img", id, x, y, w, h, img);
            ding_windows_l[id] = [x, y, w, h];
        });

        ding_window.setAlwaysOnTop(true, "screen-saver");
    } else {
        var id = new Date().getTime();
        ding_window.webContents.send("img", id, x, y, w, h, img);
        ding_windows_l[id] = [x, y, w, h];
    }
    var can_c_ignore = true;
    ipcMain.on("ding_ignore", (event, v) => {
        can_c_ignore = v;
        if (!v) ding_window.setIgnoreMouseEvents(false);
    });
    ipcMain.on("ding_p_s", (event, wid, p_s) => {
        ding_windows_l[wid] = p_s;
    });
    // 关闭窗口
    ipcMain.on("ding_close", (event, wid) => {
        delete ding_windows_l[wid];
        if (Object.keys(ding_windows_l).length == 1) {
            ding_window.close();
        }
    });
    // 自动改变鼠标穿透
    function ding_click_through() {
        var n_xy = screen.getCursorScreenPoint();
        var ratio = screen.getPrimaryDisplay().scaleFactor;
        var in_window = 0;
        for (let i in Object.values(ding_windows_l)) {
            let ii = Object.values(ding_windows_l)[i];
            // 如果光标在某个窗口上，不穿透
            var x2 = ii[0] + ii[2],
                y2 = ii[1] + ii[3];
            if (ii[0] <= n_xy.x * ratio && n_xy.x * ratio <= x2 && ii[1] <= n_xy.y * ratio && n_xy.y * ratio <= y2) {
                in_window += 1;
            }
        }
        // 窗口可能destroyed
        try {
            if (can_c_ignore)
                if (in_window > 0) {
                    ding_window.setIgnoreMouseEvents(false);
                } else {
                    ding_window.setIgnoreMouseEvents(true);
                }
        } catch {}
        setTimeout(ding_click_through, 10);
    }
    ding_click_through();
}

// 主窗口
/**
 * @type {Object.<number, BrowserWindow>}
 */
var main_window_l = {};
/**
 * @type {Object.<number, Array.<number>>}
 */
var main_to_search_l = {};
function create_main_window(web_page, t, about) {
    var window_name = new Date().getTime();
    var [w, h, m] = store.get("主窗口大小");
    var x = screen.getCursorScreenPoint().x - w,
        y = screen.getCursorScreenPoint().y - h;
    var main_window = (main_window_l[window_name] = new BrowserWindow({
        x: x < 0 ? 0 : x,
        y: y < 0 ? 0 : y,
        width: w,
        height: h,
        minWidth: 800,
        backgroundColor: nativeTheme.shouldUseDarkColors ? "#0f0f0f" : "#ffffff",
        icon: the_icon,
        webPreferences: {
            nodeIntegration: true,
            contextIsolation: false,
        },
        show: false,
    }));

    main_to_search_l[window_name] = [];

    if (m) main_window.maximize();

    // 自定义界面
    main_window.loadFile(web_page || "index.html");

    if (dev) main_window.webContents.openDevTools();
    main_window.webContents.on("did-finish-load", () => {
        main_window.show();
        main_window.webContents.setZoomFactor(store.get("全局.缩放") || 1.0);
        t = t || [""];
        // 确保切换到index时能传递window_name
        main_window.webContents.send("text", window_name, [t[0], t[1] || "auto"]);

        if (web_page == "setting.html") {
            main_window.webContents.send("about", about);
            main_window.webContents.send("setting", main_window.id);
        }
    });

    main_window.on("close", () => {
        store.set("主窗口大小", [
            main_window.getNormalBounds().width,
            main_window.getNormalBounds().height,
            main_window.isMaximized(),
        ]);
    });

    var 失焦关闭 = false;
    main_window.on("closed", () => {
        delete main_window_l[window_name];

        if (store.get("关闭窗口.子窗口跟随主窗口关") && !失焦关闭) {
            var search_l = [...main_to_search_l[window_name]];
            for (let i of search_l) {
                search_window_l[i].close();
            }
        }
    });

    main_window.on("blur", () => {
        if (store.get("关闭窗口.失焦.主窗口")) {
            失焦关闭 = true;
            main_window.close();
        }
    });

    main_window.webContents.on("found-in-page", (e, r) => {
        main_window.webContents.send("found", r.activeMatchOrdinal, r.matches);
    });
}

/**
 * 向聚焦的主界面发送事件信息
 * @param {String} m
 */
function main_edit(window, m) {
    window.webContents.send("edit", m);
}

/**
 * @type {Object.<number, BrowserWindow | BrowserView>}
 */
var search_window_l = {};
ipcMain.on("open_url", (event, window_name, url) => {
    create_browser(window_name, url);
});
async function create_browser(window_name, url) {
    var win_name = new Date().getTime();
    var [w, h, m] = store.get("主窗口大小");
    var search_window = (search_window_l[win_name] = new BrowserWindow({
        width: w,
        height: h,
        webPreferences: {
            nodeIntegration: true,
            contextIsolation: false,
        },
        backgroundColor: nativeTheme.shouldUseDarkColors ? "#0f0f0f" : "#ffffff",
        icon: the_icon,
    }));

    if (m) search_window.maximize();

    if (window_name) main_to_search_l[window_name].push(win_name);
    if (dev) search_window.webContents.openDevTools();

    if (store.get("开启代理")) await search_window.webContents.session.setProxy(store.get("代理"));
    search_window.loadFile("browser.html");

    search_window.on("resize", () => {
        var w = search_window.getBounds().width,
            h = search_window.getBounds().height;
        for (let i of search_window.getBrowserViews()) {
            i.setBounds({ x: 0, y: 30, width: w, height: h - 30 });
            if (search_window.isMaximized() || search_window.isFullScreen()) {
                search_window.setSize(
                    screen.getPrimaryDisplay().workArea.width,
                    screen.getPrimaryDisplay().workArea.height
                );
                i.setBounds({ x: 0, y: 30, width: w, height: h - 30 });
            }
        }
    });

    function new_browser_view(url) {
        if (search_window.isDestroyed()) return;
        var view = new Date().getTime();
        var search_view = (search_window_l[view] = new BrowserView());
        search_view.setBackgroundColor(nativeTheme.shouldUseDarkColors ? "#0f0f0f" : "#ffffff");
        search_window.addBrowserView(search_view);
        search_view.webContents.loadURL(url);
        var w = search_window.getBounds().width,
            h = search_window.getBounds().height;
        search_view.setBounds({ x: 0, y: 30, width: w, height: h - 30 });
        search_window.setSize(w, h + 1);
        search_window.setSize(w, h);
        search_view.webContents.setWindowOpenHandler(({ url }) => {
            new_browser_view(url);
            return { action: "deny" };
        });
        if (!search_window.isDestroyed()) search_window.webContents.send("url", win_name, view, "new", url);
        search_view.webContents.on("page-title-updated", (event, title) => {
            if (!search_window.isDestroyed()) search_window.webContents.send("url", win_name, view, "title", title);
            search_window.setTitle(`eSearch - ${title}`);
        });
        search_view.webContents.on("page-favicon-updated", (event, favlogo) => {
            if (!search_window.isDestroyed()) search_window.webContents.send("url", win_name, view, "icon", favlogo);
        });
        search_view.webContents.on("did-navigate", (event, url) => {
            if (!search_window.isDestroyed()) search_window.webContents.send("url", win_name, view, "url", url);
        });
        search_view.webContents.on("did-start-loading", () => {
            if (!search_window.isDestroyed()) search_window.webContents.send("url", win_name, view, "load", true);
        });
        search_view.webContents.on("did-stop-loading", () => {
            if (!search_window.isDestroyed()) search_window.webContents.send("url", win_name, view, "load", false);
        });
    }

    var 失焦关闭 = false;
    search_window.webContents.on("did-finish-load", () => {
        new_browser_view(url);
        search_window.on("blur", () => {
            if (store.get("关闭窗口.失焦.搜索窗口")) {
                失焦关闭 = true;
                search_window.close();
            }
        });
    });
    search_window.on("close", () => {
        close_win();
    });

    function close_win() {
        var views = search_window.getBrowserViews();
        for (let i of views) {
            search_window.removeBrowserView(i);
        }
        if (!window_name) return;
        // 清除列表中的索引
        for (let i in main_to_search_l[window_name]) {
            if (main_to_search_l[window_name][i] == win_name) {
                main_to_search_l[window_name].splice(Number(i), 1);
            }
        }
        // 自动关闭
        if (main_to_search_l[window_name].length == 0 && store.get("关闭窗口.主窗口跟随子窗口关") && !失焦关闭) {
            if (main_window_l[window_name]) main_window_l[window_name].close();
        }

        delete search_window_l[window_name];
    }
}
/**
 * 标签页事件
 * @param {BrowserWindow} w 浏览器
 * @param {String} arg 事件字符
 */
function view_events(w, arg) {
    w.webContents.send("view_events", arg);
}

ipcMain.on("tab_view", (e, pid, id, arg, arg2) => {
    var main_window = search_window_l[pid],
        search_window = search_window_l[id];
    if (arg == "close") {
        main_window.removeBrowserView(search_window);
        search_window.webContents.setAudioMuted(true);
        if (main_window.getBrowserViews().length == 0) main_window.close();
    } else if (arg == "top") {
        main_window.setTopBrowserView(search_window);
    } else if (arg == "back") {
        search_window.webContents.goBack();
    } else if (arg == "forward") {
        search_window.webContents.goForward();
    } else if (arg == "stop") {
        search_window.webContents.stop();
    } else if (arg == "reload") {
        search_window.webContents.reload();
    }
});

function get_file_name() {
    const f = require("./lib/time_format");
    var save_name_time = f(store.get("保存名称.时间"), new Date()).replace("\\", "");
    var file_name = store.get("保存名称.前缀") + save_name_time + store.get("保存名称.后缀");
    return file_name;
}
// 快速截屏
function quick_clip() {
    var x = robot.screen.capture();
    var image = nativeImage.createFromBuffer(Buffer.from(x.image), { width: x.width, height: x.height });
    if (store.get("快速截屏.模式") == "clip") {
        clipboard.writeImage(image);
    } else if (store.get("快速截屏.模式") == "path" && store.get("快速截屏.路径")) {
        var file_name = `${store.get("快速截屏.路径")}${get_file_name()}.png`;
        function check_file(n, name) {
            // 检查文件是否存在于当前目录中。
            fs.access(name, fs.constants.F_OK, (err) => {
                if (!err) {
                    /* 存在文件，需要重命名 */
                    name = file_name.replace(/\.png$/, `(${n}).png`);
                    check_file(n + 1, name);
                } else {
                    file_name = name;
                    fs.writeFile(
                        file_name,
                        Buffer.from(image.toDataURL().replace(/^data:image\/\w+;base64,/, ""), "base64"),
                        (err) => {
                            if (err) return;
                            noti(file_name);
                        }
                    );
                }
            });
        }
        check_file(1, file_name);
    }
    x = image = null;
}

function noti(file_path) {
    var notification = new Notification({
        title: `${app.name} 保存图像成功`,
        body: `已保存图像到 ${file_path}`,
        icon: `${run_path}/assets/logo/64x64.png`,
    });
    notification.on("click", () => {
        shell.showItemInFolder(file_path);
    });
    notification.show();
}

ipcMain.on("get_save_path", (event, path) => {
    if (!path) path = app.getPath("pictures");
    dialog
        .showOpenDialog({
            title: "选择要保存的位置",
            defaultPath: path,
            properties: ["openDirectory"],
        })
        .then((x) => {
            if (x.filePaths[0]) event.sender.send("get_save_path", x.filePaths[0] + "/");
        });
});

ipcMain.on("theme", (e, v) => {
    nativeTheme.themeSource = v;
    store.set("全局.深色模式", v);
});

// 默认设置
var default_setting = {
    首次运行: false,
    启动提示: true,
    快捷键: {
        自动识别: {
            key: "Alt+C",
        },
        截屏搜索: {},
        选中搜索: {},
        剪贴板搜索: {},
        快速截屏: {},
        主页面: {},
    },
    其他快捷键: {
        关闭: "Escape",
        OCR: "Enter",
        以图搜图: "",
        QR码: "",
        图像编辑: isMac ? "Command+D" : "Control+D",
        其他应用打开: "",
        放在屏幕上: "",
        复制: isMac ? "Command+C" : "Control+C",
        保存: isMac ? "Command+S" : "Control+S",
        复制颜色: "K",
    },
    主搜索功能: {
        自动搜索排除: [],
        剪贴板选区搜索: true,
    },
    全局: {
        模糊: 25,
        缩放: 1,
        不透明度: 0.4,
        深色模式: "system",
    },
    工具栏: {
        按钮大小: 60,
        按钮图标比例: 0.7,
    },
    字体: {
        主要字体: "",
        等宽字体: "",
        记住: false,
        大小: 16,
    },
    编辑器: {
        自动换行: true,
        拼写检查: false,
        行号: true,
        tab: 2,
        光标动画: 0.05,
    },
    工具栏跟随: "展示内容优先",
    取色器默认格式: "HEX",
    自动搜索: true,
    遮罩颜色: "#0008",
    选区颜色: "#0000",
    像素大小: 10,
    取色器大小: 15,
    显示四角坐标: true,
    其他应用打开: "",
    框选: {
        自动框选: {
            开启: false,
            最小阈值: 50,
            最大阈值: 150,
        },
    },
    图像编辑: {
        默认属性: {
            填充颜色: "#fff",
            边框颜色: "#333",
            边框宽度: 1,
            画笔颜色: "#333",
            画笔粗细: 2,
        },
        复制偏移: {
            x: 10,
            y: 10,
        },
    },
    OCR: {
        类型: "离线",
        检查OCR: true,
        离线切换: true,
        det: "",
        rec: "",
        字典: "",
        记住: false,
    },
    在线OCR: {
        baidu: {
            url: "",
            id: "",
            secret: "",
        },
        youdao: {
            id: "",
            secret: "",
        },
    },
    以图搜图: {
        引擎: "baidu",
        记住: false,
    },
    自动打开链接: false,
    自动搜索中文占比: 0.2,
    浏览器中打开: false,
    搜索窗口自动关闭: true,
    保存: {
        默认格式: "png",
    },
    保存路径: app.getPath("pictures") + "/",
    保存名称: { 前缀: "eSearch-", 时间: "YYYY-MM-DD-HH-mm-ss-S", 后缀: "" },
    jpg质量: 1,
    框选后默认操作: "no",
    快速截屏: { 模式: "clip", 路径: "" },
    搜索引擎: [
        ["Google", "https://www.google.com/search?q=%s"],
        ["百度", "https://www.baidu.com/s?wd=%s"],
        ["必应", "https://cn.bing.com/search?q=%s"],
        ["Yandex", "https://yandex.com/search/?text=%s"],
    ],
    翻译引擎: [
        ["Google", "https://translate.google.cn/?op=translate&text=%s"],
        ["Deepl", "https://www.deepl.com/translator#any/any/%s"],
        ["金山词霸", "http://www.iciba.com/word?w=%s"],
        ["百度", "https://fanyi.baidu.com/#auto/auto/%s"],
        ["腾讯", "https://fanyi.qq.com/?text=%s"],
    ],
    引擎: {
        记住: false,
        默认搜索引擎: "百度",
        默认翻译引擎: "Google",
    },
    历史记录设置: {
        保留历史记录: true,
        自动清除历史记录: false,
        d: 14,
        h: 0,
    },
    ding_dock: [0, 0],
    开启代理: false,
    代理: {
        pacScript: "",
        proxyRules: "",
        proxyBypassRules: "",
    },
    主窗口大小: [800, 600, false],
    关闭窗口: {
        失焦: { 主窗口: false, 搜索窗口: false },
        子窗口跟随主窗口关: false,
        主窗口跟随子窗口关: false,
    },
    记录截屏: {
        记录: true,
        限定保留: true,
        保留次数: 4,
    },
    时间格式: "MM/DD hh:mm:ss",
    硬件加速: true,
    更新: {
        检查更新: true,
    },
    录屏: {
        自动录制: 3,
        视频比特率: 2.5,
        屏幕: {
            帧率: 30,
        },
        摄像头: {
            默认开启: false,
            记住开启状态: false,
        },
        音频: {
            默认开启: false,
            记住开启状态: false,
        },
    },
};
function set_default_setting() {
    for (let i in default_setting) {
        store.set(i, default_setting[i]);
    }
}

// 增加设置项后，防止undefined
function fix_setting_tree() {
    var tree = "default_setting";
    walk(tree);
    function walk(path) {
        var x = eval(path);
        if (Object.keys(x).length == 0) {
            path = path.slice(tree.length + 1); /* 去除开头主tree */
            if (store.get(path) === undefined) store.set(path, x);
        } else {
            for (let i in x) {
                var c_path = path + "." + i;
                if (x[i].constructor === Object) {
                    walk(c_path);
                } else {
                    c_path = c_path.slice(tree.length + 1); /* 去除开头主tree */
                    if (store.get(c_path) === undefined) store.set(c_path, x[i]);
                }
            }
        }
    }
}