@humanode-hyperlane/rebalancer 0.0.0-stage → 27.2.2

package/dist/strategy/StrategyFactory.js

@@ -0,0 +1,79 @@
+import { ExecutionType, RebalancerStrategyOptions, } from '../config/types.js';
+import { CollateralDeficitStrategy } from './CollateralDeficitStrategy.js';
+import { CompositeStrategy } from './CompositeStrategy.js';
+import { MinAmountStrategy } from './MinAmountStrategy.js';
+import { WeightedStrategy } from './WeightedStrategy.js';
+export class StrategyFactory {
+    /**
+     * Creates a strategy from an array of strategy configs.
+     * - Single strategy (array with 1 element): Creates that strategy directly
+     * - Multiple strategies (array with 2+ elements): Creates CompositeStrategy
+     *
+     * @param strategyConfigs Array of strategy configurations (always array format)
+     * @param tokensByChainName A map of chain->token to ease the lookup of token by chain
+     * @param initialTotalCollateral The initial total collateral of the rebalancer
+     * @param logger The logger to use for the strategy
+     * @param metrics The metrics to use for the strategy
+     * @param minAmountsByChain Optional minimum amounts per chain for filtering routes
+     * @returns A concrete strategy implementation
+     */
+    static createStrategy(strategyConfigs, tokensByChainName, initialTotalCollateral, logger, metrics, minAmountsByChain) {
+        if (strategyConfigs.length === 0) {
+            throw new Error('At least one strategy must be configured');
+        }
+        // Single strategy - create directly without CompositeStrategy wrapper
+        if (strategyConfigs.length === 1) {
+            return this.createSingleStrategy(strategyConfigs[0], tokensByChainName, initialTotalCollateral, logger, metrics, minAmountsByChain);
+        }
+        // Multiple strategies - create CompositeStrategy
+        const subStrategies = strategyConfigs.map((config) => this.createSingleStrategy(config, tokensByChainName, initialTotalCollateral, logger, metrics, minAmountsByChain));
+        return new CompositeStrategy(subStrategies, logger);
+    }
+    /**
+     * Create a single strategy from config.
+     */
+    static createSingleStrategy(strategyConfig, tokensByChainName, initialTotalCollateral, logger, metrics, _minAmountsByChain) {
+        const bridgeConfigs = this.extractBridgeConfigs(strategyConfig);
+        switch (strategyConfig.rebalanceStrategy) {
+            case RebalancerStrategyOptions.Weighted: {
+                return new WeightedStrategy(strategyConfig.chains, logger, bridgeConfigs, metrics, tokensByChainName);
+            }
+            case RebalancerStrategyOptions.MinAmount: {
+                return new MinAmountStrategy(strategyConfig.chains, tokensByChainName, initialTotalCollateral, logger, bridgeConfigs, metrics);
+            }
+            case RebalancerStrategyOptions.CollateralDeficit: {
+                return new CollateralDeficitStrategy(strategyConfig.chains, tokensByChainName, logger, bridgeConfigs, metrics);
+            }
+            default: {
+                throw new Error('Unsupported strategy type');
+            }
+        }
+    }
+    static extractBridgeConfigs(strategyConfig) {
+        const bridgeConfigs = {};
+        for (const [chain, config] of Object.entries(strategyConfig.chains)) {
+            const baseConfig = {
+                bridgeMinAcceptedAmount: config.bridgeMinAcceptedAmount ?? 0,
+            };
+            const executionType = config.executionType ?? ExecutionType.MovableCollateral;
+            if (executionType === ExecutionType.Inventory) {
+                bridgeConfigs[chain] = {
+                    ...baseConfig,
+                    executionType: 'inventory',
+                    externalBridge: config.externalBridge, // Validated by config schema
+                    override: config.override,
+                };
+            }
+            else {
+                bridgeConfigs[chain] = {
+                    ...baseConfig,
+                    executionType: 'movableCollateral',
+                    bridge: config.bridge, // Validated by config schema
+                    override: config.override,
+                };
+            }
+        }
+        return bridgeConfigs;
+    }
+}
+//# sourceMappingURL=StrategyFactory.js.map

