endurox-rs 0.1.1

use crate::{raw, AtmiCtx, TypedBuffer, TypedUbf};
use core::ffi::c_char;
use std::ffi::CStr;

/// Safe Rust wrapper for TPSVCINFO passed into a service callback.
///
/// Owns the service buffer until the handler calls `take_data()` /
/// `take_data_ubf()` and passes it to `tpreturn_ubf`.  If the handler
/// returns without taking the data the buffer is freed automatically via
/// the `TypedBuffer` Drop impl (correct: service must either return or free it).
#[derive(Debug)]
pub struct TpSvcInfo<'ctx> {
    raw: *mut raw::TPSVCINFO,
    ctx: &'ctx AtmiCtx,
    /// Owned data buffer from TPSVCINFO::data.  Moved out on take_data*().
    data: Option<TypedBuffer<'ctx>>,
}

impl<'ctx> TpSvcInfo<'ctx> {
    /// # Safety
    /// - `raw` must be a valid TPSVCINFO pointer supplied by XATMI.
    /// - `ctx` must be the current ATMI context for this thread.
    pub(crate) unsafe fn from_raw(ctx: &'ctx AtmiCtx, raw: *mut raw::TPSVCINFO) -> Self {
        let data_ptr = (*raw).data as *mut c_char;
        let data = if data_ptr.is_null() {
            None
        } else {
            Some(TypedBuffer::from_raw(ctx, data_ptr))
        };
        TpSvcInfo { raw, ctx, data }
    }

    #[inline]
    fn raw(&self) -> &raw::TPSVCINFO {
        unsafe { &*self.raw }
    }

    #[inline]
    fn raw_mut(&mut self) -> &mut raw::TPSVCINFO {
        unsafe { &mut *self.raw }
    }

    /// Name of the service.
    pub fn name(&self) -> &str {
        unsafe {
            CStr::from_ptr(self.raw().name.as_ptr())
                .to_str()
                .unwrap_or("")
        }
    }

    /// Name of the advertised function (may differ from service name).
    pub fn fname(&self) -> &str {
        unsafe {
            CStr::from_ptr(self.raw().fname.as_ptr())
                .to_str()
                .unwrap_or("")
        }
    }

    /// Input buffer length as reported by the framework.
    pub fn len(&self) -> i64 {
        self.raw().len
    }

    pub fn set_len(&mut self, len: i64) {
        self.raw_mut().len = len;
    }

    pub fn flags(&self) -> i64 {
        self.raw().flags
    }

    pub fn set_flags(&mut self, flags: i64) {
        self.raw_mut().flags = flags;
    }

    pub fn cd(&self) -> i32 {
        self.raw().cd
    }

    pub fn appkey(&self) -> i64 {
        self.raw().appkey
    }

    pub fn cltid(&self) -> crate::ClientId {
        self.raw().cltid
    }

    pub(crate) fn data_ptr(&self) -> *mut c_char {
        self.data
            .as_ref()
            .map(|b| b.as_ptr())
            .unwrap_or(std::ptr::null_mut())
    }

    /// Transfer ownership of the service data buffer.
    ///
    /// Returns `None` if the buffer was already taken or was originally null.
    /// After calling this, the caller is responsible for passing the buffer
    /// to `tpreturn_buffer` (or freeing it before calling `tpreturn`).
    pub fn take_data(&mut self) -> Option<TypedBuffer<'ctx>> {
        self.data.take()
    }

    /// Take the service data buffer as a typed UBF buffer.
    ///
    /// Convenience wrapper over `take_data()` for the common case where the
    /// request was sent as a UBF message.
    pub fn take_data_ubf(&mut self) -> Option<TypedUbf<'ctx>> {
        self.data.take().map(TypedUbf::from_typed)
    }
}