@usageflow/core 0.1.3 → 0.1.5

package/dist/base.d.ts

@@ -1,23 +1,29 @@
-import { Route, UsageFlowConfig, RoutesMap, Headers } from "./types";
+import { Route, UsageFlowConfig, RoutesMap, Headers, RequestForAllocation, RequestMetadata, UsageFlowRequest } from "./types";
+import { UsageFlowSocketManger } from "./socket";
 export declare abstract class UsageFlowAPI {
     protected apiKey: string | null;
     protected usageflowUrl: string;
-    protected apiConfig: UsageFlowConfig | null;
+    protected webServer: 'express' | 'fastify' | 'nestjs';
+    protected apiConfigs: UsageFlowConfig[];
     private configUpdateInterval;
+    socketManager: UsageFlowSocketManger | null;
+    private applicationId;
     /**
      * Initialize the UsageFlow API with credentials
      */
     init(apiKey: string, usageflowUrl?: string): UsageFlowAPI;
+    setWebServer(webServer: 'express' | 'fastify' | 'nestjs'): void;
     getApiKey(): string | null;
     getUsageflowUrl(): string;
     /**
      * Start background config update process
      */
     private startConfigUpdater;
-    /**
-     * Fetch API configuration from UsageFlow
-     */
-    private fetchApiConfig;
+    protected getRoutePattern(request: UsageFlowRequest): string;
+    guessLedgerId(request: UsageFlowRequest): string;
+    private fetchApiPolicies;
+    useAllocationRequest(payload: RequestForAllocation): Promise<void>;
+    allocationRequest(request: UsageFlowRequest, payload: RequestForAllocation, metadata: RequestMetadata): Promise<void>;
     /**
      * Convert routes list to a map for faster lookup
      */
@@ -39,6 +45,10 @@ export declare abstract class UsageFlowAPI {
      */
     destroy(): void;
     extractBearerToken(authHeader?: string): string | null;
+    /**
+     * Get header value from headers object (handles both Record and Headers types)
+     */
+    private getHeaderValue;
     /**
      * Decodes a JWT token without verifying the signature
      * @param token The JWT token to decode

package/dist/base.js

@@ -1,17 +1,16 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.UsageFlowAPI = void 0;
-const axios_1 = __importDefault(require("axios"));
-const types_1 = require("./types");
+const socket_1 = require("./socket");
 class UsageFlowAPI {
     constructor() {
         this.apiKey = null;
         this.usageflowUrl = "https://api.usageflow.io";
-        this.apiConfig = null;
+        this.webServer = 'express';
+        this.apiConfigs = [];
         this.configUpdateInterval = null;
+        this.socketManager = null;
+        this.applicationId = false;
     }
     /**
      * Initialize the UsageFlow API with credentials
@@ -19,9 +18,19 @@ class UsageFlowAPI {
     init(apiKey, usageflowUrl = "https://api.usageflow.io") {
         this.apiKey = apiKey;
         this.usageflowUrl = usageflowUrl;
-        this.startConfigUpdater();
+        // this.startConfigUpdater();
+        this.socketManager = new socket_1.UsageFlowSocketManger(apiKey);
+        // Connect the socket manager
+        this.socketManager.connect().catch((error) => {
+            console.error("[UsageFlow] Failed to establish WebSocket connection:", error);
+        }).then(() => {
+            this.startConfigUpdater();
+        });
         return this;
     }
+    setWebServer(webServer) {
+        this.webServer = webServer;
+    }
     getApiKey() {
         return this.apiKey;
     }
@@ -36,35 +45,231 @@ class UsageFlowAPI {
             clearInterval(this.configUpdateInterval);
         }
         this.configUpdateInterval = setInterval(async () => {
-            try {
-                const config = await this.fetchApiConfig();
-                if (config) {
-                    this.apiConfig = config;
+            await this.fetchApiPolicies();
+        }, 60000);
+        // this.fetchApiConfig().catch(console.error);
+    }
+    getRoutePattern(request) {
+        var _a, _b, _c, _d, _e, _f, _g, _h, _j;
+        if (this.webServer === 'fastify') {
+            return ((_a = request === null || request === void 0 ? void 0 : request.routeOptions) === null || _a === void 0 ? void 0 : _a.url) || request.url || '';
+        }
+        const routePattern = ((_b = request.route) === null || _b === void 0 ? void 0 : _b.path) || ((_d = (_c = request.app._router.stack.find((route) => {
+            // a => a.path == request.url
+            if (!route.route)
+                return false;
+            if (route.path) {
+                return route.path == request.url;
+            }
+            if (route.regexp) {
+                return route.regexp.test(request.url);
+            }
+        })) === null || _c === void 0 ? void 0 : _c.route) === null || _d === void 0 ? void 0 : _d.path);
+        if (routePattern) {
+            return routePattern;
+        }
+        // Method 1: Use request.route.path (available after route matching)
+        if ((_e = request.route) === null || _e === void 0 ? void 0 : _e.path) {
+            const baseUrl = request.baseUrl || "";
+            const routePath = request.route.path;
+            // Combine baseUrl and routePath to get full pattern
+            const fullPath = baseUrl + routePath;
+            return fullPath || "/";
+        }
+        // Method 2: Try to find route from router stack
+        try {
+            const app = request.app;
+            const router = app._router;
+            if (router && router.stack) {
+                const method = request.method.toLowerCase();
+                const path = request.path || ((_f = request.url) === null || _f === void 0 ? void 0 : _f.split("?")[0]) || "";
+                // Search through router stack for matching route
+                for (const layer of router.stack) {
+                    if (layer.route) {
+                        const route = layer.route;
+                        const routePath = route.path;
+                        const routeMethods = Object.keys(route.methods).map(m => m.toLowerCase());
+                        // Check if method matches and path pattern matches
+                        if (routeMethods.includes(method) || routeMethods.includes("*")) {
+                            // Try to match the pattern by checking if params match
+                            const pattern = (request.baseUrl || "") + routePath;
+                            // Simple check: if we have params and the route has params, it might match
+                            if (request.params && Object.keys(request.params).length > 0) {
+                                // Check if route path contains the param names
+                                const paramNames = Object.keys(request.params);
+                                const routeHasParams = paramNames.some(param => routePath.includes(`:${param}`));
+                                if (routeHasParams) {
+                                    return pattern;
+                                }
+                            }
+                            else if (!routePath.includes(":")) {
+                                // If no params in request and route has no params, check exact match
+                                if (path === pattern || path === routePath) {
+                                    return pattern;
+                                }
+                            }
+                        }
+                    }
+                    else if (layer.name === "router" && layer.handle && layer.handle.stack) {
+                        // Handle router middleware (nested routers)
+                        const mountPath = layer.regexp.source
+                            .replace("\\/?", "")
+                            .replace("(?=\\/|$)", "")
+                            .replace(/\\\//g, "/")
+                            .replace(/\^/g, "")
+                            .replace(/\$/g, "")
+                            .replace(/\\/g, "");
+                        for (const subLayer of layer.handle.stack) {
+                            if (subLayer.route) {
+                                const route = subLayer.route;
+                                const routePath = route.path;
+                                const routeMethods = Object.keys(route.methods).map(m => m.toLowerCase());
+                                if (routeMethods.includes(method) || routeMethods.includes("*")) {
+                                    const fullPath = mountPath + routePath;
+                                    // Check if this route matches by examining params
+                                    if (request.params && Object.keys(request.params).length > 0) {
+                                        const paramNames = Object.keys(request.params);
+                                        const routeHasParams = paramNames.some(param => routePath.includes(`:${param}`));
+                                        if (routeHasParams) {
+                                            return fullPath;
+                                        }
+                                    }
+                                    else if (!routePath.includes(":")) {
+                                        const currentPath = request.path || ((_g = request.url) === null || _g === void 0 ? void 0 : _g.split("?")[0]) || "";
+                                        if (currentPath === fullPath || currentPath === routePath) {
+                                            return fullPath;
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
                 }
             }
-            catch (error) {
-                console.error("Error fetching API config:", error);
+        }
+        catch (error) {
+            // Silently fail and try next method
+            console.debug("[UsageFlow] Could not extract route from router stack:", error);
+        }
+        // Method 3: Reconstruct pattern from params and path
+        if (request.params && Object.keys(request.params).length > 0) {
+            let path = request.path || ((_h = request.url) === null || _h === void 0 ? void 0 : _h.split("?")[0]) || "/";
+            const baseUrl = request.baseUrl || "";
+            // Split path into segments and replace matching segments with param names
+            const pathSegments = path.split("/");
+            const paramEntries = Object.entries(request.params);
+            // Replace segments that match param values with :paramName
+            for (let i = 0; i < pathSegments.length; i++) {
+                for (const [paramName, paramValue] of paramEntries) {
+                    if (pathSegments[i] === paramValue) {
+                        pathSegments[i] = `:${paramName}`;
+                        break; // Only replace once per segment
+                    }
+                }
             }
-        }, 60000);
-        this.fetchApiConfig().catch(console.error);
+            const pattern = baseUrl + pathSegments.join("/");
+            return pattern;
+        }
+        // Fallback: return baseUrl + path or just path
+        const baseUrl = request.baseUrl || "";
+        const path = request.path || ((_j = request.url) === null || _j === void 0 ? void 0 : _j.split("?")[0]) || "/";
+        return baseUrl + path || "/";
     }
-    /**
-     * Fetch API configuration from UsageFlow
-     */
-    async fetchApiConfig() {
-        if (!this.apiKey) {
-            throw new types_1.UsageFlowError("API key not initialized");
+    guessLedgerId(request) {
+        var _a, _b, _c;
+        const method = request.method;
+        const url = this.getRoutePattern(request);
+        const configs = this.apiConfigs;
+        if (!configs.length) {
+            return `${method} ${url}`;
         }
-        try {
-            const response = await axios_1.default.get(`${this.usageflowUrl}/api/v1/strategies/application`, {
-                headers: { "x-usage-key": this.apiKey },
-                timeout: 10000,
+        for (const config of configs) {
+            const fieldName = config.identityFieldName;
+            const location = config.identityFieldLocation;
+            switch (location) {
+                case "path_params":
+                    if ((_a = request.params) === null || _a === void 0 ? void 0 : _a[fieldName]) {
+                        return `${method} ${url} ${request.params[fieldName]}`;
+                    }
+                    break;
+                case "query_params":
+                    if ((_b = request.query) === null || _b === void 0 ? void 0 : _b[fieldName]) {
+                        return `${method} ${url} ${request.query[fieldName]}`;
+                    }
+                    break;
+                case "body":
+                    if ((_c = request.body) === null || _c === void 0 ? void 0 : _c[fieldName]) {
+                        return `${method} ${url} ${request.body[fieldName]}`;
+                    }
+                    break;
+                case "bearer_token":
+                    const authHeader = this.getHeaderValue(request.headers, 'authorization');
+                    const token = this.extractBearerToken(authHeader || undefined);
+                    if (token) {
+                        const claims = this.decodeJwtUnverified(token);
+                        if (claims === null || claims === void 0 ? void 0 : claims[fieldName]) {
+                            return `${method} ${url} ${claims[fieldName]}`;
+                        }
+                    }
+                    break;
+                case "headers": {
+                    const headerValue = this.getHeaderValue(request.headers, fieldName);
+                    if (headerValue) {
+                        return `${method} ${url} ${headerValue}`;
+                    }
+                    break;
+                }
+            }
+        }
+        return `${method} ${url}`;
+    }
+    async fetchApiPolicies() {
+        var _a;
+        if (this.socketManager && this.socketManager.isConnected()) {
+            const response = await this.socketManager.sendAsync({
+                type: "get_application_policies",
+                payload: null,
             });
-            return response.data;
+            if (response.type === 'success') {
+                this.apiConfigs = ((_a = response.payload) === null || _a === void 0 ? void 0 : _a.policies) || [];
+            }
         }
-        catch (error) {
-            console.error("Error fetching API config:", error);
-            return null;
+    }
+    async useAllocationRequest(payload) {
+        if (this.socketManager && this.socketManager.isConnected()) {
+            this.socketManager.sendAsync({
+                type: "use_allocation",
+                payload
+            }).catch((error) => {
+                console.error("[UsageFlow] Error sending finalization via WebSocket:", error);
+                throw error;
+            });
+        }
+    }
+    async allocationRequest(request, payload, metadata) {
+        if (this.socketManager && this.socketManager.isConnected()) {
+            try {
+                const allocationResponse = await this.socketManager.sendAsync({
+                    type: "request_for_allocation",
+                    payload
+                });
+                if (allocationResponse.type === 'error') {
+                    throw new Error(allocationResponse.message || allocationResponse.error);
+                }
+                if (allocationResponse.type === 'success') {
+                    request.usageflow.eventId = allocationResponse.payload.allocationId;
+                    request.usageflow.metadata = metadata;
+                    return;
+                }
+                else {
+                    throw new Error(allocationResponse.message || "Unknown error occurred");
+                }
+            }
+            catch (error) {
+                console.error("[UsageFlow] WebSocket allocation failed, falling back to HTTP:", error);
+                throw error;
+                // Fall through to HTTP request
+            }
         }
     }
     /**
@@ -152,6 +357,24 @@ class UsageFlowAPI {
         }
         return parts[1];
     }
+    /**
+     * Get header value from headers object (handles both Record and Headers types)
+     */
+    getHeaderValue(headers, key) {
+        // Check if it's a Headers object (from Fetch API) by checking for the 'get' method
+        if (headers && typeof headers.get === 'function') {
+            return headers.get(key) || null;
+        }
+        // Otherwise, treat it as a Record
+        const value = headers[key];
+        if (typeof value === 'string') {
+            return value;
+        }
+        if (Array.isArray(value) && value.length > 0) {
+            return value[0];
+        }
+        return null;
+    }
     /**
      * Decodes a JWT token without verifying the signature
      * @param token The JWT token to decode

package/dist/base.js.map

@@ -1 +1 @@
-{"version":3,"file":"base.js","sourceRoot":"","sources":["../src/base.ts"],"names":[],"mappings":";;;;;;AAAA,kDAA0B;AAC1B,mCAMiB;AAEjB,MAAsB,YAAY;IAAlC;QACc,WAAM,GAAkB,IAAI,CAAC;QAC7B,iBAAY,GAAW,0BAA0B,CAAC;QAElD,cAAS,GAA2B,IAAI,CAAC;QAC3C,yBAAoB,GAA0B,IAAI,CAAC;IAwM/D,CAAC;IAtMG;;OAEG;IACI,IAAI,CACP,MAAc,EACd,eAAuB,0BAA0B;QAEjD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC;IAChB,CAAC;IAEM,SAAS;QACZ,OAAO,IAAI,CAAC,MAAM,CAAC;IACvB,CAAC;IAEM,eAAe;QAClB,OAAO,IAAI,CAAC,YAAY,CAAC;IAC7B,CAAC;IAED;;OAEG;IACK,kBAAkB;QACtB,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5B,aAAa,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,CAAC,oBAAoB,GAAG,WAAW,CAAC,KAAK,IAAI,EAAE;YAC/C,IAAI,CAAC;gBACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;gBAC3C,IAAI,MAAM,EAAE,CAAC;oBACT,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC;gBAC5B,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,4BAA4B,EAAE,KAAK,CAAC,CAAC;YACvD,CAAC;QACL,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAI,CAAC,cAAc,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc;QACxB,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,IAAI,sBAAc,CAAC,yBAAyB,CAAC,CAAC;QACxD,CAAC;QAED,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,eAAK,CAAC,GAAG,CAC5B,GAAG,IAAI,CAAC,YAAY,gCAAgC,EACpD;gBACI,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,CAAC,MAAM,EAAE;gBACvC,OAAO,EAAE,KAAK;aACjB,CACJ,CAAC;YACF,OAAO,QAAQ,CAAC,IAAI,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,4BAA4B,EAAE,KAAK,CAAC,CAAC;YACnD,OAAO,IAAI,CAAC;QAChB,CAAC;IACL,CAAC;IAED;;OAEG;IACI,eAAe,CAAC,MAAe;QAClC,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACzB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC1B,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;YAChC,CAAC;YACD,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;QAC7C,CAAC;QACD,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED;;OAEG;IACI,eAAe,CAClB,MAAc,EACd,GAAW,EACX,YAAuB;QAEvB,IACI,YAAY,CAAC,MAAM,CAAC;YACpB,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,EAC1D,CAAC;YACC,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,IACI,YAAY,CAAC,GAAG,CAAC;YACjB,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EACpD,CAAC;YACC,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAED;;OAEG;IACI,kBAAkB,CACrB,MAAc,EACd,GAAW,EACX,SAAoB;QAEpB,IACI,SAAS,CAAC,MAAM,CAAC;YACjB,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,EACpD,CAAC;YACC,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,IAAI,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;YACjE,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAED;;OAEG;IACI,eAAe,CAAC,OAAgB;QACnC,MAAM,SAAS,GAAY,EAAE,CAAC;QAC9B,MAAM,QAAQ,GAAG,YAAY,CAAC;QAE9B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACjD,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;YACnC,IAAI,QAAQ,KAAK,eAAe,EAAE,CAAC;gBAC/B,IAAI,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC9B,SAAS,CAAC,GAAG,CAAC,GAAG,aAAa,CAAC;gBACnC,CAAC;qBAAM,CAAC;oBACJ,SAAS,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;gBAC5B,CAAC;YACL,CAAC;iBAAM,IAAI,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,SAAS,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;YAC5B,CAAC;iBAAM,CAAC;gBACJ,SAAS,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YAC3B,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAED;;OAEG;IACI,OAAO;QACV,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5B,aAAa,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACzC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QACrC,CAAC;IACL,CAAC;IAEM,kBAAkB,CAAC,UAAmB;QACzC,IAAI,CAAC,UAAU,EAAE,CAAC;YACd,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,+BAA+B;QAC/B,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,KAAK,QAAQ,EAAE,CAAC;YAC5D,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;IAED;;;;OAIG;IACI,mBAAmB,CAAC,KAAa;QACpC,IAAI,CAAC;YACD,6BAA6B;YAC7B,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC/B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACrB,OAAO,IAAI,CAAC;YAChB,CAAC;YAED,8BAA8B;YAC9B,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACzB,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YACjE,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,IAAI,CAAC;QAChB,CAAC;IACL,CAAC;CACJ;AA7MD,oCA6MC"}
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../src/base.ts"],"names":[],"mappings":";;;AAWA,qCAAiD;AAEjD,MAAsB,YAAY;IAAlC;QACc,WAAM,GAAkB,IAAI,CAAC;QAC7B,iBAAY,GAAW,0BAA0B,CAAC;QAClD,cAAS,GAAqC,SAAS,CAAC;QACxD,eAAU,GAAsB,EAAE,CAAC;QACrC,yBAAoB,GAA0B,IAAI,CAAC;QAC3D,kBAAa,GAAiC,IAAI,CAAC;QAC3C,kBAAa,GAAY,KAAK,CAAC;IAgc3C,CAAC;IA/bG;;OAEG;IACI,IAAI,CACP,MAAc,EACd,eAAuB,0BAA0B;QAEjD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,6BAA6B;QAC7B,IAAI,CAAC,aAAa,GAAG,IAAI,8BAAqB,CAAC,MAAM,CAAC,CAAC;QACvD,6BAA6B;QAC7B,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACzC,OAAO,CAAC,KAAK,CAAC,uDAAuD,EAAE,KAAK,CAAC,CAAC;QAClF,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE;YACT,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAGH,OAAO,IAAI,CAAC;IAChB,CAAC;IAEM,YAAY,CAAC,SAA2C;QAC3D,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC/B,CAAC;IAEM,SAAS;QACZ,OAAO,IAAI,CAAC,MAAM,CAAC;IACvB,CAAC;IAEM,eAAe;QAClB,OAAO,IAAI,CAAC,YAAY,CAAC;IAC7B,CAAC;IAED;;OAEG;IACK,kBAAkB;QACtB,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5B,aAAa,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,CAAC,oBAAoB,GAAG,WAAW,CAAC,KAAK,IAAI,EAAE;YAC/C,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAClC,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,8CAA8C;IAClD,CAAC;IAES,eAAe,CAAC,OAAyB;;QAE/C,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YAC/B,OAAO,CAAA,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,YAAY,0CAAE,GAAG,KAAI,OAAO,CAAC,GAAG,IAAI,EAAE,CAAC;QAC3D,CAAC;QAED,MAAM,YAAY,GAAG,CAAA,MAAA,OAAO,CAAC,KAAK,0CAAE,IAAI,MAAI,MAAA,MAAA,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAU,EAAE,EAAE;YACtF,6BAA6B;YAC7B,IAAI,CAAC,KAAK,CAAC,KAAK;gBAAE,OAAO,KAAK,CAAC;YAC/B,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC,IAAI,IAAI,OAAO,CAAC,GAAG,CAAC;YACrC,CAAC;YAED,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBACf,OAAO,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAC1C,CAAC;QAEL,CAAC,CAAC,0CAAE,KAAK,0CAAE,IAAI,CAAA,CAAA;QAEf,IAAI,YAAY,EAAE,CAAC;YACf,OAAO,YAAY,CAAC;QACxB,CAAC;QAED,oEAAoE;QACpE,IAAI,MAAA,OAAO,CAAC,KAAK,0CAAE,IAAI,EAAE,CAAC;YACtB,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;YACtC,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC;YACrC,oDAAoD;YACpD,MAAM,QAAQ,GAAG,OAAO,GAAG,SAAS,CAAC;YACrC,OAAO,QAAQ,IAAI,GAAG,CAAC;QAC3B,CAAC;QAED,gDAAgD;QAChD,IAAI,CAAC;YACD,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;YACxB,MAAM,MAAM,GAAI,GAAW,CAAC,OAAO,CAAC;YACpC,IAAI,MAAM,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;gBAC5C,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,KAAI,MAAA,OAAO,CAAC,GAAG,0CAAE,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAA,IAAI,EAAE,CAAC;gBAE9D,iDAAiD;gBACjD,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;oBAC/B,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;wBACd,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;wBAC1B,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC;wBAC7B,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;wBAE1E,mDAAmD;wBACnD,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;4BAC9D,uDAAuD;4BACvD,MAAM,OAAO,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC,GAAG,SAAS,CAAC;4BACpD,2EAA2E;4BAC3E,IAAI,OAAO,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gCAC3D,+CAA+C;gCAC/C,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gCAC/C,MAAM,cAAc,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC;gCACjF,IAAI,cAAc,EAAE,CAAC;oCACjB,OAAO,OAAO,CAAC;gCACnB,CAAC;4BACL,CAAC;iCAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gCAClC,qEAAqE;gCACrE,IAAI,IAAI,KAAK,OAAO,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;oCACzC,OAAO,OAAO,CAAC;gCACnB,CAAC;4BACL,CAAC;wBACL,CAAC;oBACL,CAAC;yBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;wBACvE,4CAA4C;wBAC5C,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM;6BAChC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;6BACnB,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;6BACxB,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC;6BACrB,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;6BAClB,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;6BAClB,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;wBAExB,KAAK,MAAM,QAAQ,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;4BACxC,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;gCACjB,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;gCAC7B,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC;gCAC7B,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;gCAE1E,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;oCAC9D,MAAM,QAAQ,GAAG,SAAS,GAAG,SAAS,CAAC;oCACvC,kDAAkD;oCAClD,IAAI,OAAO,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wCAC3D,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;wCAC/C,MAAM,cAAc,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC;wCACjF,IAAI,cAAc,EAAE,CAAC;4CACjB,OAAO,QAAQ,CAAC;wCACpB,CAAC;oCACL,CAAC;yCAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;wCAClC,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,KAAI,MAAA,OAAO,CAAC,GAAG,0CAAE,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAA,IAAI,EAAE,CAAC;wCACrE,IAAI,WAAW,KAAK,QAAQ,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;4CACxD,OAAO,QAAQ,CAAC;wCACpB,CAAC;oCACL,CAAC;gCACL,CAAC;4BACL,CAAC;wBACL,CAAC;oBACL,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,oCAAoC;YACpC,OAAO,CAAC,KAAK,CAAC,wDAAwD,EAAE,KAAK,CAAC,CAAC;QACnF,CAAC;QAED,qDAAqD;QACrD,IAAI,OAAO,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3D,IAAI,IAAI,GAAG,OAAO,CAAC,IAAI,KAAI,MAAA,OAAO,CAAC,GAAG,0CAAE,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAA,IAAI,GAAG,CAAC;YAC7D,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;YAEtC,0EAA0E;YAC1E,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YACrC,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAEpD,2DAA2D;YAC3D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3C,KAAK,MAAM,CAAC,SAAS,EAAE,UAAU,CAAC,IAAI,YAAY,EAAE,CAAC;oBACjD,IAAI,YAAY,CAAC,CAAC,CAAC,KAAK,UAAU,EAAE,CAAC;wBACjC,YAAY,CAAC,CAAC,CAAC,GAAG,IAAI,SAAS,EAAE,CAAC;wBAClC,MAAM,CAAC,gCAAgC;oBAC3C,CAAC;gBACL,CAAC;YACL,CAAC;YAED,MAAM,OAAO,GAAG,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACjD,OAAO,OAAO,CAAC;QACnB,CAAC;QAED,+CAA+C;QAC/C,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,KAAI,MAAA,OAAO,CAAC,GAAG,0CAAE,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAA,IAAI,GAAG,CAAC;QAC/D,OAAO,OAAO,GAAG,IAAI,IAAI,GAAG,CAAC;IACjC,CAAC;IAEM,aAAa,CAAC,OAAyB;;QAC1C,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAA;QAE/B,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAClB,OAAO,GAAG,MAAM,IAAI,GAAG,EAAE,CAAC;QAC9B,CAAC;QAGD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC3B,MAAM,SAAS,GAAG,MAAM,CAAC,iBAAkB,CAAC;YAC5C,MAAM,QAAQ,GAAG,MAAM,CAAC,qBAAqB,CAAC;YAE9C,QAAQ,QAAQ,EAAE,CAAC;gBACf,KAAK,aAAa;oBACd,IAAI,MAAA,OAAO,CAAC,MAAM,0CAAG,SAAS,CAAC,EAAE,CAAC;wBAC9B,OAAO,GAAG,MAAM,IAAI,GAAG,IAAI,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC3D,CAAC;oBACD,MAAM;gBACV,KAAK,cAAc;oBACf,IAAI,MAAA,OAAO,CAAC,KAAK,0CAAG,SAAS,CAAC,EAAE,CAAC;wBAC7B,OAAO,GAAG,MAAM,IAAI,GAAG,IAAI,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC1D,CAAC;oBACD,MAAM;gBACV,KAAK,MAAM;oBACP,IAAI,MAAA,OAAO,CAAC,IAAI,0CAAG,SAAS,CAAC,EAAE,CAAC;wBAC5B,OAAO,GAAG,MAAM,IAAI,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;oBACzD,CAAC;oBACD,MAAM;gBACV,KAAK,cAAc;oBACf,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;oBACzE,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,UAAU,IAAI,SAAS,CAAC,CAAC;oBAC/D,IAAI,KAAK,EAAE,CAAC;wBACR,MAAM,MAAM,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;wBAC/C,IAAI,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAG,SAAS,CAAC,EAAE,CAAC;4BACtB,OAAO,GAAG,MAAM,IAAI,GAAG,IAAI,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;wBACnD,CAAC;oBACL,CAAC;oBACD,MAAM;gBAEV,KAAK,SAAS,CAAC,CAAC,CAAC;oBACb,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;oBACpE,IAAI,WAAW,EAAE,CAAC;wBACd,OAAO,GAAG,MAAM,IAAI,GAAG,IAAI,WAAW,EAAE,CAAC;oBAC7C,CAAC;oBACD,MAAM;gBACV,CAAC;YACL,CAAC;QACL,CAAC;QAED,OAAO,GAAG,MAAM,IAAI,GAAG,EAAE,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,gBAAgB;;QAC1B,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,EAAE,CAAC;YACzD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAiD;gBAChG,IAAI,EAAE,0BAA0B;gBAChC,OAAO,EAAE,IAAI;aAChB,CAAC,CAAC;YACH,IAAI,QAAQ,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC9B,IAAI,CAAC,UAAU,GAAG,CAAA,MAAA,QAAQ,CAAC,OAAO,0CAAE,QAAQ,KAAI,EAAE,CAAC;YACvD,CAAC;QACL,CAAC;IACL,CAAC;IAED,KAAK,CAAC,oBAAoB,CACtB,OAA6B;QAE7B,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,EAAE,CAAC;YACzD,IAAI,CAAC,aAAa,CAAC,SAAS,CAAM;gBAC9B,IAAI,EAAE,gBAAgB;gBACtB,OAAO;aACV,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACf,OAAO,CAAC,KAAK,CAAC,uDAAuD,EAAE,KAAK,CAAC,CAAC;gBAC9E,MAAM,KAAK,CAAC;YAChB,CAAC,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAGD,KAAK,CAAC,iBAAiB,CACnB,OAAyB,EACzB,OAA6B,EAC7B,QAAyB;QAEzB,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,EAAE,CAAC;YACzD,IAAI,CAAC;gBACD,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAM;oBAC/D,IAAI,EAAE,wBAAwB;oBAC9B,OAAO;iBACV,CAAC,CAAC;gBAEH,IAAI,kBAAkB,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBACtC,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,OAAO,IAAI,kBAAkB,CAAC,KAAK,CAAC,CAAC;gBAC5E,CAAC;gBACD,IAAI,kBAAkB,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;oBACxC,OAAO,CAAC,SAAU,CAAC,OAAO,GAAG,kBAAkB,CAAC,OAAO,CAAC,YAAY,CAAC;oBACrE,OAAO,CAAC,SAAU,CAAC,QAAQ,GAAG,QAAQ,CAAC;oBACvC,OAAO;gBACX,CAAC;qBAAM,CAAC;oBACJ,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,OAAO,IAAI,wBAAwB,CAAC,CAAC;gBAC5E,CAAC;YACL,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBAClB,OAAO,CAAC,KAAK,CAAC,gEAAgE,EAAE,KAAK,CAAC,CAAC;gBACvF,MAAM,KAAK,CAAC;gBACZ,+BAA+B;YACnC,CAAC;QACL,CAAC;IACL,CAAC;IAED;;OAEG;IACI,eAAe,CAAC,MAAe;QAClC,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACzB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC1B,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;YAChC,CAAC;YACD,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;QAC7C,CAAC;QACD,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED;;OAEG;IACI,eAAe,CAClB,MAAc,EACd,GAAW,EACX,YAAuB;QAEvB,IACI,YAAY,CAAC,MAAM,CAAC;YACpB,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,EAC1D,CAAC;YACC,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,IACI,YAAY,CAAC,GAAG,CAAC;YACjB,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EACpD,CAAC;YACC,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAED;;OAEG;IACI,kBAAkB,CACrB,MAAc,EACd,GAAW,EACX,SAAoB;QAEpB,IACI,SAAS,CAAC,MAAM,CAAC;YACjB,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,EACpD,CAAC;YACC,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,IAAI,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;YACjE,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAED;;OAEG;IACI,eAAe,CAAC,OAAgB;QACnC,MAAM,SAAS,GAAY,EAAE,CAAC;QAC9B,MAAM,QAAQ,GAAG,YAAY,CAAC;QAE9B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACjD,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;YACnC,IAAI,QAAQ,KAAK,eAAe,EAAE,CAAC;gBAC/B,IAAI,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC9B,SAAS,CAAC,GAAG,CAAC,GAAG,aAAa,CAAC;gBACnC,CAAC;qBAAM,CAAC;oBACJ,SAAS,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;gBAC5B,CAAC;YACL,CAAC;iBAAM,IAAI,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,SAAS,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;YAC5B,CAAC;iBAAM,CAAC;gBACJ,SAAS,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YAC3B,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAED;;OAEG;IACI,OAAO;QACV,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5B,aAAa,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACzC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QACrC,CAAC;IACL,CAAC;IAEM,kBAAkB,CAAC,UAAmB;QACzC,IAAI,CAAC,UAAU,EAAE,CAAC;YACd,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,+BAA+B;QAC/B,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,KAAK,QAAQ,EAAE,CAAC;YAC5D,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,OAAgE,EAAE,GAAW;QAChG,mFAAmF;QACnF,IAAI,OAAO,IAAI,OAAQ,OAAe,CAAC,GAAG,KAAK,UAAU,EAAE,CAAC;YACxD,OAAQ,OAAe,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC;QAC7C,CAAC;QACD,kCAAkC;QAClC,MAAM,KAAK,GAAI,OAAyD,CAAC,GAAG,CAAC,CAAC;QAC9E,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC5B,OAAO,KAAK,CAAC;QACjB,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3C,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;;;OAIG;IACI,mBAAmB,CAAC,KAAa;QACpC,IAAI,CAAC;YACD,6BAA6B;YAC7B,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC/B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACrB,OAAO,IAAI,CAAC;YAChB,CAAC;YAED,8BAA8B;YAC9B,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACzB,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YACjE,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,IAAI,CAAC;QAChB,CAAC;IACL,CAAC;CACJ;AAvcD,oCAucC"}

package/dist/index.d.ts

@@ -1,2 +1,3 @@
 export * from './types';
 export * from './base';
+export * from './socket';

package/dist/index.js

@@ -16,4 +16,5 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./types"), exports);
 __exportStar(require("./base"), exports);
+__exportStar(require("./socket"), exports);
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,0CAAwB;AACxB,yCAAuB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,0CAAwB;AACxB,yCAAuB;AACvB,2CAAyB"}

package/dist/socket.d.ts

@@ -0,0 +1,21 @@
+import WebSocket from "ws";
+import { UsageFlowSocketMessage, UsageFlowSocketResponse } from "./types";
+export declare class UsageFlowSocketManger {
+    private apiKey;
+    private ws;
+    private wsUrl;
+    private wsConnected;
+    constructor(apiKey: string);
+    establishWebSocketConnection(): Promise<void>;
+    connect(): Promise<void>;
+    sendAsync<T>(payload: UsageFlowSocketMessage): Promise<UsageFlowSocketResponse<T>>;
+    send(payload: UsageFlowSocketMessage): void;
+    close(): void;
+    isConnected(): boolean;
+    getWs(): WebSocket | null;
+    private _asyncSend;
+    /**
+     * Clean up resources including WebSocket connection
+     */
+    destroy(): void;
+}

package/dist/socket.js

@@ -0,0 +1,216 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UsageFlowSocketManger = void 0;
+const ws_1 = __importDefault(require("ws"));
+class UsageFlowSocketManger {
+    constructor(apiKey) {
+        this.apiKey = apiKey;
+        this.ws = null;
+        this.wsUrl = "wss://api.usageflow.io/ws";
+        this.wsConnected = false;
+        this.apiKey = apiKey;
+        // Connection is explicit - call connect() when ready
+        // Do NOT auto-connect in constructor to avoid immediate execution
+    }
+    async establishWebSocketConnection() {
+        if (this.ws && this.ws.readyState === ws_1.default.OPEN) {
+            this.wsConnected = true;
+            return;
+        }
+        if (!this.apiKey) {
+            console.warn("[UsageFlow] Cannot establish WebSocket connection: API key not initialized");
+            return;
+        }
+        return new Promise((resolve, reject) => {
+            try {
+                if (!this.apiKey) {
+                    reject(new Error("API key not available"));
+                    return;
+                }
+                const headers = {
+                    "x-usage-key": this.apiKey,
+                };
+                this.ws = new ws_1.default(this.wsUrl, {
+                    headers,
+                    family: 4 // This forces IPv4
+                });
+                this.ws.on("open", () => {
+                    this.wsConnected = true;
+                    resolve();
+                });
+                this.ws.on("error", (error) => {
+                    console.error("[UsageFlow] WebSocket error:", error);
+                    this.wsConnected = false;
+                    reject(error);
+                });
+                this.ws.on("close", () => {
+                    this.wsConnected = false;
+                    // Attempt to reconnect after a delay
+                    setTimeout(() => {
+                        if (this.apiKey) {
+                            this.establishWebSocketConnection().catch(console.error);
+                        }
+                    }, 5000);
+                });
+                this.ws.on("message", (data) => {
+                    // Message handling done in _asyncSend
+                });
+            }
+            catch (error) {
+                console.error("[UsageFlow] Error creating WebSocket connection:", error);
+                this.wsConnected = false;
+                reject(error);
+            }
+        });
+    }
+    async connect() {
+        if (!this.apiKey) {
+            console.warn("[UsageFlow] Cannot establish WebSocket connection: API key not initialized");
+            return;
+        }
+        return new Promise((resolve, reject) => {
+            try {
+                const headers = {
+                    "x-usage-key": this.apiKey,
+                };
+                this.ws = new ws_1.default(this.wsUrl, {
+                    headers,
+                    family: 4 // This forces IPv4
+                });
+                this.ws.on("open", () => {
+                    console.debug("[UsageFlow] WebSocket connection established");
+                    this.wsConnected = true;
+                    resolve();
+                });
+                this.ws.on("error", (error) => {
+                    console.debug("[UsageFlow] WebSocket error:", error);
+                    this.wsConnected = false;
+                    reject(error);
+                });
+                this.ws.on("close", () => {
+                    console.debug("[UsageFlow] WebSocket connection closed");
+                    this.wsConnected = false;
+                    // Attempt to reconnect after a delay
+                    setTimeout(() => {
+                        if (this.apiKey) {
+                            this.connect().catch(console.error);
+                        }
+                    }, 5000);
+                });
+                this.ws.on("message", (data) => {
+                    console.debug("[UsageFlow] WebSocket message received:", data.toString());
+                });
+            }
+            catch (error) {
+                console.debug("[UsageFlow] Error creating WebSocket connection:", error);
+                this.wsConnected = false;
+                reject(error);
+            }
+        });
+    }
+    async sendAsync(payload) {
+        if (!this.ws || !this.wsConnected) {
+            throw new Error("WebSocket not connected");
+        }
+        const socketResponse = await this._asyncSend(payload);
+        return socketResponse;
+    }
+    send(payload) {
+        if (!this.ws || !this.wsConnected) {
+            throw new Error("WebSocket not connected");
+        }
+        this.ws.send(JSON.stringify(payload));
+    }
+    close() {
+        if (this.ws) {
+            this.ws.close();
+        }
+        this.ws = null;
+        this.wsConnected = false;
+    }
+    isConnected() {
+        return this.wsConnected;
+    }
+    getWs() {
+        return this.ws;
+    }
+    async _asyncSend(payload) {
+        return new Promise((resolve, reject) => {
+            if (!this.ws || !this.wsConnected) {
+                reject(new Error("WebSocket not connected"));
+                return;
+            }
+            const id = Math.random().toString(36).slice(2) + Date.now().toString(36);
+            const message = { id, ...payload };
+            let isResolved = false;
+            // Set up timeout to prevent handlers from accumulating forever
+            const timeout = setTimeout(() => {
+                if (!isResolved) {
+                    isResolved = true;
+                    if (this.ws) {
+                        this.ws.off('message', handleMessage);
+                    }
+                    reject(new Error(`WebSocket request timeout for id: ${id}`));
+                }
+            }, 30000); // 30 second timeout
+            // Wrap resolve to clear timeout
+            const wrappedResolve = (value) => {
+                if (!isResolved) {
+                    clearTimeout(timeout);
+                    isResolved = true;
+                    resolve(value);
+                }
+            };
+            const handleMessage = (data) => {
+                if (isResolved)
+                    return; // Prevent multiple processing
+                try {
+                    const response = JSON.parse(data.toString());
+                    // Use strict equality and check if this response matches our request
+                    if (response.id === id || response.replyTo === id) {
+                        console.debug("[UsageFlow] WebSocket message received for id:", id, response);
+                        // Remove handler immediately before resolving
+                        if (this.ws) {
+                            this.ws.off('message', handleMessage);
+                        }
+                        wrappedResolve(response);
+                    }
+                }
+                catch (error) {
+                    // Ignore parse errors for messages that don't match
+                }
+            };
+            this.ws.on('message', handleMessage);
+            this.ws.send(JSON.stringify(message), (err) => {
+                if (err) {
+                    clearTimeout(timeout);
+                    if (!isResolved) {
+                        isResolved = true;
+                        if (this.ws) {
+                            this.ws.off('message', handleMessage);
+                        }
+                        reject(err);
+                    }
+                }
+            });
+        });
+    }
+    /**
+     * Clean up resources including WebSocket connection
+     */
+    destroy() {
+        if (this.ws) {
+            this.ws.removeAllListeners();
+            if (this.ws.readyState === ws_1.default.OPEN || this.ws.readyState === ws_1.default.CONNECTING) {
+                this.ws.close();
+            }
+            this.ws = null;
+            this.wsConnected = false;
+        }
+    }
+}
+exports.UsageFlowSocketManger = UsageFlowSocketManger;
+//# sourceMappingURL=socket.js.map