@arvoretech/runtime-lens-mcp 1.1.0 → 1.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/agent/index.ts +281 -212
- package/dist/agent/index.js +92 -36
- package/dist/agent/index.js.map +2 -2
- package/dist/agent-bridge.d.ts +20 -0
- package/dist/agent-bridge.d.ts.map +1 -0
- package/dist/agent-bridge.js +185 -0
- package/dist/agent-bridge.js.map +1 -0
- package/dist/extension/extension.js +58 -32
- package/dist/extension/extension.js.map +2 -2
- package/dist/server.d.ts +1 -0
- package/dist/server.d.ts.map +1 -1
- package/dist/server.js +115 -100
- package/dist/server.js.map +1 -1
- package/package.json +1 -1
- package/src/agent-bridge.ts +218 -0
- package/src/server.ts +405 -358
package/dist/server.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EACN,qBAAqB,EACrB,0BAA0B,EAC1B,0BAA0B,EAC1B,sBAAsB,EACtB,oBAAoB,GACpB,MAAM,YAAY,CAAC;AAEpB,MAAM,OAAO,oBAAoB;IACf,MAAM,CAAY;IAClB,SAAS,CAAe;IACxB,SAAS,CAAmB;IAC5B,WAAW,CAAqB;IAChC,WAAW,CAAc;IAE1C,YAAY,WAAoB,EAAE,QAAmB;QACpD,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC;YAC3B,IAAI,EAAE,kBAAkB;YACxB,OAAO,EAAE,OAAO;SAChB,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,IAAI,YAAY,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QACzD,IAAI,CAAC,SAAS,GAAG,IAAI,gBAAgB,CAAC,WAAW,CAAC,CAAC;QACnD,IAAI,CAAC,WAAW,GAAG,IAAI,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC1D,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACnD,IAAI,CAAC,UAAU,EAAE,CAAC;IACnB,CAAC;IAED,MAAM,CAAC,eAAe;QACrB,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,yBAAyB,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;QAC3E,MAAM,QAAQ,GACb,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QACtE,OAAO,IAAI,oBAAoB,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;IACxD,CAAC;IAEO,UAAU;QACjB,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAC/B,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC9B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAChC,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAC/B,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC3B,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC3B,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC7B,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACzB,CAAC;IAEO,gBAAgB;QACvB,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,WAAW,EACX;YACC,KAAK,EAAE,WAAW;YAClB,WAAW,EACV,8GAA8G;YAC/G,WAAW,EAAE;gBACZ,KAAK,EAAE,oBAAoB,CAAC,KAAK,CAAC,KAAK;gBACvC,KAAK,EAAE,oBAAoB,CAAC,KAAK,CAAC,KAAK;gBACvC,SAAS,EAAE,oBAAoB,CAAC,KAAK,CAAC,SAAS;gBAC/C,MAAM,EAAE,oBAAoB,CAAC,KAAK,CAAC,MAAM;aACzC;SACD,EACD,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YAClB,OAAO,EAAE;gBACR;oBACC,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;iBAC7D;aACD;SACD,CAAC,CACF,CAAC;IACH,CAAC;IAEO,kBAAkB;QACzB,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,aAAa,EACb;YACC,KAAK,EAAE,aAAa;YACpB,WAAW,EACV,iGAAiG;YAClG,WAAW,EAAE;gBACZ,KAAK,EAAE,sBAAsB,CAAC,KAAK,CAAC,KAAK;gBACzC,KAAK,EAAE,sBAAsB,CAAC,KAAK,CAAC,KAAK;gBACzC,SAAS,EAAE,sBAAsB,CAAC,KAAK,CAAC,SAAS;gBACjD,KAAK,EAAE,sBAAsB,CAAC,KAAK,CAAC,KAAK;gBACzC,KAAK,EAAE,sBAAsB,CAAC,KAAK,CAAC,KAAK;aACzC;SACD,EACD,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YAClB,OAAO,EAAE;gBACR;oBACC,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CACnB,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAa,CAAC,EACxC,IAAI,EACJ,CAAC,CACD;iBACD;aACD;SACD,CAAC,CACF,CAAC;IACH,CAAC;IAEO,iBAAiB;QACxB,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,YAAY,EACZ;YACC,KAAK,EAAE,YAAY;YACnB,WAAW,EACV,0GAA0G;YAC3G,WAAW,EAAE;gBACZ,KAAK,EAAE,qBAAqB,CAAC,KAAK,CAAC,KAAK;gBACxC,SAAS,EAAE,qBAAqB,CAAC,KAAK,CAAC,SAAS;gBAChD,OAAO,EAAE,qBAAqB,CAAC,KAAK,CAAC,OAAO;aAC5C;SACD,EACD,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YAClB,OAAO,EAAE;gBACR;oBACC,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;iBAC/D;aACD;SACD,CAAC,CACF,CAAC;IACH,CAAC;IAEO,uBAAuB;QAC9B,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,kBAAkB,EAClB;YACC,KAAK,EAAE,uBAAuB;YAC9B,WAAW,EACV,mHAAmH;YACpH,WAAW,EAAE;gBACZ,EAAE,EAAE,0BAA0B,CAAC,KAAK,CAAC,EAAE;gBACvC,MAAM,EAAE,0BAA0B,CAAC,KAAK,CAAC,MAAM;gBAC/C,UAAU,EAAE,0BAA0B,CAAC,KAAK,CAAC,UAAU;gBACvD,UAAU,EAAE,0BAA0B,CAAC,KAAK,CAAC,UAAU;gBACvD,KAAK,EAAE,0BAA0B,CAAC,KAAK,CAAC,KAAK;aAC7C;SACD,EACD,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YAClB,OAAO,EAAE;gBACR;oBACC,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;iBACjE;aACD;SACD,CAAC,CACF,CAAC;IACH,CAAC;IAEO,sBAAsB;QAC7B,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,iBAAiB,EACjB;YACC,KAAK,EAAE,yBAAyB;YAChC,WAAW,EACV,oGAAoG;YACrG,WAAW,EAAE;gBACZ,MAAM,EAAE,0BAA0B,CAAC,KAAK,CAAC,MAAM;gBAC/C,KAAK,EAAE,0BAA0B,CAAC,KAAK,CAAC,KAAK;gBAC7C,KAAK,EAAE,0BAA0B,CAAC,KAAK,CAAC,KAAK;aAC7C;SACD,EACD,KAAK,EAAE,MAAM,EAAE,EAAE;YAChB,MAAM,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,CAAC;YAC1C,OAAO;gBACN,OAAO,EAAE;oBACR;wBACC,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;qBAChE;iBACD;aACD,CAAC;QACH,CAAC,CACD,CAAC;IACH,CAAC;IAEO,kBAAkB;QACzB,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,cAAc,EACd;YACC,KAAK,EAAE,sBAAsB;YAC7B,WAAW,EACV,oJAAoJ;YACrJ,WAAW,EAAE,EAAE;SACf,EACD,KAAK,IAAI,EAAE,CAAC,CAAC;YACZ,OAAO,EAAE;gBACR;oBACC,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CACnB,MAAM,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,EACzC,IAAI,EACJ,CAAC,CACD;iBACD;aACD;SACD,CAAC,CACF,CAAC;IACH,CAAC;IAEO,iBAAiB;QACxB,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,YAAY,EACZ;YACC,KAAK,EAAE,kBAAkB;YACzB,WAAW,EACV,kEAAkE;YACnE,WAAW,EAAE,EAAE;SACf,EACD,KAAK,IAAI,EAAE;YACV,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC;YAC1C,OAAO;gBACN,OAAO,EAAE;oBACR;wBACC,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;4BACpB,OAAO,EAAE,WAAW,MAAM,CAAC,OAAO,UAAU;yBAC5C,CAAC;qBACF;iBACD;aACD,CAAC;QACH,CAAC,CACD,CAAC;IACH,CAAC;IAEO,gBAAgB;QACvB,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,WAAW,EACX;YACC,KAAK,EAAE,oBAAoB;YAC3B,WAAW,EACV,8EAA8E;YAC/E,WAAW,EAAE,EAAE;SACf,EACD,KAAK,IAAI,EAAE,CAAC,CAAC;YACZ,OAAO,EAAE;gBACR;oBACC,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;iBACxD;aACD;SACD,CAAC,CACF,CAAC;IACH,CAAC;IAEO,wBAAwB;QAC/B,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,mBAAmB,EACnB;YACC,KAAK,EAAE,2BAA2B;YAClC,WAAW,EACV,oIAAoI;YACrI,WAAW,EAAE,EAAE;SACf,EACD,KAAK,IAAI,EAAE;YACV,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,CAAC;YACrC,OAAO;gBACN,OAAO,EAAE;oBACR;wBACC,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;4BACpB,OAAO,EAAE,6BAA6B;4BACtC,MAAM,EAAE,IAAI;yBACZ,CAAC;qBACF;iBACD;aACD,CAAC;QACH,CAAC,CACD,CAAC;IACH,CAAC;IAEO,uBAAuB;QAC9B,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,kBAAkB,EAClB;YACC,KAAK,EAAE,0BAA0B;YACjC,WAAW,EACV,kFAAkF;YACnF,WAAW,EAAE,EAAE;SACf,EACD,KAAK,IAAI,EAAE;YACV,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC;YACpC,OAAO;gBACN,OAAO,EAAE;oBACR;wBACC,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;4BACpB,OAAO,EAAE,6BAA6B;4BACtC,MAAM,EAAE,KAAK;yBACb,CAAC;qBACF;iBACD;aACD,CAAC;QACH,CAAC,CACD,CAAC;IACH,CAAC;IAEO,mBAAmB;QAC1B,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,oBAAoB,EACpB;YACC,KAAK,EAAE,yBAAyB;YAChC,WAAW,EACV,wIAAwI;YACzI,WAAW,EAAE,EAAE;SACf,EACD,KAAK,IAAI,EAAE;YACV,MAAM,IAAI,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC;YACxC,OAAO;gBACN,OAAO,EAAE;oBACR;wBACC,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;4BACpB,OAAO,EAAE,yBAAyB;4BAClC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE;yBAChC,CAAC;qBACF;iBACD;aACD,CAAC;QACH,CAAC,CACD,CAAC;IACH,CAAC;IAEO,mBAAmB;QAC1B,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,cAAc,EACd;YACC,KAAK,EAAE,wBAAwB;YAC/B,WAAW,EACV,+GAA+G;YAChH,WAAW,EAAE,EAAE;SACf,EACD,KAAK,IAAI,EAAE,CAAC,CAAC;YACZ,OAAO,EAAE;gBACR;oBACC,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CACnB,MAAM,IAAI,CAAC,SAAS,CAAC,oBAAoB,EAAE,EAC3C,IAAI,EACJ,CAAC,CACD;iBACD;aACD;SACD,CAAC,CACF,CAAC;IACH,CAAC;IAEO,qBAAqB;QAC5B,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,gBAAgB,EAChB;YACC,KAAK,EAAE,6BAA6B;YACpC,WAAW,EACV,mFAAmF;YACpF,WAAW,EAAE,EAAE;SACf,EACD,KAAK,IAAI,EAAE,CAAC,CAAC;YACZ,OAAO,EAAE;gBACR;oBACC,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CACnB,MAAM,IAAI,CAAC,SAAS,CAAC,oBAAoB,EAAE,EAC3C,IAAI,EACJ,CAAC,CACD;iBACD;aACD;SACD,CAAC,CACF,CAAC;IACH,CAAC;IAEO,gBAAgB;QACvB,IAAI,CAAC,MAAM,CAAC,YAAY,CACvB,qBAAqB,EACrB;YACC,KAAK,EAAE,qBAAqB;YAC5B,WAAW,EACV,sEAAsE;YACvE,WAAW,EAAE,EAAE;SACf,EACD,KAAK,IAAI,EAAE,CAAC,CAAC;YACZ,OAAO,EAAE;gBACR;oBACC,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CACnB,MAAM,IAAI,CAAC,SAAS,CAAC,gBAAgB,EAAE,EACvC,IAAI,EACJ,CAAC,CACD;iBACD;aACD;SACD,CAAC,CACF,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACV,IAAI,CAAC;YACJ,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YAC3B,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;YAC7C,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACrC,OAAO,CAAC,KAAK,CAAC,8CAA8C,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CACZ,0CAA0C,EAC1C,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAC9C,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;IACF,CAAC;IAED,qBAAqB;QACpB,MAAM,QAAQ,GAAG,KAAK,EAAE,MAAc,EAAiB,EAAE;YACxD,OAAO,CAAC,KAAK,CAAC,YAAY,MAAM,+BAA+B,CAAC,CAAC;YACjE,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;YAC9B,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC;YACpC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC,CAAC;QACF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC/C,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;IAClD,CAAC;CACD"}
|
package/package.json
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@arvoretech/runtime-lens-mcp",
|
|
3
3
|
"displayName": "Runtime Lens",
|
|
4
|
-
"version": "1.
|
|
4
|
+
"version": "1.2.0",
|
|
5
5
|
"description": "Runtime Lens - Runtime inspection with inline values for React, NestJS and Next.js",
|
|
6
6
|
"publisher": "arvore",
|
|
7
7
|
"main": "dist/extension/extension.js",
|
|
@@ -0,0 +1,218 @@
|
|
|
1
|
+
import { randomBytes } from "node:crypto";
|
|
2
|
+
import { request } from "node:http";
|
|
3
|
+
import type { Socket } from "node:net";
|
|
4
|
+
import type { LogCollector } from "./log-collector.js";
|
|
5
|
+
|
|
6
|
+
interface AgentMessage {
|
|
7
|
+
type: "log" | "error" | "warn" | "info" | "debug" | "result" | "buffer";
|
|
8
|
+
file: string;
|
|
9
|
+
line: number;
|
|
10
|
+
column: number;
|
|
11
|
+
values: string[];
|
|
12
|
+
timestamp: number;
|
|
13
|
+
expression?: string;
|
|
14
|
+
logs?: AgentMessage[];
|
|
15
|
+
}
|
|
16
|
+
|
|
17
|
+
const LEVEL_MAP: Record<string, "debug" | "info" | "warn" | "error" | "fatal"> =
|
|
18
|
+
{
|
|
19
|
+
log: "info",
|
|
20
|
+
info: "info",
|
|
21
|
+
warn: "warn",
|
|
22
|
+
error: "error",
|
|
23
|
+
debug: "debug",
|
|
24
|
+
result: "info",
|
|
25
|
+
};
|
|
26
|
+
|
|
27
|
+
export class AgentBridge {
|
|
28
|
+
private socket: Socket | null = null;
|
|
29
|
+
private reconnectTimer: ReturnType<typeof setTimeout> | null = null;
|
|
30
|
+
private connected = false;
|
|
31
|
+
private readonly port: number;
|
|
32
|
+
private readonly collector: LogCollector;
|
|
33
|
+
private dataBuffer: Buffer = Buffer.alloc(0);
|
|
34
|
+
private stopped = false;
|
|
35
|
+
|
|
36
|
+
constructor(collector: LogCollector, port = 9500) {
|
|
37
|
+
this.collector = collector;
|
|
38
|
+
this.port = parseInt(process.env.RUNTIME_LENS_PORT || String(port), 10);
|
|
39
|
+
}
|
|
40
|
+
|
|
41
|
+
connect(): void {
|
|
42
|
+
if (this.socket || this.stopped) return;
|
|
43
|
+
|
|
44
|
+
const key = randomBytes(16).toString("base64");
|
|
45
|
+
|
|
46
|
+
const req = request({
|
|
47
|
+
hostname: "localhost",
|
|
48
|
+
port: this.port,
|
|
49
|
+
path: "/",
|
|
50
|
+
method: "GET",
|
|
51
|
+
timeout: 3000,
|
|
52
|
+
headers: {
|
|
53
|
+
Upgrade: "websocket",
|
|
54
|
+
Connection: "Upgrade",
|
|
55
|
+
"Sec-WebSocket-Key": key,
|
|
56
|
+
"Sec-WebSocket-Version": "13",
|
|
57
|
+
},
|
|
58
|
+
});
|
|
59
|
+
|
|
60
|
+
req.on("upgrade", (_res, socket) => {
|
|
61
|
+
this.socket = socket;
|
|
62
|
+
this.connected = true;
|
|
63
|
+
this.dataBuffer = Buffer.alloc(0);
|
|
64
|
+
console.error("[runtime-lens-mcp] Connected to agent on port", this.port);
|
|
65
|
+
|
|
66
|
+
// Request buffered logs
|
|
67
|
+
this.sendFrame(JSON.stringify({ type: "get_buffer" }));
|
|
68
|
+
|
|
69
|
+
socket.on("data", (data: Buffer) => {
|
|
70
|
+
this.dataBuffer = Buffer.concat([this.dataBuffer, data]);
|
|
71
|
+
this.processFrames();
|
|
72
|
+
});
|
|
73
|
+
|
|
74
|
+
socket.on("close", () => {
|
|
75
|
+
this.connected = false;
|
|
76
|
+
this.socket = null;
|
|
77
|
+
this.scheduleReconnect();
|
|
78
|
+
});
|
|
79
|
+
|
|
80
|
+
socket.on("error", () => {
|
|
81
|
+
this.connected = false;
|
|
82
|
+
this.socket = null;
|
|
83
|
+
});
|
|
84
|
+
});
|
|
85
|
+
|
|
86
|
+
req.on("error", () => {
|
|
87
|
+
this.scheduleReconnect();
|
|
88
|
+
});
|
|
89
|
+
|
|
90
|
+
req.on("timeout", () => {
|
|
91
|
+
req.destroy();
|
|
92
|
+
this.scheduleReconnect();
|
|
93
|
+
});
|
|
94
|
+
|
|
95
|
+
req.end();
|
|
96
|
+
}
|
|
97
|
+
|
|
98
|
+
disconnect(): void {
|
|
99
|
+
this.stopped = true;
|
|
100
|
+
if (this.reconnectTimer) {
|
|
101
|
+
clearTimeout(this.reconnectTimer);
|
|
102
|
+
this.reconnectTimer = null;
|
|
103
|
+
}
|
|
104
|
+
if (this.socket) {
|
|
105
|
+
this.socket.end();
|
|
106
|
+
this.socket = null;
|
|
107
|
+
}
|
|
108
|
+
this.connected = false;
|
|
109
|
+
}
|
|
110
|
+
|
|
111
|
+
isConnected(): boolean {
|
|
112
|
+
return this.connected;
|
|
113
|
+
}
|
|
114
|
+
|
|
115
|
+
private processFrames(): void {
|
|
116
|
+
while (this.dataBuffer.length >= 2) {
|
|
117
|
+
const firstByte = this.dataBuffer[0];
|
|
118
|
+
const secondByte = this.dataBuffer[1];
|
|
119
|
+
const isFin = (firstByte & 0x80) !== 0;
|
|
120
|
+
const opcode = firstByte & 0x0f;
|
|
121
|
+
|
|
122
|
+
if (opcode === 0x08) {
|
|
123
|
+
this.socket?.end();
|
|
124
|
+
return;
|
|
125
|
+
}
|
|
126
|
+
|
|
127
|
+
let payloadLen = secondByte & 0x7f;
|
|
128
|
+
let headerLen = 2;
|
|
129
|
+
|
|
130
|
+
if (payloadLen === 126) {
|
|
131
|
+
if (this.dataBuffer.length < 4) return;
|
|
132
|
+
payloadLen = this.dataBuffer.readUInt16BE(2);
|
|
133
|
+
headerLen = 4;
|
|
134
|
+
} else if (payloadLen === 127) {
|
|
135
|
+
if (this.dataBuffer.length < 10) return;
|
|
136
|
+
payloadLen = Number(this.dataBuffer.readBigUInt64BE(2));
|
|
137
|
+
headerLen = 10;
|
|
138
|
+
}
|
|
139
|
+
|
|
140
|
+
const totalLen = headerLen + payloadLen;
|
|
141
|
+
if (this.dataBuffer.length < totalLen) return;
|
|
142
|
+
|
|
143
|
+
const payload = this.dataBuffer.subarray(headerLen, totalLen);
|
|
144
|
+
this.dataBuffer = this.dataBuffer.subarray(totalLen);
|
|
145
|
+
|
|
146
|
+
if (isFin && (opcode === 0x01 || opcode === 0x02)) {
|
|
147
|
+
const text = payload.toString("utf-8");
|
|
148
|
+
try {
|
|
149
|
+
const msg: AgentMessage = JSON.parse(text);
|
|
150
|
+
this.handleMessage(msg);
|
|
151
|
+
} catch {
|
|
152
|
+
// invalid JSON
|
|
153
|
+
}
|
|
154
|
+
}
|
|
155
|
+
}
|
|
156
|
+
}
|
|
157
|
+
|
|
158
|
+
private handleMessage(msg: AgentMessage): void {
|
|
159
|
+
if (msg.type === "buffer" && msg.logs) {
|
|
160
|
+
for (const log of msg.logs) {
|
|
161
|
+
this.ingestLog(log);
|
|
162
|
+
}
|
|
163
|
+
return;
|
|
164
|
+
}
|
|
165
|
+
|
|
166
|
+
this.ingestLog(msg);
|
|
167
|
+
}
|
|
168
|
+
|
|
169
|
+
private ingestLog(msg: AgentMessage): void {
|
|
170
|
+
const level = LEVEL_MAP[msg.type] || "info";
|
|
171
|
+
this.collector.addLog({
|
|
172
|
+
level,
|
|
173
|
+
message: msg.values.join(", "),
|
|
174
|
+
source: msg.file || "agent",
|
|
175
|
+
framework: "unknown",
|
|
176
|
+
metadata: {
|
|
177
|
+
line: msg.line,
|
|
178
|
+
column: msg.column,
|
|
179
|
+
agentType: msg.type,
|
|
180
|
+
expression: msg.expression,
|
|
181
|
+
originalTimestamp: msg.timestamp,
|
|
182
|
+
},
|
|
183
|
+
});
|
|
184
|
+
}
|
|
185
|
+
|
|
186
|
+
private sendFrame(text: string): void {
|
|
187
|
+
if (!this.socket) return;
|
|
188
|
+
const data = Buffer.from(text, "utf-8");
|
|
189
|
+
const len = data.length;
|
|
190
|
+
let header: Buffer;
|
|
191
|
+
|
|
192
|
+
if (len < 126) {
|
|
193
|
+
header = Buffer.alloc(2);
|
|
194
|
+
header[0] = 0x81;
|
|
195
|
+
header[1] = len;
|
|
196
|
+
} else if (len < 65536) {
|
|
197
|
+
header = Buffer.alloc(4);
|
|
198
|
+
header[0] = 0x81;
|
|
199
|
+
header[1] = 126;
|
|
200
|
+
header.writeUInt16BE(len, 2);
|
|
201
|
+
} else {
|
|
202
|
+
header = Buffer.alloc(10);
|
|
203
|
+
header[0] = 0x81;
|
|
204
|
+
header[1] = 127;
|
|
205
|
+
header.writeBigUInt64BE(BigInt(len), 2);
|
|
206
|
+
}
|
|
207
|
+
|
|
208
|
+
this.socket.write(Buffer.concat([header, data]));
|
|
209
|
+
}
|
|
210
|
+
|
|
211
|
+
private scheduleReconnect(): void {
|
|
212
|
+
if (this.reconnectTimer || this.stopped) return;
|
|
213
|
+
this.reconnectTimer = setTimeout(() => {
|
|
214
|
+
this.reconnectTimer = null;
|
|
215
|
+
this.connect();
|
|
216
|
+
}, 5000);
|
|
217
|
+
}
|
|
218
|
+
}
|