From d47174601a5e381ee709fa5e83d8f00b7c4f2f5c Mon Sep 17 00:00:00 2001
From: William Venner <william@venner.io>
Date: Tue, 21 Sep 2021 18:13:19 +0100
Subject: [PATCH] add some lua bindings

---
 gmod/src/lua/import.rs    |  6 ++++++
 gmod/src/lua/lua_state.rs | 11 +++++++++++
 2 files changed, 17 insertions(+)

diff --git a/gmod/src/lua/import.rs b/gmod/src/lua/import.rs
index eed1a6c..d2de566 100644
--- a/gmod/src/lua/import.rs
+++ b/gmod/src/lua/import.rs
@@ -1,3 +1,5 @@
+use std::ffi::c_void;
+
 use libloading::{Library, Symbol};
 
 macro_rules! find_library {
@@ -78,6 +80,7 @@ pub(crate) struct LuaShared {
 	pub lual_loadbuffer: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, buff: LuaString, sz: LuaSize, name: LuaString) -> i32>,
 	pub lua_getfield: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, index: i32, k: LuaString)>,
 	pub lua_pushvalue: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, index: i32)>,
+	pub lua_pushlightuserdata: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, data: *mut c_void)>,
 	pub lua_pushboolean: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, bool: i32)>,
 	pub lua_tolstring: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, index: i32, out_size: *mut LuaSize) -> LuaString>,
 	pub lua_pcall: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, nargs: i32, nresults: i32, errfunc: i32) -> i32>,
@@ -99,6 +102,7 @@ pub(crate) struct LuaShared {
 	pub lual_checkinteger: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, arg: i32) -> LuaInt>,
 	pub lual_checklstring: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, arg: i32, out_size: *mut LuaSize) -> LuaString>,
 	pub lua_toboolean: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, index: i32) -> i32>,
+	pub lual_checktype: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, index: i32, r#type: i32)>,
 	pub lua_setmetatable: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, index: i32) -> i32>,
 	pub lua_pushinteger: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, int: LuaInt)>,
 	pub lua_pushnumber: Symbol<'static, unsafe extern "C-unwind" fn(state: LuaState, int: LuaNumber)>,
@@ -133,6 +137,8 @@ impl LuaShared {
 			}
 
 			Self {
+				lua_pushlightuserdata: find_symbol!("lua_pushlightuserdata"),
+				lual_checktype: find_symbol!("luaL_checktype"),
 				lual_loadfile: find_symbol!("luaL_loadfile"),
 				lual_loadstring: find_symbol!("luaL_loadstring"),
 				lual_loadbuffer: find_symbol!("luaL_loadbuffer"),
diff --git a/gmod/src/lua/lua_state.rs b/gmod/src/lua/lua_state.rs
index e844658..a4c5acd 100644
--- a/gmod/src/lua/lua_state.rs
+++ b/gmod/src/lua/lua_state.rs
@@ -110,6 +110,11 @@ impl LuaState {
 		(LUA_SHARED.lua_pushvalue)(*self, index)
 	}
 
+	#[inline]
+	pub unsafe fn push_lightuserdata(&self, data: *mut c_void) {
+		(LUA_SHARED.lua_pushlightuserdata)(*self, data)
+	}
+
 	#[inline]
 	pub unsafe fn get_field(&self, index: i32, k: LuaString) {
 		(LUA_SHARED.lua_getfield)(*self, index, k)
@@ -320,6 +325,12 @@ impl LuaState {
 		(LUA_SHARED.lual_checknumber)(*self, arg)
 	}
 
+	#[inline]
+	pub unsafe fn check_boolean(&self, arg: i32) -> bool {
+		(LUA_SHARED.lual_checktype)(*self, arg, LUA_TBOOLEAN);
+		(LUA_SHARED.lua_toboolean)(*self, arg) == 1
+	}
+
 	#[inline]
 	pub unsafe fn to_integer(&self, index: i32) -> LuaInt {
 		(LUA_SHARED.lua_tointeger)(*self, index)