use crate::{FromInner, HandleTrait, Inner, IntoInner};
use std::borrow::Cow;
use std::convert::TryFrom;
use std::ffi::{CStr, CString};
use uv::{
uv_fs_event_getpath, uv_fs_event_init, uv_fs_event_start, uv_fs_event_stop, uv_fs_event_t,
};
bitflags! {
pub struct FsEventFlags: u32 {
const WATCHENTRY = uv::uv_fs_event_flags_UV_FS_EVENT_WATCH_ENTRY as _;
const STAT = uv::uv_fs_event_flags_UV_FS_EVENT_STAT as _;
const RECURSIVE = uv::uv_fs_event_flags_UV_FS_EVENT_RECURSIVE as _;
}
}
bitflags! {
pub struct FsEvent: u32 {
const RENAME = uv::uv_fs_event_UV_RENAME as _;
const CHANGE = uv::uv_fs_event_UV_CHANGE as _;
}
}
callbacks! {
pub FsEventCB(
handle: FsEventHandle,
filename: Option<Cow<str>>,
events: FsEvent,
status: crate::Result<u32>
);
}
#[derive(Default)]
pub(crate) struct FsEventDataFields<'a> {
fs_event_cb: FsEventCB<'a>,
}
extern "C" fn uv_fs_event_cb(
handle: *mut uv_fs_event_t,
filename: *const std::os::raw::c_char,
events: std::os::raw::c_int,
status: std::os::raw::c_int,
) {
let dataptr = crate::Handle::get_data(uv_handle!(handle));
if !dataptr.is_null() {
unsafe {
if let super::FsEventData(d) = &mut (*dataptr).addl {
let filename = if filename.is_null() {
None
} else {
Some(CStr::from_ptr(filename).to_string_lossy())
};
let status = if status < 0 {
Err(crate::Error::from_inner(status as uv::uv_errno_t))
} else {
Ok(status as _)
};
d.fs_event_cb.call(
handle.into_inner(),
filename,
FsEvent::from_bits_truncate(events as _),
status,
);
}
}
}
}
#[derive(Clone, Copy)]
pub struct FsEventHandle {
handle: *mut uv_fs_event_t,
}
impl FsEventHandle {
pub fn new(r#loop: &crate::Loop) -> crate::Result<FsEventHandle> {
let layout = std::alloc::Layout::new::<uv_fs_event_t>();
let handle = unsafe { std::alloc::alloc(layout) as *mut uv_fs_event_t };
if handle.is_null() {
return Err(crate::Error::ENOMEM);
}
let ret = unsafe { uv_fs_event_init(r#loop.into_inner(), handle) };
if ret < 0 {
unsafe { std::alloc::dealloc(handle as _, layout) };
return Err(crate::Error::from_inner(ret as uv::uv_errno_t));
}
crate::Handle::initialize_data(uv_handle!(handle), super::FsEventData(Default::default()));
Ok(FsEventHandle { handle })
}
pub fn start<CB: Into<FsEventCB<'static>>>(
&mut self,
path: &str,
flags: FsEventFlags,
cb: CB,
) -> Result<(), Box<dyn std::error::Error>> {
let path = CString::new(path)?;
let cb = cb.into();
let uv_cb = use_c_callback!(uv_fs_event_cb, cb);
let dataptr = crate::Handle::get_data(uv_handle!(self.handle));
if !dataptr.is_null() {
if let super::FsEventData(d) = unsafe { &mut (*dataptr).addl } {
d.fs_event_cb = cb;
}
}
crate::uvret(unsafe { uv_fs_event_start(self.handle, uv_cb, path.as_ptr(), flags.bits()) })
.map_err(|e| Box::new(e) as _)
}
pub fn stop(&mut self) -> crate::Result<()> {
crate::uvret(unsafe { uv_fs_event_stop(self.handle) })
}
pub fn getpath(&self) -> crate::Result<String> {
let mut size = 0usize;
let result = crate::uvret(unsafe {
uv_fs_event_getpath(self.handle, std::ptr::null_mut(), &mut size as _)
});
if let Err(e) = result {
if e != crate::Error::ENOBUFS {
return Err(e);
}
}
let mut buf: Vec<std::os::raw::c_uchar> = Vec::with_capacity(size as _);
crate::uvret(unsafe {
uv_fs_event_getpath(self.handle, buf.as_mut_ptr() as _, &mut size as _)
})
.map(|_| {
unsafe { buf.set_len((size as usize) + 1) };
unsafe { CStr::from_bytes_with_nul_unchecked(&buf) }
.to_string_lossy()
.into_owned()
})
}
}
impl FromInner<*mut uv_fs_event_t> for FsEventHandle {
fn from_inner(handle: *mut uv_fs_event_t) -> FsEventHandle {
FsEventHandle { handle }
}
}
impl Inner<*mut uv::uv_handle_t> for FsEventHandle {
fn inner(&self) -> *mut uv::uv_handle_t {
uv_handle!(self.handle)
}
}
impl From<FsEventHandle> for crate::Handle {
fn from(fs_event: FsEventHandle) -> crate::Handle {
crate::Handle::from_inner(Inner::<*mut uv::uv_handle_t>::inner(&fs_event))
}
}
impl crate::ToHandle for FsEventHandle {
fn to_handle(&self) -> crate::Handle {
crate::Handle::from_inner(Inner::<*mut uv::uv_handle_t>::inner(self))
}
}
impl TryFrom<crate::Handle> for FsEventHandle {
type Error = crate::ConversionError;
fn try_from(handle: crate::Handle) -> Result<Self, Self::Error> {
let t = handle.get_type();
if t != crate::HandleType::FS_EVENT {
Err(crate::ConversionError::new(t, crate::HandleType::FS_EVENT))
} else {
Ok((handle.inner() as *mut uv_fs_event_t).into_inner())
}
}
}
impl HandleTrait for FsEventHandle {}
impl crate::Loop {
pub fn fs_event(&self) -> crate::Result<FsEventHandle> {
FsEventHandle::new(self)
}
}