@hyperlane-xyz/rebalancer 0.1.2 → 1.0.0

package/dist/core/WithSemaphore.js

@@ -1,67 +0,0 @@
-/**
- * Prevents frequent rebalancing operations while bridges complete.
- */
-export class WithSemaphore {
-    config;
-    rebalancer;
-    // Timestamp until which rebalancing should be blocked
-    waitUntil = 0;
-    // Lock to prevent concurrent rebalance execution
-    executing = false;
-    logger;
-    constructor(config, rebalancer, logger) {
-        this.config = config;
-        this.rebalancer = rebalancer;
-        this.logger = logger.child({ class: WithSemaphore.name });
-    }
-    /**
-     * Rebalance with timing control
-     * @param routes - Routes to process
-     */
-    async rebalance(routes) {
-        if (this.executing) {
-            this.logger.info('Currently executing rebalance. Skipping.');
-            return;
-        }
-        // No routes mean the system is balanced so we reset the timer to allow new rebalancing
-        if (!routes.length) {
-            this.logger.info('No routes to execute. Assuming rebalance is complete. Resetting semaphore timer.');
-            this.waitUntil = 0;
-            return;
-        }
-        // Skip if still in waiting period
-        if (Date.now() < this.waitUntil) {
-            this.logger.info('Still in waiting period. Skipping rebalance.');
-            return;
-        }
-        // The wait period will be determined by the bridge with the highest wait tolerance
-        const highestTolerance = this.getHighestLockTime(routes);
-        try {
-            // Execute rebalance
-            this.executing = true;
-            await this.rebalancer.rebalance(routes);
-        }
-        finally {
-            this.executing = false;
-        }
-        // Set new waiting period
-        this.waitUntil = Date.now() + highestTolerance;
-        this.logger.info({
-            highestTolerance,
-            waitUntil: this.waitUntil,
-        }, 'Rebalance semaphore locked');
-    }
-    getHighestLockTime(routes) {
-        return routes.reduce((highest, route) => {
-            const origin = this.config.strategyConfig.chains[route.origin];
-            if (!origin) {
-                this.logger.error({ route }, 'Chain not found in config. Skipping.');
-                throw new Error(`Chain ${route.origin} not found in config`);
-            }
-            const bridgeLockTime = origin.bridgeLockTime;
-            const overrideLockTime = origin.override?.[route.destination]?.bridgeLockTime ?? 0;
-            return Math.max(highest, bridgeLockTime, overrideLockTime);
-        }, 0);
-    }
-}
-//# sourceMappingURL=WithSemaphore.js.map

package/dist/core/WithSemaphore.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"WithSemaphore.js","sourceRoot":"","sources":["../../src/core/WithSemaphore.ts"],"names":[],"mappings":"AAMA;;GAEG;AACH,MAAM,OAAO,aAAa;IAQL;IACA;IARnB,sDAAsD;IAC9C,SAAS,GAAW,CAAC,CAAC;IAC9B,iDAAiD;IACzC,SAAS,GAAY,KAAK,CAAC;IAClB,MAAM,CAAS;IAEhC,YACmB,MAAwB,EACxB,UAAuB,EACxC,MAAc;QAFG,WAAM,GAAN,MAAM,CAAkB;QACxB,eAAU,GAAV,UAAU,CAAa;QAGxC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,SAAS,CAAC,MAA0B;QACxC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;YAE7D,OAAO;QACT,CAAC;QAED,uFAAuF;QACvF,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,kFAAkF,CACnF,CAAC;YAEF,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC;YACnB,OAAO;QACT,CAAC;QAED,kCAAkC;QAClC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;YAChC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;YAEjE,OAAO;QACT,CAAC;QAED,mFAAmF;QACnF,MAAM,gBAAgB,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAEzD,IAAI,CAAC;YACH,oBAAoB;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACzB,CAAC;QAED,yBAAyB;QACzB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB,CAAC;QAE/C,IAAI,CAAC,MAAM,CAAC,IAAI,CACd;YACE,gBAAgB;YAChB,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,EACD,4BAA4B,CAC7B,CAAC;IACJ,CAAC;IAEO,kBAAkB,CAAC,MAA0B;QACnD,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;YACtC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAE/D,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,sCAAsC,CAAC,CAAC;gBACrE,MAAM,IAAI,KAAK,CAAC,SAAS,KAAK,CAAC,MAAM,sBAAsB,CAAC,CAAC;YAC/D,CAAC;YAED,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;YAC7C,MAAM,gBAAgB,GACpB,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,WAAW,CAAC,EAAE,cAAc,IAAI,CAAC,CAAC;YAE5D,OAAO,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,cAAc,EAAE,gBAAgB,CAAC,CAAC;QAC7D,CAAC,EAAE,CAAC,CAAC,CAAC;IACR,CAAC;CACF"}

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

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

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

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

