import "../_dnt.polyfills.js";
import { Rune, RunicArgs, ValueRune } from "../rune/mod.js"
import { Chain } from "./ChainRune.js"
import { ConstantRune } from "./ConstantRune.js"
import { PatternRune } from "./PatternRune.js"
import { StorageRune } from "./StorageRune.js"

export class PalletRune<
  out C extends Chain,
  out P extends Chain.PalletName<C>,
  out U,
> extends PatternRune<Chain.Pallet<C, P>, C, U> {
  storage<S extends Chain.StorageName<C, P>, X>(...args: RunicArgs<X, [storageName: S]>) {
    const [storageName] = RunicArgs.resolve(args)
    return this
      .into(ValueRune)
      .access("storage", storageName.as(Rune))
      .into(StorageRune, this.chain)
  }

  constant<K extends Chain.ConstantName<C, P>, X>(...args: RunicArgs<X, [constantName: K]>) {
    const [constantName] = RunicArgs.resolve(args)
    return this
      .into(ValueRune)
      .access("constants", constantName.as(Rune))
      .into(ConstantRune, this.chain)
  }
}