@newyorkcompute/kalshi-core 0.1.1 → 0.3.0

package/dist/rate-limiter.js

@@ -0,0 +1,180 @@
+/**
+ * Rate Limiter
+ *
+ * Implements exponential backoff and circuit breaker patterns
+ * for handling API rate limits gracefully.
+ */
+const DEFAULT_CONFIG = {
+    minInterval: 1000,
+    maxInterval: 60000,
+    backoffMultiplier: 2,
+    circuitBreakerThreshold: 5,
+    circuitResetTimeout: 30000,
+};
+/**
+ * Create a rate limiter instance
+ *
+ * @example
+ * ```ts
+ * const limiter = createRateLimiter();
+ *
+ * async function fetchData() {
+ *   if (limiter.shouldBlock()) {
+ *     return; // Skip request
+ *   }
+ *
+ *   try {
+ *     const data = await api.getData();
+ *     limiter.recordSuccess();
+ *     return data;
+ *   } catch (err) {
+ *     if (isRateLimitError(err)) {
+ *       limiter.recordRateLimitError();
+ *     } else {
+ *       limiter.recordFailure();
+ *     }
+ *     throw err;
+ *   }
+ * }
+ * ```
+ */
+export function createRateLimiter(config = {}) {
+    const cfg = { ...DEFAULT_CONFIG, ...config };
+    let state = {
+        currentInterval: cfg.minInterval,
+        consecutiveFailures: 0,
+        isCircuitOpen: false,
+        isRateLimited: false,
+        lastSuccessTime: 0,
+        lastFailureTime: 0,
+    };
+    let circuitResetTimer = null;
+    /**
+     * Check if requests should be blocked
+     */
+    function shouldBlock() {
+        return state.isCircuitOpen;
+    }
+    /**
+     * Get the current recommended interval between requests
+     */
+    function getCurrentInterval() {
+        return state.currentInterval;
+    }
+    /**
+     * Get the current state (for debugging/display)
+     */
+    function getState() {
+        return { ...state };
+    }
+    /**
+     * Record a successful request - resets backoff
+     */
+    function recordSuccess() {
+        state.consecutiveFailures = 0;
+        state.currentInterval = cfg.minInterval;
+        state.isRateLimited = false;
+        state.lastSuccessTime = Date.now();
+        // Close circuit if it was open
+        if (state.isCircuitOpen) {
+            state.isCircuitOpen = false;
+            if (circuitResetTimer) {
+                clearTimeout(circuitResetTimer);
+                circuitResetTimer = null;
+            }
+        }
+    }
+    /**
+     * Record a rate limit error (429) - applies exponential backoff
+     */
+    function recordRateLimitError() {
+        state.consecutiveFailures++;
+        state.isRateLimited = true;
+        state.lastFailureTime = Date.now();
+        // Apply exponential backoff
+        state.currentInterval = Math.min(state.currentInterval * cfg.backoffMultiplier, cfg.maxInterval);
+        // Open circuit if threshold reached
+        if (state.consecutiveFailures >= cfg.circuitBreakerThreshold) {
+            openCircuit();
+        }
+    }
+    /**
+     * Record a general failure (not rate limit)
+     */
+    function recordFailure() {
+        state.consecutiveFailures++;
+        state.lastFailureTime = Date.now();
+        // Open circuit if threshold reached
+        if (state.consecutiveFailures >= cfg.circuitBreakerThreshold) {
+            openCircuit();
+        }
+    }
+    /**
+     * Open the circuit breaker (block all requests temporarily)
+     */
+    function openCircuit() {
+        if (state.isCircuitOpen)
+            return;
+        state.isCircuitOpen = true;
+        // Schedule circuit reset
+        circuitResetTimer = setTimeout(() => {
+            state.isCircuitOpen = false;
+            state.consecutiveFailures = 0;
+            state.currentInterval = cfg.minInterval;
+            circuitResetTimer = null;
+        }, cfg.circuitResetTimeout);
+    }
+    /**
+     * Reset the rate limiter to initial state
+     */
+    function reset() {
+        if (circuitResetTimer) {
+            clearTimeout(circuitResetTimer);
+            circuitResetTimer = null;
+        }
+        state = {
+            currentInterval: cfg.minInterval,
+            consecutiveFailures: 0,
+            isCircuitOpen: false,
+            isRateLimited: false,
+            lastSuccessTime: 0,
+            lastFailureTime: 0,
+        };
+    }
+    return {
+        shouldBlock,
+        getCurrentInterval,
+        getState,
+        recordSuccess,
+        recordRateLimitError,
+        recordFailure,
+        reset,
+    };
+}
+/**
+ * Check if an error is a rate limit error (HTTP 429)
+ *
+ * @param error - Error to check
+ * @returns true if the error is a rate limit error
+ */
+export function isRateLimitError(error) {
+    if (!error || typeof error !== 'object')
+        return false;
+    // Check for status code 429
+    if ('status' in error && error.status === 429)
+        return true;
+    if ('response' in error) {
+        const response = error.response;
+        if (response?.status === 429)
+            return true;
+    }
+    // Check error message
+    if ('message' in error && typeof error.message === 'string') {
+        const msg = error.message.toLowerCase();
+        if (msg.includes('429') || msg.includes('rate limit') || msg.includes('too many requests')) {
+            return true;
+        }
+    }
+    return false;
+}
+//# sourceMappingURL=rate-limiter.js.map

