use crate::{FromInner, HandleTrait, Inner, IntoInner};
use std::convert::TryFrom;
use uv::{uv_signal_init, uv_signal_start, uv_signal_start_oneshot, uv_signal_stop, uv_signal_t};
callbacks! {
pub SignalCB(handle: SignalHandle, signum: i32);
}
#[derive(Default)]
pub(crate) struct SignalDataFields<'a> {
signal_cb: SignalCB<'a>,
}
extern "C" fn uv_signal_cb(handle: *mut uv_signal_t, signum: std::os::raw::c_int) {
let dataptr = crate::Handle::get_data(uv_handle!(handle));
if !dataptr.is_null() {
unsafe {
if let super::SignalData(d) = &mut (*dataptr).addl {
d.signal_cb.call(handle.into_inner(), signum as _);
}
}
}
}
#[derive(Clone, Copy)]
pub struct SignalHandle {
handle: *mut uv_signal_t,
}
impl SignalHandle {
pub fn new(r#loop: &crate::Loop) -> crate::Result<SignalHandle> {
let layout = std::alloc::Layout::new::<uv_signal_t>();
let handle = unsafe { std::alloc::alloc(layout) as *mut uv_signal_t };
if handle.is_null() {
return Err(crate::Error::ENOMEM);
}
let ret = unsafe { uv_signal_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::SignalData(Default::default()));
Ok(SignalHandle { handle })
}
pub fn start<CB: Into<SignalCB<'static>>>(&mut self, cb: CB, signum: i32) -> crate::Result<()> {
let cb = cb.into();
let uv_cb = use_c_callback!(uv_signal_cb, cb);
let dataptr = crate::Handle::get_data(uv_handle!(self.handle));
if !dataptr.is_null() {
if let super::SignalData(d) = unsafe { &mut (*dataptr).addl } {
d.signal_cb = cb;
}
}
crate::uvret(unsafe { uv_signal_start(self.handle, uv_cb, signum as _) })
}
pub fn start_oneshot<CB: Into<SignalCB<'static>>>(
&mut self,
cb: CB,
signum: i32,
) -> crate::Result<()> {
let cb = cb.into();
let uv_cb = use_c_callback!(uv_signal_cb, cb);
let dataptr = crate::Handle::get_data(uv_handle!(self.handle));
if !dataptr.is_null() {
if let super::SignalData(d) = unsafe { &mut (*dataptr).addl } {
d.signal_cb = cb;
}
}
crate::uvret(unsafe { uv_signal_start_oneshot(self.handle, uv_cb, signum as _) })
}
pub fn stop(&mut self) -> crate::Result<()> {
crate::uvret(unsafe { uv_signal_stop(self.handle) })
}
pub fn signum(&self) -> i32 {
unsafe { (*self.handle).signum }
}
}
impl FromInner<*mut uv_signal_t> for SignalHandle {
fn from_inner(handle: *mut uv_signal_t) -> SignalHandle {
SignalHandle { handle }
}
}
impl Inner<*mut uv::uv_handle_t> for SignalHandle {
fn inner(&self) -> *mut uv::uv_handle_t {
uv_handle!(self.handle)
}
}
impl From<SignalHandle> for crate::Handle {
fn from(signal: SignalHandle) -> crate::Handle {
crate::Handle::from_inner(Inner::<*mut uv::uv_handle_t>::inner(&signal))
}
}
impl crate::ToHandle for SignalHandle {
fn to_handle(&self) -> crate::Handle {
crate::Handle::from_inner(Inner::<*mut uv::uv_handle_t>::inner(self))
}
}
impl TryFrom<crate::Handle> for SignalHandle {
type Error = crate::ConversionError;
fn try_from(handle: crate::Handle) -> Result<Self, Self::Error> {
let t = handle.get_type();
if t != crate::HandleType::SIGNAL {
Err(crate::ConversionError::new(t, crate::HandleType::SIGNAL))
} else {
Ok((handle.inner() as *mut uv_signal_t).into_inner())
}
}
}
impl HandleTrait for SignalHandle {}
impl crate::Loop {
pub fn signal(&self) -> crate::Result<SignalHandle> {
SignalHandle::new(self)
}
}