restake 2.0.1 → 2.2.0

package/dist/index.js

@@ -29,6 +29,9 @@ const postageBatchId = cafe_utility_1.Arrays.requireStringArgument(process_1.arg
 const externalWallet = cafe_utility_1.Arrays.requireStringArgument(process_1.argv, 'external-wallet');
 const privateKeysPath = cafe_utility_1.Arrays.requireStringArgument(process_1.argv, 'private-keys-path');
 const jsonRpcUrl = cafe_utility_1.Arrays.requireStringArgument(process_1.argv, 'json-rpc-url');
+const telegramToken = cafe_utility_1.Arrays.requireStringArgument(process_1.argv, 'telegram-token');
+const telegramChatId = cafe_utility_1.Arrays.requireStringArgument(process_1.argv, 'telegram-chat-id');
+const BASE_PORT = 1633;
 const privateKeys = (0, fs_1.readFileSync)(privateKeysPath, 'utf-8').split('\n').filter(Boolean);
 if (n !== privateKeys.length) {
     console.error(`Expected ${n} private keys, but got ${privateKeys.length}`);
@@ -42,7 +45,7 @@ async function main() {
 async function validatePrivateKeys() {
     for (let i = 0; i < n; i++) {
         const privateKey = privateKeys[i];
-        const port = 1633 + i;
+        const port = BASE_PORT + i;
         const bee = new bee_js_1.Bee(`http://localhost:${port}`);
         const addresses = await bee.getNodeAddresses();
         const address = (0, accounts_1.privateKeyToAccount)(ensure0x(privateKey)).address;
@@ -54,34 +57,92 @@ async function validatePrivateKeys() {
 }
 function runLoop() {
     cafe_utility_1.System.forever(async () => {
-        const port = 1633 + cafe_utility_1.Random.intBetween(0, n - 1);
-        const bee = new bee_js_1.Bee(`http://localhost:${port}`);
-        const wallet = await bee.getWalletBalance();
-        if (wallet.bzzBalance.lt(bzz)) {
-            console.log(`:${port} balance is ${wallet.bzzBalance.toDecimalString()} BZZ, skipping`);
+        const eligiblePorts = await findEligiblePorts();
+        if (eligiblePorts.length === 0) {
+            console.log('no eligible nodes found, skipping');
             return;
         }
-        if (cafe_utility_1.Random.chance(1 / 3)) {
-            console.log(`:${port} depositing ${bzz.toDecimalString()} BZZ as stake`);
-            await bee.depositStake(bzz);
+        const port = cafe_utility_1.Arrays.pick(eligiblePorts);
+        const bee = new bee_js_1.Bee(`http://localhost:${port}`);
+        if (cafe_utility_1.Random.chance(1 / 4)) {
+            await runAction(port, `stake ${bzz.toDecimalString()} BZZ`, () => stake(bee, port));
+        }
+        else if (cafe_utility_1.Random.chance(1 / 3)) {
+            await runAction(port, `transfer ${bzz.toDecimalString()} BZZ to external wallet`, () => transfer(port));
         }
         else if (cafe_utility_1.Random.chance(1 / 2)) {
-            console.log(`:${port} transferring ${bzz.toDecimalString()} BZZ to external wallet`);
-            await transferBZZ(privateKeys[port - 1633], externalWallet, bzz);
+            await runAction(port, `top up postage batch ${postageBatchId}`, () => topup(bee, port));
         }
         else {
-            const postageBatches = await bee.getGlobalPostageBatches();
-            const postageBatch = postageBatches.find(batch => batch.batchID.toString() === postageBatchId);
-            if (!postageBatch) {
-                console.error(`Postage batch with ID ${postageBatchId} not found`);
-                return;
-            }
-            const { amount } = await bee.calculateTopUpForBzz(postageBatch.depth, bzz);
-            console.log(`:${port} topping up postage batch ${postageBatchId} with ${amount} amount`);
-            await bee.topUpBatch(postageBatch.batchID, amount);
+            await runAction(port, `aid weakest node with ${bzz.toDecimalString()} BZZ`, () => aid(port));
         }
     }, sleep, console.error);
 }
+async function findEligiblePorts() {
+    const eligible = [];
+    for (let i = 0; i < n; i++) {
+        const port = BASE_PORT + i;
+        const bee = new bee_js_1.Bee(`http://localhost:${port}`);
+        const [wallet, redistribution] = await Promise.all([bee.getWalletBalance(), bee.getRedistributionState()]);
+        if (!redistribution.isFrozen && wallet.bzzBalance.gte(bzz)) {
+            eligible.push(port);
+        }
+    }
+    return eligible;
+}
+async function runAction(port, description, action) {
+    try {
+        await action();
+        await sendTelegramMessage(`Port ${port}: ${description} succeeded`);
+    }
+    catch (error) {
+        await sendTelegramMessage(`Port ${port}: ${description} failed: ${error}`);
+    }
+}
+async function stake(bee, port) {
+    console.log(`:${port} depositing ${bzz.toDecimalString()} BZZ as stake`);
+    await bee.depositStake(bzz);
+}
+async function transfer(port) {
+    console.log(`:${port} transferring ${bzz.toDecimalString()} BZZ to external wallet`);
+    await transferBZZ(privateKeys[port - BASE_PORT], externalWallet, bzz);
+}
+async function topup(bee, port) {
+    const postageBatches = await bee.getGlobalPostageBatches();
+    const postageBatch = postageBatches.find(batch => batch.batchID.toString() === postageBatchId);
+    if (!postageBatch) {
+        console.error(`Postage batch with ID ${postageBatchId} not found`);
+        return;
+    }
+    const { amount } = await bee.calculateTopUpForBzz(postageBatch.depth, bzz);
+    console.log(`:${port} topping up postage batch ${postageBatchId} with ${amount} amount`);
+    await bee.topUpBatch(postageBatch.batchID, amount);
+}
+async function aid(port) {
+    const weakestAddress = await findWeakestNodeAddress();
+    const currentAddress = (0, accounts_1.privateKeyToAccount)(ensure0x(privateKeys[port - BASE_PORT])).address;
+    if (weakestAddress && weakestAddress.toLowerCase() !== currentAddress.toLowerCase()) {
+        console.log(`:${port} aiding weakest node ${weakestAddress} with ${bzz.toDecimalString()} BZZ`);
+        await transferBZZ(privateKeys[port - BASE_PORT], weakestAddress, bzz);
+    }
+    else {
+        console.log(`:${port} weakest node is self, skipping aid`);
+    }
+}
+async function findWeakestNodeAddress() {
+    let minStake = null;
+    let weakestAddress = null;
+    for (let i = 0; i < n; i++) {
+        const port = BASE_PORT + i;
+        const bee = new bee_js_1.Bee(`http://localhost:${port}`);
+        const stake = await bee.getStake();
+        if (minStake === null || stake.lt(minStake)) {
+            minStake = stake;
+            weakestAddress = (0, accounts_1.privateKeyToAccount)(ensure0x(privateKeys[i])).address;
+        }
+    }
+    return weakestAddress;
+}
 async function transferBZZ(privateKey, targetAddress, amount) {
     const account = (0, accounts_1.privateKeyToAccount)(ensure0x(privateKey));
     const client = (0, viem_1.createWalletClient)({
@@ -107,11 +168,12 @@ async function getGasPrice() {
     return bee_js_1.DAI.fromWei(BigInt(price));
 }
 async function getTransactionCount(address) {
-    address = address.toLowerCase();
-    if (address.startsWith('0x')) {
-        address = address.slice(2);
-    }
-    const payload = { jsonrpc: '2.0', id: 1, method: 'eth_getTransactionCount', params: [`0x${address}`, 'latest'] };
+    const payload = {
+        jsonrpc: '2.0',
+        id: 1,
+        method: 'eth_getTransactionCount',
+        params: [address.toLowerCase(), 'latest']
+    };
     const count = await fetchJsonRpcHexString(payload);
     return parseInt(count, 16);
 }
@@ -128,5 +190,16 @@ async function fetchJsonRpcHexString(payload) {
 function ensure0x(value) {
     return value.startsWith('0x') ? value : `0x${value}`;
 }
+async function sendTelegramMessage(text) {
+    console.log('Sending Telegram message:', text);
+    await fetch(`https://api.telegram.org/bot${telegramToken}/sendMessage`, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ chat_id: telegramChatId, text }),
+        signal: AbortSignal.timeout(cafe_utility_1.Dates.seconds(10))
+    }).catch(error => {
+        console.error('Failed to send Telegram message:', error);
+    });
+}
 process.on('unhandledRejection', console.error);
 process.on('uncaughtException', console.error);

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "restake",
-    "version": "2.0.1",
+    "version": "2.2.0",
     "description": "",
     "bin": {
         "restake": "./dist/index.js"