#![allow(unsafe_code)]
pub struct StrBuf<const N: usize>([u8; N]);
impl<const N: usize> StrBuf<N> {
pub const unsafe fn new_unchecked(buf: [u8; N]) -> Self {
#[cfg(debug_assertions)]
{
assert!(core::str::from_utf8(&buf).is_ok())
}
Self(buf)
}
pub const fn as_str(&self) -> &str {
unsafe { core::str::from_utf8_unchecked(&self.0) }
}
pub const fn as_bytes(&self) -> &[u8] {
&self.0
}
pub const fn from_str(s: &str) -> Self {
let buf = crate::bytes::clone::<N>(s.as_bytes());
unsafe { Self::new_unchecked(buf) }
}
}
#[macro_export]
macro_rules! from_utf8 {
($s: expr) => {{
use ::core::primitive::str;
const OUTPUT: &str = match ::core::str::from_utf8($s) {
Ok(s) => s,
Err(_) => panic!("invalid utf-8 bytes"),
};
OUTPUT
}};
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_strbuf_runtime() {
let buf: StrBuf<5> = StrBuf::from_str("hello");
assert_eq!(buf.as_str(), "hello");
assert_eq!(buf.as_bytes(), b"hello");
let buf_empty: StrBuf<0> = StrBuf::from_str("");
assert_eq!(buf_empty.as_str(), "");
assert_eq!(buf_empty.as_bytes(), b"");
let buf_unicode: StrBuf<9> = StrBuf::from_str("你好!");
assert_eq!(buf_unicode.as_str(), "你好!");
let raw_buf = [b'a', b'b', b'c'];
let buf_unsafe: StrBuf<3> = unsafe { StrBuf::new_unchecked(raw_buf) };
assert_eq!(buf_unsafe.as_str(), "abc");
}
}