package/dist/strategy/StrategyFactory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"StrategyFactory.js","sourceRoot":"","sources":["../../src/strategy/StrategyFactory.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,aAAa,EACb,yBAAyB,GAE1B,MAAM,oBAAoB,CAAC;AAS5B,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAC3E,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzD,MAAM,OAAO,eAAe;IAC1B;;;;;;;;;;;;OAYG;IACH,MAAM,CAAC,cAAc,CACnB,eAAiC,EACjC,iBAAkC,EAClC,sBAA8B,EAC9B,MAAc,EACd,OAAiB,EACjB,iBAAoC;QAEpC,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC9D,CAAC;QAED,sEAAsE;QACtE,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO,IAAI,CAAC,oBAAoB,CAC9B,eAAe,CAAC,CAAC,CAAC,EAClB,iBAAiB,EACjB,sBAAsB,EACtB,MAAM,EACN,OAAO,EACP,iBAAiB,CAClB,CAAC;QACJ,CAAC;QAED,iDAAiD;QACjD,MAAM,aAAa,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACnD,IAAI,CAAC,oBAAoB,CACvB,MAAM,EACN,iBAAiB,EACjB,sBAAsB,EACtB,MAAM,EACN,OAAO,EACP,iBAAiB,CAClB,CACF,CAAC;QACF,OAAO,IAAI,iBAAiB,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IACtD,CAAC;IAED;;OAEG;IACK,MAAM,CAAC,oBAAoB,CACjC,cAA8B,EAC9B,iBAAkC,EAClC,sBAA8B,EAC9B,MAAc,EACd,OAAiB,EACjB,kBAAqC;QAErC,MAAM,aAAa,GAAG,IAAI,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;QAEhE,QAAQ,cAAc,CAAC,iBAAiB,EAAE,CAAC;YACzC,KAAK,yBAAyB,CAAC,QAAQ,CAAC,CAAC,CAAC;gBACxC,OAAO,IAAI,gBAAgB,CACzB,cAAc,CAAC,MAAM,EACrB,MAAM,EACN,aAAa,EACb,OAAO,EACP,iBAAiB,CAClB,CAAC;YACJ,CAAC;YACD,KAAK,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC;gBACzC,OAAO,IAAI,iBAAiB,CAC1B,cAAc,CAAC,MAAM,EACrB,iBAAiB,EACjB,sBAAsB,EACtB,MAAM,EACN,aAAa,EACb,OAAO,CACR,CAAC;YACJ,CAAC;YACD,KAAK,yBAAyB,CAAC,iBAAiB,CAAC,CAAC,CAAC;gBACjD,OAAO,IAAI,yBAAyB,CAClC,cAAc,CAAC,MAAM,EACrB,iBAAiB,EACjB,MAAM,EACN,aAAa,EACb,OAAO,CACR,CAAC;YACJ,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACR,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,oBAAoB,CACjC,cAA8B;QAE9B,MAAM,aAAa,GAAuC,EAAE,CAAC;QAE7D,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC;YACpE,MAAM,UAAU,GAAG;gBACjB,uBAAuB,EAAE,MAAM,CAAC,uBAAuB,IAAI,CAAC;aAC7D,CAAC;YAEF,MAAM,aAAa,GACjB,MAAM,CAAC,aAAa,IAAI,aAAa,CAAC,iBAAiB,CAAC;YAE1D,IAAI,aAAa,KAAK,aAAa,CAAC,SAAS,EAAE,CAAC;gBAC9C,aAAa,CAAC,KAAK,CAAC,GAAG;oBACrB,GAAG,UAAU;oBACb,aAAa,EAAE,WAAW;oBAC1B,cAAc,EAAE,MAAM,CAAC,cAAe,EAAE,6BAA6B;oBACrE,QAAQ,EAAE,MAAM,CAAC,QAAoD;iBACtE,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,KAAK,CAAC,GAAG;oBACrB,GAAG,UAAU;oBACb,aAAa,EAAE,mBAAmB;oBAClC,MAAM,EAAE,MAAM,CAAC,MAAO,EAAE,6BAA6B;oBACrD,QAAQ,EAAE,MAAM,CAAC,QAEhB;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,aAAa,CAAC;IACvB,CAAC;CACF"}

package/dist/strategy/StrategyFactory.test.d.ts

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

package/dist/strategy/StrategyFactory.test.d.ts.map

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

package/dist/strategy/StrategyFactory.test.js

