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.
- package/.eslintrc.json +22 -0
- package/.github/workflows/ci.yml +57 -0
- package/.prettierrc +8 -0
- package/README.md +197 -0
- package/dist/config.d.ts +20 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +41 -0
- package/dist/config.js.map +1 -0
- package/dist/contextBuilder.d.ts +16 -0
- package/dist/contextBuilder.d.ts.map +1 -0
- package/dist/contextBuilder.js +139 -0
- package/dist/contextBuilder.js.map +1 -0
- package/dist/index.d.ts +45 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +88 -0
- package/dist/index.js.map +1 -0
- package/dist/licensing.d.ts +45 -0
- package/dist/licensing.d.ts.map +1 -0
- package/dist/licensing.js +99 -0
- package/dist/licensing.js.map +1 -0
- package/dist/memoryManager.d.ts +35 -0
- package/dist/memoryManager.d.ts.map +1 -0
- package/dist/memoryManager.js +265 -0
- package/dist/memoryManager.js.map +1 -0
- package/dist/metadataStore.d.ts +24 -0
- package/dist/metadataStore.d.ts.map +1 -0
- package/dist/metadataStore.js +247 -0
- package/dist/metadataStore.js.map +1 -0
- package/dist/planManager.d.ts +80 -0
- package/dist/planManager.d.ts.map +1 -0
- package/dist/planManager.js +327 -0
- package/dist/planManager.js.map +1 -0
- package/dist/rateLimiter.d.ts +49 -0
- package/dist/rateLimiter.d.ts.map +1 -0
- package/dist/rateLimiter.js +142 -0
- package/dist/rateLimiter.js.map +1 -0
- package/dist/storage/index.d.ts +3 -0
- package/dist/storage/index.d.ts.map +1 -0
- package/dist/storage/index.js +3 -0
- package/dist/storage/index.js.map +1 -0
- package/dist/storage/postgres.d.ts +31 -0
- package/dist/storage/postgres.d.ts.map +1 -0
- package/dist/storage/postgres.js +171 -0
- package/dist/storage/postgres.js.map +1 -0
- package/dist/storage/redis.d.ts +34 -0
- package/dist/storage/redis.d.ts.map +1 -0
- package/dist/storage/redis.js +101 -0
- package/dist/storage/redis.js.map +1 -0
- package/dist/types.d.ts +95 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/usageTracker.d.ts +63 -0
- package/dist/usageTracker.d.ts.map +1 -0
- package/dist/usageTracker.js +238 -0
- package/dist/usageTracker.js.map +1 -0
- package/dist/vectorStore.d.ts +18 -0
- package/dist/vectorStore.d.ts.map +1 -0
- package/dist/vectorStore.js +97 -0
- package/dist/vectorStore.js.map +1 -0
- package/examples/advanced.ts +164 -0
- package/examples/basic.ts +87 -0
- package/package.json +60 -0
- package/src/config.ts +65 -0
- package/src/contextBuilder.ts +184 -0
- package/src/index.ts +209 -0
- package/src/licensing.ts +138 -0
- package/src/memoryManager.ts +340 -0
- package/src/metadataStore.ts +298 -0
- package/src/planManager.ts +417 -0
- package/src/rateLimiter.ts +186 -0
- package/src/storage/index.ts +2 -0
- package/src/storage/postgres.ts +209 -0
- package/src/storage/redis.ts +117 -0
- package/src/types.ts +114 -0
- package/src/usageTracker.ts +325 -0
- package/src/vectorStore.ts +116 -0
- package/tests/aibrain.test.ts +171 -0
- package/tests/contextBuilder.test.ts +138 -0
- package/tests/memoryManager.test.ts +205 -0
- package/tests/metadataStore.test.ts +131 -0
- package/tests/rateLimiter.test.ts +57 -0
- package/tests/usageTracker.test.ts +62 -0
- package/tests/vectorStore.test.ts +106 -0
- package/tsconfig.json +25 -0
- 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 @@
|
|
|
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 @@
|
|
|
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"}
|