package/dist/rate-limiter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limiter.js","sourceRoot":"","sources":["../src/rate-limiter.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AA8BH,MAAM,cAAc,GAAsB;IACxC,WAAW,EAAE,IAAI;IACjB,WAAW,EAAE,KAAK;IAClB,iBAAiB,EAAE,CAAC;IACpB,uBAAuB,EAAE,CAAC;IAC1B,mBAAmB,EAAE,KAAK;CAC3B,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,MAAM,UAAU,iBAAiB,CAAC,SAAqC,EAAE;IACvE,MAAM,GAAG,GAAsB,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;IAEhE,IAAI,KAAK,GAAqB;QAC5B,eAAe,EAAE,GAAG,CAAC,WAAW;QAChC,mBAAmB,EAAE,CAAC;QACtB,aAAa,EAAE,KAAK;QACpB,aAAa,EAAE,KAAK;QACpB,eAAe,EAAE,CAAC;QAClB,eAAe,EAAE,CAAC;KACnB,CAAC;IAEF,IAAI,iBAAiB,GAAyC,IAAI,CAAC;IAEnE;;OAEG;IACH,SAAS,WAAW;QAClB,OAAO,KAAK,CAAC,aAAa,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,SAAS,kBAAkB;QACzB,OAAO,KAAK,CAAC,eAAe,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,SAAS,QAAQ;QACf,OAAO,EAAE,GAAG,KAAK,EAAE,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,SAAS,aAAa;QACpB,KAAK,CAAC,mBAAmB,GAAG,CAAC,CAAC;QAC9B,KAAK,CAAC,eAAe,GAAG,GAAG,CAAC,WAAW,CAAC;QACxC,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;QAC5B,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEnC,+BAA+B;QAC/B,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;YACxB,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;YAC5B,IAAI,iBAAiB,EAAE,CAAC;gBACtB,YAAY,CAAC,iBAAiB,CAAC,CAAC;gBAChC,iBAAiB,GAAG,IAAI,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,SAAS,oBAAoB;QAC3B,KAAK,CAAC,mBAAmB,EAAE,CAAC;QAC5B,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;QAC3B,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEnC,4BAA4B;QAC5B,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,CAC9B,KAAK,CAAC,eAAe,GAAG,GAAG,CAAC,iBAAiB,EAC7C,GAAG,CAAC,WAAW,CAChB,CAAC;QAEF,oCAAoC;QACpC,IAAI,KAAK,CAAC,mBAAmB,IAAI,GAAG,CAAC,uBAAuB,EAAE,CAAC;YAC7D,WAAW,EAAE,CAAC;QAChB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,SAAS,aAAa;QACpB,KAAK,CAAC,mBAAmB,EAAE,CAAC;QAC5B,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEnC,oCAAoC;QACpC,IAAI,KAAK,CAAC,mBAAmB,IAAI,GAAG,CAAC,uBAAuB,EAAE,CAAC;YAC7D,WAAW,EAAE,CAAC;QAChB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,SAAS,WAAW;QAClB,IAAI,KAAK,CAAC,aAAa;YAAE,OAAO;QAEhC,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;QAE3B,yBAAyB;QACzB,iBAAiB,GAAG,UAAU,CAAC,GAAG,EAAE;YAClC,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;YAC5B,KAAK,CAAC,mBAAmB,GAAG,CAAC,CAAC;YAC9B,KAAK,CAAC,eAAe,GAAG,GAAG,CAAC,WAAW,CAAC;YACxC,iBAAiB,GAAG,IAAI,CAAC;QAC3B,CAAC,EAAE,GAAG,CAAC,mBAAmB,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,SAAS,KAAK;QACZ,IAAI,iBAAiB,EAAE,CAAC;YACtB,YAAY,CAAC,iBAAiB,CAAC,CAAC;YAChC,iBAAiB,GAAG,IAAI,CAAC;QAC3B,CAAC;QAED,KAAK,GAAG;YACN,eAAe,EAAE,GAAG,CAAC,WAAW;YAChC,mBAAmB,EAAE,CAAC;YACtB,aAAa,EAAE,KAAK;YACpB,aAAa,EAAE,KAAK;YACpB,eAAe,EAAE,CAAC;YAClB,eAAe,EAAE,CAAC;SACnB,CAAC;IACJ,CAAC;IAED,OAAO;QACL,WAAW;QACX,kBAAkB;QAClB,QAAQ;QACR,aAAa;QACb,oBAAoB;QACpB,aAAa;QACb,KAAK;KACN,CAAC;AACJ,CAAC;AAOD;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC7C,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAEtD,4BAA4B;IAC5B,IAAI,QAAQ,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG;QAAE,OAAO,IAAI,CAAC;IAC3D,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;QACxB,MAAM,QAAQ,GAAG,KAAK,CAAC,QAA2C,CAAC;QACnE,IAAI,QAAQ,EAAE,MAAM,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC;IAC5C,CAAC;IAED,sBAAsB;IACtB,IAAI,SAAS,IAAI,KAAK,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QAC5D,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QACxC,IAAI,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAC3F,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/rate-limiter.test.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Rate Limiter Tests
+ */
+export {};
+//# sourceMappingURL=rate-limiter.test.d.ts.map