@@ -0,0 +1,81 @@
+import { expect } from 'chai';
+import { ethers } from 'ethers';
+import { pino } from 'pino';
+import { Token, TokenStandard } from '@humanode-hyperlane/sdk';
+import { RebalancerMinAmountType, RebalancerStrategyOptions, } from '../config/types.js';
+import { MinAmountStrategy } from './MinAmountStrategy.js';
+import { StrategyFactory } from './StrategyFactory.js';
+import { WeightedStrategy } from './WeightedStrategy.js';
+const testLogger = pino({ level: 'silent' });
+describe('StrategyFactory', () => {
+    const chain1 = 'chain1';
+    const chain2 = 'chain2';
+    const totalCollateral = BigInt(20e18);
+    const tokensByChainName = {};
+    const tokenArgs = {
+        name: 'token',
+        decimals: 18,
+        symbol: 'TOKEN',
+        standard: TokenStandard.ERC20,
+        addressOrDenom: '',
+    };
+    tokensByChainName[chain1] = new Token({ ...tokenArgs, chainName: chain1 });
+    tokensByChainName[chain2] = new Token({ ...tokenArgs, chainName: chain2 });
+    describe('createStrategy', () => {
+        it('creates a WeightedStrategy when given weighted configuration', () => {
+            const config = {
+                [chain1]: {
+                    weighted: {
+                        weight: 100n,
+                        tolerance: 0n,
+                    },
+                    bridge: ethers.constants.AddressZero,
+                    bridgeLockTime: 1,
+                },
+                [chain2]: {
+                    weighted: {
+                        weight: 100n,
+                        tolerance: 0n,
+                    },
+                    bridge: ethers.constants.AddressZero,
+                    bridgeLockTime: 1,
+                },
+            };
+            const strategyConfig = {
+                rebalanceStrategy: RebalancerStrategyOptions.Weighted,
+                chains: config,
+            };
+            const strategy = StrategyFactory.createStrategy([strategyConfig], tokensByChainName, totalCollateral, testLogger);
+            expect(strategy).to.be.instanceOf(WeightedStrategy);
+        });
+        it('creates a MinAmountStrategy when given minAmount configuration', () => {
+            const config = {
+                [chain1]: {
+                    minAmount: {
+                        min: 8,
+                        target: 10,
+                        type: RebalancerMinAmountType.Absolute,
+                    },
+                    bridge: ethers.constants.AddressZero,
+                    bridgeLockTime: 1,
+                },
+                [chain2]: {
+                    minAmount: {
+                        min: 8,
+                        target: 10,
+                        type: RebalancerMinAmountType.Absolute,
+                    },
+                    bridge: ethers.constants.AddressZero,
+                    bridgeLockTime: 1,
+                },
+            };
+            const strategyConfig = {
+                rebalanceStrategy: RebalancerStrategyOptions.MinAmount,
+                chains: config,
+            };
+            const strategy = StrategyFactory.createStrategy([strategyConfig], tokensByChainName, totalCollateral, testLogger);
+            expect(strategy).to.be.instanceOf(MinAmountStrategy);
+        });
+    });
+});
+//# sourceMappingURL=StrategyFactory.test.js.map

