xous-core/services/ime-plugin-api/src/lib.rs

#![cfg_attr(target_os = "none", no_std)]

use num_traits::{FromPrimitive, ToPrimitive};
use xous::{send_message, Message, CID};
use xous_ipc::{Buffer, String};

#[derive(Debug, rkyv::Archive, rkyv::Serialize, rkyv::Deserialize)]
pub struct Prediction {
    pub index: u32,
    pub valid: bool,
    pub string: String<1000>,
}

#[derive(Debug, Default, Copy, Clone)]
pub struct PredictionTriggers {
    /// trigger line predictions on newline -- if set, sends the *whole* line to the predictor
    /// if just wanting the last word, set `punctuation = true`
    pub newline: bool,
    /// trigger word predictions punctuation
    pub punctuation: bool,
    /// trigger word predictions on whitespace
    pub whitespace: bool,
}
impl Into<usize> for PredictionTriggers {
    fn into(self) -> usize {
        let mut ret: usize = 0;
        if self.newline {
            ret |= 0x1;
        }
        if self.punctuation {
            ret |= 0x2;
        }
        if self.whitespace {
            ret |= 0x4;
        }
        ret
    }
}
impl From<usize> for PredictionTriggers {
    fn from(code: usize) -> PredictionTriggers {
        PredictionTriggers {
            newline: (code & 0x1) != 0,
            punctuation: (code & 0x2) != 0,
            whitespace: (code & 0x4) != 0,
        }
    }
}

#[derive(Debug, num_derive::FromPrimitive, num_derive::ToPrimitive)]
pub enum Opcode {
    /// update with the latest input candidate. Replaces the previous input.
    Input, //(String<4000>),

    /// feed back to the IME plugin as to what was picked, so predictions can be updated
    Picked, //(String<4000>),

    /// Undo the last Picked value. To be used when a user hits backspace after picking a prediction
    /// note that repeated calls to Unpick will have an implementation-defined behavior
    Unpick,

    /// fetch the prediction at a given index, where the index is ordered from 0..N, where 0 is the most likely prediction
    /// if there is no prediction available, just return an empty string
    Prediction, //(Prediction),

    /// return the prediction triggers used by this IME. These are characters that can indicate that a
    /// whole predictive unit has been entered.
    GetPredictionTriggers,

    Quit,
}

#[derive(Debug, rkyv::Archive, rkyv::Serialize, rkyv::Deserialize)]
pub enum Return {
    Prediction(Prediction),
    Failure,
}

pub trait PredictionApi {
    fn get_prediction_triggers(&self) -> Result<PredictionTriggers, xous::Error>;
    fn unpick(&self) -> Result<(), xous::Error>;
    fn set_input(&self, s: String<4000>) -> Result<(), xous::Error>;
    fn feedback_picked(&self, s: String<4000>) -> Result<(), xous::Error>;
    fn get_prediction(&self, index: u32) -> Result<Option<String<4000>>, xous::Error>;
}

// provide a convenience version of the API for generic/standard calls
#[derive(Debug, Default, Copy, Clone)]
pub struct PredictionPlugin {
    pub connection: Option<CID>,
}

impl PredictionApi for PredictionPlugin {
    fn get_prediction_triggers(&self) -> Result<PredictionTriggers, xous::Error> {
        match self.connection {
            Some(cid) => {
                let response = send_message(
                    cid,
                    Message::new_blocking_scalar(
                        Opcode::GetPredictionTriggers.to_usize().unwrap(),
                        0,
                        0,
                        0,
                        0,
                    ),
                )?;
                if let xous::Result::Scalar1(code) = response {
                    Ok(code.into())
                } else {
                    Err(xous::Error::InternalError)
                }
            }
            _ => Err(xous::Error::UseBeforeInit),
        }
    }

    fn unpick(&self) -> Result<(), xous::Error> {
        match self.connection {
            Some(cid) => {
                send_message(
                    cid,
                    Message::new_scalar(Opcode::Unpick.to_usize().unwrap(), 0, 0, 0, 0),
                )?;
                Ok(())
            }
            _ => Err(xous::Error::UseBeforeInit),
        }
    }

    fn set_input(&self, s: String<4000>) -> Result<(), xous::Error> {
        match self.connection {
            Some(cid) => {
                let buf = Buffer::into_buf(s).or(Err(xous::Error::InternalError))?;
                buf.lend(cid, Opcode::Input.to_u32().unwrap())
                    .expect("|API: set_input operation failure");
                Ok(())
            }
            _ => Err(xous::Error::UseBeforeInit),
        }
    }