package/dist/rate-limiter.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limiter.test.d.ts","sourceRoot":"","sources":["../src/rate-limiter.test.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/dist/rate-limiter.test.js

@@ -0,0 +1,175 @@
+/**
+ * Rate Limiter Tests
+ */
+import { describe, it, expect, beforeEach, vi, afterEach } from 'vitest';
+import { createRateLimiter, isRateLimitError, } from './rate-limiter.js';
+describe('createRateLimiter', () => {
+    beforeEach(() => {
+        vi.useFakeTimers();
+    });
+    afterEach(() => {
+        vi.useRealTimers();
+    });
+    describe('initial state', () => {
+        it('should not block requests initially', () => {
+            const limiter = createRateLimiter();
+            expect(limiter.shouldBlock()).toBe(false);
+        });
+        it('should have minimum interval initially', () => {
+            const limiter = createRateLimiter({ minInterval: 1000 });
+            expect(limiter.getCurrentInterval()).toBe(1000);
+        });
+        it('should have correct initial state', () => {
+            const limiter = createRateLimiter();
+            const state = limiter.getState();
+            expect(state.consecutiveFailures).toBe(0);
+            expect(state.isCircuitOpen).toBe(false);
+            expect(state.isRateLimited).toBe(false);
+        });
+    });
+    describe('recordSuccess', () => {
+        it('should reset consecutive failures', () => {
+            const limiter = createRateLimiter();
+            limiter.recordFailure();
+            limiter.recordFailure();
+            expect(limiter.getState().consecutiveFailures).toBe(2);
+            limiter.recordSuccess();
+            expect(limiter.getState().consecutiveFailures).toBe(0);
+        });
+        it('should reset interval to minimum', () => {
+            const limiter = createRateLimiter({ minInterval: 1000 });
+            limiter.recordRateLimitError();
+            expect(limiter.getCurrentInterval()).toBeGreaterThan(1000);
+            limiter.recordSuccess();
+            expect(limiter.getCurrentInterval()).toBe(1000);
+        });
+        it('should clear rate limited flag', () => {
+            const limiter = createRateLimiter();
+            limiter.recordRateLimitError();
+            expect(limiter.getState().isRateLimited).toBe(true);
+            limiter.recordSuccess();
+            expect(limiter.getState().isRateLimited).toBe(false);
+        });
+    });
+    describe('recordRateLimitError', () => {
+        it('should increase interval with exponential backoff', () => {
+            const limiter = createRateLimiter({
+                minInterval: 1000,
+                backoffMultiplier: 2,
+            });
+            limiter.recordRateLimitError();
+            expect(limiter.getCurrentInterval()).toBe(2000);
+            limiter.recordRateLimitError();
+            expect(limiter.getCurrentInterval()).toBe(4000);
+            limiter.recordRateLimitError();
+            expect(limiter.getCurrentInterval()).toBe(8000);
+        });
+        it('should not exceed max interval', () => {
+            const limiter = createRateLimiter({
+                minInterval: 1000,
+                maxInterval: 5000,
+                backoffMultiplier: 2,
+            });
+            // 1000 -> 2000 -> 4000 -> 5000 (capped)
+            limiter.recordRateLimitError();
+            limiter.recordRateLimitError();
+            limiter.recordRateLimitError();
+            limiter.recordRateLimitError();
+            expect(limiter.getCurrentInterval()).toBe(5000);
+        });
+        it('should set rate limited flag', () => {
+            const limiter = createRateLimiter();
+            limiter.recordRateLimitError();
+            expect(limiter.getState().isRateLimited).toBe(true);
+        });
+        it('should increment consecutive failures', () => {
+            const limiter = createRateLimiter();
+            limiter.recordRateLimitError();
+            expect(limiter.getState().consecutiveFailures).toBe(1);
+            limiter.recordRateLimitError();
+            expect(limiter.getState().consecutiveFailures).toBe(2);
+        });
+    });
+    describe('circuit breaker', () => {
+        it('should open circuit after threshold failures', () => {
+            const limiter = createRateLimiter({
+                circuitBreakerThreshold: 3,
+            });
+            limiter.recordFailure();
+            limiter.recordFailure();
+            expect(limiter.shouldBlock()).toBe(false);
+            limiter.recordFailure();
+            expect(limiter.shouldBlock()).toBe(true);
+        });
+        it('should reset circuit after timeout', () => {
+            const limiter = createRateLimiter({
+                circuitBreakerThreshold: 2,
+                circuitResetTimeout: 5000,
+            });
+            limiter.recordFailure();
+            limiter.recordFailure();
+            expect(limiter.shouldBlock()).toBe(true);
+            // Advance time past reset timeout
+            vi.advanceTimersByTime(5001);
+            expect(limiter.shouldBlock()).toBe(false);
+            expect(limiter.getState().consecutiveFailures).toBe(0);
+        });
+        it('should close circuit on success', () => {
+            const limiter = createRateLimiter({
+                circuitBreakerThreshold: 2,
+            });
+            limiter.recordFailure();
+            limiter.recordFailure();
+            expect(limiter.shouldBlock()).toBe(true);
+            limiter.recordSuccess();
+            expect(limiter.shouldBlock()).toBe(false);
+        });
+    });
+    describe('reset', () => {
+        it('should reset all state', () => {
+            const limiter = createRateLimiter({
+                minInterval: 1000,
+                circuitBreakerThreshold: 2,
+            });
+            limiter.recordRateLimitError();
+            limiter.recordRateLimitError();
+            expect(limiter.shouldBlock()).toBe(true);
+            expect(limiter.getCurrentInterval()).toBeGreaterThan(1000);
+            limiter.reset();
+            expect(limiter.shouldBlock()).toBe(false);
+            expect(limiter.getCurrentInterval()).toBe(1000);
+            expect(limiter.getState().consecutiveFailures).toBe(0);
+            expect(limiter.getState().isRateLimited).toBe(false);
+        });
+    });
+});
+describe('isRateLimitError', () => {
+    it('should return true for error with status 429', () => {
+        expect(isRateLimitError({ status: 429 })).toBe(true);
+    });
+    it('should return true for error with response.status 429', () => {
+        expect(isRateLimitError({ response: { status: 429 } })).toBe(true);
+    });
+    it('should return true for error message containing 429', () => {
+        expect(isRateLimitError({ message: 'Error 429: Too Many Requests' })).toBe(true);
+    });
+    it('should return true for error message containing rate limit', () => {
+        expect(isRateLimitError({ message: 'Rate limit exceeded' })).toBe(true);
+    });
+    it('should return true for error message containing too many requests', () => {
+        expect(isRateLimitError({ message: 'too many requests' })).toBe(true);
+    });
+    it('should return false for other errors', () => {
+        expect(isRateLimitError({ status: 500 })).toBe(false);
+        expect(isRateLimitError({ message: 'Internal server error' })).toBe(false);
+    });
+    it('should return false for null/undefined', () => {
+        expect(isRateLimitError(null)).toBe(false);
+        expect(isRateLimitError(undefined)).toBe(false);
+    });
+    it('should return false for non-objects', () => {
+        expect(isRateLimitError('error')).toBe(false);
+        expect(isRateLimitError(123)).toBe(false);
+    });
+});
+//# sourceMappingURL=rate-limiter.test.js.map

