use crate::*;
use std::{
borrow::Cow,
fmt::{Debug, Formatter},
iter::FromIterator,
ptr,
};
pub struct UstrPtr<'a> {
ustrs: Vec<Cow<'a, Ustr>>,
ptrs: Vec<*const c::c_char>,
}
impl<'a> UstrPtr<'a> {
pub fn new() -> Self {
Self {
ustrs: vec![],
ptrs: vec![ptr::null()],
}
}
pub fn push(&mut self, s: impl IntoUstr<'a>) {
let s = s.into_ustr();
self.ustrs.reserve_exact(1);
self.ptrs.reserve_exact(1);
self.ptrs.pop();
self.ptrs.push(s.as_ptr());
self.ptrs.push(ptr::null());
self.ustrs.push(s);
}
pub fn as_ptr(&self) -> &*const c::c_char {
&self.ptrs[0]
}
}
impl<'a, T: IntoUstr<'a>> Extend<T> for UstrPtr<'a> {
fn extend<U: IntoIterator<Item = T>>(&mut self, iter: U) {
for ustr in iter {
self.push(ustr);
}
}
}
impl<'a, T: IntoUstr<'a>> FromIterator<T> for UstrPtr<'a> {
fn from_iter<U: IntoIterator<Item = T>>(iter: U) -> Self {
let mut buf = UstrPtr::new();
buf.extend(iter);
buf
}
}
impl<'a> Default for UstrPtr<'a> {
fn default() -> Self {
UstrPtr::new()
}
}
impl<'a> Debug for UstrPtr<'a> {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
Debug::fmt(&self.ustrs, f)
}
}