aimemory-core 1.0.1

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 (86) hide show
  1. package/.eslintrc.json +22 -0
  2. package/.github/workflows/ci.yml +57 -0
  3. package/.prettierrc +8 -0
  4. package/README.md +197 -0
  5. package/dist/config.d.ts +20 -0
  6. package/dist/config.d.ts.map +1 -0
  7. package/dist/config.js +41 -0
  8. package/dist/config.js.map +1 -0
  9. package/dist/contextBuilder.d.ts +16 -0
  10. package/dist/contextBuilder.d.ts.map +1 -0
  11. package/dist/contextBuilder.js +139 -0
  12. package/dist/contextBuilder.js.map +1 -0
  13. package/dist/index.d.ts +45 -0
  14. package/dist/index.d.ts.map +1 -0
  15. package/dist/index.js +88 -0
  16. package/dist/index.js.map +1 -0
  17. package/dist/licensing.d.ts +45 -0
  18. package/dist/licensing.d.ts.map +1 -0
  19. package/dist/licensing.js +99 -0
  20. package/dist/licensing.js.map +1 -0
  21. package/dist/memoryManager.d.ts +35 -0
  22. package/dist/memoryManager.d.ts.map +1 -0
  23. package/dist/memoryManager.js +265 -0
  24. package/dist/memoryManager.js.map +1 -0
  25. package/dist/metadataStore.d.ts +24 -0
  26. package/dist/metadataStore.d.ts.map +1 -0
  27. package/dist/metadataStore.js +247 -0
  28. package/dist/metadataStore.js.map +1 -0
  29. package/dist/planManager.d.ts +80 -0
  30. package/dist/planManager.d.ts.map +1 -0
  31. package/dist/planManager.js +327 -0
  32. package/dist/planManager.js.map +1 -0
  33. package/dist/rateLimiter.d.ts +49 -0
  34. package/dist/rateLimiter.d.ts.map +1 -0
  35. package/dist/rateLimiter.js +142 -0
  36. package/dist/rateLimiter.js.map +1 -0
  37. package/dist/storage/index.d.ts +3 -0
  38. package/dist/storage/index.d.ts.map +1 -0
  39. package/dist/storage/index.js +3 -0
  40. package/dist/storage/index.js.map +1 -0
  41. package/dist/storage/postgres.d.ts +31 -0
  42. package/dist/storage/postgres.d.ts.map +1 -0
  43. package/dist/storage/postgres.js +171 -0
  44. package/dist/storage/postgres.js.map +1 -0
  45. package/dist/storage/redis.d.ts +34 -0
  46. package/dist/storage/redis.d.ts.map +1 -0
  47. package/dist/storage/redis.js +101 -0
  48. package/dist/storage/redis.js.map +1 -0
  49. package/dist/types.d.ts +95 -0
  50. package/dist/types.d.ts.map +1 -0
  51. package/dist/types.js +2 -0
  52. package/dist/types.js.map +1 -0
  53. package/dist/usageTracker.d.ts +63 -0
  54. package/dist/usageTracker.d.ts.map +1 -0
  55. package/dist/usageTracker.js +238 -0
  56. package/dist/usageTracker.js.map +1 -0
  57. package/dist/vectorStore.d.ts +18 -0
  58. package/dist/vectorStore.d.ts.map +1 -0
  59. package/dist/vectorStore.js +97 -0
  60. package/dist/vectorStore.js.map +1 -0
  61. package/examples/advanced.ts +164 -0
  62. package/examples/basic.ts +87 -0
  63. package/package.json +60 -0
  64. package/src/config.ts +65 -0
  65. package/src/contextBuilder.ts +184 -0
  66. package/src/index.ts +209 -0
  67. package/src/licensing.ts +138 -0
  68. package/src/memoryManager.ts +340 -0
  69. package/src/metadataStore.ts +298 -0
  70. package/src/planManager.ts +417 -0
  71. package/src/rateLimiter.ts +186 -0
  72. package/src/storage/index.ts +2 -0
  73. package/src/storage/postgres.ts +209 -0
  74. package/src/storage/redis.ts +117 -0
  75. package/src/types.ts +114 -0
  76. package/src/usageTracker.ts +325 -0
  77. package/src/vectorStore.ts +116 -0
  78. package/tests/aibrain.test.ts +171 -0
  79. package/tests/contextBuilder.test.ts +138 -0
  80. package/tests/memoryManager.test.ts +205 -0
  81. package/tests/metadataStore.test.ts +131 -0
  82. package/tests/rateLimiter.test.ts +57 -0
  83. package/tests/usageTracker.test.ts +62 -0
  84. package/tests/vectorStore.test.ts +106 -0
  85. package/tsconfig.json +25 -0
  86. package/vitest.config.ts +12 -0