package/dist/rate-limiter.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limiter.test.js","sourceRoot":"","sources":["../src/rate-limiter.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EACL,iBAAiB,EACjB,gBAAgB,GACjB,MAAM,mBAAmB,CAAC;AAE3B,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;YACpC,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,OAAO,GAAG,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;YACzD,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YAEjC,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;YAEpC,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAEvD,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,OAAO,GAAG,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;YAEzD,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAE3D,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;YAEpC,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEpD,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,OAAO,GAAG,iBAAiB,CAAC;gBAChC,WAAW,EAAE,IAAI;gBACjB,iBAAiB,EAAE,CAAC;aACrB,CAAC,CAAC;YAEH,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEhD,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEhD,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,OAAO,GAAG,iBAAiB,CAAC;gBAChC,WAAW,EAAE,IAAI;gBACjB,WAAW,EAAE,IAAI;gBACjB,iBAAiB,EAAE,CAAC;aACrB,CAAC,CAAC;YAEH,wCAAwC;YACxC,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAE/B,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;YAEpC,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;YAEpC,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAEvD,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,OAAO,GAAG,iBAAiB,CAAC;gBAChC,uBAAuB,EAAE,CAAC;aAC3B,CAAC,CAAC;YAEH,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAE1C,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,OAAO,GAAG,iBAAiB,CAAC;gBAChC,uBAAuB,EAAE,CAAC;gBAC1B,mBAAmB,EAAE,IAAI;aAC1B,CAAC,CAAC;YAEH,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEzC,kCAAkC;YAClC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAE7B,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,OAAO,GAAG,iBAAiB,CAAC;gBAChC,uBAAuB,EAAE,CAAC;aAC3B,CAAC,CAAC;YAEH,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEzC,OAAO,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAChC,MAAM,OAAO,GAAG,iBAAiB,CAAC;gBAChC,WAAW,EAAE,IAAI;gBACjB,uBAAuB,EAAE,CAAC;aAC3B,CAAC,CAAC;YAEH,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAE3D,OAAO,CAAC,KAAK,EAAE,CAAC;YAEhB,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChD,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACvD,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,CAAC,gBAAgB,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,8BAA8B,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC3E,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,uBAAuB,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9C,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/websocket/auth.d.ts

