Prep for streaming objects

history.txt

@@ -64,3 +64,20 @@ ls
 cargo
 cargo build
 cargo run
+git status
+git add .
+git commit
+git push
+git status
+git add .
+git commit
+git push
+git config --global core.autocrlf input
+git config
+ls
+cd target
+ls
+cd target
+cd ..
+cd target
+ls

src/commands/args.rs

@@ -0,0 +1,6 @@
+use crate::Value;
+
+#[derive(Debug, Clone)]
+pub struct Args {
+    args: Vec<Value>,
+}

src/commands/cd.rs

@@ -6,16 +6,35 @@ use std::path::{Path, PathBuf};
 use sysinfo::SystemExt;
 
 #[derive(new)]
-pub struct Cd;
+pub struct CdBlueprint;
 
-impl crate::Command for Cd {
-    fn run(
-        &mut self,
+impl crate::CommandBlueprint for CdBlueprint {
+    fn create(
+        &self,
         args: Vec<String>,
-        _host: &dyn crate::Host,
+        host: &dyn crate::Host,
         env: &mut crate::Environment,
-    ) -> Result<Value, ShellError> {
-        env.cwd = dunce::canonicalize(env.cwd().join(&args[0]).as_path())?;
-        Ok(Value::nothing())
+    ) -> Box<dyn crate::Command> {
+        Box::new(Cd {
+            cwd: env.cwd().to_path_buf(),
+            target: args[0].clone(),
+        })
+    }
+}
+
+#[derive(new)]
+pub struct Cd {
+    cwd: PathBuf,
+    target: String,
+}
+
+impl crate::Command for Cd {
+    fn run(&mut self) -> Result<crate::CommandSuccess, ShellError> {
+        Ok(crate::CommandSuccess {
+            value: Value::nothing(),
+            action: vec![crate::CommandAction::ChangeCwd(dunce::canonicalize(
+                self.cwd.join(&self.target).as_path(),
+            )?)],
+        })
     }
 }

src/commands/command.rs

@@ -1,11 +1,31 @@
 use crate::errors::ShellError;
 use crate::object::Value;
+use std::path::PathBuf;
 
-pub trait Command {
-    fn run(
-        &mut self,
+pub trait CommandBlueprint {
+    fn create(
+        &self,
         args: Vec<String>,
         host: &dyn crate::Host,
         env: &mut crate::Environment,
-    ) -> Result<Value, ShellError>;
+    ) -> Box<dyn Command>;
+}
+
+crate enum CommandAction {
+    ChangeCwd(PathBuf),
+}
+
+pub struct CommandSuccess {
+    crate value: Value,
+    crate action: Vec<CommandAction>,
+}
+
+pub trait Command {
+    fn begin(&mut self) -> Result<(), ShellError> {
+        Ok(())
+    }
+    fn run(&mut self) -> Result<CommandSuccess, ShellError>;
+    fn end(&mut self) -> Result<(), ShellError> {
+        Ok(())
+    }
 }

src/commands/ls.rs

@@ -1,21 +1,36 @@
 use crate::errors::ShellError;
 use crate::object::process::Process;
 use crate::object::{DirEntry, ShellObject, Value};
+use crate::{Command, CommandSuccess};
 use derive_new::new;
+use std::path::PathBuf;
 use sysinfo::SystemExt;
 
 #[derive(new)]
-pub struct Ls;
+pub struct LsBlueprint;
+
+impl crate::CommandBlueprint for LsBlueprint {
+    fn create(
+        &self,
+        args: Vec<String>,
+        host: &dyn crate::Host,
+        env: &mut crate::Environment,
+    ) -> Box<dyn Command> {
+        Box::new(Ls {
+            cwd: env.cwd().to_path_buf(),
+        })
+    }
+}
+
+#[derive(new)]
+pub struct Ls {
+    cwd: PathBuf,
+}
 
 impl crate::Command for Ls {
-    fn run(
-        &mut self,
-        _args: Vec<String>,
-        _host: &dyn crate::Host,
-        env: &mut crate::Environment,
-    ) -> Result<Value, ShellError> {
+    fn run(&mut self) -> Result<CommandSuccess, ShellError> {
         let entries =
-            std::fs::read_dir(env.cwd()).map_err((|e| ShellError::new(format!("{:?}", e))))?;
+            std::fs::read_dir(&self.cwd).map_err((|e| ShellError::new(format!("{:?}", e))))?;
 
         let mut shell_entries = vec![];
 
@@ -24,6 +39,9 @@ impl crate::Command for Ls {
             shell_entries.push(value)
         }
 
-        Ok(Value::list(shell_entries))
+        Ok(CommandSuccess {
+            value: Value::list(shell_entries),
+            action: vec![],
+        })
     }
 }

src/commands/ps.rs

@@ -1,24 +1,39 @@
 use crate::errors::ShellError;
 use crate::object::process::Process;
 use crate::object::{ShellObject, Value};
+use crate::Command;
 use derive_new::new;
+use std::cell::RefCell;
+use std::rc::Rc;
 use sysinfo::SystemExt;
 
+#[derive(new)]
+pub struct PsBlueprint {
+    system: Rc<RefCell<sysinfo::System>>,
+}
+
+impl crate::CommandBlueprint for PsBlueprint {
+    fn create(
+        &self,
+        args: Vec<String>,
+        host: &dyn crate::Host,
+        env: &mut crate::Environment,
+    ) -> Box<dyn Command> {
+        Box::new(Ps::new(self.system.clone()))
+    }
+}
+
 #[derive(new)]
 pub struct Ps {
-    system: sysinfo::System,
+    system: Rc<RefCell<sysinfo::System>>,
 }
 
 impl crate::Command for Ps {
-    fn run(
-        &mut self,
-        _args: Vec<String>,
-        _host: &dyn crate::Host,
-        _env: &mut crate::Environment,
-    ) -> Result<Value, ShellError> {
-        self.system.refresh_all();
+    fn run(&mut self) -> Result<crate::CommandSuccess, ShellError> {
+        let mut system = self.system.borrow_mut();
+        system.refresh_all();
 
-        let list = self.system.get_process_list();
+        let list = system.get_process_list();
 
         let list = list
             .into_iter()
@@ -26,6 +41,9 @@ impl crate::Command for Ps {
             .take(5)
             .collect();
 
-        Ok(Value::List(list))
+        Ok(crate::CommandSuccess {
+            value: Value::List(list),
+            action: vec![],
+        })
     }
 }

src/main.rs

@@ -9,7 +9,7 @@ mod format;
 mod object;
 mod parser;
 
-crate use crate::commands::command::Command;
+crate use crate::commands::command::{Command, CommandAction, CommandBlueprint, CommandSuccess};
 crate use crate::env::{Environment, Host};
 crate use crate::errors::ShellError;
 crate use crate::format::RenderView;
@@ -19,8 +19,10 @@ use conch_parser::lexer::Lexer;
 use conch_parser::parse::DefaultParser;
 use rustyline::error::ReadlineError;
 use rustyline::Editor;
+use std::cell::RefCell;
 use std::collections::BTreeMap;
 use std::error::Error;
+use std::rc::Rc;
 use subprocess::Exec;
 use sysinfo::{self, SystemExt};
 
@@ -48,12 +50,12 @@ fn main() -> Result<(), Box<Error>> {
     let mut host = crate::env::host::BasicHost;
     let mut env = crate::Environment::basic()?;
 
-    let mut commands = BTreeMap::<String, Box<dyn crate::Command>>::new();
+    let mut commands = BTreeMap::<String, Box<dyn crate::CommandBlueprint>>::new();
 
-    let mut system = sysinfo::System::new();
-    let mut ps = crate::commands::ps::Ps::new(system);
-    let mut ls = crate::commands::ls::Ls;
-    let mut cd = crate::commands::cd::Cd;
+    let mut system = Rc::new(RefCell::new(sysinfo::System::new()));
+    let mut ps = crate::commands::ps::PsBlueprint::new(system);
+    let mut ls = crate::commands::ls::LsBlueprint;
+    let mut cd = crate::commands::cd::CdBlueprint;
 
     commands.insert("ps".to_string(), Box::new(ps));
     commands.insert("ls".to_string(), Box::new(ls));
@@ -86,8 +88,16 @@ fn main() -> Result<(), Box<Error>> {
 
                 match commands.get_mut(*command) {
                     Some(command) => {
-                        let result = command.run(args, &mut host, &mut env).unwrap();
-                        let view = result.to_generic_view();
+                        let mut instance = command.create(args, &mut host, &mut env);
+                        let result = instance.run()?;
+
+                        for action in result.action {
+                            match action {
+                                crate::CommandAction::ChangeCwd(cwd) => env.cwd = cwd,
+                            }
+                        }
+
+                        let view = result.value.to_generic_view();
                         let rendered = view.render_view(&mut host);
 
                         for line in rendered {