Atproto, add uri struct and parsing

src/atproto.rs

@@ -1,67 +1,55 @@
-use atrium_api::types::string::RecordKey;
-// use regex::Regex;
+use lazy_regex::regex_captures;
+use core::str::FromStr;
 
-pub use atrium_api::types::string::{
-  Nsid,
-  Did,
-  Handle,
+pub use atrium_api::types::{
+  Collection,
+  string::{
+    Nsid,
+    RecordKey,
+    AtIdentifier as Authority,
+  }
 };
 
-enum Authority {
-  Did(Did),
-  Handle(Handle),
-}
-
-// impl Authority {
-//   pub fn new(authority: String) -> Result<Self, &'static str> {
-//   }
-// }
-
-// This implementation does not support Query or Fragments, and thus follows
-// the following schema: "at://" AUTHORITY [ "/" COLLECTION [ "/" RKEY ] ]
 pub struct Uri {
-  authority: Authority,
-  collection: Option<Nsid>,
-  rkey: Option<RecordKey>,
+  whole: String,
+  // These fields could be useful in the future, 
+  // so I'm leaving the code for them.
+  // authority: Authority,
+  // collection: Option<Nsid>,
+  // rkey: Option<RecordKey>,
+}
+
+impl Uri {
+  pub fn as_str(&self) -> &str {
+    self.whole.as_str()
+  }
+
+  pub fn from_str(uri: String) -> Result<Self, &'static str> {
+    if uri.len() > 8000 {
+      return Err("Uri too long")
+    }
+
+    let Some((
+      whole, unchecked_authority, unchecked_collection, unchecked_rkey
+    )) = regex_captures!(
+      r"/^at:\/\/([\w\.\-_~:]+)(?:\/([\w\.\-_~:]+)(?:)\/([\w\.\-_~:]+))?$/i",
+      &uri,
+    ) else { 
+      return Err("Invalid Uri"); 
+    };
+
+    // This parsing is required, but the values don't need to be used yet.
+    // No compute cost to use them, just storage cost
+    let _authority = Authority::from_str(unchecked_authority)?;
+
+    let _collection = if unchecked_collection.is_empty() { None }
+      else { Some(Nsid::new(unchecked_collection.to_string())?) };
+
+    let _rkey = if unchecked_rkey.is_empty() { None }
+      else { Some(RecordKey::new(unchecked_rkey.to_string())?) };
+
+    // Ok(Uri{ whole: whole.to_string(), authority, collection, rkey })
+    Ok(Uri { whole: whole.to_string() })
+  }
 }
 
-// TODO: Replace super basic URI regex with real uri parsing
-// const URI_REGEX: Regex = Regex::new(
-//     r"/^at:\/\/([\w\.\-_~:]+)(?:\/([\w\.\-_~:]+)(?:)\/([\w\.\-_~:]+))?$/i"
-//   ).expect("valid regex");
-//
-// impl Uri {
-//   pub fn new(uri: String) -> Result<Self, &'static str> {
-//     let Some(captures) = URI_REGEX.captures(&uri) else {
-//       return Err("Invalid Uri");
-//     };
-//     // TODO: Convert authority if its a did or a handle
-//     let Some(Ok(authority)) = captures.get(1).map(|mtch| {
-//       Authority::new(mtch.as_str().to_string())
-//     }) else {
-//       return Err("Invalid Authority")
-//     };
-//     let collection = captures.get(2).map(|mtch| {
-//       Nsid::new(mtch.as_str().to_string())
-//     });
-//     let rkey = captures.get(3).map(|mtch| {
-//       RecordKey::new(mtch.as_str().to_string())
-//     });
-//     Ok(Uri { authority, collection, rkey })
-//   }
-//
-//   pub fn as_string(&self) -> String {
-//     let mut uri = String::from("at://");
-//     uri.push_str(match &self.authority {
-//       Authority::Handle(h) => &*h,
-//       Authority::Did(d) => &*d,
-//     });
-//     if let Some(nsid) = &self.collection {
-//       uri.push_str(&*nsid);
-//     }
-//     if let Some(rkey) = &self.rkey {
-//       uri.push_str(&*rkey);
-//     }
-//     uri
-//   }
-// }