SHELL.SYSTEM

Cargo.lock

@@ -194,9 +194,9 @@ dependencies = [
 
 [[package]]
 name = "linux-raw-sys"
-version = "0.4.14"
+version = "0.4.15"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "78b3ae25bc7c8c38cec158d1f2757ee79e9b3740fbc7ccf0e59e4b08d793fa89"
+checksum = "d26c52dbd32dccf2d10cac7725f8eae5296885fb5703b261f7d0a0739ec807ab"
 
 [[package]]
 name = "lock_api"
@@ -384,9 +384,9 @@ checksum = "c7fb8039b3032c191086b10f11f319a6e99e1e82889c5cc6046f515c9db1d497"
 
 [[package]]
 name = "rustix"
-version = "0.38.42"
+version = "0.38.43"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f93dc38ecbab2eb790ff964bb77fa94faf256fd3e73285fd7ba0903b76bedb85"
+checksum = "a78891ee6bf2340288408954ac787aa063d8e8817e9f53abb37c695c6d834ef6"
 dependencies = [
  "bitflags",
  "errno",
@@ -476,18 +476,18 @@ dependencies = [
 
 [[package]]
 name = "thiserror"
-version = "2.0.9"
+version = "2.0.10"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f072643fd0190df67a8bab670c20ef5d8737177d6ac6b2e9a236cb096206b2cc"
+checksum = "a3ac7f54ca534db81081ef1c1e7f6ea8a3ef428d2fc069097c079443d24124d3"
 dependencies = [
  "thiserror-impl",
 ]
 
 [[package]]
 name = "thiserror-impl"
-version = "2.0.9"
+version = "2.0.10"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "7b50fa271071aae2e6ee85f842e2e28ba8cd2c5fb67f11fcb1fd70b276f9e7d4"
+checksum = "9e9465d30713b56a37ede7185763c3492a91be2f5fa68d958c44e41ab9248beb"
 dependencies = [
  "proc-macro2",
  "quote",

Cargo.toml

@@ -11,10 +11,4 @@ home = "0.5.9"
 mlua = { version = "0.10.0", features = ["luau-jit", "vendored"] }
 thiserror = "2.0.9"
 uzers = "0.12.1"
-whoami = "1.5.2"
-
-[profile.release]
-strip = true
-opt-level = "z"
-lto = true
-codegen-units = 1
+whoami = "1.5.2"

src/ps.rs

@@ -1,27 +1,28 @@
 pub const DEFAULT_PS: &str = concat!("lambdashell-", env!("CARGO_PKG_VERSION"), " ");
 
-#[derive(Debug, Clone)]
+#[derive(Debug)]
 pub struct Ps(String);
 impl Ps {
-	pub fn set(prompt: String) -> Self {
+	pub const fn set(prompt: String) -> Self {
 		Self(prompt)
 	}
-	pub fn get(self) -> String {
-		self.0
+	//rustc: `std::string::String::as_str` is not yet stable as a const fn
+	pub fn get(&self) -> &str {
+		self.0.as_str()
+	}
+	pub fn modify(&mut self, prompt: String) {
+		self.0 = prompt
 	}
 	pub fn display(&self) {
 		print!("{}", self.0);
 	}
 
 	pub 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(),
-				}
-			})
-		})
+		std::env::current_dir().map_or("?".to_owned(), |path| path.file_name().map_or("?".to_owned(), |name| {
+			match name.to_os_string() == whoami::username_os() {
+				true => "~".to_owned(),
+				false => name.to_string_lossy().to_string(),
+			}
+		}))
 	}
 }

src/session.rs

