mirror of
https://github.com/pinnacle-comp/pinnacle.git
synced 2025-01-13 08:01:05 +01:00
Restructure lua api for future work
This commit is contained in:
parent
a9a7f3a0d2
commit
a885d8dec6
3 changed files with 94 additions and 86 deletions
|
@ -1,80 +0,0 @@
|
|||
-- require("luarocks.loader") TODO:
|
||||
|
||||
local socket = require("posix.sys.socket")
|
||||
local msgpack = require("msgpack")
|
||||
|
||||
local SOCKET_PATH = "/tmp/pinnacle_socket"
|
||||
|
||||
---@type integer
|
||||
local socket_fd = assert(socket.socket(socket.AF_UNIX, socket.SOCK_STREAM, 0), "Failed to create socket")
|
||||
print("created socket at fd " .. socket_fd)
|
||||
|
||||
assert(0 == socket.connect(socket_fd, {
|
||||
family = socket.AF_UNIX,
|
||||
path = SOCKET_PATH,
|
||||
}), "Failed to connect to Pinnacle socket")
|
||||
|
||||
function SendMsg(data)
|
||||
local encoded = msgpack.encode(data)
|
||||
assert(encoded)
|
||||
local len = encoded:len()
|
||||
socket.send(socket_fd, string.pack("=I4", len))
|
||||
socket.send(socket_fd, encoded)
|
||||
end
|
||||
|
||||
---@type fun()[]
|
||||
CallbackTable = {}
|
||||
|
||||
local CONFIG_PATH = os.getenv("PINNACLE_CONFIG")
|
||||
dofile(CONFIG_PATH)
|
||||
|
||||
---Read the specified number of bytes.
|
||||
---@param socket_fd integer The socket file descriptor
|
||||
---@param count integer The amount of bytes to read
|
||||
---@return string|nil data
|
||||
---@return string|nil err_msg
|
||||
---@return integer|nil err_num
|
||||
local function read_exact(socket_fd, count)
|
||||
local len_to_read = count
|
||||
local data = ""
|
||||
while len_to_read > 0 do
|
||||
local bytes, err_msg, errnum = socket.recv(socket_fd, len_to_read)
|
||||
|
||||
if bytes == nil then
|
||||
-- TODO: handle errors
|
||||
print("bytes was nil")
|
||||
return bytes, err_msg, errnum
|
||||
end
|
||||
|
||||
---@type integer
|
||||
local recv_len = bytes:len()
|
||||
|
||||
if recv_len == 0 then
|
||||
print("stream closed")
|
||||
break
|
||||
end
|
||||
|
||||
len_to_read = len_to_read - recv_len
|
||||
assert(len_to_read >= 0, "Overread message boundary")
|
||||
|
||||
data = data .. bytes
|
||||
end
|
||||
return data
|
||||
end
|
||||
|
||||
while true do
|
||||
local msg_len_bytes, err_msg, err_num = read_exact(socket_fd, 4)
|
||||
assert(msg_len_bytes)
|
||||
|
||||
---@type integer
|
||||
local msg_len = string.unpack("=I4", msg_len_bytes)
|
||||
|
||||
local msg_bytes, err_msg2, err_num2 = read_exact(socket_fd, msg_len)
|
||||
assert(msg_bytes)
|
||||
|
||||
local tb = msgpack.decode(msg_bytes)
|
||||
|
||||
if tb.CallCallback then
|
||||
CallbackTable[tb.CallCallback]()
|
||||
end
|
||||
end
|
92
api/lua/pinnacle.lua
Normal file
92
api/lua/pinnacle.lua
Normal file
|
@ -0,0 +1,92 @@
|
|||
-- require("luarocks.loader") TODO:
|
||||
local socket = require("posix.sys.socket")
|
||||
local msgpack = require("msgpack")
|
||||
|
||||
local M = {}
|
||||
local SOCKET_PATH = "/tmp/pinnacle_socket"
|
||||
|
||||
---Read the specified number of bytes.
|
||||
---@param socket_fd integer The socket file descriptor
|
||||
---@param count integer The amount of bytes to read
|
||||
---@return string|nil data
|
||||
---@return string|nil err_msg
|
||||
---@return integer|nil err_num
|
||||
local function read_exact(socket_fd, count)
|
||||
local len_to_read = count
|
||||
local data = ""
|
||||
while len_to_read > 0 do
|
||||
local bytes, err_msg, errnum = socket.recv(socket_fd, len_to_read)
|
||||
|
||||
if bytes == nil then
|
||||
-- TODO: handle errors
|
||||
print("bytes was nil")
|
||||
return bytes, err_msg, errnum
|
||||
end
|
||||
|
||||
---@type integer
|
||||
local recv_len = bytes:len()
|
||||
|
||||
if recv_len == 0 then
|
||||
print("stream closed")
|
||||
break
|
||||
end
|
||||
|
||||
len_to_read = len_to_read - recv_len
|
||||
assert(len_to_read >= 0, "Overread message boundary")
|
||||
|
||||
data = data .. bytes
|
||||
end
|
||||
return data
|
||||
end
|
||||
|
||||
---@class Pinnacle
|
||||
local pinnacle = {
|
||||
input = require("input"),
|
||||
client = require("client"),
|
||||
}
|
||||
|
||||
---Configure Pinnacle. You should put mostly eveything into the config_func to avoid invalid state.
|
||||
---The function takes one argument: the Pinnacle table, which is how you'll access all of the available config options.
|
||||
---@param config_func fun(pinnacle: Pinnacle)
|
||||
function M.setup(config_func)
|
||||
---@type integer
|
||||
local socket_fd = assert(socket.socket(socket.AF_UNIX, socket.SOCK_STREAM, 0), "Failed to create socket")
|
||||
print("created socket at fd " .. socket_fd)
|
||||
|
||||
assert(0 == socket.connect(socket_fd, {
|
||||
family = socket.AF_UNIX,
|
||||
path = SOCKET_PATH,
|
||||
}), "Failed to connect to Pinnacle socket")
|
||||
|
||||
---@type fun()[]
|
||||
CallbackTable = {}
|
||||
|
||||
function SendMsg(data)
|
||||
local encoded = msgpack.encode(data)
|
||||
assert(encoded)
|
||||
local len = encoded:len()
|
||||
socket.send(socket_fd, string.pack("=I4", len))
|
||||
socket.send(socket_fd, encoded)
|
||||
end
|
||||
|
||||
config_func(pinnacle)
|
||||
|
||||
while true do
|
||||
local msg_len_bytes, err_msg, err_num = read_exact(socket_fd, 4)
|
||||
assert(msg_len_bytes)
|
||||
|
||||
---@type integer
|
||||
local msg_len = string.unpack("=I4", msg_len_bytes)
|
||||
|
||||
local msg_bytes, err_msg2, err_num2 = read_exact(socket_fd, msg_len)
|
||||
assert(msg_bytes)
|
||||
|
||||
local tb = msgpack.decode(msg_bytes)
|
||||
|
||||
if tb.CallCallback then
|
||||
CallbackTable[tb.CallCallback]()
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
|
@ -163,7 +163,7 @@ impl<B: Backend> State<B> {
|
|||
.unwrap()
|
||||
.to_string_lossy()
|
||||
.to_string();
|
||||
local_lua_path.push_str("/pinnacle_api_lua"); // TODO: get from crate root
|
||||
local_lua_path.push_str("/api/lua"); // TODO: get from crate root and do dynamically
|
||||
let new_lua_path =
|
||||
format!("{local_lua_path}/?.lua;{local_lua_path}/?/init.lua;{local_lua_path}/lib/?.lua;{local_lua_path}/lib/?/init.lua;{lua_path}");
|
||||
|
||||
|
@ -171,11 +171,7 @@ impl<B: Backend> State<B> {
|
|||
let new_lua_cpath = format!("{local_lua_path}/lib/?.so;{lua_cpath}");
|
||||
|
||||
std::process::Command::new("lua5.4")
|
||||
.arg(format!(
|
||||
"{}/pinnacle_api_lua/init.lua",
|
||||
std::env::current_dir().unwrap().to_string_lossy()
|
||||
))
|
||||
.env("PINNACLE_CONFIG", config_path)
|
||||
.arg(config_path)
|
||||
.env("LUA_PATH", new_lua_path)
|
||||
.env("LUA_CPATH", new_lua_cpath)
|
||||
.spawn()
|
||||
|
|
Loading…
Reference in a new issue