100 lines
2.6 KiB
JavaScript
100 lines
2.6 KiB
JavaScript
|
/* eslint-disable complexity */
|
||
|
|
'use strict';
|
||
|
|
|
||
|
|
var getAll = require('./get');
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Get the list of children and grandchildren pids of the given PID.
|
||
|
|
* @param {Number|String} PID A PID. If -1 will return all the pids.
|
||
|
|
* @param {Object} [options] Optional options object.
|
||
|
|
* @param {Boolean} [options.root=false] Include the provided PID in the list.
|
||
|
|
* @param {Boolean} [options.advanced=false] Returns a list of objects in the
|
||
|
|
* format {pid: X, ppid: Y}.
|
||
|
|
* @param {pidCallback} callback Called when the list is ready.
|
||
|
|
*/
|
||
|
|
function list(PID, options, callback) {
|
||
|
|
if (typeof options === 'function') {
|
||
|
|
callback = options;
|
||
|
|
options = {};
|
||
|
|
}
|
||
|
|
if (typeof options !== 'object') {
|
||
|
|
options = {};
|
||
|
|
}
|
||
|
|
|
||
|
|
PID = parseInt(PID, 10);
|
||
|
|
if (isNaN(PID) || PID < -1) {
|
||
|
|
callback(new TypeError('The pid provided is invalid'));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
getAll(function(err, list) {
|
||
|
|
if (err) {
|
||
|
|
callback(err);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
// If the user wants the whole list just return it
|
||
|
|
if (PID === -1) {
|
||
|
|
for (var i = 0; i < list.length; i++) {
|
||
|
|
list[i] = options.advanced
|
||
|
|
? {ppid: list[i][0], pid: list[i][1]}
|
||
|
|
: (list[i] = list[i][1]);
|
||
|
|
}
|
||
|
|
callback(null, list);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
var root;
|
||
|
|
for (var l = 0; l < list.length; l++) {
|
||
|
|
if (list[l][1] === PID) {
|
||
|
|
root = options.advanced ? {ppid: list[l][0], pid: PID} : PID;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
if (list[l][0] === PID) {
|
||
|
|
root = options.advanced ? {pid: PID} : PID; // Special pids like 0 on *nix
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (!root) {
|
||
|
|
callback(new Error('No maching pid found'));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Build the adiacency Hash Map (pid -> [children of pid])
|
||
|
|
var tree = {};
|
||
|
|
while (list.length > 0) {
|
||
|
|
var e = list.pop();
|
||
|
|
if (tree[e[0]]) {
|
||
|
|
tree[e[0]].push(e[1]);
|
||
|
|
} else {
|
||
|
|
tree[e[0]] = [e[1]];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Starting by the PID provided by the user, traverse the tree using the
|
||
|
|
// adiacency Hash Map until the whole subtree is visited.
|
||
|
|
// Each pid encountered while visiting is added to the pids array.
|
||
|
|
var idx = 0;
|
||
|
|
var pids = [root];
|
||
|
|
while (idx < pids.length) {
|
||
|
|
var curpid = options.advanced ? pids[idx++].pid : pids[idx++];
|
||
|
|
if (!tree[curpid]) continue;
|
||
|
|
var len = tree[curpid].length;
|
||
|
|
for (var j = 0; j < len; j++) {
|
||
|
|
pids.push(
|
||
|
|
options.advanced
|
||
|
|
? {ppid: curpid, pid: tree[curpid][j]}
|
||
|
|
: tree[curpid][j]
|
||
|
|
);
|
||
|
|
}
|
||
|
|
delete tree[curpid];
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!options.root) {
|
||
|
|
pids.shift(); // Remove root
|
||
|
|
}
|
||
|
|
callback(null, pids);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = list;
|