import "../_dnt.polyfills.js";
import * as $ from "../deps/scale.js"
import { Sr25519 as WatTheSr25519 } from "../deps/wat_the_crypto.js"

const ctx = new TextEncoder().encode("substrate")

/** For the time being, this should only be used for testing */
export class Sr25519 {
  keypair: WatTheSr25519
  address

  constructor(keypair: { publicKey: Uint8Array; secretKey: Uint8Array }) {
    this.keypair = keypair instanceof WatTheSr25519
      ? keypair
      : new WatTheSr25519(keypair.publicKey, keypair.secretKey)
    this.address = { type: "Id" as const, value: this.keypair.publicKey }
  }

  static fromSecret(secret: Uint8Array) {
    return new Sr25519(WatTheSr25519.fromSecret(secret))
  }

  static fromSeed64(seed: Uint8Array) {
    return new Sr25519(WatTheSr25519.fromSeed64(seed))
  }

  get publicKey() {
    return this.keypair.publicKey
  }

  get secretKey() {
    return this.keypair.secretKey
  }

  sign = (msg: Uint8Array) => ({
    type: "Sr25519" as const,
    value: this.keypair.sign(ctx, msg),
  })
}

export const $sr25519 = $.instance(
  Sr25519,
  $.tuple($.object(
    $.field("secretKey", $.sizedUint8Array(64)),
    $.field("publicKey", $.sizedUint8Array(32)),
  )),
  (sr25519: Sr25519) => [sr25519.keypair],
)