mirror of
https://github.com/fofolee/uTools-quickcommand.git
synced 2025-06-08 22:51:25 +08:00
330 lines
9.4 KiB
JavaScript
330 lines
9.4 KiB
JavaScript
module.exports = {
|
|
// 获取当前地址
|
|
getUrl: async function (browser = "Microsoft Edge") {
|
|
const result = await quickcommand.runAppleScript(`
|
|
tell application "${browser}"
|
|
set currentTab to active tab of front window
|
|
return URL of currentTab
|
|
end tell
|
|
`);
|
|
return result;
|
|
},
|
|
|
|
// 设置当前地址
|
|
setUrl: async function (browser = "Microsoft Edge", url) {
|
|
return await quickcommand.runAppleScript(`
|
|
tell application "${browser}"
|
|
set URL of active tab of front window to "${url}"
|
|
end tell
|
|
`);
|
|
},
|
|
|
|
// 获取标签列表
|
|
getTabs: async function (browser = "Microsoft Edge") {
|
|
const result = await quickcommand.runAppleScript(`
|
|
tell application "${browser}"
|
|
set tabList to every tab of front window
|
|
set json to "["
|
|
repeat with i from 1 to count of tabList
|
|
set currentTab to item i of tabList
|
|
set json to json & "{"
|
|
set json to json & "\\"title\\":\\"" & title of currentTab & "\\","
|
|
set json to json & "\\"url\\":\\"" & URL of currentTab & "\\","
|
|
set json to json & "\\"index\\":" & i & ","
|
|
set json to json & "\\"loading\\":" & loading of currentTab
|
|
set json to json & "}"
|
|
if i < count of tabList then
|
|
set json to json & ","
|
|
end if
|
|
end repeat
|
|
set json to json & "]"
|
|
return json
|
|
end tell
|
|
`);
|
|
return JSON.parse(result);
|
|
},
|
|
|
|
// 切换标签
|
|
activateTab: async function (browser = "Microsoft Edge", index) {
|
|
return await quickcommand.runAppleScript(`
|
|
tell application "${browser}"
|
|
activate
|
|
tell front window
|
|
set active tab index to ${index}
|
|
end tell
|
|
end tell
|
|
`);
|
|
},
|
|
|
|
// 执行脚本
|
|
executeScript: async function (
|
|
browser = "Microsoft Edge",
|
|
script,
|
|
args = {}
|
|
) {
|
|
try {
|
|
// 构建参数列表
|
|
const argNames = Object.keys(args);
|
|
const argValues = Object.values(args).map((v) => JSON.stringify(v));
|
|
|
|
script = `
|
|
(function(${argNames.join(", ")}) {
|
|
${script}
|
|
})(${argValues.join(", ")})
|
|
`;
|
|
// 使用JSON.stringify处理转义
|
|
const escapedScript = JSON.stringify(script);
|
|
|
|
let result = await quickcommand.runAppleScript(`
|
|
tell application "${browser}"
|
|
tell front window
|
|
tell active tab
|
|
execute javascript ${escapedScript}
|
|
end tell
|
|
end tell
|
|
end tell
|
|
`);
|
|
|
|
result = result?.trim() || "";
|
|
if (result === "missing value") {
|
|
// 脚本没有返回值
|
|
return;
|
|
} else if (result === "##execute_script_success##") {
|
|
// 模拟操作执行成功
|
|
return true;
|
|
} else if (result === "##execute_script_failed##") {
|
|
// 模拟操作执行失败
|
|
return false;
|
|
}
|
|
|
|
return result;
|
|
} catch (error) {
|
|
quickcommand.showSystemMessageBox("执行失败", error.toString());
|
|
throw error;
|
|
}
|
|
},
|
|
|
|
// 点击元素
|
|
clickElement: async function (browser = "Microsoft Edge", selector) {
|
|
const script = `
|
|
const element = document.querySelector('${selector}');
|
|
if (element) {
|
|
element.click();
|
|
return '##execute_script_success##';
|
|
} else {
|
|
return '##execute_script_failed##';
|
|
}
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 输入文本
|
|
inputText: async function (browser = "Microsoft Edge", selector, text) {
|
|
const script = `
|
|
const element = document.querySelector('${selector}');
|
|
if (element) {
|
|
element.value = '${text}';
|
|
element.dispatchEvent(new Event('input'));
|
|
element.dispatchEvent(new Event('change'));
|
|
return '##execute_script_success##';
|
|
} else {
|
|
return '##execute_script_failed##';
|
|
}
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 获取文本
|
|
getText: async function (browser = "Microsoft Edge", selector) {
|
|
const script = `
|
|
const element = document.querySelector('${selector}');
|
|
return element ? element.textContent : '';
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 获取HTML
|
|
getHtml: async function (browser = "Microsoft Edge", selector) {
|
|
const script = `
|
|
const element = document.querySelector('${selector}');
|
|
return element ? element.innerHTML : '';
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 隐藏元素
|
|
hideElement: async function (browser = "Microsoft Edge", selector) {
|
|
const script = `
|
|
const element = document.querySelector('${selector}');
|
|
if (element) {
|
|
element.style.display = 'none';
|
|
return '##execute_script_success##';
|
|
} else {
|
|
return '##execute_script_failed##';
|
|
}
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 显示元素
|
|
showElement: async function (browser = "Microsoft Edge", selector) {
|
|
const script = `
|
|
const element = document.querySelector('${selector}');
|
|
if (element) {
|
|
element.style.display = '';
|
|
return '##execute_script_success##';
|
|
} else {
|
|
return '##execute_script_failed##';
|
|
}
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 注入CSS
|
|
injectCSS: async function (browser = "Microsoft Edge", css) {
|
|
const script = `
|
|
const style = document.createElement('style');
|
|
style.textContent = \`${css}\`;
|
|
document.head.appendChild(style);
|
|
return '##execute_script_success##';
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 设置Cookie
|
|
setCookie: async function (
|
|
browser = "Microsoft Edge",
|
|
cookies,
|
|
options = {}
|
|
) {
|
|
const cookieStatements = cookies
|
|
.map((cookie) => {
|
|
let cookieString = `${cookie.name}=${cookie.value}`;
|
|
if (options.expires) {
|
|
const expiresDate = new Date(
|
|
Date.now() + options.expires * 60 * 60 * 1000
|
|
);
|
|
cookieString += `;expires=${expiresDate.toUTCString()}`;
|
|
}
|
|
if (options.path) cookieString += `;path=${options.path}`;
|
|
if (options.domain) cookieString += `;domain=${options.domain}`;
|
|
if (options.secure) cookieString += ";secure";
|
|
return `document.cookie = ${JSON.stringify(cookieString)};`;
|
|
})
|
|
.join("\n");
|
|
|
|
const script = `
|
|
${cookieStatements}
|
|
return '##execute_script_success##';
|
|
`;
|
|
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 获取Cookie
|
|
getCookie: async function (browser = "Microsoft Edge", name) {
|
|
const script = `
|
|
const value = document.cookie
|
|
.split('; ')
|
|
.find(row => row.startsWith('${name}='))
|
|
?.split('=')[1];
|
|
return value || '';
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 删除Cookie
|
|
deleteCookie: async function (
|
|
browser = "Microsoft Edge",
|
|
name,
|
|
path = "/",
|
|
domain = ""
|
|
) {
|
|
const script = `
|
|
document.cookie = '${name}=; expires=Thu, 01 Jan 1970 00:00:00 GMT' +
|
|
(${domain ? `'; domain=${domain}'` : "''"}) +
|
|
(${path ? `'; path=${path}'` : "''"});
|
|
return '##execute_script_success##';
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 滚动到指定位置
|
|
scrollTo: async function (browser = "Microsoft Edge", x, y) {
|
|
const script = `
|
|
window.scrollTo(${x}, ${y});
|
|
return '##execute_script_success##';
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 滚动到元素位置
|
|
scrollToElement: async function (browser = "Microsoft Edge", selector) {
|
|
const script = `
|
|
const element = document.querySelector('${selector}');
|
|
if (element) {
|
|
element.scrollIntoView({ behavior: 'smooth', block: 'center' });
|
|
return '##execute_script_success##';
|
|
} else {
|
|
return '##execute_script_failed##';
|
|
}
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 获取滚动位置
|
|
getScrollPosition: async function (browser = "Microsoft Edge") {
|
|
const script = `
|
|
return JSON.stringify({
|
|
x: window.pageXOffset || document.documentElement.scrollLeft,
|
|
y: window.pageYOffset || document.documentElement.scrollTop
|
|
});
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 获取页面尺寸
|
|
getPageSize: async function (browser = "Microsoft Edge") {
|
|
const script = `
|
|
return JSON.stringify({
|
|
width: Math.max(
|
|
document.documentElement.scrollWidth,
|
|
document.documentElement.clientWidth
|
|
),
|
|
height: Math.max(
|
|
document.documentElement.scrollHeight,
|
|
document.documentElement.clientHeight
|
|
)
|
|
});
|
|
`;
|
|
return await this.executeScript(browser, script);
|
|
},
|
|
|
|
// 等待元素出现
|
|
waitForElement: async function (
|
|
browser = "Microsoft Edge",
|
|
selector,
|
|
timeout = 5000
|
|
) {
|
|
const startTime = Date.now();
|
|
|
|
// 检查元素是否存在的函数
|
|
const checkScript = `
|
|
const element = document.querySelector('${selector}');
|
|
return element ? '##execute_script_success##' : '##execute_script_failed##';
|
|
`;
|
|
|
|
// 轮询检查元素
|
|
while (Date.now() - startTime < timeout) {
|
|
const result = await this.executeScript(browser, checkScript);
|
|
if (result === true) {
|
|
return true;
|
|
}
|
|
// 等待100ms再次检查
|
|
await new Promise((resolve) => setTimeout(resolve, 100));
|
|
}
|
|
|
|
return false;
|
|
},
|
|
};
|