// Copyright (c) RoochNetwork
// SPDX-License-Identifier: Apache-2.0

import { toB64 } from '../../types/bcs'

/**
 * Value to be converted into public key.
 */
export type PublicKeyInitData = string | Uint8Array | Iterable<number>

export function bytesEqual(a: Uint8Array, b: Uint8Array) {
  if (a === b) return true

  if (a.length !== b.length) {
    return false
  }

  for (let i = 0; i < a.length; i++) {
    if (a[i] !== b[i]) {
      return false
    }
  }
  return true
}

/**
 * A public key
 */
export abstract class PublicKey {
  /**
   * Checks if two public keys are equal
   */
  equals(publicKey: PublicKey) {
    return bytesEqual(this.toBytes(), publicKey.toBytes())
  }

  /**
   * Return the base-64 representation of the public key
   */
  toBase64() {
    return toB64(this.toBytes())
  }

  /**
   * Return the Rooch representation of the public key encoded in
   * base-64. A Rooch public key is formed by the concatenation
   * of the scheme flag with the raw bytes of the public key
   */
  toRoochPublicKey(): string {
    const bytes = this.toBytes()
    const roochPublicKey = new Uint8Array(bytes.length + 1)
    roochPublicKey.set([this.flag()])
    roochPublicKey.set(bytes, 1)
    return toB64(roochPublicKey)
  }

  /**
   * Return the byte array representation of the public key
   */
  abstract toBytes(): Uint8Array

  /**
   * Return the Rooch address associated with this public key
   */
  abstract toRoochAddress(): string

  /**
   * Return signature scheme flag of the public key
   */
  abstract flag(): number
}