codemp/src/ffi/lua/mod.rs

mod client;
mod workspace;
mod cursor;
mod buffer;
mod ext;

use mlua_codemp_patch as mlua;
use mlua::prelude::*;
use crate::prelude::*;

// define multiple entrypoints, so this library can have multiple names and still work
#[mlua::lua_module(name = "codemp")] fn entry_1(lua: &Lua) -> LuaResult<LuaTable> { entrypoint(lua) }
#[mlua::lua_module(name = "libcodemp")] fn entry_2(lua: &Lua) -> LuaResult<LuaTable> { entrypoint(lua) }
#[mlua::lua_module(name = "codemp_native")] fn entry_3(lua: &Lua) -> LuaResult<LuaTable> { entrypoint(lua) }
#[mlua::lua_module(name = "codemp_lua")] fn entry_4(lua: &Lua) -> LuaResult<LuaTable> { entrypoint(lua) }

fn entrypoint(lua: &Lua) -> LuaResult<LuaTable> {
	let exports = lua.create_table()?;

	// entrypoint
	exports.set("connect", lua.create_function(|_, (config,):(CodempConfig,)|
		ext::a_sync::a_sync! { => CodempClient::connect(config).await? }
	)?)?;

	// utils
	exports.set("hash", lua.create_function(|_, (txt,):(String,)|
		Ok(crate::ext::hash(txt))
	)?)?;

	exports.set("version", lua.create_function(|_, ()|
		Ok(crate::version())
	)?)?;

	// runtime
	exports.set("setup_driver", lua.create_function(ext::a_sync::setup_driver)?)?;
	exports.set("poll_callback", lua.create_function(|lua, ()| {
		let mut val = LuaMultiValue::new();
		match ext::callback().recv() {
			None => {},
			Some(ext::callback::LuaCallback::Invoke(cb, arg)) => {
				val.push_back(LuaValue::Function(cb));
				val.push_back(arg.into_lua(lua)?);
			}
			Some(ext::callback::LuaCallback::Fail(msg)) => {
				val.push_back(false.into_lua(lua)?);
				val.push_back(msg.into_lua(lua)?);
			},
		}
		Ok(val)
	})?)?;

	// logging
	exports.set("setup_tracing", lua.create_function(ext::log::setup_tracing)?)?;

	Ok(exports)
}

impl From::<crate::errors::ConnectionError> for LuaError {
	fn from(value: crate::errors::ConnectionError) -> Self {
		LuaError::runtime(value.to_string())
	}
}

impl From::<crate::errors::RemoteError> for LuaError {
	fn from(value: crate::errors::RemoteError) -> Self {
		LuaError::runtime(value.to_string())
	}
}

impl From::<crate::errors::ControllerError> for LuaError {
	fn from(value: crate::errors::ControllerError) -> Self {
		LuaError::runtime(value.to_string())
	}
}