mirror of
https://github.com/fofolee/uTools-quickcommand.git
synced 2025-06-08 22:51:25 +08:00
535 lines
18 KiB
JavaScript
535 lines
18 KiB
JavaScript
const fs = require('fs');
|
|
const os = require('os');
|
|
const child_process = require("child_process")
|
|
const iconv = require('iconv-lite')
|
|
const electron = require('electron')
|
|
const { NodeVM } = require('vm2')
|
|
const path = require("path")
|
|
const util = require("util")
|
|
fofoCommon = require('./common').fofo
|
|
|
|
|
|
if (!utools.isWindows()) process.env.PATH += ':/usr/local/bin:/usr/local/sbin'
|
|
|
|
// window.startTime = new Date().getTime()
|
|
|
|
const shortCodes = [
|
|
|
|
open = path => {
|
|
utools.shellOpenItem(path)
|
|
},
|
|
|
|
locate = path => {
|
|
utools.shellShowItemInFolder(path);
|
|
},
|
|
|
|
visit = url => {
|
|
utools.shellOpenExternal(url);
|
|
},
|
|
|
|
system = cmd => {
|
|
child_process.exec(cmd);
|
|
},
|
|
|
|
message = msg => {
|
|
utools.showNotification(msg)
|
|
},
|
|
|
|
keyTap = (key, ...modifier) => utools.simulateKeyboardTap(key, ...modifier),
|
|
|
|
copyTo = text => {
|
|
electron.clipboard.writeText(text)
|
|
},
|
|
|
|
send = text => {
|
|
copyTo(text);
|
|
quickcommand.simulatePaste();
|
|
}
|
|
]
|
|
|
|
const quickcommand = {
|
|
simulateCopy: function() {
|
|
var ctlKey = utools.isMacOs() ? 'command' : 'control';
|
|
utools.simulateKeyboardTap('c', ctlKey);
|
|
},
|
|
|
|
simulatePaste: function() {
|
|
var ctlKey = utools.isMacOs() ? 'command' : 'control';
|
|
utools.simulateKeyboardTap('v', ctlKey);
|
|
},
|
|
|
|
// setTimout 不能在 vm2 中使用，同时在 electron 中有 bug
|
|
sleep: function(ms) {
|
|
var start = new Date().getTime()
|
|
var cmd, tempFilePath
|
|
if (utools.isWindows()) {
|
|
tempFilePath = getQuickCommandScriptFile('vbs')
|
|
cmd = `echo set ws=CreateObject("Wscript.Shell") > ${tempFilePath} && echo Wscript.sleep ${ms} >> ${tempFilePath} && cscript /nologo ${tempFilePath}`
|
|
} else {
|
|
cmd = `sleep ${ms / 1000}`
|
|
}
|
|
try {
|
|
child_process.execSync(cmd, { timeout: ms, })
|
|
} catch (ex) {
|
|
if (ex.code !== 'ETIMEDOUT') {
|
|
throw ex;
|
|
}
|
|
}
|
|
var end = new Date().getTime()
|
|
return (end - start)
|
|
},
|
|
|
|
showInputBox: function (callback, placeHolders) {
|
|
let helps = `正确用法：
|
|
quickcommand.showInputBox(yourinput => {
|
|
do something...
|
|
}, [placeholder of input1, placeholder of input2...])`
|
|
if (!(callback instanceof Function)) throw helps
|
|
placeHolders || (placeHolders = [""])
|
|
if (!(placeHolders instanceof Array)) throw helps
|
|
utools.setExpendHeight(600)
|
|
var html = ""
|
|
var inputBoxNumbers = placeHolders.length
|
|
for (let i = 0; i < inputBoxNumbers; i++) {
|
|
html += `<input class="swal2-input" id="inputBox${i}" placeholder="${placeHolders[i]}">`
|
|
}
|
|
var result = []
|
|
var options = {
|
|
html: html,
|
|
focusConfirm: false,
|
|
preConfirm: () => {
|
|
for (let i = 0; i < inputBoxNumbers; i++) {
|
|
result.push(document.getElementById(`inputBox${i}`).value)
|
|
}
|
|
callback(result)
|
|
}
|
|
}
|
|
swalOneByOne(options)
|
|
},
|
|
|
|
showSelectBox: function (callback, selects) {
|
|
let helps = `正确用法：
|
|
quickcommand.showSelectBox(choise => {
|
|
var index = choise.index
|
|
var text = choise.text
|
|
//do something...
|
|
}, [option1, option2...])`
|
|
if (!(callback instanceof Function)) throw helps
|
|
if (!(selects instanceof Array) || (selects && !selects.length)) throw helps
|
|
// 调整插件高度
|
|
let modWindowHeight = num => {
|
|
if(!$("#customize").is(":parent")) utools.setExpendHeight(num > 11 ? 600 : 50 * (num + 1));
|
|
}
|
|
var html = `<div id="quickselect"><select id="selectBox">`
|
|
var selectBoxNumbers = selects.length
|
|
modWindowHeight(selectBoxNumbers)
|
|
for (let i = 0; i < selectBoxNumbers; i++) {
|
|
html += `<option value="${i}">${selects[i]}</option>`
|
|
}
|
|
html += `</select></div>`
|
|
$("body").append(html)
|
|
$('#selectBox').select2({
|
|
width: "100%",
|
|
dropdownParent: $("#quickselect")
|
|
})
|
|
$('#selectBox').val(null).trigger('change');
|
|
$('#selectBox').select2('open')
|
|
$('#quickselect .select2').hide()
|
|
$('#selectBox').on('select2:select', function (e) {
|
|
$('#selectBox').off('select2:select');
|
|
callback({ index: $(this).val(), text: selects[$(this).val()] })
|
|
$("#quickselect").remove()
|
|
})
|
|
$('#quickselect .select2-search__field').bind("input propertychange change",function(event){
|
|
modWindowHeight($('.select2-results__option').length)
|
|
});
|
|
},
|
|
|
|
showButtonBox: function (callback, buttons) {
|
|
let helps = `正确用法：
|
|
quickcommand.showButtonBox(yourchoise => {
|
|
var index = choise.index
|
|
var text = choise.text
|
|
//do something...
|
|
}, [button1, button2...])`
|
|
if (!(callback instanceof Function)) throw helps
|
|
if (!(buttons instanceof Array) || (buttons && !buttons.length)) throw helps
|
|
utools.setExpendHeight(600)
|
|
var html = ``
|
|
var buttonBoxNumbers = buttons.length
|
|
for (let i = 0; i < buttonBoxNumbers; i++) {
|
|
html += `<button class="swal2-confirm swal2-styled" style="width: 80%" onclick="clickButton(${i})">${buttons[i]}</button>`
|
|
}
|
|
var options = {
|
|
onBeforeOpen: () => {
|
|
clickButton = i => {
|
|
callback({ index: i, text: buttons[i] })
|
|
swal.clickConfirm()
|
|
}
|
|
},
|
|
html: html,
|
|
showConfirmButton: false
|
|
}
|
|
swalOneByOne(options)
|
|
}
|
|
}
|
|
|
|
swalOneByOne = options => {
|
|
swal.getQueueStep() && Swal.insertQueueStep(options) || Swal.queue([options])
|
|
}
|
|
|
|
var getSandboxFuns = () => {
|
|
var sandbox = {
|
|
utools: utools,
|
|
quickcommand: quickcommand,
|
|
electron: electron,
|
|
fs: fs,
|
|
path: path,
|
|
os: os,
|
|
child_process: child_process,
|
|
util: util,
|
|
alert: alert,
|
|
$: {
|
|
get: $.get,
|
|
post: $.post,
|
|
ajax: $.ajax
|
|
}
|
|
}
|
|
shortCodes.forEach(f => {
|
|
sandbox[f.name] = f
|
|
})
|
|
return sandbox
|
|
}
|
|
|
|
runCodeInVm = (cmd, cb) => {
|
|
const vm = new NodeVM({
|
|
require: {
|
|
external: true,
|
|
builtin: ["*"],
|
|
},
|
|
console: 'redirect',
|
|
env: process.env,
|
|
sandbox: getSandboxFuns()
|
|
});
|
|
|
|
var parseItem = item => {
|
|
if (typeof (item) == "object") {
|
|
if (Buffer.isBuffer(item)) {
|
|
var bufferString = `[Buffer ${item.slice(0, 50).toString('hex').match(/\w{1,2}/g).join(" ")}`
|
|
if (item.length > 50) bufferString += `... ${(item.length / 1000).toFixed(2)}kb`
|
|
return bufferString + ']'
|
|
} else {
|
|
try {
|
|
var cache = [];
|
|
var string = JSON.stringify(item, (key, value) => {
|
|
if (typeof value === 'object' && value !== null) {
|
|
if (cache.indexOf(value) !== -1) return
|
|
cache.push(value);
|
|
}
|
|
return value;
|
|
}, '\t')
|
|
if (string != "{}") return string
|
|
} catch (error) { }
|
|
}
|
|
} else if (typeof (item) == "undefined") {
|
|
return "undefined"
|
|
}
|
|
return item.toString()
|
|
}
|
|
|
|
//重定向 console
|
|
vm.on('console.log', stdout => {
|
|
cb(parseItem(stdout), null)
|
|
});
|
|
|
|
vm.on('console.error', stderr => {
|
|
cb(null, stderr.toString())
|
|
});
|
|
|
|
// 错误处理
|
|
try {
|
|
vm.run(cmd, path.join(__dirname, 'preload.js'));
|
|
} catch (error) {
|
|
cb(null, error.toString())
|
|
}
|
|
|
|
let cbUnhandledError = e => {
|
|
window.removeEventListener('error', cbUnhandledError)
|
|
cb(null, e.error.toString())
|
|
}
|
|
|
|
let cbUnhandledRejection = e => {
|
|
window.removeEventListener('unhandledrejection', cbUnhandledRejection)
|
|
cb(null, e.reason.toString())
|
|
}
|
|
|
|
window.addEventListener('error', cbUnhandledError)
|
|
window.addEventListener('unhandledrejection', cbUnhandledRejection);
|
|
}
|
|
|
|
// shell 以环境变量下命令作为代码提示
|
|
getShellCommand = () => {
|
|
var bin = localStorage['shellcommand']
|
|
if (bin) {
|
|
bin = JSON.parse(bin)
|
|
} else {
|
|
bin = []
|
|
if (!utools.isWindows()) {
|
|
process.env.PATH.split(':').forEach(d => {
|
|
try {
|
|
bin = bin.concat(fs.readdirSync(d).filter(x => x[0] != "."))
|
|
} catch (e) { }
|
|
})
|
|
localStorage['shellcommand'] = JSON.stringify(bin)
|
|
}
|
|
}
|
|
return bin
|
|
}
|
|
|
|
// cmd 以环境变量下命令作为代码提示
|
|
getCmdCommand = () => {
|
|
var bin = localStorage['cmdcommand']
|
|
if (bin) {
|
|
bin = JSON.parse(bin)
|
|
} else {
|
|
bin = []
|
|
if (utools.isWindows()) {
|
|
process.env.Path.split(';').forEach(d => {
|
|
try {
|
|
bin = bin.concat(fs.readdirSync(d).filter(x => x.length > 4 && x.slice(-4) == '.exe'))
|
|
} catch (e) { }
|
|
})
|
|
bin = bin.concat(bin).join("|").replace(/\.exe/g, '').split("|")
|
|
localStorage['cmdcommand'] = JSON.stringify(bin)
|
|
}
|
|
}
|
|
return bin
|
|
}
|
|
|
|
// NodeJs 代码提示
|
|
getNodeJsCommand = () => {
|
|
var obj = getSandboxFuns()
|
|
obj.Buffer = Buffer
|
|
return obj
|
|
}
|
|
|
|
htmlEncode = (value, raw) => {
|
|
return raw ? String(value).replace(/&/g, "&").replace(/>/g, ">").replace(/</g, "<").replace(/"/g, """) : value
|
|
}
|
|
|
|
py_beautify = (code, cb) => {
|
|
var file = getQuickCommandScriptFile('py')
|
|
fs.writeFile(file, code, { encoding: 'utf8' }, err => {
|
|
var cmd = `python "${fofoCommon.GetFilePath('assets/plugins', 'autopep8.py')}" "${file}"`
|
|
child_process.exec(cmd, { encoding: "buffer" }, (err, stdout, stderr) => {
|
|
var codec = utools.isWindows() ? 'cp936' : 'utf8'
|
|
cb(iconv.decode(stdout, codec).trim())
|
|
})
|
|
})
|
|
}
|
|
|
|
getQuickCommandScriptFile = ext => {
|
|
return path.join(os.tmpdir(), `QuickCommandTempScript.${ext}`)
|
|
}
|
|
|
|
getBase64Ico = path => {
|
|
return fs.readFileSync(path, 'base64');
|
|
}
|
|
|
|
openFileInDialog = (options, readfile) => {
|
|
var dialog = utools.showOpenDialog(options);
|
|
if (!dialog) return false
|
|
var file = dialog[0]
|
|
var information = {
|
|
name: path.basename(file),
|
|
path: file
|
|
}
|
|
if(readfile) information.data = fs.readFileSync(file)
|
|
return information
|
|
}
|
|
|
|
saveFile = (options, content) => {
|
|
var filename = utools.showSaveDialog(options)
|
|
filename && fs.writeFile(filename, content, 'utf8', err => {
|
|
err && console.log(err)
|
|
})
|
|
}
|
|
|
|
// 保存剪贴板
|
|
// storeClip = () => {
|
|
// var formats = electron.clipboard.availableFormats("clipboard");
|
|
// if (formats.includes("text/plain")) {
|
|
// return ['text', electron.clipboard.readText()]
|
|
// }
|
|
// if (formats.includes("image/png") || formats.includes("image/jpeg")) {
|
|
// return ['image', electron.clipboard.readImage()]
|
|
// }
|
|
// var file;
|
|
// if (utools.isWindows()) {
|
|
// file = electron.clipboard.readBuffer('FileNameW').toString('ucs2').replace(/\\/g, '/');
|
|
// file = file.replace(new RegExp(String.fromCharCode(0), 'g'), '');
|
|
// } else {
|
|
// file = electron.clipboard.read('public.file-url').replace('file://', '');
|
|
// }
|
|
// if (file) {
|
|
// return ['file', file]
|
|
// }
|
|
// return []
|
|
// }
|
|
|
|
// 恢复剪贴板
|
|
// restoreClip = historyData => {
|
|
// if (historyData[0] == 'text') {
|
|
// electron.clipboard.writeText(historyData[1]);
|
|
// return
|
|
// }
|
|
// if (historyData[0] == 'image') {
|
|
// electron.clipboard.writeImage(historyData[1]);
|
|
// return
|
|
// }
|
|
// if (historyData[0] == 'file') {
|
|
// utools.copyFile(historyData[1])
|
|
// return
|
|
// }
|
|
// electron.clipboard.writeText('')
|
|
// }
|
|
|
|
// getSelectText = () => {
|
|
// var historyData = storeClip();
|
|
// electron.clipboard.writeText('');
|
|
// quickcommand.simulateCopy();
|
|
// var selectText = electron.clipboard.readText()
|
|
// setTimeout(() => {
|
|
// restoreClip(historyData)
|
|
// }, 500);
|
|
// return selectText
|
|
// }
|
|
|
|
getSelectFile = hwnd =>
|
|
new Promise((reslove, reject) => {
|
|
if (utools.isWindows()) {
|
|
var cmd = `powershell.exe -NoProfile "(New-Object -COM 'Shell.Application').Windows() | Where-Object { $_.HWND -eq ${hwnd} } | Select-Object -Expand Document | select @{ n='SelectItems'; e={$_.SelectedItems()} } | select -Expand SelectItems | select -Expand Path "`;
|
|
child_process.exec(cmd, { encoding: "buffer" }, (err, stdout, stderr) => {
|
|
if (err) reject(stderr)
|
|
reslove(iconv.decode(stdout, 'GBK').trim().replace(/\\/g, '/'));
|
|
})
|
|
} else {
|
|
var cmd = `osascript -e 'tell application "Finder" to set selectedItems to selection as alias list
|
|
if selectedItems is {} then return
|
|
set parentPath to do shell script "dirname " & quoted form of POSIX path of (item 1 of selectedItems)
|
|
set pathData to ""
|
|
repeat with theItem in selectedItems
|
|
set pathData to pathData & POSIX path of theItem & linefeed
|
|
end repeat
|
|
'
|
|
`
|
|
child_process.exec(cmd, (err, stdout, stderr) => {
|
|
if (err) reject(stderr)
|
|
reslove(stdout.trim());
|
|
});
|
|
}
|
|
})
|
|
|
|
special = cmd => {
|
|
// 判断是否 windows 系统
|
|
if (cmd.includes('{{isWin}}')) {
|
|
let repl = utools.isWindows() ? 1 : 0;
|
|
cmd = cmd.replace(/\{\{isWin\}\}/mg, repl)
|
|
}
|
|
|
|
// 获取本机唯一ID
|
|
if (cmd.includes('{{LocalId}}')) {
|
|
let repl = utools.getLocalId();
|
|
cmd = cmd.replace(/\{\{LocalId\}\}/mg, repl)
|
|
}
|
|
|
|
// 获取浏览器当前链接
|
|
if (cmd.includes('{{BrowserUrl}}')) {
|
|
let repl = utools.getCurrentBrowserUrl();
|
|
cmd = cmd.replace(/\{\{BrowserUrl\}\}/mg, repl)
|
|
}
|
|
|
|
// 获取剪切板的文本
|
|
if (cmd.includes('{{ClipText}}')) {
|
|
let repl = electron.clipboard.readText();
|
|
cmd = cmd.replace(/\{\{ClipText\}\}/mg, repl)
|
|
}
|
|
|
|
// 获取选中的文本
|
|
// if (cmd.includes('{{SelectText}}')) {
|
|
// let repl = getSelectText();
|
|
// cmd = cmd.replace(/\{\{SelectText\}\}/mg, repl)
|
|
// }
|
|
return cmd;
|
|
}
|
|
|
|
runCodeFile = (cmd, option, terminal, callback) => {
|
|
var bin = option.bin,
|
|
argv = option.argv,
|
|
ext = option.ext;
|
|
let script = getQuickCommandScriptFile(ext)
|
|
// 批处理和 powershell 默认编码为 GBK, 解决批处理的换行问题
|
|
if (ext == 'bat' || ext == 'ps1') cmd = iconv.encode(cmd.replace(/\n/g, '\r\n'), 'GBK');
|
|
fs.writeFileSync(script, cmd);
|
|
// var argvs = [script]
|
|
// if (argv) {
|
|
// argvs = argv.split(' ')
|
|
// argvs.push(script);
|
|
// }
|
|
var child, cmdline
|
|
if (bin.slice(-7) == 'csc.exe') {
|
|
cmdline = `${bin} ${argv} /out:"${script.slice(0, -2) + 'exe'}" "${script}" && "${script.slice(0, -2) + 'exe'}"`
|
|
} else if (bin == 'gcc') {
|
|
var suffix = utools.isWindows() ? '.exe' : ''
|
|
cmdline = `${bin} ${argv} "${script.slice(0, -2)}" "${script}" && "${script.slice(0, -2) + suffix}"`
|
|
} else if (utools.isWindows() && bin == 'bash') {
|
|
cmdline = `${bin} ${argv} "${script.replace(/\\/g, '/').replace(/C:/i, '/mnt/c')}"`
|
|
} else {
|
|
cmdline = `${bin} ${argv} "${script}"`
|
|
}
|
|
// 在终端中输出
|
|
if (terminal) {
|
|
if (utools.isWindows()) {
|
|
if (bin.slice(-7) == 'csc.exe' || bin == 'gcc') {
|
|
cmdline = cmdline.split("&&")
|
|
cmdline = cmdline[0] + "&& start cmd /k " + cmdline[1]
|
|
} else {
|
|
cmdline = `start cmd /k ${cmdline}`
|
|
}
|
|
} else if(utools.isMacOs()){
|
|
var appleScript = `if application "Terminal" is running then
|
|
tell application "Terminal"
|
|
do script "clear;${cmdline.replace(/"/g, `\\"`)}"
|
|
activate
|
|
end tell
|
|
else
|
|
tell application "Terminal"
|
|
do script "clear;${cmdline.replace(/"/g, `\\"`)}" in window 1
|
|
activate
|
|
end tell
|
|
end if`;
|
|
cmdline = `osascript -e '${appleScript}'`
|
|
} else {
|
|
return message('Linux 不支持在终端输出')
|
|
}
|
|
}
|
|
child = child_process.spawn(cmdline, { encoding: 'buffer', shell: true })
|
|
// var chunks = [],
|
|
// err_chunks = [];
|
|
console.log('running: ' + cmdline);
|
|
child.stdout.on('data', chunk => {
|
|
if (option.codec) chunk = iconv.decode(chunk, option.codec)
|
|
callback(chunk, null)
|
|
// chunks.push(chunk)
|
|
})
|
|
child.stderr.on('data', stderr => {
|
|
if (option.codec) stderr = iconv.decode(stderr, option.codec)
|
|
callback(null, stderr)
|
|
// err_chunks.push(err_chunk)
|
|
})
|
|
// child.on('close', code => {
|
|
// let stdout = chunks.join("");
|
|
// let stderr = err_chunks.join("");
|
|
// callback(stdout, stderr)
|
|
// })
|
|
} |