snekdown 0.33.4

/*
 * Snekdown - Custom Markdown flavour and parser
 * Copyright (C) 2021  Trivernis
 * See LICENSE for more information.
 */

use platform_dirs::{AppDirs, AppUI};
use sha2::Digest;
use std::fs;
use std::io;
use std::path::PathBuf;

#[derive(Clone, Debug)]
pub struct CacheStorage {
    location: PathBuf,
}

impl CacheStorage {
    pub fn new() -> Self {
        lazy_static::lazy_static! {
            static ref APP_DIRS: AppDirs = AppDirs::new(Some("snekdown"), AppUI::CommandLine).unwrap();
        }

        Self {
            location: APP_DIRS.cache_dir.clone(),
        }
    }

    /// Returns the cache path for a given file
    pub fn get_file_path(&self, path: &PathBuf) -> PathBuf {
        let mut hasher = sha2::Sha256::default();
        hasher.update(path.to_string_lossy().as_bytes());
        let mut file_name = PathBuf::from(format!("{:x}", hasher.finalize()));

        if let Some(extension) = path.extension() {
            file_name.set_extension(extension);
        }
        log::trace!("Cache path is {:?}", path);

        return self.location.join(PathBuf::from(file_name));
    }

    /// Returns if the given file exists in the cache
    pub fn has_file(&self, path: &PathBuf) -> bool {
        let cache_path = self.get_file_path(path);

        cache_path.exists()
    }

    /// Writes into the corresponding cache file
    pub fn read(&self, path: &PathBuf) -> io::Result<Vec<u8>> {
        let cache_path = self.get_file_path(path);

        fs::read(cache_path)
    }

    /// Reads the corresponding cache file
    pub fn write<R: AsRef<[u8]>>(&self, path: &PathBuf, contents: R) -> io::Result<()> {
        let cache_path = self.get_file_path(path);

        fs::write(cache_path, contents)
    }

    /// Clears the cache directory by deleting and recreating it
    pub fn clear(&self) -> io::Result<()> {
        fs::remove_dir_all(&self.location)?;
        fs::create_dir(&self.location)
    }
}