import "../../_dnt.polyfills.js";
import { mapEntries } from "../../deps/std/collections/map_entries.js"
import { Chain, ChainRune, MetaRune, Rune, RunicArgs, ValueRune } from "../../mod.js"

export function storageSizes<U, X>(
  chain: ChainRune<Chain, U>,
  ...[blockHash]: RunicArgs<X, [blockHash?: string]>
) {
  return chain.metadata
    .into(ValueRune)
    .map(({ pallets }) =>
      Rune.object(mapEntries(pallets, ([palletName, pallet]) => [
        palletName,
        Rune.object(
          mapEntries(pallet.storage, ([storageName]) => [
            storageName,
            chain.pallet(pallet.name).storage(storageName).size(null, blockHash),
          ]),
        ),
      ]))
    )
    .into(MetaRune)
    .flat()
}