xbot-py-helper/scripts/content.js

(function () {

    console.log("Content script injected into:", window.location.href);

    // 清除之前的高亮样式
    let previousHighlight = null;
    let copyButton = null;
    let isCopyModeEnabled = false;
    let copyMessage = null;

    // 创建复制按钮
    function createCopyButton() {
        const button = document.createElement('button');
        button.innerHTML = '📋'; // 使用图标展示拷贝
        button.style.position = 'absolute';
        button.style.zIndex = '1000';
        button.style.display = 'none';
        button.style.background = 'transparent';
        button.style.border = 'none';
        button.style.cursor = 'pointer';
        button.style.fontSize = '16px';
        button.addEventListener('click', function () {
            if (previousHighlight) {
                const xpath = getElementXPath(previousHighlight);
                const action = previousHighlight.tagName.toLowerCase() === 'input' ? 'input("your_text")' : 'click()';
                const command = `web.find_by_xpath("${xpath}").${action}`;
                navigator.clipboard.writeText(command).then(() => {
                    console.log('Command copied to clipboard:', command);
                    copyMessage.style.display = 'block';
                    setTimeout(() => {
                        copyMessage.style.display = 'none';
                    }, 3000);
                });
            }
        });
        document.body.appendChild(button);
        return button;
    }

    // 创建复制消息提示
    function createCopyMessage() {
        const message = document.createElement('div');
        message.innerHTML = '已拷贝! <span id="copied-xpath" style="color:rgb(191, 201, 214);"></span><span style="cursor: pointer;">x</span> ';
        message.style.position = 'fixed';
        message.style.top = '10px';
        message.style.left = '50%';
        message.style.transform = 'translateX(-50%)';
        message.style.background = '#1E3A8A';
        message.style.color = '#F9FAFB';
        message.style.padding = '10px';
        message.style.borderRadius = '5px';
        message.style.display = 'none';
        message.style.zIndex = '1001';
        message.style.fontSize = '16px'; // 增大字体
        message.querySelector('span').addEventListener('click', function () {
            message.style.display = 'none';
        });
        document.body.appendChild(message);
        return message;
    }

    // 获取元素的 XPath
    function getElementXPath(element) {
        if (element.id) {
            return `//*[@id="${element.id}"]`;
        }

        const parts = [];
        while (element && element.nodeType === Node.ELEMENT_NODE) {
            let selector = element.nodeName.toLowerCase();
            if (element.className) {
                selector += `[@class="${element.className}"]`;
            } else if (element.textContent.trim()) {
                selector += `[text()="${element.textContent.trim()}"]`;
            } else if (element.id) {
                selector += `[@id="${element.id}"]`;
            } else {
                let sibling = element;
                let index = 1;
                while (sibling.previousElementSibling) {
                    sibling = sibling.previousElementSibling;
                    if (sibling.nodeName.toLowerCase() === selector) {
                        index++;
                    }
                }
                selector += `[${index}]`;
            }
            parts.unshift(selector);
            element = element.parentNode;
        }
        return parts.length ? '/' + parts.join('/') : null;
    }

    // 监听鼠标移动事件
    document.addEventListener('mousemove', function (event) {
        if (!isCopyModeEnabled) return;

        // 移除上一次高亮
        if (previousHighlight) {
            previousHighlight.style.background = '';
            previousHighlight.style.backgroundImage = '';
            previousHighlight.style.border = '';
            copyButton.style.display = 'none';
        }

        // 获取鼠标位置
        const mouseX = event.clientX;
        const mouseY = event.clientY;

        // 获取鼠标下的元素
        const element = document.elementFromPoint(mouseX, mouseY);

        // 如果有元素，添加高亮样式并输出对应的 XPath
        if (element && element !== document.body) {
            previousHighlight = element;
            element.style.background = 'rgba(255, 182, 193, 0.5)'; // 浅红色背景
            element.style.backgroundImage = 'linear-gradient(45deg, rgba(255, 182, 193, 0.5) 25%, transparent 25%, transparent 50%, rgba(255, 182, 193, 0.5) 50%, rgba(255, 182, 193, 0.5) 75%, transparent 75%, transparent)'; // 浅红色斜线
            element.style.backgroundSize = '10px 10px';
            element.style.border = '2px solid rgba(255, 105, 135, 0.8)'; // 边框颜色稍深
            element.style.borderRadius = '5px';

            // 设置复制按钮位置并显示
            const rect = element.getBoundingClientRect();
            copyButton.style.left = `${rect.right + window.scrollX - 20}px`;
            copyButton.style.top = `${rect.top + window.scrollY}px`;
            copyButton.style.display = 'block';

            // 输出 XPath
            const xpath = getElementXPath(element);
            console.log('Element XPath:', xpath);
        }
    });

    // 监听消息以启用或禁用拷贝模式
    chrome.runtime.onMessage.addListener(function (request, sender, sendResponse) {
        if (request.action === 'toggleCopyMode') {
            isCopyModeEnabled = !isCopyModeEnabled;
            if (!isCopyModeEnabled && previousHighlight) {
                previousHighlight.style.border = '';
                previousHighlight.style.borderRadius = '';
                previousHighlight.style.boxShadow = '';
                copyButton.style.display = 'none';
            }
            sendResponse({ copyModeEnabled: isCopyModeEnabled });
        }
    });

    // 监听键盘事件以启用或禁用拷贝模式
    document.addEventListener('keydown', function (event) {
        console.log(event.key);
        if (event.altKey && event.key === 'r') {
            isCopyModeEnabled = true;
            console.log('Copy mode enabled');
        } else if (event.key === 'Escape') {
            isCopyModeEnabled = false;
            if (previousHighlight) {
                previousHighlight.style.background = '';
                previousHighlight.style.backgroundImage = '';
                previousHighlight.style.border = '';
                copyButton.style.display = 'none';
                previousHighlight = null;
            }
            console.log('Copy mode disabled');
        } else if (event.ctrlKey && event.key === 'c' && previousHighlight) {
            const xpath = getElementXPath(previousHighlight);
            const action = previousHighlight.tagName.toLowerCase() === 'input' ? 'input("your_text")' : 'click()';
            const command = `web.find_by_xpath("${xpath}").${action}`;
            navigator.clipboard.writeText(command).then(() => {
                console.log('Command copied to clipboard:', command);
                const truncatedXPath = xpath.length > 20 ? xpath.substring(0, 20) + '...' : xpath;
                document.getElementById('copied-xpath').innerText = truncatedXPath;
                copyMessage.style.display = 'block';
                setTimeout(() => {
                    copyMessage.style.display = 'none';
                }, 3000);
                isCopyModeEnabled = false;
                if (previousHighlight) {
                    previousHighlight.style.background = '';
                    previousHighlight.style.backgroundImage = '';
                    previousHighlight.style.border = '';
                    copyButton.style.display = 'none';
                    previousHighlight = null;
                }
                console.log('Copy mode disabled');
            });
        }
    });

    copyButton = createCopyButton();
    copyMessage = createCopyMessage();
})();