@@ -0,0 +1 @@
1
+ {"version":3,"file":"planManager.js","sourceRoot":"","sources":["../src/planManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAmB,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACrF,OAAO,EAAE,YAAY,EAAiC,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAItG,OAAO,EAAkB,cAAc,EAAE,MAAM,aAAa,CAAC;AA0C7D,MAAM,CAAC,MAAM,KAAK,GAAoC;IACpD,YAAY;IACZ,IAAI,EAAE;QACJ,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,CAAC;QACR,aAAa,EAAE,SAAS;QACxB,WAAW,EAAE,KAAK;QAClB,QAAQ,EAAE;YACR,WAAW,EAAE,IAAI;YACjB,eAAe,EAAE,GAAG;YACpB,gBAAgB,EAAE,GAAG;YACrB,iBAAiB,EAAE,IAAI;YACvB,YAAY,EAAE,EAAE;YAChB,kBAAkB,EAAE,IAAI;YACxB,WAAW,EAAE,QAAQ;YACrB,OAAO,EAAE,WAAW;YACpB,kBAAkB,EAAE,KAAK;YACzB,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,KAAK;YAChB,cAAc,EAAE,KAAK;YACrB,WAAW,EAAE,KAAK;SACnB;KACF;IACD,OAAO,EAAE;QACP,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,CAAC;QACR,aAAa,EAAE,SAAS;QACxB,WAAW,EAAE,WAAW;QACxB,QAAQ,EAAE;YACR,WAAW,EAAE,KAAK;YAClB,eAAe,EAAE,IAAI;YACrB,gBAAgB,EAAE,GAAG;YACrB,iBAAiB,EAAE,KAAK;YACxB,YAAY,EAAE,GAAG;YACjB,kBAAkB,EAAE,IAAI;YACxB,WAAW,EAAE,QAAQ;YACrB,OAAO,EAAE,WAAW;YACpB,kBAAkB,EAAE,KAAK;YACzB,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,IAAI;YACf,cAAc,EAAE,KAAK;YACrB,WAAW,EAAE,KAAK;SACnB;KACF;IACD,GAAG,EAAE;QACH,IAAI,EAAE,KAAK;QACX,KAAK,EAAE,EAAE;QACT,aAAa,EAAE,SAAS;QACxB,WAAW,EAAE,KAAK;QAClB,QAAQ,EAAE;YACR,WAAW,EAAE,KAAK;YAClB,eAAe,EAAE,KAAK;YACtB,gBAAgB,EAAE,IAAI;YACtB,iBAAiB,EAAE,MAAM;YACzB,YAAY,EAAE,GAAG;YACjB,kBAAkB,EAAE,IAAI;YACxB,WAAW,EAAE,OAAO;YACpB,OAAO,EAAE,OAAO;YAChB,kBAAkB,EAAE,KAAK;YACzB,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,IAAI;YACf,cAAc,EAAE,IAAI;YACpB,WAAW,EAAE,KAAK;SACnB;KACF;IACD,IAAI,EAAE;QACJ,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,CAAC;QACR,aAAa,EAAE,SAAS;QACxB,WAAW,EAAE,WAAW;QACxB,QAAQ,EAAE;YACR,WAAW,EAAE,MAAM;YACnB,eAAe,EAAE,IAAI;YACrB,gBAAgB,EAAE,IAAI;YACtB,iBAAiB,EAAE,KAAK;YACxB,YAAY,EAAE,GAAG;YACjB,kBAAkB,EAAE,IAAI;YACxB,WAAW,EAAE,QAAQ;YACrB,OAAO,EAAE,OAAO;YAChB,kBAAkB,EAAE,KAAK;YACzB,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,IAAI;YACf,cAAc,EAAE,IAAI;YACpB,WAAW,EAAE,KAAK;SACnB;KACF;IACD,OAAO,EAAE;QACP,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,EAAE;QACT,aAAa,EAAE,SAAS;QACxB,WAAW,EAAE,WAAW;QACxB,QAAQ,EAAE;YACR,WAAW,EAAE,CAAC,CAAC;YACf,eAAe,EAAE,CAAC,CAAC;YACnB,gBAAgB,EAAE,CAAC,CAAC;YACpB,iBAAiB,EAAE,CAAC,CAAC;YACrB,YAAY,EAAE,CAAC,CAAC;YAChB,kBAAkB,EAAE,IAAI;YACxB,WAAW,EAAE,OAAO;YACpB,OAAO,EAAE,UAAU;YACnB,kBAAkB,EAAE,IAAI;YACxB,OAAO,EAAE,EAAE;YACX,WAAW,EAAE,EAAE;YACf,SAAS,EAAE,IAAI;YACf,cAAc,EAAE,IAAI;YACpB,WAAW,EAAE,IAAI;SAClB;KACF;IACD,aAAa;IACb,IAAI,EAAE;QACJ,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,GAAG;QACV,aAAa,EAAE,SAAS;QACxB,WAAW,EAAE,MAAM;QACnB,QAAQ,EAAE;YACR,WAAW,EAAE,MAAM;YACnB,eAAe,EAAE,MAAM;YACvB,gBAAgB,EAAE,KAAK;YACvB,iBAAiB,EAAE,OAAO;YAC1B,YAAY,EAAE,IAAI;YAClB,kBAAkB,EAAE,IAAI;YACxB,WAAW,EAAE,OAAO;YACpB,OAAO,EAAE,OAAO;YAChB,kBAAkB,EAAE,IAAI;YACxB,OAAO,EAAE,EAAE;YACX,WAAW,EAAE,EAAE;YACf,SAAS,EAAE,IAAI;YACf,cAAc,EAAE,IAAI;YACpB,WAAW,EAAE,IAAI;SAClB;KACF;IACD,QAAQ,EAAE;QACR,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,GAAG;QACV,aAAa,EAAE,SAAS;QACxB,WAAW,EAAE,MAAM;QACnB,QAAQ,EAAE;YACR,WAAW,EAAE,OAAO;YACpB,eAAe,EAAE,MAAM;YACvB,gBAAgB,EAAE,MAAM;YACxB,iBAAiB,EAAE,OAAO;YAC1B,YAAY,EAAE,KAAK;YACnB,kBAAkB,EAAE,IAAI;YACxB,WAAW,EAAE,UAAU;YACvB,OAAO,EAAE,UAAU;YACnB,kBAAkB,EAAE,IAAI;YACxB,OAAO,EAAE,GAAG;YACZ,WAAW,EAAE,EAAE;YACf,SAAS,EAAE,IAAI;YACf,cAAc,EAAE,IAAI;YACpB,WAAW,EAAE,IAAI;SAClB;KACF;IACD,UAAU,EAAE;QACV,IAAI,EAAE,YAAY;QAClB,KAAK,EAAE,GAAG;QACV,aAAa,EAAE,SAAS;QACxB,WAAW,EAAE,MAAM;QACnB,QAAQ,EAAE;YACR,WAAW,EAAE,CAAC,CAAC;YACf,eAAe,EAAE,CAAC,CAAC;YACnB,gBAAgB,EAAE,CAAC,CAAC;YACpB,iBAAiB,EAAE,CAAC,CAAC;YACrB,YAAY,EAAE,CAAC,CAAC;YAChB,kBAAkB,EAAE,IAAI;YACxB,WAAW,EAAE,UAAU;YACvB,OAAO,EAAE,UAAU;YACnB,kBAAkB,EAAE,IAAI;YACxB,OAAO,EAAE,CAAC,CAAC;YACX,WAAW,EAAE,CAAC,CAAC;YACf,SAAS,EAAE,IAAI;YACf,cAAc,EAAE,IAAI;YACpB,WAAW,EAAE,IAAI;SAClB;KACF;CACF,CAAC;AAEF,MAAM,UAAU,aAAa,CAAC,IAAiB;IAC7C,MAAM,MAAM,GAAG,mBAAmB,CAAC,IAAI,CAAC,IAAI,mBAAmB,CAAC,IAAI,CAAC;IACrE,MAAM,KAAK,GAAG,oBAAoB,CAAC,IAAI,CAAC,IAAI,oBAAoB,CAAC,IAAI,CAAC;IACtE,OAAO;QACL,SAAS,EAAE,MAAM;QACjB,KAAK,EAAE,KAAK;KACb,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,OAAO,CAAC,IAAiB;IACvC,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC;AACrB,CAAC;AAED,MAAM,OAAO,WAAW;IACd,SAAS,GAA6B,IAAI,GAAG,EAAE,CAAC;IAChD,YAAY,GAA6B,IAAI,GAAG,EAAE,CAAC;IACnD,aAAa,GAA8B,IAAI,GAAG,EAAE,CAAC;IACrD,WAAW,GAAgB,MAAM,CAAC;IAE1C,WAAW,CAAC,MAAc,EAAE,IAAiB;QAC3C,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAEjC,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QACjE,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,WAAW,CAAC,MAAc;QACxB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,WAAW,CAAC;IACxD,CAAC;IAED,iBAAiB,CAAC,MAAc;QAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QACtC,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC;IACrB,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,IAAI,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAE5C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YACnC,OAAO,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC5C,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACzC,CAAC;QAED,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,MAAc,EAAE,MAA8E;QAClH,IAAI,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAE7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YACnC,OAAO,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC1C,CAAC;QAED,OAAO,OAAO,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,MAA8E;QAC9G,IAAI,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAE7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YACnC,OAAO,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC1C,CAAC;QAED,OAAO,OAAO,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,MAAc,EAAE,MAAsC;QAC1E,IAAI,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAE7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YACnC,OAAO,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC1C,CAAC;QAED,OAAO,OAAO,CAAC,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjD,CAAC;IAED,UAAU,CAAC,MAAc;QACvB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC9B,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACjC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACpC,CAAC;IAED,cAAc,CAAC,IAAiB;QAC9B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAED,WAAW;QACT,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC;IAED,cAAc,CAAC,IAAiB;QAC9B,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,KAAK,IAAI,CAAC,CAAC;IAClE,CAAC;IAED,aAAa,CAAC,IAAY;QACxB,OAAO,KAAK,CAAC,IAAmB,CAAC,IAAI,IAAI,CAAC;IAC5C,CAAC;CACF;AAED,MAAM,OAAO,WAAW;IACd,WAAW,CAAc;IACzB,MAAM,CAAiB;IACvB,iBAAiB,GAA6B,IAAI,CAAC;IAE3D,YAAY,MAAgC,EAAE,WAAyB;QACrE,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAC3C,IAAI,CAAC,WAAW,GAAG,WAAW,IAAI,IAAI,WAAW,EAAE,CAAC;IACtD,CAAC;IAED,oBAAoB,CAAC,EAAqB;QACxC,IAAI,CAAC,iBAAiB,GAAG,EAAE,CAAC;IAC9B,CAAC;IAED,WAAW,CAAC,MAAc,EAAE,IAAiB;QAC3C,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,MAAc;QAC9B,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAEjE,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,oCAAoC,UAAU,CAAC,UAAU,WAAW;aAC7E,CAAC;QACJ,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAEhF,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,oDAAoD;aAC7D,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,SAAS,CACb,MAAc,EACd,OAAe,EACf,QAAkC;QAElC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAE9C,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,MAAO,EAAE,CAAC;QACnC,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;QAEpF,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,OAAO,EAAE,KAAK,EAAE,0CAA0C,EAAE,CAAC;QAC/D,CAAC;QAED,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAExD,OAAO,EAAE,KAAK,EAAE,2CAA2C,EAAE,CAAC;IAChE,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED,aAAa,CAAC,MAAc;QAC1B,OAAO,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IACpD,CAAC;CACF;AAED,MAAM,UAAU,iBAAiB;IAC/B,OAAO,IAAI,WAAW,EAAE,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,iBAAiB;IAC/B,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC9B,CAAC"}
@@ -0,0 +1,49 @@
1
+ export interface RateLimitConfig {
2
+ windowMs: number;
3
+ maxRequests: number;
4
+ keyPrefix?: string;
5
+ }
6
+ export interface RateLimitResult {
7
+ allowed: boolean;
8
+ remaining: number;
9
+ resetTime: number;
10
+ retryAfter?: number;
11
+ }
12
+ export interface RateLimitStore {
13
+ get(key: string): Promise<{
14
+ count: number;
15
+ resetTime: number;
16
+ } | null>;
17
+ set(key: string, count: number, windowMs: number, resetTime: number): Promise<void>;
18
+ increment(key: string): Promise<{
19
+ count: number;
20
+ resetTime: number;
21
+ }>;
22
+ delete(key: string): Promise<void>;
23
+ reset(): Promise<void>;
24
+ }
25
+ export declare class InMemoryRateLimitStore implements RateLimitStore {
26
+ private store;
27
+ get(key: string): Promise<{
28
+ count: number;
29
+ resetTime: number;
30
+ } | null>;
31
+ set(key: string, count: number, _windowMs: number, resetTime: number): Promise<void>;
32
+ increment(key: string): Promise<{
33
+ count: number;
34
+ resetTime: number;
35
+ }>;
36
+ delete(key: string): Promise<void>;
37
+ reset(): Promise<void>;
38
+ }
39
+ export declare class RateLimiter {
40
+ private store;
41
+ private config;
42
+ constructor(config: RateLimitConfig, store?: RateLimitStore);
43
+ check(identifier: string): Promise<RateLimitResult>;
44
+ reset(identifier: string): Promise<void>;
45
+ updateConfig(config: Partial<RateLimitConfig>): void;
46
+ }
47
+ export declare const DEFAULT_RATE_LIMITS: Record<string, RateLimitConfig>;
48
+ export declare function createRateLimiter(plan?: string, store?: RateLimitStore): RateLimiter;
49
+ //# sourceMappingURL=rateLimiter.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"rateLimiter.d.ts","sourceRoot":"","sources":["../src/rateLimiter.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,cAAc;IAC7B,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC,CAAC;IACvE,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACpF,SAAS,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACtE,MAAM,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACnC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACxB;AAED,qBAAa,sBAAuB,YAAW,cAAc;IAC3D,OAAO,CAAC,KAAK,CAAgE;IAEvE,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IAYtE,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpF,SAAS,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;IAerE,MAAM,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIlC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAG7B;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,KAAK,CAAiB;IAC9B,OAAO,CAAC,MAAM,CAAkB;gBAEpB,MAAM,EAAE,eAAe,EAAE,KAAK,CAAC,EAAE,cAAc;IAKrD,KAAK,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IAgDnD,KAAK,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAK9C,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,eAAe,CAAC,GAAG,IAAI;CAGrD;AAED,eAAO,MAAM,mBAAmB,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAiD/D,CAAC;AAEF,wBAAgB,iBAAiB,CAAC,IAAI,GAAE,MAAe,EAAE,KAAK,CAAC,EAAE,cAAc,GAAG,WAAW,CAG5F"}
@@ -0,0 +1,142 @@
1
+ export class InMemoryRateLimitStore {
2
+ store = new Map();
3
+ async get(key) {
4
+ const entry = this.store.get(key);
5
+ if (!entry)
6
+ return null;
7
+ if (Date.now() > entry.resetTime) {
8
+ this.store.delete(key);
9
+ return null;
10
+ }
11
+ return entry;
12
+ }
13
+ async set(key, count, _windowMs, resetTime) {
14
+ this.store.set(key, { count, resetTime });
15
+ }
16
+ async increment(key) {
17
+ const now = Date.now();
18
+ const entry = this.store.get(key);
19
+ if (!entry || now > entry.resetTime) {
20
+ const resetTime = now + 60000;
21
+ this.store.set(key, { count: 1, resetTime });
22
+ return { count: 1, resetTime };
23
+ }
24
+ const newCount = entry.count + 1;
25
+ this.store.set(key, { count: newCount, resetTime: entry.resetTime });
26
+ return { count: newCount, resetTime: entry.resetTime };
27
+ }
28
+ async delete(key) {
29
+ this.store.delete(key);
30
+ }
31
+ async reset() {
32
+ this.store.clear();
33
+ }
34
+ }
35
+ export class RateLimiter {
36
+ store;
37
+ config;
38
+ constructor(config, store) {
39
+ this.config = config;
40
+ this.store = store || new InMemoryRateLimitStore();
41
+ }
42
+ async check(identifier) {
43
+ const key = `${this.config.keyPrefix || 'ratelimit'}:${identifier}`;
44
+ const now = Date.now();
45
+ const entry = await this.store.get(key);
46
+ if (!entry) {
47
+ const resetTime = now + this.config.windowMs;
48
+ await this.store.set(key, 1, this.config.windowMs, resetTime);
49
+ return {
50
+ allowed: true,
51
+ remaining: this.config.maxRequests - 1,
52
+ resetTime,
53
+ };
54
+ }
55
+ if (now >= entry.resetTime) {
56
+ const resetTime = now + this.config.windowMs;
57
+ await this.store.set(key, 1, this.config.windowMs, resetTime);
58
+ return {
59
+ allowed: true,
60
+ remaining: this.config.maxRequests - 1,
61
+ resetTime,
62
+ };
63
+ }
64
+ if (entry.count >= this.config.maxRequests) {
65
+ const retryAfter = Math.ceil((entry.resetTime - now) / 1000);
66
+ return {
67
+ allowed: false,
68
+ remaining: 0,
69
+ resetTime: entry.resetTime,
70
+ retryAfter,
71
+ };
72
+ }
73
+ const newEntry = await this.store.increment(key);
74
+ return {
75
+ allowed: true,
76
+ remaining: this.config.maxRequests - newEntry.count,
77
+ resetTime: newEntry.resetTime,
78
+ };
79
+ }
80
+ async reset(identifier) {
81
+ const key = `${this.config.keyPrefix || 'ratelimit'}:${identifier}`;
82
+ await this.store.delete(key);
83
+ }
84
+ updateConfig(config) {
85
+ this.config = { ...this.config, ...config };
86
+ }
87
+ }
88
+ export const DEFAULT_RATE_LIMITS = {
89
+ // NPM & Extension Free
90
+ free: {
91
+ windowMs: 60 * 1000,
92
+ maxRequests: 100,
93
+ keyPrefix: 'free',
94
+ },
95
+ // Extension Starter
96
+ starter: {
97
+ windowMs: 60 * 1000,
98
+ maxRequests: 200,
99
+ keyPrefix: 'starter',
100
+ },
101
+ // NPM Pro
102
+ pro: {
103
+ windowMs: 60 * 1000,
104
+ maxRequests: 500,
105
+ keyPrefix: 'pro',
106
+ },
107
+ // Extension Plus
108
+ plus: {
109
+ windowMs: 60 * 1000,
110
+ maxRequests: 300,
111
+ keyPrefix: 'plus',
112
+ },
113
+ // Extension Premium
114
+ premium: {
115
+ windowMs: 60 * 1000,
116
+ maxRequests: 1000,
117
+ keyPrefix: 'premium',
118
+ },
119
+ // Team
120
+ team: {
121
+ windowMs: 60 * 1000,
122
+ maxRequests: 2000,
123
+ keyPrefix: 'team',
124
+ },
125
+ // Business
126
+ business: {
127
+ windowMs: 60 * 1000,
128
+ maxRequests: 5000,
129
+ keyPrefix: 'business',
130
+ },
131
+ // Enterprise
132
+ enterprise: {
133
+ windowMs: 60 * 1000,
134
+ maxRequests: 10000,
135
+ keyPrefix: 'enterprise',
136
+ },
137
+ };
138
+ export function createRateLimiter(plan = 'free', store) {
139
+ const config = DEFAULT_RATE_LIMITS[plan] || DEFAULT_RATE_LIMITS.free;
140
+ return new RateLimiter(config, store);
141
+ }
142
+ //# sourceMappingURL=rateLimiter.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"rateLimiter.js","sourceRoot":"","sources":["../src/rateLimiter.ts"],"names":[],"mappings":"AAqBA,MAAM,OAAO,sBAAsB;IACzB,KAAK,GAAsD,IAAI,GAAG,EAAE,CAAC;IAE7E,KAAK,CAAC,GAAG,CAAC,GAAW;QACnB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;YACjC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACvB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,GAAW,EAAE,KAAa,EAAE,SAAiB,EAAE,SAAiB;QACxE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,GAAW;QACzB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAElC,IAAI,CAAC,KAAK,IAAI,GAAG,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;YACpC,MAAM,SAAS,GAAG,GAAG,GAAG,KAAK,CAAC;YAC9B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;YAC7C,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC;QACjC,CAAC;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC;QACrE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,GAAW;QACtB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;CACF;AAED,MAAM,OAAO,WAAW;IACd,KAAK,CAAiB;IACtB,MAAM,CAAkB;IAEhC,YAAY,MAAuB,EAAE,KAAsB;QACzD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,IAAI,IAAI,sBAAsB,EAAE,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,UAAkB;QAC5B,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,WAAW,IAAI,UAAU,EAAE,CAAC;QACpE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAExC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,SAAS,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;YAC7C,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;YAE9D,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,CAAC;gBACtC,SAAS;aACV,CAAC;QACJ,CAAC;QAED,IAAI,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YAC3B,MAAM,SAAS,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;YAC7C,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;YAE9D,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,CAAC;gBACtC,SAAS;aACV,CAAC;QACJ,CAAC;QAED,IAAI,KAAK,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,SAAS,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;YAE7D,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,UAAU;aACX,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAEjD,OAAO;YACL,OAAO,EAAE,IAAI;YACb,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,QAAQ,CAAC,KAAK;YACnD,SAAS,EAAE,QAAQ,CAAC,SAAS;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,UAAkB;QAC5B,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,WAAW,IAAI,UAAU,EAAE,CAAC;QACpE,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC;IAED,YAAY,CAAC,MAAgC;QAC3C,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;IAC9C,CAAC;CACF;AAED,MAAM,CAAC,MAAM,mBAAmB,GAAoC;IAClE,uBAAuB;IACvB,IAAI,EAAE;QACJ,QAAQ,EAAE,EAAE,GAAG,IAAI;QACnB,WAAW,EAAE,GAAG;QAChB,SAAS,EAAE,MAAM;KAClB;IACD,oBAAoB;IACpB,OAAO,EAAE;QACP,QAAQ,EAAE,EAAE,GAAG,IAAI;QACnB,WAAW,EAAE,GAAG;QAChB,SAAS,EAAE,SAAS;KACrB;IACD,UAAU;IACV,GAAG,EAAE;QACH,QAAQ,EAAE,EAAE,GAAG,IAAI;QACnB,WAAW,EAAE,GAAG;QAChB,SAAS,EAAE,KAAK;KACjB;IACD,iBAAiB;IACjB,IAAI,EAAE;QACJ,QAAQ,EAAE,EAAE,GAAG,IAAI;QACnB,WAAW,EAAE,GAAG;QAChB,SAAS,EAAE,MAAM;KAClB;IACD,oBAAoB;IACpB,OAAO,EAAE;QACP,QAAQ,EAAE,EAAE,GAAG,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,SAAS,EAAE,SAAS;KACrB;IACD,OAAO;IACP,IAAI,EAAE;QACJ,QAAQ,EAAE,EAAE,GAAG,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,SAAS,EAAE,MAAM;KAClB;IACD,WAAW;IACX,QAAQ,EAAE;QACR,QAAQ,EAAE,EAAE,GAAG,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,SAAS,EAAE,UAAU;KACtB;IACD,aAAa;IACb,UAAU,EAAE;QACV,QAAQ,EAAE,EAAE,GAAG,IAAI;QACnB,WAAW,EAAE,KAAK;QAClB,SAAS,EAAE,YAAY;KACxB;CACF,CAAC;AAEF,MAAM,UAAU,iBAAiB,CAAC,OAAe,MAAM,EAAE,KAAsB;IAC7E,MAAM,MAAM,GAAG,mBAAmB,CAAC,IAAI,CAAC,IAAI,mBAAmB,CAAC,IAAI,CAAC;IACrE,OAAO,IAAI,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;AACxC,CAAC"}
@@ -0,0 +1,3 @@
1
+ export { RedisRateLimitStore, createRedisRateLimiter, type RedisConfig } from './redis.js';
2
+ export { PostgresUsageStore, createPostgresUsageStore, type PostgresConfig } from './postgres.js';
3
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/storage/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,KAAK,WAAW,EAAE,MAAM,YAAY,CAAC;AAC3F,OAAO,EAAE,kBAAkB,EAAE,wBAAwB,EAAE,KAAK,cAAc,EAAE,MAAM,eAAe,CAAC"}
@@ -0,0 +1,3 @@
1
+ export { RedisRateLimitStore, createRedisRateLimiter } from './redis.js';
2
+ export { PostgresUsageStore, createPostgresUsageStore } from './postgres.js';
3
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/storage/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,sBAAsB,EAAoB,MAAM,YAAY,CAAC;AAC3F,OAAO,EAAE,kBAAkB,EAAE,wBAAwB,EAAuB,MAAM,eAAe,CAAC"}
@@ -0,0 +1,31 @@
1
+ import { UsageStore, UsageRecord, UsageMetric } from '../usageTracker';
2
+ export interface PostgresConfig {
3
+ host?: string;
4
+ port?: number;
5
+ database?: string;
6
+ user?: string;
7
+ password?: string;
8
+ connectionString?: string;
9
+ max?: number;
10
+ idleTimeoutMillis?: number;
11
+ connectionTimeoutMillis?: number;
12
+ ssl?: boolean;
13
+ }
14
+ export declare class PostgresUsageStore implements UsageStore {
15
+ private config;
16
+ private pool;
17
+ private initialized;
18
+ constructor(config: PostgresConfig);
19
+ private getPool;
20
+ private initialize;
21
+ get(userId: string, date: string): Promise<UsageRecord | null>;
22
+ set(userId: string, date: string, record: UsageRecord): Promise<void>;
23
+ increment(userId: string, metric: UsageMetric, amount?: number): Promise<void>;
24
+ getRange(userId: string, startDate: string, endDate: string): Promise<UsageRecord[]>;
25
+ reset(userId: string, date: string): Promise<void>;
26
+ deleteUser(userId: string): Promise<void>;
27
+ disconnect(): Promise<void>;
28
+ isInitialized(): boolean;
29
+ }
30
+ export declare function createPostgresUsageStore(config: PostgresConfig): PostgresUsageStore;
31
+ //# sourceMappingURL=postgres.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../../src/storage/postgres.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAEvE,MAAM,WAAW,cAAc;IAC7B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,GAAG,CAAC,EAAE,OAAO,CAAC;CACf;AAED,qBAAa,kBAAmB,YAAW,UAAU;IAIvC,OAAO,CAAC,MAAM;IAH1B,OAAO,CAAC,IAAI,CAAM;IAClB,OAAO,CAAC,WAAW,CAAkB;gBAEjB,MAAM,EAAE,cAAc;YAE5B,OAAO;YA6BP,UAAU;IAyBlB,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IA2B9D,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IA4BrE,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAE,MAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAmBjF,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAwBpF,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAYlD,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASzC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAOjC,aAAa,IAAI,OAAO;CAGzB;AAED,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,cAAc,GAAG,kBAAkB,CAEnF"}
@@ -0,0 +1,171 @@
1
+ export class PostgresUsageStore {
2
+ config;
3
+ pool;
4
+ initialized = false;
5
+ constructor(config) {
6
+ this.config = config;
7
+ }
8
+ async getPool() {
9
+ if (!this.initialized) {
10
+ try {
11
+ const { default: pg } = await import('pg');
12
+ const { Pool } = pg;
13
+ this.pool = new Pool({
14
+ host: this.config.host || 'localhost',
15
+ port: this.config.port || 5432,
16
+ database: this.config.database || 'ai_brain',
17
+ user: this.config.user || 'postgres',
18
+ password: this.config.password,
19
+ connectionString: this.config.connectionString,
20
+ max: this.config.max || 20,
21
+ idleTimeoutMillis: this.config.idleTimeoutMillis || 30000,
22
+ connectionTimeoutMillis: this.config.connectionTimeoutMillis || 5000,
23
+ ssl: this.config.ssl ? { rejectUnauthorized: false } : false,
24
+ });
25
+ await this.initialize();
26
+ this.initialized = true;
27
+ }
28
+ catch (error) {
29
+ console.error('Failed to initialize PostgreSQL:', error);
30
+ throw new Error('PostgreSQL not available');
31
+ }
32
+ }
33
+ return this.pool;
34
+ }
35
+ async initialize() {
36
+ const pool = this.pool;
37
+ await pool.query(`
38
+ CREATE TABLE IF NOT EXISTS usage_records (
39
+ id SERIAL PRIMARY KEY,
40
+ user_id VARCHAR(255) NOT NULL,
41
+ date DATE NOT NULL,
42
+ memories_added INTEGER DEFAULT 0,
43
+ memories_searched INTEGER DEFAULT 0,
44
+ context_built INTEGER DEFAULT 0,
45
+ api_calls INTEGER DEFAULT 0,
46
+ storage_used INTEGER DEFAULT 0,
47
+ created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
48
+ updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
49
+ UNIQUE(user_id, date)
50
+ );
51
+ `);
52
+ await pool.query(`
53
+ CREATE INDEX IF NOT EXISTS idx_usage_user_date
54
+ ON usage_records(user_id, date);
55
+ `);
56
+ }
57
+ async get(userId, date) {
58
+ try {
59
+ const pool = await this.getPool();
60
+ const result = await pool.query('SELECT * FROM usage_records WHERE user_id = $1 AND date = $2', [userId, date]);
61
+ if (result.rows.length === 0)
62
+ return null;
63
+ const row = result.rows[0];
64
+ return {
65
+ userId: row.user_id,
66
+ date: row.date.toISOString().split('T')[0],
67
+ metrics: {
68
+ memories_added: row.memories_added,
69
+ memories_searched: row.memories_searched,
70
+ context_built: row.context_built,
71
+ api_calls: row.api_calls,
72
+ storage_used: row.storage_used,
73
+ },
74
+ };
75
+ }
76
+ catch {
77
+ return null;
78
+ }
79
+ }
80
+ async set(userId, date, record) {
81
+ try {
82
+ const pool = await this.getPool();
83
+ await pool.query(`INSERT INTO usage_records (user_id, date, memories_added, memories_searched, context_built, api_calls, storage_used)
84
+ VALUES ($1, $2, $3, $4, $5, $6, $7)
85
+ ON CONFLICT (user_id, date) DO UPDATE SET
86
+ memories_added = EXCLUDED.memories_added,
87
+ memories_searched = EXCLUDED.memories_searched,
88
+ context_built = EXCLUDED.context_built,
89
+ api_calls = EXCLUDED.api_calls,
90
+ storage_used = EXCLUDED.storage_used,
91
+ updated_at = CURRENT_TIMESTAMP`, [
92
+ userId,
93
+ date,
94
+ record.metrics.memories_added,
95
+ record.metrics.memories_searched,
96
+ record.metrics.context_built,
97
+ record.metrics.api_calls,
98
+ record.metrics.storage_used,
99
+ ]);
100
+ }
101
+ catch (error) {
102
+ console.error('PostgreSQL set error:', error);
103
+ }
104
+ }
105
+ async increment(userId, metric, amount = 1) {
106
+ const today = new Date().toISOString().split('T')[0];
107
+ const metricColumn = metric;
108
+ try {
109
+ const pool = await this.getPool();
110
+ await pool.query(`INSERT INTO usage_records (user_id, date, ${metricColumn})
111
+ VALUES ($1, $2, $3)
112
+ ON CONFLICT (user_id, date) DO UPDATE SET
113
+ ${metricColumn} = usage_records.${metricColumn} + EXCLUDED.${metricColumn},
114
+ updated_at = CURRENT_TIMESTAMP`, [userId, today, amount]);
115
+ }
116
+ catch (error) {
117
+ console.error('PostgreSQL increment error:', error);
118
+ }
119
+ }
120
+ async getRange(userId, startDate, endDate) {
121
+ try {
122
+ const pool = await this.getPool();
123
+ const result = await pool.query('SELECT * FROM usage_records WHERE user_id = $1 AND date >= $2 AND date <= $3 ORDER BY date', [userId, startDate, endDate]);
124
+ return result.rows.map((row) => ({
125
+ userId: row.user_id,
126
+ date: row.date.toISOString().split('T')[0],
127
+ metrics: {
128
+ memories_added: row.memories_added,
129
+ memories_searched: row.memories_searched,
130
+ context_built: row.context_built,
131
+ api_calls: row.api_calls,
132
+ storage_used: row.storage_used,
133
+ },
134
+ }));
135
+ }
136
+ catch {
137
+ return [];
138
+ }
139
+ }
140
+ async reset(userId, date) {
141
+ try {
142
+ const pool = await this.getPool();
143
+ await pool.query('DELETE FROM usage_records WHERE user_id = $1 AND date = $2', [userId, date]);
144
+ }
145
+ catch (error) {
146
+ console.error('PostgreSQL reset error:', error);
147
+ }
148
+ }
149
+ async deleteUser(userId) {
150
+ try {
151
+ const pool = await this.getPool();
152
+ await pool.query('DELETE FROM usage_records WHERE user_id = $1', [userId]);
153
+ }
154
+ catch (error) {
155
+ console.error('PostgreSQL deleteUser error:', error);
156
+ }
157
+ }
158
+ async disconnect() {
159
+ if (this.pool) {
160
+ await this.pool.end();
161
+ this.initialized = false;
162
+ }
163
+ }
164
+ isInitialized() {
165
+ return this.initialized;
166
+ }
167
+ }
168
+ export function createPostgresUsageStore(config) {
169
+ return new PostgresUsageStore(config);
170
+ }
171
+ //# sourceMappingURL=postgres.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"postgres.js","sourceRoot":"","sources":["../../src/storage/postgres.ts"],"names":[],"mappings":"AAeA,MAAM,OAAO,kBAAkB;IAIT;IAHZ,IAAI,CAAM;IACV,WAAW,GAAY,KAAK,CAAC;IAErC,YAAoB,MAAsB;QAAtB,WAAM,GAAN,MAAM,CAAgB;IAAG,CAAC;IAEtC,KAAK,CAAC,OAAO;QACnB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC;gBACH,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC3C,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;gBAEpB,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,CAAC;oBACnB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,WAAW;oBACrC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,IAAI;oBAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,UAAU;oBAC5C,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,UAAU;oBACpC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;oBAC9B,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;oBAC9C,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,EAAE;oBAC1B,iBAAiB,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,KAAK;oBACzD,uBAAuB,EAAE,IAAI,CAAC,MAAM,CAAC,uBAAuB,IAAI,IAAI;oBACpE,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,KAAK;iBAC7D,CAAC,CAAC;gBAEH,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;gBACxB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YAC1B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;gBACzD,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QAEvB,MAAM,IAAI,CAAC,KAAK,CAAC;;;;;;;;;;;;;;KAchB,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,KAAK,CAAC;;;KAGhB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAAc,EAAE,IAAY;QACpC,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B,8DAA8D,EAC9D,CAAC,MAAM,EAAE,IAAI,CAAC,CACf,CAAC;YAEF,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC;YAE1C,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,OAAO;gBACL,MAAM,EAAE,GAAG,CAAC,OAAO;gBACnB,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC1C,OAAO,EAAE;oBACP,cAAc,EAAE,GAAG,CAAC,cAAc;oBAClC,iBAAiB,EAAE,GAAG,CAAC,iBAAiB;oBACxC,aAAa,EAAE,GAAG,CAAC,aAAa;oBAChC,SAAS,EAAE,GAAG,CAAC,SAAS;oBACxB,YAAY,EAAE,GAAG,CAAC,YAAY;iBAC/B;aACF,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAAc,EAAE,IAAY,EAAE,MAAmB;QACzD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,IAAI,CAAC,KAAK,CACd;;;;;;;;0CAQkC,EAClC;gBACE,MAAM;gBACN,IAAI;gBACJ,MAAM,CAAC,OAAO,CAAC,cAAc;gBAC7B,MAAM,CAAC,OAAO,CAAC,iBAAiB;gBAChC,MAAM,CAAC,OAAO,CAAC,aAAa;gBAC5B,MAAM,CAAC,OAAO,CAAC,SAAS;gBACxB,MAAM,CAAC,OAAO,CAAC,YAAY;aAC5B,CACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,MAAmB,EAAE,SAAiB,CAAC;QACrE,MAAM,KAAK,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrD,MAAM,YAAY,GAAG,MAAM,CAAC;QAE5B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,IAAI,CAAC,KAAK,CACd,6CAA6C,YAAY;;;aAGpD,YAAY,oBAAoB,YAAY,eAAe,YAAY;0CAC1C,EAClC,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,CACxB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,SAAiB,EAAE,OAAe;QAC/D,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B,4FAA4F,EAC5F,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,CAAC,CAC7B,CAAC;YAEF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,CAAC;gBACpC,MAAM,EAAE,GAAG,CAAC,OAAO;gBACnB,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC1C,OAAO,EAAE;oBACP,cAAc,EAAE,GAAG,CAAC,cAAc;oBAClC,iBAAiB,EAAE,GAAG,CAAC,iBAAiB;oBACxC,aAAa,EAAE,GAAG,CAAC,aAAa;oBAChC,SAAS,EAAE,GAAG,CAAC,SAAS;oBACxB,YAAY,EAAE,GAAG,CAAC,YAAY;iBAC/B;aACF,CAAC,CAAC,CAAC;QACN,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAc,EAAE,IAAY;QACtC,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,IAAI,CAAC,KAAK,CACd,4DAA4D,EAC5D,CAAC,MAAM,EAAE,IAAI,CAAC,CACf,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAc;QAC7B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,IAAI,CAAC,KAAK,CAAC,8CAA8C,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;QAC7E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;CACF;AAED,MAAM,UAAU,wBAAwB,CAAC,MAAsB;IAC7D,OAAO,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;AACxC,CAAC"}
@@ -0,0 +1,34 @@
1
+ import { RateLimitStore } from '../rateLimiter';
2
+ export interface RedisConfig {
3
+ host?: string;
4
+ port?: number;
5
+ url?: string;
6
+ password?: string;
7
+ db?: number;
8
+ keyPrefix?: string;
9
+ connectionTimeout?: number;
10
+ maxRetriesPerRequest?: number;
11
+ }
12
+ export declare class RedisRateLimitStore implements RateLimitStore {
13
+ private config;
14
+ private redis;
15
+ private keyPrefix;
16
+ private connected;
17
+ constructor(config: RedisConfig);
18
+ private getClient;
19
+ get(key: string): Promise<{
20
+ count: number;
21
+ resetTime: number;
22
+ } | null>;
23
+ set(key: string, count: number, windowMs: number, resetTime: number): Promise<void>;
24
+ increment(key: string): Promise<{
25
+ count: number;
26
+ resetTime: number;
27
+ }>;
28
+ delete(key: string): Promise<void>;
29
+ reset(): Promise<void>;
30
+ disconnect(): Promise<void>;
31
+ isConnected(): boolean;
32
+ }
33
+ export declare function createRedisRateLimiter(config: RedisConfig): RedisRateLimitStore;
34
+ //# sourceMappingURL=redis.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"redis.d.ts","sourceRoot":"","sources":["../../src/storage/redis.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAEhD,MAAM,WAAW,WAAW;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B;AAED,qBAAa,mBAAoB,YAAW,cAAc;IAK5C,OAAO,CAAC,MAAM;IAJ1B,OAAO,CAAC,KAAK,CAAM;IACnB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,SAAS,CAAkB;gBAEf,MAAM,EAAE,WAAW;YAIzB,SAAS;IAqBjB,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IAWtE,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAWnF,SAAS,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;IAgBrE,MAAM,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASlC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAYtB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAOjC,WAAW,IAAI,OAAO;CAGvB;AAED,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,WAAW,GAAG,mBAAmB,CAE/E"}
@@ -0,0 +1,101 @@
1
+ export class RedisRateLimitStore {
2
+ config;
3
+ redis;
4
+ keyPrefix;
5
+ connected = false;
6
+ constructor(config) {
7
+ this.config = config;
8
+ this.keyPrefix = config.keyPrefix || 'ai-memory:ratelimit:';
9
+ }
10
+ async getClient() {
11
+ if (!this.connected) {
12
+ try {
13
+ const Redis = (await import('ioredis')).default;
14
+ this.redis = new Redis({
15
+ host: this.config.host || 'localhost',
16
+ port: this.config.port || 6379,
17
+ password: this.config.password,
18
+ db: this.config.db || 0,
19
+ lazyConnect: true,
20
+ });
21
+ await this.redis.connect();
22
+ this.connected = true;
23
+ }
24
+ catch (error) {
25
+ console.error('Failed to connect to Redis:', error);
26
+ throw new Error('Redis not available');
27
+ }
28
+ }
29
+ return this.redis;
30
+ }
31
+ async get(key) {
32
+ try {
33
+ const client = await this.getClient();
34
+ const data = await client.get(this.keyPrefix + key);
35
+ if (!data)
36
+ return null;
37
+ return JSON.parse(data);
38
+ }
39
+ catch {
40
+ return null;
41
+ }
42
+ }
43
+ async set(key, count, windowMs, resetTime) {
44
+ try {
45
+ const client = await this.getClient();
46
+ const data = JSON.stringify({ count, resetTime });
47
+ const ttlSeconds = Math.ceil(windowMs / 1000);
48
+ await client.setex(this.keyPrefix + key, ttlSeconds, data);
49
+ }
50
+ catch (error) {
51
+ console.error('Redis set error:', error);
52
+ }
53
+ }
54
+ async increment(key) {
55
+ const client = await this.getClient();
56
+ const fullKey = this.keyPrefix + key;
57
+ const existing = await this.get(key);
58
+ if (!existing) {
59
+ const resetTime = Date.now() + 60000;
60
+ await this.set(key, 1, 60000, resetTime);
61
+ return { count: 1, resetTime };
62
+ }
63
+ const newCount = existing.count + 1;
64
+ await this.set(key, newCount, 60000, existing.resetTime);
65
+ return { count: newCount, resetTime: existing.resetTime };
66
+ }
67
+ async delete(key) {
68
+ try {
69
+ const client = await this.getClient();
70
+ await client.del(this.keyPrefix + key);
71
+ }
72
+ catch (error) {
73
+ console.error('Redis delete error:', error);
74
+ }
75
+ }
76
+ async reset() {
77
+ try {
78
+ const client = await this.getClient();
79
+ const keys = await client.keys(this.keyPrefix + '*');
80
+ if (keys.length > 0) {
81
+ await client.del(...keys);
82
+ }
83
+ }
84
+ catch (error) {
85
+ console.error('Redis reset error:', error);
86
+ }
87
+ }
88
+ async disconnect() {
89
+ if (this.redis && this.connected) {
90
+ await this.redis.quit();
91
+ this.connected = false;
92
+ }
93
+ }
94
+ isConnected() {
95
+ return this.connected;
96
+ }
97
+ }
98
+ export function createRedisRateLimiter(config) {
99
+ return new RedisRateLimitStore(config);
100
+ }
101
+ //# sourceMappingURL=redis.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"redis.js","sourceRoot":"","sources":["../../src/storage/redis.ts"],"names":[],"mappings":"AAaA,MAAM,OAAO,mBAAmB;IAKV;IAJZ,KAAK,CAAM;IACX,SAAS,CAAS;IAClB,SAAS,GAAY,KAAK,CAAC;IAEnC,YAAoB,MAAmB;QAAnB,WAAM,GAAN,MAAM,CAAa;QACrC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,sBAAsB,CAAC;IAC9D,CAAC;IAEO,KAAK,CAAC,SAAS;QACrB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,CAAC,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;gBAChD,IAAI,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC;oBACrB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,WAAW;oBACrC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,IAAI;oBAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;oBAC9B,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC;oBACvB,WAAW,EAAE,IAAI;iBAClB,CAAC,CAAC;gBACH,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACxB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;gBACpD,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,GAAW;QACnB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC;YACpD,IAAI,CAAC,IAAI;gBAAE,OAAO,IAAI,CAAC;YACvB,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,GAAW,EAAE,KAAa,EAAE,QAAgB,EAAE,SAAiB;QACvE,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;YAClD,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,CAAC;YAC9C,MAAM,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,GAAG,GAAG,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;QAC7D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,GAAW;QACzB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC;QAErC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;YACrC,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;YACzC,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC;QACjC,CAAC;QAED,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,GAAG,CAAC,CAAC;QACpC,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;QACzD,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,CAAC,SAAS,EAAE,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,GAAW;QACtB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,MAAM,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,KAAK,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC;YACrD,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpB,MAAM,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACjC,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YACxB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACzB,CAAC;IACH,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;CACF;AAED,MAAM,UAAU,sBAAsB,CAAC,MAAmB;IACxD,OAAO,IAAI,mBAAmB,CAAC,MAAM,CAAC,CAAC;AACzC,CAAC"}