use std::convert::TryInto;
use ergo_lib::ergo_chain_types::EcPoint;
use ergo_lib::ergotree_interpreter::sigma_protocol::private_input::DlogProverInput;
use ergo_lib::wallet;
use wasm_bindgen::prelude::*;
use crate::address::Address;
extern crate derive_more;
use derive_more::{From, Into};
#[wasm_bindgen]
#[derive(PartialEq, Eq, Debug, Clone, From, Into)]
pub struct SecretKey(wallet::secret_key::SecretKey);
#[wasm_bindgen]
impl SecretKey {
pub fn random_dlog() -> SecretKey {
SecretKey(wallet::secret_key::SecretKey::random_dlog())
}
pub fn dlog_from_bytes(bytes: &[u8]) -> Result<SecretKey, JsValue> {
let sized_bytes: &[u8; DlogProverInput::SIZE_BYTES] = bytes.try_into().map_err(|_| {
JsValue::from_str(&format!(
"expected byte array of size {}, found {}",
DlogProverInput::SIZE_BYTES,
bytes.len()
))
})?;
wallet::secret_key::SecretKey::dlog_from_bytes(sized_bytes)
.map(SecretKey)
.ok_or_else(|| JsValue::from_str("failed to parse scalar"))
}
pub fn dht_from_bytes(
secret: &[u8],
g: &[u8],
h: &[u8],
u: &[u8],
v: &[u8],
) -> Result<SecretKey, JsValue> {
let sized_secret: &[u8; DlogProverInput::SIZE_BYTES] = secret.try_into().map_err(|_| {
JsValue::from_str(&format!(
"expected secret byte array of size {}, found {}",
DlogProverInput::SIZE_BYTES,
secret.len()
))
})?;
let sized_g: &[u8; EcPoint::GROUP_SIZE] = g.try_into().map_err(|_| {
JsValue::from_str(&format!(
"expected g byte array of size {}, found {}",
EcPoint::GROUP_SIZE,
g.len()
))
})?;
let sized_h: &[u8; EcPoint::GROUP_SIZE] = h.try_into().map_err(|_| {
JsValue::from_str(&format!(
"expected h byte array of size {}, found {}",
EcPoint::GROUP_SIZE,
h.len()
))
})?;
let sized_u: &[u8; EcPoint::GROUP_SIZE] = u.try_into().map_err(|_| {
JsValue::from_str(&format!(
"expected u byte array of size {}, found {}",
EcPoint::GROUP_SIZE,
u.len()
))
})?;
let sized_v: &[u8; EcPoint::GROUP_SIZE] = v.try_into().map_err(|_| {
JsValue::from_str(&format!(
"expected v byte array of size {}, found {}",
EcPoint::GROUP_SIZE,
v.len()
))
})?;
wallet::secret_key::SecretKey::dht_from_bytes_fields(
sized_secret,
sized_g,
sized_h,
sized_u,
sized_v,
)
.map(SecretKey::from)
.ok_or_else(|| JsValue::from_str("failed to parse Diffie-Hellman tuple"))
}
pub fn get_address(&self) -> Address {
self.0.get_address_from_public_image().into()
}
pub fn from_bytes(bytes: &[u8]) -> Result<SecretKey, JsValue> {
wallet::secret_key::SecretKey::from_bytes(bytes)
.map(SecretKey)
.map_err(|e| JsValue::from_str(&format!("failed to parse SecretKey from bytes: {}", e)))
}
pub fn to_bytes(&self) -> Vec<u8> {
self.0.to_bytes()
}
pub fn from_json(json_str: &str) -> Result<SecretKey, JsValue> {
serde_json::from_str(json_str)
.map(SecretKey)
.map_err(|e| JsValue::from_str(&format!("failed to parse SecretKey from JSON: {}", e)))
}
pub fn to_json(&self) -> Result<String, JsValue> {
serde_json::to_string(&self.0)
.map_err(|e| JsValue::from_str(&format!("failed to encode SecretKey to JSON: {}", e)))
}
}
#[wasm_bindgen]
pub struct SecretKeys(Vec<SecretKey>);
#[wasm_bindgen]
impl SecretKeys {
#[wasm_bindgen(constructor)]
pub fn new() -> Self {
SecretKeys(vec![])
}
pub fn len(&self) -> usize {
self.0.len()
}
pub fn get(&self, index: usize) -> SecretKey {
self.0[index].clone()
}
pub fn add(&mut self, elem: &SecretKey) {
self.0.push(elem.clone());
}
}
impl From<&SecretKeys> for Vec<wallet::secret_key::SecretKey> {
fn from(v: &SecretKeys) -> Self {
v.0.clone().iter().map(|i| i.0.clone()).collect()
}
}
impl From<Vec<wallet::secret_key::SecretKey>> for SecretKeys {
fn from(v: Vec<wallet::secret_key::SecretKey>) -> Self {
SecretKeys(v.into_iter().map(SecretKey::from).collect())
}
}