use super::*;
use crate::prelude::*;
use crate::shim::address::Payload;
use bls_signatures::{PublicKey as BlsPublicKey, Serialize as _};
impl StateManager {
pub fn get_bls_public_key(
db: &(impl Blockstore + ShallowClone),
addr: Address,
state_cid: Cid,
) -> Result<BlsPublicKey, Error> {
let state =
StateTree::new_from_root(db, &state_cid).map_err(|e| Error::Other(e.to_string()))?;
let kaddr = state
.resolve_to_deterministic_address(db, addr)
.context("Failed to resolve key address")?;
match kaddr.into_payload() {
Payload::BLS(key) => BlsPublicKey::from_bytes(&key)
.context("Failed to construct bls public key")
.map_err(Error::from),
_ => Err(Error::state(
"Address must be BLS address to load bls public key",
)),
}
}
pub fn lookup_id(&self, addr: &Address, ts: &Tipset) -> Result<Option<Address>, Error> {
let state_tree =
StateTree::new_from_root(self.db(), ts.parent_state()).map_err(|e| format!("{e:?}"))?;
Ok(state_tree
.lookup_id(addr)
.map_err(|e| Error::Other(e.to_string()))?
.map(Address::new_id))
}
pub fn lookup_required_id(&self, addr: &Address, ts: &Tipset) -> Result<Address, Error> {
self.lookup_id(addr, ts)?
.ok_or_else(|| Error::Other(format!("Failed to lookup the id address {addr}")))
}
pub async fn resolve_to_deterministic_address(
&self,
address: Address,
ts: &Tipset,
) -> anyhow::Result<Address> {
use crate::shim::address::Protocol::*;
match address.protocol() {
BLS | Secp256k1 | Delegated => Ok(address),
Actor => anyhow::bail!("cannot resolve actor address to key address"),
ID => {
let id = address.id()?;
self.id_to_deterministic_address_cache
.get_or_insert_async(&id, async {
let resolved = if let Ok(state) =
StateTree::new_from_root(self.db(), ts.parent_state())
&& let Ok(address) = state
.resolve_to_deterministic_address(self.chain_store().db(), address)
{
address
} else {
let TipsetState { state_root, .. } = self.load_tipset_state(ts).await?;
let state = StateTree::new_from_root(self.db(), &state_root)?;
state.resolve_to_deterministic_address(
self.chain_store().db(),
address,
)?
};
Ok(resolved)
})
.await
}
}
}
}