123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170
const { build, run, readAll } = Deno;
export interface Process {  command: string; // Command to run this process  ppid: number; // The parent process ID of the process  pid: number; // Process ID  stat: string; // Process status  children?: Process[];}
export interface KillOptions {  force?: boolean;  ignoreCase?: boolean;  tree?: boolean;}
/** * Get the single process information. * Requires `--allow-run` flag * @param pid */export async function get(pid: number): Promise<Process | void> {  return (await getAll()).find((v) => v.pid === pid);}
/** * Get process list * Requires `--allow-run` flag */export async function getAll(): Promise<Process[]> {  const commands = build.os == "windows"    ? ["wmic.exe", "PROCESS", "GET", "Name,ProcessId,ParentProcessId,Status"]    : ["ps", "-A", "-o", "comm,ppid,pid,stat"];
  const ps = run({    cmd: commands,    stdout: "piped",  });
  const output = new TextDecoder().decode(await readAll(ps.stdout!));
  const { success, code } = await ps.status();
  ps.stdout?.close();
  ps.close();
  if (!success || code !== 0) {    throw new Error("Fail to get process.");  }
  const lines = output.split("\n").filter((v: string): string => v.trim());  lines.shift();
  const processList: Process[] = lines.map(    (line: string): Process => {      const columns = line.trim().split(/\s+/);      return {        command: columns[0],        ppid: +columns[1],        pid: +columns[2],        stat: columns[3],      };    },  );
  return processList;}
/** * Get process tree * Requires `--allow-run` flag */export async function getTree(): Promise<Process[]> {  const items = await getAll();  const nest = (items: Process[], pid: number = 0): Process[] => {    return items      .filter((item) => item.ppid === pid)      .map((item) => {        const children = nest(items, item.pid);        if (!children.length) {          return item;        } else {          return { ...item, children };        }      }) as Process[];  };
  return nest(items);}
function getKillCommand(  pidOrName: number | string,  options: KillOptions = {},): string[] {  const killByName = typeof pidOrName === "string";  if (build.os === "windows") {    const commands = ["taskkill"];
    if (options.force) {      commands.push("/f");    }
    if (options.tree) {      commands.push("/t");    }
    commands.push(killByName ? "/im" : "/pid", pidOrName + "");
    return commands;  } else if (build.os === "linux") {    const commands = [killByName ? "killall" : "kill"];
    if (options.force) {      commands.push("-9");    }
    if (killByName && options.ignoreCase) {      commands.push("-I");    }
    commands.push(pidOrName + "");
    return commands;  } else {    const commands = [killByName ? "pkill" : "kill"];
    if (options.force) {      commands.push("-9");    }
    if (killByName && options.ignoreCase) {      commands.push("-i");    }
    commands.push(pidOrName + "");
    return commands;  }}
/** * kill process * Requires `--allow-run` flag * @param pidOrName pid or process name * @param options */export async function kill(  pidOrName: number | string,  options: KillOptions = {},): Promise<void> {  const commands = getKillCommand(pidOrName, options);
  const ps = run({    cmd: commands,    stderr: "piped",  });
  const { success, code } = await ps.status();
  ps.stderr?.close();
  ps.close();
  if (!success || code !== 0) {    const msg = new TextDecoder().decode(await readAll(ps.stderr!));    throw new Error(msg || "exit with code: " + code);  }}