@qwickapps/qwickbrain-proxy 1.0.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.
Files changed (90) hide show
  1. package/.github/workflows/publish.yml +92 -0
  2. package/CHANGELOG.md +47 -0
  3. package/LICENSE +45 -0
  4. package/README.md +165 -0
  5. package/dist/bin/cli.d.ts +3 -0
  6. package/dist/bin/cli.d.ts.map +1 -0
  7. package/dist/bin/cli.js +142 -0
  8. package/dist/bin/cli.js.map +1 -0
  9. package/dist/db/client.d.ts +10 -0
  10. package/dist/db/client.d.ts.map +1 -0
  11. package/dist/db/client.js +23 -0
  12. package/dist/db/client.js.map +1 -0
  13. package/dist/db/schema.d.ts +551 -0
  14. package/dist/db/schema.d.ts.map +1 -0
  15. package/dist/db/schema.js +65 -0
  16. package/dist/db/schema.js.map +1 -0
  17. package/dist/index.d.ts +7 -0
  18. package/dist/index.d.ts.map +1 -0
  19. package/dist/index.js +5 -0
  20. package/dist/index.js.map +1 -0
  21. package/dist/lib/__tests__/cache-manager.test.d.ts +2 -0
  22. package/dist/lib/__tests__/cache-manager.test.d.ts.map +1 -0
  23. package/dist/lib/__tests__/cache-manager.test.js +202 -0
  24. package/dist/lib/__tests__/cache-manager.test.js.map +1 -0
  25. package/dist/lib/__tests__/connection-manager.test.d.ts +2 -0
  26. package/dist/lib/__tests__/connection-manager.test.d.ts.map +1 -0
  27. package/dist/lib/__tests__/connection-manager.test.js +188 -0
  28. package/dist/lib/__tests__/connection-manager.test.js.map +1 -0
  29. package/dist/lib/__tests__/proxy-server.test.d.ts +2 -0
  30. package/dist/lib/__tests__/proxy-server.test.d.ts.map +1 -0
  31. package/dist/lib/__tests__/proxy-server.test.js +205 -0
  32. package/dist/lib/__tests__/proxy-server.test.js.map +1 -0
  33. package/dist/lib/__tests__/qwickbrain-client.test.d.ts +2 -0
  34. package/dist/lib/__tests__/qwickbrain-client.test.d.ts.map +1 -0
  35. package/dist/lib/__tests__/qwickbrain-client.test.js +233 -0
  36. package/dist/lib/__tests__/qwickbrain-client.test.js.map +1 -0
  37. package/dist/lib/cache-manager.d.ts +25 -0
  38. package/dist/lib/cache-manager.d.ts.map +1 -0
  39. package/dist/lib/cache-manager.js +149 -0
  40. package/dist/lib/cache-manager.js.map +1 -0
  41. package/dist/lib/connection-manager.d.ts +26 -0
  42. package/dist/lib/connection-manager.d.ts.map +1 -0
  43. package/dist/lib/connection-manager.js +130 -0
  44. package/dist/lib/connection-manager.js.map +1 -0
  45. package/dist/lib/proxy-server.d.ts +19 -0
  46. package/dist/lib/proxy-server.d.ts.map +1 -0
  47. package/dist/lib/proxy-server.js +258 -0
  48. package/dist/lib/proxy-server.js.map +1 -0
  49. package/dist/lib/qwickbrain-client.d.ts +24 -0
  50. package/dist/lib/qwickbrain-client.d.ts.map +1 -0
  51. package/dist/lib/qwickbrain-client.js +197 -0
  52. package/dist/lib/qwickbrain-client.js.map +1 -0
  53. package/dist/types/config.d.ts +186 -0
  54. package/dist/types/config.d.ts.map +1 -0
  55. package/dist/types/config.js +42 -0
  56. package/dist/types/config.js.map +1 -0
  57. package/dist/types/mcp.d.ts +223 -0
  58. package/dist/types/mcp.d.ts.map +1 -0
  59. package/dist/types/mcp.js +78 -0
  60. package/dist/types/mcp.js.map +1 -0
  61. package/dist/version.d.ts +2 -0
  62. package/dist/version.d.ts.map +1 -0
  63. package/dist/version.js +9 -0
  64. package/dist/version.js.map +1 -0
  65. package/drizzle/0000_fat_rafael_vega.sql +41 -0
  66. package/drizzle/0001_goofy_invisible_woman.sql +2 -0
  67. package/drizzle/meta/0000_snapshot.json +276 -0
  68. package/drizzle/meta/0001_snapshot.json +295 -0
  69. package/drizzle/meta/_journal.json +20 -0
  70. package/drizzle.config.ts +12 -0
  71. package/package.json +65 -0
  72. package/src/bin/cli.ts +158 -0
  73. package/src/db/client.ts +34 -0
  74. package/src/db/schema.ts +68 -0
  75. package/src/index.ts +6 -0
  76. package/src/lib/__tests__/cache-manager.test.ts +264 -0
  77. package/src/lib/__tests__/connection-manager.test.ts +255 -0
  78. package/src/lib/__tests__/proxy-server.test.ts +261 -0
  79. package/src/lib/__tests__/qwickbrain-client.test.ts +310 -0
  80. package/src/lib/cache-manager.ts +201 -0
  81. package/src/lib/connection-manager.ts +156 -0
  82. package/src/lib/proxy-server.ts +320 -0
  83. package/src/lib/qwickbrain-client.ts +260 -0
  84. package/src/types/config.ts +47 -0
  85. package/src/types/mcp.ts +97 -0
  86. package/src/version.ts +11 -0
  87. package/test/fixtures/test-mcp.json +5 -0
  88. package/test-mcp-client.js +67 -0
  89. package/test-proxy.sh +25 -0
  90. package/tsconfig.json +22 -0
