@defisaver/automation-sdk 3.1.3 → 3.2.0

package/esm/automation/private/StrategiesAutomation.js

@@ -1,175 +1,184 @@
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-import Dec from 'decimal.js';
-import { Strategies, ProtocolIdentifiers } from '../../types/enums';
-import { addToObjectIf, isDefined, isUndefined } from '../../services/utils';
-import { getAbiItem, makeSubStorageContract } from '../../services/contractService';
-import { getEventsFromContract, multicall } from '../../services/ethereumService';
-import { parseStrategiesAutomatedPosition } from '../../services/strategiesService';
-import Automation from './Automation';
-export default class StrategiesAutomation extends Automation {
-    constructor(args) {
-        super();
-        this.web3 = args.provider;
-        this.web3Fork = args.providerFork;
-        this.chainId = args.chainId;
-        this.subStorageContract = makeSubStorageContract(this.web3, this.chainId);
-        this.subStorageContractFork = this.web3Fork ? makeSubStorageContract(this.web3Fork, this.chainId) : null;
-        this.assert();
-    }
-    getEventFromSubStorage(event, options) {
-        var _a;
-        return __awaiter(this, void 0, void 0, function* () {
-            // only used for backfilling, so in case options?.fromBlock in undefined
-            // (just like we omit fromBlock when we call from app when not on fork), we still want to fetch events
-            if (new Dec(this.subStorageContract.createdBlock.toString()).gt(((_a = options === null || options === void 0 ? void 0 : options.fromBlock) === null || _a === void 0 ? void 0 : _a.toString()) || this.subStorageContract.createdBlock.toString())) {
-                return [];
-            }
-            return getEventsFromContract(this.subStorageContract, this.subStorageContractFork, event, options);
-        });
-    }
-    getStrategiesSubs(subIds, fromBlock = 'latest') {
-        return __awaiter(this, void 0, void 0, function* () {
-            let options;
-            let web3;
-            if (this.web3Fork && this.subStorageContractFork) {
-                options = {
-                    target: this.subStorageContractFork.address,
-                    abiItem: getAbiItem(this.subStorageContractFork.abi, 'strategiesSubs'),
-                };
-                web3 = this.web3Fork;
-            }
-            else {
-                options = {
-                    target: this.subStorageContract.address,
-                    abiItem: getAbiItem(this.subStorageContract.abi, 'strategiesSubs'),
-                };
-                web3 = this.web3;
-            }
-            const multicallCalls = subIds.map((subId) => (Object.assign(Object.assign({}, options), { params: [subId] })));
-            return multicall(web3, this.chainId, multicallCalls, fromBlock);
-        });
-    }
-    getSubscriptionEventsFromSubStorage(options) {
-        return __awaiter(this, void 0, void 0, function* () {
-            return this.getEventFromSubStorage('Subscribe', options);
-        });
-    }
-    getUpdateDataEventsFromSubStorage(options) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const events = yield this.getEventFromSubStorage('UpdateData', options);
-            /** @dev - Some RPCs sort events differently */
-            return events.sort((a, b) => a.blockNumber - b.blockNumber);
-        });
-    }
-    getParsedSubscriptions(parseData) {
-        return parseStrategiesAutomatedPosition(parseData);
-    }
-    /**
-     * @description Removes expired Limit Order subscriptions
-     */
-    removeExpiredSubscriptions(subscriptions) {
-        return subscriptions.filter((subscription) => {
-            if (!subscription) {
-                return true;
-            }
-            const { protocol, strategy, strategyData } = subscription;
-            if (protocol.id === ProtocolIdentifiers.StrategiesAutomation.Exchange
-                && strategy.strategyId === Strategies.Identifiers.LimitOrder) {
-                return new Dec(strategyData.decoded.triggerData.goodUntil).gt(new Dec(Date.now()).div(1000));
-            }
-            return true;
-        });
-    }
-    _mergeCheck(s, current) {
-        return s.owner === current.owner
-            && s.strategy.strategyId === current.strategy.strategyId
-            && s.protocol.id === current.protocol.id
-            && s.specific.mergeId === current.specific.mergeWithId
-            && (s.protocol.id !== ProtocolIdentifiers.StrategiesAutomation.MakerDAO // reflexer needs to get added if we have it
-                || s.strategyData.decoded.subData.vaultId === current.strategyData.decoded.triggerData.vaultId)
-            && (s.protocol.id !== ProtocolIdentifiers.StrategiesAutomation.CrvUSD // merge only crvUSD leverage management for the same market
-                || s.strategyData.decoded.subData.controller.toLowerCase() === current.strategyData.decoded.triggerData.controller.toLowerCase())
-            && (s.protocol.id !== ProtocolIdentifiers.StrategiesAutomation.MorphoBlue // merge morpho blue with the same marketId
-                || s.strategyData.decoded.triggerData.marketId.toLowerCase() === current.strategyData.decoded.triggerData.marketId.toLowerCase());
-    }
-    mergeSubs(_subscriptions) {
-        const mergeBase = _subscriptions.filter(s => isDefined(s) && isDefined(s.specific.mergeWithId));
-        const mergeExtension = _subscriptions.filter(s => isDefined(s) && isDefined(s.specific.mergeId));
-        let subscriptions = _subscriptions.filter(s => isDefined(s) && isUndefined(s.specific.mergeWithId) && isUndefined(s.specific.mergeId)).map((s) => (Object.assign(Object.assign({}, s), { subIds: [s.subId] })));
-        mergeBase.forEach((current) => {
-            const mergePairIndexSubEnabled = mergeExtension.findIndex(s => this._mergeCheck(s, current) && s.isEnabled === current.isEnabled);
-            const mergePairIndexSubDisabled = mergeExtension.findIndex(s => this._mergeCheck(s, current));
-            const mergePairIndex = mergePairIndexSubEnabled !== -1 ? mergePairIndexSubEnabled : mergePairIndexSubDisabled;
-            if (mergePairIndex !== -1) {
-                const mergePair = mergeExtension[mergePairIndex];
-                mergeExtension.splice(mergePairIndex, 1);
-                subscriptions.push(Object.assign(Object.assign(Object.assign({}, mergePair), current), { 
-                    // @ts-ignore
-                    blockNumber: Dec.max(mergePair.blockNumber, current.blockNumber).toNumber(), subIds: [current.subId, mergePair.subId], isEnabled: mergePair.isEnabled || current.isEnabled, specific: Object.assign(Object.assign({}, mergePair.specific), current.specific) }));
-            }
-            else {
-                subscriptions.push(Object.assign(Object.assign({}, current), { subIds: [current.subId] }));
-            }
-        });
-        if (mergeExtension.length > 0) {
-            console.error('Not all merge-able extensions were used', mergeExtension);
-            subscriptions = [...subscriptions, ...mergeExtension.map((s) => (Object.assign(Object.assign({}, s), { subIds: [s.subId] })))];
-        }
-        return subscriptions;
-    }
-    _getSubscriptions(addresses, options) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const _options = Object.assign(Object.assign({}, addToObjectIf(isDefined(options), options)), addToObjectIf(isDefined(addresses), { filter: { proxy: addresses } }));
-            let subscriptionEvents = (yield this.getSubscriptionEventsFromSubStorage(_options)); // TODO PlaceholderType
-            let subscriptions = [];
-            if (subscriptionEvents) {
-                let strategiesSubs = yield this.getStrategiesSubs(subscriptionEvents.map((e) => e.returnValues.subId), _options.toBlock);
-                if (_options.enabledOnly) {
-                    const filteredSubscriptionEvents = [];
-                    strategiesSubs = strategiesSubs.filter((sub, index) => {
-                        if (sub === null || sub === void 0 ? void 0 : sub.isEnabled)
-                            filteredSubscriptionEvents.push(subscriptionEvents[index]);
-                        return sub === null || sub === void 0 ? void 0 : sub.isEnabled;
-                    });
-                    subscriptionEvents = filteredSubscriptionEvents;
-                }
-                subscriptions = yield Promise.all(strategiesSubs.map((sub, index) => __awaiter(this, void 0, void 0, function* () {
-                    var _a;
-                    let latestUpdate = subscriptionEvents[index].returnValues;
-                    if (latestUpdate.subHash !== (sub === null || sub === void 0 ? void 0 : sub.strategySubHash)) {
-                        const updates = yield this.getUpdateDataEventsFromSubStorage(Object.assign(Object.assign({}, addToObjectIf(!!_options, _options)), { filter: { subId: latestUpdate.subId } }));
-                        latestUpdate = Object.assign(Object.assign({}, latestUpdate), (_a = updates === null || updates === void 0 ? void 0 : updates[updates.length - 1]) === null || _a === void 0 ? void 0 : _a.returnValues);
-                    }
-                    return this.getParsedSubscriptions({
-                        chainId: this.chainId,
-                        blockNumber: subscriptionEvents[index].blockNumber,
-                        subscriptionEventData: latestUpdate,
-                        strategiesSubsData: sub,
-                    });
-                })));
-                if (options === null || options === void 0 ? void 0 : options.mergeSubs) {
-                    subscriptions = this.mergeSubs(subscriptions);
-                }
-            }
-            return _options.unexpiredOnly ? this.removeExpiredSubscriptions(subscriptions) : subscriptions;
-        });
-    }
-    getSubscriptions(options) {
-        return __awaiter(this, void 0, void 0, function* () {
-            return this._getSubscriptions(undefined, options);
-        });
-    }
-    getSubscriptionsFor(addresses, options) {
-        return __awaiter(this, void 0, void 0, function* () {
-            return this._getSubscriptions(addresses, options);
-        });
-    }
-}
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import Dec from 'decimal.js';
+import PromisePool from 'es6-promise-pool';
+import { Strategies, ProtocolIdentifiers } from '../../types/enums';
+import { addToObjectIf, isDefined, isUndefined } from '../../services/utils';
+import { getAbiItem, makeSubStorageContract } from '../../services/contractService';
+import { getEventsFromContract, multicall } from '../../services/ethereumService';
+import { parseStrategiesAutomatedPosition } from '../../services/strategiesService';
+import Automation from './Automation';
+export default class StrategiesAutomation extends Automation {
+    constructor(args) {
+        super();
+        this.web3 = args.provider;
+        this.web3Fork = args.providerFork;
+        this.chainId = args.chainId;
+        this.subStorageContract = makeSubStorageContract(this.web3, this.chainId);
+        this.subStorageContractFork = this.web3Fork ? makeSubStorageContract(this.web3Fork, this.chainId) : null;
+        this.assert();
+    }
+    getEventFromSubStorage(event, options) {
+        var _a;
+        return __awaiter(this, void 0, void 0, function* () {
+            // only used for backfilling, so in case options?.fromBlock in undefined
+            // (just like we omit fromBlock when we call from app when not on fork), we still want to fetch events
+            if (new Dec(this.subStorageContract.createdBlock.toString()).gt(((_a = options === null || options === void 0 ? void 0 : options.fromBlock) === null || _a === void 0 ? void 0 : _a.toString()) || this.subStorageContract.createdBlock.toString())) {
+                return [];
+            }
+            return getEventsFromContract(this.subStorageContract, this.subStorageContractFork, event, options);
+        });
+    }
+    getStrategiesSubs(subIds, fromBlock = 'latest') {
+        return __awaiter(this, void 0, void 0, function* () {
+            let options;
+            let web3;
+            if (this.web3Fork && this.subStorageContractFork) {
+                options = {
+                    target: this.subStorageContractFork.address,
+                    abiItem: getAbiItem(this.subStorageContractFork.abi, 'strategiesSubs'),
+                };
+                web3 = this.web3Fork;
+            }
+            else {
+                options = {
+                    target: this.subStorageContract.address,
+                    abiItem: getAbiItem(this.subStorageContract.abi, 'strategiesSubs'),
+                };
+                web3 = this.web3;
+            }
+            const multicallCalls = subIds.map((subId) => (Object.assign(Object.assign({}, options), { params: [subId] })));
+            return multicall(web3, this.chainId, multicallCalls, fromBlock);
+        });
+    }
+    getSubscriptionEventsFromSubStorage(options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this.getEventFromSubStorage('Subscribe', options);
+        });
+    }
+    getUpdateDataEventsFromSubStorage(options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const events = yield this.getEventFromSubStorage('UpdateData', options);
+            /** @dev - Some RPCs sort events differently */
+            return events.sort((a, b) => a.blockNumber - b.blockNumber);
+        });
+    }
+    getParsedSubscriptions(parseData) {
+        return parseStrategiesAutomatedPosition(parseData);
+    }
+    /**
+     * @description Removes expired Limit Order subscriptions
+     */
+    removeExpiredSubscriptions(subscriptions) {
+        return subscriptions.filter((subscription) => {
+            if (!subscription) {
+                return true;
+            }
+            const { protocol, strategy, strategyData } = subscription;
+            if (protocol.id === ProtocolIdentifiers.StrategiesAutomation.Exchange
+                && strategy.strategyId === Strategies.Identifiers.LimitOrder) {
+                return new Dec(strategyData.decoded.triggerData.goodUntil).gt(new Dec(Date.now()).div(1000));
+            }
+            return true;
+        });
+    }
+    _mergeCheck(s, current) {
+        return s.owner === current.owner
+            && s.strategy.strategyId === current.strategy.strategyId
+            && s.protocol.id === current.protocol.id
+            && s.specific.mergeId === current.specific.mergeWithId
+            && (s.protocol.id !== ProtocolIdentifiers.StrategiesAutomation.MakerDAO // reflexer needs to get added if we have it
+                || s.strategyData.decoded.subData.vaultId === current.strategyData.decoded.triggerData.vaultId)
+            && (s.protocol.id !== ProtocolIdentifiers.StrategiesAutomation.CrvUSD // merge only crvUSD leverage management for the same market
+                || s.strategyData.decoded.subData.controller.toLowerCase() === current.strategyData.decoded.triggerData.controller.toLowerCase())
+            && (s.protocol.id !== ProtocolIdentifiers.StrategiesAutomation.MorphoBlue // merge morpho blue with the same marketId
+                || s.strategyData.decoded.triggerData.marketId.toLowerCase() === current.strategyData.decoded.triggerData.marketId.toLowerCase());
+    }
+    mergeSubs(_subscriptions) {
+        const mergeBase = _subscriptions.filter(s => isDefined(s) && isDefined(s.specific.mergeWithId));
+        const mergeExtension = _subscriptions.filter(s => isDefined(s) && isDefined(s.specific.mergeId));
+        let subscriptions = _subscriptions.filter(s => isDefined(s) && isUndefined(s.specific.mergeWithId) && isUndefined(s.specific.mergeId)).map((s) => (Object.assign(Object.assign({}, s), { subIds: [s.subId] })));
+        mergeBase.forEach((current) => {
+            const mergePairIndexSubEnabled = mergeExtension.findIndex(s => this._mergeCheck(s, current) && s.isEnabled === current.isEnabled);
+            const mergePairIndexSubDisabled = mergeExtension.findIndex(s => this._mergeCheck(s, current));
+            const mergePairIndex = mergePairIndexSubEnabled !== -1 ? mergePairIndexSubEnabled : mergePairIndexSubDisabled;
+            if (mergePairIndex !== -1) {
+                const mergePair = mergeExtension[mergePairIndex];
+                mergeExtension.splice(mergePairIndex, 1);
+                subscriptions.push(Object.assign(Object.assign(Object.assign({}, mergePair), current), { 
+                    // @ts-ignore
+                    blockNumber: Dec.max(mergePair.blockNumber, current.blockNumber).toNumber(), subIds: [current.subId, mergePair.subId], isEnabled: mergePair.isEnabled || current.isEnabled, specific: Object.assign(Object.assign({}, mergePair.specific), current.specific) }));
+            }
+            else {
+                subscriptions.push(Object.assign(Object.assign({}, current), { subIds: [current.subId] }));
+            }
+        });
+        if (mergeExtension.length > 0) {
+            console.error('Not all merge-able extensions were used', mergeExtension);
+            subscriptions = [...subscriptions, ...mergeExtension.map((s) => (Object.assign(Object.assign({}, s), { subIds: [s.subId] })))];
+        }
+        return subscriptions;
+    }
+    _getSubscriptions(addresses, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const _options = Object.assign(Object.assign({}, addToObjectIf(isDefined(options), options)), addToObjectIf(isDefined(addresses), { filter: { proxy: addresses } }));
+            let subscriptionEvents = yield this.getSubscriptionEventsFromSubStorage(_options);
+            let subscriptions = [];
+            if (subscriptionEvents) {
+                let strategiesSubs = yield this.getStrategiesSubs(subscriptionEvents.map((e) => +e.returnValues.subId), _options.toBlock);
+                if (_options.enabledOnly) {
+                    const filteredSubscriptionEvents = [];
+                    strategiesSubs = strategiesSubs.filter((sub, index) => {
+                        if (sub === null || sub === void 0 ? void 0 : sub.isEnabled)
+                            filteredSubscriptionEvents.push(subscriptionEvents[index]);
+                        return sub === null || sub === void 0 ? void 0 : sub.isEnabled;
+                    });
+                    subscriptionEvents = filteredSubscriptionEvents;
+                }
+                const replaceSubWithUpdate = (index) => __awaiter(this, void 0, void 0, function* () {
+                    var _a;
+                    const sub = strategiesSubs[index];
+                    let latestUpdate = subscriptionEvents[index].returnValues;
+                    if (latestUpdate.subHash !== (sub === null || sub === void 0 ? void 0 : sub.strategySubHash)) {
+                        const updates = yield this.getUpdateDataEventsFromSubStorage(Object.assign(Object.assign({}, addToObjectIf(!!_options, _options)), { filter: { subId: latestUpdate.subId } }));
+                        latestUpdate = Object.assign(Object.assign({}, latestUpdate), (_a = updates === null || updates === void 0 ? void 0 : updates[updates.length - 1]) === null || _a === void 0 ? void 0 : _a.returnValues);
+                    }
+                    subscriptions.push(this.getParsedSubscriptions({
+                        chainId: this.chainId,
+                        blockNumber: subscriptionEvents[index].blockNumber,
+                        subscriptionEventData: latestUpdate,
+                        strategiesSubsData: sub,
+                    }));
+                });
+                const generatePromises = function* () {
+                    for (let index = 0; index < strategiesSubs.length; index++) {
+                        yield replaceSubWithUpdate(index);
+                    }
+                };
+                const pool = new PromisePool(generatePromises, 50);
+                yield pool.start();
+                if (options === null || options === void 0 ? void 0 : options.mergeSubs) {
+                    subscriptions = this.mergeSubs(subscriptions);
+                }
+            }
+            return _options.unexpiredOnly ? this.removeExpiredSubscriptions(subscriptions) : subscriptions;
+        });
+    }
+    getSubscriptions(options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this._getSubscriptions(undefined, options);
+        });
+    }
+    getSubscriptionsFor(addresses, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this._getSubscriptions(addresses, options);
+        });
+    }
+}

package/esm/automation/private/StrategiesAutomation.test.d.ts

@@ -1 +1 @@
-import '../../configuration';
+import '../../configuration';