@@ -1,4 +1,4 @@
-use std::{fs, io::{self}};
+use std::{cell::RefCell, fs, io::{self}, rc::Rc};
 use core::fmt;
 
 use crate::{
@@ -14,16 +14,16 @@ pub struct LambdaShell {
 	terminate: bool,
 	config: Config,
 	vm: LuauVm,
-	ps: Ps,
+	ps: Rc<RefCell<Ps>>,
 }
 impl LambdaShell {
 	pub fn create(config: Config) -> Self {
-		let ps = Ps::set(ps::DEFAULT_PS.to_owned());
+		let ps = Rc::new(RefCell::new(Ps::set(ps::DEFAULT_PS.to_owned())));
 		Self {
-			vm: vm::LuauVm::new(ps.to_owned()),
+			ps: Rc::clone(&ps),
+			vm: vm::LuauVm::new(ps),
 			terminate: false,
 			config,
-			ps
 		}
 	}
 
@@ -48,12 +48,12 @@ impl LambdaShell {
 				fs::read_to_string(conf_file).map_or_display(|luau_conf| self.vm_exec(luau_conf));
 			}
 		}
-		self.ps.display();
 
+		self.ps.borrow().display();
 		loop {
 			if self.terminate { break } else {
 				match self.wait() {
-			        Ok(()) => self.ps.display(),
+			        Ok(()) => self.ps.borrow().display(),
 			        Err(flush_err) => self.error(flush_err),
 			    }
 			}

src/vm/mod.rs

@@ -1,5 +1,6 @@
 use mlua::{Function, Lua as Luau, MultiValue, Result as lResult, Table, Value};
 use color_print::{cformat, ceprintln};
+use std::{cell::RefCell, rc::Rc};
 use core::fmt;
 use shell::Shell;
 
@@ -34,7 +35,7 @@ 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.vm.create_function(move |this, args: MultiValue| -> lResult<()> {
+		luau_globals.raw_set("warn", self.vm.create_function(move |this, args: MultiValue| -> lResult<()> {
 			let luau_multi_values = args.into_iter()
 				.map(|value| cformat!("<bold,y>{}</>", value.to_string().unwrap_or(Self::CONV_ERROR.to_owned())))
 				.map(|arg_v| Value::String(this.create_string(arg_v).unwrap()))
@@ -46,27 +47,27 @@ impl Globals for LuauVm {
 
 	fn global_version(&self, luau_globals: &Table) -> lResult<()> {
 		let luau_info = luau_globals.get::<String>("_VERSION")?;
-		luau_globals.set("_VERSION", format!("{luau_info}, liblambdashell {}", Self::LIB_VERSION))
+		luau_globals.raw_set("_VERSION", format!("{luau_info}, liblambdashell {}", Self::LIB_VERSION))
 	}
 }
 
 pub struct LuauVm {
 	vm: Luau,
-	ps: Ps,
+	ps: Rc<RefCell<Ps>>
 }
 impl LuauVm {
-	pub(crate) fn new(ps: Ps) -> Self {
+	pub(crate) fn new(ps: Rc<RefCell<Ps>>) -> Self {
 		Self { vm: Luau::new(), ps }
 	}
 
 	fn set_shell_globals(&self) -> lResult<()> {
 		let luau_globals = self.vm.globals();
+		self.global_shell(&luau_globals)?;
+		self.global_terminal(&luau_globals)?;
 		self.global_warn(&luau_globals)?;
 		self.global_version(&luau_globals)?;
-		self.global_terminal(&luau_globals)?;
-		self.global_shell(&luau_globals)?;
-		luau_globals.set("getfenv", mlua::Nil)?;
-		luau_globals.set("setfenv", mlua::Nil)?;
+		luau_globals.raw_set("getfenv", mlua::Nil)?;
+		luau_globals.raw_set("setfenv", mlua::Nil)?;
 		self.vm.sandbox(true)?;
 		Ok(())
 	}

src/vm/shell.rs

@@ -1,46 +1,38 @@
-use mlua::{Result as lResult, Table};
+use mlua::{Lua as Luau, MetaMethod, Result as lResult, Table, UserData, UserDataFields, UserDataMethods};
+use std::{cell::RefCell, rc::Rc};
 use whoami::fallible;
 
-use crate::vm::LuauVm;
+use crate::{ps::Ps, vm::LuauVm};
 
-trait PsPrompt {
-	fn ps_prompt(&self) -> lResult<Table>;
+const DEFAULT_HOSTNAME: &str = "hostname";
+
+fn luau_sys_details(luau: &Luau) -> lResult<Table> {
+	let system = luau.create_table()?;
+	system.raw_set("DESKTOP_ENV", whoami::desktop_env().to_string())?;
+	system.raw_set("DEVICENAME", whoami::devicename().to_string())?;
+	system.raw_set("USERNAME", whoami::username().to_string())?;
+	system.raw_set("REALNAME", whoami::realname().to_string())?;
+	system.raw_set("PLATFORM", whoami::platform().to_string())?;
+	system.raw_set("DISTRO", whoami::distro().to_string())?;
+	system.raw_set("ARCH", whoami::arch().to_string())?;
+	system.raw_set("HOSTNAME", fallible::hostname().unwrap_or(DEFAULT_HOSTNAME.to_owned()))?;
+	Ok(system)
 }
-impl PsPrompt for LuauVm {
-	fn ps_prompt(&self) -> lResult<Table> {
-		let prompt_table = self.vm.create_table()?;
-		let prompt_metatable = self.vm.create_table()?;
-		let ps_owned = self.ps.to_owned();
-		prompt_metatable.set("__index", self.vm.create_function(move |_, (s, s1): (Table, String)| -> lResult<String> {
-			Ok(ps_owned.clone().get())
-		})?)?;
-		prompt_metatable.set("__newindex", self.vm.create_function(|_, _: String| -> lResult<String> {
-			Ok("placeholder".to_owned())
-		})?)?;
-		// prompt_table.set("__metatable", mlua::Nil)?;
-		prompt_table.set_metatable(Some(prompt_metatable));
-		prompt_table.set_readonly(false);
-		Ok(prompt_table)
+
+struct ShellUserdata(Rc<RefCell<Ps>>);
+impl UserData for ShellUserdata {
+	fn add_fields<F: UserDataFields<Self>>(fields: &mut F) {
+		fields.add_field_method_get("PROMPT", |_, this| Ok(this.0.borrow().get().to_owned()));
+		fields.add_field_method_get("SYSTEM", |luau, _| luau_sys_details(luau));
 	}
-}
 
-trait System {
-	const DEFAULT_HOSTNAME: &str;
-	fn sys_details(&self) -> lResult<Table>;
-}
-impl System for LuauVm {
-	const DEFAULT_HOSTNAME: &str = "hostname";
-
-	fn sys_details(&self) -> lResult<Table> {
-		let system = self.vm.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(Self::DEFAULT_HOSTNAME.to_owned()))?;
-		Ok(system)
+	fn add_methods<M: UserDataMethods<Self>>(methods: &mut M) {
+		methods.add_meta_method_mut(MetaMethod::NewIndex, |_, this, (tindex, tvalue): (String, String)| -> lResult<()> {
+			if tindex == "PROMPT" {
+				this.0.borrow_mut().modify(tvalue);
+			}
+			Ok(())
+		});
 	}
 }
 
@@ -49,10 +41,10 @@ pub trait Shell {
 }
 impl Shell for LuauVm {
 	fn global_shell(&self, luau_globals: &Table) -> lResult<()> {
-		let shell = self.vm.create_table()?;
-		shell.set("SYSTEM", self.sys_details()?)?;
-		shell.set("PROMPT", self.ps_prompt()?)?;
-		luau_globals.set("SHELL", shell)?;
+		// let shell = self.vm.create_table()?;
+		// shell.set("SYSTEM", self.sys_details()?)?;
+		// shell.set("PROMPT", self.ps_prompt()?)?;
+		luau_globals.set("SHELL", ShellUserdata(Rc::clone(&self.ps)))?;
 		Ok(())
 	}
 }

src/vm/terminal.rs

@@ -7,7 +7,7 @@ use crate::vm::LuauVm;
 macro_rules! foreground_styles_luau {
 	($self:expr, $style_table:expr, $($color:ident)+) => {
 		$(
-			$style_table.set(stringify!($color).to_ascii_uppercase(), $self.vm.create_function(|_, text: String| -> lResult<String> {
+			$style_table.raw_set(stringify!($color).to_ascii_uppercase(), $self.vm.create_function(|_, text: String| -> lResult<String> {
             	Ok(text.$color().to_string())
         	})?)?;
         )+
@@ -16,7 +16,7 @@ macro_rules! foreground_styles_luau {
 macro_rules! background_styles_luau {
 	($self:expr, $style_table:expr, $($color:ident)+) => {
 		$(
-			$style_table.set(
+			$style_table.raw_set(
 				str_split!(stringify!($color), "_")[1..].join("_").to_ascii_uppercase(),
 			$self.vm.create_function(|_, text: String| -> lResult<String> {
 				Ok(text.$color().to_string())
@@ -44,7 +44,7 @@ impl Colors for LuauVm {
 			underline_blue        underline_magenta      underline_cyan       underline_white
 			bold
 		);
-		term_out_table.set("FOREGROUND", foreground_table)
+		term_out_table.raw_set("FOREGROUND", foreground_table)
 	}
 
 	fn foreground(&self, term_out_table: &Table) -> lResult<()> {
@@ -57,7 +57,7 @@ impl Colors for LuauVm {
 		    on_blue  on_magenta
 		    on_cyan  on_white
 		);
-		term_out_table.set("BACKGROUND", background_table)
+		term_out_table.raw_set("BACKGROUND", background_table)
 	}
 }
 
@@ -73,14 +73,12 @@ impl Write for LuauVm {
 			Ok(())
 		})
 	}
-
 	fn write_error(&self) -> lResult<Function> {
 		self.vm.create_function(|_, s: String| -> lResult<()> {
 			eprint!("{s}");
 			Ok(())
 		})
 	}
-
 	fn write_error_ln(&self) -> lResult<Function> {
 		self.vm.create_function(|_, s: String| -> lResult<()> {
 			eprintln!("{s}");
@@ -103,10 +101,10 @@ impl Terminal for LuauVm {
 
 	fn global_terminal(&self, luau_globals: &Table) -> lResult<()> {
 		let term_table = self.vm.create_table()?;
-		term_table.set("OUT", self.out()?)?;
-		term_table.set("WRITE", self.write()?)?;
-		term_table.set("WRITE_ERROR", self.write_error()?)?;
-		term_table.set("WRITE_ERROR_LN", self.write_error_ln()?)?;
-		luau_globals.set("TERMINAL", term_table)
+		term_table.raw_set("OUT", self.out()?)?;
+		term_table.raw_set("WRITE", self.write()?)?;
+		term_table.raw_set("WRITE_ERROR", self.write_error()?)?;
+		term_table.raw_set("WRITE_ERROR_LN", self.write_error_ln()?)?;
+		luau_globals.raw_set("TERMINAL", term_table)
 	}
 }