mirror of
				https://github.com/rm-dr/daisy
				synced 2025-10-30 05:54:24 -07:00 
			
		
		
		
	Compare commits
	
		
			5 Commits
		
	
	
		
			v1.1.0
			...
			599c9742d2
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 599c9742d2 | |||
| 9f9cc5d084 | |||
| c8ebec59ae | |||
| fde1220a96 | |||
| 2f1f8a0801 | 
							
								
								
									
										12
									
								
								Cargo.toml
									
									
									
									
									
								
							
							
						
						
									
										12
									
								
								Cargo.toml
									
									
									
									
									
								
							| @ -13,6 +13,10 @@ readme = "README.md" | ||||
| name = "daisy" | ||||
| path = "src/main.rs" | ||||
|  | ||||
| [lib] | ||||
| name = "daisycalc" | ||||
| path = "src/lib.rs" | ||||
|  | ||||
| [profile.release] | ||||
| opt-level = 3 | ||||
| debug = 0 | ||||
| @ -25,11 +29,13 @@ panic = "abort" | ||||
|  | ||||
| [dependencies] | ||||
| cfg-if = "1.0.0" | ||||
|  | ||||
| [target.'cfg(target_family = "unix")'.dependencies] | ||||
| termion = "2.0.1" | ||||
| num = "0.4.1" | ||||
| #astro-float = "0.7.1" | ||||
|  | ||||
|  | ||||
| [target.'cfg(target_family = "unix")'.dependencies] | ||||
| termion = "2.0.1" | ||||
|  | ||||
|  | ||||
| [build-dependencies] | ||||
| toml = "0.7.4" | ||||
| @ -16,6 +16,7 @@ pub fn is_command( | ||||
| 		| "vars" | ||||
| 		| "consts" | "constants" | ||||
| 		| "del" | "delete" | ||||
| 		| "flags" | ||||
| 		=> true, | ||||
| 		_ => false | ||||
| 	} | ||||
| @ -81,6 +82,27 @@ pub fn do_command( | ||||
| 			return t; | ||||
| 		}, | ||||
|  | ||||
| 		"flags" => { | ||||
| 			return FormattedText::new( | ||||
| 				concat!( | ||||
| 					"\n", | ||||
| 					"A list of command-line arguments is below\n", | ||||
| 					"\n", | ||||
| 					"╞════ [t]Flag[n] ════╪════════════════ [t]Function[n] ════════════════╡\n", | ||||
| 					"  [c]--help[n]        Show help\n", | ||||
| 					"  [c]--version[n]     Show version\n", | ||||
| 					"  [c]--info[n]        Show system information\n", | ||||
| 					"  [c]--256color[n]    Use full color support (default)\n", | ||||
| 					"  [c]--8color[n]      Use reduced colors (ANSI, no styling)\n", | ||||
| 					"  [c]--nocolor[n]     Do not use colors\n", | ||||
| 					"  [c]--nosub[n]       Disable inline substitution\n", | ||||
| 					"  [c]--nosuper[n]     Disable superscript powers\n", | ||||
| 					"  [c]--nooneover[n]   Disable \"one-over\" fractions as -1 power\n", | ||||
| 					"\n\n" | ||||
| 				).to_string() | ||||
| 			); | ||||
| 		}, | ||||
|  | ||||
| 		"clear" => { | ||||
| 			return FormattedText::new("[clear]".to_string()); | ||||
| 		}, | ||||
|  | ||||
| @ -1,13 +0,0 @@ | ||||
|  | ||||
|  | ||||
| // Select main script for target system | ||||
| cfg_if::cfg_if! { | ||||
| 	if #[cfg(target_family = "unix")] { | ||||
| 		mod unix; | ||||
| 		pub use unix::main as main_e; | ||||
| 	} else { | ||||
| 		pub fn main_e() -> Result<(), std::io::Error> { | ||||
| 			unimplemented!("Not yet implemented."); | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @ -1,3 +0,0 @@ | ||||
| mod unix; | ||||
| mod promptbuffer; | ||||
| pub use self::unix::main; | ||||
| @ -1,115 +0,0 @@ | ||||
| use std::io::stdout; | ||||
| use std::io::stdin; | ||||
| use std::env; | ||||
|  | ||||
| use termion::{ | ||||
| 	event::Key, | ||||
| 	input::TermRead, | ||||
| 	raw::IntoRawMode, | ||||
| 	color::DetectColors | ||||
| }; | ||||
|  | ||||
| use super::promptbuffer::PromptBuffer; | ||||
| use crate::command; | ||||
| use crate::context::Context; | ||||
| use crate::FormattedText; | ||||
|  | ||||
| #[inline(always)] | ||||
| pub fn main() -> Result<(), std::io::Error> { | ||||
| 	let mut stdout = stdout().into_raw_mode().unwrap(); | ||||
| 	let mut pb: PromptBuffer = PromptBuffer::new(64); | ||||
| 	let mut context = Context::new(); | ||||
|  | ||||
| 	// Set color compatibilty | ||||
| 	let term_colors = stdout.available_colors().unwrap_or(0); | ||||
| 	if term_colors >= 256 { | ||||
| 		context.config.term_color_type = 2; | ||||
| 	} else if term_colors >= 8 { | ||||
| 		context.config.term_color_type = 1; | ||||
| 	} else { | ||||
| 		context.config.term_color_type = 0; | ||||
| 	} | ||||
|  | ||||
| 	context.config.check(); | ||||
|  | ||||
|  | ||||
|  | ||||
| 	// Handle command-line arguments | ||||
| 	let args: Vec<String> = env::args().collect(); | ||||
| 	if args.iter().any(|s| s == "--help") { | ||||
| 		let t = command::do_command(&mut context, &String::from("help")); | ||||
| 		t.write(&context, &mut stdout)?; | ||||
| 		return Ok(()); | ||||
| 	} else if args.iter().any(|s| s == "--version") { | ||||
| 		let t = FormattedText::new(format!( | ||||
| 			"Daisy v{}\n", env!("CARGO_PKG_VERSION") | ||||
| 		)); | ||||
| 		t.write(&context, &mut stdout)?; | ||||
| 		return Ok(()); | ||||
| 	} else if args.iter().any(|s| s == "--debug") { | ||||
| 		let t = FormattedText::new(format!( | ||||
| 			concat!( | ||||
| 				"Daisy v{}\n", | ||||
| 				"Your terminal supports {} colors.\n" | ||||
| 			), | ||||
| 			env!("CARGO_PKG_VERSION"), | ||||
| 			term_colors | ||||
| 		)); | ||||
| 		t.write(&context, &mut stdout)?; | ||||
| 		return Ok(()); | ||||
| 	} | ||||
|  | ||||
| 	'outer: loop { | ||||
|  | ||||
| 		pb.write_prompt(&mut context, &mut stdout)?; | ||||
|  | ||||
| 		let stdin = stdin(); | ||||
| 		for c in stdin.keys() { | ||||
| 			if let Key::Char(q) = c.as_ref().unwrap() { | ||||
| 				match q { | ||||
| 					'\n' => { | ||||
| 						// Print again without cursor, in case we pressed enter | ||||
| 						// while inside a substitution | ||||
| 						pb.write_prompt_nocursor(&mut context, &mut stdout)?; | ||||
| 						let in_str = pb.enter(); | ||||
| 						FormattedText::newline(&mut stdout)?; | ||||
| 						if in_str == "" { break; } | ||||
|  | ||||
| 						if in_str.trim() == "quit" { | ||||
| 							break 'outer; | ||||
| 						} else { | ||||
| 							let r = crate::do_string(&mut context, &in_str); | ||||
|  | ||||
| 							match r { | ||||
| 								Ok(t) | Err(t) => { | ||||
| 									t.write(&context, &mut stdout).unwrap(); | ||||
| 								} | ||||
| 							} | ||||
| 						} | ||||
|  | ||||
| 						break; | ||||
| 					}, | ||||
| 					_ => { pb.add_char(*q); } | ||||
| 				}; | ||||
| 			} else { | ||||
| 				match c.unwrap() { | ||||
| 					Key::Backspace => { pb.backspace(); }, | ||||
| 					Key::Delete => { pb.delete(); }, | ||||
| 					Key::Left => { pb.cursor_left(); }, | ||||
| 					Key::Right => { pb.cursor_right(); }, | ||||
| 					Key::Up => { pb.hist_up(); }, | ||||
| 					Key::Down => { pb.hist_down(); }, | ||||
|  | ||||
| 					Key::Ctrl('d') | | ||||
| 					Key::Ctrl('c') => { break 'outer; }, | ||||
| 					_ => {} | ||||
| 				}; | ||||
| 			}; | ||||
|  | ||||
| 			pb.write_prompt(&mut context, &mut stdout)?; | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	FormattedText::newline(&mut stdout)?; | ||||
| 	return Ok(()); | ||||
| } | ||||
| @ -102,14 +102,7 @@ fn format_map_full(c: char) -> Option<String> { | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| pub fn format_map(c: char, context: &Context) -> Option<String> { | ||||
| 	match context.config.term_color_type { | ||||
| 		0 => format_map_none(c), | ||||
| 		1 => format_map_ansi(c), | ||||
| 		2 => format_map_full(c), | ||||
| 		_ => unreachable!("Invalid term_color_type") | ||||
| 	} | ||||
| } | ||||
|  | ||||
|  | ||||
|  | ||||
| impl FormattedText { | ||||
| @ -117,6 +110,15 @@ impl FormattedText { | ||||
| 		write!(stdout, "\n")?; | ||||
| 		return Ok(()); | ||||
| 	} | ||||
|  | ||||
| 	pub fn format_map(c: char, context: &Context) -> Option<String> { | ||||
| 		match context.config.term_color_type { | ||||
| 			0 => format_map_none(c), | ||||
| 			1 => format_map_ansi(c), | ||||
| 			2 => format_map_full(c), | ||||
| 			_ => unreachable!("Invalid term_color_type") | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
|  | ||||
| @ -161,7 +163,7 @@ impl FormattedText { | ||||
| 					match (a, b) { | ||||
| 						(c, ']') => { // Normal text | ||||
|  | ||||
| 							let q = format_map(c, context); | ||||
| 							let q = Self::format_map(c, context); | ||||
|  | ||||
| 							if q.is_some() { | ||||
| 								s.push_str(&q.unwrap()); | ||||
|  | ||||
							
								
								
									
										283
									
								
								src/lib.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										283
									
								
								src/lib.rs
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,283 @@ | ||||
| pub mod parser; | ||||
| pub mod command; | ||||
| pub mod quantity; | ||||
|  | ||||
| use crate::parser::substitute; | ||||
| use crate::parser::LineLocation; | ||||
|  | ||||
|  | ||||
| mod context; | ||||
| mod formattedtext; | ||||
| mod errors; | ||||
| mod evaluate; | ||||
|  | ||||
| pub use crate::formattedtext::FormattedText; | ||||
| pub use crate::context::Context; | ||||
| pub use crate::errors::DaisyError; | ||||
| pub use crate::evaluate::evaluate; | ||||
|  | ||||
|  | ||||
|  | ||||
| #[inline(always)] | ||||
| pub fn do_string( | ||||
| 	context: &mut Context, | ||||
| 	s: &String | ||||
| ) -> Result<FormattedText, FormattedText> { | ||||
|  | ||||
| 	let r: (LineLocation, DaisyError); | ||||
| 	if command::is_command(s) { | ||||
| 		return Ok(command::do_command(context, s)); | ||||
| 	} else if s.contains("=") { | ||||
| 		let x = do_assignment(context, s); | ||||
| 		match x { | ||||
| 			Ok(t) => { return Ok(t) }, | ||||
| 			Err(t) => { r = t } | ||||
| 		}; | ||||
| 	} else { | ||||
| 		let x = do_expression(context, s); | ||||
| 		match x { | ||||
| 			Ok((t, e)) => { context.push_hist(e); return Ok(t) }, | ||||
| 			Err(t) => { r = t } | ||||
| 		}; | ||||
| 	} | ||||
|  | ||||
| 	let (l, e) = r; | ||||
| 	let mut t = FormattedText::new("".to_string()); | ||||
| 	if l.zero() { | ||||
| 		t.push(&format!( | ||||
| 			"\n  {}\n\n", | ||||
| 			e.text().to_string(), | ||||
| 		)); | ||||
| 	} else { | ||||
| 		t.push(&format!( | ||||
| 			concat!( | ||||
| 				"{}[e]{}[n]\n", | ||||
| 				"  {}\n\n" | ||||
| 			), | ||||
| 			" ".repeat(l.pos + 4), | ||||
| 			"^".repeat(l.len), | ||||
| 			e.text().to_string(), | ||||
| 		)); | ||||
| 	} | ||||
|  | ||||
| 	return Err(t); | ||||
| } | ||||
|  | ||||
| // Handle a simple evaluation string. | ||||
| // Returns a FormattedText with output that should be printed. | ||||
| #[inline(always)] | ||||
| fn do_expression( | ||||
| 	context: &mut Context, | ||||
| 	s: &String | ||||
| ) -> Result<(FormattedText, parser::Expression), (LineLocation, DaisyError)> { | ||||
|  | ||||
| 	let mut output = FormattedText::new("".to_string()); | ||||
|  | ||||
| 	let g = parser::parse(context, &s)?; | ||||
| 	let g_evaluated = evaluate::evaluate(context, &g)?; | ||||
|  | ||||
| 	// Display parsed string | ||||
| 	output.push(&format!( | ||||
| 		" [s]=>[n] {}\n\n", | ||||
| 		g.display(context) | ||||
| 	)); | ||||
|  | ||||
| 	// Display result | ||||
| 	output.push(&format!( | ||||
| 		"  [r]=[n] {}\n\n", | ||||
| 		g_evaluated.display_outer(context), | ||||
| 	)); | ||||
|  | ||||
| 	return Ok((output, g_evaluated)); | ||||
| } | ||||
|  | ||||
|  | ||||
| // Handle a variable or function definition string. | ||||
| // Returns a FormattedText with output that should be printed. | ||||
| #[inline(always)] | ||||
| fn do_assignment( | ||||
| 	context: &mut Context, | ||||
| 	s: &String | ||||
| ) -> Result<FormattedText, (LineLocation, DaisyError)> { | ||||
|  | ||||
| 	let mut output = FormattedText::new("".to_string()); | ||||
|  | ||||
| 	let parts = s.split("=").collect::<Vec<&str>>(); | ||||
| 	if parts.len() != 2 { | ||||
| 		return Err(( | ||||
| 			LineLocation::new_zero(), | ||||
| 			DaisyError::Syntax | ||||
| 		)); | ||||
| 	} | ||||
|  | ||||
| 	// Index of first non-whitespace character in left | ||||
| 	// (relative to whole prompt) | ||||
| 	let starting_left = parts[0] | ||||
| 		.char_indices() | ||||
| 		.find(|(_, ch)| !(ch.is_whitespace() && *ch != '\n')) | ||||
| 		.map(|(i, _)| i) | ||||
| 		.unwrap_or_else(|| parts[0].len()); | ||||
|  | ||||
| 	// Index of first non-whitespace character in right | ||||
| 	// (relative to whole prompt) | ||||
| 	// +1 accounts for equals sign | ||||
| 	let starting_right = parts[0].chars().count() + 1 + | ||||
| 		parts[1] | ||||
| 			.char_indices() | ||||
| 			.find(|(_, ch)| !(ch.is_whitespace() && *ch != '\n')) | ||||
| 			.map(|(i, _)| i) | ||||
| 			.unwrap_or_else(|| parts[0].len()); | ||||
|  | ||||
|  | ||||
| 	let left = substitute(context, &parts[0].trim().to_string()); | ||||
| 	let right = substitute(context, &parts[1].trim().to_string()); | ||||
| 	let is_function = left.contains("("); | ||||
|  | ||||
| 	// The order of methods below is a bit odd. | ||||
| 	// This is intentional, since we want to check a definition's | ||||
| 	// variable name before even attempting to parse its content. | ||||
| 	if is_function { | ||||
| 		let mut mode = 0; | ||||
| 		let mut name = String::new(); | ||||
| 		let mut args = String::new(); | ||||
| 		for c in left.chars() { | ||||
| 			match mode { | ||||
|  | ||||
| 				// Mode 0: reading function name | ||||
| 				0 => { | ||||
| 					if c == '(' { | ||||
| 						mode = 1; continue; | ||||
| 					} else { name.push(c); } | ||||
| 				}, | ||||
|  | ||||
| 				// Mode 1: reading arguments | ||||
| 				1 => { | ||||
| 					if c == ')' { | ||||
| 						mode = 2; continue; | ||||
| 					} else { args.push(c); } | ||||
| 				}, | ||||
|  | ||||
| 				// Mode 2: we should be done by now. | ||||
| 				// That close paren should've been the last character. | ||||
| 				2 => { | ||||
| 					return Err(( | ||||
| 						LineLocation{ pos: starting_left, len: left.chars().count() }, | ||||
| 						DaisyError::Syntax | ||||
| 					)); | ||||
| 				}, | ||||
|  | ||||
| 				_ => unreachable!() | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
|  | ||||
| 		let args = args | ||||
| 			.split(",").collect::<Vec<&str>>() | ||||
| 			.iter().map(|x| x.trim().to_string()).collect::<Vec<String>>(); | ||||
|  | ||||
| 		if name.len() == 0 { | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: starting_left, len: left.chars().count() }, | ||||
| 				DaisyError::Syntax | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		if !context.valid_function(&name) { | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: starting_left, len: left.chars().count() }, | ||||
| 				DaisyError::BadFunction | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		if args.iter().find(|x| &x[..] == "").is_some() { | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: starting_left, len: left.chars().count() }, | ||||
| 				DaisyError::Syntax | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		for a in &args { | ||||
| 			if !context.valid_varible(a) { | ||||
| 				return Err(( | ||||
| 					LineLocation{ pos: starting_left, len: left.chars().count() }, | ||||
| 					DaisyError::BadVariable | ||||
| 				)); | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		// Parse right hand side | ||||
| 		let g = parser::parse(context, &right); | ||||
| 		let Ok(g) = g else { | ||||
| 			let Err((l, e)) = g else { unreachable!() }; | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: l.pos + starting_right, len: l.len}, | ||||
| 				e | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		// Display parsed string | ||||
| 		output.push(&format!( | ||||
| 			" [s]=>[n] {left} = {}\n\n", | ||||
| 			g.display(context) | ||||
| 		)); | ||||
|  | ||||
| 		// Evaluate expression with shadow variables | ||||
| 		for a in &args { context.add_shadow(a.to_string(), None);} | ||||
| 		let g_evaluated = evaluate::evaluate(context, &g); | ||||
| 		context.clear_shadow(); | ||||
| 		let Ok(_g_evaluated) = g_evaluated else { | ||||
| 			let Err((l, e)) = g_evaluated else { unreachable!() }; | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: l.pos + starting_right, len: l.len}, | ||||
| 				e | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		// We could push g_evaluated instead, but an un-evaluated string | ||||
| 		// makes the 'vars' command prettier. | ||||
| 		// | ||||
| 		// We still need to evaluate g above, though, to make sure it works. | ||||
| 		context.push_function(name, args, g).unwrap(); | ||||
| 	} else { | ||||
|  | ||||
| 		if !context.valid_varible(&left) { | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: starting_left, len: left.chars().count() }, | ||||
| 				DaisyError::BadVariable | ||||
| 			)); | ||||
| 		} | ||||
|  | ||||
| 		// Parse right hand side | ||||
| 		let g = parser::parse(context, &right); | ||||
| 		let Ok(g) = g else { | ||||
| 			let Err((l, e)) = g else { unreachable!() }; | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: l.pos + starting_right, len: l.len}, | ||||
| 				e | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		// Display parsed string | ||||
| 		output.push(&format!( | ||||
| 			" [t]=>[n] {left} = {}\n\n", | ||||
| 			g.display(context) | ||||
| 		)); | ||||
|  | ||||
| 		// Evaluate expression | ||||
| 		let g_evaluated = evaluate::evaluate(context, &g); | ||||
| 		let Ok(g_evaluated) = g_evaluated else { | ||||
| 			let Err((l, e)) = g_evaluated else { unreachable!() }; | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: l.pos + starting_right, len: l.len}, | ||||
| 				e | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		context.push_variable(left.to_string(), g_evaluated).unwrap(); | ||||
| 	} | ||||
|  | ||||
| 	return Ok(output); | ||||
|  | ||||
| } | ||||
|  | ||||
|  | ||||
							
								
								
									
										388
									
								
								src/main.rs
									
									
									
									
									
								
							
							
						
						
									
										388
									
								
								src/main.rs
									
									
									
									
									
								
							| @ -1,289 +1,149 @@ | ||||
| pub mod parser; | ||||
| pub mod command; | ||||
| pub mod quantity; | ||||
| pub mod evaluate; | ||||
| pub mod context; | ||||
| pub mod errors; | ||||
| pub mod formattedtext; | ||||
| pub mod promptbuffer; | ||||
|  | ||||
| use crate::parser::substitute; | ||||
| use crate::errors::DaisyError; | ||||
| use crate::formattedtext::FormattedText; | ||||
| use crate::context::Context; | ||||
| use crate::parser::LineLocation; | ||||
| use std::io::stdout; | ||||
| use std::io::stdin; | ||||
| use std::env; | ||||
|  | ||||
| use termion::{ | ||||
| 	event::Key, | ||||
| 	input::TermRead, | ||||
| 	raw::IntoRawMode, | ||||
| 	color::DetectColors | ||||
| }; | ||||
|  | ||||
| // Run main script for target system | ||||
| mod entrypoint; | ||||
| use crate::entrypoint::main_e; | ||||
| use promptbuffer::PromptBuffer; | ||||
| use daisycalc::command; | ||||
| use daisycalc::Context; | ||||
| use daisycalc::FormattedText; | ||||
|  | ||||
| use daisycalc::do_string; | ||||
|  | ||||
| #[cfg(test)] | ||||
| mod tests; | ||||
|  | ||||
| fn main() -> Result<(), std::io::Error> { | ||||
| 	return main_e(); | ||||
| } | ||||
|  | ||||
|  | ||||
| #[inline(always)] | ||||
| pub fn do_string( | ||||
| 	context: &mut Context, | ||||
| 	s: &String | ||||
| ) -> Result<FormattedText, FormattedText> { | ||||
| pub fn main() -> Result<(), std::io::Error> { | ||||
| 	let mut stdout = stdout().into_raw_mode().unwrap(); | ||||
| 	let mut pb: PromptBuffer = PromptBuffer::new(64); | ||||
| 	let mut context = Context::new(); | ||||
|  | ||||
| 	let r: (LineLocation, DaisyError); | ||||
| 	if command::is_command(s) { | ||||
| 		return Ok(command::do_command(context, s)); | ||||
| 	} else if s.contains("=") { | ||||
| 		let x = do_assignment(context, s); | ||||
| 		match x { | ||||
| 			Ok(t) => { return Ok(t) }, | ||||
| 			Err(t) => { r = t } | ||||
| 		}; | ||||
| 	// Detect color compatibilty | ||||
| 	// Currently unused, this is slow. | ||||
| 	/* | ||||
| 	let term_colors = stdout.available_colors().unwrap_or(0); | ||||
| 	if term_colors >= 256 { | ||||
| 		context.config.term_color_type = 2; | ||||
| 	} else if term_colors >= 8 { | ||||
| 		context.config.term_color_type = 1; | ||||
| 	} else { | ||||
| 		let x = do_expression(context, s); | ||||
| 		match x { | ||||
| 			Ok((t, e)) => { context.push_hist(e); return Ok(t) }, | ||||
| 			Err(t) => { r = t } | ||||
| 		}; | ||||
| 		context.config.term_color_type = 0; | ||||
| 	} | ||||
| 	*/ | ||||
|  | ||||
| 	let (l, e) = r; | ||||
| 	let mut t = FormattedText::new("".to_string()); | ||||
| 	if l.zero() { | ||||
| 		t.push(&format!( | ||||
| 			"\n  {}\n\n", | ||||
| 			e.text().to_string(), | ||||
|  | ||||
| 	// Handle command-line arguments | ||||
| 	let args: Vec<String> = env::args().collect(); | ||||
| 	if args.iter().any(|s| s == "--help") { | ||||
| 		let t = command::do_command(&mut context, &String::from("help")); | ||||
| 		t.write(&context, &mut stdout)?; | ||||
| 		let t = command::do_command(&mut context, &String::from("flags")); | ||||
| 		t.write(&context, &mut stdout)?; | ||||
| 		return Ok(()); | ||||
| 	} else if args.iter().any(|s| s == "--version") { | ||||
| 		let t = FormattedText::new(format!( | ||||
| 			"Daisy v{}\n", env!("CARGO_PKG_VERSION") | ||||
| 		)); | ||||
| 	} else { | ||||
| 		t.push(&format!( | ||||
| 		t.write(&context, &mut stdout)?; | ||||
| 		return Ok(()); | ||||
| 	} else if args.iter().any(|s| s == "--info") { | ||||
| 		let t = FormattedText::new(format!( | ||||
| 			concat!( | ||||
| 				"{}[e]{}[n]\n", | ||||
| 				"  {}\n\n" | ||||
| 				"Daisy v{}\n", | ||||
| 				"Your terminal supports {} colors.\n" | ||||
| 			), | ||||
| 			" ".repeat(l.pos + 4), | ||||
| 			"^".repeat(l.len), | ||||
| 			e.text().to_string(), | ||||
| 			env!("CARGO_PKG_VERSION"), | ||||
| 			stdout.available_colors().unwrap_or(0) | ||||
| 		)); | ||||
| 		t.write(&context, &mut stdout)?; | ||||
| 		return Ok(()); | ||||
| 	} else if args.iter().any(|s| s == "--256color") { | ||||
| 		context.config.term_color_type = 2; | ||||
| 	} else if args.iter().any(|s| s == "--8color") { | ||||
| 		context.config.term_color_type = 1; | ||||
| 	} else if args.iter().any(|s| s == "--0color") { | ||||
| 		context.config.term_color_type = 0; | ||||
| 	} else if args.iter().any(|s| s == "--nosub") { | ||||
| 		context.config.enable_substituion = false; | ||||
| 	} else if args.iter().any(|s| s == "--nosuper") { | ||||
| 		context.config.enable_super_powers = false; | ||||
| 	} else if args.iter().any(|s| s == "--nooneover") { | ||||
| 		context.config.enable_one_over_power = false; | ||||
| 	} | ||||
|  | ||||
| 	return Err(t); | ||||
| } | ||||
|  | ||||
| // Handle a simple evaluation string. | ||||
| // Returns a FormattedText with output that should be printed. | ||||
| #[inline(always)] | ||||
| fn do_expression( | ||||
| 	context: &mut Context, | ||||
| 	s: &String | ||||
| ) -> Result<(FormattedText, parser::Expression), (LineLocation, DaisyError)> { | ||||
|  | ||||
| 	let mut output = FormattedText::new("".to_string()); | ||||
|  | ||||
| 	let g = parser::parse(context, &s)?; | ||||
| 	let g_evaluated = evaluate::evaluate(context, &g)?; | ||||
|  | ||||
| 	// Display parsed string | ||||
| 	output.push(&format!( | ||||
| 		" [s]=>[n] {}\n\n", | ||||
| 		g.display(context) | ||||
| 	)); | ||||
|  | ||||
| 	// Display result | ||||
| 	output.push(&format!( | ||||
| 		"  [r]=[n] {}\n\n", | ||||
| 		g_evaluated.display_outer(context), | ||||
| 	)); | ||||
|  | ||||
| 	return Ok((output, g_evaluated)); | ||||
| } | ||||
| 	context.config.check(); | ||||
|  | ||||
|  | ||||
| // Handle a variable or function definition string. | ||||
| // Returns a FormattedText with output that should be printed. | ||||
| #[inline(always)] | ||||
| fn do_assignment( | ||||
| 	context: &mut Context, | ||||
| 	s: &String | ||||
| ) -> Result<FormattedText, (LineLocation, DaisyError)> { | ||||
| 	'outer: loop { | ||||
|  | ||||
| 	let mut output = FormattedText::new("".to_string()); | ||||
| 		pb.write_prompt(&mut context, &mut stdout)?; | ||||
|  | ||||
| 	let parts = s.split("=").collect::<Vec<&str>>(); | ||||
| 	if parts.len() != 2 { | ||||
| 		return Err(( | ||||
| 			LineLocation::new_zero(), | ||||
| 			DaisyError::Syntax | ||||
| 		)); | ||||
| 		let stdin = stdin(); | ||||
| 		for c in stdin.keys() { | ||||
| 			if let Key::Char(q) = c.as_ref().unwrap() { | ||||
| 				match q { | ||||
| 					'\n' => { | ||||
| 						// Print again without cursor, in case we pressed enter | ||||
| 						// while inside a substitution | ||||
| 						pb.write_prompt_nocursor(&mut context, &mut stdout)?; | ||||
| 						let in_str = pb.enter(); | ||||
| 						FormattedText::newline(&mut stdout)?; | ||||
| 						if in_str == "" { break; } | ||||
|  | ||||
| 						if in_str.trim() == "quit" { | ||||
| 							break 'outer; | ||||
| 						} else { | ||||
| 							let r = crate::do_string(&mut context, &in_str); | ||||
|  | ||||
| 							match r { | ||||
| 								Ok(t) | Err(t) => { | ||||
| 									t.write(&context, &mut stdout).unwrap(); | ||||
| 								} | ||||
| 							} | ||||
| 						} | ||||
|  | ||||
| 						break; | ||||
| 					}, | ||||
|  | ||||
| 					// Only process sane characters | ||||
| 					'a'..='z' | 'A'..='Z' | '0'..='9' | ||||
| 					|'!'|'@'|'#'|'$'|'%'|'^'|'&'|'*'|'('|')' | ||||
| 					|'?'|'~'|','|'.'|'['|']' | ||||
| 					|'<'|'>'|'/'|'_'|'-'|':'|'|'|'='|'+'|';' | ||||
| 					=> { pb.add_char(*q); }, | ||||
|  | ||||
| 					_ => {} | ||||
| 				}; | ||||
| 			} else { | ||||
| 				match c.unwrap() { | ||||
| 					Key::Backspace => { pb.backspace(); }, | ||||
| 					Key::Delete => { pb.delete(); }, | ||||
| 					Key::Left => { pb.cursor_left(); }, | ||||
| 					Key::Right => { pb.cursor_right(); }, | ||||
| 					Key::Up => { pb.hist_up(); }, | ||||
| 					Key::Down => { pb.hist_down(); }, | ||||
|  | ||||
| 					Key::Ctrl('d') | | ||||
| 					Key::Ctrl('c') => { break 'outer; }, | ||||
| 					_ => {} | ||||
| 				}; | ||||
| 			}; | ||||
|  | ||||
| 			pb.write_prompt(&mut context, &mut stdout)?; | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Index of first non-whitespace character in left | ||||
| 	// (relative to whole prompt) | ||||
| 	let starting_left = parts[0] | ||||
| 		.char_indices() | ||||
| 		.find(|(_, ch)| !(ch.is_whitespace() && *ch != '\n')) | ||||
| 		.map(|(i, _)| i) | ||||
| 		.unwrap_or_else(|| parts[0].len()); | ||||
|  | ||||
| 	// Index of first non-whitespace character in right | ||||
| 	// (relative to whole prompt) | ||||
| 	// +1 accounts for equals sign | ||||
| 	let starting_right = parts[0].chars().count() + 1 + | ||||
| 		parts[1] | ||||
| 			.char_indices() | ||||
| 			.find(|(_, ch)| !(ch.is_whitespace() && *ch != '\n')) | ||||
| 			.map(|(i, _)| i) | ||||
| 			.unwrap_or_else(|| parts[0].len()); | ||||
|  | ||||
|  | ||||
| 	let left = substitute(context, &parts[0].trim().to_string()); | ||||
| 	let right = substitute(context, &parts[1].trim().to_string()); | ||||
| 	let is_function = left.contains("("); | ||||
|  | ||||
| 	// The order of methods below is a bit odd. | ||||
| 	// This is intentional, since we want to check a definition's | ||||
| 	// variable name before even attempting to parse its content. | ||||
| 	if is_function { | ||||
| 		let mut mode = 0; | ||||
| 		let mut name = String::new(); | ||||
| 		let mut args = String::new(); | ||||
| 		for c in left.chars() { | ||||
| 			match mode { | ||||
|  | ||||
| 				// Mode 0: reading function name | ||||
| 				0 => { | ||||
| 					if c == '(' { | ||||
| 						mode = 1; continue; | ||||
| 					} else { name.push(c); } | ||||
| 				}, | ||||
|  | ||||
| 				// Mode 1: reading arguments | ||||
| 				1 => { | ||||
| 					if c == ')' { | ||||
| 						mode = 2; continue; | ||||
| 					} else { args.push(c); } | ||||
| 				}, | ||||
|  | ||||
| 				// Mode 2: we should be done by now. | ||||
| 				// That close paren should've been the last character. | ||||
| 				2 => { | ||||
| 					return Err(( | ||||
| 						LineLocation{ pos: starting_left, len: left.chars().count() }, | ||||
| 						DaisyError::Syntax | ||||
| 					)); | ||||
| 				}, | ||||
|  | ||||
| 				_ => unreachable!() | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
|  | ||||
| 		let args = args | ||||
| 			.split(",").collect::<Vec<&str>>() | ||||
| 			.iter().map(|x| x.trim().to_string()).collect::<Vec<String>>(); | ||||
|  | ||||
| 		if name.len() == 0 { | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: starting_left, len: left.chars().count() }, | ||||
| 				DaisyError::Syntax | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		if !context.valid_function(&name) { | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: starting_left, len: left.chars().count() }, | ||||
| 				DaisyError::BadFunction | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		if args.iter().find(|x| &x[..] == "").is_some() { | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: starting_left, len: left.chars().count() }, | ||||
| 				DaisyError::Syntax | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		for a in &args { | ||||
| 			if !context.valid_varible(a) { | ||||
| 				return Err(( | ||||
| 					LineLocation{ pos: starting_left, len: left.chars().count() }, | ||||
| 					DaisyError::BadVariable | ||||
| 				)); | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		// Parse right hand side | ||||
| 		let g = parser::parse(context, &right); | ||||
| 		let Ok(g) = g else { | ||||
| 			let Err((l, e)) = g else { unreachable!() }; | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: l.pos + starting_right, len: l.len}, | ||||
| 				e | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		// Display parsed string | ||||
| 		output.push(&format!( | ||||
| 			" [s]=>[n] {left} = {}\n\n", | ||||
| 			g.display(context) | ||||
| 		)); | ||||
|  | ||||
| 		// Evaluate expression with shadow variables | ||||
| 		for a in &args { context.add_shadow(a.to_string(), None);} | ||||
| 		let g_evaluated = evaluate::evaluate(context, &g); | ||||
| 		context.clear_shadow(); | ||||
| 		let Ok(_g_evaluated) = g_evaluated else { | ||||
| 			let Err((l, e)) = g_evaluated else { unreachable!() }; | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: l.pos + starting_right, len: l.len}, | ||||
| 				e | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		// We could push g_evaluated instead, but an un-evaluated string | ||||
| 		// makes the 'vars' command prettier. | ||||
| 		// | ||||
| 		// We still need to evaluate g above, though, to make sure it works. | ||||
| 		context.push_function(name, args, g).unwrap(); | ||||
| 	} else { | ||||
|  | ||||
| 		if !context.valid_varible(&left) { | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: starting_left, len: left.chars().count() }, | ||||
| 				DaisyError::BadVariable | ||||
| 			)); | ||||
| 		} | ||||
|  | ||||
| 		// Parse right hand side | ||||
| 		let g = parser::parse(context, &right); | ||||
| 		let Ok(g) = g else { | ||||
| 			let Err((l, e)) = g else { unreachable!() }; | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: l.pos + starting_right, len: l.len}, | ||||
| 				e | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		// Display parsed string | ||||
| 		output.push(&format!( | ||||
| 			" [t]=>[n] {left} = {}\n\n", | ||||
| 			g.display(context) | ||||
| 		)); | ||||
|  | ||||
| 		// Evaluate expression | ||||
| 		let g_evaluated = evaluate::evaluate(context, &g); | ||||
| 		let Ok(g_evaluated) = g_evaluated else { | ||||
| 			let Err((l, e)) = g_evaluated else { unreachable!() }; | ||||
| 			return Err(( | ||||
| 				LineLocation{ pos: l.pos + starting_right, len: l.len}, | ||||
| 				e | ||||
| 			)); | ||||
| 		}; | ||||
|  | ||||
| 		context.push_variable(left.to_string(), g_evaluated).unwrap(); | ||||
| 	} | ||||
|  | ||||
| 	return Ok(output); | ||||
|  | ||||
| } | ||||
|  | ||||
|  | ||||
| 	FormattedText::newline(&mut stdout)?; | ||||
| 	return Ok(()); | ||||
| } | ||||
| @ -1,9 +1,9 @@ | ||||
| use std::collections::VecDeque; | ||||
| use std::io::Write; | ||||
| use termion::raw::RawTerminal; | ||||
| use crate::formattedtext; | ||||
| use crate::parser::substitute_cursor; | ||||
| use crate::context::Context; | ||||
| use daisycalc::FormattedText; | ||||
| use daisycalc::parser::substitute_cursor; | ||||
| use daisycalc::Context; | ||||
| 
 | ||||
| const PROMPT_STR: &str = "==> "; | ||||
| 
 | ||||
| @ -55,8 +55,8 @@ impl PromptBuffer { | ||||
| 	
 | ||||
| 		write!( | ||||
| 			stdout, "\r{}{PROMPT_STR}{}{}", | ||||
| 			formattedtext::format_map('p', context).unwrap(), | ||||
| 			formattedtext::format_map('n', context).unwrap(), | ||||
| 			FormattedText::format_map('p', context).unwrap(), | ||||
| 			FormattedText::format_map('n', context).unwrap(), | ||||
| 			s | ||||
| 		)?; | ||||
| 
 | ||||
| @ -1,7 +1,7 @@ | ||||
| // Many of these have been borrowed from insect. | ||||
| use crate::parser; | ||||
| use crate::evaluate::evaluate; | ||||
| use crate::context::Context; | ||||
| use daisycalc::parser; | ||||
| use daisycalc::evaluate; | ||||
| use daisycalc::Context; | ||||
|  | ||||
| fn eval_to_str(s: &str) -> Result<String, ()> { | ||||
| 	let g = match parser::parse_no_context(&String::from(s)) { | ||||
|  | ||||
		Reference in New Issue
	
	Block a user