Skip to main content

luaur_vm/functions/
utfchar.rs

1use crate::functions::buffutfchar::buffutfchar;
2use crate::functions::lua_gettop::lua_gettop;
3use crate::functions::lua_l_addlstring::lua_l_addlstring;
4use crate::functions::lua_l_buffinit::lua_l_buffinit;
5use crate::functions::lua_l_pushresult::lua_l_pushresult;
6use crate::functions::lua_pushlstring::lua_pushlstring;
7use crate::records::lua_l_strbuf::LuaLStrbuf;
8use crate::type_aliases::lua_state::lua_State;
9
10use core::ffi::{c_char, c_int};
11
12const UTF8BUFFSZ: usize = 8;
13
14#[no_mangle]
15pub unsafe fn utfchar(L: *mut lua_State) -> c_int {
16    let mut buff = [0 as c_char; UTF8BUFFSZ];
17    let mut charstr = core::ptr::null::<c_char>();
18
19    let n = lua_gettop(L); // number of arguments
20    if n == 1 {
21        // optimize common case of single char
22        let l = buffutfchar(L, 1, buff.as_mut_ptr(), &mut charstr as *mut *const c_char);
23        lua_pushlstring(L, charstr, l as usize);
24    } else {
25        let mut b = LuaLStrbuf {
26            p: core::ptr::null_mut(),
27            end: core::ptr::null_mut(),
28            L: core::ptr::null_mut(),
29            storage: core::ptr::null_mut(),
30            buffer: [0; 512],
31        };
32        lua_l_buffinit(L, &mut b as *mut LuaLStrbuf);
33        let mut i = 1;
34        while i <= n {
35            let l = buffutfchar(L, i, buff.as_mut_ptr(), &mut charstr as *mut *const c_char);
36            lua_l_addlstring(&mut b as *mut LuaLStrbuf, charstr, l as usize);
37            i += 1;
38        }
39        lua_l_pushresult(&mut b as *mut LuaLStrbuf);
40    }
41    1
42}