@streamr/node 100.0.0-rc.0

package/dist/src/plugins/operator/ContractFacade.js

@@ -0,0 +1,364 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ContractFacade = exports.parsePartitionFromReviewRequestMetadata = exports.ParseError = exports.VOTE_NO_KICK = exports.VOTE_KICK = void 0;
+const network_contracts_1 = require("@streamr/network-contracts");
+const protocol_1 = require("@streamr/protocol");
+const utils_1 = require("@streamr/utils");
+const ethers_1 = require("ethers");
+const sample_1 = __importDefault(require("lodash/sample"));
+const node_fetch_1 = __importDefault(require("node-fetch"));
+exports.VOTE_KICK = '0x0000000000000000000000000000000000000000000000000000000000000001';
+exports.VOTE_NO_KICK = '0x0000000000000000000000000000000000000000000000000000000000000000';
+class ParseError extends Error {
+    reasonText;
+    constructor(reasonText) {
+        super(`Failed to parse metadata: ${reasonText}`);
+        this.reasonText = reasonText;
+    }
+}
+exports.ParseError = ParseError;
+function parsePartitionFromReviewRequestMetadata(metadataAsString) {
+    if (metadataAsString === undefined) {
+        throw new ParseError('no metadata');
+    }
+    let metadata;
+    try {
+        metadata = JSON.parse(metadataAsString);
+    }
+    catch {
+        throw new ParseError('malformed metadata');
+    }
+    const partition = Number(metadata.partition);
+    if (isNaN(partition)) {
+        throw new ParseError('invalid or missing "partition" field');
+    }
+    try {
+        (0, protocol_1.ensureValidStreamPartitionIndex)(partition);
+    }
+    catch {
+        throw new ParseError('invalid partition numbering');
+    }
+    return partition;
+}
+exports.parsePartitionFromReviewRequestMetadata = parsePartitionFromReviewRequestMetadata;
+const logger = new utils_1.Logger(module);
+class ContractFacade {
+    operatorContract;
+    theGraphClient;
+    config;
+    // for tests
+    constructor(operatorContract, theGraphClient, config) {
+        this.operatorContract = operatorContract;
+        this.theGraphClient = theGraphClient;
+        this.config = config;
+    }
+    static createInstance(config) {
+        return new ContractFacade(new ethers_1.Contract(config.operatorContractAddress, network_contracts_1.operatorABI, config.signer), new utils_1.TheGraphClient({
+            serverUrl: config.theGraphUrl,
+            fetch: node_fetch_1.default,
+            logger
+        }), config);
+    }
+    async writeHeartbeat(nodeDescriptor) {
+        const metadata = JSON.stringify(nodeDescriptor);
+        await (await this.operatorContract.heartbeat(metadata, this.getEthersOverrides())).wait();
+    }
+    async getTimestampOfLastHeartbeat() {
+        const result = await this.theGraphClient.queryEntity({
+            query: `{
+                operator(id: "${this.getOperatorContractAddress()}") {
+                    latestHeartbeatTimestamp
+                }
+            }`
+        });
+        if (result.operator === null || result.operator.latestHeartbeatTimestamp === null) {
+            return undefined;
+        }
+        else {
+            const timestampInSecs = parseInt(result.operator.latestHeartbeatTimestamp);
+            if (isNaN(timestampInSecs)) {
+                throw new Error('Assertion failed: unexpected non-integer latestHeartbeatTimestamp'); // should never happen
+            }
+            return timestampInSecs * 1000;
+        }
+    }
+    getOperatorContractAddress() {
+        return (0, utils_1.toEthereumAddress)(this.operatorContract.address);
+    }
+    async getSponsorshipsOfOperator(operatorAddress) {
+        const createQuery = (lastId, pageSize) => {
+            return {
+                query: `
+                    {
+                        operator(id: "${operatorAddress}") {
+                            stakes(where: {id_gt: "${lastId}"}, first: ${pageSize}) {
+                                id
+                                sponsorship {
+                                    id
+                                    operatorCount
+                                    stream {
+                                        id
+                                    }
+                                }
+                            }
+                        }
+                    }
+                    `
+            };
+        };
+        const parseItems = (response) => {
+            return response.operator?.stakes ?? [];
+        };
+        const queryResult = this.theGraphClient.queryEntities(createQuery, parseItems);
+        const results = [];
+        for await (const stake of queryResult) {
+            results.push({
+                sponsorshipAddress: (0, utils_1.toEthereumAddress)(stake.sponsorship.id),
+                streamId: (0, protocol_1.toStreamID)(stake.sponsorship.stream.id),
+                operatorCount: stake.sponsorship.operatorCount
+            });
+        }
+        return results;
+    }
+    async getExpiredFlags(sponsorships, maxAgeInMs) {
+        const maxFlagStartTime = Math.floor((Date.now() - maxAgeInMs) / 1000);
+        const createQuery = (lastId, pageSize) => {
+            return {
+                query: `
+                {
+                    flags (where : {
+                        id_gt: "${lastId}",
+                        flaggingTimestamp_lt: ${maxFlagStartTime},
+                        result_in: ["waiting", "voting"],
+                        sponsorship_in: ${JSON.stringify(sponsorships)}
+                    }, first: ${pageSize}) {
+                        id
+                        flaggingTimestamp
+                        target {
+                            id
+                        }
+                        sponsorship {
+                            id
+                        }
+                    }
+                }`
+            };
+        };
+        const flagEntities = this.theGraphClient.queryEntities(createQuery);
+        const flags = [];
+        for await (const flag of flagEntities) {
+            flags.push(flag);
+        }
+        return flags;
+    }
+    async getOperatorsInSponsorship(sponsorshipAddress) {
+        const createQuery = (lastId, pageSize) => {
+            return {
+                query: `
+                    {
+                        sponsorship(id: "${sponsorshipAddress}") {
+                            stakes(where: {id_gt: "${lastId}"}, first: ${pageSize}) {
+                                id
+                                operator {
+                                    id
+                                }
+                            }
+                        }
+                    }
+                    `
+            };
+        };
+        const parseItems = (response) => {
+            return response.sponsorship?.stakes ?? [];
+        };
+        const queryResult = this.theGraphClient.queryEntities(createQuery, parseItems);
+        const operatorIds = [];
+        for await (const stake of queryResult) {
+            operatorIds.push((0, utils_1.toEthereumAddress)(stake.operator.id));
+        }
+        return operatorIds;
+    }
+    async flag(sponsorship, operator, partition) {
+        const metadata = JSON.stringify({ partition });
+        await (await this.operatorContract.flag(sponsorship, operator, metadata, this.getEthersOverrides())).wait();
+    }
+    async getRandomOperator() {
+        const latestBlock = await this.operatorContract.provider.getBlockNumber();
+        const operators = await this.getOperatorAddresses(latestBlock);
+        const excluded = this.getOperatorContractAddress();
+        const operatorAddresses = operators.filter((id) => id !== excluded);
+        logger.debug(`Found ${operatorAddresses.length} operators`);
+        return (0, sample_1.default)(operatorAddresses);
+    }
+    /**
+     * Find the sum of earnings in Sponsorships (that the Operator must withdraw before the sum reaches a limit),
+     * SUBJECT TO the constraints, set in the OperatorServiceConfig:
+     *  - only take at most maxSponsorshipsInWithdraw addresses (those with most earnings), or all if undefined
+     *  - only take sponsorships that have more than minSponsorshipEarningsInWithdraw, or all if undefined
+     */
+    async getEarningsOf(operatorContractAddress, minSponsorshipEarningsInWithdraw, maxSponsorshipsInWithdraw) {
+        const operator = new ethers_1.Contract(operatorContractAddress, network_contracts_1.operatorABI, this.config.signer);
+        const minSponsorshipEarningsInWithdrawWei = BigInt(minSponsorshipEarningsInWithdraw ?? 0);
+        const { addresses: allSponsorshipAddresses, earnings, maxAllowedEarnings, } = await operator.getSponsorshipsAndEarnings();
+        const sponsorships = allSponsorshipAddresses
+            .map((address, i) => ({ address, earnings: earnings[i].toBigInt() }))
+            .filter((sponsorship) => sponsorship.earnings >= minSponsorshipEarningsInWithdrawWei)
+            .sort((a, b) => Number(b.earnings - a.earnings)) // TODO: after Node 20, use .toSorted() instead
+            .slice(0, maxSponsorshipsInWithdraw); // take all if maxSponsorshipsInWithdraw is undefined
+        return {
+            sponsorshipAddresses: sponsorships.map((sponsorship) => (0, utils_1.toEthereumAddress)(sponsorship.address)),
+            sumDataWei: sponsorships.reduce((sum, sponsorship) => sum += sponsorship.earnings, 0n),
+            maxAllowedEarningsDataWei: maxAllowedEarnings.toBigInt()
+        };
+    }
+    async getMyEarnings(minSponsorshipEarningsInWithdraw, maxSponsorshipsInWithdraw) {
+        return this.getEarningsOf(this.getOperatorContractAddress(), minSponsorshipEarningsInWithdraw, maxSponsorshipsInWithdraw);
+    }
+    async withdrawMyEarningsFromSponsorships(sponsorshipAddresses) {
+        await (await this.operatorContract.withdrawEarningsFromSponsorships(sponsorshipAddresses, this.getEthersOverrides())).wait();
+    }
+    async triggerWithdraw(targetOperatorAddress, sponsorshipAddresses) {
+        await (await this.operatorContract.triggerAnotherOperatorWithdraw(targetOperatorAddress, sponsorshipAddresses, this.getEthersOverrides())).wait();
+    }
+    async getOperatorAddresses(requiredBlockNumber) {
+        // TODO: use pagination or find a clever efficient way of selecting a random operator (NET-1113)
+        const createQuery = () => {
+            return {
+                query: `
+                    {
+                        operators(first: 1000) {
+                            id
+                        }
+                    }
+                    `
+            };
+        };
+        this.theGraphClient.updateRequiredBlockNumber(requiredBlockNumber);
+        const queryResult = this.theGraphClient.queryEntities(createQuery);
+        const operatorAddresses = [];
+        for await (const operator of queryResult) {
+            operatorAddresses.push((0, utils_1.toEthereumAddress)(operator.id));
+        }
+        return operatorAddresses;
+    }
+    pullStakedStreams(requiredBlockNumber) {
+        const createQuery = (lastId, pageSize) => {
+            return {
+                query: `
+                    {
+                        operator(id: "${this.getOperatorContractAddress()}") {
+                            stakes(where: {id_gt: "${lastId}"}, first: ${pageSize}) {
+                                sponsorship {
+                                    id
+                                    stream {
+                                        id
+                                    }
+                                }
+                            }
+                        }
+                        _meta {
+                            block {
+                            number
+                            }
+                        }
+                    }
+                    `
+            };
+        };
+        const parseItems = (response) => {
+            if (!response.operator) {
+                logger.error('Unable to find operator in The Graph', { operatorContractAddress: this.operatorContract.address });
+                return [];
+            }
+            return response.operator.stakes;
+        };
+        this.theGraphClient.updateRequiredBlockNumber(requiredBlockNumber);
+        return this.theGraphClient.queryEntities(createQuery, parseItems);
+    }
+    async hasOpenFlag(operatorAddress, sponsorshipAddress) {
+        const createQuery = () => {
+            return {
+                query: `
+                    {
+                        flags(where: {
+                            sponsorship: "${sponsorshipAddress}",
+                            target: "${operatorAddress}",
+                            result_in: ["waiting", "voting"]
+                        }) {
+                            id
+                        }
+                    }
+                    `
+            };
+        };
+        const queryResult = this.theGraphClient.queryEntities(createQuery);
+        const flags = await (0, utils_1.collect)(queryResult, 1);
+        if (flags.length > 0) {
+            logger.debug('Found open flag', { flag: flags[0] });
+            return true;
+        }
+        else {
+            return false;
+        }
+    }
+    addReviewRequestListener(listener, abortSignal) {
+        (0, utils_1.addManagedEventListener)(this.operatorContract, 'ReviewRequest', (sponsorship, targetOperator, voteStartTimestampInSecs, voteEndTimestampInSecs, metadataAsString) => {
+            let partition;
+            try {
+                partition = parsePartitionFromReviewRequestMetadata(metadataAsString);
+            }
+            catch (err) {
+                if (err instanceof ParseError) {
+                    logger.warn(`Skip review request (${err.reasonText})`, {
+                        address: this.operatorContract.address,
+                        sponsorship,
+                        targetOperator,
+                    });
+                }
+                else {
+                    logger.warn('Encountered unexpected error', { err });
+                }
+                return;
+            }
+            logger.debug('Receive review request', {
+                address: this.operatorContract.address,
+                sponsorship,
+                targetOperator,
+                partition
+            });
+            listener((0, utils_1.toEthereumAddress)(sponsorship), (0, utils_1.toEthereumAddress)(targetOperator), partition, voteStartTimestampInSecs * 1000, voteEndTimestampInSecs * 1000);
+        }, abortSignal);
+    }
+    async getStreamId(sponsorshipAddress) {
+        const sponsorship = new ethers_1.Contract(sponsorshipAddress, network_contracts_1.sponsorshipABI, this.config.signer);
+        return (0, protocol_1.toStreamID)(await sponsorship.streamId());
+    }
+    async voteOnFlag(sponsorship, targetOperator, kick) {
+        const voteData = kick ? exports.VOTE_KICK : exports.VOTE_NO_KICK;
+        // typical gas cost 99336, but this has shown insufficient sometimes
+        // TODO should we set gasLimit only here, or also for other transactions made by ContractFacade?
+        await (await this.operatorContract.voteOnFlag(sponsorship, targetOperator, voteData, { ...this.getEthersOverrides(), gasLimit: '1000000' })).wait();
+    }
+    async closeFlag(sponsorship, targetOperator) {
+        // voteOnFlag is not used to vote here but to close the expired flag. The vote data gets ignored.
+        // Anyone can call this function at this point.
+        await this.voteOnFlag(sponsorship, targetOperator, false);
+    }
+    addOperatorContractStakeEventListener(eventName, listener) {
+        this.operatorContract.on(eventName, listener);
+    }
+    removeOperatorContractStakeEventListener(eventName, listener) {
+        this.operatorContract.off(eventName, listener);
+    }
+    getProvider() {
+        return this.config.signer.provider;
+    }
+    getEthersOverrides() {
+        return this.config.getEthersOverrides();
+    }
+}
+exports.ContractFacade = ContractFacade;
+//# sourceMappingURL=ContractFacade.js.map

