71 lines
1.7 KiB
Rust
71 lines
1.7 KiB
Rust
|
use atrium_api::types::string::{
|
||
|
|
AtIdentifier,
|
||
|
|
RecordKey,
|
||
|
|
};
|
||
|
|
use regex::Regex;
|
||
|
|
|
||
|
|
pub use atrium_api::types::string::{
|
||
|
|
Nsid,
|
||
|
|
Did,
|
||
|
|
Handle,
|
||
|
|
};
|
||
|
|
|
||
|
|
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>,
|
||
|
|
}
|
||
|
|
|
||
|
|
// 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
|
||
|
|
}
|
||
|
|
}
|