42 lines
926 B
Rust
42 lines
926 B
Rust
use std::borrow::Cow;
|
|
use std::fmt;
|
|
|
|
use rocket::request::FromParam;
|
|
|
|
use rand::{self, distributions::Alphanumeric, Rng};
|
|
|
|
pub struct PasteId<'a>(Cow<'a, str>);
|
|
|
|
fn valid_id(id: &str) -> bool {
|
|
id.chars().all(char::is_alphanumeric)
|
|
}
|
|
|
|
impl<'a> PasteId<'a> {
|
|
pub fn new(size: usize) -> PasteId<'static> {
|
|
let id: String = rand::thread_rng()
|
|
.sample_iter(&Alphanumeric)
|
|
.take(size)
|
|
.map(char::from)
|
|
.collect();
|
|
|
|
PasteId(Cow::Owned(id))
|
|
}
|
|
}
|
|
|
|
impl<'a> FromParam<'a> for PasteId<'a> {
|
|
type Error = &'a str;
|
|
|
|
fn from_param(param: &'a str) -> Result<Self, Self::Error> {
|
|
match valid_id(param) {
|
|
true => Ok(PasteId(Cow::Borrowed(param))),
|
|
false => Err(param),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'a> fmt::Display for PasteId<'a> {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
write!(f, "{}", self.0)
|
|
}
|
|
}
|