package/dist/core/WithSemaphore.test.js

@@ -1,83 +0,0 @@
-import chai, { expect } from 'chai';
-import chaiAsPromised from 'chai-as-promised';
-import { pino } from 'pino';
-import Sinon from 'sinon';
-import { RebalancerStrategyOptions } from '../config/types.js';
-import { MockRebalancer, buildTestConfig } from '../test/helpers.js';
-import { WithSemaphore } from './WithSemaphore.js';
-chai.use(chaiAsPromised);
-const testLogger = pino({ level: 'silent' });
-describe('WithSemaphore', () => {
-    it('should call the underlying rebalancer', async () => {
-        const config = buildTestConfig();
-        const routes = [
-            {
-                origin: 'chain1',
-            },
-        ];
-        const rebalancer = new MockRebalancer();
-        const rebalanceSpy = Sinon.spy(rebalancer, 'rebalance');
-        const withSemaphore = new WithSemaphore(config, rebalancer, testLogger);
-        await withSemaphore.rebalance(routes);
-        expect(rebalanceSpy.calledOnce).to.be.true;
-        expect(rebalanceSpy.calledWith(routes)).to.be.true;
-    });
-    it('should return early if there are no routes', async () => {
-        const config = buildTestConfig();
-        const rebalancer = new MockRebalancer();
-        const rebalanceSpy = Sinon.spy(rebalancer, 'rebalance');
-        const withSemaphore = new WithSemaphore(config, rebalancer, testLogger);
-        await withSemaphore.rebalance([]);
-        expect(rebalanceSpy.calledOnce).to.be.false;
-    });
-    it('should return early if rebalance occurs before waitUntil is reached', async () => {
-        const config = buildTestConfig();
-        const routes = [
-            {
-                origin: 'chain1',
-            },
-        ];
-        const rebalancer = new MockRebalancer();
-        const rebalanceSpy = Sinon.spy(rebalancer, 'rebalance');
-        const withSemaphore = new WithSemaphore(config, rebalancer, testLogger);
-        await withSemaphore.rebalance(routes);
-        expect(rebalanceSpy.calledOnce).to.be.true;
-        expect(rebalanceSpy.calledWith(routes)).to.be.true;
-        rebalanceSpy.resetHistory();
-        await withSemaphore.rebalance(routes);
-        expect(rebalanceSpy.calledOnce).to.be.false;
-    });
-    it('should throw if a chain is missing', async () => {
-        const config = buildTestConfig({
-            strategyConfig: {
-                rebalanceStrategy: RebalancerStrategyOptions.Weighted,
-                chains: {},
-            },
-        });
-        const routes = [
-            {
-                origin: 'chain1',
-            },
-        ];
-        const rebalancer = new MockRebalancer();
-        const withSemaphore = new WithSemaphore(config, rebalancer, testLogger);
-        await expect(withSemaphore.rebalance(routes)).to.be.rejectedWith(`Chain ${routes[0].origin} not found in config`);
-    });
-    it('should not execute if another rebalance is currently executing', async () => {
-        const config = buildTestConfig();
-        const routes = [
-            {
-                origin: 'chain1',
-            },
-        ];
-        const rebalancer = new MockRebalancer();
-        const rebalanceSpy = Sinon.spy(rebalancer, 'rebalance');
-        const withSemaphore = new WithSemaphore(config, rebalancer, testLogger);
-        const rebalancePromise1 = withSemaphore.rebalance(routes);
-        const rebalancePromise2 = withSemaphore.rebalance(routes);
-        await rebalancePromise1;
-        await rebalancePromise2;
-        expect(rebalanceSpy.calledOnce).to.be.true;
-    });
-});
-//# sourceMappingURL=WithSemaphore.test.js.map

