uTools-quickcommand/src/preload.js

const fs = require('fs');
const os = require('os');
const { spawn, exec, execSync } = require("child_process")
const iconv = require('iconv-lite')
const { clipboard } = require('electron')

const path = require("path")

pluginInfo = JSON.parse(fs.readFileSync(path.join(__dirname, 'plugin.json')));

// fix PATH
process.env.PATH += ':/usr/local/bin:/usr/local/sbin'


open = path => {
    utools.shellOpenItem(path)
}

locate = path => {
    utools.shellShowItemInFolder(path);
}

visit = url => {
    utools.shellOpenExternal(url);
}

system = cmd => {
    return execSync(cmd);
}

message = msg => {
    utools.showNotification(msg)
}

keyTap = utools.simulateKeyboardTap

sleep = ms => new Promise((r, j) => setTimeout(r, ms))

readFile = fs.readFileSync

writeFile = fs.writeFileSync
// ------------------------

isWin = os.platform() == 'win32' ? true : false;

isDev = /[a-zA-Z0-9\-]+\.asar/.test(__dirname) ? false : true

basename = path.basename;
dirname = __dirname;
resolve = path.resolve;
exists = fs.existsSync;
tmpdir = os.tmpdir(),

getBase64Ico = path => {
    return fs.readFileSync(path, 'base64');
}

openFolder = options => {
    return utools.showOpenDialog(options);
}

saveFile = (options, content) => {
    var filename = utools.showSaveDialog(options)
    filename && fs.writeFile(filename, content, 'utf8', err => {
        err && console.log(err)
    })
}

copy = () => {
    var ctlKey = isWin ? 'control' : 'command';
    utools.simulateKeyboardTap('c', ctlKey);
}

copyTo = text => {
    clipboard.writeText(text)
}

paste = () => {
    var ctlKey = isWin ? 'control' : 'command';
    utools.simulateKeyboardTap('v', ctlKey);
}

send = text => {
    var historyData = storeClip();
    copyTo(text);
    paste();
    setTimeout(() => {
        restoreClip(historyData);
    }, 500);
}


// 保存剪贴板
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()
    setTimeout(() => {
        restoreClip(historyData)
    }, 500);
    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());
            });
        }
    })   

// 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 = 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(/\{\{LocalId\}\}/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 = (cmd, option, terminal, callback) => {
    var bin = option.bin,
        argv = option.argv,
        ext = option.ext;
    let script = path.join(tmpdir, `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 (option.codec) chunk = iconv.decode(chunk, option.codec)
        chunks.push(chunk)
    })
    child.stderr.on('data', err_chunk => {
        if (option.codec) err_chunk = iconv.decode(err_chunk, option.codec)
        err_chunks.push(err_chunk)
    })
    child.on('close', code => {
        let stdout = chunks.join("");
        let stderr = err_chunks.join("");
        callback(stdout, stderr)
    })
}