yazi/yazi-shared/src/osstr.rs

use std::{borrow::Cow, ffi::{OsStr, OsString}};

pub trait IntoOsStr<'a> {
	type Error;

	fn into_os_str(self) -> Result<Cow<'a, OsStr>, Self::Error>;
}

impl<'a> IntoOsStr<'a> for Cow<'a, [u8]> {
	type Error = anyhow::Error;

	fn into_os_str(self) -> Result<Cow<'a, OsStr>, Self::Error> {
		#[cfg(unix)]
		{
			use std::os::unix::ffi::{OsStrExt, OsStringExt};
			Ok(match self {
				Cow::Borrowed(b) => Cow::Borrowed(OsStr::from_bytes(b)),
				Cow::Owned(b) => Cow::Owned(OsString::from_vec(b)),
			})
		}
		#[cfg(windows)]
		{
			Ok(match self {
				Cow::Borrowed(b) => Cow::Borrowed(OsStr::new(str::from_utf8(b)?)),
				Cow::Owned(b) => Cow::Owned(OsString::from(String::from_utf8(b)?)),
			})
		}
	}
}

impl<'a> IntoOsStr<'a> for &'a [u8] {
	type Error = anyhow::Error;

	fn into_os_str(self) -> Result<Cow<'a, OsStr>, Self::Error> { Cow::Borrowed(self).into_os_str() }
}