use crate::commands::command::EvaluatedWholeStreamCommandArgs; use crate::commands::cp::CopyArgs; use crate::commands::mkdir::MkdirArgs; use crate::commands::mv::MoveArgs; use crate::commands::rm::RemoveArgs; use crate::context::SourceMap; use crate::prelude::*; use crate::shell::shell::Shell; use crate::utils::ValueStructure; use std::ffi::OsStr; use std::path::{Path, PathBuf}; #[derive(Clone, Debug)] pub struct ValueShell { pub(crate) path: String, pub(crate) value: Tagged, } impl ValueShell { pub fn new(value: Tagged) -> ValueShell { ValueShell { path: "/".to_string(), value, } } fn members_under(&self, path: &Path) -> VecDeque> { let mut shell_entries = VecDeque::new(); let full_path = path.to_path_buf(); let mut viewed = self.value.clone(); let sep_string = std::path::MAIN_SEPARATOR.to_string(); let sep = OsStr::new(&sep_string); for p in full_path.iter() { match p { x if x == sep => {} step => match viewed.get_data_by_key(step.to_str().unwrap()) { Some(v) => { viewed = v.clone(); } _ => {} }, } } match viewed { Tagged { item: Value::List(l), .. } => { for item in l { shell_entries.push_back(item.clone()); } } x => { shell_entries.push_back(x.clone()); } } shell_entries } fn members(&self) -> VecDeque> { self.members_under(Path::new(".")) } } impl Shell for ValueShell { fn name(&self, source_map: &SourceMap) -> String { let origin_name = self.value.origin_name(source_map); format!( "{}", match origin_name { Some(x) => format!("{{{}}}", x), None => format!("<{}>", self.value.item.type_name(),), } ) } fn homedir(&self) -> Option { dirs::home_dir() } fn ls(&self, args: EvaluatedWholeStreamCommandArgs) -> Result { let mut full_path = PathBuf::from(self.path()); let target = args.nth(0); match target { Some(value) => full_path.push(Path::new(&value.as_path()?)), _ => {} } let mut value_system = ValueStructure::new(); value_system.walk_decorate(&self.value)?; if !value_system.exists(&full_path) { if let Some(target) = target { return Err(ShellError::labeled_error( "Can not list entries inside", "No such path exists", target.span(), )); } return Err(ShellError::labeled_error( "Can not list entries inside", "No such path exists", args.call_info.name_span, )); } Ok(self .members_under(full_path.as_path()) .map(|x| ReturnSuccess::value(x)) .to_output_stream()) } fn cd(&self, args: EvaluatedWholeStreamCommandArgs) -> Result { let destination = args.nth(0); let path = match destination { None => "/".to_string(), Some(v) => { let target = v.as_path()?; let mut cwd = PathBuf::from(&self.path); if target == PathBuf::from("..") { cwd.pop(); } else { match target.to_str() { Some(target) => match target.chars().nth(0) { Some(x) if x == '/' => cwd = PathBuf::from(target), _ => cwd.push(target), }, None => cwd.push(target), } } cwd.to_string_lossy().to_string() } }; let mut value_system = ValueStructure::new(); value_system.walk_decorate(&self.value)?; if !value_system.exists(&PathBuf::from(&path)) { if let Some(destination) = destination { return Err(ShellError::labeled_error( "Can not change to path inside", "No such path exists", destination.span(), )); } return Err(ShellError::labeled_error( "Can not change to path inside", "No such path exists", args.call_info.name_span, )); } let mut stream = VecDeque::new(); stream.push_back(ReturnSuccess::change_cwd(path)); Ok(stream.into()) } fn cp(&self, _args: CopyArgs, name: Span, _path: &str) -> Result { Err(ShellError::labeled_error( "cp not currently supported on values", "not currently supported", name, )) } fn mv(&self, _args: MoveArgs, name: Span, _path: &str) -> Result { Err(ShellError::labeled_error( "mv not currently supported on values", "not currently supported", name, )) } fn mkdir(&self, _args: MkdirArgs, name: Span, _path: &str) -> Result { Err(ShellError::labeled_error( "mkdir not currently supported on values", "not currently supported", name, )) } fn rm(&self, _args: RemoveArgs, name: Span, _path: &str) -> Result { Err(ShellError::labeled_error( "rm not currently supported on values", "not currently supported", name, )) } fn path(&self) -> String { self.path.clone() } fn set_path(&mut self, path: String) { let _ = std::env::set_current_dir(&path); self.path = path.clone(); } fn complete( &self, line: &str, pos: usize, _ctx: &rustyline::Context<'_>, ) -> Result<(usize, Vec), rustyline::error::ReadlineError> { let mut completions = vec![]; let mut possible_completion = vec![]; let members = self.members(); for member in members { match member { Tagged { item, .. } => { for desc in item.data_descriptors() { possible_completion.push(desc); } } } } let line_chars: Vec<_> = line.chars().collect(); let mut replace_pos = pos; while replace_pos > 0 { if line_chars[replace_pos - 1] == ' ' { break; } replace_pos -= 1; } for command in possible_completion.iter() { let mut pos = replace_pos; let mut matched = true; if pos < line_chars.len() { for chr in command.chars() { if line_chars[pos] != chr { matched = false; break; } pos += 1; if pos == line_chars.len() { break; } } } if matched { completions.push(rustyline::completion::Pair { display: command.to_string(), replacement: command.to_string(), }); } } Ok((replace_pos, completions)) } fn hint(&self, _line: &str, _pos: usize, _ctx: &rustyline::Context<'_>) -> Option { None } }