package/dist/src/plugins/operator/ContractFacade.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ContractFacade.js","sourceRoot":"","sources":["../../../../src/plugins/operator/ContractFacade.ts"],"names":[],"mappings":";;;;;;AACA,kEAA+F;AAC/F,gDAAyF;AACzF,0CAOuB;AACvB,mCAA4C;AAC5C,2DAAkC;AAClC,4DAA8B;AAcjB,QAAA,SAAS,GAAG,oEAAoE,CAAA;AAChF,QAAA,YAAY,GAAG,oEAAoE,CAAA;AAEhG,MAAa,UAAW,SAAQ,KAAK;IACjB,UAAU,CAAQ;IAElC,YAAY,UAAkB;QAC1B,KAAK,CAAC,6BAA6B,UAAU,EAAE,CAAC,CAAA;QAChD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAA;IAChC,CAAC;CACJ;AAPD,gCAOC;AAED,SAAgB,uCAAuC,CAAC,gBAAoC;IACxF,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;QACjC,MAAM,IAAI,UAAU,CAAC,aAAa,CAAC,CAAA;IACvC,CAAC;IAED,IAAI,QAAiC,CAAA;IACrC,IAAI,CAAC;QACD,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAA;IAC3C,CAAC;IAAC,MAAM,CAAC;QACL,MAAM,IAAI,UAAU,CAAC,oBAAoB,CAAC,CAAA;IAC9C,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAA;IAC5C,IAAI,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;QACnB,MAAM,IAAI,UAAU,CAAC,sCAAsC,CAAC,CAAA;IAChE,CAAC;IAED,IAAI,CAAC;QACD,IAAA,0CAA+B,EAAC,SAAS,CAAC,CAAA;IAC9C,CAAC;IAAC,MAAM,CAAC;QACL,MAAM,IAAI,UAAU,CAAC,6BAA6B,CAAC,CAAA;IACvD,CAAC;IAED,OAAO,SAAS,CAAA;AACpB,CAAC;AAxBD,0FAwBC;AAUD,MAAM,MAAM,GAAG,IAAI,cAAM,CAAC,MAAM,CAAC,CAAA;AAmBjC,MAAa,cAAc;IAEN,gBAAgB,CAAU;IAC1B,cAAc,CAAgB;IAC9B,MAAM,CAAuB;IAE9C,YAAY;IACZ,YAAY,gBAA0B,EAAE,cAA8B,EAAE,MAA6B;QACjG,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAA;QACxC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAA;QACpC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACxB,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,MAA6B;QAC/C,OAAO,IAAI,cAAc,CACrB,IAAI,iBAAQ,CAAC,MAAM,CAAC,uBAAuB,EAAE,+BAAW,EAAE,MAAM,CAAC,MAAM,CAAwB,EAC/F,IAAI,sBAAc,CAAC;YACf,SAAS,EAAE,MAAM,CAAC,WAAW;YAC7B,KAAK,EAAL,oBAAK;YACL,MAAM;SACT,CAAC,EACF,MAAM,CACT,CAAA;IACL,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,cAAqC;QACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;QAC/C,MAAM,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;IAC7F,CAAC;IAED,KAAK,CAAC,2BAA2B;QAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,CAAY;YAC5D,KAAK,EAAE;gCACa,IAAI,CAAC,0BAA0B,EAAE;;;cAGnD;SACL,CAAC,CAAA;QACF,IAAI,MAAM,CAAC,QAAQ,KAAK,IAAI,IAAI,MAAM,CAAC,QAAQ,CAAC,wBAAwB,KAAK,IAAI,EAAE,CAAC;YAChF,OAAO,SAAS,CAAA;QACpB,CAAC;aAAM,CAAC;YACJ,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAA;YAC1E,IAAI,KAAK,CAAC,eAAe,CAAC,EAAE,CAAC;gBACzB,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAA,CAAC,sBAAsB;YAC/G,CAAC;YACD,OAAO,eAAe,GAAG,IAAI,CAAA;QACjC,CAAC;IACL,CAAC;IAED,0BAA0B;QACtB,OAAO,IAAA,yBAAiB,EAAC,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAA;IAC3D,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,eAAgC;QAW5D,MAAM,WAAW,GAAG,CAAC,MAAc,EAAE,QAAgB,EAAE,EAAE;YACrD,OAAO;gBACH,KAAK,EAAE;;wCAEiB,eAAe;qDACF,MAAM,cAAc,QAAQ;;;;;;;;;;;;qBAY5D;aACR,CAAA;QACL,CAAC,CAAA;QACD,MAAM,UAAU,GAAG,CAAC,QAA4C,EAAW,EAAE;YACzE,OAAO,QAAQ,CAAC,QAAQ,EAAE,MAAM,IAAI,EAAE,CAAA;QAC1C,CAAC,CAAA;QACD,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAQ,WAAW,EAAE,UAAU,CAAC,CAAA;QACrF,MAAM,OAAO,GAAwB,EAAE,CAAA;QACvC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YACpC,OAAO,CAAC,IAAI,CAAC;gBACT,kBAAkB,EAAE,IAAA,yBAAiB,EAAC,KAAK,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC3D,QAAQ,EAAE,IAAA,qBAAU,EAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC;gBACjD,aAAa,EAAE,KAAK,CAAC,WAAW,CAAC,aAAa;aACjD,CAAC,CAAA;QACN,CAAC;QACD,OAAO,OAAO,CAAA;IAClB,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,YAA+B,EAAE,UAAkB;QACrE,MAAM,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,UAAU,CAAC,GAAG,IAAI,CAAC,CAAA;QACrE,MAAM,WAAW,GAAG,CAAC,MAAc,EAAE,QAAgB,EAAE,EAAE;YACrD,OAAO;gBACH,KAAK,EAAE;;;kCAGW,MAAM;gDACQ,gBAAgB;;0CAEtB,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC;gCACtC,QAAQ;;;;;;;;;;kBAUtB;aACL,CAAA;QACL,CAAC,CAAA;QACD,MAAM,YAAY,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAO,WAAW,CAAC,CAAA;QACzE,MAAM,KAAK,GAAW,EAAE,CAAA;QACxB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;YACpC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACpB,CAAC;QACD,OAAO,KAAK,CAAA;IAChB,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,kBAAmC;QAO/D,MAAM,WAAW,GAAG,CAAC,MAAc,EAAE,QAAgB,EAAE,EAAE;YACrD,OAAO;gBACH,KAAK,EAAE;;2CAEoB,kBAAkB;qDACR,MAAM,cAAc,QAAQ;;;;;;;;qBAQ5D;aACR,CAAA;QACL,CAAC,CAAA;QACD,MAAM,UAAU,GAAG,CAAC,QAA+C,EAAY,EAAE;YAC7E,OAAO,QAAQ,CAAC,WAAW,EAAE,MAAM,IAAI,EAAE,CAAA;QAC7C,CAAC,CAAA;QACD,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAQ,WAAW,EAAE,UAAU,CAAC,CAAA;QACrF,MAAM,WAAW,GAAsB,EAAE,CAAA;QACzC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YACpC,WAAW,CAAC,IAAI,CAAC,IAAA,yBAAiB,EAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAA;QAC1D,CAAC;QACD,OAAO,WAAW,CAAA;IACtB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,WAA4B,EAAE,QAAyB,EAAE,SAAiB;QACjF,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC,CAAA;QAC9C,MAAM,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;IAC/G,CAAC;IAED,KAAK,CAAC,iBAAiB;QACnB,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAA;QACzE,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAA;QAC9D,MAAM,QAAQ,GAAG,IAAI,CAAC,0BAA0B,EAAE,CAAA;QAClD,MAAM,iBAAiB,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,QAAQ,CAAC,CAAA;QACnE,MAAM,CAAC,KAAK,CAAC,SAAS,iBAAiB,CAAC,MAAM,YAAY,CAAC,CAAA;QAC3D,OAAO,IAAA,gBAAM,EAAC,iBAAiB,CAAC,CAAA;IACpC,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,aAAa,CACf,uBAAwC,EACxC,gCAAwC,EACxC,yBAAiC;QAEjC,MAAM,QAAQ,GAAG,IAAI,iBAAQ,CAAC,uBAAuB,EAAE,+BAAW,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAwB,CAAA;QAC9G,MAAM,mCAAmC,GAAG,MAAM,CAAC,gCAAgC,IAAI,CAAC,CAAC,CAAA;QACzF,MAAM,EACF,SAAS,EAAE,uBAAuB,EAClC,QAAQ,EACR,kBAAkB,GACrB,GAAG,MAAM,QAAQ,CAAC,0BAA0B,EAAE,CAAA;QAE/C,MAAM,YAAY,GAAG,uBAAuB;aACvC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;aACpE,MAAM,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,QAAQ,IAAI,mCAAmC,CAAC;aACpF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,+CAA+C;aAC/F,KAAK,CAAC,CAAC,EAAE,yBAAyB,CAAC,CAAA,CAAC,qDAAqD;QAE9F,OAAO;YACH,oBAAoB,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAA,yBAAiB,EAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAC/F,UAAU,EAAE,YAAY,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,WAAW,EAAE,EAAE,CAAC,GAAG,IAAI,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC;YACtF,yBAAyB,EAAE,kBAAkB,CAAC,QAAQ,EAAE;SAC3D,CAAA;IACL,CAAC;IAED,KAAK,CAAC,aAAa,CACf,gCAAwC,EACxC,yBAAiC;QAEjC,OAAO,IAAI,CAAC,aAAa,CACrB,IAAI,CAAC,0BAA0B,EAAE,EACjC,gCAAgC,EAChC,yBAAyB,CAC5B,CAAA;IACL,CAAC;IAED,KAAK,CAAC,kCAAkC,CAAC,oBAAuC;QAC5E,MAAM,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,gCAAgC,CAC/D,oBAAoB,EACpB,IAAI,CAAC,kBAAkB,EAAE,CAC5B,CAAC,CAAC,IAAI,EAAE,CAAA;IACb,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,qBAAsC,EAAE,oBAAuC;QACjG,MAAM,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,8BAA8B,CAC7D,qBAAqB,EACrB,oBAAoB,EACpB,IAAI,CAAC,kBAAkB,EAAE,CAC5B,CAAC,CAAC,IAAI,EAAE,CAAA;IACb,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAAC,mBAA2B;QAC1D,gGAAgG;QAChG,MAAM,WAAW,GAAG,GAAG,EAAE;YACrB,OAAO;gBACH,KAAK,EAAE;;;;;;qBAMF;aACR,CAAA;QACL,CAAC,CAAA;QACD,IAAI,CAAC,cAAc,CAAC,yBAAyB,CAAC,mBAAmB,CAAC,CAAA;QAClE,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAiB,WAAW,CAAC,CAAA;QAElF,MAAM,iBAAiB,GAAsB,EAAE,CAAA;QAC/C,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,WAAW,EAAE,CAAC;YACvC,iBAAiB,CAAC,IAAI,CAAC,IAAA,yBAAiB,EAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAA;QAC1D,CAAC;QACD,OAAO,iBAAiB,CAAA;IAC5B,CAAC;IAED,iBAAiB,CAAC,mBAA2B;QACzC,MAAM,WAAW,GAAG,CAAC,MAAc,EAAE,QAAgB,EAAE,EAAE;YACrD,OAAO;gBACH,KAAK,EAAE;;wCAEiB,IAAI,CAAC,0BAA0B,EAAE;qDACpB,MAAM,cAAc,QAAQ;;;;;;;;;;;;;;;qBAe5D;aACR,CAAA;QACL,CAAC,CAAA;QACD,MAAM,UAAU,GAAG,CAAC,QAAa,EAAE,EAAE;YACjC,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACrB,MAAM,CAAC,KAAK,CAAC,sCAAsC,EAAE,EAAE,uBAAuB,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAA;gBAChH,OAAO,EAAE,CAAA;YACb,CAAC;YACD,OAAO,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAA;QACnC,CAAC,CAAA;QACD,IAAI,CAAC,cAAc,CAAC,yBAAyB,CAAC,mBAAmB,CAAC,CAAA;QAClE,OAAO,IAAI,CAAC,cAAc,CAAC,aAAa,CAAsE,WAAW,EAAE,UAAU,CAAC,CAAA;IAC1I,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,eAAgC,EAAE,kBAAmC;QACnF,MAAM,WAAW,GAAG,GAAG,EAAE;YACrB,OAAO;gBACH,KAAK,EAAE;;;4CAGqB,kBAAkB;uCACvB,eAAe;;;;;;qBAMjC;aACR,CAAA;QACL,CAAC,CAAA;QACD,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAiB,WAAW,CAAC,CAAA;QAElF,MAAM,KAAK,GAAG,MAAM,IAAA,eAAO,EAAC,WAAW,EAAE,CAAC,CAAC,CAAA;QAC3C,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnB,MAAM,CAAC,KAAK,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;YACnD,OAAO,IAAI,CAAA;QACf,CAAC;aAAM,CAAC;YACJ,OAAO,KAAK,CAAA;QAChB,CAAC;IACL,CAAC;IAED,wBAAwB,CAAC,QAA+B,EAAE,WAAwB;QAC9E,IAAA,+BAAuB,EACnB,IAAI,CAAC,gBAAuB,EAC5B,eAAe,EACf,CACI,WAAmB,EACnB,cAAsB,EACtB,wBAAgC,EAChC,sBAA8B,EAC9B,gBAAyB,EAC3B,EAAE;YACA,IAAI,SAAiB,CAAA;YACrB,IAAI,CAAC;gBACD,SAAS,GAAG,uCAAuC,CAAC,gBAAgB,CAAC,CAAA;YACzE,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACX,IAAI,GAAG,YAAY,UAAU,EAAE,CAAC;oBAC5B,MAAM,CAAC,IAAI,CAAC,wBAAwB,GAAG,CAAC,UAAU,GAAG,EAAE;wBACnD,OAAO,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO;wBACtC,WAAW;wBACX,cAAc;qBACjB,CAAC,CAAA;gBACN,CAAC;qBAAM,CAAC;oBACJ,MAAM,CAAC,IAAI,CAAC,8BAA8B,EAAE,EAAE,GAAG,EAAE,CAAC,CAAA;gBACxD,CAAC;gBACD,OAAM;YACV,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE;gBACnC,OAAO,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO;gBACtC,WAAW;gBACX,cAAc;gBACd,SAAS;aACZ,CAAC,CAAA;YACF,QAAQ,CACJ,IAAA,yBAAiB,EAAC,WAAW,CAAC,EAC9B,IAAA,yBAAiB,EAAC,cAAc,CAAC,EACjC,SAAS,EACT,wBAAwB,GAAG,IAAI,EAC/B,sBAAsB,GAAG,IAAI,CAChC,CAAA;QACL,CAAC,EACD,WAAW,CACd,CAAA;IACL,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,kBAA0B;QACxC,MAAM,WAAW,GAAG,IAAI,iBAAQ,CAAC,kBAAkB,EAAE,kCAAc,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAA2B,CAAA;QAClH,OAAO,IAAA,qBAAU,EAAC,MAAM,WAAW,CAAC,QAAQ,EAAE,CAAC,CAAA;IACnD,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,WAAmB,EAAE,cAAsB,EAAE,IAAa;QACvE,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,iBAAS,CAAC,CAAC,CAAC,oBAAY,CAAA;QAChD,oEAAoE;QACpE,gGAAgG;QAChG,MAAM,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,UAAU,CACzC,WAAW,EACX,cAAc,EACd,QAAQ,EACR,EAAE,GAAG,IAAI,CAAC,kBAAkB,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CACxD,CAAC,CAAC,IAAI,EAAE,CAAA;IACb,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,WAAmB,EAAE,cAAsB;QACvD,iGAAiG;QACjG,+CAA+C;QAC/C,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,cAAc,EAAE,KAAK,CAAC,CAAA;IAC7D,CAAC;IAED,qCAAqC,CAAC,SAAgC,EAAE,QAA0C;QAC9G,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAA;IACjD,CAAC;IAED,wCAAwC,CAAC,SAAgC,EAAE,QAA0C;QACjH,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAA;IAClD,CAAC;IAED,WAAW;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAA;IACtC,CAAC;IAED,kBAAkB;QACd,OAAO,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAA;IAC3C,CAAC;CACJ;AApZD,wCAoZC"}

