btake 0.0.1

package/README.md

@@ -0,0 +1 @@
+# insight

package/bin/index.js

@@ -0,0 +1,121 @@
+#!/usr/bin/env node
+const yargs = require('yargs')(process.argv.slice(2));
+const lsofi = require('lsofi');
+const systeminformation = require('systeminformation');
+const { Inspector, ToolName } = require('../lib');
+
+const commonOptions = {
+  p: {
+    alias: 'pid',
+    demandOption: false,
+    describe: 'target pid of your Node.js process.',
+    type: 'number',
+  },
+  a: {
+    alias: 'addr',
+    demandOption: false,
+    describe:
+      'The inspect address of your Node.js process, e.g. 127.0.0.1:9229',
+    type: 'string',
+  },
+};
+
+const options = [];
+
+options.push({
+  command: 'cpuprofile',
+  desc: `${ToolName} cpuprofile -p=<pid> [<args>]\ntake the cpu profile for a process.`,
+  optsFunc: (yargs) => {
+    yargs.options({
+      d: {
+        alias: 'duration',
+        demandOption: false,
+        describe: 'How long the profiling will last, unit: ms.',
+        default: 10000,
+        type: 'number',
+      },
+      ...commonOptions,
+    })
+  },
+  commandFunc: function (argv) {
+    new Inspector(argv).takeCpuProfile();
+  },
+})
+
+options.push({
+  command: 'heapsnapshot',
+  desc: `${ToolName} heapsnapshot -p=<pid> [<args>]\ntake the heapsnapshot of a process.`,
+  optsFunc: (yargs) => {
+    yargs.options({
+      ...commonOptions,
+    })
+  },
+  commandFunc: function (argv) {
+    new Inspector(argv).takeHeapSnapshot(argv);
+  },
+})
+
+options.push({
+    command: 'who',
+    desc: `${ToolName} who -p=<port> [<args>]\nwho is listening the port`,
+    optsFunc: (yargs) => {
+      yargs.options({
+        p: {
+            alias: 'port',
+            type: 'number',
+        },
+      })
+    },
+    commandFunc: async function (argv) {
+        try {
+            const pid = await lsofi(argv.port);
+            if (pid) {
+                console.log(`process(${pid}) is listening the port(${argv.port})`);
+            } else {
+                console.log(`no process is listening the port(${argv.port})`);
+            }
+        } catch(e) {
+            console.error(`${ToolName} error: err: ${e.message}`);
+        }
+    },
+});
+
+options.push({
+    command: 'port',
+    desc: `${ToolName} port p=<pid> [<args>]\ntake the tcp ports for process`,
+    optsFunc: (yargs) => {
+      yargs.options({
+        p: {
+            alias: 'pid',
+            type: 'number',
+        },
+      })
+    },
+    commandFunc: async function (argv) {
+        systeminformation.networkConnections(function(connections) {
+            const ports = connections
+            .filter(conn => {
+                return conn.pid === argv.pid && conn.protocol.startsWith('tcp') && conn.state === 'LISTEN';
+            }).map(conn => ({
+                host: conn.localAddress,
+                port: conn.localPort,
+            }));
+            console.log(`process(${argv.pid}) tcp listen ports: `, ports);
+        });
+    },
+});
+
+  
+options.forEach((option) => {
+  const { commandFunc, desc, command, optsFunc } = option
+  yargs.command(command, desc, optsFunc, commandFunc)
+})
+
+yargs
+  .usage(`usage: ${ToolName} <command> [<args>]`)
+  .scriptName(ToolName)
+  .help()
+  .locale('en')
+  .demandCommand()
+  .strict()
+  .parse()

package/lib/index.d.ts

@@ -0,0 +1,10 @@
+export declare const ToolName: any;
+export declare class Inspector {
+    options: any;
+    constructor(options: any);
+    private openInspector;
+    private genFilename;
+    takeHeapSnapshot(): Promise<void>;
+    takeCpuProfile(): Promise<void>;
+}
+//# sourceMappingURL=index.d.ts.map

package/lib/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAMA,eAAO,MAAM,QAAQ,KAAO,CAAC;AA+B7B,qBAAa,SAAS;IAClB,OAAO,EAAE,GAAG,CAAC;gBAED,OAAO,EAAE,GAAG;IAiB3B,OAAO,CAAC,aAAa;IAgBrB,OAAO,CAAC,WAAW;IAQb,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;IAoBjC,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;CAgBrC"}

package/lib/index.js

