aljaz/pubky-core
1
0
Fork 0
mirror of https://github.com/aljazceru/pubky-core.git synced 2025-12-31 12:54:35 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat(pubky): enable get and put methods

Browse Source
This commit is contained in:
nazeh
2024-07-27 19:31:26 +03:00
parent ba50429b7a
commit c40ba8390b
2 changed files with 9 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pubky/src/client.rs
View File

@@ -1,6 +1,6 @@
mod auth;
mod pkarr;
// mod public;
mod public;
use std::{collections::HashMap, fmt::format, time::Duration};

32
pubky/src/client/public.rs
View File

@@ -5,41 +5,31 @@ use pkarr::PublicKey;
use super::{PubkyClient, Result};
impl PubkyClient {
pub fn put(&self, pubky: &PublicKey, path: &str, content: &[u8]) -> Result<()> {
pub async fn put(&self, pubky: &PublicKey, path: &str, content: &[u8]) -> Result<()> {
let path = normalize_path(path);
let (_, mut url) = self.resolve_pubky_homeserver(pubky)?;
let (_, mut url) = self.resolve_pubky_homeserver(pubky).await?;
url.set_path(&format!("/{pubky}/{path}"));
self.request(super::HttpMethod::Put, &url)
.send_bytes(content)?;
self.http.put(url).body(content.to_owned()).send().await?;
Ok(())
}
pub fn get(&self, pubky: &PublicKey, path: &str) -> Result<Bytes> {
pub async fn get(&self, pubky: &PublicKey, path: &str) -> Result<Bytes> {
let path = normalize_path(path);
let (_, mut url) = self.resolve_pubky_homeserver(pubky)?;
let (_, mut url) = self.resolve_pubky_homeserver(pubky).await?;
url.set_path(&format!("/{pubky}/{path}"));
let response = self.request(super::HttpMethod::Get, &url).call()?;
let len = response
.header("Content-Length")
.and_then(|s| s.parse::<u64>().ok())
// TODO: return an error in case content-length header is missing
.unwrap_or(0);
let response = self.http.get(url).send().await?;
// TODO: bail on too large files.
let bytes = response.bytes().await?;
let mut bytes = vec![0; len as usize];
response.into_reader().read_exact(&mut bytes);
Ok(bytes.into())
Ok(bytes)
}
}
@@ -86,12 +76,6 @@ mod tests {
.put(&keypair.public_key(), "/pub/foo.txt", &[0, 1, 2, 3, 4])
.await;
if let Err(Error::Ureq(ureqerror)) = response {
if let Some(r) = ureqerror.into_response() {
dbg!(r.into_string());
}
}
let response = client
.get(&keypair.public_key(), "/pub/foo.txt")
.await