preload.js

const { ipcRenderer } = require("electron");
const fs = require("fs");
const utils = require('./utils.js')

const mineMap = {
    bmp: "image/bmp",
    gif: "image/gif",
    heic: "image/heic",
    jpeg: "image/jpeg",
    jpg: "image/jpeg",
    jpe: "image/jpeg",
    png: "image/png",
    svg: "image/svg+xml",
    webp: "image/webp",
    ico: "image/x-icon",
    html: "html",
    htm: "html",
};

function uuidv4() {
    return "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx".replace(/[xy]/g, function (c) {
        var r = (Math.random() * 16) | 0,
            v = c == "x" ? r : (r & 0x3) | 0x8;
        return v.toString(16);
    });
}

function show(payload, filePath) {
    let params = {};
    let paramsIndex = payload.lastIndexOf("#");
    // 解析参数到 params 对象
    if (paramsIndex > 0) {
        payload
            .substring(paramsIndex + 1)
            .split("&")
            .forEach((item) => {
                params[item.split("=")[0]] = item.split("=")[1];
            });
    }
    utils.log(payload ? payload.substring(0, 30): null, filePath);
    // 网页的情况
    let isPage = filePath && (filePath.endsWith('htm') || filePath.endsWith('html'));
    isPage ||= payload.startsWith("http://") || payload.startsWith("https://");
    if (!isPage) {
        let img = new Image();
        img.src = payload;
        img.onload = function () {
            let width = img.width / (utools.isMacOs() ? 2 : 1);
            let height = img.height / (utools.isMacOs() ? 2 : 1);
            utils.log(`图片宽高: width: ${width}, height: ${height}`);
            loadWindow(width, height, payload, filePath, params);
        };
    } else {
        let width = utils.config.defaultWidth / (utools.isMacOs() ? 2 : 1);
        let height = utils.config.defaultHeight / (utools.isMacOs() ? 2 : 1);
        loadWindow(width, height, payload, filePath, params);
    }
}

function loadWindow(width, height, payload, filePath, params) {
    let scale = width / (height * 1.0);
    //图片大小不能超过当前显示器80%，否则缩放
    let cursorPoint = utools.getCursorScreenPoint();
    let display = utools.getDisplayNearestPoint(cursorPoint);
    if (display) {
        width = Math.min(width, display.size.width * 0.8);
        height = width / scale;
        height = Math.min(height, display.size.height * 0.8);
        width = height * scale;
    }
    let eleKey = uuidv4();
    //通过localStorage传参,解决url传参的大小限制问题
    localStorage.setItem(eleKey, payload);
    localStorage.setItem(eleKey + "_file", filePath);
    let windowOptions = {
        title: "悬浮",
        x: params.x ? parseInt(params.x) : cursorPoint.x,
        y: params.y ? parseInt(params.y) : cursorPoint.y,
        width: parseInt(width),
        height: parseInt(height),
        useContentSize: true,
        //不能最大最小化
        minimizable: false,
        maximizable: false,
        fullscreenable: false,
        //背景透明，防止放大缩小时出现白框
        transparent: true,
        backgroundColor: "#00000000",
        frame: false,
        alwaysOnTop: true,
        webPreferences: {
            preload: "suspend.js",
            // devTools: true,
        },
    }
    utils.log(windowOptions);
    let browserWindow = utools.createBrowserWindow("suspend.html?#" + eleKey, windowOptions, () => {
        // browserWindow.webContents.openDevTools();
        ipcRenderer.sendTo(browserWindow.webContents.id, "init");
        for (var i = 1; i <= 5; i++) {
            setTimeout(
                () => ipcRenderer.sendTo(browserWindow.webContents.id, "init"),
                i * 200
            );
        }
        ipcRenderer.on("resize", (event, changed, proportion) => {
            if (event.senderId == browserWindow.webContents.id) {
                let nowBounds = browserWindow.getBounds();
                let widthChanged = nowBounds.width + changed;
                browserWindow.setSize(
                    Math.ceil(widthChanged),
                    Math.ceil(widthChanged * proportion)
                );
            }
        });
        ipcRenderer.on("moveBounds", (event, x, y, width, height) => {
            if (event.senderId == browserWindow.webContents.id) {
                let bound = browserWindow.getBounds();
                let newBounds = {
                    x: parseInt(bound.x + x),
                    y: parseInt(bound.y + y),
                    width: parseInt(width || bound.width),
                    height: parseInt(height || bound.height),
                };
                browserWindow.setBounds(newBounds);
            }
        });
        ipcRenderer.on("toEdit", (event) => {
            if (event.senderId == browserWindow.webContents.id) {
                let bound = browserWindow.getBounds();
                browserWindow.capturePage().then((img) => {
                    utools.redirect("截图工具", {
                        type: "img",
                        data: `data:image/png;base64,${_arrayBufferToBase64(img)}#x=${bound.x}&y=${bound.y}`,
                    });
                    browserWindow.close();
                });
            }
        });
        ipcRenderer.on("copyNowImage", (event) => {
            if (event.senderId == browserWindow.webContents.id) {
                browserWindow.capturePage().then((img) => {
                    utools.copyImage(
                        `data:image/png;base64,${_arrayBufferToBase64(img)}`
                    );
                    utools.showNotification("图片已经拷贝至剪切板");
                    ipcRenderer.sendTo(browserWindow.webContents.id, "reduction");
                });
            }
        });
        ipcRenderer.on("saveNowImage", (event) => {
            if (event.senderId == browserWindow.webContents.id) {
                browserWindow.capturePage().then((img) => {
                    let defaultPath =
                        utools.getPath("downloads") +
                        "/suspend_" +
                        new Date().getTime() +
                        ".png";
                    let savePath = utools.showSaveDialog({
                        title: "保存图片",
                        defaultPath: defaultPath,
                        buttonLabel: "保存",
                    });
                    if (savePath) {
                        fs.writeFileSync(savePath, img);
                        utools.showNotification("保存成功");
                    }
                    ipcRenderer.sendTo(browserWindow.webContents.id, "reduction");
                });
            }
        });
        browserWindow.on("will-resize", (event, newBounds) => {
            event.preventDefault();
            ipcRenderer.sendTo(
                browserWindow.webContents.id,
                "will-resize",
                newBounds
            );
        });
    });
    if (utils.config.debug && utils.config.devTool) {
        browserWindow.webContents.openDevTools();
    }
}