package/dist/core/WithSemaphore.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"WithSemaphore.test.js","sourceRoot":"","sources":["../../src/core/WithSemaphore.test.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,EAAE,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AACpC,OAAO,cAAc,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAE/D,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErE,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;AAEzB,MAAM,UAAU,GAAG,IAAI,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;AAE7C,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,MAAM,GAAG,eAAe,EAAE,CAAC;QAEjC,MAAM,MAAM,GAAG;YACb;gBACE,MAAM,EAAE,QAAQ;aACU;SAC7B,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,MAAM,YAAY,GAAG,KAAK,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QACxD,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QACxE,MAAM,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAEtC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAC3C,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,MAAM,GAAG,eAAe,EAAE,CAAC;QAEjC,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,MAAM,YAAY,GAAG,KAAK,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QACxD,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QACxE,MAAM,aAAa,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAElC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,MAAM,GAAG,eAAe,EAAE,CAAC;QAEjC,MAAM,MAAM,GAAG;YACb;gBACE,MAAM,EAAE,QAAQ;aACU;SAC7B,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,MAAM,YAAY,GAAG,KAAK,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QACxD,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QACxE,MAAM,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAEtC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAC3C,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAEnD,YAAY,CAAC,YAAY,EAAE,CAAC;QAC5B,MAAM,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAEtC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,MAAM,GAAG,eAAe,CAAC;YAC7B,cAAc,EAAE;gBACd,iBAAiB,EAAE,yBAAyB,CAAC,QAAQ;gBACrD,MAAM,EAAE,EAAE;aACX;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG;YACb;gBACE,MAAM,EAAE,QAAQ;aACU;SAC7B,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QAExE,MAAM,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,YAAY,CAC9D,SAAS,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,sBAAsB,CAChD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,MAAM,GAAG,eAAe,EAAE,CAAC;QAEjC,MAAM,MAAM,GAAG;YACb;gBACE,MAAM,EAAE,QAAQ;aACU;SAC7B,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,MAAM,YAAY,GAAG,KAAK,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QACxD,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QAExE,MAAM,iBAAiB,GAAG,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC1D,MAAM,iBAAiB,GAAG,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC1D,MAAM,iBAAiB,CAAC;QACxB,MAAM,iBAAiB,CAAC;QAExB,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IAC7C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/src/core/WithInflightGuard.test.ts

@@ -1,131 +0,0 @@
-import chai, { expect } from 'chai';
-import chaiAsPromised from 'chai-as-promised';
-import { ethers } from 'ethers';
-import { pino } from 'pino';
-import Sinon from 'sinon';
-
-import { chainMetadata } from '@hyperlane-xyz/registry';
-import { ChainMetadataManager } from '@hyperlane-xyz/sdk';
-
-import { type RebalancingRoute } from '../interfaces/IStrategy.js';
-import { MockRebalancer, buildTestConfig } from '../test/helpers.js';
-import { ExplorerClient } from '../utils/ExplorerClient.js';
-
-import { WithInflightGuard } from './WithInflightGuard.js';
-
-chai.use(chaiAsPromised);
-
-const testLogger = pino({ level: 'silent' });
-
-describe('WithInflightGuard', () => {
-  it('forwards empty routes without calling Explorer', async () => {
-    const config = buildTestConfig();
-
-    const rebalancer = new MockRebalancer();
-    const rebalanceSpy = Sinon.spy(rebalancer, 'rebalance');
-
-    const explorer = new ExplorerClient('http://localhost');
-    const explorerSpy = Sinon.stub(explorer, 'hasUndeliveredRebalance');
-
-    const guard = new WithInflightGuard(
-      config,
-      rebalancer,
-      explorer,
-      ethers.Wallet.createRandom().address,
-      new ChainMetadataManager(chainMetadata as any),
-      testLogger,
-    );
-
-    await guard.rebalance([]);
-
-    expect(explorerSpy.called).to.be.false;
-    expect(rebalanceSpy.calledOnce).to.be.true;
-    expect(rebalanceSpy.calledWith([])).to.be.true;
-  });
-
-  it('calls underlying rebalancer when no inflight is detected', async () => {
-    const config = buildTestConfig({}, ['ethereum', 'arbitrum']);
-    const routes: RebalancingRoute[] = [{ origin: 'ethereum' } as any];
-
-    const rebalancer = new MockRebalancer();
-    const rebalanceSpy = Sinon.spy(rebalancer, 'rebalance');
-
-    const explorer = new ExplorerClient('http://localhost');
-    const explorerSpy = Sinon.stub(
-      explorer,
-      'hasUndeliveredRebalance',
-    ).resolves(false);
-
-    const guard = new WithInflightGuard(
-      config,
-      rebalancer,
-      explorer,
-      ethers.Wallet.createRandom().address,
-      new ChainMetadataManager(chainMetadata as any),
-      testLogger,
-    );
-
-    await guard.rebalance(routes);
-
-    expect(explorerSpy.calledOnce).to.be.true;
-    expect(rebalanceSpy.calledOnce).to.be.true;
-    expect(rebalanceSpy.calledWith(routes)).to.be.true;
-  });
-
-  it('skips rebalancing when inflight is detected', async () => {
-    const config = buildTestConfig({}, ['ethereum', 'arbitrum']);
-    const routes: RebalancingRoute[] = [{ origin: 'ethereum' } as any];
-
-    const rebalancer = new MockRebalancer();
-    const rebalanceSpy = Sinon.spy(rebalancer, 'rebalance');
-
-    const explorer = new ExplorerClient('http://localhost');
-    const explorerSpy = Sinon.stub(
-      explorer,
-      'hasUndeliveredRebalance',
-    ).resolves(true);
-
-    const guard = new WithInflightGuard(
-      config,
-      rebalancer,
-      explorer,
-      ethers.Wallet.createRandom().address,
-      new ChainMetadataManager(chainMetadata as any),
-      testLogger,
-    );
-
-    await guard.rebalance(routes);
-
-    expect(explorerSpy.calledOnce).to.be.true;
-    expect(rebalanceSpy.called).to.be.false;
-  });
-
-  it('propagates explorer query error', async () => {
-    const config = buildTestConfig({}, ['ethereum', 'arbitrum']);
-    const routes: RebalancingRoute[] = [{ origin: 'ethereum' } as any];
-
-    const rebalancer = new MockRebalancer();
-    const rebalanceSpy = Sinon.spy(rebalancer, 'rebalance');
-
-    const explorer = new ExplorerClient('http://localhost');
-    const explorerSpy = Sinon.stub(explorer, 'hasUndeliveredRebalance').rejects(
-      new Error('Explorer HTTP 405'),
-    );
-
-    const guard = new WithInflightGuard(
-      config,
-      rebalancer,
-      explorer,
-      ethers.Wallet.createRandom().address,
-      new ChainMetadataManager(chainMetadata as any),
-      testLogger,
-    );
-
-    await expect(guard.rebalance(routes)).to.be.rejectedWith(
-      'Explorer HTTP 405',
-    );
-
-    expect(explorerSpy.calledOnce).to.be.true;
-    expect(rebalanceSpy.called).to.be.false;
-  });
-});

package/src/core/WithInflightGuard.ts

@@ -1,67 +0,0 @@
-import type { Logger } from 'pino';
-
-import { type ChainMetadataManager } from '@hyperlane-xyz/sdk';
-
-import { type RebalancerConfig } from '../config/RebalancerConfig.js';
-import type { IRebalancer } from '../interfaces/IRebalancer.js';
-import type { RebalancingRoute } from '../interfaces/IStrategy.js';
-import { type ExplorerClient } from '../utils/ExplorerClient.js';
-
-/**
- * Prevents rebalancing if there are inflight rebalances for the warp route.
- */
-export class WithInflightGuard implements IRebalancer {
-  private readonly logger: Logger;
-
-  constructor(
-    private readonly config: RebalancerConfig,
-    private readonly rebalancer: IRebalancer,
-    private readonly explorer: ExplorerClient,
-    private readonly txSender: string,
-    private readonly chainManager: ChainMetadataManager,
-    logger: Logger,
-  ) {
-    this.logger = logger.child({ class: WithInflightGuard.name });
-  }
-
-  async rebalance(routes: RebalancingRoute[]): Promise<void> {
-    // Always enforce the inflight guard
-    if (routes.length === 0) {
-      return this.rebalancer.rebalance(routes);
-    }
-
-    const chains = Object.keys(this.config.strategyConfig.chains);
-    const bridges = chains.map(
-      (chain) => this.config.strategyConfig.chains[chain].bridge,
-    );
-    const domains = chains.map((chain) => this.chainManager.getDomainId(chain));
-
-    let hasInflightRebalances = false;
-    try {
-      hasInflightRebalances = await this.explorer.hasUndeliveredRebalance(
-        {
-          bridges,
-          domains: Array.from(new Set(domains)),
-          txSender: this.txSender,
-          limit: 5,
-        },
-        this.logger,
-      );
-    } catch (e: any) {
-      this.logger.error(
-        { status: e.status, body: e.body },
-        'Explorer inflight query failed',
-      );
-      throw e;
-    }
-
-    if (hasInflightRebalances) {
-      this.logger.info(
-        'Inflight rebalance detected via Explorer; skipping this cycle',
-      );
-      return;
-    }
-
-    return this.rebalancer.rebalance(routes);
-  }
-}

package/src/core/WithSemaphore.test.ts

@@ -1,111 +0,0 @@
-import chai, { expect } from 'chai';
-import chaiAsPromised from 'chai-as-promised';
-import { pino } from 'pino';
-import Sinon from 'sinon';
-
-import { RebalancerStrategyOptions } from '../config/types.js';
-import { type RebalancingRoute } from '../interfaces/IStrategy.js';
-import { MockRebalancer, buildTestConfig } from '../test/helpers.js';
-
-import { WithSemaphore } from './WithSemaphore.js';
-
-chai.use(chaiAsPromised);
-
-const testLogger = pino({ level: 'silent' });
-
-describe('WithSemaphore', () => {
-  it('should call the underlying rebalancer', async () => {
-    const config = buildTestConfig();
-
-    const routes = [
-      {
-        origin: 'chain1',
-      } as any as RebalancingRoute,
-    ];
-
-    const rebalancer = new MockRebalancer();
-    const rebalanceSpy = Sinon.spy(rebalancer, 'rebalance');
-    const withSemaphore = new WithSemaphore(config, rebalancer, testLogger);
-    await withSemaphore.rebalance(routes);
-
-    expect(rebalanceSpy.calledOnce).to.be.true;
-    expect(rebalanceSpy.calledWith(routes)).to.be.true;
-  });
-
-  it('should return early if there are no routes', async () => {
-    const config = buildTestConfig();
-
-    const rebalancer = new MockRebalancer();
-    const rebalanceSpy = Sinon.spy(rebalancer, 'rebalance');
-    const withSemaphore = new WithSemaphore(config, rebalancer, testLogger);
-    await withSemaphore.rebalance([]);
-
-    expect(rebalanceSpy.calledOnce).to.be.false;
-  });
-
-  it('should return early if rebalance occurs before waitUntil is reached', async () => {
-    const config = buildTestConfig();
-
-    const routes = [
-      {
-        origin: 'chain1',
-      } as any as RebalancingRoute,
-    ];
-
-    const rebalancer = new MockRebalancer();
-    const rebalanceSpy = Sinon.spy(rebalancer, 'rebalance');
-    const withSemaphore = new WithSemaphore(config, rebalancer, testLogger);
-    await withSemaphore.rebalance(routes);
-
-    expect(rebalanceSpy.calledOnce).to.be.true;
-    expect(rebalanceSpy.calledWith(routes)).to.be.true;
-
-    rebalanceSpy.resetHistory();
-    await withSemaphore.rebalance(routes);
-
-    expect(rebalanceSpy.calledOnce).to.be.false;
-  });
-
-  it('should throw if a chain is missing', async () => {
-    const config = buildTestConfig({
-      strategyConfig: {
-        rebalanceStrategy: RebalancerStrategyOptions.Weighted,
-        chains: {},
-      },
-    });
-
-    const routes = [
-      {
-        origin: 'chain1',
-      } as any as RebalancingRoute,
-    ];
-
-    const rebalancer = new MockRebalancer();
-    const withSemaphore = new WithSemaphore(config, rebalancer, testLogger);
-
-    await expect(withSemaphore.rebalance(routes)).to.be.rejectedWith(
-      `Chain ${routes[0].origin} not found in config`,
-    );
-  });
-
-  it('should not execute if another rebalance is currently executing', async () => {
-    const config = buildTestConfig();
-
-    const routes = [
-      {
-        origin: 'chain1',
-      } as any as RebalancingRoute,
-    ];
-
-    const rebalancer = new MockRebalancer();
-    const rebalanceSpy = Sinon.spy(rebalancer, 'rebalance');
-    const withSemaphore = new WithSemaphore(config, rebalancer, testLogger);
-
-    const rebalancePromise1 = withSemaphore.rebalance(routes);
-    const rebalancePromise2 = withSemaphore.rebalance(routes);
-    await rebalancePromise1;
-    await rebalancePromise2;
-
-    expect(rebalanceSpy.calledOnce).to.be.true;
-  });
-});

package/src/core/WithSemaphore.ts

@@ -1,92 +0,0 @@
-import type { Logger } from 'pino';
-
-import { type RebalancerConfig } from '../config/RebalancerConfig.js';
-import type { IRebalancer } from '../interfaces/IRebalancer.js';
-import type { RebalancingRoute } from '../interfaces/IStrategy.js';
-
-/**
- * Prevents frequent rebalancing operations while bridges complete.
- */
-export class WithSemaphore implements IRebalancer {
-  // Timestamp until which rebalancing should be blocked
-  private waitUntil: number = 0;
-  // Lock to prevent concurrent rebalance execution
-  private executing: boolean = false;
-  private readonly logger: Logger;
-
-  constructor(
-    private readonly config: RebalancerConfig,
-    private readonly rebalancer: IRebalancer,
-    logger: Logger,
-  ) {
-    this.logger = logger.child({ class: WithSemaphore.name });
-  }
-
-  /**
-   * Rebalance with timing control
-   * @param routes - Routes to process
-   */
-  async rebalance(routes: RebalancingRoute[]): Promise<void> {
-    if (this.executing) {
-      this.logger.info('Currently executing rebalance. Skipping.');
-
-      return;
-    }
-
-    // No routes mean the system is balanced so we reset the timer to allow new rebalancing
-    if (!routes.length) {
-      this.logger.info(
-        'No routes to execute. Assuming rebalance is complete. Resetting semaphore timer.',
-      );
-
-      this.waitUntil = 0;
-      return;
-    }
-
-    // Skip if still in waiting period
-    if (Date.now() < this.waitUntil) {
-      this.logger.info('Still in waiting period. Skipping rebalance.');
-
-      return;
-    }
-
-    // The wait period will be determined by the bridge with the highest wait tolerance
-    const highestTolerance = this.getHighestLockTime(routes);
-
-    try {
-      // Execute rebalance
-      this.executing = true;
-      await this.rebalancer.rebalance(routes);
-    } finally {
-      this.executing = false;
-    }
-
-    // Set new waiting period
-    this.waitUntil = Date.now() + highestTolerance;
-
-    this.logger.info(
-      {
-        highestTolerance,
-        waitUntil: this.waitUntil,
-      },
-      'Rebalance semaphore locked',
-    );
-  }
-
-  private getHighestLockTime(routes: RebalancingRoute[]) {
-    return routes.reduce((highest, route) => {
-      const origin = this.config.strategyConfig.chains[route.origin];
-
-      if (!origin) {
-        this.logger.error({ route }, 'Chain not found in config. Skipping.');
-        throw new Error(`Chain ${route.origin} not found in config`);
-      }
-
-      const bridgeLockTime = origin.bridgeLockTime;
-      const overrideLockTime =
-        origin.override?.[route.destination]?.bridgeLockTime ?? 0;
-
-      return Math.max(highest, bridgeLockTime, overrideLockTime);
-    }, 0);
-  }
-}