    fn feedback_picked(&self, s: String<4000>) -> Result<(), xous::Error> {
        match self.connection {
            Some(cid) => {
                let buf = Buffer::into_buf(s).or(Err(xous::Error::InternalError))?;
                buf.lend(cid, Opcode::Picked.to_u32().unwrap())
                    .expect("|API: feedback_picked operation failure");
                Ok(())
            }
            _ => Err(xous::Error::UseBeforeInit),
        }
    }

    fn get_prediction(&self, index: u32) -> Result<Option<String<4000>>, xous::Error> {
        match self.connection {
            Some(cid) => {
                let prediction = Prediction {
                    index,
                    string: String::<1000>::new(),
                    valid: false,
                };
                let mut buf = Buffer::into_buf(prediction).or(Err(xous::Error::InternalError))?;
                buf.lend_mut(cid, Opcode::Prediction.to_u32().unwrap())
                    .or(Err(xous::Error::InternalError))?;

                log::trace!("IME|API: returned from get_prediction");

                match buf.to_original().unwrap() {
                    Return::Prediction(pred) => {
                        log::trace!("|API: got {:?}", pred);
                        if pred.valid {
                            let mut ret = String::<4000>::new();
                            use core::fmt::Write as CoreWrite;
                            write!(ret, "{}", pred.string).unwrap();
                            Ok(Some(ret))
                        } else {
                            Ok(None)
                        }
                    }
                    _ => {
                        log::error!("API get_prediction returned an invalid result");
                        Err(xous::Error::InternalError)
                    }
                }
            }
            _ => Err(xous::Error::UseBeforeInit),
        }
    }
}

//////////////////////////////////////////////////////
//////////////////// FRONT END API
//////////////////////////////////////////////////////
// Most people won't need to touch this, but it's packaged
// in this crate so we can break circular dependencies
// between the IMEF, GAM, and graphics server

#[derive(Debug, num_derive::FromPrimitive, num_derive::ToPrimitive)]
pub enum ImefOpcode {
    /// connect an backend configuration
    ConnectBackend,

    /// register a listener for finalized inputs
    RegisterListener, //(String<64>),

    /// internal use for passing keyboard events from the keyboard callback
    ProcessKeys,

    /// force a redraw of the UI
    Redraw,

    Quit,
}
#[derive(Debug, num_derive::FromPrimitive, num_derive::ToPrimitive)]
pub enum ImefCallback {
    GotInputLine, //(String<4000>),
    Drop,
}

#[derive(Debug, rkyv::Archive, rkyv::Serialize, rkyv::Deserialize, Copy, Clone)]
pub struct ImefDescriptor {
    pub input_canvas: Option<graphics_server::Gid>,
    pub prediction_canvas: Option<graphics_server::Gid>,
    pub predictor: Option<String<64>>,
    pub token: [u32; 4], // token used to lookup our connected app inside the GAM
}

pub trait ImeFrontEndApi {
    fn connect_backend(&self, descriptor: ImefDescriptor) -> Result<(), xous::Error>;
    fn hook_listener_callback(&mut self, cb: fn(String<4000>)) -> Result<(), xous::Error>;
    fn redraw(&self, force_all: bool) -> Result<(), xous::Error>;
    fn send_keyevent(&self, keys: [char; 4]) -> Result<(), xous::Error>;
    fn conn(&self) -> xous::CID;
    fn getop_process_keys(&self) -> u32;
}

pub const SERVER_NAME_IME_FRONT: &str = "_IME front end_";
static mut INPUT_CB: Option<fn(String<4000>)> = None;

