xsd-parser 1.1.0

use std::io::BufReader;

use reqwest::{
    blocking::{Client, Response},
    Error,
};
use url::Url;

use super::{ResolveRequest, Resolver};

/// Implements a [`Resolver`] that resolves `http` and `https` resources.
#[must_use]
#[derive(Default, Debug)]
pub struct WebResolver {
    client: Option<Client>,
}

impl WebResolver {
    /// Create a new [`WebResolver`] instance.
    pub fn new() -> Self {
        Self::default()
    }
}

impl Resolver for WebResolver {
    type Buffer = BufReader<Response>;
    type Error = Error;

    fn resolve(
        &mut self,
        req: &ResolveRequest,
    ) -> Result<Option<(Url, Self::Buffer)>, Self::Error> {
        let url = if let Some(current) = &req.current_location {
            current.join(&req.requested_location)
        } else {
            Url::parse(&req.requested_location)
        };

        let Ok(url) = url else {
            return Ok(None);
        };

        if !matches!(url.scheme(), "http" | "https") {
            return Ok(None);
        }

        let client = self.client.get_or_insert_with(Client::new);

        let res = client.get(url.clone()).send()?;
        let buffer = BufReader::new(res);

        Ok(Some((url, buffer)))
    }
}