neopdf 0.2.0-alpha4

//! This module provides management utilities for PDF set installation, download, and path resolution.
//!
//! It defines types and methods for ensuring that PDF sets are available locally, downloading them if
//! necessary, and handling different PDF set formats (LHAPDF, NeoPDF).
use flate2::read::GzDecoder;
use indicatif::{ProgressBar, ProgressStyle};
use serde::{Deserialize, Serialize};
use std::error::Error;
use std::io::Read;
use std::path::{Path, PathBuf};
use tar::Archive;

/// TODO
#[derive(Debug, Deserialize, Serialize)]
pub enum PdfSetFormat {
    /// TODO
    Lhapdf,
    /// TODO
    Neopdf,
}

#[derive(Debug, Deserialize, Serialize)]
pub struct ManageData {
    neopdf_path: PathBuf,
    set_name: String,
    pdfset_path: PathBuf,
    pdfset_format: PdfSetFormat,
}

impl ManageData {
    pub fn new(set_name: &str, format: PdfSetFormat) -> Self {
        let data_path = Self::get_data_path();
        let xpdf_path = data_path.join(set_name);

        let manager = Self {
            neopdf_path: data_path,
            set_name: set_name.to_string(),
            pdfset_path: xpdf_path,
            pdfset_format: format,
        };
        manager.ensure_pdf_installed().unwrap();

        manager
    }

    pub fn get_data_path() -> PathBuf {
        // Check for NEOPDF_DATA_PATH environment variable first
        if let Ok(neopdf_data_path) = std::env::var("NEOPDF_DATA_PATH") {
            let neopdf_dir = PathBuf::from(neopdf_data_path);

            if !neopdf_dir.exists() {
                std::fs::create_dir_all(&neopdf_dir).unwrap();
            }

            return neopdf_dir;
        }

        // Falls back to the XDG data directory if the env. variable is not set.
        // TODO: Make this more robust and not platform-dependent
        let home = std::env::var("HOME")
            .map_err(|_| {
                std::io::Error::new(
                    std::io::ErrorKind::NotFound,
                    "HOME environment variable not found",
                )
            })
            .unwrap();
        let data_dir = PathBuf::from(home).join(".local").join("share");
        let neopdf_dir = data_dir.join("neopdf");

        if !neopdf_dir.exists() {
            std::fs::create_dir_all(&neopdf_dir).unwrap();
        }

        neopdf_dir
    }

    /// Download the PDF set and extract it into the designated path.
    /// The download happens in memory so no `*.tar.*` is written.
    pub fn download_pdf(&self) -> Result<(), Box<dyn Error>> {
        let url = format!(
            "https://lhapdfsets.web.cern.ch/current/{}.tar.gz",
            self.set_name
        );
        println!("Downloading PDF set from: {}", url);

        let response = reqwest::blocking::Client::builder()
            .timeout(None)
            .build()?
            .get(&url)
            .send()?;

        if !response.status().is_success() {
            return Err(format!(
                "Failed to download PDF set '{}': HTTP {}",
                self.set_name,
                response.status()
            )
            .into());
        }

        let total_size = response
            .headers()
            .get(reqwest::header::CONTENT_LENGTH)
            .and_then(|v| v.to_str().ok())
            .and_then(|s| s.parse::<u64>().ok())
            .unwrap_or(0);

        let pb = ProgressBar::new(total_size);
        pb.set_style(ProgressStyle::default_bar()
            .template("{spinner:.green} [{elapsed_precise}] [{bar:40.cyan/blue}] {bytes}/{total_bytes} ({eta})")?
            .progress_chars("=>-"));

        let mut response_bytes = Vec::new();
        let mut decorated_response = pb.wrap_read(response);
        decorated_response.read_to_end(&mut response_bytes)?;

        let tar = GzDecoder::new(&response_bytes[..]);
        let mut archive = Archive::new(tar);

        archive.unpack(&self.neopdf_path)?;

        Ok(())
    }

    /// Check that the PDF set is installed in the correct path.
    pub fn is_pdf_installed(&self) -> bool {
        match self.pdfset_format {
            PdfSetFormat::Neopdf => self.pdfset_path.exists() && self.pdfset_path.is_file(),
            _ => self.pdfset_path.exists() && self.pdfset_path.is_dir(),
        }
    }

    /// Ensure that the PDF set is installed, otherwise download it.
    pub fn ensure_pdf_installed(&self) -> Result<(), Box<dyn Error>> {
        if self.is_pdf_installed() {
            return Ok(());
        }

        println!("PDF set '{}' not found, downloading...", self.set_name);
        self.download_pdf()
    }

    /// Get the name of the PDF set.
    pub fn set_name(&self) -> &str {
        &self.set_name
    }

    /// Get the path where PDF sets are stored.
    pub fn data_path(&self) -> &Path {
        &self.neopdf_path
    }

    /// Get the full path to this specific PDF set.
    pub fn set_path(&self) -> &Path {
        &self.pdfset_path
    }
}