@@ -0,0 +1,28 @@
+/**
+ * Kalshi WebSocket Authentication
+ *
+ * Handles signature generation for authenticated WebSocket connections.
+ * Uses the same RSA-PSS signing as the REST API.
+ */
+/**
+ * Generate authentication headers for WebSocket connection
+ *
+ * @param apiKeyId - Your Kalshi API key ID
+ * @param privateKey - Your private key in PEM format
+ * @param timestamp - Unix timestamp in milliseconds
+ * @returns Headers object for WebSocket connection
+ */
+export declare function generateWsAuthHeaders(apiKeyId: string, privateKey: string, timestamp?: number): Record<string, string>;
+/**
+ * Generate a signed WebSocket URL with auth params
+ *
+ * Some WebSocket implementations don't support custom headers,
+ * so we can pass auth as query parameters instead.
+ *
+ * @param baseUrl - WebSocket endpoint URL
+ * @param apiKeyId - Your Kalshi API key ID
+ * @param privateKey - Your private key in PEM format
+ * @returns Full WebSocket URL with auth parameters
+ */
+export declare function generateSignedWsUrl(baseUrl: string, apiKeyId: string, privateKey: string): string;
+//# sourceMappingURL=auth.d.ts.map

package/dist/websocket/auth.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.d.ts","sourceRoot":"","sources":["../../src/websocket/auth.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH;;;;;;;GAOG;AACH,wBAAgB,qBAAqB,CACnC,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,SAAS,GAAE,MAAmB,GAC7B,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAqBxB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,GACjB,MAAM,CAUR"}

package/dist/websocket/auth.js

@@ -0,0 +1,55 @@
+/**
+ * Kalshi WebSocket Authentication
+ *
+ * Handles signature generation for authenticated WebSocket connections.
+ * Uses the same RSA-PSS signing as the REST API.
+ */
+import * as crypto from 'crypto';
+/**
+ * Generate authentication headers for WebSocket connection
+ *
+ * @param apiKeyId - Your Kalshi API key ID
+ * @param privateKey - Your private key in PEM format
+ * @param timestamp - Unix timestamp in milliseconds
+ * @returns Headers object for WebSocket connection
+ */
+export function generateWsAuthHeaders(apiKeyId, privateKey, timestamp = Date.now()) {
+    // The message to sign for WebSocket is typically the timestamp
+    const timestampStr = timestamp.toString();
+    // Create signature using RSA-PSS with SHA-256
+    const sign = crypto.createSign('RSA-SHA256');
+    sign.update(timestampStr);
+    sign.end();
+    // Sign with PSS padding
+    const signature = sign.sign({
+        key: privateKey,
+        padding: crypto.constants.RSA_PKCS1_PSS_PADDING,
+        saltLength: crypto.constants.RSA_PSS_SALTLEN_DIGEST,
+    }, 'base64');
+    return {
+        'KALSHI-ACCESS-KEY': apiKeyId,
+        'KALSHI-ACCESS-SIGNATURE': signature,
+        'KALSHI-ACCESS-TIMESTAMP': timestampStr,
+    };
+}
+/**
+ * Generate a signed WebSocket URL with auth params
+ *
+ * Some WebSocket implementations don't support custom headers,
+ * so we can pass auth as query parameters instead.
+ *
+ * @param baseUrl - WebSocket endpoint URL
+ * @param apiKeyId - Your Kalshi API key ID
+ * @param privateKey - Your private key in PEM format
+ * @returns Full WebSocket URL with auth parameters
+ */
+export function generateSignedWsUrl(baseUrl, apiKeyId, privateKey) {
+    const timestamp = Date.now();
+    const headers = generateWsAuthHeaders(apiKeyId, privateKey, timestamp);
+    const url = new URL(baseUrl);
+    url.searchParams.set('api_key', headers['KALSHI-ACCESS-KEY']);
+    url.searchParams.set('signature', headers['KALSHI-ACCESS-SIGNATURE']);
+    url.searchParams.set('timestamp', headers['KALSHI-ACCESS-TIMESTAMP']);
+    return url.toString();
+}
+//# sourceMappingURL=auth.js.map

