yazi_plugin/
lua.rs

1use anyhow::{Context, Result};
2use futures::executor::block_on;
3use mlua::Lua;
4use yazi_binding::{Runtime, runtime_mut};
5use yazi_boot::BOOT;
6use yazi_macro::plugin_preset as preset;
7use yazi_shared::RoCell;
8
9pub static LUA: RoCell<Lua> = RoCell::new();
10
11pub(super) fn init_lua() -> Result<()> {
12	LUA.init(Lua::new());
13
14	stage_1(&LUA).context("Lua setup failed")?;
15	stage_2(&LUA).context("Lua runtime failed")?;
16	Ok(())
17}
18
19fn stage_1(lua: &'static Lua) -> Result<()> {
20	lua.set_app_data(Runtime::new());
21
22	// Base
23	let globals = lua.globals();
24	globals.raw_set("ui", crate::elements::compose())?;
25	globals.raw_set("ya", crate::utils::compose(false))?;
26	globals.raw_set("fs", crate::fs::compose())?;
27	globals.raw_set("ps", crate::pubsub::compose())?;
28	globals.raw_set("rt", crate::runtime::compose())?;
29	globals.raw_set("th", crate::theme::compose())?;
30
31	yazi_binding::Error::install(lua)?;
32	yazi_binding::Cha::install(lua)?;
33	crate::loader::install(lua)?;
34	crate::process::install(lua)?;
35	yazi_binding::File::install(lua)?;
36	yazi_binding::Url::install(lua)?;
37	yazi_binding::Path::install(lua)?;
38
39	// Addons
40	lua.load(preset!("ya")).set_name("ya.lua").exec()?;
41
42	// Components
43	lua.load(preset!("components/current")).set_name("current.lua").exec()?;
44	lua.load(preset!("components/entity")).set_name("entity.lua").exec()?;
45	lua.load(preset!("components/header")).set_name("header.lua").exec()?;
46	lua.load(preset!("components/linemode")).set_name("linemode.lua").exec()?;
47
48	lua.load(preset!("components/marker")).set_name("marker.lua").exec()?;
49	lua.load(preset!("components/modal")).set_name("modal.lua").exec()?;
50	lua.load(preset!("components/parent")).set_name("parent.lua").exec()?;
51	lua.load(preset!("components/preview")).set_name("preview.lua").exec()?;
52	lua.load(preset!("components/progress")).set_name("progress.lua").exec()?;
53	lua.load(preset!("components/rail")).set_name("rail.lua").exec()?;
54	lua.load(preset!("components/root")).set_name("root.lua").exec()?;
55	lua.load(preset!("components/status")).set_name("status.lua").exec()?;
56	lua.load(preset!("components/tab")).set_name("tab.lua").exec()?;
57	lua.load(preset!("components/tabs")).set_name("tabs.lua").exec()?;
58	lua.load(preset!("components/tasks")).set_name("tasks.lua").exec()?;
59
60	Ok(())
61}
62
63fn stage_2(lua: &'static Lua) -> mlua::Result<()> {
64	lua.load(preset!("setup")).set_name("setup.lua").exec()?;
65	lua.load(preset!("compat")).set_name("compat.lua").exec()?;
66
67	if let Ok(b) = std::fs::read(BOOT.config_dir.join("init.lua")) {
68		block_on(lua.load(b).set_name("init.lua").exec_async())?;
69	}
70
71	runtime_mut!(lua)?.blocking = false;
72	Ok(())
73}
yazi_plugin/lua.rs

yazi_plugin/
lua.rs