@@ -0,0 +1,134 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Inspector = exports.ToolName = void 0;
+const chrome_remote_interface_1 = __importDefault(require("chrome-remote-interface"));
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const { name } = JSON.parse(fs_1.default.readFileSync(path_1.default.join(__dirname, '../package.json'), 'utf8'));
+exports.ToolName = name;
+const closeInspectorScript = `(function() {
+ try {
+  if (typeof process._debugEnd === 'function') {
+   process._debugEnd();
+  } else {
+   require('inspector').close();
+  }
+ } catch (e) {
+ }
+})();`;
+function sleep(ms) {
+    return new Promise(resolve => setTimeout(resolve, ms));
+}
+function retry(task, delay = 100, retries = 3) {
+    return __awaiter(this, void 0, void 0, function* () {
+        let lastError;
+        for (let i = 0; i < retries; i++) {
+            try {
+                return yield task();
+            }
+            catch (error) {
+                lastError = error;
+                yield sleep(delay);
+            }
+        }
+        throw lastError;
+    });
+}
+class Inspector {
+    constructor(options) {
+        this.options = options;
+        if (this.options.addr) {
+            const list = this.options.addr.split(':');
+            if (list.length >= 2) {
+                this.options.host = list[0];
+                this.options.port = Number(list[1]);
+            }
+            else {
+                this.options.host = 'localhost';
+                this.options.port = Number(list[0]);
+            }
+        }
+        else {
+            this.options.host = 'localhost';
+            this.options.port = 9229;
+        }
+    }
+    openInspector() {
+        try {
+            if (this.options.pid) {
+                // @ts-ignore
+                if (typeof process._debugProcess === 'function') {
+                    // @ts-ignore
+                    process._debugProcess(this.options.pid);
+                }
+                else {
+                    process.kill(this.options.pid, 'SIGUSR1');
+                }
+            }
+        }
+        catch (e) {
+            console.error('open inspector failed: err', e.message);
+        }
+    }
+    genFilename(ext) {
+        if (this.options.pid) {
+            return `${name}_${this.options.pid}_${Date.now()}.${ext}`;
+        }
+        else {
+            return `${name}_${this.options.host}_${this.options.port}_${Date.now()}.${ext}`;
+        }
+    }
+    takeHeapSnapshot() {
+        return __awaiter(this, void 0, void 0, function* () {
+            this.openInspector();
+            retry(() => __awaiter(this, void 0, void 0, function* () {
+                const client = yield (0, chrome_remote_interface_1.default)({ port: this.options.port, host: this.options.host });
+                let chunk = '';
+                const handler = (data) => {
+                    chunk += data.chunk;
+                };
+                client.on('HeapProfiler.addHeapSnapshotChunk', handler);
+                yield client.HeapProfiler.takeHeapSnapshot({});
+                // @ts-ignore
+                client.off('HeapProfiler.addHeapSnapshotChunk', handler);
+                yield client.Runtime.evaluate({
+                    includeCommandLineAPI: true,
+                    expression: closeInspectorScript,
+                }).catch(() => { });
+                fs_1.default.writeFileSync(this.genFilename('heapsnapshot'), chunk);
+            }));
+        });
+    }
+    takeCpuProfile() {
+        return __awaiter(this, void 0, void 0, function* () {
+            this.openInspector();
+            retry(() => __awaiter(this, void 0, void 0, function* () {
+                const client = yield (0, chrome_remote_interface_1.default)({ port: this.options.port, host: this.options.host });
+                yield client.Profiler.enable();
+                yield client.Profiler.start();
+                yield sleep(this.options.duration || 1000);
+                const result = yield client.Profiler.stop();
+                yield client.Profiler.disable();
+                yield client.Runtime.evaluate({
+                    includeCommandLineAPI: true,
+                    expression: closeInspectorScript,
+                }).catch(() => { });
+                fs_1.default.writeFileSync(this.genFilename('cpuprofile'), JSON.stringify(result.profile));
+            }));
+        });
+    }
+}
+exports.Inspector = Inspector;
+//# sourceMappingURL=index.js.map

package/lib/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,sFAAyC;AACzC,4CAAoB;AACpB,gDAAwB;AAExB,MAAM,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,YAAE,CAAC,YAAY,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,iBAAiB,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;AAEjF,QAAA,QAAQ,GAAG,IAAI,CAAC;AAE7B,MAAM,oBAAoB,GAAG;;;;;;;;;MASvB,CAAC;AAEP,SAAS,KAAK,CAAC,EAAU;IACxB,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AACxD,CAAC;AAEA,SAAe,KAAK,CAAI,IAAsB,EAAE,KAAK,GAAG,GAAG,EAAE,OAAO,GAAG,CAAC;;QACxE,IAAI,SAAc,CAAC;QACnB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE,EAAE;YACjC,IAAI;gBACH,OAAO,MAAM,IAAI,EAAE,CAAC;aACpB;YAAC,OAAO,KAAK,EAAE;gBACf,SAAS,GAAG,KAAK,CAAC;gBAClB,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;aACnB;SACD;QAED,MAAM,SAAS,CAAC;IACjB,CAAC;CAAA;AAED,MAAa,SAAS;IAGlB,YAAY,OAAY;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE;YACtB,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC1C,IAAI,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE;gBACrB,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;gBAC5B,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;aACpC;iBAAM;gBACN,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,WAAW,CAAC;gBAChC,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;aACpC;SACD;aAAM;YACN,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,WAAW,CAAC;YACvB,IAAI,CAAC,OAAO,CAAC,IAAI,GAAI,IAAI,CAAC;SACnC;IACC,CAAC;IAEI,aAAa;QACd,IAAI;YACT,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE;gBACrB,aAAa;gBACb,IAAI,OAAO,OAAO,CAAC,aAAa,KAAK,UAAU,EAAE;oBAChD,aAAa;oBACb,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;iBACxC;qBAAM;oBACN,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAa,EAAE,SAAS,CAAC,CAAC;iBACpD;aACD;SACD;QAAC,OAAO,CAAC,EAAE;YACX,OAAO,CAAC,KAAK,CAAC,4BAA4B,EAAG,CAAW,CAAC,OAAO,CAAC,CAAC;SAClE;IACF,CAAC;IAEO,WAAW,CAAC,GAAW;QAC9B,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE;YACrB,OAAO,GAAG,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,GAAG,EAAE,CAAC;SAC1D;aAAM;YACN,OAAO,GAAG,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,GAAG,EAAE,CAAC;SAChF;IACF,CAAC;IAEK,gBAAgB;;YACrB,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,KAAK,CAAC,GAAS,EAAE;gBAChB,MAAM,MAAM,GAAG,MAAM,IAAA,iCAAG,EAAC,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC/E,IAAI,KAAK,GAAG,EAAE,CAAC;gBACf,MAAM,OAAO,GAAG,CAAC,IAAuB,EAAE,EAAE;oBAC3C,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC;gBACrB,CAAC,CAAC;gBACF,MAAM,CAAC,EAAE,CAAC,mCAAmC,EAAE,OAAO,CAAC,CAAC;gBACxD,MAAM,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;gBAC/C,aAAa;gBACb,MAAM,CAAC,GAAG,CAAC,mCAAmC,EAAE,OAAO,CAAC,CAAC;gBAChD,MAAM,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;oBAC1B,qBAAqB,EAAE,IAAI;oBAC3B,UAAU,EAAE,oBAAoB;iBACnC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;gBACpB,YAAE,CAAC,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,KAAK,CAAC,CAAC;YACpE,CAAC,CAAA,CAAC,CAAC;QACJ,CAAC;KAAA;IAEK,cAAc;;YACnB,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,KAAK,CAAC,GAAS,EAAE;gBAChB,MAAM,MAAM,GAAG,MAAM,IAAA,iCAAG,EAAC,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC/E,MAAM,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACtB,MAAM,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAC9B,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC,CAAC;gBAC3C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;gBAC5C,MAAM,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;gBAChC,MAAM,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;oBAC1B,qBAAqB,EAAE,IAAI;oBAC3B,UAAU,EAAE,oBAAoB;iBACnC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;gBAC7B,YAAE,CAAC,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;YAClF,CAAC,CAAA,CAAC,CAAC;QACJ,CAAC;KAAA;CACD;AAhFD,8BAgFC"}

package/package.json

@@ -0,0 +1,35 @@
+{
+  "name": "btake",
+  "version": "0.0.1",
+  "description": "a tool for taking node.js profile",
+  "main": "/lib/index.js",
+  "bin": {
+    "btake": "./bin/index.js"
+  },
+  "scripts": {
+    "publish": "npm run build && npm publish",
+    "beta": "npm run build && npm publish --tag beta",
+    "dev": "tsc -w",
+    "build": "rm -rf lib && tsc"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/theanarkh/btake.git"
+  },
+  "author": "theanarkh",
+  "license": "ISC",
+  "bugs": {
+    "url": "https://github.com/theanarkh/btake/issues"
+  },
+  "homepage": "https://github.com/theanarkh/btake#readme",
+  "dependencies": {
+    "chrome-remote-interface": "^0.33.3",
+    "lsofi": "^1.0.0",
+    "systeminformation": "^5.30.1",
+    "yargs": "^14.0.0"
+  },
+  "devDependencies": {
+    "@types/chrome-remote-interface": "^0.33.0",
+    "@types/node": "^25.0.3"
+  }
+}

