@wildix/xbees-conversations-utils 1.2.0 → 1.2.2

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 (75) hide show
  1. package/dist-cjs/normalization.js +0 -2
  2. package/dist-cjs/streamRateLimitHandlingProxy/helpers/executeWithRateLimitHandling.js +5 -4
  3. package/dist-cjs/streamRateLimitHandlingProxy/helpers/sleep.js +6 -0
  4. package/dist-es/normalization.js +0 -2
  5. package/dist-es/streamRateLimitHandlingProxy/helpers/executeWithRateLimitHandling.js +1 -1
  6. package/dist-es/streamRateLimitHandlingProxy/helpers/sleep.js +4 -0
  7. package/dist-types/streamRateLimitHandlingProxy/helpers/sleep.d.ts +2 -0
  8. package/dist-types/types.d.ts +0 -1
  9. package/package.json +1 -1
  10. package/dist-cjs/rateLimits/RedisCooldown.js +0 -66
  11. package/dist-cjs/rateLimits/constants.js +0 -43
  12. package/dist-cjs/rateLimits/createRateLimitedStreamProxy.js +0 -92
  13. package/dist-cjs/rateLimits/exception/createRateLimitExceededException.js +0 -21
  14. package/dist-cjs/rateLimits/exception/processStreamRateLimitException.js +0 -47
  15. package/dist-cjs/rateLimits/executeStreamRequestWithRateLimitHandling.js +0 -94
  16. package/dist-cjs/rateLimits/generatedMethodNames.js +0 -320
  17. package/dist-cjs/rateLimits/helpers/RedisCooldown.js +0 -174
  18. package/dist-cjs/rateLimits/helpers/createRateLimitExceededException.js +0 -32
  19. package/dist-cjs/rateLimits/helpers/executeWithRateLimitHandling.js +0 -144
  20. package/dist-cjs/rateLimits/helpers/headerUtils.js +0 -40
  21. package/dist-cjs/rateLimits/helpers/isStreamChannelLike.js +0 -23
  22. package/dist-cjs/rateLimits/helpers/processStreamBudgetHeaders.js +0 -93
  23. package/dist-cjs/rateLimits/helpers/processStreamRateLimitException.js +0 -63
  24. package/dist-cjs/rateLimits/helpers/processStreamRateLimitHeaders.js +0 -50
  25. package/dist-cjs/rateLimits/helpers/rateLimitSnapshot.js +0 -31
  26. package/dist-cjs/rateLimits/helpers/registerStreamBudgetInterceptors.js +0 -48
  27. package/dist-cjs/rateLimits/helpers/registerStreamInterceptors.js +0 -58
  28. package/dist-cjs/rateLimits/helpers/splitCallArgsAndOptions.js +0 -21
  29. package/dist-cjs/rateLimits/index.js +0 -7
  30. package/dist-cjs/rateLimits/types.js +0 -3
  31. package/dist-cjs/rateLimits/withStreamRateLimitOptions.js +0 -10
  32. package/dist-es/rateLimits/RedisCooldown.js +0 -62
  33. package/dist-es/rateLimits/constants.js +0 -33
  34. package/dist-es/rateLimits/createRateLimitedStreamProxy.js +0 -88
  35. package/dist-es/rateLimits/exception/createRateLimitExceededException.js +0 -17
  36. package/dist-es/rateLimits/exception/processStreamRateLimitException.js +0 -37
  37. package/dist-es/rateLimits/executeStreamRequestWithRateLimitHandling.js +0 -93
  38. package/dist-es/rateLimits/generatedMethodNames.js +0 -317
  39. package/dist-es/rateLimits/helpers/RedisCooldown.js +0 -170
  40. package/dist-es/rateLimits/helpers/createRateLimitExceededException.js +0 -28
  41. package/dist-es/rateLimits/helpers/executeWithRateLimitHandling.js +0 -140
  42. package/dist-es/rateLimits/helpers/headerUtils.js +0 -34
  43. package/dist-es/rateLimits/helpers/isStreamChannelLike.js +0 -19
  44. package/dist-es/rateLimits/helpers/processStreamBudgetHeaders.js +0 -89
  45. package/dist-es/rateLimits/helpers/processStreamRateLimitException.js +0 -59
  46. package/dist-es/rateLimits/helpers/processStreamRateLimitHeaders.js +0 -46
  47. package/dist-es/rateLimits/helpers/rateLimitSnapshot.js +0 -26
  48. package/dist-es/rateLimits/helpers/registerStreamBudgetInterceptors.js +0 -44
  49. package/dist-es/rateLimits/helpers/registerStreamInterceptors.js +0 -54
  50. package/dist-es/rateLimits/helpers/splitCallArgsAndOptions.js +0 -17
  51. package/dist-es/rateLimits/index.js +0 -4
  52. package/dist-es/rateLimits/types.js +0 -1
  53. package/dist-es/rateLimits/withStreamRateLimitOptions.js +0 -6
  54. package/dist-types/rateLimits/RedisCooldown.d.ts +0 -7
  55. package/dist-types/rateLimits/constants.d.ts +0 -29
  56. package/dist-types/rateLimits/createRateLimitedStreamProxy.d.ts +0 -11
  57. package/dist-types/rateLimits/exception/createRateLimitExceededException.d.ts +0 -12
  58. package/dist-types/rateLimits/exception/processStreamRateLimitException.d.ts +0 -8
  59. package/dist-types/rateLimits/executeStreamRequestWithRateLimitHandling.d.ts +0 -8
  60. package/dist-types/rateLimits/generatedMethodNames.d.ts +0 -11
  61. package/dist-types/rateLimits/helpers/RedisCooldown.d.ts +0 -11
  62. package/dist-types/rateLimits/helpers/createRateLimitExceededException.d.ts +0 -10
  63. package/dist-types/rateLimits/helpers/executeWithRateLimitHandling.d.ts +0 -2
  64. package/dist-types/rateLimits/helpers/headerUtils.d.ts +0 -4
  65. package/dist-types/rateLimits/helpers/isStreamChannelLike.d.ts +0 -6
  66. package/dist-types/rateLimits/helpers/processStreamBudgetHeaders.d.ts +0 -9
  67. package/dist-types/rateLimits/helpers/processStreamRateLimitException.d.ts +0 -3
  68. package/dist-types/rateLimits/helpers/processStreamRateLimitHeaders.d.ts +0 -6
  69. package/dist-types/rateLimits/helpers/rateLimitSnapshot.d.ts +0 -3
  70. package/dist-types/rateLimits/helpers/registerStreamBudgetInterceptors.d.ts +0 -2
  71. package/dist-types/rateLimits/helpers/registerStreamInterceptors.d.ts +0 -2
  72. package/dist-types/rateLimits/helpers/splitCallArgsAndOptions.d.ts +0 -6
  73. package/dist-types/rateLimits/index.d.ts +0 -4
  74. package/dist-types/rateLimits/types.d.ts +0 -68
  75. package/dist-types/rateLimits/withStreamRateLimitOptions.d.ts +0 -6
