1use dcbor::prelude::*;
2use anyhow::{ bail, Result, Error };
3
4use crate::{tags, Digest, PrivateKeyBase, PublicKeys, Reference, ReferenceProvider, SigningPrivateKey, SigningPublicKey};
5
6#[derive(Clone, Copy, Eq, PartialEq, Hash)]
23pub struct XID([u8; Self::XID_SIZE]);
24
25impl XID {
26 pub const XID_SIZE: usize = 32;
27
28 pub fn from_data(data: [u8; Self::XID_SIZE]) -> Self {
30 Self(data)
31 }
32
33 pub fn from_data_ref(data: impl AsRef<[u8]>) -> Result<Self> {
37 let data = data.as_ref();
38 if data.len() != Self::XID_SIZE {
39 bail!("Invalid XID size");
40 }
41 let mut arr = [0u8; Self::XID_SIZE];
42 arr.copy_from_slice(data.as_ref());
43 Ok(Self::from_data(arr))
44 }
45
46 pub fn data(&self) -> &[u8; Self::XID_SIZE] {
48 self.into()
49 }
50
51 pub fn new(genesis_key: impl AsRef<SigningPublicKey>) -> Self {
55 let key_cbor_data = genesis_key.as_ref().to_cbor_data();
56 let digest = Digest::from_image(key_cbor_data);
57 Self::from_data(*digest.data())
58 }
59
60 pub fn validate(&self, key: &SigningPublicKey) -> bool {
62 let key_data = key.to_cbor_data();
63 let digest = Digest::from_image(key_data);
64 *digest.data() == self.0
65 }
66
67 pub fn from_hex(hex: impl AsRef<str>) -> Self {
72 Self::from_data_ref(hex::decode(hex.as_ref()).unwrap()).unwrap()
73 }
74
75 pub fn to_hex(&self) -> String {
77 hex::encode(self.0)
78 }
79
80 pub fn short_description(&self) -> String {
82 self.ref_hex_short()
83 }
84
85 pub fn bytewords_identifier(&self, prefix: bool) -> String {
87 self.ref_bytewords(if prefix {Some("๐
ง")} else {None})
88 }
89
90 pub fn bytemoji_identifier(&self, prefix: bool) -> String {
92 self.ref_bytemoji(if prefix {Some("๐
ง")} else {None})
93 }
94}
95
96pub trait XIDProvider {
98 fn xid(&self) -> XID;
100}
101
102impl XIDProvider for XID {
104 fn xid(&self) -> XID {
105 *self
106 }
107}
108
109impl XIDProvider for SigningPublicKey {
111 fn xid(&self) -> XID {
112 XID::new(self)
113 }
114}
115
116impl ReferenceProvider for XID {
118 fn reference(&self) -> Reference {
119 Reference::from_data(*self.data())
120 }
121}
122
123impl<'a> From<&'a XID> for &'a [u8; XID::XID_SIZE] {
125 fn from(value: &'a XID) -> Self {
126 &value.0
127 }
128}
129
130impl<'a> From<&'a XID> for &'a [u8] {
132 fn from(value: &'a XID) -> Self {
133 &value.0
134 }
135}
136
137impl AsRef<[u8]> for XID {
139 fn as_ref(&self) -> &[u8] {
140 &self.0
141 }
142}
143
144impl std::cmp::PartialOrd for XID {
146 fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
147 Some(self.0.cmp(&other.0))
148 }
149}
150
151impl std::cmp::Ord for XID {
153 fn cmp(&self, other: &Self) -> std::cmp::Ordering {
154 self.0.cmp(&other.0)
155 }
156}
157
158impl std::fmt::Debug for XID {
160 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
161 write!(f, "XID({})", hex::encode(self.0))
162 }
163}
164
165impl std::fmt::Display for XID {
167 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
168 write!(f, "XID({})", self.short_description())
169 }
170}
171
172impl CBORTagged for XID {
174 fn cbor_tags() -> Vec<Tag> {
175 tags_for_values(&[tags::TAG_XID])
176 }
177}
178
179impl From<XID> for CBOR {
181 fn from(value: XID) -> Self {
182 value.tagged_cbor()
183 }
184}
185
186impl From<&SigningPublicKey> for XID {
188 fn from(key: &SigningPublicKey) -> Self {
189 Self::new(key)
190 }
191}
192
193impl TryFrom<&SigningPrivateKey> for XID {
195 type Error = Error;
196
197 fn try_from(key: &SigningPrivateKey) -> Result<Self, Self::Error> {
198 Ok(Self::new(&key.public_key()?))
199 }
200}
201
202impl From<&PublicKeys> for XID {
204 fn from(key: &PublicKeys) -> Self {
205 Self::new(key.signing_public_key())
206 }
207}
208
209impl From<&PrivateKeyBase> for XID {
211 fn from(key: &PrivateKeyBase) -> Self {
212 Self::new(key.schnorr_signing_private_key().public_key().unwrap())
213 }
214}
215
216impl CBORTaggedEncodable for XID {
218 fn untagged_cbor(&self) -> CBOR {
219 CBOR::to_byte_string(self.0)
220 }
221}
222
223impl TryFrom<CBOR> for XID {
225 type Error = Error;
226
227 fn try_from(cbor: CBOR) -> Result<Self, Self::Error> {
228 Self::from_tagged_cbor(cbor)
229 }
230}
231
232impl CBORTaggedDecodable for XID {
234 fn from_untagged_cbor(cbor: CBOR) -> Result<Self> {
235 let data = CBOR::try_into_byte_string(cbor)?;
236 Self::from_data_ref(data)
237 }
238}
239
240impl From<XID> for Vec<u8> {
242 fn from(xid: XID) -> Self {
243 xid.0.to_vec()
244 }
245}
246
247#[cfg(test)]
248mod tests {
249 use crate::{ECPrivateKey, SigningPrivateKey};
250
251 use super::*;
252 use bc_ur::prelude::*;
253 use hex_literal::hex;
254 use indoc::indoc;
255
256 #[test]
257 fn test_xid() {
258 crate::register_tags();
259 let xid = XID::from_data_ref(hex!("de2853684ae55803a08b36dd7f4e566649970601927330299fd333f33fecc037")).unwrap();
260 assert_eq!(xid.to_hex(), "de2853684ae55803a08b36dd7f4e566649970601927330299fd333f33fecc037");
261 assert_eq!(xid.short_description(), "de285368");
262 assert_eq!(xid.data(), &hex!("de2853684ae55803a08b36dd7f4e566649970601927330299fd333f33fecc037"));
263 assert_eq!(format!("{:?}", xid), "XID(de2853684ae55803a08b36dd7f4e566649970601927330299fd333f33fecc037)");
264 assert_eq!(format!("{}", xid), "XID(de285368)");
265
266 let xid_string = xid.ur_string();
267 assert_eq!(xid_string, "ur:xid/hdcxuedeguisgevwhdaxnbluenutlbglhfiygamsamadmojkdydtneteeowffhwprtemcaatledk");
268 assert_eq!(XID::from_ur_string(xid_string).unwrap(), xid);
269 assert_eq!(xid.bytewords_identifier(true), "๐
ง URGE DICE GURU IRIS");
270 assert_eq!(xid.bytemoji_identifier(true), "๐
ง ๐ป ๐ป ๐ ๐");
271 }
272
273 #[test]
274 fn test_xid_from_key() {
275 crate::register_tags();
276 let private_key = SigningPrivateKey::new_schnorr(
277 ECPrivateKey::from_data(
278 hex!("322b5c1dd5a17c3481c2297990c85c232ed3c17b52ce9905c6ec5193ad132c36")
279 )
280 );
281 let public_key = private_key.public_key().unwrap();
282
283 let key_cbor = public_key.to_cbor();
284 assert_eq!(key_cbor.diagnostic(), indoc! {"
285 40022(
286 h'e8251dc3a17e0f2c07865ed191139ecbcddcbdd070ec1ff65df5148c7ef4005a'
287 )
288 "}.trim());
289 assert_eq!(key_cbor.hex_annotated(), indoc! {"
290 d9 9c56 # tag(40022) signing-public-key
291 5820 # bytes(32)
292 e8251dc3a17e0f2c07865ed191139ecbcddcbdd070ec1ff65df5148c7ef4005a
293 "}.trim());
294 let key_cbor_data = key_cbor.to_cbor_data();
295 assert_eq!(key_cbor_data, hex!("d99c565820e8251dc3a17e0f2c07865ed191139ecbcddcbdd070ec1ff65df5148c7ef4005a"));
296 let digest = Digest::from_image(&key_cbor_data);
297 assert_eq!(digest.data(), &hex!("d40e0602674df1b732f5e025d04c45f2e74ed1652c5ae1740f6a5502dbbdcd47"));
298
299 let xid = XID::new(&public_key);
300 assert_eq!(format!("{:?}", xid), "XID(d40e0602674df1b732f5e025d04c45f2e74ed1652c5ae1740f6a5502dbbdcd47)");
301 xid.validate(&public_key);
302
303 assert_eq!(format!("{}", xid), "XID(d40e0602)");
304 let reference = xid.reference();
305 assert_eq!(format!("{reference}"), "Reference(d40e0602)");
306
307 assert_eq!(reference.bytewords_identifier(None), "TINY BETA ATOM ALSO");
308 assert_eq!(reference.bytemoji_identifier(None), "๐งฆ ๐คจ ๐ ๐");
309
310 assert_eq!(digest.data(), xid.data());
311 }
312}