use crate::errors::ValidationError;
use crate::traits::ValueObject;
pub type HexColorInput = String;
pub type HexColorOutput = String;
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[cfg_attr(feature = "serde", serde(transparent))]
pub struct HexColor(String);
impl ValueObject for HexColor {
type Input = HexColorInput;
type Output = HexColorOutput;
type Error = ValidationError;
fn new(value: Self::Input) -> Result<Self, Self::Error> {
let s = value.trim().to_uppercase();
if s.is_empty() {
return Err(ValidationError::empty("HexColor"));
}
let hex = s
.strip_prefix('#')
.ok_or_else(|| ValidationError::invalid("HexColor", &s))?;
let expanded = match hex.len() {
3 => {
if !hex.chars().all(|c| c.is_ascii_hexdigit()) {
return Err(ValidationError::invalid("HexColor", &s));
}
let chars: Vec<char> = hex.chars().collect();
format!("#{0}{0}{1}{1}{2}{2}", chars[0], chars[1], chars[2])
}
6 => {
if !hex.chars().all(|c| c.is_ascii_hexdigit()) {
return Err(ValidationError::invalid("HexColor", &s));
}
s
}
_ => return Err(ValidationError::invalid("HexColor", &s)),
};
Ok(Self(expanded))
}
fn value(&self) -> &Self::Output {
&self.0
}
fn into_inner(self) -> Self::Input {
self.0
}
}
impl HexColor {
fn channel(s: &str, offset: usize) -> u8 {
u8::from_str_radix(&s[offset..offset + 2], 16).unwrap_or(0)
}
pub fn r(&self) -> u8 {
Self::channel(&self.0, 1)
}
pub fn g(&self) -> u8 {
Self::channel(&self.0, 3)
}
pub fn b(&self) -> u8 {
Self::channel(&self.0, 5)
}
}
impl TryFrom<&str> for HexColor {
type Error = ValidationError;
fn try_from(value: &str) -> Result<Self, Self::Error> {
Self::new(value.to_owned())
}
}
impl std::fmt::Display for HexColor {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "{}", self.0)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn accepts_six_digit_form() {
let c = HexColor::new("#FF0000".into()).unwrap();
assert_eq!(c.value(), "#FF0000");
}
#[test]
fn normalises_to_uppercase() {
let c = HexColor::new("#ff0000".into()).unwrap();
assert_eq!(c.value(), "#FF0000");
}
#[test]
fn expands_three_digit_shorthand() {
let c = HexColor::new("#F0A".into()).unwrap();
assert_eq!(c.value(), "#FF00AA");
}
#[test]
fn expands_three_digit_lowercase() {
let c = HexColor::new("#f0a".into()).unwrap();
assert_eq!(c.value(), "#FF00AA");
}
#[test]
fn rejects_missing_hash() {
assert!(HexColor::new("FF0000".into()).is_err());
}
#[test]
fn rejects_invalid_chars() {
assert!(HexColor::new("#GGGGGG".into()).is_err());
}
#[test]
fn rejects_wrong_length() {
assert!(HexColor::new("#FFFF".into()).is_err());
}
#[test]
fn rejects_empty() {
assert!(HexColor::new(String::new()).is_err());
}
#[test]
fn rgb_channels() {
let c = HexColor::new("#1A2B3C".into()).unwrap();
assert_eq!(c.r(), 0x1A);
assert_eq!(c.g(), 0x2B);
assert_eq!(c.b(), 0x3C);
}
#[test]
fn try_from_str() {
let c: HexColor = "#ABC".try_into().unwrap();
assert_eq!(c.value(), "#AABBCC");
}
}