wof 1.0.0

The Who's On First rust library and command line.
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

use crate::wof::WOFGeoJSON;
use regex::Regex;
use std::io::Result;
use std::path::Path;
use walkdir::{DirEntry, IntoIter, WalkDir};

/// Walk through a WOF directory and iterate over all GeoJSONs using the powerful [`walkdir`](https://crates.io/crates/walkdir) crate.
pub struct Walk {
  walker: IntoIter,
  geojson_regex: Regex,
  alt_regex: Regex,
  with_alt: bool,
  with_deprecated: bool,
}

impl Walk {
  pub fn new<P: AsRef<Path>>(directory: P, with_alt: bool, with_deprecated: bool) -> Self {
    Walk {
      walker: WalkDir::new(directory).into_iter(),
      with_alt,
      with_deprecated,
      geojson_regex: Regex::new(r"\.geojson$").unwrap(),
      alt_regex: Regex::new(r"^\d+-alt.*\.geojson$").unwrap(),
    }
  }

  pub fn json_or_geojson<P: AsRef<Path>>(directory: P) -> Self {
    Walk {
      walker: WalkDir::new(directory).into_iter(),
      with_alt: true,
      with_deprecated: true,
      geojson_regex: Regex::new(r"\.(geojson|json)$").unwrap(),
      alt_regex: Regex::new(r"^\d+-alt.*\.(geojson|json)$").unwrap(),
    }
  }

  fn should_skip<P: AsRef<Path>>(&self, path: P) -> std::result::Result<bool, String> {
    if !self.with_deprecated {
      let json = crate::parse_file_to_json(path)?;
      let geojson = WOFGeoJSON::as_valid_wof_geojson(&json)?;
      if geojson.is_doc_deprecated() {
        return Ok(true);
      }
    }
    Ok(false)
  }
}

impl Iterator for Walk {
  type Item = Result<DirEntry>;

  fn next(&mut self) -> Option<Result<DirEntry>> {
    loop {
      match self.walker.next() {
        Some(Ok(path)) => {
          if path.file_type().is_file() {
            let (is_geojson, is_altname) = if let Some(file_name) = path.file_name().to_str() {
              (
                self.geojson_regex.is_match(file_name),
                self.alt_regex.is_match(file_name),
              )
            } else {
              continue;
            };
            if is_geojson && (self.with_alt || !is_altname) {
              if !self.should_skip(path.path()).unwrap_or(true) {
                return Some(Ok(path));
              }
            }
          }
        }
        Some(Err(err)) => {
          return Some(Err(err.into_io_error().unwrap_or(std::io::Error::new(
            std::io::ErrorKind::Other,
            "file system loop found",
          ))))
        }
        None => return None,
      }
    }
  }
}