@@ -1,40 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.normalizeResetToEpochMs = exports.parseHeaderNumber = exports.getHeaderCaseInsensitive = void 0;
4
- function getHeaderCaseInsensitive(headers, key) {
5
- if (!headers) {
6
- return;
7
- }
8
- const direct = headers[key];
9
- if (direct !== undefined) {
10
- return direct;
11
- }
12
- const normalizedKey = key.toLowerCase();
13
- for (const [headerKey, headerValue] of Object.entries(headers)) {
14
- if (headerKey.toLowerCase() === normalizedKey) {
15
- return headerValue;
16
- }
17
- }
18
- }
19
- exports.getHeaderCaseInsensitive = getHeaderCaseInsensitive;
20
- function parseHeaderNumber(headers, key) {
21
- const raw = getHeaderCaseInsensitive(headers, key);
22
- if (typeof raw === 'number' && Number.isFinite(raw)) {
23
- return raw;
24
- }
25
- if (typeof raw === 'string') {
26
- const parsed = Number(raw);
27
- if (Number.isFinite(parsed)) {
28
- return parsed;
29
- }
30
- }
31
- }
32
- exports.parseHeaderNumber = parseHeaderNumber;
33
- function normalizeResetToEpochMs(reset) {
34
- if (reset < 10000)
35
- return Date.now() + Math.round(reset * 1000);
36
- if (reset < 10000000000)
37
- return Math.round(reset * 1000);
38
- return Math.round(reset);
39
- }
40
- exports.normalizeResetToEpochMs = normalizeResetToEpochMs;
@@ -1,23 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.isStreamChannelLike = void 0;
4
- const stream_chat_1 = require("stream-chat");
5
- function hasFunctionProperty(value, key) {
6
- return key in value && typeof value[key] === 'function';
7
- }
8
- function hasStringProperty(value, key) {
9
- return key in value && typeof value[key] === 'string';
10
- }
11
- function isStreamChannelLike(value) {
12
- if (typeof value !== 'object' || value === null) {
13
- return false;
14
- }
15
- if (value instanceof stream_chat_1.Channel) {
16
- return true;
17
- }
18
- return (hasStringProperty(value, 'cid') &&
19
- hasFunctionProperty(value, 'sendMessage') &&
20
- hasFunctionProperty(value, 'queryMembers') &&
21
- hasFunctionProperty(value, 'updatePartial'));
22
- }
23
- exports.isStreamChannelLike = isStreamChannelLike;
@@ -1,93 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.processStreamBudgetHeaders = void 0;
4
- const constants_1 = require("../constants");
5
- const headerUtils_1 = require("./headerUtils");
6
- const RedisCooldown_1 = require("./RedisCooldown");
7
- function extractBudgetHeaders(headers) {
8
- return {
9
- limitMs: (0, headerUtils_1.parseHeaderNumber)(headers ?? {}, 'x-budget-limit-ms'),
10
- remainingMs: (0, headerUtils_1.parseHeaderNumber)(headers ?? {}, 'x-budget-remaining-ms'),
11
- usedMs: (0, headerUtils_1.parseHeaderNumber)(headers ?? {}, 'x-budget-used-ms'),
12
- };
13
- }
14
- function isBudgetThresholdReached(limitMs, remainingMs, usedMs) {
15
- if (limitMs <= 0) {
16
- return false;
17
- }
18
- if (typeof usedMs === 'number' && usedMs / limitMs >= constants_1.BUDGET_THRESHOLD) {
19
- return true;
20
- }
21
- if (typeof remainingMs === 'number') {
22
- const remainingThreshold = Math.ceil((1 - constants_1.BUDGET_THRESHOLD) * limitMs);
23
- return remainingMs <= remainingThreshold;
24
- }
25
- return false;
26
- }
27
- function getRandomDelayInRange(minMs, maxMs) {
28
- return Math.round(minMs + Math.random() * (maxMs - minMs));
29
- }
30
- function getBudgetUsage(limitMs, remainingMs, usedMs) {
31
- if (limitMs <= 0) {
32
- return 0;
33
- }
34
- if (typeof usedMs === 'number' && Number.isFinite(usedMs) && usedMs >= 0) {
35
- return usedMs / limitMs;
36
- }
37
- if (typeof remainingMs === 'number' && Number.isFinite(remainingMs)) {
38
- return Math.max(0, 1 - remainingMs / limitMs);
39
- }
40
- return 0;
41
- }
42
- async function processStreamBudgetHeaders(headers, context) {
43
- const { logger } = context;
44
- const { limitMs, remainingMs, usedMs } = extractBudgetHeaders(headers);
45
- if (limitMs === undefined && remainingMs === undefined && usedMs === undefined) {
46
- return;
47
- }
48
- if (typeof limitMs !== 'number' || !Number.isFinite(limitMs)) {
49
- return;
50
- }
51
- const usage = getBudgetUsage(limitMs, remainingMs, usedMs);
52
- const thresholdReached = isBudgetThresholdReached(limitMs, remainingMs, usedMs);
53
- if (thresholdReached) {
54
- const delayMs = getRandomDelayInRange(constants_1.BUDGET_DELAY_MIN_MS, constants_1.BUDGET_DELAY_MAX_MS);
55
- await RedisCooldown_1.RedisCooldown.setRemainingMsExact(constants_1.STREAM_BUDGET_COOLDOWN_KEY, delayMs, context, {
56
- keyPrefix: constants_1.STREAM_BUDGET_COOLDOWN_KEY_PREFIX,
57
- });
58
- logger.warn('Stream budget threshold reached, enabling cooldown', {
59
- budgetLimitMs: limitMs,
60
- budgetRemainingMs: remainingMs,
61
- budgetUsedMs: usedMs,
62
- budgetUsage: usage,
63
- cooldownMs: delayMs,
64
- });
65
- return;
66
- }
67
- if (usage < constants_1.BUDGET_CLEAR_USAGE_THRESHOLD) {
68
- await RedisCooldown_1.RedisCooldown.clear(constants_1.STREAM_BUDGET_COOLDOWN_KEY, context, {
69
- keyPrefix: constants_1.STREAM_BUDGET_COOLDOWN_KEY_PREFIX,
70
- });
71
- logger.info('Stream budget is low, clearing cooldown', {
72
- budgetLimitMs: limitMs,
73
- budgetRemainingMs: remainingMs,
74
- budgetUsedMs: usedMs,
75
- budgetUsage: usage,
76
- });
77
- return;
78
- }
79
- const delayMs = usage >= constants_1.BUDGET_RELAXED_USAGE_THRESHOLD
80
- ? getRandomDelayInRange(constants_1.BUDGET_RELAXED_DELAY_MIN_MS, constants_1.BUDGET_RELAXED_DELAY_MAX_MS)
81
- : getRandomDelayInRange(constants_1.BUDGET_MINIMAL_DELAY_MIN_MS, constants_1.BUDGET_MINIMAL_DELAY_MAX_MS);
82
- await RedisCooldown_1.RedisCooldown.setRemainingMsExact(constants_1.STREAM_BUDGET_COOLDOWN_KEY, delayMs, context, {
83
- keyPrefix: constants_1.STREAM_BUDGET_COOLDOWN_KEY_PREFIX,
84
- });
85
- logger.info('Stream budget is below threshold, reducing cooldown', {
86
- budgetLimitMs: limitMs,
87
- budgetRemainingMs: remainingMs,
88
- budgetUsedMs: usedMs,
89
- budgetUsage: usage,
90
- cooldownMs: delayMs,
91
- });
92
- }
93
- exports.processStreamBudgetHeaders = processStreamBudgetHeaders;
@@ -1,63 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.processStreamRateLimitException = void 0;
4
- const createRateLimitExceededException_1 = require("./createRateLimitExceededException");
5
- const headerUtils_1 = require("./headerUtils");
6
- function parseRetryAfterMs(headers) {
7
- const retryAfter = (0, headerUtils_1.getHeaderCaseInsensitive)(headers, 'retry-after');
8
- if (typeof retryAfter === 'number' && Number.isFinite(retryAfter) && retryAfter >= 0) {
9
- return Math.round(retryAfter * 1000);
10
- }
11
- if (typeof retryAfter !== 'string') {
12
- return;
13
- }
14
- const asSeconds = Number(retryAfter);
15
- if (Number.isFinite(asSeconds) && asSeconds >= 0) {
16
- return Math.round(asSeconds * 1000);
17
- }
18
- const asDateMs = Date.parse(retryAfter);
19
- if (!Number.isNaN(asDateMs)) {
20
- return Math.max(0, asDateMs - Date.now());
21
- }
22
- }
23
- function getRateLimitFromHeaders(headers) {
24
- const limit = (0, headerUtils_1.parseHeaderNumber)(headers, 'x-ratelimit-limit');
25
- const remaining = (0, headerUtils_1.parseHeaderNumber)(headers, 'x-ratelimit-remaining');
26
- const resetRaw = (0, headerUtils_1.parseHeaderNumber)(headers, 'x-ratelimit-reset');
27
- if (limit === undefined || remaining === undefined || resetRaw === undefined) {
28
- return;
29
- }
30
- return {
31
- limit,
32
- remaining,
33
- reset: (0, headerUtils_1.normalizeResetToEpochMs)(resetRaw),
34
- };
35
- }
36
- function isStreamHttpRateLimitError(error) {
37
- return (typeof error === 'object' &&
38
- error !== null &&
39
- (('status' in error && error.status === 429) ||
40
- ('response' in error &&
41
- typeof error.response === 'object' &&
42
- error.response !== null &&
43
- error.response?.status === 429)));
44
- }
45
- function processStreamRateLimitException(error, logger) {
46
- if (isStreamHttpRateLimitError(error)) {
47
- const headers = error.response?.headers ?? {};
48
- const rateLimit = getRateLimitFromHeaders(headers);
49
- const retryAfterMs = parseRetryAfterMs(headers);
50
- const message = error.response?.data?.message || error.message || 'Rate limit exceeded';
51
- logger.warn('Stream rate limit exceeded [HTTP 429]', {
52
- status: error.status ?? error.response?.status,
53
- retryAfterMs,
54
- rateLimit,
55
- });
56
- return (0, createRateLimitExceededException_1.createRateLimitExceededException)({
57
- rateLimit,
58
- retryAfterMs,
59
- message,
60
- });
61
- }
62
- }
63
- exports.processStreamRateLimitException = processStreamRateLimitException;
@@ -1,50 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.processStreamRateLimitHeaders = void 0;
4
- const constants_1 = require("../constants");
5
- const headerUtils_1 = require("./headerUtils");
6
- const RedisCooldown_1 = require("./RedisCooldown");
7
- function computeSoftThrottleDelayMs(limit, remaining) {
8
- if (limit <= 0) {
9
- return 0;
10
- }
11
- const usage = 1 - remaining / limit;
12
- if (usage >= constants_1.SOFT_THROTTLE_USAGE_HIGH) {
13
- return constants_1.SOFT_THROTTLE_DELAY_HIGH_MS;
14
- }
15
- if (usage >= constants_1.SOFT_THROTTLE_USAGE_MEDIUM) {
16
- return constants_1.SOFT_THROTTLE_DELAY_MEDIUM_MS;
17
- }
18
- if (usage >= constants_1.SOFT_THROTTLE_USAGE_LOW) {
19
- return constants_1.SOFT_THROTTLE_DELAY_LOW_MS;
20
- }
21
- return 0;
22
- }
23
- async function processStreamRateLimitHeaders(headers, context) {
24
- const { logger } = context;
25
- const limit = (0, headerUtils_1.parseHeaderNumber)(headers ?? {}, 'x-ratelimit-limit');
26
- const remaining = (0, headerUtils_1.parseHeaderNumber)(headers ?? {}, 'x-ratelimit-remaining');
27
- if (limit === undefined || remaining === undefined) {
28
- return;
29
- }
30
- if (limit <= 0 || remaining < 0) {
31
- return;
32
- }
33
- const delayMs = computeSoftThrottleDelayMs(limit, remaining);
34
- if (delayMs > 0) {
35
- await RedisCooldown_1.RedisCooldown.setRemainingMs(constants_1.STREAM_RATE_LIMIT_SOFT_COOLDOWN_KEY, delayMs, context, {
36
- keyPrefix: constants_1.STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX,
37
- });
38
- logger.debug('Enabling soft stream rate-limit cooldown', {
39
- rateLimit: limit,
40
- rateLimitRemaining: remaining,
41
- cooldownMs: delayMs,
42
- });
43
- return;
44
- }
45
- await RedisCooldown_1.RedisCooldown.clear(constants_1.STREAM_RATE_LIMIT_SOFT_COOLDOWN_KEY, context, {
46
- keyPrefix: constants_1.STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX,
47
- });
48
- logger.debug('Clearing soft stream rate-limit cooldown');
49
- }
50
- exports.processStreamRateLimitHeaders = processStreamRateLimitHeaders;
@@ -1,31 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.parseRateLimitSnapshot = exports.serializeRateLimitSnapshot = void 0;
4
- const RATE_LIMIT_SNAPSHOT_PARTS_COUNT = 3;
5
- function serializeRateLimitSnapshot(snapshot) {
6
- return `${snapshot.limit},${snapshot.remaining},${snapshot.reset}`;
7
- }
8
- exports.serializeRateLimitSnapshot = serializeRateLimitSnapshot;
9
- function parseRateLimitSnapshot(snapshot) {
10
- if (typeof snapshot !== 'string' || snapshot.length === 0) {
11
- return;
12
- }
13
- const parts = snapshot.split(',');
14
- if (parts.length !== RATE_LIMIT_SNAPSHOT_PARTS_COUNT) {
15
- return;
16
- }
17
- const [limitRaw, remainingRaw, resetRaw] = parts;
18
- const limit = Number(limitRaw);
19
- const remaining = Number(remainingRaw);
20
- const reset = Number(resetRaw);
21
- if (!Number.isFinite(limit) ||
22
- !Number.isFinite(remaining) ||
23
- !Number.isFinite(reset) ||
24
- limit < 0 ||
25
- remaining < 0 ||
26
- reset <= 0) {
27
- return;
28
- }
29
- return { limit, remaining, reset };
30
- }
31
- exports.parseRateLimitSnapshot = parseRateLimitSnapshot;
@@ -1,48 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.registerStreamBudgetInterceptors = void 0;
4
- const processStreamBudgetHeaders_1 = require("./processStreamBudgetHeaders");
5
- const processStreamRateLimitHeaders_1 = require("./processStreamRateLimitHeaders");
6
- const attachedInterceptors = new WeakSet();
7
- function hasResponseInterceptors(value) {
8
- return (typeof value === 'object' &&
9
- value !== null &&
10
- typeof value.interceptors?.response?.use === 'function');
11
- }
12
- function resolveAxiosInstance(stream) {
13
- if (hasResponseInterceptors(stream)) {
14
- return stream;
15
- }
16
- if (typeof stream !== 'object' || stream === null) {
17
- return;
18
- }
19
- const maybeStream = stream;
20
- if (hasResponseInterceptors(maybeStream.axiosInstance)) {
21
- return maybeStream.axiosInstance;
22
- }
23
- if (hasResponseInterceptors(maybeStream._client?.axiosInstance)) {
24
- return maybeStream._client?.axiosInstance;
25
- }
26
- }
27
- function registerStreamBudgetInterceptors(stream, context) {
28
- const { logger } = context;
29
- const axiosInstance = resolveAxiosInstance(stream);
30
- if (!axiosInstance || !axiosInstance.interceptors?.response?.use) {
31
- logger.warn('Unable to register stream budget interceptor: axios instance not found');
32
- return;
33
- }
34
- if (attachedInterceptors.has(axiosInstance)) {
35
- return;
36
- }
37
- axiosInstance.interceptors.response.use(async (response) => {
38
- await (0, processStreamBudgetHeaders_1.processStreamBudgetHeaders)(response?.headers, context);
39
- await (0, processStreamRateLimitHeaders_1.processStreamRateLimitHeaders)(response?.headers, context);
40
- return response;
41
- }, async (error) => {
42
- await (0, processStreamBudgetHeaders_1.processStreamBudgetHeaders)(error?.response?.headers, context);
43
- await (0, processStreamRateLimitHeaders_1.processStreamRateLimitHeaders)(error?.response?.headers, context);
44
- return Promise.reject(error);
45
- });
46
- attachedInterceptors.add(axiosInstance);
47
- }
48
- exports.registerStreamBudgetInterceptors = registerStreamBudgetInterceptors;
@@ -1,58 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.registerStreamInterceptors = void 0;
4
- const processStreamBudgetHeaders_1 = require("./processStreamBudgetHeaders");
5
- const processStreamRateLimitHeaders_1 = require("./processStreamRateLimitHeaders");
6
- const attachedInterceptors = new WeakSet();
7
- function hasResponseInterceptors(value) {
8
- return (typeof value === 'object' &&
9
- value !== null &&
10
- typeof value.interceptors?.response?.use === 'function');
11
- }
12
- function resolveAxiosInstance(stream) {
13
- if (hasResponseInterceptors(stream)) {
14
- return stream;
15
- }
16
- if (typeof stream !== 'object' || stream === null) {
17
- return;
18
- }
19
- const maybeStream = stream;
20
- if (hasResponseInterceptors(maybeStream.axiosInstance)) {
21
- return maybeStream.axiosInstance;
22
- }
23
- if (hasResponseInterceptors(maybeStream._client?.axiosInstance)) {
24
- return maybeStream._client?.axiosInstance;
25
- }
26
- }
27
- function registerStreamInterceptors(stream, context) {
28
- const { logger } = context;
29
- const axiosInstance = resolveAxiosInstance(stream);
30
- if (!axiosInstance || !axiosInstance.interceptors?.response?.use) {
31
- logger.warn('Unable to register stream budget & rate limit interceptors: axios instance not found');
32
- return;
33
- }
34
- if (attachedInterceptors.has(axiosInstance)) {
35
- return;
36
- }
37
- attachedInterceptors.add(axiosInstance);
38
- axiosInstance.interceptors.response.use(async (response) => {
39
- try {
40
- await (0, processStreamBudgetHeaders_1.processStreamBudgetHeaders)(response?.headers, context);
41
- await (0, processStreamRateLimitHeaders_1.processStreamRateLimitHeaders)(response?.headers, context);
42
- }
43
- catch (error) {
44
- logger.warn('Failed to process Stream response headers in interceptor', { error });
45
- }
46
- return response;
47
- }, async (error) => {
48
- try {
49
- await (0, processStreamBudgetHeaders_1.processStreamBudgetHeaders)(error?.response?.headers, context);
50
- await (0, processStreamRateLimitHeaders_1.processStreamRateLimitHeaders)(error?.response?.headers, context);
51
- }
52
- catch (interceptorError) {
53
- logger.warn('Failed to process Stream error headers in interceptor', { error: interceptorError });
54
- }
55
- return Promise.reject(error);
56
- });
57
- }
58
- exports.registerStreamInterceptors = registerStreamInterceptors;
@@ -1,21 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.splitCallArgsAndOptions = void 0;
4
- const constants_1 = require("../constants");
5
- function isStreamRateLimitOptionsToken(value) {
6
- return typeof value === 'object' && value !== null && constants_1.STREAM_RATE_LIMIT_OPTIONS_MARKER in value;
7
- }
8
- function splitCallArgsAndOptions(args) {
9
- if (args.length === 0) {
10
- return { callArgs: args, callOptions: {} };
11
- }
12
- const lastArg = args[args.length - 1];
13
- if (isStreamRateLimitOptionsToken(lastArg)) {
14
- return {
15
- callArgs: args.slice(0, -1),
16
- callOptions: lastArg[constants_1.STREAM_RATE_LIMIT_OPTIONS_MARKER],
17
- };
18
- }
19
- return { callArgs: args, callOptions: {} };
20
- }
21
- exports.splitCallArgsAndOptions = splitCallArgsAndOptions;
@@ -1,7 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- const tslib_1 = require("tslib");
4
- tslib_1.__exportStar(require("./createRateLimitedStreamProxy"), exports);
5
- tslib_1.__exportStar(require("./helpers/createRateLimitExceededException"), exports);
6
- tslib_1.__exportStar(require("./helpers/processStreamRateLimitException"), exports);
7
- tslib_1.__exportStar(require("./withStreamRateLimitOptions"), exports);
@@ -1,3 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- const constants_1 = require("./constants");
@@ -1,10 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.withStreamRateLimitOptions = void 0;
4
- const constants_1 = require("./constants");
5
- function withStreamRateLimitOptions(options) {
6
- return {
7
- [constants_1.STREAM_RATE_LIMIT_OPTIONS_MARKER]: options,
8
- };
9
- }
10
- exports.withStreamRateLimitOptions = withStreamRateLimitOptions;
@@ -1,62 +0,0 @@
1
- import {STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX} from './constants';
2
-
3
- export class RedisCooldown {
4
- static async getRemainingMs(cooldownKey, context) {
5
- const {redis} = context;
6
-
7
- try {
8
- const ttlMs = await redis.pttl(`${STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX}${cooldownKey}`);
9
-
10
- if (ttlMs > 0) {
11
- return ttlMs;
12
- }
13
- } catch (error) {
14
- console.warn('Failed to read stream rate-limit cooldown from Redis', {
15
- cooldownKey,
16
- error,
17
- });
18
- }
19
-
20
- return 0;
21
- }
22
-
23
- static async setRemainingMs(cooldownKey, cooldownMs, context) {
24
- if (cooldownMs <= 0) {
25
- return;
26
- }
27
-
28
- const {redis} = context;
29
- const redisKey = `${STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX}${cooldownKey}`;
30
-
31
- try {
32
- await redis.eval(
33
- `
34
- local key = KEYS[1]
35
- local new_ttl = tonumber(ARGV[1])
36
- local current_ttl = redis.call('PTTL', key)
37
-
38
- if current_ttl == -2 then
39
- redis.call('SET', key, '1', 'PX', new_ttl)
40
- return 1
41
- end
42
-
43
- if current_ttl == -1 or current_ttl < new_ttl then
44
- redis.call('PEXPIRE', key, new_ttl)
45
- return 2
46
- end
47
-
48
- return 0
49
- `,
50
- 1,
51
- redisKey,
52
- cooldownMs.toString(),
53
- );
54
- } catch (error) {
55
- console.warn('Failed to write stream rate-limit cooldown to Redis', {
56
- cooldownKey,
57
- cooldownMs,
58
- error,
59
- });
60
- }
61
- }
62
- }
@@ -1,33 +0,0 @@
1
- export { CHANNEL_METHOD_NAMES, CHANNEL_SYNC_METHOD_NAMES, CHANNEL_SYNC_PROXY_RESULT_METHOD_NAMES, STREAM_CHANNEL_RESULT_METHOD_NAMES, STREAM_METHOD_NAMES, STREAM_SYNC_CHANNEL_RESULT_METHOD_NAMES, STREAM_SYNC_METHOD_NAMES, } from './generatedMethodNames';
2
- export const DEFAULT_MAX_DELAY_MS = 5000;
3
- export const DEFAULT_RETRY_AFTER_MS = 10000;
4
- export const DEFAULT_JITTER_MS = 150;
5
- export const RETRY_AFTER_BUFFER_MS = 1000;
6
- export const STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX = 'stream-rate-limit-cooldown:';
7
- export const STREAM_BUDGET_COOLDOWN_KEY_PREFIX = 'stream-budget-cooldown:';
8
- export const STREAM_RATE_LIMIT_OPTIONS_MARKER = '__xbsRateLimitOptions';
9
- export const STREAM_RATE_LIMIT_SOFT_COOLDOWN_KEY = 'stream-rate-limit-soft-global';
10
- export const STREAM_RATE_LIMIT_METADATA_KEY_SUFFIX = ':meta';
11
- export const STREAM_BUDGET_COOLDOWN_KEY = 'stream-budget-global';
12
- export const BUDGET_THRESHOLD = 0.8;
13
- export const BUDGET_DELAY_MIN_MS = 30000;
14
- export const BUDGET_DELAY_MAX_MS = 60000;
15
- export const BUDGET_CLEAR_USAGE_THRESHOLD = 0.6;
16
- export const BUDGET_RELAXED_USAGE_THRESHOLD = 0.7;
17
- export const BUDGET_RELAXED_DELAY_MIN_MS = 5000;
18
- export const BUDGET_RELAXED_DELAY_MAX_MS = 10000;
19
- export const BUDGET_MINIMAL_DELAY_MIN_MS = 1000;
20
- export const BUDGET_MINIMAL_DELAY_MAX_MS = 2000;
21
- export const BUDGET_COOLDOWN_SLEEP_CHUNK_MS = 3000;
22
- export const SOFT_THROTTLE_USAGE_LOW = 0.7;
23
- export const SOFT_THROTTLE_USAGE_MEDIUM = 0.85;
24
- export const SOFT_THROTTLE_USAGE_HIGH = 0.95;
25
- export const SOFT_THROTTLE_DELAY_LOW_MS = 500;
26
- export const SOFT_THROTTLE_DELAY_MEDIUM_MS = 1500;
27
- export const SOFT_THROTTLE_DELAY_HIGH_MS = 3000;
28
- export const DEFAULT_OPTIONS = {
29
- maxAttempts: 3,
30
- enableCooldown: true,
31
- maxDelayMs: 5000,
32
- maxRetryableDelayMs: 10000,
33
- };
@@ -1,88 +0,0 @@
1
- import { CHANNEL_SYNC_METHOD_NAMES, CHANNEL_SYNC_PROXY_RESULT_METHOD_NAMES, DEFAULT_OPTIONS, STREAM_CHANNEL_RESULT_METHOD_NAMES, STREAM_SYNC_CHANNEL_RESULT_METHOD_NAMES, STREAM_SYNC_METHOD_NAMES, } from './constants';
2
- import { executeWithRateLimitHandling } from './helpers/executeWithRateLimitHandling';
3
- import { isStreamChannelLike } from './helpers/isStreamChannelLike';
4
- import { registerStreamInterceptors } from './helpers/registerStreamInterceptors';
5
- import { splitCallArgsAndOptions } from './helpers/splitCallArgsAndOptions';
6
- const STREAM_SYNC_METHODS = new Set(STREAM_SYNC_METHOD_NAMES);
7
- const STREAM_CHANNEL_RESULT_METHODS = new Set(STREAM_CHANNEL_RESULT_METHOD_NAMES);
8
- const STREAM_SYNC_CHANNEL_RESULT_METHODS = new Set(STREAM_SYNC_CHANNEL_RESULT_METHOD_NAMES);
9
- const CHANNEL_SYNC_METHODS = new Set(CHANNEL_SYNC_METHOD_NAMES);
10
- const CHANNEL_SYNC_PROXY_RESULT_METHODS = new Set(CHANNEL_SYNC_PROXY_RESULT_METHOD_NAMES);
11
- function wrapKnownChannelResult(result, dependencies) {
12
- if (Array.isArray(result)) {
13
- return result.map((item) => wrapKnownChannelResult(item, dependencies));
14
- }
15
- if (typeof result !== 'object' || result === null) {
16
- return result;
17
- }
18
- return wrapTarget(result, dependencies);
19
- }
20
- function normalizeStreamCallResult(result, dependencies, wrapAsChannelResult = false) {
21
- if (wrapAsChannelResult) {
22
- return wrapKnownChannelResult(result, dependencies);
23
- }
24
- if (Array.isArray(result)) {
25
- return result.map((item) => normalizeStreamCallResult(item, dependencies));
26
- }
27
- if (!isStreamChannelLike(result)) {
28
- return result;
29
- }
30
- return wrapTarget(result, dependencies);
31
- }
32
- function createRateLimitedMethodWrapper(target, value, operation, dependencies, wrapAsChannelResult) {
33
- return async (...args) => {
34
- const { callArgs, callOptions } = splitCallArgsAndOptions(args);
35
- const result = await executeWithRateLimitHandling(() => Reflect.apply(value, target, callArgs), operation, dependencies.context, {
36
- ...DEFAULT_OPTIONS,
37
- ...callOptions,
38
- });
39
- return normalizeStreamCallResult(result, dependencies, wrapAsChannelResult);
40
- };
41
- }
42
- function wrapTarget(target, dependencies) {
43
- const cachedObject = dependencies.cache.get(target);
44
- if (cachedObject) {
45
- return cachedObject;
46
- }
47
- const proxy = new Proxy(target, {
48
- get: (originalTarget, property, receiver) => {
49
- const value = Reflect.get(originalTarget, property, receiver);
50
- if (typeof property !== 'string' || typeof value !== 'function') {
51
- return value;
52
- }
53
- if (STREAM_SYNC_CHANNEL_RESULT_METHODS.has(property)) {
54
- return (...args) => {
55
- const { callArgs } = splitCallArgsAndOptions(args);
56
- const result = Reflect.apply(value, originalTarget, callArgs);
57
- return normalizeStreamCallResult(result, dependencies, true);
58
- };
59
- }
60
- if (CHANNEL_SYNC_PROXY_RESULT_METHODS.has(property)) {
61
- return (...args) => {
62
- const { callArgs } = splitCallArgsAndOptions(args);
63
- const result = Reflect.apply(value, originalTarget, callArgs);
64
- if (typeof result === 'object' && result !== null) {
65
- return wrapTarget(result, dependencies);
66
- }
67
- return result;
68
- };
69
- }
70
- if (STREAM_SYNC_METHODS.has(property) || CHANNEL_SYNC_METHODS.has(property)) {
71
- return (...args) => {
72
- const { callArgs } = splitCallArgsAndOptions(args);
73
- return Reflect.apply(value, originalTarget, callArgs);
74
- };
75
- }
76
- return createRateLimitedMethodWrapper(originalTarget, value, property, dependencies, STREAM_CHANNEL_RESULT_METHODS.has(property));
77
- },
78
- });
79
- dependencies.cache.set(target, proxy);
80
- return proxy;
81
- }
82
- export function createRateLimitedStreamProxy(stream, context) {
83
- registerStreamInterceptors(stream, context);
84
- return wrapTarget(stream, {
85
- context,
86
- cache: new WeakMap(),
87
- });
88
- }
@@ -1,17 +0,0 @@
1
- import {RateLimitExceededException} from '@wildix/xbees-conversations-client';
2
-
3
- import {DEFAULT_RETRY_AFTER_MS} from '../constants';
4
-
5
- export function createRateLimitExceededException(options = {}) {
6
- const {rateLimit, message, operation, retryAfterMs = DEFAULT_RETRY_AFTER_MS} = options;
7
- const defaultMessage = operation ? `Rate limit exceeded for ${operation}` : 'Rate limit exceeded';
8
-
9
- return new RateLimitExceededException({
10
- $metadata: {},
11
- message: message || defaultMessage,
12
- rateLimitRemaining: rateLimit?.remaining || 0,
13
- retryAfter: rateLimit?.reset ? rateLimit.reset - Date.now() : retryAfterMs,
14
- rateLimit: rateLimit?.limit?.toString() || 'unknown',
15
- rateLimitReset: rateLimit?.reset || Date.now() + retryAfterMs,
16
- });
17
- }