oxigdal-cli 0.1.4

Command-line interface for OxiGDAL geospatial operations
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

//! Utility modules for CLI operations

pub mod cloud;
pub mod creation_options;
pub mod parallel;
pub mod progress;
pub mod raster;
pub mod vector;

use std::path::Path;

/// Detect file format from extension
pub fn detect_format(path: &Path) -> Option<&'static str> {
    path.extension()
        .and_then(|ext| ext.to_str())
        .and_then(|ext| match ext.to_lowercase().as_str() {
            "tif" | "tiff" => Some("GeoTIFF"),
            "json" | "geojson" => Some("GeoJSON"),
            "shp" => Some("Shapefile"),
            "fgb" => Some("FlatGeobuf"),
            "parquet" | "geoparquet" => Some("GeoParquet"),
            "zarr" => Some("Zarr"),
            "gpkg" => Some("GeoPackage"),
            "jp2" => Some("JPEG2000"),
            "copc" => Some("COPC"),
            "pmtiles" => Some("PMTiles"),
            "mbtiles" => Some("MBTiles"),
            _ => None,
        })
}

/// Format file size for human-readable output
pub fn format_size(bytes: u64) -> String {
    const UNITS: &[&str] = &["B", "KB", "MB", "GB", "TB"];
    let mut size = bytes as f64;
    let mut unit_index = 0;

    while size >= 1024.0 && unit_index < UNITS.len() - 1 {
        size /= 1024.0;
        unit_index += 1;
    }

    if unit_index == 0 {
        format!("{} {}", bytes, UNITS[unit_index])
    } else {
        format!("{:.2} {}", size, UNITS[unit_index])
    }
}

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

    #[test]
    fn test_detect_format() {
        assert_eq!(detect_format(Path::new("test.tif")), Some("GeoTIFF"));
        assert_eq!(detect_format(Path::new("test.tiff")), Some("GeoTIFF"));
        assert_eq!(detect_format(Path::new("test.geojson")), Some("GeoJSON"));
        assert_eq!(detect_format(Path::new("test.json")), Some("GeoJSON"));
        assert_eq!(detect_format(Path::new("test.shp")), Some("Shapefile"));
        assert_eq!(detect_format(Path::new("test.fgb")), Some("FlatGeobuf"));
        assert_eq!(detect_format(Path::new("test.parquet")), Some("GeoParquet"));
        assert_eq!(detect_format(Path::new("test.zarr")), Some("Zarr"));
        assert_eq!(detect_format(Path::new("test.gpkg")), Some("GeoPackage"));
        assert_eq!(detect_format(Path::new("test.jp2")), Some("JPEG2000"));
        assert_eq!(detect_format(Path::new("test.copc")), Some("COPC"));
        assert_eq!(detect_format(Path::new("test.pmtiles")), Some("PMTiles"));
        assert_eq!(detect_format(Path::new("test.mbtiles")), Some("MBTiles"));
        assert_eq!(detect_format(Path::new("test.unknown")), None);
    }

    #[test]
    fn test_format_size() {
        assert_eq!(format_size(512), "512 B");
        assert_eq!(format_size(1024), "1.00 KB");
        assert_eq!(format_size(1536), "1.50 KB");
        assert_eq!(format_size(1_048_576), "1.00 MB");
    }
}