diff options
author | Lars Wirzenius <liw@liw.fi> | 2021-09-15 18:12:50 +0300 |
---|---|---|
committer | Lars Wirzenius <liw@liw.fi> | 2021-09-18 17:52:36 +0300 |
commit | 6ed6b1bc75b1995a7740ff28bc26a908b91f37c8 (patch) | |
tree | 48575e9335123d5f6a228038f71565e42dbf9366 /src/checksummer.rs | |
parent | d6728974b98a821b8d633197abd97cf2fc9357f5 (diff) | |
download | obnam2-6ed6b1bc75b1995a7740ff28bc26a908b91f37c8.tar.gz |
refactor: define a Checksum type and use it where appropriate
This will make it harder to compare, say, a SHA-256 and a SHA3, later,
when we add more checksum types.
Sponsored-by: author
Diffstat (limited to 'src/checksummer.rs')
-rw-r--r-- | src/checksummer.rs | 32 |
1 files changed, 27 insertions, 5 deletions
diff --git a/src/checksummer.rs b/src/checksummer.rs index 162c26b..18b8afb 100644 --- a/src/checksummer.rs +++ b/src/checksummer.rs @@ -1,8 +1,30 @@ use sha2::{Digest, Sha256}; +use std::fmt; -pub fn sha256(data: &[u8]) -> String { - let mut hasher = Sha256::new(); - hasher.update(data); - let hash = hasher.finalize(); - format!("{:x}", hash) +/// A checksum of some data. +#[derive(Debug, Clone)] +pub enum Checksum { + Sha256(String), +} + +impl Checksum { + pub fn sha256(data: &[u8]) -> Self { + let mut hasher = Sha256::new(); + hasher.update(data); + let hash = hasher.finalize(); + Self::Sha256(format!("{:x}", hash)) + } + + pub fn sha256_from_str_unchecked(hash: &str) -> Self { + Self::Sha256(hash.to_string()) + } +} + +impl fmt::Display for Checksum { + fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { + let hash = match self { + Self::Sha256(hash) => hash, + }; + write!(f, "{}", hash) + } } |