/
Lachklen
/
MYtgstation
Обзор
Документация
Войти
/
Lachklen
/
MYtgstation
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
code/modules/admin/verbs/lua/lua_state.dm
236 строк
8 KB
Lucy
Add a `DISABLE_DREAMLUAU` define, for disabling dreamluau at compile-time (#89359)
06 фев 2025, 04:27
Не верифицирован
06 фев 2025, 04:27
348413a
Код
Авторство
О чём код?
#ifndef DISABLE_DREAMLUAU #define MAX_LOG_REPEAT_LOOKBACK 5 GLOBAL_DATUM(lua_usr, /mob) GLOBAL_PROTECT(lua_usr) GLOBAL_LIST_EMPTY_TYPED(lua_state_stack, /datum/weakref) GLOBAL_PROTECT(lua_state_stack) /datum/lua_state var/display_name /// The internal ID of the lua state stored in dreamluau's state list var/internal_id /// A log of every return, yield, and error for each chunk execution and function call var/list/log = list() /// A list of all the variables in the state's environment var/list/globals = list() /// Ckey of the last user who ran a script on this lua state. var/ckey_last_runner = "" /// Whether the timer.lua script has been included into this lua context state. var/timer_enabled = FALSE /// Whether to supress logging BYOND runtimes for this state. var/supress_runtimes = FALSE /// Callbacks that need to be ran on next tick var/list/functions_to_execute = list() /datum/lua_state/vv_edit_var(var_name, var_value) . = ..() if(var_name == NAMEOF(src, internal_id)) return FALSE /datum/lua_state/New(_name) if(SSlua.initialized != TRUE) qdel(src) return display_name = _name internal_id = DREAMLUAU_NEW_STATE() if(isnull(internal_id)) stack_trace("dreamluau is not loaded") qdel(src) else if(!isnum(internal_id)) stack_trace(internal_id) qdel(src) /datum/lua_state/proc/check_if_slept(result) if(result["status"] == "sleep") SSlua.sleeps += src /datum/lua_state/proc/log_result(result, verbose = TRUE) if(!islist(result)) return var/status = result["status"] if(!verbose && status != "error" && status != "panic" && status != "runtime" && !(result["name"] == "input" && (status == "finished" || length(result["return_values"])))) return if(status == "runtime" && supress_runtimes) return var/append_to_log = TRUE var/index_of_log if(log.len) for(var/index in log.len to max(log.len - MAX_LOG_REPEAT_LOOKBACK, 1) step -1) var/list/entry = log[index] if(!compare_lua_logs(entry, result)) continue if(!entry["repeats"]) entry["repeats"] = 0 index_of_log = index entry["repeats"]++ append_to_log = FALSE break if(append_to_log) if(islist(result["return_values"])) add_lua_return_value_variants(result["return_values"], result["variants"]) result["return_values"] = weakrefify_list(result["return_values"]) log += list(result) index_of_log = log.len INVOKE_ASYNC(src, TYPE_PROC_REF(/datum/lua_state, update_editors)) return index_of_log /datum/lua_state/proc/parse_error(message, name) if(copytext(message, 1, 7) == "PANIC:") return list("status" = "panic", "message" = copytext(message, 7), "name" = name) else return list("status" = "error", "message" = message, "name" = name) /datum/lua_state/proc/load_script(script) var/tmp_usr = GLOB.lua_usr GLOB.lua_usr = usr DREAMLUAU_SET_USR GLOB.lua_state_stack += WEAKREF(src) var/result = DREAMLUAU_LOAD(internal_id, script, "input") SSlua.needs_gc_cycle |= src pop(GLOB.lua_state_stack) GLOB.lua_usr = tmp_usr // Internal errors unrelated to the code being executed are returned as text rather than lists if(isnull(result)) result = list("status" = "error", "message" = "load returned null (it may have runtimed - check the runtime logs)", "name" = "input") if(istext(result)) result = parse_error(result, "input") result["chunk"] = script check_if_slept(result) log_lua("[key_name(usr)] executed the following lua code:\n<code>[script]</code>") return result /datum/lua_state/process(seconds_per_tick) if(timer_enabled) var/result = call_function("__Timer_timer_process", seconds_per_tick) log_result(result, verbose = FALSE) for(var/function as anything in functions_to_execute) result = call_function(list("__Timer_callbacks", function)) log_result(result, verbose = FALSE) functions_to_execute.Cut() /datum/lua_state/proc/call_function(function, ...) var/call_args = length(args) > 1 ? args.Copy(2) : list() if(islist(function)) var/list/new_function_path = list() for(var/path_element in function) if(isweakref(path_element)) var/datum/weakref/weak_ref = path_element var/resolved = weak_ref.hard_resolve() if(!resolved) return list("status" = "error", "message" = "Weakref in function path ([weak_ref] [text_ref(weak_ref)]) resolved to null.", "name" = jointext(function, ".")) new_function_path += resolved else new_function_path += path_element function = new_function_path else function = list(function) var/tmp_usr = GLOB.lua_usr GLOB.lua_usr = usr DREAMLUAU_SET_USR GLOB.lua_state_stack += WEAKREF(src) var/result = DREAMLUAU_CALL_FUNCTION(internal_id, function, call_args) SSlua.needs_gc_cycle |= src pop(GLOB.lua_state_stack) GLOB.lua_usr = tmp_usr if(isnull(result)) result = list("status" = "error", "message" = "call_function returned null (it may have runtimed - check the runtime logs)", "name" = jointext(function, ".")) if(istext(result)) result = parse_error(result, jointext(function, ".")) check_if_slept(result) return result /datum/lua_state/proc/call_function_return_first(function, ...) SHOULD_NOT_SLEEP(TRUE) // This function is meant to be used for signal handlers. var/list/result = call_function(arglist(args)) INVOKE_ASYNC(src, PROC_REF(log_result), deep_copy_list(result), /*verbose = */FALSE) if(length(result)) if(islist(result["return_values"]) && length(result["return_values"])) var/return_value = result["return_values"][1] var/variant = (islist(result["variants"]) && length(result["variants"])) && result["variants"][1] if(islist(return_value) && islist(variant)) remove_non_dm_variants(return_value, variant) return return_value /datum/lua_state/proc/awaken() DREAMLUAU_SET_USR GLOB.lua_state_stack += WEAKREF(src) var/result = DREAMLUAU_AWAKEN(internal_id) SSlua.needs_gc_cycle |= src pop(GLOB.lua_state_stack) if(isnull(result)) result = list("status" = "error", "message" = "awaken returned null (it may have runtimed - check the runtime logs)", "name" = "An attempted awaken") if(istext(result)) result = parse_error(result, "An attempted awaken") check_if_slept(result) return result /// Prefer calling SSlua.queue_resume over directly calling this /datum/lua_state/proc/resume(index, ...) var/call_args = length(args) > 1 ? args.Copy(2) : list() DREAMLUAU_SET_USR GLOB.lua_state_stack += WEAKREF(src) var/result = DREAMLUAU_RESUME(internal_id, index, call_args) SSlua.needs_gc_cycle |= src pop(GLOB.lua_state_stack) if(isnull(result)) result = list("status" = "error", "param" = "resume returned null (it may have runtimed - check the runtime logs)", "name" = "An attempted resume") if(istext(result)) result = parse_error(result, "An attempted resumt") check_if_slept(result) return result /datum/lua_state/proc/get_globals() var/result = DREAMLUAU_GET_GLOBALS(internal_id) if(isnull(result)) CRASH("get_globals returned null") if(istext(result)) CRASH(result) var/list/new_globals = result var/list/values = new_globals["values"] var/list/variants = new_globals["variants"] add_lua_editor_variants(values, variants) globals = list("values" = weakrefify_list(values), "variants" = variants) /datum/lua_state/proc/get_tasks() var/result = DREAMLUAU_LIST_THREADS(internal_id) if(isnull(result)) CRASH("list_threads returned null") if(istext(result)) CRASH(result) return result /datum/lua_state/proc/kill_task(is_sleep, index) var/result = is_sleep ? DREAMLUAU_KILL_SLEEPING_THREAD(internal_id, index) : DREAMLUAU_KILL_YIELDED_THREAD(internal_id, index) SSlua.needs_gc_cycle |= src return result /datum/lua_state/proc/collect_garbage() var/result = DREAMLUAU_COLLECT_GARBAGE(internal_id) if(!isnull(result)) CRASH(result) /datum/lua_state/proc/update_editors() var/list/editor_list = LAZYACCESS(SSlua.editors, text_ref(src)) if(editor_list) for(var/datum/lua_editor/editor as anything in editor_list) SStgui.update_uis(editor) #undef MAX_LOG_REPEAT_LOOKBACK #endif