kml 0.13.0

KML support for Rust
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

use std::fs::File;
use std::io::Cursor;
use std::path::Path;
use std::str::FromStr;

use zip::ZipArchive;

use crate::errors::Error;
use crate::reader::KmlReader;
use crate::types::CoordType;

#[cfg_attr(docsrs, doc(cfg(feature = "zip")))]
impl<T> KmlReader<Cursor<Vec<u8>>, T>
where
    T: CoordType + FromStr + Default,
{
    #[cfg_attr(docsrs, doc(cfg(feature = "zip")))]
    /// Create a [`KmlReader`](struct.KmlReader.html) from a KMZ file path
    ///
    /// # Example
    ///
    /// ```
    /// use std::path::Path;
    /// use kml::KmlReader;
    ///
    /// let kmz_path = Path::new(env!("CARGO_MANIFEST_DIR"))
    ///     .join("tests")
    ///     .join("fixtures")
    ///     .join("polygon.kmz");
    /// let mut kml_reader = KmlReader::<_, f64>::from_kmz_path(kmz_path).unwrap();
    /// let kml = kml_reader.read().unwrap();
    /// ```
    pub fn from_kmz_path<P: AsRef<Path>>(path: P) -> Result<KmlReader<Cursor<Vec<u8>>, T>, Error> {
        let file = File::open(path)?;
        let mut archive = ZipArchive::new(file)?;

        // Should parse the first file with a KML extension
        for i in 0..archive.len() {
            let mut kml_file = archive
                .by_index(i)
                .map_err(|e| Error::InvalidInput(format!("{e:?}")))?;
            if !kml_file.name().to_ascii_lowercase().ends_with(".kml") {
                continue;
            }
            let mut buf = Vec::with_capacity(kml_file.size() as usize);
            std::io::copy(&mut kml_file, &mut buf)?;
            return Ok(KmlReader::from_reader(Cursor::new(buf)));
        }

        Err(Error::InvalidInput(
            "Archive contains no elements".to_string(),
        ))
    }
}

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

    #[test]
    fn test_read_kmz() {
        let kmz_path = Path::new(env!("CARGO_MANIFEST_DIR"))
            .join("tests")
            .join("fixtures")
            .join("polygon.kmz");
        let mut kml_reader = KmlReader::<_, f64>::from_kmz_path(kmz_path).unwrap();
        let kml = kml_reader.read().unwrap();

        assert!(matches!(kml, Kml::Polygon(_)))
    }
}