agentic-qe 3.3.4 → 3.4.0

package/v3/dist/adapters/a2a/auth/token-store.js

@@ -0,0 +1,453 @@
+/**
+ * A2A Token Store
+ *
+ * In-memory token storage with TTL support for OAuth 2.0 tokens.
+ * Provides storage for access tokens, refresh tokens, and authorization codes.
+ *
+ * @module adapters/a2a/auth/token-store
+ */
+/**
+ * Default token store configuration
+ */
+export const DEFAULT_TOKEN_STORE_CONFIG = {
+    cleanupInterval: 60000,
+    maxTokens: 100000,
+    enableAutoCleanup: true,
+};
+// ============================================================================
+// Token Store Implementation
+// ============================================================================
+/**
+ * In-memory token store with TTL support
+ *
+ * Provides storage for OAuth 2.0 tokens with automatic cleanup
+ * of expired tokens.
+ */
+export class TokenStore {
+    config;
+    accessTokens = new Map();
+    refreshTokens = new Map();
+    authorizationCodes = new Map();
+    revokedTokens = new Set();
+    cleanupTimer;
+    lastCleanup = null;
+    totalExpiredCleaned = 0;
+    constructor(config = {}) {
+        this.config = { ...DEFAULT_TOKEN_STORE_CONFIG, ...config };
+        if (this.config.enableAutoCleanup && this.config.cleanupInterval > 0) {
+            this.startCleanupTimer();
+        }
+    }
+    // ============================================================================
+    // Access Token Operations
+    // ============================================================================
+    /**
+     * Store an access token
+     *
+     * @param token - The token string (JWT)
+     * @param claims - The token claims
+     */
+    storeAccessToken(token, claims) {
+        this.checkCapacity();
+        const fullClaims = {
+            ...claims,
+            tokenType: 'access',
+        };
+        this.accessTokens.set(token, fullClaims);
+    }
+    /**
+     * Get access token claims
+     *
+     * @param token - The token string
+     * @returns The token claims or null if not found/expired/revoked
+     */
+    getAccessTokenClaims(token) {
+        // Check if revoked
+        if (this.isRevoked(token)) {
+            return null;
+        }
+        const claims = this.accessTokens.get(token);
+        if (!claims) {
+            return null;
+        }
+        // Check if expired
+        if (this.isExpired(claims.expiresAt)) {
+            this.accessTokens.delete(token);
+            return null;
+        }
+        return claims;
+    }
+    /**
+     * Check if an access token is valid
+     *
+     * @param token - The token string
+     * @returns True if the token is valid
+     */
+    isAccessTokenValid(token) {
+        return this.getAccessTokenClaims(token) !== null;
+    }
+    // ============================================================================
+    // Refresh Token Operations
+    // ============================================================================
+    /**
+     * Store a refresh token
+     *
+     * @param token - The token string
+     * @param claims - The token claims
+     */
+    storeRefreshToken(token, claims) {
+        this.checkCapacity();
+        const fullClaims = {
+            ...claims,
+            tokenType: 'refresh',
+        };
+        this.refreshTokens.set(token, fullClaims);
+    }
+    /**
+     * Get refresh token claims
+     *
+     * @param token - The token string
+     * @returns The token claims or null if not found/expired/revoked
+     */
+    getRefreshTokenClaims(token) {
+        if (this.isRevoked(token)) {
+            return null;
+        }
+        const claims = this.refreshTokens.get(token);
+        if (!claims) {
+            return null;
+        }
+        if (this.isExpired(claims.expiresAt)) {
+            this.refreshTokens.delete(token);
+            return null;
+        }
+        return claims;
+    }
+    /**
+     * Increment the use count for a refresh token
+     *
+     * @param token - The refresh token
+     * @returns The updated claims or null if not found
+     */
+    incrementRefreshTokenUseCount(token) {
+        const claims = this.refreshTokens.get(token);
+        if (!claims) {
+            return null;
+        }
+        const updatedClaims = {
+            ...claims,
+            useCount: claims.useCount + 1,
+        };
+        this.refreshTokens.set(token, updatedClaims);
+        return updatedClaims;
+    }
+    // ============================================================================
+    // Authorization Code Operations
+    // ============================================================================
+    /**
+     * Store an authorization code
+     *
+     * @param code - The authorization code
+     * @param claims - The code claims
+     */
+    storeAuthorizationCode(code, claims) {
+        this.authorizationCodes.set(code, {
+            ...claims,
+            used: false,
+        });
+    }
+    /**
+     * Get authorization code claims
+     *
+     * @param code - The authorization code
+     * @returns The code claims or null if not found/expired/used
+     */
+    getAuthorizationCodeClaims(code) {
+        const claims = this.authorizationCodes.get(code);
+        if (!claims) {
+            return null;
+        }
+        // Check if expired
+        if (this.isExpired(claims.expiresAt)) {
+            this.authorizationCodes.delete(code);
+            return null;
+        }
+        // Check if already used
+        if (claims.used) {
+            return null;
+        }
+        return claims;
+    }
+    /**
+     * Mark an authorization code as used
+     *
+     * @param code - The authorization code
+     * @returns True if the code was marked as used
+     */
+    markAuthorizationCodeUsed(code) {
+        const claims = this.authorizationCodes.get(code);
+        if (!claims || claims.used) {
+            return false;
+        }
+        claims.used = true;
+        return true;
+    }
+    /**
+     * Consume an authorization code (get and mark as used)
+     *
+     * @param code - The authorization code
+     * @returns The code claims or null if not valid
+     */
+    consumeAuthorizationCode(code) {
+        const claims = this.getAuthorizationCodeClaims(code);
+        if (!claims) {
+            return null;
+        }
+        this.markAuthorizationCodeUsed(code);
+        return claims;
+    }
+    // ============================================================================
+    // Token Revocation
+    // ============================================================================
+    /**
+     * Revoke a token
+     *
+     * @param token - The token string
+     * @returns True if the token was revoked
+     */
+    revokeToken(token) {
+        // Add to revocation list
+        this.revokedTokens.add(token);
+        // Remove from active stores
+        const wasAccessToken = this.accessTokens.delete(token);
+        const wasRefreshToken = this.refreshTokens.delete(token);
+        return wasAccessToken || wasRefreshToken;
+    }
+    /**
+     * Revoke all tokens for a client
+     *
+     * @param clientId - The client ID
+     * @returns Number of tokens revoked
+     */
+    revokeTokensByClient(clientId) {
+        let count = 0;
+        // Revoke access tokens
+        for (const [token, claims] of this.accessTokens) {
+            if (claims.clientId === clientId) {
+                this.revokeToken(token);
+                count++;
+            }
+        }
+        // Revoke refresh tokens
+        for (const [token, claims] of this.refreshTokens) {
+            if (claims.clientId === clientId) {
+                this.revokeToken(token);
+                count++;
+            }
+        }
+        return count;
+    }
+    /**
+     * Check if a token is revoked
+     *
+     * @param token - The token string
+     * @returns True if the token is revoked
+     */
+    isRevoked(token) {
+        return this.revokedTokens.has(token);
+    }
+    // ============================================================================
+    // Generic Token Operations
+    // ============================================================================
+    /**
+     * Get claims for any token type
+     *
+     * @param token - The token string
+     * @returns The token claims or null
+     */
+    getTokenClaims(token) {
+        const accessClaims = this.getAccessTokenClaims(token);
+        if (accessClaims) {
+            return accessClaims;
+        }
+        return this.getRefreshTokenClaims(token);
+    }
+    /**
+     * Check if a token exists and is valid
+     *
+     * @param token - The token string
+     * @returns True if the token is valid
+     */
+    isTokenValid(token) {
+        return this.getTokenClaims(token) !== null;
+    }
+    // ============================================================================
+    // Cleanup Operations
+    // ============================================================================
+    /**
+     * Clean up expired tokens
+     *
+     * @returns Number of tokens cleaned up
+     */
+    cleanup() {
+        const now = Math.floor(Date.now() / 1000);
+        let cleaned = 0;
+        // Clean access tokens
+        for (const [token, claims] of this.accessTokens) {
+            if (this.isExpired(claims.expiresAt)) {
+                this.accessTokens.delete(token);
+                cleaned++;
+            }
+        }
+        // Clean refresh tokens
+        for (const [token, claims] of this.refreshTokens) {
+            if (this.isExpired(claims.expiresAt)) {
+                this.refreshTokens.delete(token);
+                cleaned++;
+            }
+        }
+        // Clean authorization codes
+        for (const [code, claims] of this.authorizationCodes) {
+            if (this.isExpired(claims.expiresAt)) {
+                this.authorizationCodes.delete(code);
+                cleaned++;
+            }
+        }
+        // Clean old revoked tokens (keep for 24 hours)
+        const dayAgo = now - 86400;
+        for (const token of this.revokedTokens) {
+            // We can't know when it was revoked, so we check if there are too many
+            if (this.revokedTokens.size > 10000) {
+                // Just remove some to prevent unbounded growth
+                this.revokedTokens.delete(token);
+                cleaned++;
+            }
+        }
+        this.lastCleanup = now;
+        this.totalExpiredCleaned += cleaned;
+        return cleaned;
+    }
+    /**
+     * Start the automatic cleanup timer
+     */
+    startCleanupTimer() {
+        this.cleanupTimer = setInterval(() => {
+            this.cleanup();
+        }, this.config.cleanupInterval);
+        // Don't prevent process exit
+        if (typeof this.cleanupTimer.unref === 'function') {
+            this.cleanupTimer.unref();
+        }
+    }
+    /**
+     * Stop the automatic cleanup timer
+     */
+    stopCleanupTimer() {
+        if (this.cleanupTimer) {
+            clearInterval(this.cleanupTimer);
+            this.cleanupTimer = undefined;
+        }
+    }
+    /**
+     * Check capacity and evict if necessary
+     */
+    checkCapacity() {
+        if (this.config.maxTokens <= 0) {
+            return;
+        }
+        const totalTokens = this.accessTokens.size + this.refreshTokens.size;
+        if (totalTokens >= this.config.maxTokens) {
+            // Run cleanup first
+            this.cleanup();
+            // If still over capacity, evict oldest tokens
+            const stillOver = this.accessTokens.size + this.refreshTokens.size - this.config.maxTokens;
+            if (stillOver > 0) {
+                this.evictOldestTokens(stillOver + 100); // Evict some extra
+            }
+        }
+    }
+    /**
+     * Evict the oldest tokens
+     *
+     * @param count - Number of tokens to evict
+     */
+    evictOldestTokens(count) {
+        // Get all tokens sorted by issuedAt
+        const allTokens = [];
+        for (const [token, claims] of this.accessTokens) {
+            allTokens.push({ token, issuedAt: claims.issuedAt, type: 'access' });
+        }
+        for (const [token, claims] of this.refreshTokens) {
+            allTokens.push({ token, issuedAt: claims.issuedAt, type: 'refresh' });
+        }
+        allTokens.sort((a, b) => a.issuedAt - b.issuedAt);
+        // Evict oldest
+        const toEvict = allTokens.slice(0, count);
+        for (const { token, type } of toEvict) {
+            if (type === 'access') {
+                this.accessTokens.delete(token);
+            }
+            else {
+                this.refreshTokens.delete(token);
+            }
+        }
+    }
+    /**
+     * Check if a timestamp has expired
+     */
+    isExpired(expiresAt) {
+        const now = Math.floor(Date.now() / 1000);
+        return expiresAt <= now;
+    }
+    // ============================================================================
+    // Statistics and Management
+    // ============================================================================
+    /**
+     * Get store statistics
+     */
+    getStats() {
+        return {
+            accessTokenCount: this.accessTokens.size,
+            refreshTokenCount: this.refreshTokens.size,
+            authCodeCount: this.authorizationCodes.size,
+            revokedTokenCount: this.revokedTokens.size,
+            lastCleanup: this.lastCleanup,
+            totalExpiredCleaned: this.totalExpiredCleaned,
+        };
+    }
+    /**
+     * Clear all tokens
+     */
+    clear() {
+        this.accessTokens.clear();
+        this.refreshTokens.clear();
+        this.authorizationCodes.clear();
+        this.revokedTokens.clear();
+    }
+    /**
+     * Destroy the store
+     */
+    destroy() {
+        this.stopCleanupTimer();
+        this.clear();
+    }
+    /**
+     * Get the total number of tokens
+     */
+    get size() {
+        return this.accessTokens.size + this.refreshTokens.size;
+    }
+}
+// ============================================================================
+// Factory Function
+// ============================================================================
+/**
+ * Create a new TokenStore instance
+ *
+ * @param config - Store configuration
+ * @returns A new TokenStore
+ */
+export function createTokenStore(config = {}) {
+    return new TokenStore(config);
+}
+//# sourceMappingURL=token-store.js.map

