1use std::str::FromStr;
5
6use iota_types::{
7 TypeTag,
8 base_types::{IotaAddress, ObjectID},
9 supported_protocol_versions::Chain,
10};
11use serde::{Deserialize, Serialize};
12
13use crate::Name;
14
15#[derive(Clone, Debug, Serialize, Deserialize, Eq, PartialEq)]
16#[serde(rename_all = "kebab-case")]
17pub struct IotaNamesConfig {
18 pub package_address: IotaAddress,
20 pub object_id: ObjectID,
22 pub payments_package_address: IotaAddress,
24 pub registry_id: ObjectID,
26 pub reverse_registry_id: ObjectID,
28}
29
30impl Default for IotaNamesConfig {
31 fn default() -> Self {
32 Self::devnet()
35 }
36}
37
38impl IotaNamesConfig {
39 pub fn new(
40 package_address: IotaAddress,
41 object_id: ObjectID,
42 payments_package_address: IotaAddress,
43 registry_id: ObjectID,
44 reverse_registry_id: ObjectID,
45 ) -> Self {
46 Self {
47 package_address,
48 object_id,
49 payments_package_address,
50 registry_id,
51 reverse_registry_id,
52 }
53 }
54
55 pub fn from_env() -> anyhow::Result<Self> {
56 Ok(Self::new(
57 std::env::var("IOTA_NAMES_PACKAGE_ADDRESS")?.parse()?,
58 std::env::var("IOTA_NAMES_OBJECT_ID")?.parse()?,
59 std::env::var("IOTA_NAMES_PAYMENTS_PACKAGE_ADDRESS")?.parse()?,
60 std::env::var("IOTA_NAMES_REGISTRY_ID")?.parse()?,
61 std::env::var("IOTA_NAMES_REVERSE_REGISTRY_ID")?.parse()?,
62 ))
63 }
64
65 pub fn from_chain(chain: &Chain) -> Self {
66 match chain {
67 Chain::Mainnet => todo!("https://github.com/iotaledger/iota/issues/6532"),
68 Chain::Testnet => todo!("https://github.com/iotaledger/iota/issues/6531"),
69 Chain::Unknown => IotaNamesConfig::devnet(),
70 }
71 }
72
73 pub fn record_field_id(&self, name: &Name) -> ObjectID {
74 let name_type_tag = Name::type_(self.package_address);
75 let name_bytes = bcs::to_bytes(name).unwrap();
76
77 iota_types::dynamic_field::derive_dynamic_field_id(
78 self.registry_id,
79 &TypeTag::Struct(Box::new(name_type_tag)),
80 &name_bytes,
81 )
82 .unwrap()
83 }
84
85 pub fn reverse_record_field_id(&self, address: &IotaAddress) -> ObjectID {
86 iota_types::dynamic_field::derive_dynamic_field_id(
87 self.reverse_registry_id,
88 &TypeTag::Address,
89 address.as_ref(),
90 )
91 .unwrap()
92 }
93
94 pub fn devnet() -> Self {
100 const PACKAGE_ADDRESS: &str =
101 "0xb9d617f24c84826bf660a2f4031951678cc80c264aebc4413459fb2a95ada9ba";
102 const OBJECT_ID: &str =
103 "0x07c59b37bd7d036bf78fa30561a2ab9f7a970837487656ec29466e817f879342";
104 const PAYMENTS_PACKAGE_ADDRESS: &str =
105 "0x98b9b33b7c2347a8f4e8b8716fb4c7e6e1af846ec2ea063a47bba81ffe03b440";
106 const REGISTRY_ID: &str =
107 "0xe00b2f2400c33b4dbd3081c4dcf2e289d0544caba23a3d130b264bd756403c07";
108 const REVERSE_REGISTRY_ID: &str =
109 "0x1c1da17843cc453ad4079b05ce55e103b7a8cdd4db6ab42dc367b47ed6d8994d";
110
111 let package_address = IotaAddress::from_str(PACKAGE_ADDRESS).unwrap();
112 let object_id = ObjectID::from_str(OBJECT_ID).unwrap();
113 let payments_package_address = IotaAddress::from_str(PAYMENTS_PACKAGE_ADDRESS).unwrap();
114 let registry_id = ObjectID::from_str(REGISTRY_ID).unwrap();
115 let reverse_registry_id = ObjectID::from_str(REVERSE_REGISTRY_ID).unwrap();
116
117 Self::new(
118 package_address,
119 object_id,
120 payments_package_address,
121 registry_id,
122 reverse_registry_id,
123 )
124 }
125}