zvault/src/repository/config.rs

use serde_yaml;

use std::fs::File;
use std::path::Path;

use ::util::*;
use ::chunker::ChunkerType;


impl HashMethod {
    fn from_yaml(yaml: String) -> Result<Self, &'static str> {
        HashMethod::from(&yaml)
    }

    fn to_yaml(&self) -> String {
        self.name().to_string()
    }
}



impl ChecksumType {
    fn from_yaml(yaml: String) -> Result<Self, &'static str> {
        ChecksumType::from(&yaml)
    }

    fn to_yaml(&self) -> String {
        self.name().to_string()
    }
}



struct ChunkerYaml {
    method: String,
    avg_size: usize,
    seed: u64
}
impl Default for ChunkerYaml {
    fn default() -> Self {
        ChunkerYaml {
            method: "fastcdc".to_string(),
            avg_size: 16*1024,
            seed: 0
        }
    }
}
serde_impl!(ChunkerYaml(String) {
    method: String => "method",
    avg_size: usize => "avg_size",
    seed: u64 => "seed"
});

impl ChunkerType {
    fn from_yaml(yaml: ChunkerYaml) -> Result<Self, &'static str> {
        ChunkerType::from(&yaml.method, yaml.avg_size, yaml.seed)
    }

    fn to_yaml(&self) -> ChunkerYaml {
        ChunkerYaml {
            method: self.name().to_string(),
            avg_size: self.avg_size(),
            seed: self.seed()
        }
    }
}



struct CompressionYaml {
    codec: String,
    level: Option<u8>
}
impl Default for CompressionYaml {
    fn default() -> Self {
        CompressionYaml {
            codec: "brotli".to_string(),
            level: None
        }
    }
}
serde_impl!(CompressionYaml(String) {
    codec: String => "codec",
    level: Option<u8> => "level"
});

impl Compression {
    fn from_yaml(yaml: CompressionYaml) -> Result<Self, &'static str> {
        match &yaml.codec as &str {
            "snappy" => Ok(Compression::Snappy(())),
            "zstd" => Ok(Compression::ZStd(yaml.level.unwrap_or(5))),
            "deflate" | "zlib" | "gzip" => Ok(Compression::Deflate(yaml.level.unwrap_or(5))),
            "brotli" => Ok(Compression::Brotli(yaml.level.unwrap_or(5))),
            "lzma2" => Ok(Compression::Lzma2(yaml.level.unwrap_or(5))),
            _ => Err("Unsupported codec")
        }
    }

    fn to_yaml(&self) -> CompressionYaml {
        CompressionYaml {
            codec: self.name().to_string(),
            level: self.level()
        }
    }
}



struct ConfigYaml {
    compression: Option<CompressionYaml>,
    bundle_size: usize,
    chunker: ChunkerYaml,
    checksum: String,
    hash: String,
}
impl Default for ConfigYaml {
    fn default() -> Self {
        ConfigYaml {
            compression: Some(CompressionYaml { codec: "brotli".to_string(), level: Some(5) }),
            bundle_size: 25*1024*1024,
            chunker: ChunkerYaml::default(),
            checksum: "blake2_256".to_string(),
            hash: "blake2".to_string()
        }
    }
}
serde_impl!(ConfigYaml(String) {
    compression: Option<CompressionYaml> => "compression",
    bundle_size: usize => "bundle_size",
    chunker: ChunkerYaml => "chunker",
    checksum: String => "checksum",
    hash: String => "hash"
});



#[derive(Debug)]
pub struct Config {
    pub compression: Option<Compression>,
    pub bundle_size: usize,
    pub chunker: ChunkerType,
    pub checksum: ChecksumType,
    pub hash: HashMethod
}
impl Config {
    fn from_yaml(yaml: ConfigYaml) -> Result<Self, &'static str> {
        let compression = if let Some(c) = yaml.compression {
            Some(try!(Compression::from_yaml(c)))
        } else {
            None
        };
        Ok(Config{
            compression: compression,
            bundle_size: yaml.bundle_size,
            chunker: try!(ChunkerType::from_yaml(yaml.chunker)),
            checksum: try!(ChecksumType::from_yaml(yaml.checksum)),
            hash: try!(HashMethod::from_yaml(yaml.hash))
        })
    }

    fn to_yaml(&self) -> ConfigYaml {
        ConfigYaml {
            compression: self.compression.as_ref().map(|c| c.to_yaml()),
            bundle_size: self.bundle_size,
            chunker: self.chunker.to_yaml(),
            checksum: self.checksum.to_yaml(),
            hash: self.hash.to_yaml()
        }
    }

    pub fn load<P: AsRef<Path>>(path: P) -> Result<Self, &'static str> {
        let f = try!(File::open(path).map_err(|_| "Failed to open config"));
        let config = try!(serde_yaml::from_reader(f).map_err(|_| "Failed to parse config"));
        Config::from_yaml(config)
    }

    pub fn save<P: AsRef<Path>>(&self, path: P) -> Result<(), &'static str> {
        let mut f = try!(File::create(path).map_err(|_| "Failed to open config"));
        try!(serde_yaml::to_writer(&mut f, &self.to_yaml()).map_err(|_| "Failed to wrtie config"));
        Ok(())
    }
}