package/v3/dist/adapters/a2a/auth/token-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"token-store.js","sourceRoot":"","sources":["../../../../src/adapters/a2a/auth/token-store.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AA6GH;;GAEG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAA+B;IACpE,eAAe,EAAE,KAAK;IACtB,SAAS,EAAE,MAAM;IACjB,iBAAiB,EAAE,IAAI;CACxB,CAAC;AA6BF,+EAA+E;AAC/E,6BAA6B;AAC7B,+EAA+E;AAE/E;;;;;GAKG;AACH,MAAM,OAAO,UAAU;IACJ,MAAM,CAA6B;IACnC,YAAY,GAAmC,IAAI,GAAG,EAAE,CAAC;IACzD,aAAa,GAAoC,IAAI,GAAG,EAAE,CAAC;IAC3D,kBAAkB,GAAyC,IAAI,GAAG,EAAE,CAAC;IACrE,aAAa,GAAgB,IAAI,GAAG,EAAE,CAAC;IAChD,YAAY,CAAkC;IAC9C,WAAW,GAAkB,IAAI,CAAC;IAClC,mBAAmB,GAAG,CAAC,CAAC;IAEhC,YAAY,SAA2B,EAAE;QACvC,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,0BAA0B,EAAE,GAAG,MAAM,EAAE,CAAC;QAE3D,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;YACrE,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,+EAA+E;IAC/E,0BAA0B;IAC1B,+EAA+E;IAE/E;;;;;OAKG;IACH,gBAAgB,CAAC,KAAa,EAAE,MAA4C;QAC1E,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,MAAM,UAAU,GAAsB;YACpC,GAAG,MAAM;YACT,SAAS,EAAE,QAAQ;SACpB,CAAC;QAEF,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IAC3C,CAAC;IAED;;;;;OAKG;IACH,oBAAoB,CAAC,KAAa;QAChC,mBAAmB;QACnB,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC;QACd,CAAC;QAED,mBAAmB;QACnB,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YACrC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAChC,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;OAKG;IACH,kBAAkB,CAAC,KAAa;QAC9B,OAAO,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,KAAK,IAAI,CAAC;IACnD,CAAC;IAED,+EAA+E;IAC/E,2BAA2B;IAC3B,+EAA+E;IAE/E;;;;;OAKG;IACH,iBAAiB,CAAC,KAAa,EAAE,MAA6C;QAC5E,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,MAAM,UAAU,GAAuB;YACrC,GAAG,MAAM;YACT,SAAS,EAAE,SAAS;SACrB,CAAC;QAEF,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IAC5C,CAAC;IAED;;;;;OAKG;IACH,qBAAqB,CAAC,KAAa;QACjC,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YACrC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACjC,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;OAKG;IACH,6BAA6B,CAAC,KAAa;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,aAAa,GAAuB;YACxC,GAAG,MAAM;YACT,QAAQ,EAAE,MAAM,CAAC,QAAQ,GAAG,CAAC;SAC9B,CAAC;QAEF,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;QAC7C,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,+EAA+E;IAC/E,gCAAgC;IAChC,+EAA+E;IAE/E;;;;;OAKG;IACH,sBAAsB,CAAC,IAAY,EAAE,MAA6C;QAChF,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,EAAE;YAChC,GAAG,MAAM;YACT,IAAI,EAAE,KAAK;SACZ,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACH,0BAA0B,CAAC,IAAY;QACrC,MAAM,MAAM,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACjD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC;QACd,CAAC;QAED,mBAAmB;QACnB,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YACrC,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACrC,OAAO,IAAI,CAAC;QACd,CAAC;QAED,wBAAwB;QACxB,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAChB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;OAKG;IACH,yBAAyB,CAAC,IAAY;QACpC,MAAM,MAAM,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACjD,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAC3B,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;QACnB,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;OAKG;IACH,wBAAwB,CAAC,IAAY;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC;QACrC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,+EAA+E;IAC/E,mBAAmB;IACnB,+EAA+E;IAE/E;;;;;OAKG;IACH,WAAW,CAAC,KAAa;QACvB,yBAAyB;QACzB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAE9B,4BAA4B;QAC5B,MAAM,cAAc,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvD,MAAM,eAAe,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAEzD,OAAO,cAAc,IAAI,eAAe,CAAC;IAC3C,CAAC;IAED;;;;;OAKG;IACH,oBAAoB,CAAC,QAAgB;QACnC,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,uBAAuB;QACvB,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YAChD,IAAI,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACjC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBACxB,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QAED,wBAAwB;QACxB,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACjD,IAAI,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACjC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBACxB,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;;;OAKG;IACH,SAAS,CAAC,KAAa;QACrB,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC;IAED,+EAA+E;IAC/E,2BAA2B;IAC3B,+EAA+E;IAE/E;;;;;OAKG;IACH,cAAc,CAAC,KAAa;QAC1B,MAAM,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACtD,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,OAAO,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;IAC3C,CAAC;IAED;;;;;OAKG;IACH,YAAY,CAAC,KAAa;QACxB,OAAO,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,KAAK,IAAI,CAAC;IAC7C,CAAC;IAED,+EAA+E;IAC/E,qBAAqB;IACrB,+EAA+E;IAE/E;;;;OAIG;IACH,OAAO;QACL,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;QAC1C,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,sBAAsB;QACtB,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YAChD,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChC,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QAED,uBAAuB;QACvB,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACjD,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACjC,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QAED,4BAA4B;QAC5B,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACrD,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrC,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACrC,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QAED,+CAA+C;QAC/C,MAAM,MAAM,GAAG,GAAG,GAAG,KAAK,CAAC;QAC3B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvC,uEAAuE;YACvE,IAAI,IAAI,CAAC,aAAa,CAAC,IAAI,GAAG,KAAK,EAAE,CAAC;gBACpC,+CAA+C;gBAC/C,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACjC,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC;QACvB,IAAI,CAAC,mBAAmB,IAAI,OAAO,CAAC;QAEpC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACK,iBAAiB;QACvB,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE;YACnC,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;QAEhC,6BAA6B;QAC7B,IAAI,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;YAClD,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACjC,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;QAChC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa;QACnB,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,CAAC,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;QACrE,IAAI,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YACzC,oBAAoB;YACpB,IAAI,CAAC,OAAO,EAAE,CAAC;YAEf,8CAA8C;YAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;YAC3F,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;gBAClB,IAAI,CAAC,iBAAiB,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC,CAAC,mBAAmB;YAC9D,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,iBAAiB,CAAC,KAAa;QACrC,oCAAoC;QACpC,MAAM,SAAS,GAA2E,EAAE,CAAC;QAE7F,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YAChD,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QACvE,CAAC;QAED,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACjD,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QACxE,CAAC;QAED,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;QAElD,eAAe;QACf,MAAM,OAAO,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QAC1C,KAAK,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,OAAO,EAAE,CAAC;YACtC,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtB,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAClC,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,SAAS,CAAC,SAAiB;QACjC,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;QAC1C,OAAO,SAAS,IAAI,GAAG,CAAC;IAC1B,CAAC;IAED,+EAA+E;IAC/E,4BAA4B;IAC5B,+EAA+E;IAE/E;;OAEG;IACH,QAAQ;QACN,OAAO;YACL,gBAAgB,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;YACxC,iBAAiB,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI;YAC1C,aAAa,EAAE,IAAI,CAAC,kBAAkB,CAAC,IAAI;YAC3C,iBAAiB,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI;YAC1C,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,mBAAmB,EAAE,IAAI,CAAC,mBAAmB;SAC9C,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC1B,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;QAC3B,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;QAChC,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;IAED;;OAEG;IACH,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;IAC1D,CAAC;CACF;AAED,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAAC,SAA2B,EAAE;IAC5D,OAAO,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;AAChC,CAAC"}