package/dist/websocket/auth.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.js","sourceRoot":"","sources":["../../src/websocket/auth.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AAEjC;;;;;;;GAOG;AACH,MAAM,UAAU,qBAAqB,CACnC,QAAgB,EAChB,UAAkB,EAClB,YAAoB,IAAI,CAAC,GAAG,EAAE;IAE9B,+DAA+D;IAC/D,MAAM,YAAY,GAAG,SAAS,CAAC,QAAQ,EAAE,CAAC;IAE1C,8CAA8C;IAC9C,MAAM,IAAI,GAAG,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;IAC7C,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IAC1B,IAAI,CAAC,GAAG,EAAE,CAAC;IAEX,wBAAwB;IACxB,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC;QAC1B,GAAG,EAAE,UAAU;QACf,OAAO,EAAE,MAAM,CAAC,SAAS,CAAC,qBAAqB;QAC/C,UAAU,EAAE,MAAM,CAAC,SAAS,CAAC,sBAAsB;KACpD,EAAE,QAAQ,CAAC,CAAC;IAEb,OAAO;QACL,mBAAmB,EAAE,QAAQ;QAC7B,yBAAyB,EAAE,SAAS;QACpC,yBAAyB,EAAE,YAAY;KACxC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAM,UAAU,mBAAmB,CACjC,OAAe,EACf,QAAgB,EAChB,UAAkB;IAElB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,OAAO,GAAG,qBAAqB,CAAC,QAAQ,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;IAEvE,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;IAC7B,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC,CAAC;IAC9D,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,WAAW,EAAE,OAAO,CAAC,yBAAyB,CAAC,CAAC,CAAC;IACtE,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,WAAW,EAAE,OAAO,CAAC,yBAAyB,CAAC,CAAC,CAAC;IAEtE,OAAO,GAAG,CAAC,QAAQ,EAAE,CAAC;AACxB,CAAC"}

package/dist/websocket/auth.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=auth.test.d.ts.map

package/dist/websocket/auth.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.test.d.ts","sourceRoot":"","sources":["../../src/websocket/auth.test.ts"],"names":[],"mappings":""}

package/dist/websocket/auth.test.js

