xenofem/poop-graph
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Wiki Activity

make directories configurable

Browse source
This commit is contained in:
xenofem 2022-04-16 01:57:34 -04:00
parent 774f3195de
commit 4eeabbb770
3 changed files with 24 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/fetch.rs
View file

@ -1,4 +1,4 @@
use std::time::{Duration, Instant, SystemTime};
use std::{time::{Duration, Instant, SystemTime}, path::PathBuf};
use futures::{sink::SinkExt, TryStreamExt};
use lazy_static::lazy_static;
@ -14,7 +14,6 @@ lazy_static! {
static ref MIN_CHECK_INTERVAL: Duration = Duration::from_secs(300);
}
const CACHED_PDF_PATH: &str = "data.pdf";
const LAST_MODIFIED_FORMAT: &[time::format_description::FormatItem] = time::macros::format_description!(
"[weekday repr:short], [day] [month repr:short] [year] [hour repr:24]:[minute]:[second] GMT"
);
@ -39,13 +38,15 @@ pub enum Error {
pub struct PdfFetcher {
client: reqwest::Client,
cached_pdf_path: PathBuf,
last_checked: Option<Instant>,
}
impl PdfFetcher {
pub fn new() -> Result<Self, Error> {
pub fn new(cached_pdf_path: PathBuf) -> Result<Self, Error> {
Ok(Self {
client: reqwest::ClientBuilder::new().build()?,
cached_pdf_path,
last_checked: None,
})
}
@ -53,7 +54,7 @@ impl PdfFetcher {
pub async fn fetch(&mut self) -> Result<pdf::file::File<Vec<u8>>, Error> {
info!("Fetching data PDF");
let cache_modtime = match tokio::fs::File::open(CACHED_PDF_PATH).await {
let cache_modtime = match tokio::fs::File::open(&self.cached_pdf_path).await {
Ok(file) => Some(
file.metadata()
.await?
@ -125,7 +126,7 @@ impl PdfFetcher {
.bytes_stream()
.map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e));
let cache_file = tokio::fs::File::create(CACHED_PDF_PATH).await?;
let cache_file = tokio::fs::File::create(&self.cached_pdf_path).await?;
let mut sink = codec::FramedWrite::new(cache_file, codec::BytesCodec::new());
sink.send_all(&mut pdf_stream).await?;
<dyn futures::Sink<bytes::Bytes, Error = std::io::Error> + Unpin>::close(&mut sink)
@ -138,6 +139,6 @@ impl PdfFetcher {
}
fn cached_pdf(&self) -> Result<pdf::file::File<Vec<u8>>, Error> {
Ok(pdf::file::File::open(CACHED_PDF_PATH)?)
Ok(pdf::file::File::open(&self.cached_pdf_path)?)
}
}