nix-rust/src/store/binary_cache_store.rs


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

use super::{PathInfo, Store, StorePath};
use crate::Error;

pub struct BinaryCacheStore {
    base_uri: String,
    client: reqwest::r#async::Client,
}

impl BinaryCacheStore {
    pub fn new(base_uri: String) -> Self {
        Self {
            base_uri,
            client: reqwest::r#async::Client::new(),
        }
    }
}

impl Store for BinaryCacheStore {
    fn query_path_info(
        &self,
        path: &StorePath,
    ) -> std::pin::Pin<Box<dyn std::future::Future<Output = Result<PathInfo, Error>> + Send>> {
        let uri = format!("{}/{}.narinfo", self.base_uri.clone(), path.hash);
        let path = path.clone();
        let client = self.client.clone();
        let store_dir = self.store_dir().to_string();

        Box::pin(async move {
            let response = client.get(&uri).send().await?;

            if response.status() == reqwest::StatusCode::NOT_FOUND
                || response.status() == reqwest::StatusCode::FORBIDDEN
            {
                return Err(Error::InvalidPath(path));
            }

            let response = response.error_for_status()?;

            let body = response.text().await?;

            PathInfo::parse_nar_info(&body, &store_dir)
        })
    }
}