use anyhow::anyhow;
use cid::Cid;
use fil_actors_shared::v8::{
make_empty_map, make_map_with_root_and_bitwidth, FIRST_NON_SINGLETON_ADDR,
};
use fvm_ipld_blockstore::Blockstore;
use fvm_ipld_encoding::tuple::*;
use fvm_ipld_hamt::Error as HamtError;
use fvm_shared::address::{Address, Protocol};
use fvm_shared::{ActorID, HAMT_BIT_WIDTH};
#[derive(Debug, Serialize_tuple, Deserialize_tuple)]
pub struct State {
pub address_map: Cid,
pub next_id: ActorID,
pub network_name: String,
}
impl State {
pub fn new<BS: Blockstore>(store: &BS, network_name: String) -> anyhow::Result<Self> {
let empty_map = make_empty_map::<_, ()>(store, HAMT_BIT_WIDTH)
.flush()
.map_err(|e| anyhow!("failed to create empty map: {}", e))?;
Ok(Self {
address_map: empty_map,
next_id: FIRST_NON_SINGLETON_ADDR,
network_name,
})
}
pub fn map_address_to_new_id<BS: Blockstore>(
&mut self,
store: &BS,
addr: &Address,
) -> Result<ActorID, HamtError> {
let id = self.next_id;
self.next_id += 1;
let mut map = make_map_with_root_and_bitwidth(&self.address_map, store, HAMT_BIT_WIDTH)?;
map.set(addr.to_bytes().into(), id)?;
self.address_map = map.flush()?;
Ok(id)
}
pub fn resolve_address<BS: Blockstore>(
&self,
store: &BS,
addr: &Address,
) -> anyhow::Result<Option<Address>> {
if addr.protocol() == Protocol::ID {
return Ok(Some(*addr));
}
let map = make_map_with_root_and_bitwidth(&self.address_map, store, HAMT_BIT_WIDTH)?;
Ok(map.get(&addr.to_bytes())?.copied().map(Address::new_id))
}
}