pub struct ImeFrontEnd {
    cid: CID,
    callback_sid: Option<xous::SID>,
}
impl ImeFrontEnd {
    pub fn new(xns: &xous_names::XousNames) -> Result<Self, xous::Error> {
        REFCOUNT.store(REFCOUNT.load(Ordering::Relaxed) + 1, Ordering::Relaxed);
        let conn = xns
            .request_connection_blocking(SERVER_NAME_IME_FRONT)
            .expect("Can't connect to IMEF");
        Ok(ImeFrontEnd {
            cid: conn,
            callback_sid: None,
        })
    }
}
use core::sync::atomic::{AtomicU32, Ordering};
static REFCOUNT: AtomicU32 = AtomicU32::new(0);
impl Drop for ImeFrontEnd {
    fn drop(&mut self) {
        if let Some(sid) = self.callback_sid.take() {
            // no need to tell the pstream server we're quitting: the next time a callback processes,
            // it will automatically remove my entry as it will receive a ServerNotFound error.

            // tell my handler thread to quit
            let cid = xous::connect(sid).unwrap();
            xous::send_message(
                cid,
                Message::new_blocking_scalar(ImefCallback::Drop.to_usize().unwrap(), 0, 0, 0, 0),
            )
            .unwrap();
            unsafe {
                xous::disconnect(cid).unwrap();
            }
            xous::destroy_server(sid).unwrap();
        }
        if REFCOUNT.load(Ordering::Relaxed) == 0 {
            unsafe {
                xous::disconnect(self.cid).unwrap();
            }
        }
    }
}

impl ImeFrontEndApi for ImeFrontEnd {
    fn conn(&self) -> xous::CID {
        self.cid
    }
    fn getop_process_keys(&self) -> u32 {
        ImefOpcode::ProcessKeys.to_u32().unwrap()
    }
    fn connect_backend(&self, descriptor: ImefDescriptor) -> Result<(), xous::Error> {
        let buf = Buffer::into_buf(descriptor).or(Err(xous::Error::InternalError))?;
        buf.lend(self.cid, ImefOpcode::ConnectBackend.to_u32().unwrap())
            .or(Err(xous::Error::InternalError))
            .map(|_| ())
    }

    fn send_keyevent(&self, keys: [char; 4]) -> Result<(), xous::Error> {
        log::trace!("sending keys: {:?}", keys);
        xous::send_message(
            self.cid,
            xous::Message::new_scalar(
                ImefOpcode::ProcessKeys.to_usize().unwrap(),
                keys[0] as u32 as usize,
                keys[1] as u32 as usize,
                keys[2] as u32 as usize,
                keys[3] as u32 as usize,
            ),
        )
        .map(|_| ())
    }

    fn hook_listener_callback(&mut self, cb: fn(String<4000>)) -> Result<(), xous::Error> {
        if unsafe { INPUT_CB }.is_some() {
            return Err(xous::Error::MemoryInUse); // can't hook it twice
        }
        unsafe { INPUT_CB = Some(cb) };
        if self.callback_sid.is_none() {
            let sid = xous::create_server().unwrap();
            self.callback_sid = Some(sid);
            let sid_tuple = sid.to_u32();
            xous::create_thread_4(
                callback_server,
                sid_tuple.0 as usize,
                sid_tuple.1 as usize,
                sid_tuple.2 as usize,
                sid_tuple.3 as usize,
            )
            .unwrap();
            xous::send_message(
                self.cid,
                Message::new_scalar(
                    ImefOpcode::RegisterListener.to_usize().unwrap(),
                    sid_tuple.0 as usize,
                    sid_tuple.1 as usize,
                    sid_tuple.2 as usize,
                    sid_tuple.3 as usize,
                ),
            )
            .unwrap();
        }
        Ok(())
    }

    fn redraw(&self, force_all: bool) -> Result<(), xous::Error> {
        let arg = if force_all { 1 } else { 0 };
        send_message(
            self.cid,
            Message::new_scalar(ImefOpcode::Redraw.to_usize().unwrap(), arg, 0, 0, 0),
        )?;
        Ok(())
    }
}

/// handles callback messages from server, in the library user's process space.
fn callback_server(sid0: usize, sid1: usize, sid2: usize, sid3: usize) {
    let sid = xous::SID::from_u32(sid0 as u32, sid1 as u32, sid2 as u32, sid3 as u32);
    loop {
        let msg = xous::receive_message(sid).unwrap();
        match FromPrimitive::from_usize(msg.body.id()) {
            Some(ImefCallback::GotInputLine) => {
                let buffer =
                    unsafe { Buffer::from_memory_message(msg.body.memory_message().unwrap()) };
                let inputline = buffer.to_original::<String<4000>, _>().unwrap();
                unsafe {
                    if let Some(cb) = INPUT_CB {
                        cb(inputline)
                    }
                }
            }
            Some(ImefCallback::Drop) => {
                break; // this exits the loop and kills the thread
            }
            None => (),
        }
    }
}