use std::{borrow::Borrow, fmt, hash};
use bytes::Bytes;
use serde::{
de::{Error, Visitor},
Deserialize, Deserializer, Serialize,
};
#[derive(Clone, Eq, PartialEq, PartialOrd, Ord)]
pub struct ByteString(Bytes);
impl ByteString {
pub(crate) fn new(origin: &Bytes, string: &str) -> Self {
ByteString(origin.slice_ref(string.as_bytes()))
}
pub fn as_str(&self) -> &str {
unsafe { std::str::from_utf8_unchecked(&self.0) }
}
pub fn inner(&self) -> &Bytes {
&self.0
}
}
impl From<String> for ByteString {
fn from(s: String) -> Self {
ByteString(s.into())
}
}
impl From<&str> for ByteString {
fn from(s: &str) -> Self {
ByteString(s.to_string().into())
}
}
impl PartialEq<ByteString> for String {
fn eq(&self, other: &ByteString) -> bool {
self.as_bytes() == other.0
}
}
impl Borrow<str> for ByteString {
fn borrow(&self) -> &str {
self.as_str()
}
}
impl std::fmt::Debug for ByteString {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
fmt::Debug::fmt(self.as_str(), f)
}
}
impl hash::Hash for ByteString {
fn hash<H: hash::Hasher>(&self, hasher: &mut H) {
self.as_str().hash(hasher)
}
}
struct ByteStringVisitor;
impl<'de> Visitor<'de> for ByteStringVisitor {
type Value = ByteString;
fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
formatter.write_str("a string")
}
fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
where
E: Error,
{
Ok(v.into())
}
fn visit_string<E>(self, v: String) -> Result<Self::Value, E>
where
E: Error,
{
Ok(v.into())
}
}
impl<'de> Deserialize<'de> for ByteString {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
deserializer.deserialize_str(ByteStringVisitor)
}
}
impl Serialize for ByteString {
#[inline]
fn serialize<S>(&self, serializer: S) -> std::result::Result<S::Ok, S::Error>
where
S: ::serde::Serializer,
{
serializer.serialize_str(self.as_str())
}
}