mirror of
https://github.com/fofolee/uTools-quickcommand.git
synced 2025-06-08 05:55:34 +08:00
278 lines
9.1 KiB
JavaScript
278 lines
9.1 KiB
JavaScript
const fs = require('fs');
|
|
const os = require('os');
|
|
const { spawn, exec } = require("child_process")
|
|
const iconv = require('iconv-lite')
|
|
const { clipboard } = require('electron')
|
|
const robot = utools.robot
|
|
// const jschardet = require("jschardet")
|
|
|
|
const path = require("path")
|
|
const { dialog, BrowserWindow, nativeImage } = require('electron').remote
|
|
const { shell } = require('electron');
|
|
|
|
pluginInfo = JSON.parse(fs.readFileSync(path.join(__dirname, 'plugin.json')));
|
|
logo = nativeImage.createFromPath(path.join(__dirname, 'logo.png'));
|
|
|
|
// fix PATH
|
|
process.env.PATH += ':/usr/local/bin:/usr/local/sbin'
|
|
|
|
messageBox = (options, callback) => {
|
|
dialog.showMessageBox(BrowserWindow.getFocusedWindow(), options, index => {
|
|
utools.showMainWindow()
|
|
callback(index)
|
|
})
|
|
}
|
|
|
|
open = url => {
|
|
shell.openExternal(url);
|
|
}
|
|
// ------------------------
|
|
|
|
isWin = os.platform() == 'win32' ? true : false;
|
|
|
|
isDev = /unsafe-\w+\.asar/.test(__dirname) ? false : true
|
|
|
|
basename = path.basename;
|
|
|
|
dirname = __dirname;
|
|
|
|
resolve = path.resolve;
|
|
|
|
exists = fs.existsSync;
|
|
|
|
getBase64Ico = path => {
|
|
return fs.readFileSync(path, 'base64');
|
|
}
|
|
|
|
openFolder = options => {
|
|
return dialog.showOpenDialogSync(BrowserWindow.getFocusedWindow(), options);
|
|
}
|
|
|
|
saveFile = (options, content) => {
|
|
dialog.showSaveDialog(BrowserWindow.getFocusedWindow(), options, filename => {
|
|
filename && fs.writeFile(filename, content, 'utf8', err => {
|
|
err && console.log(err)
|
|
})
|
|
})
|
|
}
|
|
|
|
copy = () => {
|
|
var ctlKey = isWin ? 'control' : 'command';
|
|
robot.keyTap('c', ctlKey);
|
|
robot.setKeyboardDelay(20);
|
|
}
|
|
|
|
copyTo = text => {
|
|
clipboard.writeText(text)
|
|
}
|
|
|
|
paste = () => {
|
|
var ctlKey = isWin ? 'control' : 'command';
|
|
robot.keyTap('v', ctlKey);
|
|
}
|
|
|
|
// 保存剪贴板
|
|
storeClip = () => {
|
|
var formats = clipboard.availableFormats("clipboard");
|
|
if (formats.includes("text/plain")) {
|
|
return ['text', clipboard.readText()]
|
|
}
|
|
if (formats.includes("image/png") || formats.includes("image/jpeg")) {
|
|
return ['image', clipboard.readImage()]
|
|
}
|
|
var file;
|
|
if (isWin) {
|
|
file = clipboard.readBuffer('FileNameW').toString('ucs2').replace(/\\/g, '/');
|
|
file = file.replace(new RegExp(String.fromCharCode(0), 'g'), '');
|
|
} else {
|
|
file = clipboard.read('public.file-url').replace('file://', '');
|
|
}
|
|
if (file) {
|
|
return ['file', file]
|
|
}
|
|
return []
|
|
}
|
|
|
|
// 恢复剪贴板
|
|
restoreClip = historyData => {
|
|
if (historyData[0] == 'text') {
|
|
clipboard.writeText(historyData[1]);
|
|
return
|
|
}
|
|
if (historyData[0] == 'image') {
|
|
clipboard.writeImage(historyData[1]);
|
|
return
|
|
}
|
|
if (historyData[0] == 'file') {
|
|
utools.copyFile(historyData[1])
|
|
return
|
|
}
|
|
clipboard.writeText('')
|
|
}
|
|
|
|
getSelectText = () => {
|
|
var historyData = storeClip();
|
|
clipboard.writeText('');
|
|
copy();
|
|
var selectText = clipboard.readText()
|
|
restoreClip(historyData)
|
|
return selectText
|
|
}
|
|
|
|
getSelectFile = hwnd =>
|
|
new Promise((reslove, reject) => {
|
|
if (isWin) {
|
|
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 "`;
|
|
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
|
|
'
|
|
`
|
|
exec(cmd, (err, stdout, stderr) => {
|
|
if (err) reject(stderr)
|
|
reslove(stdout.trim());
|
|
});
|
|
}
|
|
})
|
|
|
|
// GetBinPath = () => {
|
|
// if (isDev) {
|
|
// return path.join(__dirname.replace(/(unsafe-\w+\.asar)/,'$1.unpacked'), 'bin', 'GetForegroundWindow.exe')
|
|
// } else {
|
|
// return path.join(__dirname, 'bin', 'GetForegroundWindow.exe')
|
|
// }
|
|
// }
|
|
|
|
// 获取前台窗口句柄
|
|
// GetForegroundWindow = callback =>
|
|
// exec(`"${GetBinPath()}"`, (error, stdout, stderr) => {
|
|
// callback(stdout);
|
|
// });
|
|
|
|
|
|
pwd = hwnd =>
|
|
new Promise((reslove, reject) => {
|
|
if (isWin) {
|
|
var cmd = `powershell.exe -NoProfile "((New-Object -COM 'Shell.Application').Windows() | Where-Object { $_.HWND -eq (${hwnd}) } | Select-Object -Expand LocationURL).replace('file:///','')"`;
|
|
exec(cmd, { encoding: "buffer" }, (err, stdout, stderr) => {
|
|
if (err) {
|
|
console.log(iconv.decode(stderr, 'GBK'));
|
|
reslove(`${os.homedir().replace(/\\/g, '/')}/Desktop`)
|
|
} else {
|
|
reslove(decodeURIComponent(iconv.decode(stdout, 'GBK').trim()));
|
|
}
|
|
});
|
|
} else {
|
|
var cmd = `osascript -e 'tell application "Finder" to get the POSIX path of (target of front window as alias)'`
|
|
exec(cmd, (err, stdout, stderr) => {
|
|
if (err) reject(stderr)
|
|
reslove(stdout.trim());
|
|
});
|
|
}
|
|
});
|
|
|
|
special = async cmd => {
|
|
// 判断是否 windows 系统
|
|
if (cmd.includes('{{isWin}}')) {
|
|
let repl = isWin ? 1 : 0;
|
|
cmd = cmd.replace(/\{\{isWin\}\}/mg, repl)
|
|
}
|
|
|
|
// 获取本机唯一ID
|
|
if (cmd.includes('{{LocalID}}')) {
|
|
let repl = utools.getLocalId();
|
|
cmd = cmd.replace(/\{\{HostName\}\}/mg, repl)
|
|
}
|
|
|
|
// 获取浏览器当前链接
|
|
if (cmd.includes('{{BrowserUrl}}')) {
|
|
let repl = utools.getCurrentBrowserUrl();
|
|
cmd = cmd.replace(/\{\{BrowserUrl\}\}/mg, repl)
|
|
}
|
|
|
|
// 获取剪切板的文本
|
|
if (cmd.includes('{{ClipText}}')) {
|
|
let repl = clipboard.readText();
|
|
cmd = cmd.replace(/\{\{ClipText\}\}/mg, repl)
|
|
}
|
|
|
|
// 获取选中的文本
|
|
if (cmd.includes('{{SelectText}}')) {
|
|
let repl = getSelectText();
|
|
cmd = cmd.replace(/\{\{SelectText\}\}/mg, repl)
|
|
}
|
|
return cmd;
|
|
}
|
|
|
|
run = async (cmd, option, codec, terminal, callback) => {
|
|
var tmp = os.tmpdir(),
|
|
bin = option.bin,
|
|
argv = option.argv,
|
|
ext = option.ext;
|
|
cmd = await special(cmd);
|
|
let script = path.join(tmp, `QuickCommandTempScript.${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;
|
|
if (bin) {
|
|
// 在终端中输出
|
|
if (terminal) {
|
|
if (isWin) {
|
|
child = spawn('cmd', ['/c', 'start', 'cmd', '/k', bin].concat(argvs), { encoding: 'buffer' })
|
|
} else {
|
|
var appleScript = `if application "Terminal" is running then
|
|
tell application "Terminal"
|
|
# do script without "in window" will open a new window
|
|
do script "clear;${bin} ${argv} ${script}"
|
|
activate
|
|
end tell
|
|
else
|
|
tell application "Terminal"
|
|
# window 1 is guaranteed to be recently opened window
|
|
do script "clear;${bin} ${argv} ${script}" in window 1
|
|
activate
|
|
end tell
|
|
end if`;
|
|
child = spawn('osascript', ['-e', appleScript], { encoding: 'buffer' })
|
|
}
|
|
} else {
|
|
child = spawn(bin, argvs, { encoding: 'buffer' })
|
|
}
|
|
} else {
|
|
if (terminal) {
|
|
child = spawn('cmd', ['/c', 'start', 'cmd', '/k', script], { encoding: 'buffer' })
|
|
} else {
|
|
child = spawn(script, { encoding: 'buffer' })
|
|
}
|
|
}
|
|
var chunks = [],
|
|
err_chunks = [];
|
|
child.stdout.on('data', chunk => {
|
|
if (isWin) chunk = iconv.decode(chunk, 'GBK')
|
|
chunks.push(chunk)
|
|
})
|
|
child.stderr.on('data', err_chunk => {
|
|
if (isWin) err_chunk = iconv.decode(err_chunk, 'GBK')
|
|
err_chunks.push(err_chunk)
|
|
})
|
|
child.on('close', code => {
|
|
let stdout = chunks.join("");
|
|
let stderr = err_chunks.join("");
|
|
callback(stdout, stderr)
|
|
})
|
|
} |