package/v3/dist/adapters/a2a/discovery/agent-health.d.ts

@@ -0,0 +1,226 @@
+/**
+ * A2A Agent Health Checker
+ *
+ * Provides health checking functionality for registered agents.
+ * Supports individual and bulk health checks with configurable intervals.
+ *
+ * @module adapters/a2a/discovery/agent-health
+ */
+import { EventEmitter } from 'events';
+import { QEAgentCard } from '../agent-cards/schema.js';
+/**
+ * Agent health status levels
+ */
+export type HealthStatus = 'healthy' | 'degraded' | 'unavailable' | 'unknown';
+/**
+ * Health check result for an individual agent
+ */
+export interface AgentHealthStatus {
+    /** Agent identifier */
+    readonly agentId: string;
+    /** Current health status */
+    readonly status: HealthStatus;
+    /** Timestamp of the last health check */
+    readonly lastCheck: Date;
+    /** Response time in milliseconds (if available) */
+    readonly responseTime?: number;
+    /** Consecutive error count */
+    readonly errorCount: number;
+    /** Consecutive success count */
+    readonly successCount: number;
+    /** Error message (if unhealthy) */
+    readonly errorMessage?: string;
+    /** Additional health details */
+    readonly details?: Record<string, unknown>;
+}
+/**
+ * Health check configuration
+ */
+export interface HealthCheckConfig {
+    /** Timeout for health check requests in milliseconds */
+    readonly timeoutMs?: number;
+    /** Number of consecutive failures before marking as unhealthy */
+    readonly failureThreshold?: number;
+    /** Number of consecutive successes to recover from unhealthy */
+    readonly recoveryThreshold?: number;
+    /** Response time threshold for degraded status (ms) */
+    readonly degradedThresholdMs?: number;
+    /** Custom health check function */
+    readonly checkFn?: (agentId: string, card: QEAgentCard) => Promise<HealthCheckResult>;
+}
+/**
+ * Default health check configuration
+ */
+export declare const DEFAULT_HEALTH_CHECK_CONFIG: Required<Omit<HealthCheckConfig, 'checkFn'>>;
+/**
+ * Result from a health check attempt
+ */
+export interface HealthCheckResult {
+    /** Whether the check was successful */
+    readonly success: boolean;
+    /** Response time in milliseconds */
+    readonly responseTime?: number;
+    /** Error if check failed */
+    readonly error?: Error;
+    /** Additional details */
+    readonly details?: Record<string, unknown>;
+}
+/**
+ * Periodic check configuration
+ */
+export interface PeriodicCheckConfig {
+    /** Interval between checks in milliseconds */
+    readonly intervalMs: number;
+    /** Whether to check all agents or only previously unhealthy ones */
+    readonly checkOnlyUnhealthy?: boolean;
+    /** Callback for status changes */
+    readonly onStatusChange?: (agentId: string, oldStatus: HealthStatus, newStatus: HealthStatus) => void;
+}
+/**
+ * Health summary statistics
+ */
+export interface HealthSummary {
+    /** Total number of agents */
+    readonly total: number;
+    /** Number of healthy agents */
+    readonly healthy: number;
+    /** Number of degraded agents */
+    readonly degraded: number;
+    /** Number of unavailable agents */
+    readonly unavailable: number;
+    /** Number of unknown status agents */
+    readonly unknown: number;
+    /** Overall health percentage (0-100) */
+    readonly healthPercentage: number;
+    /** Last summary calculation time */
+    readonly calculatedAt: Date;
+}
+/**
+ * Event map for AgentHealthChecker
+ */
+export interface HealthCheckerEvents {
+    'status-change': (agentId: string, oldStatus: HealthStatus, newStatus: HealthStatus) => void;
+    'check-complete': (agentId: string, result: AgentHealthStatus) => void;
+    'check-error': (agentId: string, error: Error) => void;
+    'all-checks-complete': (summary: HealthSummary) => void;
+}
+/**
+ * Agent Health Checker
+ *
+ * Monitors the health of registered agents through periodic or on-demand checks.
+ * Supports configurable thresholds for failure detection and recovery.
+ */
+export declare class AgentHealthChecker extends EventEmitter {
+    private readonly config;
+    private readonly healthStates;
+    private readonly agentCards;
+    private periodicCheckTimer;
+    private periodicConfig;
+    constructor(agentCards: Map<string, QEAgentCard>, config?: HealthCheckConfig);
+    /**
+     * Check the health of a specific agent
+     */
+    checkAgent(agentId: string): Promise<AgentHealthStatus>;
+    /**
+     * Check the health of all registered agents
+     */
+    checkAll(): Promise<Map<string, AgentHealthStatus>>;
+    /**
+     * Start periodic health checks
+     */
+    startPeriodicChecks(config: PeriodicCheckConfig): void;
+    /**
+     * Stop periodic health checks
+     */
+    stopPeriodicChecks(): void;
+    /**
+     * Check if periodic checks are running
+     */
+    isPeriodicCheckRunning(): boolean;
+    /**
+     * Get list of healthy agent IDs
+     */
+    getHealthyAgents(): string[];
+    /**
+     * Get list of unhealthy agent IDs (degraded or unavailable)
+     */
+    getUnhealthyAgents(): string[];
+    /**
+     * Get health status for a specific agent
+     */
+    getStatus(agentId: string): AgentHealthStatus | null;
+    /**
+     * Get all health statuses
+     */
+    getAllStatuses(): Map<string, AgentHealthStatus>;
+    /**
+     * Get health summary
+     */
+    getSummary(): HealthSummary;
+    /**
+     * Reset health state for an agent
+     */
+    resetAgent(agentId: string): void;
+    /**
+     * Reset all health states
+     */
+    resetAll(): void;
+    /**
+     * Update the agent cards reference (e.g., after hot reload)
+     */
+    updateAgentCards(cards: Map<string, QEAgentCard>): void;
+    /**
+     * Perform the actual health check
+     */
+    private performHealthCheck;
+    /**
+     * Update health state based on check result
+     */
+    private updateHealthState;
+    /**
+     * Handle a check error
+     */
+    private handleCheckError;
+    /**
+     * Convert state to status
+     */
+    private stateToStatus;
+    /**
+     * Create an unknown status
+     */
+    private createUnknownStatus;
+    /**
+     * Calculate health summary
+     */
+    private calculateSummary;
+    /**
+     * Run periodic check
+     */
+    private runPeriodicCheck;
+}
+/**
+ * Create an Agent Health Checker instance
+ *
+ * @param agentCards - Map of agent IDs to their cards
+ * @param config - Health check configuration
+ * @returns Agent health checker instance
+ *
+ * @example
+ * ```typescript
+ * const healthChecker = createAgentHealthChecker(agentCards, {
+ *   timeoutMs: 5000,
+ *   failureThreshold: 3,
+ * });
+ *
+ * // Check individual agent
+ * const status = await healthChecker.checkAgent('qe-test-architect');
+ *
+ * // Start periodic checks
+ * healthChecker.startPeriodicChecks({ intervalMs: 30000 });
+ *
+ * // Get healthy agents
+ * const healthyAgents = healthChecker.getHealthyAgents();
+ * ```
+ */
+export declare function createAgentHealthChecker(agentCards: Map<string, QEAgentCard>, config?: HealthCheckConfig): AgentHealthChecker;
+//# sourceMappingURL=agent-health.d.ts.map