RPC: ultralight_setNetworkStorage

ethereumjs · Jul 10, 2024 · a8ee2d3 · a8ee2d3
1 parent 9a8f09b
commit a8ee2d3
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/packages/cli/src/rpc/modules/ultralight.ts b/packages/cli/src/rpc/modules/ultralight.ts
@@ -26,6 +26,7 @@ const methods = [
   'ultralight_getNetworkStorageInfo',
   'ultralight_getNetworkDBSize',
   'ultralight_pruneNetworkDB',
+  'ultralight_setNetworkStorage',
 ]
 
 export class ultralight {
@@ -68,6 +69,10 @@ export class ultralight {
       [validators.networkId],
     ])
     this.pruneNetworkDB = middleware(this.pruneNetworkDB.bind(this), 1, [[validators.networkId]])
+    this.setNetworkStorage = middleware(this.setNetworkStorage.bind(this), 2, [
+      [validators.networkId],
+      [validators.megabytes],
+    ])
   }
   async methods() {
     return methods
@@ -196,4 +201,20 @@ export class ultralight {
       radius: '0x' + network.nodeRadius.toString(16),
     }
   }
+  async setNetworkStorage(params: [NetworkId, number]) {
+    const [networkId, maxStorage] = params
+    const network = this._client.networks.get(networkId)
+    if (!network) {
+      throw {
+        code: INTERNAL_ERROR,
+        message: `Invalid network id ${networkId}`,
+      }
+    }
+    await network.prune(maxStorage)
+    return {
+      maxStorage,
+      dbSize: await network.db.size(),
+      radius: '0x' + network.nodeRadius.toString(16),
+    }
+  }
 }