S.P.L.U.R.T-Station-13/tgui/packages/tgui-dev-server/dreamseeker.js

/**
 * @file
 * @copyright 2020 Aleksej Komarov
 * @license MIT
 */

import { exec } from 'child_process';
import { promisify } from 'util';
import { createLogger } from './logging.js';
import { require } from './require.js';

const axios = require('axios');
const logger = createLogger('dreamseeker');

const instanceByPid = new Map();

export class DreamSeeker {
  constructor(pid, addr) {
    this.pid = pid;
    this.addr = addr;
    this.client = axios.create({
      baseURL: `http://${addr}/`,
    });
  }

  topic(params = {}) {
    const query = Object.keys(params)
      .map(key => encodeURIComponent(key)
        + '=' + encodeURIComponent(params[key]))
      .join('&');
    logger.log(
      `topic call at ${this.client.defaults.baseURL + '/dummy?' + query}`
    );
    return this.client.get('/dummy?' + query);
  }
}

/**
 * @param {number[]} pids
 * @returns {DreamSeeker[]}
 */
DreamSeeker.getInstancesByPids = async pids => {
  if (process.platform !== 'win32') {
    return [];
  }
  const instances = [];
  const pidsToResolve = [];
  for (let pid of pids) {
    const instance = instanceByPid.get(pid);
    if (instance) {
      instances.push(instance);
    }
    else {
      pidsToResolve.push(pid);
    }
  }
  if (pidsToResolve.length > 0) {
    try {
      const command = 'netstat -ano | findstr TCP | findstr 0.0.0.0:0';
      const { stdout } = await promisify(exec)(command, {
        // Max buffer of 1MB (default is 200KB)
        maxBuffer: 1024 * 1024,
      });
      // Line format:
      // proto addr mask mode pid
      const entries = [];
      const lines = stdout.split('\r\n');
      for (let line of lines) {
        const words = line.match(/\S+/g);
        if (!words || words.length === 0) {
          continue;
        }
        const entry = {
          addr: words[1],
          pid: parseInt(words[4], 10),
        };
        if (pidsToResolve.includes(entry.pid)) {
          entries.push(entry);
        }
      }
      const len = entries.length;
      logger.log('found', len, plural('instance', len));
      for (let entry of entries) {
        const { pid, addr } = entry;
        const instance = new DreamSeeker(pid, addr);
        instances.push(instance);
        instanceByPid.set(pid, instance);
      }
    }
    catch (err) {
      if (err.code === 'ERR_CHILD_PROCESS_STDIO_MAXBUFFER') {
        logger.error(err.message, err.code);
      }
      else {
        logger.error(err);
      }
      return [];
    }
  }
  return instances;
};

const plural = (word, n) => n !== 1 ? word + 's' : word;