package/dist/src/plugins/operator/MaintainTopologyHelper.d.ts

@@ -0,0 +1,23 @@
+import { StreamID } from '@streamr/protocol';
+import { EventEmitter } from 'eventemitter3';
+import { ContractFacade } from './ContractFacade';
+export interface MaintainTopologyHelperEvents {
+    /**
+     * Emitted when staking into a Sponsorship on a stream that we haven't staked on before (in another Sponsorship)
+     */
+    addStakedStreams: (streamIds: StreamID[]) => void;
+    /**
+     * Emitted when un-staked from all Sponsorships for the given stream
+     */
+    removeStakedStream: (streamId: StreamID) => void;
+}
+export declare class MaintainTopologyHelper extends EventEmitter<MaintainTopologyHelperEvents> {
+    private readonly streamIdOfSponsorship;
+    private readonly sponsorshipCountOfStream;
+    private onStakedListener?;
+    private onUnstakedListener?;
+    private readonly contractFacade;
+    constructor(contractFacade: ContractFacade);
+    start(): Promise<void>;
+    stop(): void;
+}

package/dist/src/plugins/operator/MaintainTopologyHelper.js

@@ -0,0 +1,75 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MaintainTopologyHelper = void 0;
+const protocol_1 = require("@streamr/protocol");
+const utils_1 = require("@streamr/utils");
+const eventemitter3_1 = require("eventemitter3");
+const logger = new utils_1.Logger(module);
+class MaintainTopologyHelper extends eventemitter3_1.EventEmitter {
+    streamIdOfSponsorship = new Map();
+    sponsorshipCountOfStream = new Map();
+    onStakedListener;
+    onUnstakedListener;
+    contractFacade;
+    constructor(contractFacade) {
+        super();
+        this.contractFacade = contractFacade;
+    }
+    async start() {
+        const latestBlock = await this.contractFacade.getProvider().getBlockNumber();
+        this.onStakedListener = async (sponsorship) => {
+            logger.info('Receive "Staked" event', { sponsorship });
+            const sponsorshipAddress = (0, utils_1.toEthereumAddress)(sponsorship);
+            const streamId = await this.contractFacade.getStreamId(sponsorshipAddress); // TODO: add catching here
+            if (this.streamIdOfSponsorship.has(sponsorshipAddress)) {
+                logger.debug('Ignore already staked into sponsorship', { sponsorship });
+                return;
+            }
+            this.streamIdOfSponsorship.set(sponsorshipAddress, streamId);
+            const sponsorshipCount = (this.sponsorshipCountOfStream.get(streamId) ?? 0) + 1;
+            this.sponsorshipCountOfStream.set(streamId, sponsorshipCount);
+            if (sponsorshipCount === 1) {
+                this.emit('addStakedStreams', [streamId]);
+            }
+        };
+        this.contractFacade.addOperatorContractStakeEventListener('Staked', this.onStakedListener);
+        this.onUnstakedListener = (sponsorship) => {
+            logger.info('Receive "Unstaked" event', { sponsorship });
+            const sponsorshipAddress = (0, utils_1.toEthereumAddress)(sponsorship);
+            const streamId = this.streamIdOfSponsorship.get(sponsorshipAddress);
+            if (streamId === undefined) {
+                logger.debug('Unable to find streamId for sponsorship', { sponsorshipAddress });
+                return;
+            }
+            this.streamIdOfSponsorship.delete(sponsorshipAddress);
+            const sponsorshipCount = this.sponsorshipCountOfStream.get(streamId) - 1;
+            this.sponsorshipCountOfStream.set(streamId, sponsorshipCount);
+            if (sponsorshipCount === 0) {
+                this.sponsorshipCountOfStream.delete(streamId);
+                this.emit('removeStakedStream', streamId);
+            }
+        };
+        this.contractFacade.addOperatorContractStakeEventListener('Unstaked', this.onUnstakedListener);
+        const queryResult = this.contractFacade.pullStakedStreams(latestBlock);
+        for await (const stake of queryResult) {
+            const sponsorshipId = (0, utils_1.toEthereumAddress)(stake.sponsorship.id);
+            const streamId = (0, protocol_1.toStreamID)(stake.sponsorship.stream.id);
+            if (this.streamIdOfSponsorship.get(sponsorshipId) !== streamId) {
+                this.streamIdOfSponsorship.set(sponsorshipId, streamId);
+                const sponsorshipCount = (this.sponsorshipCountOfStream.get(streamId) || 0) + 1;
+                this.sponsorshipCountOfStream.set(streamId, sponsorshipCount);
+            }
+        }
+        if (this.sponsorshipCountOfStream.size > 0) {
+            const initialStreams = Array.from(this.sponsorshipCountOfStream.keys());
+            this.emit('addStakedStreams', initialStreams);
+        }
+    }
+    stop() {
+        this.contractFacade.removeOperatorContractStakeEventListener('Staked', this.onStakedListener);
+        this.contractFacade.removeOperatorContractStakeEventListener('Unstaked', this.onUnstakedListener);
+        this.removeAllListeners();
+    }
+}
+exports.MaintainTopologyHelper = MaintainTopologyHelper;
+//# sourceMappingURL=MaintainTopologyHelper.js.map

