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
|
use std::collections::HashMap;
use std::fs;
use std::path::Path;
use crate::config::Config;
use crate::error::{Error, Result};
use crate::utils;
use super::api::{Api, Site};
/// This structure allows interacting with locally cached StackExchange metadata.
pub struct LocalStorage {
pub sites: Vec<Site>,
}
impl LocalStorage {
fn fetch_local_sites(filename: &Path) -> Result<Option<Vec<Site>>> {
if let Some(file) = utils::open_file(filename)? {
return serde_json::from_reader(file)
.map_err(|_| Error::MalformedFile(filename.to_path_buf()));
}
Ok(None)
}
fn store_local_sites(filename: &Path, sites: &[Site]) -> Result<()> {
let file = utils::create_file(filename)?;
serde_json::to_writer(file, sites)?;
Ok(())
}
async fn init_sites(filename: &Path, update: bool) -> Result<Vec<Site>> {
if !update {
if let Some(sites) = Self::fetch_local_sites(filename)? {
return Ok(sites);
}
}
let sites = Api::new(None).sites().await?;
Self::store_local_sites(filename, &sites)?;
Ok(sites)
}
pub async fn new(update: bool) -> Result<Self> {
let project = Config::project_dir()?;
let dir = project.cache_dir();
fs::create_dir_all(&dir)?;
let sites_filename = dir.join("sites.json");
let sites = Self::init_sites(&sites_filename, update).await?;
Ok(LocalStorage { sites })
}
// TODO is this HM worth it? Probably only will ever have < 10 site codes to search...
// maybe store this as Option<HM> on self if other methods use it...
pub async fn find_invalid_site<'a, 'b>(
&'b self,
site_codes: &'a [String],
) -> Option<&'a String> {
let hm: HashMap<&str, ()> = self
.sites
.iter()
.map(|site| (site.api_site_parameter.as_str(), ()))
.collect();
site_codes.iter().find(|s| !hm.contains_key(&s.as_str()))
}
pub fn get_urls(&self, site_codes: &[String]) -> HashMap<String, String> {
self.sites
.iter()
.filter_map(move |site| {
let _ = site_codes
.iter()
.find(|&sc| *sc == site.api_site_parameter)?;
Some((site.api_site_parameter.to_owned(), site.site_url.to_owned()))
})
.collect()
}
}
|