@@ -0,0 +1,56 @@
+import { describe, it, expect, vi } from 'vitest';
+// Mock crypto module since we can't use real RSA keys in tests easily
+vi.mock('crypto', async () => {
+    const actual = await vi.importActual('crypto');
+    return {
+        ...actual,
+        createSign: vi.fn(() => ({
+            update: vi.fn().mockReturnThis(),
+            end: vi.fn().mockReturnThis(),
+            sign: vi.fn(() => 'mocked-signature-base64'),
+        })),
+    };
+});
+import { generateWsAuthHeaders, generateSignedWsUrl } from './auth.js';
+describe('WebSocket Auth', () => {
+    describe('generateWsAuthHeaders', () => {
+        it('generates all required headers', () => {
+            const headers = generateWsAuthHeaders('test-api-key', 'fake-private-key');
+            expect(headers).toHaveProperty('KALSHI-ACCESS-KEY');
+            expect(headers).toHaveProperty('KALSHI-ACCESS-SIGNATURE');
+            expect(headers).toHaveProperty('KALSHI-ACCESS-TIMESTAMP');
+        });
+        it('uses the provided API key', () => {
+            const headers = generateWsAuthHeaders('my-api-key-123', 'fake-key');
+            expect(headers['KALSHI-ACCESS-KEY']).toBe('my-api-key-123');
+        });
+        it('uses the provided timestamp', () => {
+            const timestamp = 1703721600000; // Fixed timestamp
+            const headers = generateWsAuthHeaders('test-key', 'fake-key', timestamp);
+            expect(headers['KALSHI-ACCESS-TIMESTAMP']).toBe('1703721600000');
+        });
+        it('returns a signature', () => {
+            const headers = generateWsAuthHeaders('test-key', 'fake-key');
+            expect(headers['KALSHI-ACCESS-SIGNATURE']).toBeDefined();
+            expect(headers['KALSHI-ACCESS-SIGNATURE'].length).toBeGreaterThan(0);
+        });
+    });
+    describe('generateSignedWsUrl', () => {
+        it('appends auth parameters to URL', () => {
+            const url = generateSignedWsUrl('wss://api.example.com/ws', 'test-api-key', 'fake-key');
+            expect(url).toContain('api_key=test-api-key');
+            expect(url).toContain('signature=');
+            expect(url).toContain('timestamp=');
+        });
+        it('preserves the base URL', () => {
+            const url = generateSignedWsUrl('wss://api.elections.kalshi.com/trade-api/ws/v2', 'test-key', 'fake-key');
+            expect(url.startsWith('wss://api.elections.kalshi.com/trade-api/ws/v2')).toBe(true);
+        });
+        it('properly encodes parameters', () => {
+            const url = generateSignedWsUrl('wss://api.example.com/ws', 'test+key', 'fake-key');
+            // URL should properly encode the + character
+            expect(url).toContain('api_key=test%2Bkey');
+        });
+    });
+});
+//# sourceMappingURL=auth.test.js.map

package/dist/websocket/auth.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.test.js","sourceRoot":"","sources":["../../src/websocket/auth.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAGlD,sEAAsE;AACtE,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;IAC3B,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,YAAY,CAAgB,QAAQ,CAAC,CAAC;IAC9D,OAAO;QACL,GAAG,MAAM;QACT,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;YACvB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;YAChC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;YAC7B,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC;SAC7C,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,OAAO,EAAE,qBAAqB,EAAE,mBAAmB,EAAE,MAAM,WAAW,CAAC;AAEvE,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,OAAO,GAAG,qBAAqB,CAAC,cAAc,EAAE,kBAAkB,CAAC,CAAC;YAE1E,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC;YACpD,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,yBAAyB,CAAC,CAAC;YAC1D,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,yBAAyB,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,OAAO,GAAG,qBAAqB,CAAC,gBAAgB,EAAE,UAAU,CAAC,CAAC;YAEpE,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,SAAS,GAAG,aAAa,CAAC,CAAC,kBAAkB;YACnD,MAAM,OAAO,GAAG,qBAAqB,CAAC,UAAU,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;YAEzE,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE;YAC7B,MAAM,OAAO,GAAG,qBAAqB,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YAE9D,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACzD,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,GAAG,GAAG,mBAAmB,CAC7B,0BAA0B,EAC1B,cAAc,EACd,UAAU,CACX,CAAC;YAEF,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;YAC9C,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YACpC,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAChC,MAAM,GAAG,GAAG,mBAAmB,CAC7B,gDAAgD,EAChD,UAAU,EACV,UAAU,CACX,CAAC;YAEF,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,gDAAgD,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,GAAG,GAAG,mBAAmB,CAC7B,0BAA0B,EAC1B,UAAU,EACV,UAAU,CACX,CAAC;YAEF,6CAA6C;YAC7C,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/websocket/client.d.ts

