feat: storageRune.size method + sizeTree pattern (#657)

_tasks/dnt.ts

@@ -36,6 +36,9 @@ await Promise.all([
     }, {
       name: "./patterns/identity",
       path: "./patterns/identity.ts",
+    }, {
+      name: "./patterns/size_tree",
+      path: "./patterns/sizeTree.ts",
     }, {
       name: "./server",
       path: "./server/mod.ts",

examples/size_tree.ts

@@ -0,0 +1,6 @@
+import { sizeTree } from "capi/patterns/sizeTree.ts"
+import { chain } from "polkadot_dev/mod.ts"
+
+const result = await sizeTree(chain).run()
+
+console.log(result)

fluent/StorageRune.ts

@@ -19,6 +19,17 @@ export class StorageRune<in out K extends unknown[], out V, out U> extends Rune<
     this.$value = this.pallet.metadata.codec(this.into(ValueRune).access("value"))
   }
 
+  size<X>(...[partialKey, blockHash]: RunicArgs<X, [partialKey?: unknown[], blockHash?: HexHash]>) {
+    return this.pallet.metadata.chain.connection
+      .call(
+        "state_getStorageSize",
+        this.$key.encoded(Rune.resolve(partialKey).map((x) => x ?? [])).map(hex.encode),
+        blockHash,
+      )
+      .unhandle(null)
+      .rehandle(null, () => Rune.constant(undefined))
+  }
+
   entryPageRaw<X>(
     ...[count, partialKey, start, blockHash]: RunicArgs<X, [
       count: number,

patterns/sizeTree.ts

@@ -0,0 +1,23 @@
+import { Chain, ChainRune, HexHash, MetaRune, Rune, RunicArgs, ValueRune } from "../mod.ts"
+
+export function sizeTree<U, C extends Chain, X>(
+  chain: ChainRune<U, C>,
+  ...[blockHash]: RunicArgs<X, [blockHash?: HexHash]>
+) {
+  const metadata = chain.metadata(blockHash)
+  return metadata
+    .into(ValueRune)
+    .map(({ pallets }) =>
+      Rune.rec(Object.fromEntries(pallets.map((pallet) => [
+        pallet.name,
+        Rune.rec(Object.fromEntries(
+          pallet.storage?.entries.map((entry) => [
+            entry.name,
+            metadata.pallet(pallet.name).storage(entry.name).size([], blockHash),
+          ]) || [],
+        )),
+      ])))
+    )
+    .into(MetaRune)
+    .flat()
+}