115 lines
2.7 KiB
Rust
115 lines
2.7 KiB
Rust
use std::convert::{TryFrom, TryInto};
|
|
use std::mem;
|
|
use std::os::raw::c_char;
|
|
use std::ptr;
|
|
use std::slice;
|
|
use std::str;
|
|
|
|
use super::Value;
|
|
use crate::error::check_status;
|
|
use crate::{sys, Error, JsNumber, NapiValue, Result, Status};
|
|
|
|
#[derive(Clone, Copy, Debug)]
|
|
pub struct JsString(pub(crate) Value);
|
|
|
|
impl JsString {
|
|
pub fn len(&self) -> Result<usize> {
|
|
let mut raw_length = ptr::null_mut();
|
|
unsafe {
|
|
let status = sys::napi_get_named_property(
|
|
self.0.env,
|
|
self.0.value,
|
|
"length\0".as_ptr() as *const c_char,
|
|
&mut raw_length,
|
|
);
|
|
check_status(status)?;
|
|
}
|
|
let length: JsNumber = JsNumber::from_raw_unchecked(self.0.env, raw_length);
|
|
length.try_into()
|
|
}
|
|
}
|
|
|
|
impl JsString {
|
|
#[inline]
|
|
pub fn get_ref(&self) -> Result<&[u8]> {
|
|
let mut written_char_count: u64 = 0;
|
|
let len = self.len()? + 1;
|
|
let mut result = Vec::with_capacity(len);
|
|
unsafe {
|
|
let status = sys::napi_get_value_string_utf8(
|
|
self.0.env,
|
|
self.0.value,
|
|
result.as_mut_ptr(),
|
|
len as u64,
|
|
&mut written_char_count,
|
|
);
|
|
|
|
check_status(status)?;
|
|
let ptr = result.as_ptr();
|
|
mem::forget(result);
|
|
Ok(slice::from_raw_parts(
|
|
ptr as *const u8,
|
|
written_char_count as usize,
|
|
))
|
|
}
|
|
}
|
|
|
|
pub fn as_str(&self) -> Result<&str> {
|
|
str::from_utf8(self.get_ref()?)
|
|
.map_err(|e| Error::new(Status::GenericFailure, format!("{:?}", e)))
|
|
}
|
|
|
|
pub fn get_ref_mut(&mut self) -> Result<&mut [u8]> {
|
|
let mut written_char_count: u64 = 0;
|
|
let len = self.len()? + 1;
|
|
let mut result = Vec::with_capacity(len);
|
|
unsafe {
|
|
let status = sys::napi_get_value_string_utf8(
|
|
self.0.env,
|
|
self.0.value,
|
|
result.as_mut_ptr(),
|
|
len as u64,
|
|
&mut written_char_count,
|
|
);
|
|
|
|
check_status(status)?;
|
|
let ptr = result.as_ptr();
|
|
mem::forget(result);
|
|
Ok(slice::from_raw_parts_mut(
|
|
ptr as *mut _,
|
|
written_char_count as usize,
|
|
))
|
|
}
|
|
}
|
|
}
|
|
|
|
impl TryFrom<JsString> for Vec<u16> {
|
|
type Error = Error;
|
|
|
|
fn try_from(value: JsString) -> Result<Vec<u16>> {
|
|
let mut result = Vec::with_capacity(value.len()? + 1); // Leave room for trailing null byte
|
|
|
|
unsafe {
|
|
let mut written_char_count = 0;
|
|
let status = sys::napi_get_value_string_utf16(
|
|
value.0.env,
|
|
value.0.value,
|
|
result.as_mut_ptr(),
|
|
result.capacity() as u64,
|
|
&mut written_char_count,
|
|
);
|
|
check_status(status)?;
|
|
result.set_len(written_char_count as usize);
|
|
}
|
|
|
|
Ok(result)
|
|
}
|
|
}
|
|
|
|
impl TryFrom<JsString> for String {
|
|
type Error = Error;
|
|
|
|
fn try_from(value: JsString) -> Result<String> {
|
|
Ok(value.as_str()?.to_owned())
|
|
}
|
|
}
|