use Get-Process instead of tasklist

This commit is contained in:
fofolee 2019-03-27 12:18:54 +08:00
parent 79ab0912e4
commit 20444beb82

View File

@ -1,26 +1,59 @@
const exec = require('child_process').exec; const fs = require('fs');
const iconv = require('iconv-lite'); const path = require("path");
const iconExtractor = require('icon-extractor');
const os = require('os')
const PowerShell = require("powershell");
window.tasklist = (callback) => {
// if(process.platform === 'win32'){} getico = tasks =>{
let cmd = 'tasklist /FO csv /NH /V' iconExtractor.emitter.on('icon', function (data) {
exec(cmd, function (err, stdout, stderr) { let icondir = path.join(os.tmpdir(), 'ProcessIcon')
let tasklist = []; fs.exists(icondir, exists => {
let lines = stdout.trim().split('\n'); if (!exists) { fs.mkdirSync(icondir) }
for (var line of lines){ let iconpath = path.join(icondir, `${data.Context}.png`)
tasklist.push(line.trim().split(',')); fs.exists(iconpath, exists => {
} if (!exists) {
callback(tasklist); fs.writeFile(iconpath, data.Base64ImageData, "base64", err => {
}) if (err) { console.log(err); }
})
}
})
})
});
for (var task of tasks) {
iconExtractor.getIcon(task.ProcessName, task.Path);
}
} }
window.taskkill = (taskname, callback) => { tasklist = (callback) => {
let cmd = 'TASKKILL /F /IM "' + taskname + '" /T' let ps = new PowerShell("chcp 65001;Get-Process | Format-List ProcessName,Path,Description");
exec(cmd, { encoding: "buffer"}, (err, stdout, stderr) => { ps.on("output", data => {
if (err) { let tasklist = [];
callback(iconv.decode(stderr,'cp936')); let tasks = data.trim().split('\r\n\r\n');
return; for (var task of tasks) {
dict = {}
let lines = task.split('\r\n')
for (var line of lines) {
if (line) {
let key = line.split(/\s+:\s*/)[0];
let value = line.split(/\s+:\s*/)[1];
dict[key] = value;
}
}
var icon = path.join(os.tmpdir(), 'ProcessIcon', `${dict.ProcessName}.png`);
dict.Icon = icon
tasklist.push(dict);
} }
callback(iconv.decode(stdout,'cp936')); tasklist.shift();
}) getico(tasklist);
callback(tasklist);
});
}
taskkill = (taskname, callback) => {
let ps = new PowerShell(`chcp 65001;Stop-Process -Name ${taskname}`);
ps.on("error-output", data => {
callback(data.split('\n')[0])
});
} }