100 lines
2.7 KiB
TypeScript
100 lines
2.7 KiB
TypeScript
import { luaBuildStandardEnv } from "$common/space_lua/stdlib.ts";
|
|
import { parsePageRef } from "@silverbulletmd/silverbullet/lib/page_ref";
|
|
import {
|
|
LuaEnv,
|
|
LuaNativeJSFunction,
|
|
LuaStackFrame,
|
|
LuaTable,
|
|
} from "$common/space_lua/runtime.ts";
|
|
import type { System } from "$lib/plugos/system.ts";
|
|
|
|
export function buildLuaEnv(system: System<any>) {
|
|
const env = new LuaEnv(luaBuildStandardEnv());
|
|
|
|
// Expose all syscalls to Lua
|
|
exposeSyscalls(env, system);
|
|
|
|
return env;
|
|
}
|
|
|
|
function exposeSyscalls(env: LuaEnv, system: System<any>) {
|
|
// Expose all syscalls to Lua
|
|
// Except...
|
|
const blacklist = ["template", "shell"];
|
|
const nativeFs = new LuaStackFrame(env, null);
|
|
for (const syscallName of system.registeredSyscalls.keys()) {
|
|
if (blacklist.includes(syscallName)) {
|
|
continue;
|
|
}
|
|
const [ns, fn] = syscallName.split(".");
|
|
if (!env.has(ns)) {
|
|
env.set(ns, new LuaTable(), nativeFs);
|
|
}
|
|
const luaFn = new LuaNativeJSFunction((...args) => {
|
|
return system.localSyscall(syscallName, args);
|
|
});
|
|
// Register the function with the same name as the syscall both in regular and snake_case
|
|
env.get(ns, nativeFs).set(fn, luaFn, nativeFs);
|
|
env.get(ns, nativeFs).set(snakeCase(fn), luaFn, nativeFs);
|
|
}
|
|
}
|
|
|
|
export async function buildThreadLocalEnv(
|
|
system: System<any>,
|
|
globalEnv: LuaEnv,
|
|
) {
|
|
const tl = new LuaEnv();
|
|
if (system.registeredSyscalls.has("editor.getCurrentPageMeta")) {
|
|
const currentPageMeta = await system.localSyscall(
|
|
"editor.getCurrentPageMeta",
|
|
[],
|
|
);
|
|
if (currentPageMeta) {
|
|
tl.setLocal("currentPage", currentPageMeta);
|
|
} else {
|
|
tl.setLocal("currentPage", {
|
|
name: await system.localSyscall("editor.getCurrentPage", []),
|
|
});
|
|
}
|
|
}
|
|
tl.setLocal("_GLOBAL", globalEnv);
|
|
return Promise.resolve(tl);
|
|
}
|
|
|
|
export async function handleLuaError(e: any, system: System<any>) {
|
|
console.error(
|
|
"Lua eval exception",
|
|
e.message,
|
|
e.context,
|
|
);
|
|
if (e.context && e.context.ref) {
|
|
// We got an error and actually know where it came from, let's navigate there to help debugging
|
|
const pageRef = parsePageRef(e.context.ref);
|
|
await system.localSyscall(
|
|
"editor.flashNotification",
|
|
[
|
|
`Lua error: ${e.message}`,
|
|
"error",
|
|
],
|
|
);
|
|
await system.localSyscall(
|
|
"editor.flashNotification",
|
|
[
|
|
`Navigating to the place in the code where this error occurred in ${pageRef.page}`,
|
|
"info",
|
|
],
|
|
);
|
|
await system.localSyscall("editor.navigate", [
|
|
{
|
|
page: pageRef.page,
|
|
pos: pageRef.pos + e.context.from +
|
|
"```space-lua\n".length,
|
|
},
|
|
]);
|
|
}
|
|
}
|
|
|
|
function snakeCase(s: string) {
|
|
return s.replace(/([A-Z])/g, "_$1").toLowerCase();
|
|
}
|