import { decrypt, encrypt } from '../crypto'
import { ensureUint8Array, jsonToUint8Array, Uint8ArrayToJson } from '../data/bin'

export interface Encoder<T = any> {
  encode: (data: T) => Promise<Uint8Array>
  decode: (data: Uint8Array) => Promise<T>
}

export class NoopEncoder<T> implements Encoder<T> {
  async encode(data: any): Promise<Uint8Array> {
    return data
  }

  async decode(data: Uint8Array): Promise<any> {
    return data
  }
}

export class JsonEncoder<T> implements Encoder<T> {
  async encode(data: any): Promise<Uint8Array> {
    return jsonToUint8Array(data)
  }

  async decode(data: Uint8Array): Promise<any> {
    return Uint8ArrayToJson(ensureUint8Array(data))
  }
}

export class CryptoEncoder<T> implements Encoder<T> {
  key: CryptoKey

  constructor(key: CryptoKey) {
    this.key = key
  }

  async encode(data: any): Promise<Uint8Array> {
    const plain = jsonToUint8Array(data)
    return await encrypt(plain, this.key)
  }

  async decode(data: Uint8Array): Promise<any> {
    const plain = await decrypt(ensureUint8Array(data), this.key)
    return Uint8ArrayToJson(ensureUint8Array(plain))
  }
}