dprint 0.30.2

Binary for dprint code formatter—a pluggable and configurable code formatting platform.
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

use anyhow::anyhow;
use anyhow::Result;
use std::cell::RefCell;
use std::collections::HashMap;

use crate::environment::UrlDownloader;

type CachedDownloadResult = Result<Option<Vec<u8>>, String>;

pub struct CachedDownloader<TInner: UrlDownloader> {
  inner: TInner,
  results: RefCell<HashMap<String, CachedDownloadResult>>,
}

impl<TInner: UrlDownloader> CachedDownloader<TInner> {
  pub fn new(inner: TInner) -> Self {
    Self {
      inner,
      results: Default::default(),
    }
  }
}

impl<TInner: UrlDownloader> UrlDownloader for CachedDownloader<TInner> {
  fn download_file(&self, url: &str) -> Result<Option<Vec<u8>>> {
    let mut results = self.results.borrow_mut();
    if let Some(result) = results.get(url) {
      match result {
        Ok(result) => Ok(result.clone()),
        Err(err) => Err(anyhow!("{:#}", err)),
      }
    } else {
      let result = self.inner.download_file(url);
      results.insert(
        url.to_string(),
        match &result {
          Ok(result) => Ok(result.clone()),
          Err(err) => Err(format!("{:#}", err)),
        },
      );
      result
    }
  }
}

#[cfg(test)]
mod test {
  use super::*;
  use crate::environment::TestEnvironmentBuilder;

  #[test]
  fn should_download_and_cache() {
    let mut builder = TestEnvironmentBuilder::new();
    let exists_url = "http://localhost/test.txt";
    let not_exists_url = "http://localhost/non-existent.txt";
    let environment = builder.add_remote_file(exists_url, "1").build();
    let downloader = CachedDownloader::new(environment.clone());

    // should cache when not exists
    assert!(downloader.download_file(not_exists_url).as_ref().unwrap().is_none());
    environment.add_remote_file_bytes(not_exists_url, Vec::new());
    assert!(downloader.download_file(not_exists_url).as_ref().unwrap().is_none());

    // should get data and have it cached
    assert_eq!(downloader.download_file(exists_url).as_ref().unwrap().as_ref().unwrap(), "1".as_bytes());
    environment.add_remote_file_bytes(exists_url, Vec::new());
    assert_eq!(downloader.download_file(exists_url).as_ref().unwrap().as_ref().unwrap(), "1".as_bytes());
  }
}