#![allow(dead_code)]
#![allow(unused)]
use crate::__internal::runtime::InnerProtoString;
use crate::__internal::{Private, SealedInternal};
use crate::{AsView, IntoProxied, IntoView, MapKey, Mut, MutProxied, Proxied, View};
use std::borrow::Cow;
use std::cmp::{Eq, Ord, Ordering, PartialEq, PartialOrd};
use std::convert::{AsMut, AsRef};
use std::ffi::{OsStr, OsString};
use std::fmt;
use std::hash::{Hash, Hasher};
use std::iter;
use std::ops::{Deref, DerefMut};
use std::ptr;
use std::rc::Rc;
use std::sync::Arc;
pub struct ProtoBytes {
pub(crate) inner: InnerProtoString,
}
impl ProtoBytes {
#[doc(hidden)]
pub fn into_inner(self, _private: Private) -> InnerProtoString {
self.inner
}
#[doc(hidden)]
pub fn from_inner(_private: Private, inner: InnerProtoString) -> ProtoBytes {
Self { inner }
}
pub fn as_view(&self) -> &[u8] {
self.inner.as_bytes()
}
}
impl AsRef<[u8]> for ProtoBytes {
fn as_ref(&self) -> &[u8] {
self.inner.as_bytes()
}
}
impl From<&[u8]> for ProtoBytes {
fn from(v: &[u8]) -> ProtoBytes {
ProtoBytes { inner: InnerProtoString::from(v) }
}
}
impl<const N: usize> From<&[u8; N]> for ProtoBytes {
fn from(v: &[u8; N]) -> ProtoBytes {
ProtoBytes { inner: InnerProtoString::from(v.as_ref()) }
}
}
impl SealedInternal for ProtoBytes {}
impl Proxied for ProtoBytes {
type View<'msg> = &'msg [u8];
}
impl AsView for ProtoBytes {
type Proxied = Self;
fn as_view(&self) -> &[u8] {
self.as_view()
}
}
impl IntoProxied<ProtoBytes> for &[u8] {
fn into_proxied(self, _private: Private) -> ProtoBytes {
ProtoBytes::from(self)
}
}
impl<const N: usize> IntoProxied<ProtoBytes> for &[u8; N] {
fn into_proxied(self, _private: Private) -> ProtoBytes {
ProtoBytes::from(self.as_ref())
}
}
impl IntoProxied<ProtoBytes> for Vec<u8> {
fn into_proxied(self, _private: Private) -> ProtoBytes {
ProtoBytes::from(AsRef::<[u8]>::as_ref(&self))
}
}
impl IntoProxied<ProtoBytes> for &Vec<u8> {
fn into_proxied(self, _private: Private) -> ProtoBytes {
ProtoBytes::from(AsRef::<[u8]>::as_ref(self))
}
}
impl IntoProxied<ProtoBytes> for Box<[u8]> {
fn into_proxied(self, _private: Private) -> ProtoBytes {
ProtoBytes::from(AsRef::<[u8]>::as_ref(&self))
}
}
impl IntoProxied<ProtoBytes> for Cow<'_, [u8]> {
fn into_proxied(self, _private: Private) -> ProtoBytes {
ProtoBytes::from(AsRef::<[u8]>::as_ref(&self))
}
}
impl IntoProxied<ProtoBytes> for Rc<[u8]> {
fn into_proxied(self, _private: Private) -> ProtoBytes {
ProtoBytes::from(AsRef::<[u8]>::as_ref(&self))
}
}
impl IntoProxied<ProtoBytes> for Arc<[u8]> {
fn into_proxied(self, _private: Private) -> ProtoBytes {
ProtoBytes::from(AsRef::<[u8]>::as_ref(&self))
}
}
impl SealedInternal for &[u8] {}
impl AsView for &[u8] {
type Proxied = ProtoBytes;
fn as_view(&self) -> &[u8] {
self
}
}
impl<'msg> IntoView<'msg> for &'msg [u8] {
fn into_view<'shorter>(self) -> &'shorter [u8]
where
'msg: 'shorter,
{
self
}
}
#[derive(Debug, PartialEq)]
pub struct Utf8Error {
pub(crate) inner: std::str::Utf8Error,
}
impl std::fmt::Display for Utf8Error {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
self.inner.fmt(f)
}
}
impl std::error::Error for Utf8Error {}
impl From<std::str::Utf8Error> for Utf8Error {
fn from(inner: std::str::Utf8Error) -> Utf8Error {
Utf8Error { inner }
}
}
pub struct ProtoString {
pub(crate) inner: InnerProtoString,
}
impl ProtoString {
pub fn as_view(&self) -> &ProtoStr {
ProtoStr::from_utf8_unchecked(self.as_bytes())
}
pub fn as_bytes(&self) -> &[u8] {
self.inner.as_bytes()
}
pub fn from_utf8_unchecked(v: &[u8]) -> Self {
Self { inner: InnerProtoString::from(v) }
}
#[doc(hidden)]
pub fn into_inner(self, _private: Private) -> InnerProtoString {
self.inner
}
#[doc(hidden)]
pub fn from_inner(_private: Private, inner: InnerProtoString) -> ProtoString {
Self { inner }
}
}
impl SealedInternal for ProtoString {}
impl Deref for ProtoString {
type Target = ProtoStr;
fn deref(&self) -> &Self::Target {
self.as_view()
}
}
impl AsRef<[u8]> for ProtoString {
fn as_ref(&self) -> &[u8] {
self.inner.as_bytes()
}
}
impl From<ProtoString> for ProtoBytes {
fn from(v: ProtoString) -> Self {
ProtoBytes { inner: v.inner }
}
}
impl From<&str> for ProtoString {
fn from(v: &str) -> Self {
Self::from_utf8_unchecked(v.as_bytes())
}
}
impl TryFrom<&[u8]> for ProtoString {
type Error = Utf8Error;
fn try_from(v: &[u8]) -> Result<Self, Self::Error> {
let s = std::str::from_utf8(v)?;
Ok(ProtoString::from(s))
}
}
impl SealedInternal for &str {}
impl SealedInternal for &ProtoStr {}
impl IntoProxied<ProtoString> for &str {
fn into_proxied(self, _private: Private) -> ProtoString {
ProtoString::from(self)
}
}
impl IntoProxied<ProtoString> for &ProtoStr {
fn into_proxied(self, _private: Private) -> ProtoString {
ProtoString::from_utf8_unchecked(self.as_bytes())
}
}
impl IntoProxied<ProtoString> for String {
fn into_proxied(self, _private: Private) -> ProtoString {
ProtoString::from(self.as_str())
}
}
impl IntoProxied<ProtoString> for &String {
fn into_proxied(self, _private: Private) -> ProtoString {
ProtoString::from_utf8_unchecked(self.as_bytes())
}
}
impl IntoProxied<ProtoString> for OsString {
fn into_proxied(self, private: Private) -> ProtoString {
self.as_os_str().into_proxied(private)
}
}
impl IntoProxied<ProtoString> for &OsStr {
fn into_proxied(self, _private: Private) -> ProtoString {
ProtoString::from_utf8_unchecked(self.as_encoded_bytes())
}
}
impl IntoProxied<ProtoString> for Box<str> {
fn into_proxied(self, _private: Private) -> ProtoString {
ProtoString::from(AsRef::<str>::as_ref(&self))
}
}
impl IntoProxied<ProtoString> for Cow<'_, str> {
fn into_proxied(self, _private: Private) -> ProtoString {
ProtoString::from(AsRef::<str>::as_ref(&self))
}
}
impl IntoProxied<ProtoString> for Rc<str> {
fn into_proxied(self, _private: Private) -> ProtoString {
ProtoString::from(AsRef::<str>::as_ref(&self))
}
}
impl IntoProxied<ProtoString> for Arc<str> {
fn into_proxied(self, _private: Private) -> ProtoString {
ProtoString::from(AsRef::<str>::as_ref(&self))
}
}
#[repr(transparent)]
pub struct ProtoStr([u8]);
impl ProtoStr {
pub const fn as_bytes(&self) -> &[u8] {
&self.0
}
pub const fn to_str(&self) -> Result<&str, Utf8Error> {
match std::str::from_utf8(&self.0) {
Ok(s) => Ok(s),
Err(e) => Err(Utf8Error { inner: e }),
}
}
pub fn to_cow_lossy(&self) -> Cow<'_, str> {
String::from_utf8_lossy(&self.0)
}
pub const fn is_empty(&self) -> bool {
self.0.is_empty()
}
pub const fn len(&self) -> usize {
self.0.len()
}
pub const fn from_utf8_unchecked(bytes: &[u8]) -> &Self {
unsafe { &*(bytes as *const [u8] as *const Self) }
}
pub const fn from_str(string: &str) -> &Self {
Self::from_utf8_unchecked(string.as_bytes())
}
pub const fn is_ascii(&self) -> bool {
self.0.is_ascii()
}
pub fn contains<T>(&self, other: &T) -> bool
where
T: AsRef<[u8]> + ?Sized,
{
let other = other.as_ref();
if other.is_empty() {
return true;
}
self.0.windows(other.len()).any(|window| window == other)
}
pub fn starts_with<T>(&self, other: &T) -> bool
where
T: AsRef<[u8]> + ?Sized,
{
self.0.starts_with(other.as_ref())
}
pub fn ends_with<T>(&self, other: &T) -> bool
where
T: AsRef<[u8]> + ?Sized,
{
self.0.ends_with(other.as_ref())
}
pub fn find<T>(&self, other: &T) -> Option<usize>
where
T: AsRef<[u8]> + ?Sized,
{
let other = other.as_ref();
if other.is_empty() {
return Some(0);
}
self.0.windows(other.len()).position(|window| window == other)
}
pub const fn trim_ascii(&self) -> &Self {
Self::from_utf8_unchecked(self.0.trim_ascii())
}
pub const fn trim_ascii_start(&self) -> &Self {
Self::from_utf8_unchecked(self.0.trim_ascii_start())
}
pub const fn trim_ascii_end(&self) -> &Self {
Self::from_utf8_unchecked(self.0.trim_ascii_end())
}
}
impl AsRef<[u8]> for ProtoStr {
fn as_ref(&self) -> &[u8] {
self.as_bytes()
}
}
impl<'msg> From<&'msg ProtoStr> for &'msg [u8] {
fn from(val: &'msg ProtoStr) -> &'msg [u8] {
val.as_bytes()
}
}
impl<'msg> From<&'msg str> for &'msg ProtoStr {
fn from(val: &'msg str) -> &'msg ProtoStr {
ProtoStr::from_str(val)
}
}
impl<'msg> TryFrom<&'msg ProtoStr> for &'msg str {
type Error = Utf8Error;
fn try_from(val: &'msg ProtoStr) -> Result<&'msg str, Utf8Error> {
val.to_str()
}
}
impl<'msg> TryFrom<&'msg [u8]> for &'msg ProtoStr {
type Error = Utf8Error;
fn try_from(val: &'msg [u8]) -> Result<&'msg ProtoStr, Utf8Error> {
let s = std::str::from_utf8(val)?;
Ok(ProtoStr::from_str(s))
}
}
impl fmt::Debug for ProtoStr {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "\"");
for chunk in self.as_bytes().utf8_chunks() {
for ch in chunk.valid().chars() {
write!(f, "{}", ch.escape_debug());
}
for byte in chunk.invalid() {
write!(f, "\\x{:02X}", byte);
}
}
write!(f, "\"");
Ok(())
}
}
impl fmt::Display for ProtoStr {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
fmt::Display::fmt(&String::from_utf8_lossy(self.as_bytes()), f)?;
Ok(())
}
}
impl Hash for ProtoStr {
fn hash<H: Hasher>(&self, state: &mut H) {
self.as_bytes().hash(state)
}
}
impl Eq for ProtoStr {}
impl Ord for ProtoStr {
fn cmp(&self, other: &ProtoStr) -> Ordering {
self.as_bytes().cmp(other.as_bytes())
}
}
impl Proxied for ProtoString {
type View<'msg> = &'msg ProtoStr;
}
impl MapKey for ProtoString {}
impl AsView for ProtoString {
type Proxied = Self;
fn as_view(&self) -> &ProtoStr {
self.as_view()
}
}
impl AsView for &ProtoStr {
type Proxied = ProtoString;
fn as_view(&self) -> &ProtoStr {
self
}
}
impl<'msg> IntoView<'msg> for &'msg ProtoStr {
fn into_view<'shorter>(self) -> &'shorter ProtoStr
where
'msg: 'shorter,
{
self
}
}
macro_rules! impl_bytes_partial_cmp {
($(<($($generics:tt)*)> $lhs:ty => $rhs:ty),+ $(,)?) => {
$(
impl<$($generics)*> PartialEq<$rhs> for $lhs {
fn eq(&self, other: &$rhs) -> bool {
AsRef::<[u8]>::as_ref(self) == AsRef::<[u8]>::as_ref(other)
}
}
impl<$($generics)*> PartialOrd<$rhs> for $lhs {
fn partial_cmp(&self, other: &$rhs) -> Option<Ordering> {
AsRef::<[u8]>::as_ref(self).partial_cmp(AsRef::<[u8]>::as_ref(other))
}
}
)*
};
}
impl_bytes_partial_cmp!(
<()> ProtoStr => ProtoStr,
<()> ProtoStr => str,
<()> str => ProtoStr,
);
impl std::fmt::Debug for ProtoString {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> Result<(), std::fmt::Error> {
std::fmt::Debug::fmt(self.as_view(), f)
}
}
impl std::fmt::Debug for ProtoBytes {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> Result<(), std::fmt::Error> {
std::fmt::Debug::fmt(self.as_view(), f)
}
}
unsafe impl Sync for ProtoString {}
unsafe impl Send for ProtoString {}
unsafe impl Send for ProtoBytes {}
unsafe impl Sync for ProtoBytes {}
#[cfg(test)]
mod tests {
use super::*;
use googletest::prelude::*;
#[gtest]
fn test_proto_string_try_from() -> googletest::Result<()> {
let valid_utf8 = b"hello";
let s = ProtoString::try_from(&valid_utf8[..])?;
verify_eq!(s.as_bytes(), valid_utf8)?;
let invalid_utf8 = b"\xff";
let res = ProtoString::try_from(&invalid_utf8[..]);
verify_that!(res, err(anything()))?;
Ok(())
}
#[gtest]
fn test_proto_string_from_utf8_unchecked() -> googletest::Result<()> {
let invalid_utf8 = b"\xff";
let s = ProtoString::from_utf8_unchecked(invalid_utf8);
verify_eq!(s.as_bytes(), invalid_utf8)?;
Ok(())
}
#[gtest]
fn test_proto_str_methods() -> googletest::Result<()> {
let s = ProtoStr::from_str(" hello world ");
verify_eq!(s.contains(s), true)?;
verify_eq!(s.contains("hello"), true)?;
verify_eq!(s.contains("world"), true)?;
verify_eq!(s.contains("o w"), true)?;
verify_eq!(s.contains("xyz"), false)?;
verify_eq!(s.contains(""), true)?;
verify_eq!(s.starts_with(s), true)?;
verify_eq!(s.ends_with(s), true)?;
verify_eq!(s.starts_with(" he"), true)?;
verify_eq!(s.ends_with("d "), true)?;
verify_eq!(s.starts_with("hel"), false)?;
verify_eq!(s.find(s), Some(0))?;
verify_eq!(s.find("hello"), Some(2))?;
verify_eq!(s.find("world"), Some(8))?;
verify_eq!(s.find("xyz"), None)?;
verify_eq!(s.find(""), Some(0))?;
verify_eq!(s.trim_ascii(), "hello world")?;
verify_eq!(s.trim_ascii_start(), "hello world ")?;
verify_eq!(s.trim_ascii_end(), " hello world")?;
Ok(())
}
#[gtest]
fn test_proto_string_deref() -> googletest::Result<()> {
let s = ProtoString::from(" hello ");
verify_eq!(s.contains("hello"), true)?;
verify_eq!(s.trim_ascii(), "hello")?;
let s2 = ProtoStr::from_str("he");
verify_eq!(s.contains(s2), true)?;
let s2 = ProtoStr::from_str("world");
verify_eq!(s.contains(s2), false)?;
Ok(())
}
#[gtest]
fn test_const_proto_str() -> googletest::Result<()> {
const S: &ProtoStr = ProtoStr::from_str("hello");
verify_eq!(S.contains("hello"), true)?;
const S_BYTES: &[u8] = S.as_bytes();
verify_eq!(S_BYTES, b"hello")?;
const S_TO_STR: core::result::Result<&str, Utf8Error> = S.to_str();
verify_eq!(S_TO_STR.unwrap(), "hello")?;
const S_IS_EMPTY: bool = S.is_empty();
verify_eq!(S_IS_EMPTY, false)?;
const EMPTY: &ProtoStr = ProtoStr::from_str("");
const EMPTY_IS_EMPTY: bool = EMPTY.is_empty();
verify_eq!(EMPTY_IS_EMPTY, true)?;
const S_LEN: usize = S.len();
verify_eq!(S_LEN, 5)?;
const S_IS_ASCII: bool = S.is_ascii();
verify_eq!(S_IS_ASCII, true)?;
const TRIM_ME: &ProtoStr = ProtoStr::from_str(" foo ");
const TRIMMED: &ProtoStr = TRIM_ME.trim_ascii();
const TRIMMED_START: &ProtoStr = TRIM_ME.trim_ascii_start();
const TRIMMED_END: &ProtoStr = TRIM_ME.trim_ascii_end();
verify_eq!(TRIMMED.as_bytes(), b"foo")?;
verify_eq!(TRIMMED_START.as_bytes(), b"foo ")?;
verify_eq!(TRIMMED_END.as_bytes(), b" foo")?;
const S2: &ProtoStr = ProtoStr::from_utf8_unchecked(b"world");
verify_eq!(S2.contains("world"), true)?;
Ok(())
}
}