package/dist/strategy/StrategyFactory.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"StrategyFactory.test.js","sourceRoot":"","sources":["../../src/strategy/StrategyFactory.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAC9B,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAE5B,OAAO,EAAiB,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAE9E,OAAO,EAEL,uBAAuB,EACvB,yBAAyB,GAG1B,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzD,MAAM,UAAU,GAAG,IAAI,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;AAE7C,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,MAAM,MAAM,GAAG,QAAQ,CAAC;IACxB,MAAM,MAAM,GAAG,QAAQ,CAAC;IACxB,MAAM,eAAe,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAEtC,MAAM,iBAAiB,GAAoB,EAAE,CAAC;IAC9C,MAAM,SAAS,GAAG;QAChB,IAAI,EAAE,OAAO;QACb,QAAQ,EAAE,EAAE;QACZ,MAAM,EAAE,OAAO;QACf,QAAQ,EAAE,aAAa,CAAC,KAAK;QAC7B,cAAc,EAAE,EAAE;KACnB,CAAC;IACF,iBAAiB,CAAC,MAAM,CAAC,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;IAC3E,iBAAiB,CAAC,MAAM,CAAC,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;IAE3E,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,MAAM,GAA2B;gBACrC,CAAC,MAAM,CAAC,EAAE;oBACR,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI;wBACZ,SAAS,EAAE,EAAE;qBACd;oBACD,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,WAAW;oBACpC,cAAc,EAAE,CAAC;iBAClB;gBACD,CAAC,MAAM,CAAC,EAAE;oBACR,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI;wBACZ,SAAS,EAAE,EAAE;qBACd;oBACD,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,WAAW;oBACpC,cAAc,EAAE,CAAC;iBAClB;aACF,CAAC;YAEF,MAAM,cAAc,GAAmB;gBACrC,iBAAiB,EAAE,yBAAyB,CAAC,QAAQ;gBACrD,MAAM,EAAE,MAAM;aACf,CAAC;YAEF,MAAM,QAAQ,GAAG,eAAe,CAAC,cAAc,CAC7C,CAAC,cAAc,CAAC,EAChB,iBAAiB,EACjB,eAAe,EACf,UAAU,CACX,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;YACxE,MAAM,MAAM,GAA4B;gBACtC,CAAC,MAAM,CAAC,EAAE;oBACR,SAAS,EAAE;wBACT,GAAG,EAAE,CAAC;wBACN,MAAM,EAAE,EAAE;wBACV,IAAI,EAAE,uBAAuB,CAAC,QAAQ;qBACvC;oBACD,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,WAAW;oBACpC,cAAc,EAAE,CAAC;iBAClB;gBACD,CAAC,MAAM,CAAC,EAAE;oBACR,SAAS,EAAE;wBACT,GAAG,EAAE,CAAC;wBACN,MAAM,EAAE,EAAE;wBACV,IAAI,EAAE,uBAAuB,CAAC,QAAQ;qBACvC;oBACD,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,WAAW;oBACpC,cAAc,EAAE,CAAC;iBAClB;aACF,CAAC;YAEF,MAAM,cAAc,GAAmB;gBACrC,iBAAiB,EAAE,yBAAyB,CAAC,SAAS;gBACtD,MAAM,EAAE,MAAM;aACf,CAAC;YAEF,MAAM,QAAQ,GAAG,eAAe,CAAC,cAAc,CAC7C,CAAC,cAAc,CAAC,EAChB,iBAAiB,EACjB,eAAe,EACf,UAAU,CACX,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/strategy/WeightedStrategy.d.ts

@@ -0,0 +1,32 @@
+import { type Logger } from 'pino';
+import type { ChainMap, Token } from '@humanode-hyperlane/sdk';
+import { RebalancerStrategyOptions, type WeightedStrategyConfig } from '../config/types.js';
+import type { RawBalances, Route, StrategyRoute } from '../interfaces/IStrategy.js';
+import { type Metrics } from '../metrics/Metrics.js';
+import type { BridgeConfigWithOverride } from '../utils/bridgeUtils.js';
+import { BaseStrategy, type Delta } from './BaseStrategy.js';
+/**
+ * Strategy implementation that rebalance based on weights
+ * It distributes funds across chains based on their weights
+ */
+export declare class WeightedStrategy extends BaseStrategy {
+    readonly name = RebalancerStrategyOptions.Weighted;
+    private readonly config;
+    private readonly totalWeight;
+    protected readonly logger: Logger;
+    constructor(config: WeightedStrategyConfig, logger: Logger, bridgeConfigs: ChainMap<BridgeConfigWithOverride>, metrics?: Metrics, tokensByChainName?: ChainMap<Token>);
+    /**
+     * Gets balances categorized by surplus and deficit based on weights
+     *
+     * Simulates both types of rebalances before calculating surpluses/deficits:
+     * - pendingRebalances: in-flight intents (origin tx confirmed, add to destination only)
+     * - proposedRebalances: routes from earlier strategies (subtract from origin AND add to destination)
+     *
+     * This prevents over-rebalancing when multiple strategies run in sequence.
+     */
+    protected getCategorizedBalances(rawBalances: RawBalances, pendingRebalances?: Route[], proposedRebalances?: StrategyRoute[]): {
+        surpluses: Delta[];
+        deficits: Delta[];
+    };
+}
+//# sourceMappingURL=WeightedStrategy.d.ts.map

package/dist/strategy/WeightedStrategy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"WeightedStrategy.d.ts","sourceRoot":"","sources":["../../src/strategy/WeightedStrategy.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,MAAM,EAAE,MAAM,MAAM,CAAC;AAEnC,OAAO,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAE/D,OAAO,EACL,yBAAyB,EACzB,KAAK,sBAAsB,EAC5B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EACV,WAAW,EACX,KAAK,EACL,aAAa,EACd,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,KAAK,OAAO,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AAExE,OAAO,EAAE,YAAY,EAAE,KAAK,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAE7D;;;GAGG;AACH,qBAAa,gBAAiB,SAAQ,YAAY;IAChD,QAAQ,CAAC,IAAI,sCAAsC;IACnD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAyB;IAChD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;gBAGhC,MAAM,EAAE,sBAAsB,EAC9B,MAAM,EAAE,MAAM,EACd,aAAa,EAAE,QAAQ,CAAC,wBAAwB,CAAC,EACjD,OAAO,CAAC,EAAE,OAAO,EACjB,iBAAiB,CAAC,EAAE,QAAQ,CAAC,KAAK,CAAC;IAkCrC;;;;;;;;OAQG;IACH,SAAS,CAAC,sBAAsB,CAC9B,WAAW,EAAE,WAAW,EACxB,iBAAiB,CAAC,EAAE,KAAK,EAAE,EAC3B,kBAAkB,CAAC,EAAE,aAAa,EAAE,GACnC;QACD,SAAS,EAAE,KAAK,EAAE,CAAC;QACnB,QAAQ,EAAE,KAAK,EAAE,CAAC;KACnB;CA0CF"}

package/dist/strategy/WeightedStrategy.js

@@ -0,0 +1,73 @@
+import { RebalancerStrategyOptions, } from '../config/types.js';
+import { BaseStrategy } from './BaseStrategy.js';
+/**
+ * Strategy implementation that rebalance based on weights
+ * It distributes funds across chains based on their weights
+ */
+export class WeightedStrategy extends BaseStrategy {
+    name = RebalancerStrategyOptions.Weighted;
+    config;
+    totalWeight;
+    logger;
+    constructor(config, logger, bridgeConfigs, metrics, tokensByChainName) {
+        const chains = Object.keys(config);
+        const log = logger.child({ class: WeightedStrategy.name });
+        super(chains, log, bridgeConfigs, metrics, tokensByChainName);
+        this.logger = log;
+        let totalWeight = 0n;
+        for (const chain of chains) {
+            const { weight, tolerance } = config[chain].weighted;
+            if (weight < 0n) {
+                throw new Error(`Weight (${weight}) must not be negative for ${chain}`);
+            }
+            if (tolerance < 0n || tolerance > 100n) {
+                throw new Error(`Tolerance (${tolerance}) must be between 0 and 100 for ${chain}`);
+            }
+            totalWeight += weight;
+        }
+        if (totalWeight <= 0n) {
+            throw new Error('The total weight for all chains must be greater than 0');
+        }
+        this.config = config;
+        this.totalWeight = totalWeight;
+        this.logger.info('WeightedStrategy created');
+    }
+    /**
+     * Gets balances categorized by surplus and deficit based on weights
+     *
+     * Simulates both types of rebalances before calculating surpluses/deficits:
+     * - pendingRebalances: in-flight intents (origin tx confirmed, add to destination only)
+     * - proposedRebalances: routes from earlier strategies (subtract from origin AND add to destination)
+     *
+     * This prevents over-rebalancing when multiple strategies run in sequence.
+     */
+    getCategorizedBalances(rawBalances, pendingRebalances, proposedRebalances) {
+        // Step 1: Simulate pending rebalances (in-flight, origin already deducted on-chain)
+        let simulatedBalances = this.simulatePendingRebalances(rawBalances, pendingRebalances ?? []);
+        // Step 2: Simulate proposed rebalances (from earlier strategies, not yet executed)
+        simulatedBalances = this.simulateProposedRebalances(simulatedBalances, proposedRebalances ?? []);
+        // Get the total balance from all chains (using simulated balances)
+        const total = this.chains.reduce((sum, chain) => sum + simulatedBalances[chain], 0n);
+        return this.chains.reduce((acc, chain) => {
+            const { weight, tolerance } = this.config[chain].weighted;
+            const target = (total * weight) / this.totalWeight;
+            const toleranceAmount = (target * tolerance) / 100n;
+            const balance = simulatedBalances[chain];
+            // Apply the tolerance to deficits to prevent small imbalances
+            if (balance < target - toleranceAmount) {
+                acc.deficits.push({ chain, amount: target - balance });
+            }
+            else if (balance > target) {
+                acc.surpluses.push({ chain, amount: balance - target });
+            }
+            else {
+                // Do nothing as the balance is already on target
+            }
+            return acc;
+        }, {
+            surpluses: [],
+            deficits: [],
+        });
+    }
+}
+//# sourceMappingURL=WeightedStrategy.js.map

package/dist/strategy/WeightedStrategy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WeightedStrategy.js","sourceRoot":"","sources":["../../src/strategy/WeightedStrategy.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,yBAAyB,GAE1B,MAAM,oBAAoB,CAAC;AAS5B,OAAO,EAAE,YAAY,EAAc,MAAM,mBAAmB,CAAC;AAE7D;;;GAGG;AACH,MAAM,OAAO,gBAAiB,SAAQ,YAAY;IACvC,IAAI,GAAG,yBAAyB,CAAC,QAAQ,CAAC;IAClC,MAAM,CAAyB;IAC/B,WAAW,CAAS;IAClB,MAAM,CAAS;IAElC,YACE,MAA8B,EAC9B,MAAc,EACd,aAAiD,EACjD,OAAiB,EACjB,iBAAmC;QAEnC,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,gBAAgB,CAAC,IAAI,EAAE,CAAC,CAAC;QAC3D,KAAK,CAAC,MAAM,EAAE,GAAG,EAAE,aAAa,EAAE,OAAO,EAAE,iBAAiB,CAAC,CAAC;QAC9D,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC;QAElB,IAAI,WAAW,GAAG,EAAE,CAAC;QAErB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC;YAErD,IAAI,MAAM,GAAG,EAAE,EAAE,CAAC;gBAChB,MAAM,IAAI,KAAK,CAAC,WAAW,MAAM,8BAA8B,KAAK,EAAE,CAAC,CAAC;YAC1E,CAAC;YAED,IAAI,SAAS,GAAG,EAAE,IAAI,SAAS,GAAG,IAAI,EAAE,CAAC;gBACvC,MAAM,IAAI,KAAK,CACb,cAAc,SAAS,mCAAmC,KAAK,EAAE,CAClE,CAAC;YACJ,CAAC;YAED,WAAW,IAAI,MAAM,CAAC;QACxB,CAAC;QAED,IAAI,WAAW,IAAI,EAAE,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;QAC5E,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;IAC/C,CAAC;IAED;;;;;;;;OAQG;IACO,sBAAsB,CAC9B,WAAwB,EACxB,iBAA2B,EAC3B,kBAAoC;QAKpC,oFAAoF;QACpF,IAAI,iBAAiB,GAAG,IAAI,CAAC,yBAAyB,CACpD,WAAW,EACX,iBAAiB,IAAI,EAAE,CACxB,CAAC;QAEF,mFAAmF;QACnF,iBAAiB,GAAG,IAAI,CAAC,0BAA0B,CACjD,iBAAiB,EACjB,kBAAkB,IAAI,EAAE,CACzB,CAAC;QACF,mEAAmE;QACnE,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAC9B,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,iBAAiB,CAAC,KAAK,CAAC,EAC9C,EAAE,CACH,CAAC;QAEF,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CACvB,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;YACb,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC;YAC1D,MAAM,MAAM,GAAG,CAAC,KAAK,GAAG,MAAM,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC;YACnD,MAAM,eAAe,GAAG,CAAC,MAAM,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;YACpD,MAAM,OAAO,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAEzC,8DAA8D;YAC9D,IAAI,OAAO,GAAG,MAAM,GAAG,eAAe,EAAE,CAAC;gBACvC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,EAAE,CAAC,CAAC;YACzD,CAAC;iBAAM,IAAI,OAAO,GAAG,MAAM,EAAE,CAAC;gBAC5B,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,GAAG,MAAM,EAAE,CAAC,CAAC;YAC1D,CAAC;iBAAM,CAAC;gBACN,iDAAiD;YACnD,CAAC;YAED,OAAO,GAAG,CAAC;QACb,CAAC,EACD;YACE,SAAS,EAAE,EAAa;YACxB,QAAQ,EAAE,EAAa;SACxB,CACF,CAAC;IACJ,CAAC;CACF"}

package/dist/strategy/WeightedStrategy.test.d.ts

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

package/dist/strategy/WeightedStrategy.test.d.ts.map

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