function fileUrlData(path) {
    return new Promise((resolve, reject) => {
        let postfix = path.substring(path.lastIndexOf(".") + 1);
        let supportType = mineMap[postfix.toLowerCase()];
        if (supportType) {
            fs.readFile(path, function (err, data) {
                if (err) {
                    reject(err);
                } else if ("html" === supportType) {
                    resolve(data.toString("utf-8"));
                } else {
                    resolve(`data:${supportType};base64,${data.toString("base64")}`);
                }
            });
        } else {
            reject("不支持的文件格式!");
        }
    });
}

function _arrayBufferToBase64(buffer) {
    var binary = "";
    var bytes = new Uint8Array(buffer);
    var len = bytes.byteLength;
    for (var i = 0; i < len; i++) {
        binary += String.fromCharCode(bytes[i]);
    }
    return window.btoa(binary);
}

window.exports = {
    pic_sus: {
        mode: "none",
        args: {
            enter: (action) => {
                utils.log(action)
                window.utools.hideMainWindow();
                if (action.type === "files") {
                    for (i in action.payload) {
                        fileUrlData(action.payload[i].path)
                            .then((payload) => {
                                show(payload, action.payload[i].path);
                            })
                            .catch((err) => {
                                utools.showNotification(err);
                            })
                            .finally(() => {
                                window.utools.outPlugin();
                            });
                    }
                } else if (action.type === "img") {
                    show(action.payload);
                    window.utools.outPlugin();
                }
            },
        },
    },
    shot_sus: {
        mode: "none",
        args: {
            enter: (action) => {
                //解决用户反馈的截图并识别容易截取到utools黑屏的问题
                utools.hideMainWindow();
                utools.screenCapture((base64Str) => {
                    utools.copyImage(base64Str);
                    show(base64Str);
                    utools.outPlugin();
                });
            },
        },
    },
    base_sus: {
        mode: "none",
        args: {
            enter: (action) => {
                show(action.payload);
                window.utools.outPlugin();
            },
        },
    },
    svg_sus: {
        mode: "none",
        args: {
            enter: (action) => {
                var base64 = btoa(action.payload);
                show(`data:image/svg+xml;base64,${base64}`);
                window.utools.outPlugin();
            },
        },
    },
    htm_sus: {
        mode: "none",
        args: {
            enter: (action) => {
                window.utools.hideMainWindow();
                utils.log(action);
                switch (action.type) {
                    case "files":
                        for (i in action.payload) {
                            fileUrlData(action.payload[i].path)
                                .then((payload) => {
                                    show(payload, action.payload[i].path);
                                })
                                .catch((err) => {
                                    utools.showNotification(err);
                                })
                                .finally(() => {
                                    window.utools.outPlugin();
                                });
                        }
                        break;
                    case "img":
                        show(action.payload);
                        window.utools.outPlugin();
                    default:
                }
            },
        },
    },
    web_sus: {
        mode: "none",
        args: {
            enter: (action) => {
                window.utools.hideMainWindow();
                utils.log(action);
                if (action.type === "regex") {
                    try{
                        show(action.payload, action.payload);
                    } finally {
                        window.utools.outPlugin();
                    }
                }
            }
        },
    },
};