restake 1.0.0 → 2.0.0

package/dist/index.js

@@ -1,13 +1,129 @@
 #!/usr/bin/env node
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+const bee_js_1 = require("@ethersphere/bee-js");
 const cafe_utility_1 = require("cafe-utility");
+const fs_1 = require("fs");
 const process_1 = require("process");
+const viem_1 = require("viem");
+const accounts_1 = require("viem/accounts");
+const chains_1 = require("viem/chains");
+const abi = [
+    {
+        constant: false,
+        inputs: [
+            { name: 'recipient', type: 'address' },
+            { name: 'amount', type: 'uint256' }
+        ],
+        name: 'transfer',
+        outputs: [{ name: '', type: 'bool' }],
+        payable: false,
+        stateMutability: 'nonpayable',
+        type: 'function'
+    }
+];
 const n = cafe_utility_1.Arrays.requireNumberArgument(process_1.argv, 'n');
-const bzz = cafe_utility_1.FixedPointNumber.fromFloat(cafe_utility_1.Arrays.requireNumberArgument(process_1.argv, 'bzz'), 16);
+const bzz = bee_js_1.BZZ.fromFloat(cafe_utility_1.Arrays.requireNumberArgument(process_1.argv, 'bzz'));
 const sleep = cafe_utility_1.Dates.make(cafe_utility_1.Arrays.requireStringArgument(process_1.argv, 'sleep'));
-cafe_utility_1.System.forever(async () => {
-    const port = 1633 + cafe_utility_1.Random.intBetween(0, n - 1);
-    await fetch(`http://localhost:${port}/stake/${bzz.toString()}`, { method: 'POST' });
-}, sleep, console.error);
+const postageBatchId = cafe_utility_1.Arrays.requireStringArgument(process_1.argv, 'postage-batch-id');
+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 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}`);
+    (0, process_1.exit)(1);
+}
+main();
+async function main() {
+    await validatePrivateKeys();
+    runLoop();
+}
+async function validatePrivateKeys() {
+    for (let i = 0; i < n; i++) {
+        const privateKey = privateKeys[i];
+        const port = 1633 + 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;
+        if (!addresses.ethereum.equals(address)) {
+            console.error(`Private key for port ${port} does not match node's Ethereum address`);
+            (0, process_1.exit)(1);
+        }
+    }
+}
+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`);
+            return;
+        }
+        if (cafe_utility_1.Random.chance(1 / 3)) {
+            await bee.depositStake(bzz.toString());
+        }
+        else if (cafe_utility_1.Random.chance(1 / 2)) {
+            await transferBZZ(privateKeys[port - 1633], externalWallet, bzz);
+        }
+        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);
+            await bee.topUpBatch(postageBatch.batchID, amount);
+        }
+    }, sleep, console.error);
+}
+async function transferBZZ(privateKey, targetAddress, amount) {
+    const account = (0, accounts_1.privateKeyToAccount)(ensure0x(privateKey));
+    const client = (0, viem_1.createWalletClient)({
+        chain: chains_1.gnosis,
+        transport: (0, viem_1.http)(jsonRpcUrl)
+    });
+    await client.writeContract({
+        account,
+        abi,
+        address: '0xdbf3ea6f5bee45c02255b2c26a16f300502f68da',
+        functionName: 'transfer',
+        args: [targetAddress, amount.toPLURBigInt()],
+        gas: 100000n,
+        gasPrice: (await getGasPrice()).toWeiBigInt(),
+        type: 'legacy',
+        chain: chains_1.gnosis,
+        nonce: await getTransactionCount(account.address)
+    });
+}
+async function getGasPrice() {
+    const payload = { jsonrpc: '2.0', id: 1, method: 'eth_gasPrice', params: [] };
+    const price = await fetchJsonRpcHexString(payload);
+    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 count = await fetchJsonRpcHexString(payload);
+    return parseInt(count, 16);
+}
+async function fetchJsonRpcHexString(payload) {
+    const response = await fetch(jsonRpcUrl, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify(payload)
+    });
+    const data = await response.json();
+    const object = cafe_utility_1.Types.asObject(data);
+    return cafe_utility_1.Types.asHexString(object.result, { strictPrefix: true, uneven: true });
+}
+function ensure0x(value) {
+    return value.startsWith('0x') ? value : `0x${value}`;
+}
 process.on('unhandledRejection', console.error);
+process.on('uncaughtException', console.error);

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "restake",
-    "version": "1.0.0",
+    "version": "2.0.0",
     "description": "",
     "bin": {
         "restake": "./dist/index.js"
@@ -14,7 +14,9 @@
     "license": "ISC",
     "type": "commonjs",
     "dependencies": {
-        "cafe-utility": "^33.8.0"
+        "@ethersphere/bee-js": "^12.1.0",
+        "cafe-utility": "^33.8.0",
+        "viem": "^2.48.4"
     },
     "devDependencies": {
         "@types/node": "^25.5.2"

package/tsconfig.json

@@ -4,7 +4,7 @@
 
     "compilerOptions": {
         "outDir": "dist",
-        "lib": ["ES2022"],
+        "lib": ["ES2022", "DOM"],
         "module": "CommonJS",
         "target": "ES2022",
         "declaration": false,