package/dist/src/plugins/operator/MaintainTopologyHelper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MaintainTopologyHelper.js","sourceRoot":"","sources":["../../../../src/plugins/operator/MaintainTopologyHelper.ts"],"names":[],"mappings":";;;AAAA,gDAAwD;AACxD,0CAA2E;AAC3E,iDAA4C;AAG5C,MAAM,MAAM,GAAG,IAAI,cAAM,CAAC,MAAM,CAAC,CAAA;AAcjC,MAAa,sBAAuB,SAAQ,4BAA0C;IAEjE,qBAAqB,GAAmC,IAAI,GAAG,EAAE,CAAA;IACjE,wBAAwB,GAA0B,IAAI,GAAG,EAAE,CAAA;IACpE,gBAAgB,CAAmC;IACnD,kBAAkB,CAAmC;IAC5C,cAAc,CAAgB;IAE/C,YAAY,cAA8B;QACtC,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,cAAc,GAAG,cAAc,CAAA;IACxC,CAAC;IAED,KAAK,CAAC,KAAK;QACP,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,cAAc,EAAE,CAAA;QAE5E,IAAI,CAAC,gBAAgB,GAAG,KAAK,EAAE,WAAmB,EAAE,EAAE;YAClD,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE,EAAE,WAAW,EAAE,CAAC,CAAA;YACtD,MAAM,kBAAkB,GAAG,IAAA,yBAAiB,EAAC,WAAW,CAAC,CAAA;YACzD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAA,CAAC,0BAA0B;YACrG,IAAI,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,CAAC;gBACrD,MAAM,CAAC,KAAK,CAAC,wCAAwC,EAAE,EAAE,WAAW,EAAE,CAAC,CAAA;gBACvE,OAAM;YACV,CAAC;YACD,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAA;YAE5D,MAAM,gBAAgB,GAAG,CAAC,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAA;YAC/E,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAA;YAC7D,IAAI,gBAAgB,KAAK,CAAC,EAAE,CAAC;gBACzB,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAA;YAC7C,CAAC;QACL,CAAC,CAAA;QACD,IAAI,CAAC,cAAc,CAAC,qCAAqC,CAAC,QAAQ,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAC1F,IAAI,CAAC,kBAAkB,GAAG,CAAC,WAAmB,EAAE,EAAE;YAC9C,MAAM,CAAC,IAAI,CAAC,0BAA0B,EAAE,EAAE,WAAW,EAAE,CAAC,CAAA;YACxD,MAAM,kBAAkB,GAAG,IAAA,yBAAiB,EAAC,WAAW,CAAC,CAAA;YACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAA;YACnE,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;gBACzB,MAAM,CAAC,KAAK,CAAC,yCAAyC,EAAE,EAAE,kBAAkB,EAAE,CAAC,CAAA;gBAC/E,OAAM;YACV,CAAC;YACD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAA;YACrD,MAAM,gBAAgB,GAAG,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,QAAQ,CAAE,GAAG,CAAC,CAAA;YACzE,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAA;YAC7D,IAAI,gBAAgB,KAAK,CAAC,EAAE,CAAC;gBACzB,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;gBAC9C,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,QAAQ,CAAC,CAAA;YAC7C,CAAC;QACL,CAAC,CAAA;QACD,IAAI,CAAC,cAAc,CAAC,qCAAqC,CAAC,UAAU,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAA;QAE9F,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAA;QACtE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YACpC,MAAM,aAAa,GAAG,IAAA,yBAAiB,EAAC,KAAK,CAAC,WAAW,CAAC,EAAE,CAAC,CAAA;YAC7D,MAAM,QAAQ,GAAG,IAAA,qBAAU,EAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;YACxD,IAAI,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,aAAa,CAAC,KAAK,QAAQ,EAAE,CAAC;gBAC7D,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAA;gBACvD,MAAM,gBAAgB,GAAG,CAAC,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAA;gBAC/E,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAA;YACjE,CAAC;QACL,CAAC;QACD,IAAI,IAAI,CAAC,wBAAwB,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACzC,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,CAAC,CAAA;YACvE,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,cAAc,CAAC,CAAA;QACjD,CAAC;IACL,CAAC;IAED,IAAI;QACA,IAAI,CAAC,cAAc,CAAC,wCAAwC,CAAC,QAAQ,EAAE,IAAI,CAAC,gBAAiB,CAAC,CAAA;QAC9F,IAAI,CAAC,cAAc,CAAC,wCAAwC,CAAC,UAAU,EAAE,IAAI,CAAC,kBAAmB,CAAC,CAAA;QAClG,IAAI,CAAC,kBAAkB,EAAE,CAAA;IAC7B,CAAC;CACJ;AAxED,wDAwEC"}

