quilt-rs 0.5.6

Rust library for accessing Quilt data packages.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

use aws_sdk_s3::primitives::ByteStream;
use tokio::io::AsyncReadExt;

#[derive(Clone, Debug, PartialEq, Eq)]
pub struct S3Uri {
    pub bucket: String,
    pub key: String,
    pub version: Option<String>,
}

impl S3Uri {
    pub async fn get_contents(&self) -> Result<String, String> {
        get_object_contents(self).await
    }

    pub async fn put_contents(&self, contents: impl Into<ByteStream>) -> Result<(), String> {
        put_object_contents(self, contents).await
    }
}

pub async fn get_object_bytes(uri: &S3Uri) -> Result<Vec<u8>, String> {
    // real impl
    let client = crate::s3_utils::get_client_for_bucket(&uri.bucket).await?;

    let result = client.get_object().bucket(&uri.bucket).key(&uri.key);

    let result = match &uri.version {
        Some(version) => result.version_id(version),
        None => result,
    };

    let result = result.send().await.map_err(|err| {
        err.into_service_error()
            .meta()
            .message()
            .unwrap_or("failed to download s3 object")
            .to_string()
    })?;

    let mut contents = Vec::new();

    result
        .body
        .into_async_read()
        .read_to_end(&mut contents)
        .await
        .map_err(|err| err.to_string())?;

    Ok(contents)

    // TODO: fake impl
}

pub async fn get_object_contents(uri: &S3Uri) -> Result<String, String> {
    let bytes = get_object_bytes(uri).await?;
    String::from_utf8(bytes).map_err(|err| err.to_string())
}

pub async fn put_object_contents(
    uri: &S3Uri,
    contents: impl Into<ByteStream>,
) -> Result<(), String> {
    let client = crate::s3_utils::get_client_for_bucket(&uri.bucket).await?;
    client
        .put_object()
        .bucket(&uri.bucket)
        .key(&uri.key)
        .body(contents.into())
        .send()
        .await
        .map_err(|err| {
            err.into_service_error()
                .meta()
                .message()
                .unwrap_or("failed to upload s3 object")
                .to_string()
        })?;

    Ok(())
}

// pub type MemoryBuckets = HashMap<String, MemoryFS>;
//
// pub struct FakeS3Storage<'a> {
//     buckets: &'a MemoryBuckets,
// }
//
// async fn get_object_contents(uri: &S3Uri) -> Result<String, String> {
//     // TODO: support versioning?
//     self.buckets.get(&uri.bucket).ok_or("bucket not found")?.get(&uri.key).ok_or(String::from("key not found")).cloned()
// }