113 lines
3.2 KiB
Rust
113 lines
3.2 KiB
Rust
/*
|
|
* Copyright 2024 The Dragonfly Authors
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
use std::fmt;
|
|
use std::str::FromStr;
|
|
|
|
// SEPARATOR is the separator of digest.
|
|
pub const SEPARATOR: &str = ":";
|
|
|
|
// Algorithm is a enum of the algorithm that is used to generate digest.
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
|
pub enum Algorithm {
|
|
// Sha256 is sha256 algorithm for generate digest.
|
|
Sha256,
|
|
|
|
// Sha512 is sha512 algorithm for generate digest.
|
|
Sha512,
|
|
}
|
|
|
|
// Algorithm implements the Display.
|
|
impl fmt::Display for Algorithm {
|
|
// fmt formats the value using the given formatter.
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
match self {
|
|
Algorithm::Sha256 => write!(f, "sha256"),
|
|
Algorithm::Sha512 => write!(f, "sha512"),
|
|
}
|
|
}
|
|
}
|
|
|
|
// Algorithm implements the FromStr.
|
|
impl FromStr for Algorithm {
|
|
type Err = String;
|
|
|
|
// from_str parses a algorithm string.
|
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
match s {
|
|
"sha256" => Ok(Algorithm::Sha256),
|
|
"sha512" => Ok(Algorithm::Sha512),
|
|
_ => Err(format!("invalid digest algorithm: {}", s)),
|
|
}
|
|
}
|
|
}
|
|
|
|
// Digest is a struct that is used to generate digest.
|
|
pub struct Digest {
|
|
// algorithm is the algorithm that is used to generate digest.
|
|
algorithm: Algorithm,
|
|
|
|
// encoded is the encoded digest.
|
|
encoded: String,
|
|
}
|
|
|
|
// Digest implements the Digest.
|
|
impl Digest {
|
|
// new returns a new Digest.
|
|
pub fn new(algorithm: Algorithm, encoded: String) -> Self {
|
|
Self { algorithm, encoded }
|
|
}
|
|
|
|
// algorithm returns the algorithm of the digest.
|
|
pub fn algorithm(&self) -> Algorithm {
|
|
self.algorithm
|
|
}
|
|
|
|
// encoded returns the encoded digest.
|
|
pub fn encoded(&self) -> &str {
|
|
&self.encoded
|
|
}
|
|
}
|
|
|
|
// Digest implements the Display.
|
|
impl fmt::Display for Digest {
|
|
// fmt formats the value using the given formatter.
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
write!(f, "{}{}{}", self.algorithm, SEPARATOR, self.encoded)
|
|
}
|
|
}
|
|
|
|
// Digest implements the FromStr.
|
|
impl FromStr for Digest {
|
|
type Err = String;
|
|
|
|
// from_str parses a digest string.
|
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
let parts: Vec<&str> = s.splitn(2, SEPARATOR).collect();
|
|
if parts.len() != 2 {
|
|
return Err(format!("invalid digest: {}", s));
|
|
}
|
|
|
|
let algorithm = match parts[0] {
|
|
"sha256" => Algorithm::Sha256,
|
|
"sha512" => Algorithm::Sha512,
|
|
_ => return Err(format!("invalid digest algorithm: {}", parts[0])),
|
|
};
|
|
|
|
Ok(Digest::new(algorithm, parts[1].to_string()))
|
|
}
|
|
}
|