use std::fmt;
use serde::{Deserialize, Serialize};
use crate::{Error, data_dir::valid_task_handle};
#[derive(Debug, Clone, PartialEq, Eq, Hash, PartialOrd, Ord)]
pub struct TaskHandle(String);
impl TaskHandle {
pub fn parse(handle: impl Into<String>) -> Result<Self, Error> {
let handle = handle.into();
if valid_task_handle(&handle).is_some() {
Ok(Self(handle))
} else {
Err(Error::invalid_reference(format!(
"`{handle}` is not a task handle"
)))
}
}
pub fn as_str(&self) -> &str {
&self.0
}
pub(crate) fn key(&self) -> &str {
valid_task_handle(&self.0)
.expect("a TaskHandle is only ever constructed from a valid grammar")
.0
}
}
impl fmt::Display for TaskHandle {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str(&self.0)
}
}
impl std::str::FromStr for TaskHandle {
type Err = Error;
fn from_str(handle: &str) -> Result<Self, Self::Err> {
Self::parse(handle)
}
}
impl AsRef<str> for TaskHandle {
fn as_ref(&self) -> &str {
&self.0
}
}
impl Serialize for TaskHandle {
fn serialize<S: serde::Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
serializer.serialize_str(&self.0)
}
}
impl<'de> Deserialize<'de> for TaskHandle {
fn deserialize<D: serde::Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
let handle = String::deserialize(deserializer)?;
Self::parse(handle).map_err(serde::de::Error::custom)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn a_well_formed_handle_round_trips() {
let text = format!("0123456789abcdef/{:032x}", 1);
let handle = TaskHandle::parse(text.clone()).expect("well-formed");
assert_eq!(handle.as_str(), text);
assert_eq!(handle.to_string(), text);
assert_eq!(handle.key(), "0123456789abcdef");
}
#[test]
fn a_malformed_handle_is_refused_generically() {
for bad in ["not-a-handle", "0123456789abcdef", "../../etc/passwd"] {
let error = TaskHandle::parse(bad).expect_err("refused");
assert!(error.to_string().contains("not a task handle"), "{error}");
assert!(
error.is_invalid_reference(),
"a malformed handle was never going to resolve: {error}"
);
}
}
#[test]
fn json_round_trips_as_a_bare_string() {
let text = format!("0123456789abcdef/{:032x}", 2);
let handle = TaskHandle::parse(text.clone()).unwrap();
let json = serde_json::to_string(&handle).unwrap();
assert_eq!(json, format!("\"{text}\""));
let back: TaskHandle = serde_json::from_str(&json).unwrap();
assert_eq!(back, handle);
}
}