use ciphersuite::*;
use dkg::{ThresholdKeys, Curves, Secp256k1};

use ethereum_schnorr::PublicKey;

pub(crate) struct KeyGenParams;
impl key_gen::KeyGenParams for KeyGenParams {
  const ID: &'static str = "Ethereum";

  type ExternalNetworkCiphersuite = Secp256k1;

  fn tweak_keys(
    keys: &mut ThresholdKeys<<Self::ExternalNetworkCiphersuite as Curves>::ToweringCurve>,
  ) {
    while PublicKey::new(keys.group_key()).is_none() {
      *keys = keys.clone().offset(<<Secp256k1 as Curves>::ToweringCurve as WrappedGroup>::F::ONE);
    }
  }

  fn encode_key(
    key: <<Self::ExternalNetworkCiphersuite as Curves>::ToweringCurve as WrappedGroup>::G,
  ) -> Vec<u8> {
    PublicKey::new(key).unwrap().eth_repr().to_vec()
  }

  fn decode_key(
    key: &[u8],
  ) -> Option<<<Self::ExternalNetworkCiphersuite as Curves>::ToweringCurve as WrappedGroup>::G> {
    PublicKey::from_eth_repr(key.try_into().ok()?).map(|key| key.point())
  }
}