mgrs 0.4.2

Bidirectional MGRS/lat-long coordinate conversion CLI with multi-format I/O: CSV, GeoJSON, KML, KMZ, GPX, WKT, TopoJSON, Shapefile, GeoPackage, FlatGeobuf
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

use std::io::Write;
use anyhow::Result;
use crate::formats::{ConvertedRow, OutputFormat};

pub struct CsvOutput<W: Write> {
    writer: csv::Writer<W>,
}

impl<W: Write> CsvOutput<W> {
    pub fn new(output: W) -> Self {
        Self {
            writer: csv::Writer::from_writer(output),
        }
    }
}

impl<W: Write> OutputFormat for CsvOutput<W> {
    fn write_header(&mut self, headers: &[String]) -> Result<()> {
        let mut row: Vec<&str> = headers.iter().map(|h| h.as_str()).collect();
        row.push("Latitude");
        row.push("Longitude");
        self.writer.write_record(&row)?;
        Ok(())
    }

    fn write_row(&mut self, row: &ConvertedRow) -> Result<()> {
        let mut record: Vec<String> = row.fields.clone();
        record.push(row.latitude.map(|l| l.to_string()).unwrap_or_default());
        record.push(row.longitude.map(|l| l.to_string()).unwrap_or_default());
        self.writer.write_record(&record)?;
        Ok(())
    }

    fn finish(&mut self) -> Result<()> {
        self.writer.flush()?;
        Ok(())
    }
}

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

    #[test]
    fn test_csv_output_writes_headers_with_latlon() {
        let mut buf = Vec::new();
        {
            let mut writer = CsvOutput::new(&mut buf);
            writer.write_header(&[
                "Name".to_string(),
                "MGRS".to_string(),
            ]).unwrap();
            writer.finish().unwrap();
        }
        let output = String::from_utf8(buf).unwrap();
        assert!(output.contains("Name"));
        assert!(output.contains("MGRS"));
        assert!(output.contains("Latitude"));
        assert!(output.contains("Longitude"));
    }

    #[test]
    fn test_csv_output_writes_row_with_coordinates() {
        let mut buf = Vec::new();
        {
            let mut writer = CsvOutput::new(&mut buf);
            writer.write_header(&[
                "Name".to_string(),
                "MGRS".to_string(),
            ]).unwrap();
            writer.write_row(&ConvertedRow {
                fields: vec!["DC".to_string(), "18SUJ2337006519".to_string()],
                headers: vec!["Name".to_string(), "MGRS".to_string()],
                latitude: Some(38.8977),
                longitude: Some(-77.0365),
                mgrs_source: Some("18SUJ2337006519".to_string()),
            }).unwrap();
            writer.finish().unwrap();
        }
        let output = String::from_utf8(buf).unwrap();
        assert!(output.contains("38.8977"));
        assert!(output.contains("-77.0365"));
    }

    #[test]
    fn test_csv_output_writes_empty_on_failed_conversion() {
        let mut buf = Vec::new();
        {
            let mut writer = CsvOutput::new(&mut buf);
            writer.write_header(&["Name".to_string()]).unwrap();
            writer.write_row(&ConvertedRow {
                fields: vec!["Place".to_string()],
                headers: vec!["Name".to_string()],
                latitude: None,
                longitude: None,
                mgrs_source: None,
            }).unwrap();
            writer.finish().unwrap();
        }
        let output = String::from_utf8(buf).unwrap();
        let lines: Vec<&str> = output.trim().lines().collect();
        assert_eq!(lines.len(), 2); // header + 1 row
        // Row should end with two empty fields
        assert!(lines[1].ends_with(",,"));
    }
}