add const_format with rust_1_64, and vendored feature for mlua

Cargo.lock

@@ -16,9 +16,9 @@ checksum = "b048fb63fd8b5923fc5aa7b340d8e156aec7ec02f0c78fa8a6ddc2613f6f71de"
 
 [[package]]
 name = "bstr"
-version = "1.11.1"
+version = "1.11.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "786a307d683a5bf92e6fd5fd69a7eb613751668d1d8d67d802846dfe367c62c8"
+checksum = "531a9155a481e2ee699d4f98f43c0ca4ff8ee1bfd55c31e9e98fb29d2b176fe0"
 dependencies = [
  "memchr",
  "serde",
@@ -32,9 +32,9 @@ checksum = "79296716171880943b8470b5f8d03aa55eb2e645a4874bdbb28adb49162e012c"
 
 [[package]]
 name = "cc"
-version = "1.2.6"
+version = "1.2.7"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "8d6dbb628b8f8555f86d0323c2eb39e3ec81901f4b83e091db8a6a76d316a333"
+checksum = "a012a0df96dd6d06ba9a1b29d6402d1a5d77c6befd2566afdc26e10603dc93d7"
 dependencies = [
  "shlex",
 ]
@@ -73,6 +73,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "126f97965c8ad46d6d9163268ff28432e8f6a1196a55578867832e3049df63dd"
 dependencies = [
  "const_format_proc_macros",
+ "konst",
 ]
 
 [[package]]
@@ -146,6 +147,21 @@ dependencies = [
  "wasm-bindgen",
 ]
 
+[[package]]
+name = "konst"
+version = "0.2.19"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "330f0e13e6483b8c34885f7e6c9f19b1a7bd449c673fbb948a51c99d66ef74f4"
+dependencies = [
+ "konst_macro_rules",
+]
+
+[[package]]
+name = "konst_macro_rules"
+version = "0.2.19"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a4933f3f57a8e9d9da04db23fb153356ecaf00cbd14aee46279c33dc80925c37"
+
 [[package]]
 name = "libc"
 version = "0.2.169"
@@ -198,6 +214,25 @@ version = "0.4.22"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "a7a70ba024b9dc04c27ea2f0c0548feb474ec5c54bba33a7f72f873a39d07b24"
 
+[[package]]
+name = "lua-src"
+version = "547.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1edaf29e3517b49b8b746701e5648ccb5785cde1c119062cbabbc5d5cd115e42"
+dependencies = [
+ "cc",
+]
+
+[[package]]
+name = "luajit-src"
+version = "210.5.11+97813fb"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3015551c284515db7c30c559fc1080f9cb9ee990d1f6fca315451a107c7540bb"
+dependencies = [
+ "cc",
+ "which",
+]
+
 [[package]]
 name = "luau0-src"
 version = "0.11.2+luau653"
@@ -254,6 +289,8 @@ checksum = "63a11d485edf0f3f04a508615d36c7d50d299cf61a7ee6d3e2530651e0a31771"
 dependencies = [
  "cc",
  "cfg-if",
+ "lua-src",
+ "luajit-src",
  "luau0-src",
  "pkg-config",
 ]
@@ -428,9 +465,9 @@ checksum = "3c5e1a9a646d36c3599cd173a41282daf47c44583ad367b8e6837255952e5c67"
 
 [[package]]
 name = "syn"
-version = "2.0.94"
+version = "2.0.95"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "987bc0be1cdea8b10216bd06e2ca407d40b9543468fafd3ddfb02f36e77f71f3"
+checksum = "46f71c0377baf4ef1cc3e3402ded576dccc315800fbc62dfc7fe04b009773b4a"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -555,6 +592,18 @@ dependencies = [
  "wasm-bindgen",
 ]
 
+[[package]]
+name = "which"
+version = "6.0.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b4ee928febd44d98f2f459a4a79bd4d928591333a494a10a868418ac1b39cf1f"
+dependencies = [
+ "either",
+ "home",
+ "rustix",
+ "winsafe",
+]
+
 [[package]]
 name = "whoami"
 version = "1.5.2"
@@ -669,3 +718,9 @@ name = "windows_x86_64_msvc"
 version = "0.52.6"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec"
+
+[[package]]
+name = "winsafe"
+version = "0.0.19"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d135d17ab770252ad95e9a872d365cf3090e3be864a34ab46f48555993efc904"

Cargo.toml

@@ -5,10 +5,10 @@ edition = "2021"
 
 [dependencies]
 color-print = "0.3.7"
-const_format = "0.2.33"
+const_format = { version = "0.2.34", features = ["rust_1_64"] }
 crossterm = "0.28.1"
 home = "0.5.9"
-mlua = { version = "0.10.0", features = ["luau-jit"] }
+mlua = { version = "0.10.0", features = ["luau-jit", "vendored"] }
 thiserror = "2.0.9"
 uzers = "0.12.1"
 whoami = "1.5.2"

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 rhpidfyre
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

src/lib.rs

@@ -1,13 +1,8 @@
 pub const VERSION: &str = env!("CARGO_PKG_VERSION");
 
-pub mod shell;
-mod commands;
-mod ps;
-mod rc;
+pub mod session;
+pub mod commands;
+pub mod ps;
+pub mod rc;
 
-#[path = "./luau/vm.rs"]
-mod vm;
-#[path = "./luau/alias.rs"]
-mod alias;
-#[path = "./luau/terminal.rs"]
-mod terminal;
+pub mod vm;

src/luau/vm.rs

@@ -1,84 +0,0 @@
-use mlua::{
-	Lua as Luau,
-	Result as lResult,
-	MultiValue,
-};
-use crate::VERSION;
-use crate::terminal::TerminalColors;
-use core::fmt;
-use color_print::cprintln;
-
-fn display_none<T, E>(err: E) -> Option<T>
-where
-	E: fmt::Display
-{
-	println!("{err}");
-	None
-}
-
-fn luau_error<T>(err: mlua::Error) -> Option<T> {
-	cprintln!("<bold>====</>\n<r><bold>[!]</> {err}</>\n<bold>====</>");
-	None
-}
-
-fn luau_out(luau_args: MultiValue) -> String {
-	let mut print = String::new();
-	luau_args.iter()
-		.map(|arg| arg.to_string().unwrap_or("<SHELL CONVERSION ERROR>".to_owned()))
-		.for_each(|arg| {
-			if !print.is_empty() {
-				print.push('\u{0009}');
-			};
-			print.push_str(&arg);
-		}
-	);
-	print
-}
-
-trait Globals {
-	fn print(&self) -> lResult<()>;
-	fn printraw(&self) -> lResult<()>;
-	fn version(&self) -> lResult<()>;
-}
-impl Globals for Vm {
-	fn print(&self) -> lResult<()> {
-		self.0.globals().set("print", self.0.create_function(|_this, args: MultiValue| -> lResult<()> {
-			println!("{}", luau_out(args));
-			Ok(())
-		})?)
-	}
-	fn printraw(&self) -> lResult<()> {
-		self.0.globals().set("printraw", self.0.create_function(|_this, args: MultiValue| -> lResult<()> {
-			println!("{}", luau_out(args));
-			Ok(())
-		})?)
-	}
-	fn version(&self) -> lResult<()> {
-		let luau_info = self.0.globals().get::<String>("_VERSION")?;
-		self.0.globals().set("_VERSION", format!("{}, liblambdashell {}", luau_info, VERSION))
-	}
-}
-
-pub struct Vm(pub Luau);
-impl Vm {
-	pub fn new() -> Self {
-		Self(Luau::new())
-	}
-
-	fn set_shell_globals(&self) -> mlua::Result<()> {
-		self.print()?;
-		self.printraw()?;
-		self.version()?;
-		self.terminal()?;
-		self.0.globals().set("getfenv", mlua::Nil)?;
-		self.0.globals().set("setfenv", mlua::Nil)?;
-		self.0.sandbox(true)?;
-		Ok(())
-	}
-
-	pub fn exec(&self, source: String) {
-		self.set_shell_globals().map_or_else(display_none, |()| {
-			self.0.load(source).exec().map_or_else(luau_error, |()| Some(()))
-		});
-	}
-}

src/ps.rs

@@ -1,21 +1,26 @@
 use const_format::formatcp;
-use color_print::{cformat, cprint};
 
 pub const DEFAULT_PS: &str = formatcp!("lambdashell-{}", env!("CARGO_PKG_VERSION"));
 
-pub fn working_dir_name() -> String {
-	std::env::current_dir().map_or("?".to_owned(), |path| {
-		path.file_name().map_or("?".to_owned(), |name| {
-			let name_os_string = name.to_os_string();
-			match name_os_string == whoami::username_os() && name_os_string != "root" {
-				true => "~".to_owned(),
-				false => name.to_string_lossy().to_string(),
-			}
-		})
-	})
-}
+struct Ps(String);
+impl Ps {
+	fn set(prompt: String) -> Self {
+		Self(prompt)
+	}
 
-pub fn display(ps1: &String) {
-	let working_dir_name = cformat!(" <bold>{}</> ", working_dir_name());
-	cprint!("{}{}λ ", ps1, working_dir_name);
-}
+	fn working_dir_name(&self) -> String {
+		std::env::current_dir().map_or("?".to_owned(), |path| {
+			path.file_name().map_or("?".to_owned(), |name| {
+				let name_os_string = name.to_os_string();
+				match name_os_string == whoami::username_os() && name_os_string != "root" {
+					true => "~".to_owned(),
+					false => name.to_string_lossy().to_string(),
+				}
+			})
+		})
+	}
+
+	fn display(&self) {
+		print!("{}", self.0);
+	}
+}

src/session.rs

@@ -1,20 +1,12 @@
-use crate::{ps, commands, rc, vm};
+use crate::{commands, ps, rc, vm::{LuauVm, self}};
 use std::{fs, io::{self}};
 
 pub struct Config {
 	pub norc: bool
 }
 
-trait ShellLuauVm {
-	fn shell_vm_exec(&self, source: String);
-}
-impl ShellLuauVm for LambdaShell {
-	fn shell_vm_exec(&self, source: String) {
-		vm::Vm::new().exec(source);
-	}
-}
-
 pub struct LambdaShell {
+	vm: LuauVm,
 	ps1: String,
 	config: Config,
 	terminating: bool,
@@ -22,6 +14,7 @@ pub struct LambdaShell {
 impl LambdaShell {
 	pub fn create(config: Config) -> Self {
 		Self {
+			vm: vm::LuauVm::new(),
 			ps1: ps::DEFAULT_PS.to_owned(),
 			terminating: false,
 			config,
@@ -41,30 +34,27 @@ impl LambdaShell {
 		})
 	}
 
-	fn rc_parse(&self) {
-		if !self.config.norc {
-			if let Some(conf_file) = rc::config_file() {
-				match fs::read_to_string(conf_file) {
-			        Ok(luau_conf) => self.shell_vm_exec(luau_conf),
-			        Err(read_err) => println!("{read_err}"),
-			    }
-			}
-		}
+	pub fn vm_exec(&self, source: String) {
+		self.vm.exec(source);
 	}
 
 	pub fn start(&mut self) {
-		self.rc_parse();
-
-		ps::display(&self.ps1);
-
+		if !self.config.norc {
+			if let Some(conf_file) = rc::config_file() {
+				fs::read_to_string(conf_file).map_or_else(|e| println!("{e}"), |luau_conf| self.vm_exec(luau_conf));
+			}
+		}
 		loop {
 			match self.terminating {
 				true => break,
-				false => match self.wait() {
-					Ok(()) => ps::display(&self.ps1),
-					Err(flush_error) => {
-						println!("{flush_error}");
-						break;
+				false => {
+
+					match self.wait() {
+						Ok(()) => {},
+						Err(flush_error) => {
+							println!("{flush_error}");
+							break;
+						}
 					}
 				},
 			}

src/vm/mod.rs

@@ -0,0 +1,76 @@
+use mlua::{
+	Function, Lua as Luau, MultiValue, Result as lResult, Table, Value
+};
+use crate::vm::{shell::System, terminal::Terminal};
+use crate::VERSION;
+use color_print::{cformat, cprintln};
+use core::fmt;
+
+mod shell;
+mod terminal;
+mod alias;
+
+trait Helpers {
+	fn option_display_none<T, E: fmt::Display>(&self, err: E) -> Option<T>;
+	fn luau_error<T>(&self, err: mlua::Error) -> Option<T>;
+}
+impl Helpers for LuauVm {
+	fn option_display_none<T, E: fmt::Display>(&self, err: E) -> Option<T> {
+		println!("{err}");
+		None
+	}
+
+	fn luau_error<T>(&self, err: mlua::Error) -> Option<T> {
+		cprintln!("<bold>====</>\n<r><bold>[!]:</> {err}</>\n<bold>====</>");
+		None
+	}
+}
+
+trait Globals {
+	fn global_warn(&self, luau_globals: &Table) -> lResult<()>;
+	fn global_version(&self, luau_globals: &Table) -> lResult<()>;
+}
+impl Globals for LuauVm {
+	fn global_warn(&self, luau_globals: &Table) -> lResult<()> {
+		let luau_print = luau_globals.get::<Function>("print")?;
+		luau_globals.set("warn", self.0.create_function(move |this, args: MultiValue| -> lResult<()> {
+			let luau_multi_values = args.into_iter()
+				.map(|value| cformat!("<y>{}</>", value.to_string().unwrap_or("<SHELL CONVERSION ERROR>".to_owned())))
+				.map(|arg_v| Value::String(this.create_string(arg_v).unwrap()))
+				.collect::<MultiValue>();
+			luau_print.call::<()>(luau_multi_values).unwrap();
+			Ok(())
+		})?)
+	}
+
+	fn global_version(&self, luau_globals: &Table) -> lResult<()> {
+		let luau_info = luau_globals.get::<String>("_VERSION")?;
+		luau_globals.set("_VERSION", format!("{}, liblambdashell {}", luau_info, VERSION))
+	}
+}
+
+pub struct LuauVm(pub Luau);
+impl LuauVm {
+	pub(crate) fn new() -> Self {
+		Self(Luau::new())
+	}
+
+	fn set_shell_globals(&self) -> lResult<()> {
+		let luau_globals = self.0.globals();
+		self.global_warn(&luau_globals)?;
+		self.global_version(&luau_globals)?;
+		self.global_terminal(&luau_globals)?;
+		self.shell_globals(&luau_globals)?;
+		luau_globals.set("getfenv", mlua::Nil)?;
+		luau_globals.set("setfenv", mlua::Nil)?;
+		self.0.sandbox(true)?;
+		Ok(())
+	}
+
+	pub fn exec(&self, source: String) {
+		match self.set_shell_globals() {
+		    Ok(()) => self.0.load(source).exec().map_or_else(|exec_err| self.luau_error(exec_err), Some),
+		    Err(globals_err) => self.option_display_none(globals_err),
+		};
+	}
+}

src/vm/shell.rs

@@ -0,0 +1,52 @@
+use mlua::{Result as lResult, Table, Value};
+use whoami::fallible;
+use crate::vm::LuauVm;
+
+const DEFAULT_HOSTNAME: &str = "hostname";
+
+pub trait PsPrompt {
+	fn ps_prompt(&self) -> lResult<Table>;
+}
+impl PsPrompt for LuauVm {
+	fn ps_prompt(&self) -> lResult<Table> {
+		let prompt_table = self.0.create_table()?;
+		let prompt_metatable = self.0.create_table()?;
+		prompt_metatable.set("__index", self.0.create_function(|_, (table, index): (Table, Value)| -> lResult<String> {
+			table.raw_get::<String>(index)
+		})?)?;
+		prompt_metatable.set("__newindex", self.0.create_function(|_, _: String| -> lResult<String> {
+			Ok("placeholder".to_owned())
+		})?)?;
+		prompt_table.set("__metatable", mlua::Nil)?;
+		prompt_table.set_metatable(Some(prompt_metatable));
+		Ok(prompt_table)
+	}
+}
+
+pub trait System {
+	fn sys_details(&self, shell: &Table) -> lResult<()>;
+	fn shell_globals(&self, luau_globals: &Table) -> lResult<()>;
+}
+impl System for LuauVm {
+	fn sys_details(&self, shell: &Table) -> lResult<()> {
+		let system = self.0.create_table()?;
+		system.set("DESKTOP_ENV", whoami::desktop_env().to_string())?;
+		system.set("DEVICENAME", whoami::devicename().to_string())?;
+		system.set("USERNAME", whoami::username().to_string())?;
+		system.set("REALNAME", whoami::realname().to_string())?;
+		system.set("PLATFORM", whoami::platform().to_string())?;
+		system.set("DISTRO", whoami::distro().to_string())?;
+		system.set("HOSTNAME", fallible::hostname().unwrap_or(DEFAULT_HOSTNAME.to_owned()))?;
+		shell.set("SYSTEM", system)?;
+		Ok(())
+	}
+
+	fn shell_globals(&self, luau_globals: &Table) -> lResult<()> {
+		let shell = self.0.create_table()?;
+		let ps_prompt = self.ps_prompt()?;
+		self.sys_details(&shell)?;
+		shell.set("PROMPT", ps_prompt)?;
+		luau_globals.set("SHELL", shell)?;
+		Ok(())
+	}
+}

src/vm/terminal.rs

@@ -1,6 +1,7 @@
-use mlua::{Result as lResult, Table};
+use const_format::str_split;
+use mlua::{Function, Result as lResult, Table};
 use crossterm::style::Stylize;
-use crate::vm::Vm;
+use crate::vm::LuauVm;
 
 macro_rules! foreground_styles_luau {
 	($self:expr, $style_table:expr, $($color:ident)+) => {
@@ -11,27 +12,23 @@ macro_rules! foreground_styles_luau {
         )+
     };
 }
-
 macro_rules! background_styles_luau {
 	($self:expr, $style_table:expr, $($color:ident)+) => {
 		$(
-			match stringify!($color).split_once("_") {
-			    Some((_, color_name)) => $style_table.set(color_name.to_ascii_uppercase(), $self.0.create_function(|_, text: String| -> lResult<String> {
-					Ok(text.$color().to_string())
-        		})?)?,
-			    None => panic!("Luau set error: {:?}. There was nothing to split from delimiter: \"_\"", stringify!($color)),
-			}
+			$style_table.set(str_split!(stringify!($color), "_")[1..].join("_").to_ascii_uppercase(), $self.0.create_function(|_, text: String| -> lResult<String> {
+				Ok(text.$color().to_string())
+        	})?)?;
         )+
     };
 }
 
-pub trait TerminalColors {
+#[allow(dead_code)]
+trait Colors {
 	fn background(&self, style_table: &Table) -> lResult<()>;
 	fn foreground(&self, style_table: &Table) -> lResult<()>;
-	fn styling(&self) -> lResult<Table>;
-	fn terminal(&self) -> lResult<()>;
+	fn styling(&self, term_out_table: &Table) -> lResult<()>;
 }
-impl TerminalColors for Vm {
+impl Colors for LuauVm {
 	fn background(&self, style_table: &Table) -> lResult<()> {
 		let foreground_table = self.0.create_table()?;
 		foreground_styles_luau!(self, foreground_table,
@@ -46,8 +43,7 @@ impl TerminalColors for Vm {
 			underline_blue        underline_magenta      underline_cyan       underline_white
 			bold
 		);
-		style_table.set("FOREGROUND", foreground_table)?;
-        Ok(())
+		style_table.set("FOREGROUND", foreground_table)
 	}
 
 	fn foreground(&self, style_table: &Table) -> lResult<()> {
@@ -60,21 +56,38 @@ impl TerminalColors for Vm {
 		    on_blue  on_magenta
 		    on_cyan  on_white
 		);
-		style_table.set("BACKGROUND", background_table)?;
-        Ok(())
+		style_table.set("BACKGROUND", background_table)
 	}
 
-	fn styling(&self) -> lResult<Table> {
+	fn styling(&self, term_out_table: &Table) -> lResult<()> {
 		let style_table = self.0.create_table()?;
 		self.foreground(&style_table)?;
 		self.background(&style_table)?;
-		Ok(style_table)
+		term_out_table.set("STYLE", style_table)
 	}
+}
 
-	fn terminal(&self) -> lResult<()> {
+#[allow(dead_code)]
+trait Write {
+	fn write(&self, term_out_table: &Table) -> lResult<()>;
+}
+impl Write for LuauVm {
+	fn write(&self, term_out_table: &Table) -> lResult<()> {
+		term_out_table.set("WRITE", self.0.create_function(|_, s: String| -> lResult<()> {
+			print!("{s}");
+			Ok(())
+		})?)
+	}
+}
+
+pub trait Terminal {
+	fn global_terminal(&self, luau_globals: &Table) -> lResult<()>;
+}
+impl Terminal for LuauVm {
+	fn global_terminal(&self, luau_globals: &Table) -> lResult<()> {
 		let term_table = self.0.create_table()?;
-		term_table.set("OUT", self.styling()?)?;
-		self.0.globals().set("TERMINAL", &term_table)?;
-		Ok(())
+		let term_out_table = self.0.create_table()?;
+		term_table.set("OUT", term_out_table)?;
+		luau_globals.set("TERMINAL", &term_table)
 	}
 }