@humanode-hyperlane/rebalancer 0.0.0-stage → 27.2.2

package/dist/core/RebalancerOrchestrator.js

@@ -0,0 +1,154 @@
+import { getStrategyChainNames } from '../config/types.js';
+import { isInventoryRoute, isMovableCollateralRoute, } from '../interfaces/IStrategy.js';
+import { getRawBalances } from '../utils/balanceUtils.js';
+export class RebalancerOrchestrator {
+    strategy;
+    actionTracker;
+    inflightContextAdapter;
+    rebalancerConfig;
+    logger;
+    rebalancersByType;
+    externalBridgeRegistry;
+    metrics;
+    constructor(deps) {
+        this.strategy = deps.strategy;
+        this.actionTracker = deps.actionTracker;
+        this.inflightContextAdapter = deps.inflightContextAdapter;
+        this.rebalancerConfig = deps.rebalancerConfig;
+        this.logger = deps.logger;
+        this.rebalancersByType = new Map(deps.rebalancers.map((r) => [r.rebalancerType, r]));
+        this.externalBridgeRegistry = deps.externalBridgeRegistry;
+        this.metrics = deps.metrics;
+    }
+    /**
+     * Execute a single rebalancing cycle.
+     * Processes monitor event, evaluates strategy, and executes routes.
+     */
+    async executeCycle(event) {
+        this.logger.info('Polling cycle started');
+        const { metrics } = this;
+        if (metrics) {
+            await Promise.all(event.tokensInfo.map((tokenInfo) => metrics.processToken(tokenInfo)));
+        }
+        await this.syncActionTracker(event.confirmedBlockTags);
+        const rawBalances = getRawBalances(getStrategyChainNames(this.rebalancerConfig.strategyConfig), event, this.logger);
+        this.logger.info({
+            balances: Object.entries(rawBalances).map(([chain, balance]) => ({
+                chain,
+                balance: balance.toString(),
+            })),
+        }, 'Router balances');
+        // Get inflight context for strategy decision-making
+        const inflightContext = await this.getInflightContext();
+        const strategyRoutes = this.strategy.getRebalancingRoutes(rawBalances, inflightContext);
+        let executedCount = 0;
+        let failedCount = 0;
+        if (strategyRoutes.length > 0) {
+            this.logger.info({
+                routes: strategyRoutes.map((r) => ({
+                    from: r.origin,
+                    to: r.destination,
+                    amount: r.amount.toString(),
+                })),
+            }, 'Routes proposed');
+            const results = await this.executeWithTracking(strategyRoutes, event);
+            executedCount = results.executedCount;
+            failedCount = results.failedCount;
+        }
+        else {
+            this.logger.info('No rebalancing needed');
+        }
+        const inventoryRebalancer = this.rebalancersByType.get('inventory');
+        if (inventoryRebalancer && strategyRoutes.length === 0) {
+            await this.executeRoutes([], inventoryRebalancer, event);
+        }
+        this.logger.info('Polling cycle completed');
+        return {
+            balances: rawBalances,
+            proposedRoutes: strategyRoutes,
+            executedCount,
+            failedCount,
+        };
+    }
+    /**
+     * Sync action tracker with current chain state.
+     */
+    async syncActionTracker(confirmedBlockTags) {
+        try {
+            await Promise.all([
+                this.actionTracker.syncTransfers(confirmedBlockTags),
+                this.actionTracker.syncRebalanceIntents(),
+                this.actionTracker.syncRebalanceActions(confirmedBlockTags),
+            ]);
+            // Sync inventory movement actions via external bridge API
+            if (this.externalBridgeRegistry) {
+                await this.actionTracker.syncInventoryMovementActions(this.externalBridgeRegistry);
+            }
+            await this.actionTracker.logStoreContents();
+        }
+        catch (error) {
+            this.logger.warn({ error }, 'ActionTracker sync failed, using stale data');
+        }
+    }
+    /**
+     * Get inflight context for strategy decision-making
+     */
+    async getInflightContext() {
+        return this.inflightContextAdapter.getInflightContext();
+    }
+    async executeWithTracking(routes, event) {
+        const movableCollateral = routes.filter(isMovableCollateralRoute);
+        const inventory = routes.filter(isInventoryRoute);
+        let executedCount = 0;
+        let failedCount = 0;
+        const movableCollateralRebalancer = this.rebalancersByType.get('movableCollateral');
+        if (movableCollateral.length > 0 && movableCollateralRebalancer) {
+            const results = await this.executeRoutes(movableCollateral, movableCollateralRebalancer, event);
+            executedCount = results.filter((r) => r.success).length;
+            failedCount = results.filter((r) => !r.success).length;
+        }
+        const inventoryRebalancer = this.rebalancersByType.get('inventory');
+        if (inventory.length > 0 && inventoryRebalancer) {
+            await this.executeRoutes(inventory, inventoryRebalancer, event);
+        }
+        return { executedCount, failedCount };
+    }
+    async executeRoutes(routes, rebalancer, event) {
+        if (rebalancer.rebalancerType === 'inventory' && event.inventoryBalances) {
+            rebalancer.setInventoryBalances(event.inventoryBalances);
+        }
+        try {
+            const results = await rebalancer.rebalance(routes);
+            const successful = results.filter((r) => r.success);
+            const failed = results.filter((r) => !r.success);
+            if (successful.length > 0) {
+                if (rebalancer.rebalancerType === 'movableCollateral') {
+                    this.metrics?.recordRebalancerSuccess();
+                }
+                this.logger.info({ count: successful.length, type: rebalancer.rebalancerType }, 'Rebalancer completed successfully');
+            }
+            if (failed.length > 0) {
+                if (rebalancer.rebalancerType === 'movableCollateral') {
+                    this.metrics?.recordRebalancerFailure();
+                }
+                this.logger.warn({
+                    count: failed.length,
+                    type: rebalancer.rebalancerType,
+                    errors: failed.map((r) => ({
+                        route: `${r.route.origin} -> ${r.route.destination}`,
+                        error: r.error,
+                    })),
+                }, 'Some routes failed');
+            }
+            return results;
+        }
+        catch (error) {
+            if (rebalancer.rebalancerType === 'movableCollateral') {
+                this.metrics?.recordRebalancerFailure();
+            }
+            this.logger.error({ error, type: rebalancer.rebalancerType }, 'Error while executing routes');
+            return [];
+        }
+    }
+}
+//# sourceMappingURL=RebalancerOrchestrator.js.map

