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
92
93
94
95
96
97
98
99
100
101
102
103
104
105

use color_eyre::eyre::{ensure, Result};

use crate::errors::GCSObjectURIError;

#[derive(Debug)]
pub struct GCSObjectURI {
    bucket: String,
    object: String,
}

impl GCSObjectURI {
    pub fn new(uri: &str) -> Result<Self> {
        ensure!(&uri[..5] == "gs://", GCSObjectURIError::InvalidPrefix(uri.to_string()));
        let mut parts = uri.splitn(2, "://");
        let _ = parts.next().unwrap();
        let path = parts.next().unwrap();
        let mut parts = path.splitn(2, '/');
        let bucket = parts.next().ok_or(GCSObjectURIError::MissingBucket)?;
        let object = parts.next().ok_or(GCSObjectURIError::MissingObject)?;
        ensure!(!bucket.is_empty(), GCSObjectURIError::MissingBucket);
        ensure!(!object.is_empty(), GCSObjectURIError::MissingObject);
        Ok(Self {
            bucket: bucket.to_string(),
            object: object.to_string(),
        })
    }

    pub(crate) fn bucket(&self) -> &str {
        &self.bucket
    }

    pub(crate) fn object(&self) -> &str {
        &self.object
    }

    pub(crate) fn endpoint(&self) -> String {
        let object = self.object.replace('/', "%2F");
        format!(
            "https://storage.googleapis.com/storage/v1/b/{}/o/{}",
            self.bucket, object
        )
    }

    pub(crate) fn uri(&self) -> String {
        format!("gs://{}/{}", self.bucket, self.object)
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::setup;

    #[test]
    fn test_valid_uri() {
        setup();
        let uri = "gs://my_bucket/my_object";
        let gcs_obj = GCSObjectURI::new(uri).unwrap();
        assert_eq!(gcs_obj.bucket(), "my_bucket");
        assert_eq!(gcs_obj.object(), "my_object");
    }

    #[test]
    fn test_invalid_prefix() {
        setup();
        let uri = "s3://my_bucket/my_object";
        let gcs_obj = GCSObjectURI::new(uri);
        assert!(gcs_obj.is_err());
    }

    #[test]
    fn test_missing_object() {
        setup();
        let uri = "gs://my_bucket/";
        let gcs_obj = GCSObjectURI::new(uri);
        assert!(gcs_obj.is_err());
    }

    #[test]
    fn test_missing_bucket_and_object() {
        setup();
        let uri = "gs://";
        let gcs_obj = GCSObjectURI::new(uri);
        assert!(gcs_obj.is_err());
    }

    #[test]
    fn test_uri_method() {
        setup();
        let uri = "gs://my_bucket/my_object/folder";
        let gcs_obj = GCSObjectURI::new(uri).unwrap();
        assert_eq!(gcs_obj.uri(), "gs://my_bucket/my_object/folder");
    }

    #[test]
    fn test_endpoint_method() {
        setup();
        let uri = "gs://my_bucket/my_object/folder";
        let gcs_obj = GCSObjectURI::new(uri).unwrap();
        assert_eq!(
            gcs_obj.endpoint(),
            "https://storage.googleapis.com/storage/v1/b/my_bucket/o/my_object%2Ffolder"
        );
    }
}