package/src/index.ts

@@ -0,0 +1,118 @@
+import CDP from 'chrome-remote-interface'
+import fs from 'fs';
+import path from 'path';
+
+const { name } = JSON.parse(fs.readFileSync(path.join(__dirname, '../package.json'), 'utf8'));
+
+export const ToolName = name;
+
+const closeInspectorScript = `(function() {
+ try {
+  if (typeof process._debugEnd === 'function') {
+   process._debugEnd();
+  } else {
+   require('inspector').close();
+  }
+ } catch (e) {
+ }
+})();`;
+
+function sleep(ms: number) {
+ return new Promise(resolve => setTimeout(resolve, ms));
+}
+
+ async function retry<T>(task: () => Promise<T>, delay = 100, retries = 3): Promise<T> {
+ let lastError: any;
+ for (let i = 0; i < retries; i++) {
+  try {
+   return await task();
+  } catch (error) {
+   lastError = error;
+   await sleep(delay);
+  }
+ }
+
+ throw lastError;
+}
+
+export class Inspector {
+    options: any;
+
+    constructor(options: any) {
+  this.options = options;
+  if (this.options.addr) {
+   const list = this.options.addr.split(':');
+   if (list.length >= 2) {
+    this.options.host = list[0];
+    this.options.port = Number(list[1]);
+   } else {
+    this.options.host = 'localhost';
+    this.options.port = Number(list[0]);
+   }
+  } else {
+   this.options.host = 'localhost';
+            this.options.port =  9229;
+  }
+    }
+
+ private openInspector() {
+        try {
+   if (this.options.pid) {
+    // @ts-ignore
+    if (typeof process._debugProcess === 'function') {
+     // @ts-ignore
+     process._debugProcess(this.options.pid);
+    } else {
+     process.kill(this.options.pid as number, 'SIGUSR1');
+    }
+   }
+  } catch (e) {
+   console.error('open inspector failed: err', (e as Error).message);
+  }
+ }
+
+ private genFilename(ext: string) {
+  if (this.options.pid) {
+   return `${name}_${this.options.pid}_${Date.now()}.${ext}`;
+  } else {
+   return `${name}_${this.options.host}_${this.options.port}_${Date.now()}.${ext}`;
+  }
+ }
+
+ async takeHeapSnapshot(): Promise<void> {
+  this.openInspector();
+  retry(async () => {
+   const client = await CDP({ port: this.options.port, host: this.options.host });
+   let chunk = '';
+   const handler = (data: { chunk: string }) => {
+    chunk += data.chunk;
+   };
+   client.on('HeapProfiler.addHeapSnapshotChunk', handler);
+   await client.HeapProfiler.takeHeapSnapshot({});
+   // @ts-ignore
+   client.off('HeapProfiler.addHeapSnapshotChunk', handler); 
+            await client.Runtime.evaluate({
+                includeCommandLineAPI: true,
+                expression: closeInspectorScript,
+            }).catch(() => { });
+            fs.writeFileSync(this.genFilename('heapsnapshot'), chunk);
+  });
+ }
+
+ async takeCpuProfile(): Promise<void> {
+  this.openInspector();
+  retry(async () => {
+   const client = await CDP({ port: this.options.port, host: this.options.host });
+   await client.Profiler.enable();
+            await client.Profiler.start();
+            await sleep(this.options.duration || 1000);
+            const result = await client.Profiler.stop();
+            await client.Profiler.disable();
+            await client.Runtime.evaluate({
+                includeCommandLineAPI: true,
+                expression: closeInspectorScript,
+            }).catch(() => { });
+   fs.writeFileSync(this.genFilename('cpuprofile'), JSON.stringify(result.profile));
+  });
+ }
+}

package/tsconfig.json

@@ -0,0 +1,18 @@
+{
+    // Visit https://aka.ms/tsconfig to read more about this file
+    "compilerOptions": {
+      "module": "nodenext",
+      "target": "es6",
+      "types": ["node"],
+      "sourceMap": true,
+      "declaration": true,
+      "declarationMap": true,
+      "noUncheckedIndexedAccess": true,
+      "exactOptionalPropertyTypes": true,
+      "strict": true,
+      "isolatedModules": true,
+      "moduleDetection": "force",
+      "skipLibCheck": true,
+      "outDir": "./lib",
+    }
+  }