package/dist/core/RebalancerOrchestrator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"RebalancerOrchestrator.js","sourceRoot":"","sources":["../../src/core/RebalancerOrchestrator.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAY3D,OAAO,EACL,gBAAgB,EAChB,wBAAwB,GACzB,MAAM,4BAA4B,CAAC;AAIpC,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AA4B1D,MAAM,OAAO,sBAAsB;IAChB,QAAQ,CAAY;IACpB,aAAa,CAAiB;IAC9B,sBAAsB,CAAyB;IAC/C,gBAAgB,CAAmB;IACnC,MAAM,CAAS;IACf,iBAAiB,CAAmC;IACpD,sBAAsB,CAAmC;IACzD,OAAO,CAAW;IAEnC,YAAY,IAAgC;QAC1C,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9B,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,sBAAsB,CAAC;QAC1D,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QAC9C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,iBAAiB,GAAG,IAAI,GAAG,CAC9B,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CACnD,CAAC;QACF,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,sBAAsB,CAAC;QAC1D,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;IAC9B,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,YAAY,CAAC,KAAmB;QACpC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAE1C,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;QACzB,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CACrE,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;QAEvD,MAAM,WAAW,GAAG,cAAc,CAChC,qBAAqB,CAAC,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,EAC3D,KAAK,EACL,IAAI,CAAC,MAAM,CACZ,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CACd;YACE,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC/D,KAAK;gBACL,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;aAC5B,CAAC,CAAC;SACJ,EACD,iBAAiB,CAClB,CAAC;QAEF,oDAAoD;QACpD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAExD,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CACvD,WAAW,EACX,eAAe,CAChB,CAAC;QAEF,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,IAAI,WAAW,GAAG,CAAC,CAAC;QAEpB,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CACd;gBACE,MAAM,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBACjC,IAAI,EAAE,CAAC,CAAC,MAAM;oBACd,EAAE,EAAE,CAAC,CAAC,WAAW;oBACjB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE;iBAC5B,CAAC,CAAC;aACJ,EACD,iBAAiB,CAClB,CAAC;YAEF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;YACtE,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;YACtC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACpC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC5C,CAAC;QAED,MAAM,mBAAmB,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACpE,IAAI,mBAAmB,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvD,MAAM,IAAI,CAAC,aAAa,CAAC,EAAE,EAAE,mBAAmB,EAAE,KAAK,CAAC,CAAC;QAC3D,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QAE5C,OAAO;YACL,QAAQ,EAAE,WAAW;YACrB,cAAc,EAAE,cAAc;YAC9B,aAAa;YACb,WAAW;SACZ,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB,CAC7B,kBAAuC;QAEvC,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChB,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,kBAAkB,CAAC;gBACpD,IAAI,CAAC,aAAa,CAAC,oBAAoB,EAAE;gBACzC,IAAI,CAAC,aAAa,CAAC,oBAAoB,CAAC,kBAAkB,CAAC;aAC5D,CAAC,CAAC;YAEH,0DAA0D;YAC1D,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;gBAChC,MAAM,IAAI,CAAC,aAAa,CAAC,4BAA4B,CACnD,IAAI,CAAC,sBAAsB,CAC5B,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,CAAC,aAAa,CAAC,gBAAgB,EAAE,CAAC;QAC9C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,EAAE,KAAK,EAAE,EACT,6CAA6C,CAC9C,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB;QAC9B,OAAO,IAAI,CAAC,sBAAsB,CAAC,kBAAkB,EAAE,CAAC;IAC1D,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,MAAuB,EACvB,KAAmB;QAEnB,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC;QAClE,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAElD,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,IAAI,WAAW,GAAG,CAAC,CAAC;QAEpB,MAAM,2BAA2B,GAC/B,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QAClD,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,IAAI,2BAA2B,EAAE,CAAC;YAChE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CACtC,iBAAiB,EACjB,2BAA2B,EAC3B,KAAK,CACN,CAAC;YACF,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;YACxD,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;QACzD,CAAC;QAED,MAAM,mBAAmB,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACpE,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,mBAAmB,EAAE,CAAC;YAChD,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,mBAAmB,EAAE,KAAK,CAAC,CAAC;QAClE,CAAC;QAED,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,CAAC;IACxC,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,MAAuB,EACvB,UAAuB,EACvB,KAAmB;QAEnB,IAAI,UAAU,CAAC,cAAc,KAAK,WAAW,IAAI,KAAK,CAAC,iBAAiB,EAAE,CAAC;YACxE,UAAkC,CAAC,oBAAoB,CACtD,KAAK,CAAC,iBAAiB,CACxB,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAEnD,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YAEjD,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,IAAI,UAAU,CAAC,cAAc,KAAK,mBAAmB,EAAE,CAAC;oBACtD,IAAI,CAAC,OAAO,EAAE,uBAAuB,EAAE,CAAC;gBAC1C,CAAC;gBACD,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,EAAE,KAAK,EAAE,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC,cAAc,EAAE,EAC7D,mCAAmC,CACpC,CAAC;YACJ,CAAC;YAED,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACtB,IAAI,UAAU,CAAC,cAAc,KAAK,mBAAmB,EAAE,CAAC;oBACtD,IAAI,CAAC,OAAO,EAAE,uBAAuB,EAAE,CAAC;gBAC1C,CAAC;gBACD,IAAI,CAAC,MAAM,CAAC,IAAI,CACd;oBACE,KAAK,EAAE,MAAM,CAAC,MAAM;oBACpB,IAAI,EAAE,UAAU,CAAC,cAAc;oBAC/B,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBACzB,KAAK,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,OAAO,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE;wBACpD,KAAK,EAAE,CAAC,CAAC,KAAK;qBACf,CAAC,CAAC;iBACJ,EACD,oBAAoB,CACrB,CAAC;YACJ,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,UAAU,CAAC,cAAc,KAAK,mBAAmB,EAAE,CAAC;gBACtD,IAAI,CAAC,OAAO,EAAE,uBAAuB,EAAE,CAAC;YAC1C,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,CAAC,cAAc,EAAE,EAC1C,8BAA8B,CAC/B,CAAC;YACF,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;CACF"}

package/dist/core/RebalancerOrchestrator.test.d.ts

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

package/dist/core/RebalancerOrchestrator.test.d.ts.map

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