librespot/core/src/cache.rs

use std::fs;
use std::fs::File;
use std::io;
use std::io::Read;
use std::path::Path;
use std::path::PathBuf;

use crate::authentication::Credentials;
use crate::spotify_id::FileId;
use crate::volume::Volume;

#[derive(Clone)]
pub struct Cache {
    audio_root: PathBuf,
    system_root: PathBuf,
    use_audio_cache: bool,
}

fn mkdir_existing(path: &Path) -> io::Result<()> {
    fs::create_dir(path).or_else(|err| {
        if err.kind() == io::ErrorKind::AlreadyExists {
            Ok(())
        } else {
            Err(err)
        }
    })
}

impl Cache {
    pub fn new(audio_location: PathBuf, system_location: PathBuf, use_audio_cache: bool) -> Cache {
        if use_audio_cache == true {
            mkdir_existing(&audio_location).unwrap();
            mkdir_existing(&audio_location.join("files")).unwrap();
        }
        mkdir_existing(&system_location).unwrap();

        Cache {
            audio_root: audio_location,
            system_root: system_location,
            use_audio_cache: use_audio_cache,
        }
    }
}

impl Cache {
    fn credentials_path(&self) -> PathBuf {
        self.system_root.join("credentials.json")
    }

    pub fn credentials(&self) -> Option<Credentials> {
        let path = self.credentials_path();
        Credentials::from_file(path)
    }

    pub fn save_credentials(&self, cred: &Credentials) {
        let path = self.credentials_path();
        cred.save_to_file(&path);
    }
}

// cache volume to system_root/volume
impl Cache {
    fn volume_path(&self) -> PathBuf {
        self.system_root.join("volume")
    }

    pub fn volume(&self) -> Option<u16> {
        let path = self.volume_path();
        Volume::from_file(path)
    }

    pub fn save_volume(&self, volume: Volume) {
        let path = self.volume_path();
        volume.save_to_file(&path);
    }
}

impl Cache {
    fn file_path(&self, file: FileId) -> PathBuf {
        let name = file.to_base16();
        self.audio_root
            .join("files")
            .join(&name[0..2])
            .join(&name[2..])
    }

    pub fn file(&self, file: FileId) -> Option<File> {
        File::open(self.file_path(file)).ok()
    }

    pub fn save_file(&self, file: FileId, contents: &mut dyn Read) {
        if self.use_audio_cache {
            let path = self.file_path(file);
            mkdir_existing(path.parent().unwrap()).unwrap();

            let mut cache_file = File::create(path).unwrap_or_else(|_e| {
                ::std::fs::remove_dir_all(&self.audio_root.join("files")).unwrap();
                mkdir_existing(&self.audio_root.join("files")).unwrap();

                let path = self.file_path(file);
                mkdir_existing(path.parent().unwrap()).unwrap();
                File::create(path).unwrap()
            });
            ::std::io::copy(contents, &mut cache_file).unwrap_or_else(|_e| {
                ::std::fs::remove_dir_all(&self.audio_root.join("files")).unwrap();
                mkdir_existing(&self.audio_root.join("files")).unwrap();

                let path = self.file_path(file);
                mkdir_existing(path.parent().unwrap()).unwrap();
                let mut file = File::create(path).unwrap();
                ::std::io::copy(contents, &mut file).unwrap()
            });
        }
    }
}