import "../_dnt.polyfills.js";
import { hex } from "../crypto/mod.js"
import * as $ from "../deps/scale.js"
import { $extrinsic, decodeMetadata, FrameMetadata } from "../frame_metadata/mod.js"
import { Connection } from "../rpc/mod.js"
import { is, Rune, RunicArgs, ValueRune } from "../rune/mod.js"
import { BlockHashRune } from "./BlockHashRune.js"
import { CodecRune } from "./CodecRune.js"
import { ConnectionRune } from "./ConnectionRune.js"
import { ExtrinsicRune } from "./ExtrinsicRune.js"
import { PalletRune } from "./PalletRune.js"

export interface Chain<M extends FrameMetadata = FrameMetadata> {
  connection: Connection
  metadata: M
}

export namespace Chain {
  export type Call<C extends Chain> = $.Output<C["metadata"]["extrinsic"]["call"]>
  export type Address<C extends Chain> = $.Output<C["metadata"]["extrinsic"]["address"]>
  export type Signature<C extends Chain> = $.Output<C["metadata"]["extrinsic"]["signature"]>
  export type Extra<C extends Chain> = $.Output<C["metadata"]["extrinsic"]["extra"]>
  export type Additional<C extends Chain> = $.Output<C["metadata"]["extrinsic"]["additional"]>

  export type Pallets<C extends Chain> = C["metadata"]["pallets"]
  export type PalletName<C extends Chain> = keyof Pallets<C>
  export type Pallet<C extends Chain, P extends PalletName<C>> = Pallets<C>[P]

  export type Constants<C extends Chain, P extends PalletName<C>> = Pallet<C, P>["constants"]
  export type ConstantName<C extends Chain, P extends PalletName<C>> = keyof Constants<C, P>
  export type Constant<C extends Chain, P extends PalletName<C>, K extends ConstantName<C, P>> =
    Constants<C, P>[K]

  export namespace Constant {
    export type Value<C extends Chain, P extends PalletName<C>, K extends ConstantName<C, P>> =
      $.Output<Constant<C, P, K>["codec"]>
  }

  export type StorageEntries<C extends Chain, P extends PalletName<C>> = Pallet<C, P>["storage"]
  export type StorageName<C extends Chain, P extends PalletName<C>> = keyof StorageEntries<C, P>
  export type Storage<C extends Chain, P extends PalletName<C>, S extends StorageName<C, P>> =
    StorageEntries<C, P>[S]

  export namespace Storage {
    export type Key<C extends Chain, P extends PalletName<C>, S extends StorageName<C, P>> =
      $.Output<Storage<C, P, S>["key"]>
    export type PartialKey<C extends Chain, P extends PalletName<C>, S extends StorageName<C, P>> =
      $.Output<Storage<C, P, S>["partialKey"]>
    export type Value<C extends Chain, P extends PalletName<C>, S extends StorageName<C, P>> =
      $.Output<Storage<C, P, S>["value"]>
  }
}

// TODO: do we want to represent the discovery value and conn type within the type system?
export class ChainRune<out C extends Chain, out U> extends Rune<C, U> {
  static from<M extends FrameMetadata>(
    connect: (signal: AbortSignal) => Connection,
    staticMetadata?: M,
  ) {
    const connection = ConnectionRune.from(connect)
    const metadata = staticMetadata ?? Rune
      .fn(hex.decode)
      .call(connection.call("state_getMetadata"))
      .map(decodeMetadata)
    return Rune.object({ connection, metadata }).into(this)
  }

  with(connect: (signal: AbortSignal) => Connection) {
    const connection = ConnectionRune.from(connect)
    return Rune.object({ connection, metadata: this.metadata }).into(ChainRune) as ChainRune<C, U>
  }

  connection = this.into(ValueRune<Chain, U>).access("connection").into(ConnectionRune)

  metadata = this.into(ValueRune).access("metadata")

  $extrinsic = Rune.fn($extrinsic).call(this.metadata).into(CodecRune)

  latestBlockNum = this.connection
    .subscribe("chain_subscribeNewHeads", "chain_unsubscribeNewHeads")
    .access("number")

  latestBlockHash = this.connection
    .call("chain_getBlockHash", this.latestBlockNum)
    .unsafeAs<string>()
    .into(BlockHashRune, this)

  blockHash<X>(...[blockHash]: RunicArgs<X, [blockHash?: string]>) {
    return Rune
      .resolve(blockHash)
      .handle(is(undefined), () => this.connection.call("chain_getFinalizedHead"))
      .into(BlockHashRune, this)
  }

  extrinsic<X>(...args: RunicArgs<X, [call: Chain.Call<C>]>) {
    const [call] = RunicArgs.resolve(args)
    return call.into(ExtrinsicRune, this.as(ChainRune))
  }

  pallet<P extends Chain.PalletName<C>, X>(...[palletName]: RunicArgs<X, [P]>) {
    return this.metadata
      .access("pallets", palletName)
      .unsafeAs<Chain.Pallet<C, P>>()
      .into(PalletRune, this.as(ChainRune))
  }

  addressPrefix() {
    return this
      .pallet("System")
      .constant("SS58Prefix")
      .decoded
  }

  chainVersion = this.connection.call("system_version")
}

export interface AddressPrefixChain extends Chain {
  metadata: FrameMetadata & {
    pallets: {
      System: {
        constants: {
          SS58Prefix: {
            codec: $.Codec<number>
          }
        }
      }
    }
  }
}