Scan *.tiff either

martin/src/args/root.rs

@@ -99,17 +99,17 @@ impl Args {
 
         #[cfg(feature = "pmtiles")]
         if !cli_strings.is_empty() {
-            config.pmtiles = parse_file_args(&mut cli_strings, "pmtiles", true);
+            config.pmtiles = parse_file_args(&mut cli_strings, &["pmtiles"], true);
         }
 
         #[cfg(feature = "mbtiles")]
         if !cli_strings.is_empty() {
-            config.mbtiles = parse_file_args(&mut cli_strings, "mbtiles", false);
+            config.mbtiles = parse_file_args(&mut cli_strings, &["mbtiles"], false);
         }
 
         #[cfg(feature = "cog")]
         if !cli_strings.is_empty() {
-            config.cog = parse_file_args(&mut cli_strings, "tif", false);
+            config.cog = parse_file_args(&mut cli_strings, &["tif", "tiff"], false);
         }
 
         #[cfg(feature = "sprites")]
@@ -125,35 +125,40 @@ impl Args {
     }
 }
 
-#[cfg(any(feature = "pmtiles", feature = "mbtiles"))]
-fn is_url(s: &str, extension: &str) -> bool {
+#[cfg(any(feature = "pmtiles", feature = "mbtiles", feature = "cog"))]
+fn is_url(s: &str, extension: &[&str]) -> bool {
     if s.starts_with("http") {
         if let Ok(url) = url::Url::parse(s) {
             if url.scheme() == "http" || url.scheme() == "https" {
                 if let Some(ext) = url.path().rsplit('.').next() {
-                    return ext == extension;
+                    return extension.contains(&ext);
                 }
             }
         }
     }
     false
 }
 
-#[cfg(any(feature = "pmtiles", feature = "mbtiles"))]
+#[cfg(any(feature = "pmtiles", feature = "mbtiles", feature = "cog"))]
 pub fn parse_file_args<T: crate::file_config::ConfigExtras>(
     cli_strings: &mut Arguments,
-    extension: &str,
+    extensions: &[&str],
     allow_url: bool,
 ) -> FileConfigEnum<T> {
     use crate::args::State::{Ignore, Share, Take};
 
     let paths = cli_strings.process(|s| {
         let path = PathBuf::from(s);
-        if allow_url && is_url(s, extension) {
+        if allow_url && is_url(s, extensions) {
             Take(path)
         } else if path.is_dir() {
             Share(path)
-        } else if path.is_file() && path.extension().map_or(false, |e| e == extension) {
+        } else if path.is_file()
+            && extensions.iter().any(|&expected_ext| {
+                path.extension()
+                    .is_some_and(|actual_ext| actual_ext == expected_ext)
+            })
+        {
             Take(path)
         } else {
             Ignore
@@ -275,4 +280,15 @@ mod tests {
         let bad = vec!["foobar".to_string()];
         assert!(matches!(err, UnrecognizableConnections(v) if v == bad));
     }
+
+    #[test]
+    fn cli_multiple_extensions() {
+        let args = Args::parse_from(["martin", "../tests/fixtures/cog"]);
+
+        let env = FauxEnv::default();
+        let mut config = Config::default();
+        let err = args.merge_into_config(&mut config, &env);
+        println!("{:?}", err);
+        println!("{:?}", config);
+    }
 }

martin/src/config.rs

@@ -178,21 +178,21 @@ impl Config {
         #[cfg(feature = "pmtiles")]
         if !self.pmtiles.is_empty() {
             let cfg = &mut self.pmtiles;
-            let val = crate::file_config::resolve_files(cfg, idr, cache.clone(), "pmtiles");
+            let val = crate::file_config::resolve_files(cfg, idr, cache.clone(), &["pmtiles"]);
             sources.push(Box::pin(val));
         }
 
         #[cfg(feature = "mbtiles")]
         if !self.mbtiles.is_empty() {
             let cfg = &mut self.mbtiles;
-            let val = crate::file_config::resolve_files(cfg, idr, cache.clone(), "mbtiles");
+            let val = crate::file_config::resolve_files(cfg, idr, cache.clone(), &["mbtiles"]);
             sources.push(Box::pin(val));
         }
 
         #[cfg(feature = "cog")]
         if !self.cog.is_empty() {
             let cfg = &mut self.cog;
-            let val = crate::file_config::resolve_files(cfg, idr, cache.clone(), "tif");
+            let val = crate::file_config::resolve_files(cfg, idr, cache.clone(), &["tif", "tiff"]);
             sources.push(Box::pin(val));
         }
 

martin/src/file_config.rs

@@ -235,7 +235,7 @@ pub async fn resolve_files<T: SourceConfigExtras>(
     config: &mut FileConfigEnum<T>,
     idr: &IdResolver,
     cache: OptMainCache,
-    extension: &str,
+    extension: &[&str],
 ) -> MartinResult<TileInfoSources> {
     resolve_int(config, idr, cache, extension)
         .map_err(crate::MartinError::from)
@@ -246,7 +246,7 @@ async fn resolve_int<T: SourceConfigExtras>(
     config: &mut FileConfigEnum<T>,
     idr: &IdResolver,
     cache: OptMainCache,
-    extension: &str,
+    extension: &[&str],
 ) -> FileResult<TileInfoSources> {
     let Some(cfg) = config.extract_file_config(cache)? else {
         return Ok(TileInfoSources::default());
@@ -285,16 +285,20 @@ async fn resolve_int<T: SourceConfigExtras>(
 
     for path in cfg.paths {
         if let Some(url) = parse_url(T::parse_urls(), &path)? {
-            let id = url
-                .path_segments()
-                .and_then(Iterator::last)
-                .and_then(|s| {
-                    // Strip extension and trailing dot, or keep the original string
-                    s.strip_suffix(extension)
-                        .and_then(|s| s.strip_suffix('.'))
-                        .or(Some(s))
-                })
-                .unwrap_or("pmt_web_source");
+            let target_ext = extension.iter().find(|&e| url.to_string().ends_with(e));
+            let id = if let Some(ext) = target_ext {
+                url.path_segments()
+                    .and_then(Iterator::last)
+                    .and_then(|s| {
+                        // Strip extension and trailing dot, or keep the original string
+                        s.strip_suffix(ext)
+                            .and_then(|s| s.strip_suffix('.'))
+                            .or(Some(s))
+                    })
+                    .unwrap_or("web_source")
+            } else {
+                "web_source"
+            };
 
             let id = idr.resolve(id, url.to_string());
             configs.insert(id.clone(), FileConfigSrc::Path(path));
@@ -337,13 +341,17 @@ async fn resolve_int<T: SourceConfigExtras>(
     Ok(results)
 }
 
-fn dir_to_paths(path: &Path, extension: &str) -> Result<Vec<PathBuf>, FileError> {
+fn dir_to_paths(path: &Path, extension: &[&str]) -> Result<Vec<PathBuf>, FileError> {
     Ok(path
         .read_dir()
         .map_err(|e| IoError(e, path.to_path_buf()))?
         .filter_map(Result::ok)
         .filter(|f| {
-            f.path().extension().filter(|e| *e == extension).is_some() && f.path().is_file()
+            f.path()
+                .extension()
+                .filter(|e| extension.iter().any(|x| x == e))
+                .is_some()
+                && f.path().is_file()
         })
         .map(|f| f.path())
         .collect())

tests/config.yaml

@@ -182,7 +182,7 @@ sprites:
 
 cog:
   paths:
-    - tests/fixtures/cog/rgba_u8_nodata.tif
+    - tests/fixtures/cog/rgba_u8_nodata.tiff
   sources:
     cog-src1: tests/fixtures/cog/rgba_u8.tif
     cog-src2: tests/fixtures/cog/rgb_u8.tif

tests/expected/auto/save_config.yaml

@@ -260,7 +260,7 @@ cog:
   sources:
     rgb_u8: tests/fixtures/cog/rgb_u8.tif
     rgba_u8: tests/fixtures/cog/rgba_u8.tif
-    rgba_u8_nodata: tests/fixtures/cog/rgba_u8_nodata.tif
+    rgba_u8_nodata: tests/fixtures/cog/rgba_u8_nodata.tiff
 sprites: tests/fixtures/sprites/src1
 fonts:
 - tests/fixtures/fonts/overpass-mono-regular.ttf

tests/expected/configured/save_config.yaml

@@ -170,7 +170,7 @@ cog:
   sources:
     cog-src1: tests/fixtures/cog/rgba_u8.tif
     cog-src2: tests/fixtures/cog/rgb_u8.tif
-    rgba_u8_nodata: tests/fixtures/cog/rgba_u8_nodata.tif
+    rgba_u8_nodata: tests/fixtures/cog/rgba_u8_nodata.tiff
 sprites:
   paths: tests/fixtures/sprites/src1
   sources: