feat: pick up env in store (#751)

Author: gadomski

crates/io/src/store.rs

@@ -1,5 +1,5 @@
 use crate::{Format, Readable, Result, Writeable};
-use object_store::{ObjectStore, PutResult, path::Path};
+use object_store::{ObjectStore, ObjectStoreScheme, PutResult, path::Path};
 use stac::Href;
 use std::sync::Arc;
 
@@ -17,17 +17,62 @@ where
     K: AsRef<str>,
     V: Into<String>,
 {
-    let (store, path) = match href.as_ref() {
-        Href::Url(url) => object_store::parse_url_opts(url, options)?,
-        Href::String(s) => {
-            if s.starts_with("/") {
-                object_store::parse_url_opts(&format!("file://{s}").parse()?, options)?
-            } else {
-                let s = std::env::current_dir()?.join(s);
-                object_store::parse_url_opts(&format!("file://{}", s.display()).parse()?, options)?
+    let parse = || -> Result<(Box<dyn ObjectStore>, Path)> {
+        match href.as_ref() {
+            Href::Url(url) => {
+                tracing::debug!("parsing url={url}");
+                // It's technically inefficient to parse it twice, but we're doing this to
+                // then do IO so who cares.
+                let (scheme, path) =
+                    ObjectStoreScheme::parse(url).map_err(object_store::Error::from)?;
+
+                #[cfg(feature = "store-aws")]
+                if matches!(scheme, ObjectStoreScheme::AmazonS3) {
+                    let mut builder = object_store::aws::AmazonS3Builder::from_env();
+                    for (key, value) in options {
+                        builder = builder.with_config(key.as_ref().parse()?, value);
+                    }
+                    return Ok((Box::new(builder.with_url(url.to_string()).build()?), path));
+                }
+
+                #[cfg(feature = "store-azure")]
+                if matches!(scheme, ObjectStoreScheme::AmazonS3) {
+                    let mut builder = object_store::azure::MicrosoftAzureBuilder::from_env();
+                    for (key, value) in options {
+                        builder = builder.with_config(key.as_ref().parse()?, value);
+                    }
+                    return Ok((Box::new(builder.with_url(url.to_string()).build()?), path));
+                }
+
+                #[cfg(feature = "store-gcp")]
+                if matches!(scheme, ObjectStoreScheme::GoogleCloudStorage) {
+                    let mut builder = object_store::gcp::GoogleCloudStorageBuilder::from_env();
+                    for (key, value) in options {
+                        builder = builder.with_config(key.as_ref().parse()?, value);
+                    }
+                    return Ok((Box::new(builder.with_url(url.to_string()).build()?), path));
+                }
+
+                let pair = object_store::parse_url_opts(url, options)?;
+                Ok(pair)
+            }
+            Href::String(s) => {
+                if s.starts_with("/") {
+                    let pair =
+                        object_store::parse_url_opts(&format!("file://{s}").parse()?, options)?;
+                    Ok(pair)
+                } else {
+                    let s = std::env::current_dir()?.join(s);
+                    let pair = object_store::parse_url_opts(
+                        &format!("file://{}", s.display()).parse()?,
+                        options,
+                    )?;
+                    Ok(pair)
+                }
             }
         }
     };
+    let (store, path) = parse()?;
     Ok((store.into(), path))
 }