@@ -0,0 +1 @@
1
+ {"version":3,"file":"proxy-server.test.js","sourceRoot":"","sources":["../../../src/lib/__tests__/proxy-server.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;AACzC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;AAC5B,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAGjD,eAAe;AACf,EAAE,CAAC,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QACxC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAC7C,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAC3C,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;QACjB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;KACpD,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QACxC,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;QAC1B,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAC7C,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;KAC5C,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC9B,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,yCAAyC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC5B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,IAAI,MAAc,CAAC;IACnB,IAAI,MAAc,CAAC;IACnB,IAAI,EAA2C,CAAC;IAChD,IAAI,MAAmB,CAAC;IAExB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC;QACpD,MAAM,QAAQ,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACxC,EAAE,GAAG,QAAQ,CAAC,EAAE,CAAC;QAEjB,kCAAkC;QAClC,aAAa,CAAC,EAAE,CAAC,CAAC;QAElB,MAAM,GAAG;YACP,UAAU,EAAE;gBACV,IAAI,EAAE,KAAK;gBACX,GAAG,EAAE,wBAAwB;aAC9B;YACD,KAAK,EAAE;gBACL,GAAG,EAAE,MAAM;gBACX,GAAG,EAAE;oBACH,SAAS,EAAE,IAAI;oBACf,KAAK,EAAE,IAAI;oBACX,SAAS,EAAE,IAAI;oBACf,QAAQ,EAAE,GAAG;iBACd;gBACD,OAAO,EAAE,EAAE;aACZ;YACD,UAAU,EAAE;gBACV,mBAAmB,EAAE,KAAK;gBAC1B,OAAO,EAAE,IAAI;gBACb,oBAAoB,EAAE,CAAC;gBACvB,gBAAgB,EAAE;oBAChB,OAAO,EAAE,IAAI;oBACb,UAAU,EAAE,CAAC;oBACb,GAAG,EAAE,KAAK;iBACX;aACF;YACD,IAAI,EAAE;gBACJ,QAAQ,EAAE,KAAK;gBACf,SAAS,EAAE,EAAE;aACd;SACF,CAAC;QAEF,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,MAAM,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,qBAAqB,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC;YAEjE,yDAAyD;YACzD,MAAM,CAAC,qBAAqB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACjD,MAAM,CAAE,qBAA6B,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QACrF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,2BAA2B;YAC3B,MAAM,cAAc,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC;YACrC,cAAc,CAAC,KAAK,CAAC,GAAG,GAAG;gBACzB,SAAS,EAAE,CAAC;gBACZ,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,CAAC;gBACZ,QAAQ,EAAE,CAAC;aACZ,CAAC;YAEF,MAAM,UAAU,GAAG,IAAI,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAEvD,MAAM,UAAU,CAAC,cAAc,CAAC,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAC5E,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YAEvD,qCAAqC;YACrC,MAAM,aAAa,GAAG,MAAM,UAAU,CAAC,cAAc,CAAC,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YACvF,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAErC,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC;YAEzB,wCAAwC;YACxC,MAAM,YAAY,GAAG,MAAM,UAAU,CAAC,cAAc,CAAC,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YACtF,MAAM,CAAC,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC;YAEhC,MAAM,UAAU,CAAC,IAAI,EAAE,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,oDAAoD;YACpD,MAAM,cAAc,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC;YACrC,cAAc,CAAC,KAAK,CAAC,GAAG,GAAG;gBACzB,SAAS,EAAE,CAAC;gBACZ,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,CAAC;gBACZ,QAAQ,EAAE,CAAC;aACZ,CAAC;YACF,MAAM,UAAU,GAAG,IAAI,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAEvD,oBAAoB;YACpB,MAAM,UAAU,CAAC,cAAc,CAAC,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,gBAAgB,CAAC,CAAC;YAEnF,2BAA2B;YAC3B,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YAEvD,8BAA8B;YAC9B,UAAU,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,GAAG,cAAc,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,mBAAmB,CAAC,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAEzE,MAAM,CAAE,MAAM,CAAC,IAAY,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAEpD,MAAM,UAAU,CAAC,IAAI,EAAE,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,4CAA4C;YAC5C,MAAM,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,GAAG,cAAc,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;YAE7E,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAExE,2BAA2B;YAC3B,MAAM,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,GAAG,WAAW,CAAC;YAEnD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAErE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YACnD,MAAM,CAAC,OAAO,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,SAAS,GAAG,MAAM,CAAC,gBAAgB,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YACzD,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC;gBACxB,MAAM,EAAE,OAAO;gBACf,WAAW,EAAE,GAAG;gBAChB,MAAM,EAAE,MAAM,CAAC,mBAAmB,CAAC,CAAC,QAAQ,EAAE;aAC/C,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC;YAClD,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,MAAM,EAAE,MAAM;gBACd,MAAM,EAAE,MAAM,CAAC,mBAAmB,CAAC,CAAC,QAAQ,EAAE;aAC/C,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,MAAM,CAAC,gBAAgB,CAAC,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC;YAC/D,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC;gBACxB,MAAM,EAAE,aAAa;gBACrB,WAAW,EAAE,GAAG;gBAChB,MAAM,EAAE,MAAM,CAAC,mBAAmB,CAAC,CAAC,QAAQ,EAAE;aAC/C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,6BAA6B;YAC7B,MAAM,CAAC,kBAAkB,CAAC,CAAC,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAChE,IAAI,KAAK,CAAC,eAAe,CAAC,CAC3B,CAAC;YAEF,2BAA2B;YAC3B,MAAM,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,GAAG,WAAW,CAAC;YAEnD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAErE,oEAAoE;YACpE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC,WAAW,CACtC,QAAQ,EACR,aAAa,EACb,gBAAgB,EAChB,YAAY,CACb,CAAC;YAEF,MAAM,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,GAAG,WAAW,CAAC;YAEnD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,QAAQ,EAAE,aAAa,EAAE,YAAY,CAAC,CAAC;YAExF,MAAM,CAAE,MAAM,CAAC,IAAY,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAC5D,MAAM,CAAE,MAAM,CAAC,IAAY,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;YACxE,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,cAAc,EAAE,OAAO,CAAC,CAAC;YAElF,MAAM,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,GAAG,WAAW,CAAC;YAEnD,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACvE,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAEjF,MAAM,CAAE,YAAY,CAAC,IAAY,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC/D,MAAM,CAAE,aAAa,CAAC,IAAY,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=qwickbrain-client.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"qwickbrain-client.test.d.ts","sourceRoot":"","sources":["../../../src/lib/__tests__/qwickbrain-client.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,233 @@
1
+ import { describe, it, expect, beforeEach, vi, afterEach } from 'vitest';
2
+ import { QwickBrainClient } from '../qwickbrain-client.js';
3
+ // Mock the MCP SDK
4
+ vi.mock('@modelcontextprotocol/sdk/client/index.js', () => ({
5
+ Client: vi.fn().mockImplementation(() => ({
6
+ connect: vi.fn().mockResolvedValue(undefined),
7
+ close: vi.fn().mockResolvedValue(undefined),
8
+ callTool: vi.fn(),
9
+ listTools: vi.fn().mockResolvedValue({ tools: [] }),
10
+ })),
11
+ }));
12
+ vi.mock('@modelcontextprotocol/sdk/client/stdio.js', () => ({
13
+ StdioClientTransport: vi.fn(),
14
+ }));
15
+ vi.mock('@modelcontextprotocol/sdk/client/sse.js', () => ({
16
+ SSEClientTransport: vi.fn(),
17
+ }));
18
+ // Mock fetch for HTTP mode
19
+ global.fetch = vi.fn();
20
+ describe('QwickBrainClient', () => {
21
+ afterEach(() => {
22
+ vi.clearAllMocks();
23
+ });
24
+ describe('SSE mode', () => {
25
+ let client;
26
+ let config;
27
+ beforeEach(() => {
28
+ config = {
29
+ mode: 'sse',
30
+ url: 'http://test.local:3000',
31
+ };
32
+ client = new QwickBrainClient(config);
33
+ });
34
+ it('should connect in SSE mode', async () => {
35
+ await client.connect();
36
+ expect(client['mode']).toBe('sse');
37
+ expect(client['client']).toBeDefined();
38
+ });
39
+ it('should get document via MCP in SSE mode', async () => {
40
+ await client.connect();
41
+ const mockResponse = {
42
+ content: [
43
+ {
44
+ type: 'text',
45
+ text: JSON.stringify({
46
+ document: {
47
+ content: 'test content',
48
+ metadata: { version: 1 },
49
+ },
50
+ }),
51
+ },
52
+ ],
53
+ };
54
+ client['client'].callTool.mockResolvedValue(mockResponse);
55
+ const result = await client.getDocument('workflow', 'test');
56
+ expect(result.content).toBe('test content');
57
+ expect(result.metadata).toEqual({ version: 1 });
58
+ });
59
+ it('should throw on invalid MCP response format', async () => {
60
+ await client.connect();
61
+ const mockResponse = {
62
+ content: [
63
+ {
64
+ type: 'image', // Invalid type
65
+ data: 'invalid',
66
+ },
67
+ ],
68
+ };
69
+ client['client'].callTool.mockResolvedValue(mockResponse);
70
+ await expect(client.getDocument('workflow', 'test')).rejects.toThrow();
71
+ });
72
+ });
73
+ describe('HTTP mode', () => {
74
+ let client;
75
+ let config;
76
+ beforeEach(() => {
77
+ config = {
78
+ mode: 'http',
79
+ url: 'http://api.test.com',
80
+ apiKey: 'test-key',
81
+ };
82
+ client = new QwickBrainClient(config);
83
+ });
84
+ it('should get document via HTTP', async () => {
85
+ const mockResponse = {
86
+ content: 'test content',
87
+ metadata: { version: 1 },
88
+ };
89
+ global.fetch.mockResolvedValue({
90
+ ok: true,
91
+ json: async () => mockResponse,
92
+ });
93
+ const result = await client.getDocument('workflow', 'test');
94
+ expect(result.content).toBe('test content');
95
+ expect(result.metadata).toEqual({ version: 1 });
96
+ expect(global.fetch).toHaveBeenCalledWith('http://api.test.com/mcp/document', expect.objectContaining({
97
+ method: 'POST',
98
+ headers: expect.objectContaining({
99
+ 'Content-Type': 'application/json',
100
+ 'Authorization': 'Bearer test-key',
101
+ }),
102
+ }));
103
+ });
104
+ it('should throw on HTTP error', async () => {
105
+ global.fetch.mockResolvedValue({
106
+ ok: false,
107
+ status: 404,
108
+ statusText: 'Not Found',
109
+ });
110
+ await expect(client.getDocument('workflow', 'test')).rejects.toThrow('HTTP 404');
111
+ });
112
+ it('should validate HTTP response schema', async () => {
113
+ const invalidResponse = {
114
+ // Missing required 'content' field
115
+ metadata: {},
116
+ };
117
+ global.fetch.mockResolvedValue({
118
+ ok: true,
119
+ json: async () => invalidResponse,
120
+ });
121
+ await expect(client.getDocument('workflow', 'test')).rejects.toThrow();
122
+ });
123
+ it('should get memory via HTTP', async () => {
124
+ const mockResponse = {
125
+ content: 'memory content',
126
+ metadata: {},
127
+ };
128
+ global.fetch.mockResolvedValue({
129
+ ok: true,
130
+ json: async () => mockResponse,
131
+ });
132
+ const result = await client.getMemory('test-memory');
133
+ expect(result.content).toBe('memory content');
134
+ expect(global.fetch).toHaveBeenCalledWith('http://api.test.com/mcp/memory', expect.objectContaining({
135
+ method: 'POST',
136
+ }));
137
+ });
138
+ });
139
+ describe('MCP mode', () => {
140
+ let client;
141
+ let config;
142
+ beforeEach(() => {
143
+ config = {
144
+ mode: 'mcp',
145
+ command: 'npx',
146
+ args: ['qwickbrain-server'],
147
+ };
148
+ client = new QwickBrainClient(config);
149
+ });
150
+ it('should connect in MCP mode', async () => {
151
+ await client.connect();
152
+ expect(client['mode']).toBe('mcp');
153
+ expect(client['client']).toBeDefined();
154
+ });
155
+ it('should require command in MCP mode', async () => {
156
+ const invalidConfig = {
157
+ mode: 'mcp',
158
+ // Missing command
159
+ };
160
+ const invalidClient = new QwickBrainClient(invalidConfig);
161
+ await expect(invalidClient.connect()).rejects.toThrow('MCP mode requires command');
162
+ });
163
+ });
164
+ describe('healthCheck', () => {
165
+ it('should return true for successful MCP health check', async () => {
166
+ const config = {
167
+ mode: 'sse',
168
+ url: 'http://test.local:3000',
169
+ };
170
+ const client = new QwickBrainClient(config);
171
+ await client.connect();
172
+ const isHealthy = await client.healthCheck();
173
+ expect(isHealthy).toBe(true);
174
+ });
175
+ it('should return true for successful HTTP health check', async () => {
176
+ const config = {
177
+ mode: 'http',
178
+ url: 'http://api.test.com',
179
+ };
180
+ const client = new QwickBrainClient(config);
181
+ global.fetch.mockResolvedValue({
182
+ ok: true,
183
+ });
184
+ const isHealthy = await client.healthCheck();
185
+ expect(isHealthy).toBe(true);
186
+ expect(global.fetch).toHaveBeenCalledWith('http://api.test.com/health');
187
+ });
188
+ it('should return false on health check failure', async () => {
189
+ const config = {
190
+ mode: 'sse',
191
+ url: 'http://test.local:3000',
192
+ };
193
+ const client = new QwickBrainClient(config);
194
+ await client.connect();
195
+ client['client'].listTools.mockRejectedValue(new Error('Failed'));
196
+ const isHealthy = await client.healthCheck();
197
+ expect(isHealthy).toBe(false);
198
+ });
199
+ it('should return false for missing URL in HTTP mode', async () => {
200
+ const config = {
201
+ mode: 'http',
202
+ // Missing URL
203
+ };
204
+ const client = new QwickBrainClient(config);
205
+ const isHealthy = await client.healthCheck();
206
+ expect(isHealthy).toBe(false);
207
+ });
208
+ });
209
+ describe('disconnect', () => {
210
+ it('should disconnect MCP client', async () => {
211
+ const config = {
212
+ mode: 'sse',
213
+ url: 'http://test.local:3000',
214
+ };
215
+ const client = new QwickBrainClient(config);
216
+ await client.connect();
217
+ const closeMock = client['client'].close;
218
+ await client.disconnect();
219
+ expect(closeMock).toHaveBeenCalled();
220
+ expect(client['client']).toBeNull();
221
+ });
222
+ it('should handle disconnect when not connected', async () => {
223
+ const config = {
224
+ mode: 'http',
225
+ url: 'http://api.test.com',
226
+ };
227
+ const client = new QwickBrainClient(config);
228
+ // Should not throw
229
+ await expect(client.disconnect()).resolves.toBeUndefined();
230
+ });
231
+ });
232
+ });
233
+ //# sourceMappingURL=qwickbrain-client.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"qwickbrain-client.test.js","sourceRoot":"","sources":["../../../src/lib/__tests__/qwickbrain-client.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAG3D,mBAAmB;AACnB,EAAE,CAAC,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QACxC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAC7C,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAC3C,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;QACjB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;KACpD,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC9B,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,yCAAyC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC5B,CAAC,CAAC,CAAC;AAEJ,2BAA2B;AAC3B,MAAM,CAAC,KAAK,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAEvB,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,IAAI,MAAwB,CAAC;QAC7B,IAAI,MAA4B,CAAC;QAEjC,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,GAAG;gBACP,IAAI,EAAE,KAAK;gBACX,GAAG,EAAE,wBAAwB;aAC9B,CAAC;YAEF,MAAM,GAAG,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;YAEvB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;YAEvB,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;4BACnB,QAAQ,EAAE;gCACR,OAAO,EAAE,cAAc;gCACvB,QAAQ,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE;6BACzB;yBACF,CAAC;qBACH;iBACF;aACF,CAAC;YAED,MAAM,CAAC,QAAQ,CAAE,CAAC,QAAgB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAEpE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;YAEvB,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,OAAO,EAAE,eAAe;wBAC9B,IAAI,EAAE,SAAS;qBAChB;iBACF;aACF,CAAC;YAED,MAAM,CAAC,QAAQ,CAAE,CAAC,QAAgB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAEpE,MAAM,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACzE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,IAAI,MAAwB,CAAC;QAC7B,IAAI,MAA4B,CAAC;QAEjC,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,GAAG;gBACP,IAAI,EAAE,MAAM;gBACZ,GAAG,EAAE,qBAAqB;gBAC1B,MAAM,EAAE,UAAU;aACnB,CAAC;YAEF,MAAM,GAAG,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE,cAAc;gBACvB,QAAQ,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE;aACzB,CAAC;YAED,MAAM,CAAC,KAAa,CAAC,iBAAiB,CAAC;gBACtC,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,YAAY;aAC/B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;YAEhD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CACvC,kCAAkC,EAClC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,iBAAiB;iBACnC,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,CAAC,KAAa,CAAC,iBAAiB,CAAC;gBACtC,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,GAAG;gBACX,UAAU,EAAE,WAAW;aACxB,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACnF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,eAAe,GAAG;gBACtB,mCAAmC;gBACnC,QAAQ,EAAE,EAAE;aACb,CAAC;YAED,MAAM,CAAC,KAAa,CAAC,iBAAiB,CAAC;gBACtC,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,eAAe;aAClC,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE,gBAAgB;gBACzB,QAAQ,EAAE,EAAE;aACb,CAAC;YAED,MAAM,CAAC,KAAa,CAAC,iBAAiB,CAAC;gBACtC,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,YAAY;aAC/B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAE9C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CACvC,gCAAgC,EAChC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM;aACf,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,IAAI,MAAwB,CAAC;QAC7B,IAAI,MAA4B,CAAC;QAEjC,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,GAAG;gBACP,IAAI,EAAE,KAAK;gBACX,OAAO,EAAE,KAAK;gBACd,IAAI,EAAE,CAAC,mBAAmB,CAAC;aAC5B,CAAC;YAEF,MAAM,GAAG,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;YAEvB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,aAAa,GAAyB;gBAC1C,IAAI,EAAE,KAAK;gBACX,kBAAkB;aACnB,CAAC;YAEF,MAAM,aAAa,GAAG,IAAI,gBAAgB,CAAC,aAAa,CAAC,CAAC;YAE1D,MAAM,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;QACrF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,MAAM,GAAyB;gBACnC,IAAI,EAAE,KAAK;gBACX,GAAG,EAAE,wBAAwB;aAC9B,CAAC;YAEF,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC5C,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;YAEvB,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YAE7C,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,MAAM,GAAyB;gBACnC,IAAI,EAAE,MAAM;gBACZ,GAAG,EAAE,qBAAqB;aAC3B,CAAC;YAEF,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAE3C,MAAM,CAAC,KAAa,CAAC,iBAAiB,CAAC;gBACtC,EAAE,EAAE,IAAI;aACT,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YAE7C,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,4BAA4B,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,MAAM,GAAyB;gBACnC,IAAI,EAAE,KAAK;gBACX,GAAG,EAAE,wBAAwB;aAC9B,CAAC;YAEF,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC5C,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;YAEtB,MAAM,CAAC,QAAQ,CAAE,CAAC,SAAiB,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;YAE5E,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YAE7C,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,MAAM,GAAyB;gBACnC,IAAI,EAAE,MAAM;gBACZ,cAAc;aACf,CAAC;YAEF,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAE5C,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YAE7C,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,MAAM,GAAyB;gBACnC,IAAI,EAAE,KAAK;gBACX,GAAG,EAAE,wBAAwB;aAC9B,CAAC;YAEF,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC5C,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;YAEvB,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAE,CAAC,KAAK,CAAC;YAE1C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,MAAM,GAAyB;gBACnC,IAAI,EAAE,MAAM;gBACZ,GAAG,EAAE,qBAAqB;aAC3B,CAAC;YAEF,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAE5C,mBAAmB;YACnB,MAAM,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,25 @@
1
+ import type { DB } from '../db/client.js';
2
+ import type { Config } from '../types/config.js';
3
+ interface CachedItem<T> {
4
+ data: T;
5
+ cachedAt: Date;
6
+ expiresAt: Date;
7
+ age: number;
8
+ isExpired: boolean;
9
+ }
10
+ export declare class CacheManager {
11
+ private db;
12
+ private config;
13
+ constructor(db: DB, config: Config['cache']);
14
+ private getTTL;
15
+ getDocument(docType: string, name: string, project?: string): Promise<CachedItem<any> | null>;
16
+ setDocument(docType: string, name: string, content: string, project?: string, metadata?: Record<string, unknown>): Promise<void>;
17
+ getMemory(name: string, project?: string): Promise<CachedItem<any> | null>;
18
+ setMemory(name: string, content: string, project?: string, metadata?: Record<string, unknown>): Promise<void>;
19
+ cleanupExpiredEntries(): Promise<{
20
+ documentsDeleted: number;
21
+ memoriesDeleted: number;
22
+ }>;
23
+ }
24
+ export {};
25
+ //# sourceMappingURL=cache-manager.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"cache-manager.d.ts","sourceRoot":"","sources":["../../src/lib/cache-manager.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,iBAAiB,CAAC;AAE1C,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAEjD,UAAU,UAAU,CAAC,CAAC;IACpB,IAAI,EAAE,CAAC,CAAC;IACR,QAAQ,EAAE,IAAI,CAAC;IACf,SAAS,EAAE,IAAI,CAAC;IAChB,GAAG,EAAE,MAAM,CAAC;IACZ,SAAS,EAAE,OAAO,CAAC;CACpB;AAED,qBAAa,YAAY;IAErB,OAAO,CAAC,EAAE;IACV,OAAO,CAAC,MAAM;gBADN,EAAE,EAAE,EAAE,EACN,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC;IAGjC,OAAO,CAAC,MAAM;IAUR,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;IAuC7F,WAAW,CACf,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,MAAM,EAChB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACjC,OAAO,CAAC,IAAI,CAAC;IAiCV,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;IAqC1E,SAAS,CACb,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,MAAM,EAChB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACjC,OAAO,CAAC,IAAI,CAAC;IA+BV,qBAAqB,IAAI,OAAO,CAAC;QAAE,gBAAgB,EAAE,MAAM,CAAC;QAAC,eAAe,EAAE,MAAM,CAAA;KAAE,CAAC;CAoB9F"}
@@ -0,0 +1,149 @@
1
+ import { eq, and, lte } from 'drizzle-orm';
2
+ import { documents, memories } from '../db/schema.js';
3
+ export class CacheManager {
4
+ db;
5
+ config;
6
+ constructor(db, config) {
7
+ this.db = db;
8
+ this.config = config;
9
+ }
10
+ getTTL(operation) {
11
+ const ttlMap = {
12
+ get_workflow: this.config.ttl.workflows,
13
+ get_document: this.config.ttl.documents,
14
+ get_memory: this.config.ttl.memories,
15
+ };
16
+ return ttlMap[operation] || 0;
17
+ }
18
+ async getDocument(docType, name, project) {
19
+ const projectValue = project || '';
20
+ const [cached] = await this.db
21
+ .select()
22
+ .from(documents)
23
+ .where(and(eq(documents.docType, docType), eq(documents.name, name), eq(documents.project, projectValue)))
24
+ .limit(1);
25
+ if (!cached) {
26
+ return null;
27
+ }
28
+ const now = new Date();
29
+ const age = Math.floor((now.getTime() - cached.cachedAt.getTime()) / 1000);
30
+ // Fix: Compare timestamp values explicitly to avoid Date comparison issues
31
+ const isExpired = now.getTime() > cached.expiresAt.getTime();
32
+ return {
33
+ data: {
34
+ name: cached.name,
35
+ doc_type: cached.docType,
36
+ project: cached.project,
37
+ content: cached.content,
38
+ metadata: cached.metadata ? JSON.parse(cached.metadata) : {},
39
+ },
40
+ cachedAt: cached.cachedAt,
41
+ expiresAt: cached.expiresAt,
42
+ age,
43
+ isExpired,
44
+ };
45
+ }
46
+ async setDocument(docType, name, content, project, metadata) {
47
+ const now = new Date();
48
+ const ttl = this.getTTL('get_document');
49
+ const expiresAt = new Date(now.getTime() + ttl * 1000);
50
+ // Use empty string instead of null for project to make unique constraint work
51
+ // SQLite treats NULL as distinct values in unique constraints
52
+ const projectValue = project || '';
53
+ await this.db
54
+ .insert(documents)
55
+ .values({
56
+ docType,
57
+ name,
58
+ project: projectValue,
59
+ content,
60
+ metadata: metadata ? JSON.stringify(metadata) : null,
61
+ cachedAt: now,
62
+ expiresAt,
63
+ synced: true,
64
+ })
65
+ .onConflictDoUpdate({
66
+ target: [documents.docType, documents.name, documents.project],
67
+ set: {
68
+ content,
69
+ metadata: metadata ? JSON.stringify(metadata) : null,
70
+ cachedAt: now,
71
+ expiresAt,
72
+ synced: true,
73
+ },
74
+ });
75
+ }
76
+ async getMemory(name, project) {
77
+ const projectValue = project || '';
78
+ const [cached] = await this.db
79
+ .select()
80
+ .from(memories)
81
+ .where(and(eq(memories.name, name), eq(memories.project, projectValue)))
82
+ .limit(1);
83
+ if (!cached) {
84
+ return null;
85
+ }
86
+ const now = new Date();
87
+ const age = Math.floor((now.getTime() - cached.cachedAt.getTime()) / 1000);
88
+ // Fix: Compare timestamp values explicitly to avoid Date comparison issues
89
+ const isExpired = now.getTime() > cached.expiresAt.getTime();
90
+ return {
91
+ data: {
92
+ name: cached.name,
93
+ project: cached.project,
94
+ content: cached.content,
95
+ metadata: cached.metadata ? JSON.parse(cached.metadata) : {},
96
+ },
97
+ cachedAt: cached.cachedAt,
98
+ expiresAt: cached.expiresAt,
99
+ age,
100
+ isExpired,
101
+ };
102
+ }
103
+ async setMemory(name, content, project, metadata) {
104
+ const now = new Date();
105
+ const ttl = this.getTTL('get_memory');
106
+ const expiresAt = new Date(now.getTime() + ttl * 1000);
107
+ // Use empty string instead of null for project to make unique constraint work
108
+ const projectValue = project || '';
109
+ await this.db
110
+ .insert(memories)
111
+ .values({
112
+ name,
113
+ project: projectValue,
114
+ content,
115
+ metadata: metadata ? JSON.stringify(metadata) : null,
116
+ cachedAt: now,
117
+ expiresAt,
118
+ synced: true,
119
+ })
120
+ .onConflictDoUpdate({
121
+ target: [memories.name, memories.project],
122
+ set: {
123
+ content,
124
+ metadata: metadata ? JSON.stringify(metadata) : null,
125
+ cachedAt: now,
126
+ expiresAt,
127
+ synced: true,
128
+ },
129
+ });
130
+ }
131
+ async cleanupExpiredEntries() {
132
+ const now = new Date();
133
+ // Delete expired documents (use lte to include items expiring exactly now)
134
+ const deletedDocs = await this.db
135
+ .delete(documents)
136
+ .where(lte(documents.expiresAt, now))
137
+ .returning({ id: documents.id });
138
+ // Delete expired memories (use lte to include items expiring exactly now)
139
+ const deletedMems = await this.db
140
+ .delete(memories)
141
+ .where(lte(memories.expiresAt, now))
142
+ .returning({ id: memories.id });
143
+ return {
144
+ documentsDeleted: deletedDocs.length,
145
+ memoriesDeleted: deletedMems.length,
146
+ };
147
+ }
148
+ }
149
+ //# sourceMappingURL=cache-manager.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"cache-manager.js","sourceRoot":"","sources":["../../src/lib/cache-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,GAAG,EAAM,GAAG,EAAO,MAAM,aAAa,CAAC;AAEpD,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAWtD,MAAM,OAAO,YAAY;IAEb;IACA;IAFV,YACU,EAAM,EACN,MAAuB;QADvB,OAAE,GAAF,EAAE,CAAI;QACN,WAAM,GAAN,MAAM,CAAiB;IAC9B,CAAC;IAEI,MAAM,CAAC,SAAiB;QAC9B,MAAM,MAAM,GAA2B;YACrC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS;YACvC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS;YACvC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ;SACrC,CAAC;QAEF,OAAO,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAAe,EAAE,IAAY,EAAE,OAAgB;QAC/D,MAAM,YAAY,GAAG,OAAO,IAAI,EAAE,CAAC;QAEnC,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,EAAE;aAC3B,MAAM,EAAE;aACR,IAAI,CAAC,SAAS,CAAC;aACf,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,EAC9B,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,EACxB,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,YAAY,CAAC,CACpC,CACF;aACA,KAAK,CAAC,CAAC,CAAC,CAAC;QAEZ,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;QAC3E,2EAA2E;QAC3E,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QAE7D,OAAO;YACL,IAAI,EAAE;gBACJ,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,QAAQ,EAAE,MAAM,CAAC,OAAO;gBACxB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE;aAC7D;YACD,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,GAAG;YACH,SAAS;SACV,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CACf,OAAe,EACf,IAAY,EACZ,OAAe,EACf,OAAgB,EAChB,QAAkC;QAElC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC;QAEvD,8EAA8E;QAC9E,8DAA8D;QAC9D,MAAM,YAAY,GAAG,OAAO,IAAI,EAAE,CAAC;QAEnC,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,SAAS,CAAC;aACjB,MAAM,CAAC;YACN,OAAO;YACP,IAAI;YACJ,OAAO,EAAE,YAAY;YACrB,OAAO;YACP,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI;YACpD,QAAQ,EAAE,GAAG;YACb,SAAS;YACT,MAAM,EAAE,IAAI;SACb,CAAC;aACD,kBAAkB,CAAC;YAClB,MAAM,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC,OAAO,CAAC;YAC9D,GAAG,EAAE;gBACH,OAAO;gBACP,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI;gBACpD,QAAQ,EAAE,GAAG;gBACb,SAAS;gBACT,MAAM,EAAE,IAAI;aACb;SACF,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAY,EAAE,OAAgB;QAC5C,MAAM,YAAY,GAAG,OAAO,IAAI,EAAE,CAAC;QAEnC,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,EAAE;aAC3B,MAAM,EAAE;aACR,IAAI,CAAC,QAAQ,CAAC;aACd,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,EACvB,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,YAAY,CAAC,CACnC,CACF;aACA,KAAK,CAAC,CAAC,CAAC,CAAC;QAEZ,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;QAC3E,2EAA2E;QAC3E,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QAE7D,OAAO;YACL,IAAI,EAAE;gBACJ,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE;aAC7D;YACD,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,GAAG;YACH,SAAS;SACV,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,SAAS,CACb,IAAY,EACZ,OAAe,EACf,OAAgB,EAChB,QAAkC;QAElC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACtC,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC;QAEvD,8EAA8E;QAC9E,MAAM,YAAY,GAAG,OAAO,IAAI,EAAE,CAAC;QAEnC,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,QAAQ,CAAC;aAChB,MAAM,CAAC;YACN,IAAI;YACJ,OAAO,EAAE,YAAY;YACrB,OAAO;YACP,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI;YACpD,QAAQ,EAAE,GAAG;YACb,SAAS;YACT,MAAM,EAAE,IAAI;SACb,CAAC;aACD,kBAAkB,CAAC;YAClB,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC;YACzC,GAAG,EAAE;gBACH,OAAO;gBACP,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI;gBACpD,QAAQ,EAAE,GAAG;gBACb,SAAS;gBACT,MAAM,EAAE,IAAI;aACb;SACF,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,qBAAqB;QACzB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QAEvB,2EAA2E;QAC3E,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,EAAE;aAC9B,MAAM,CAAC,SAAS,CAAC;aACjB,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;aACpC,SAAS,CAAC,EAAE,EAAE,EAAE,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC;QAEnC,0EAA0E;QAC1E,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,EAAE;aAC9B,MAAM,CAAC,QAAQ,CAAC;aAChB,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;aACnC,SAAS,CAAC,EAAE,EAAE,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC;QAElC,OAAO;YACL,gBAAgB,EAAE,WAAW,CAAC,MAAM;YACpC,eAAe,EAAE,WAAW,CAAC,MAAM;SACpC,CAAC;IACJ,CAAC;CACF"}
@@ -0,0 +1,26 @@
1
+ import { EventEmitter } from 'events';
2
+ import { ConnectionState } from '../types/mcp.js';
3
+ import type { Config } from '../types/config.js';
4
+ import type { QwickBrainClient } from './qwickbrain-client.js';
5
+ export declare class ConnectionManager extends EventEmitter {
6
+ private state;
7
+ private reconnectAttempts;
8
+ private reconnectTimer;
9
+ private healthCheckTimer;
10
+ private qwickbrainClient;
11
+ private config;
12
+ private isStopped;
13
+ private executionLock;
14
+ constructor(qwickbrainClient: QwickBrainClient, config: Config['connection']);
15
+ getState(): ConnectionState;
16
+ setState(state: ConnectionState): void;
17
+ start(): Promise<void>;
18
+ stop(): void;
19
+ healthCheck(): Promise<boolean>;
20
+ private scheduleHealthCheck;
21
+ private scheduleReconnect;
22
+ private clearReconnectTimer;
23
+ recordFailure(): void;
24
+ execute<T>(fn: () => Promise<T>): Promise<T>;
25
+ }
26
+ //# sourceMappingURL=connection-manager.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"connection-manager.d.ts","sourceRoot":"","sources":["../../src/lib/connection-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE/D,qBAAa,iBAAkB,SAAQ,YAAY;IACjD,OAAO,CAAC,KAAK,CAAmC;IAChD,OAAO,CAAC,iBAAiB,CAAK;IAC9B,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,gBAAgB,CAA+B;IACvD,OAAO,CAAC,gBAAgB,CAAmB;IAC3C,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,aAAa,CAAoC;gBAE7C,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,EAAE,MAAM,CAAC,YAAY,CAAC;IAM5E,QAAQ,IAAI,eAAe;IAI3B,QAAQ,CAAC,KAAK,EAAE,eAAe,GAAG,IAAI;IAShC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAM5B,IAAI,IAAI,IAAI;IAaN,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;IAyBrC,OAAO,CAAC,mBAAmB;IAmB3B,OAAO,CAAC,iBAAiB;IA4BzB,OAAO,CAAC,mBAAmB;IAO3B,aAAa,IAAI,IAAI;IAOf,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;CAgBnD"}
@@ -0,0 +1,130 @@
1
+ import { EventEmitter } from 'events';
2
+ export class ConnectionManager extends EventEmitter {
3
+ state = 'disconnected';
4
+ reconnectAttempts = 0;
5
+ reconnectTimer = null;
6
+ healthCheckTimer = null;
7
+ qwickbrainClient;
8
+ config;
9
+ isStopped = false;
10
+ executionLock = Promise.resolve();
11
+ constructor(qwickbrainClient, config) {
12
+ super();
13
+ this.qwickbrainClient = qwickbrainClient;
14
+ this.config = config;
15
+ }
16
+ getState() {
17
+ return this.state;
18
+ }
19
+ setState(state) {
20
+ const previousState = this.state;
21
+ this.state = state;
22
+ if (previousState !== state) {
23
+ this.emit('stateChange', { from: previousState, to: state });
24
+ }
25
+ }
26
+ async start() {
27
+ this.isStopped = false;
28
+ await this.healthCheck();
29
+ this.scheduleHealthCheck();
30
+ }
31
+ stop() {
32
+ this.isStopped = true;
33
+ if (this.healthCheckTimer) {
34
+ clearTimeout(this.healthCheckTimer);
35
+ this.healthCheckTimer = null;
36
+ }
37
+ if (this.reconnectTimer) {
38
+ clearTimeout(this.reconnectTimer);
39
+ this.reconnectTimer = null;
40
+ }
41
+ this.setState('offline');
42
+ }
43
+ async healthCheck() {
44
+ const startTime = Date.now();
45
+ try {
46
+ const isHealthy = await this.qwickbrainClient.healthCheck();
47
+ const latencyMs = Date.now() - startTime;
48
+ if (isHealthy) {
49
+ this.setState('connected');
50
+ this.reconnectAttempts = 0;
51
+ this.clearReconnectTimer();
52
+ this.emit('connected', { latencyMs });
53
+ return true;
54
+ }
55
+ else {
56
+ throw new Error('Health check failed');
57
+ }
58
+ }
59
+ catch (error) {
60
+ const errorMessage = error instanceof Error ? error.message : String(error);
61
+ this.setState('disconnected');
62
+ this.emit('disconnected', { error: errorMessage });
63
+ this.scheduleReconnect();
64
+ return false;
65
+ }
66
+ }
67
+ scheduleHealthCheck() {
68
+ // Don't schedule if stopped
69
+ if (this.isStopped) {
70
+ return;
71
+ }
72
+ if (this.healthCheckTimer) {
73
+ clearTimeout(this.healthCheckTimer);
74
+ }
75
+ this.healthCheckTimer = setTimeout(async () => {
76
+ await this.healthCheck();
77
+ // Check again before rescheduling to prevent leak after stop()
78
+ if (!this.isStopped) {
79
+ this.scheduleHealthCheck();
80
+ }
81
+ }, this.config.healthCheckInterval);
82
+ }
83
+ scheduleReconnect() {
84
+ if (this.reconnectTimer) {
85
+ return; // Already scheduled
86
+ }
87
+ if (this.reconnectAttempts >= this.config.maxReconnectAttempts) {
88
+ this.setState('offline');
89
+ this.emit('maxReconnectAttemptsReached');
90
+ return;
91
+ }
92
+ const delay = Math.min(this.config.reconnectBackoff.initial *
93
+ Math.pow(this.config.reconnectBackoff.multiplier, this.reconnectAttempts), this.config.reconnectBackoff.max);
94
+ this.reconnectAttempts++;
95
+ this.setState('reconnecting');
96
+ this.reconnectTimer = setTimeout(async () => {
97
+ this.reconnectTimer = null;
98
+ await this.healthCheck();
99
+ }, delay);
100
+ this.emit('reconnecting', { attempt: this.reconnectAttempts, delay });
101
+ }
102
+ clearReconnectTimer() {
103
+ if (this.reconnectTimer) {
104
+ clearTimeout(this.reconnectTimer);
105
+ this.reconnectTimer = null;
106
+ }
107
+ }
108
+ recordFailure() {
109
+ if (this.state === 'connected') {
110
+ this.setState('disconnected');
111
+ this.scheduleReconnect();
112
+ }
113
+ }
114
+ async execute(fn) {
115
+ // Wait for any ongoing state transitions to complete
116
+ await this.executionLock;
117
+ // Atomically check state and execute
118
+ if (this.state !== 'connected') {
119
+ throw new Error(`QwickBrain unavailable (state: ${this.state})`);
120
+ }
121
+ try {
122
+ return await fn();
123
+ }
124
+ catch (error) {
125
+ this.recordFailure();
126
+ throw error;
127
+ }
128
+ }
129
+ }
130
+ //# sourceMappingURL=connection-manager.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"connection-manager.js","sourceRoot":"","sources":["../../src/lib/connection-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAKtC,MAAM,OAAO,iBAAkB,SAAQ,YAAY;IACzC,KAAK,GAAoB,cAAc,CAAC;IACxC,iBAAiB,GAAG,CAAC,CAAC;IACtB,cAAc,GAA0B,IAAI,CAAC;IAC7C,gBAAgB,GAA0B,IAAI,CAAC;IAC/C,gBAAgB,CAAmB;IACnC,MAAM,CAAuB;IAC7B,SAAS,GAAG,KAAK,CAAC;IAClB,aAAa,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;IAEzD,YAAY,gBAAkC,EAAE,MAA4B;QAC1E,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,QAAQ,CAAC,KAAsB;QAC7B,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC;QACjC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QAEnB,IAAI,aAAa,KAAK,KAAK,EAAE,CAAC;YAC5B,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACzB,IAAI,CAAC,mBAAmB,EAAE,CAAC;IAC7B,CAAC;IAED,IAAI;QACF,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,YAAY,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACpC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;QACD,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,WAAW;QACf,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE,CAAC;YAC5D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAEzC,IAAI,SAAS,EAAE,CAAC;gBACd,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;gBAC3B,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC;gBAC3B,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC3B,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;gBACtC,OAAO,IAAI,CAAC;YACd,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;YACnD,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACzB,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAEO,mBAAmB;QACzB,4BAA4B;QAC5B,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,YAAY,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACtC,CAAC;QAED,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;YAC5C,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;YACzB,+DAA+D;YAC/D,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACpB,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC7B,CAAC;QACH,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;IACtC,CAAC;IAEO,iBAAiB;QACvB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO,CAAC,oBAAoB;QAC9B,CAAC;QAED,IAAI,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YAC/D,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YACzC,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CACpB,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO;YAClC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,UAAU,EAAE,IAAI,CAAC,iBAAiB,CAAC,EAC3E,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,GAAG,CACjC,CAAC;QAEF,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QAE9B,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;YAC1C,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;IACxE,CAAC;IAEO,mBAAmB;QACzB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;IAED,aAAa;QACX,IAAI,IAAI,CAAC,KAAK,KAAK,WAAW,EAAE,CAAC;YAC/B,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;YAC9B,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAI,EAAoB;QACnC,qDAAqD;QACrD,MAAM,IAAI,CAAC,aAAa,CAAC;QAEzB,qCAAqC;QACrC,IAAI,IAAI,CAAC,KAAK,KAAK,WAAW,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CAAC,kCAAkC,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF"}
@@ -0,0 +1,19 @@
1
+ import type { Config } from '../types/config.js';
2
+ import type { DB } from '../db/client.js';
3
+ export declare class ProxyServer {
4
+ private server;
5
+ private connectionManager;
6
+ private cacheManager;
7
+ private qwickbrainClient;
8
+ private config;
9
+ constructor(db: DB, config: Config);
10
+ private setupConnectionListeners;
11
+ private setupHandlers;
12
+ private createMetadata;
13
+ private handleGetWorkflow;
14
+ private handleGetDocument;
15
+ private handleGetMemory;
16
+ start(): Promise<void>;
17
+ stop(): Promise<void>;
18
+ }
19
+ //# sourceMappingURL=proxy-server.d.ts.map