@@ -0,0 +1,95 @@
+/**
+ * Kalshi WebSocket Client
+ *
+ * Provides real-time market data streaming from Kalshi.
+ * Features:
+ * - Automatic reconnection with exponential backoff
+ * - Subscription management
+ * - Event-based message handling
+ * - Connection health monitoring
+ */
+import { type KalshiWsConfig, type KalshiWsEventHandlers, type SubscriptionChannel, type ConnectionState } from './types.js';
+/**
+ * Kalshi WebSocket Client
+ *
+ * @example
+ * ```typescript
+ * const client = new KalshiWsClient({
+ *   apiKeyId: 'your-api-key',
+ *   privateKey: 'your-private-key',
+ * });
+ *
+ * client.on('ticker', (data) => {
+ *   console.log('Price update:', data.market_ticker, data.yes_bid);
+ * });
+ *
+ * await client.connect();
+ * client.subscribe(['ticker'], ['KXBTC-25JAN03']);
+ * ```
+ */
+export declare class KalshiWsClient {
+    private config;
+    private ws;
+    private handlers;
+    private state;
+    private reconnectAttempts;
+    private commandId;
+    private pingTimer;
+    private pongTimer;
+    private subscriptions;
+    constructor(config: KalshiWsConfig);
+    /**
+     * Get current connection state
+     */
+    get connectionState(): ConnectionState;
+    /**
+     * Check if connected
+     */
+    get isConnected(): boolean;
+    /**
+     * Register event handlers
+     */
+    on<K extends keyof KalshiWsEventHandlers>(event: K, handler: KalshiWsEventHandlers[K]): void;
+    /**
+     * Connect to WebSocket server
+     */
+    connect(): Promise<void>;
+    /**
+     * Disconnect from WebSocket server
+     */
+    disconnect(): void;
+    /**
+     * Subscribe to channels for specific market tickers
+     *
+     * @param channels - Channels to subscribe to
+     * @param marketTickers - Market tickers (not needed for 'fill' channel)
+     */
+    subscribe(channels: SubscriptionChannel[], marketTickers?: string[]): void;
+    /**
+     * Unsubscribe from channels
+     */
+    unsubscribe(channels: SubscriptionChannel[], marketTickers?: string[]): void;
+    /**
+     * Add markets to existing subscription
+     */
+    addMarkets(channels: SubscriptionChannel[], marketTickers: string[]): void;
+    /**
+     * Remove markets from existing subscription
+     */
+    removeMarkets(channels: SubscriptionChannel[], marketTickers: string[]): void;
+    /**
+     * Get currently subscribed tickers for a channel
+     */
+    getSubscribedTickers(channel: SubscriptionChannel): string[];
+    private send;
+    private handleMessage;
+    private handleClose;
+    private scheduleReconnect;
+    private resubscribeAll;
+    private updateLocalSubscriptions;
+    private startPingPong;
+    private stopPingPong;
+    private setPongTimeout;
+    private clearPongTimeout;
+}
+//# sourceMappingURL=client.d.ts.map

package/dist/websocket/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/websocket/client.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAGH,OAAO,EACL,KAAK,cAAc,EACnB,KAAK,qBAAqB,EAG1B,KAAK,mBAAmB,EACxB,KAAK,eAAe,EAGrB,MAAM,YAAY,CAAC;AAYpB;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAA2B;IACzC,OAAO,CAAC,EAAE,CAA0B;IACpC,OAAO,CAAC,QAAQ,CAA6B;IAC7C,OAAO,CAAC,KAAK,CAAmC;IAChD,OAAO,CAAC,iBAAiB,CAAK;IAC9B,OAAO,CAAC,SAAS,CAAK;IACtB,OAAO,CAAC,SAAS,CAA+C;IAChE,OAAO,CAAC,SAAS,CAA8C;IAC/D,OAAO,CAAC,aAAa,CAKnB;gBAEU,MAAM,EAAE,cAAc;IAclC;;OAEG;IACH,IAAI,eAAe,IAAI,eAAe,CAErC;IAED;;OAEG;IACH,IAAI,WAAW,IAAI,OAAO,CAEzB;IAED;;OAEG;IACH,EAAE,CAAC,CAAC,SAAS,MAAM,qBAAqB,EACtC,KAAK,EAAE,CAAC,EACR,OAAO,EAAE,qBAAqB,CAAC,CAAC,CAAC,GAChC,IAAI;IAIP;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAiD9B;;OAEG;IACH,UAAU,IAAI,IAAI;IAUlB;;;;;OAKG;IACH,SAAS,CAAC,QAAQ,EAAE,mBAAmB,EAAE,EAAE,aAAa,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI;IAoB1E;;OAEG;IACH,WAAW,CAAC,QAAQ,EAAE,mBAAmB,EAAE,EAAE,aAAa,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI;IAmB5E;;OAEG;IACH,UAAU,CAAC,QAAQ,EAAE,mBAAmB,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,GAAG,IAAI;IAoB1E;;OAEG;IACH,aAAa,CAAC,QAAQ,EAAE,mBAAmB,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,GAAG,IAAI;IAoB7E;;OAEG;IACH,oBAAoB,CAAC,OAAO,EAAE,mBAAmB,GAAG,MAAM,EAAE;IAS5D,OAAO,CAAC,IAAI;IAMZ,OAAO,CAAC,aAAa;IAyCrB,OAAO,CAAC,WAAW;IAenB,OAAO,CAAC,iBAAiB;IAgBzB,OAAO,CAAC,cAAc;IAsBtB,OAAO,CAAC,wBAAwB;IAoBhC,OAAO,CAAC,aAAa;IASrB,OAAO,CAAC,YAAY;IAQpB,OAAO,CAAC,cAAc;IAQtB,OAAO,CAAC,gBAAgB;CAMzB"}