package/dist/src/plugins/operator/MaintainTopologyService.d.ts

@@ -0,0 +1,11 @@
+import { StreamrClient } from '@streamr/sdk';
+import { StreamPartAssignments } from './StreamPartAssignments';
+export declare class MaintainTopologyService {
+    private readonly streamrClient;
+    private readonly subscriptions;
+    private readonly concurrencyLimit;
+    constructor(streamrClient: StreamrClient, assignments: StreamPartAssignments);
+    private onAddStakedStreamPart;
+    private onRemoveStakedStreamPart;
+    private concurrencyLimiter;
+}

package/dist/src/plugins/operator/MaintainTopologyService.js

@@ -0,0 +1,57 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MaintainTopologyService = void 0;
+const utils_1 = require("@streamr/utils");
+const protocol_1 = require("@streamr/protocol");
+const p_limit_1 = __importDefault(require("p-limit"));
+const logger = new utils_1.Logger(module);
+class MaintainTopologyService {
+    streamrClient;
+    subscriptions = new Map();
+    concurrencyLimit = (0, p_limit_1.default)(1);
+    constructor(streamrClient, assignments) {
+        this.streamrClient = streamrClient;
+        assignments.on('assigned', this.onAddStakedStreamPart);
+        assignments.on('unassigned', this.onRemoveStakedStreamPart);
+    }
+    onAddStakedStreamPart = this.concurrencyLimiter(async (streamPartId) => {
+        const [id, partition] = protocol_1.StreamPartIDUtils.getStreamIDAndPartition(streamPartId);
+        let subscription;
+        try {
+            logger.info('Join stream partition', { streamPartId });
+            subscription = await this.streamrClient.subscribe({
+                id,
+                partition,
+                raw: true
+            }, () => { });
+        }
+        catch (err) {
+            logger.warn('Failed to join stream partition', { streamPartId, reason: err?.reason });
+            return;
+        }
+        this.subscriptions.set(streamPartId, subscription);
+    });
+    onRemoveStakedStreamPart = this.concurrencyLimiter(async (streamPartId) => {
+        const subscription = this.subscriptions.get(streamPartId);
+        this.subscriptions.delete(streamPartId);
+        try {
+            logger.info('Leave stream partition', { streamPartId });
+            await subscription?.unsubscribe();
+        }
+        catch (err) {
+            logger.warn('Failed to leave stream partition', { streamPartId, reason: err?.reason });
+        }
+    });
+    concurrencyLimiter(fn) {
+        return (streamPartId) => {
+            this.concurrencyLimit(() => fn(streamPartId)).catch((err) => {
+                logger.warn('Encountered error while processing event', { err });
+            });
+        };
+    }
+}
+exports.MaintainTopologyService = MaintainTopologyService;
+//# sourceMappingURL=MaintainTopologyService.js.map

package/dist/src/plugins/operator/MaintainTopologyService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MaintainTopologyService.js","sourceRoot":"","sources":["../../../../src/plugins/operator/MaintainTopologyService.ts"],"names":[],"mappings":";;;;;;AAAA,0CAAuC;AAEvC,gDAAmE;AACnE,sDAA4B;AAG5B,MAAM,MAAM,GAAG,IAAI,cAAM,CAAC,MAAM,CAAC,CAAA;AAEjC,MAAa,uBAAuB;IACf,aAAa,CAAe;IAC5B,aAAa,GAAG,IAAI,GAAG,EAA8B,CAAA;IACrD,gBAAgB,GAAG,IAAA,iBAAM,EAAC,CAAC,CAAC,CAAA;IAE7C,YAAY,aAA4B,EAAE,WAAkC;QACxE,IAAI,CAAC,aAAa,GAAG,aAAa,CAAA;QAClC,WAAW,CAAC,EAAE,CAAC,UAAU,EAAE,IAAI,CAAC,qBAAqB,CAAC,CAAA;QACtD,WAAW,CAAC,EAAE,CAAC,YAAY,EAAE,IAAI,CAAC,wBAAwB,CAAC,CAAA;IAC/D,CAAC;IAEO,qBAAqB,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,YAA0B,EAAiB,EAAE;QACxG,MAAM,CAAC,EAAE,EAAE,SAAS,CAAC,GAAG,4BAAiB,CAAC,uBAAuB,CAAC,YAAY,CAAC,CAAA;QAC/E,IAAI,YAA0B,CAAA;QAC9B,IAAI,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,YAAY,EAAE,CAAC,CAAA;YACtD,YAAY,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC;gBAC9C,EAAE;gBACF,SAAS;gBACT,GAAG,EAAE,IAAI;aACZ,EAAE,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,MAAM,CAAC,IAAI,CAAC,iCAAiC,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAA;YACrF,OAAM;QACV,CAAC;QACD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,YAAY,EAAE,YAAY,CAAC,CAAA;IACtD,CAAC,CAAC,CAAA;IAEM,wBAAwB,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,YAA0B,EAAiB,EAAE;QAC3G,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,YAAY,CAAC,CAAA;QACzD,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,YAAY,CAAC,CAAA;QACvC,IAAI,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE,EAAE,YAAY,EAAE,CAAC,CAAA;YACvD,MAAM,YAAY,EAAE,WAAW,EAAE,CAAA;QACrC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAA;QAC1F,CAAC;IACL,CAAC,CAAC,CAAA;IAEM,kBAAkB,CACtB,EAAiD;QAEjD,OAAO,CAAC,YAAY,EAAE,EAAE;YACpB,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACxD,MAAM,CAAC,IAAI,CAAC,0CAA0C,EAAE,EAAE,GAAG,EAAE,CAAC,CAAA;YACpE,CAAC,CAAC,CAAA;QACN,CAAC,CAAA;IACL,CAAC;CACJ;AAhDD,0DAgDC"}

package/dist/src/plugins/operator/OperatorFleetState.d.ts

@@ -0,0 +1,33 @@
+import { DhtAddress } from '@streamr/dht';
+import { StreamID } from '@streamr/protocol';
+import { EventEmitter } from 'eventemitter3';
+import { NetworkPeerDescriptor, StreamrClient } from '@streamr/sdk';
+export interface OperatorFleetStateEvents {
+    added: (nodeId: DhtAddress) => void;
+    removed: (nodeId: DhtAddress) => void;
+}
+export type CreateOperatorFleetStateFn = (coordinationStreamId: StreamID) => OperatorFleetState;
+export declare class OperatorFleetState extends EventEmitter<OperatorFleetStateEvents> {
+    private readonly streamrClient;
+    private readonly coordinationStreamId;
+    private readonly timeProvider;
+    private readonly pruneAgeInMs;
+    private readonly pruneIntervalInMs;
+    private readonly heartbeatIntervalInMs;
+    private readonly latencyExtraInMs;
+    private readonly warmupPeriodInMs;
+    private readonly latestHeartbeats;
+    private readonly abortController;
+    private readonly ready;
+    private subscription?;
+    static createOperatorFleetStateBuilder(streamrClient: StreamrClient, heartbeatIntervalInMs: number, pruneAgeInMs: number, pruneIntervalInMs: number, latencyExtraInMs: number, warmupPeriodInMs: number, timeProvider?: () => number): CreateOperatorFleetStateFn;
+    private constructor();
+    start(): Promise<void>;
+    waitUntilReady(): Promise<void>;
+    destroy(): Promise<void>;
+    getLeaderNodeId(): DhtAddress | undefined;
+    getNodeIds(): DhtAddress[];
+    getPeerDescriptor(nodeId: DhtAddress): NetworkPeerDescriptor | undefined;
+    private launchOpenReadyGateTimer;
+    private pruneOfflineNodes;
+}