Strip away a bit of cruft and add reject

This commit is contained in:
Yehuda Katz 2019-05-15 14:44:06 -07:00
parent 8f327477e7
commit 91f7d5384f
14 changed files with 113 additions and 103 deletions

View File

@ -3,6 +3,7 @@ crate mod cd;
crate mod command; crate mod command;
crate mod ls; crate mod ls;
crate mod ps; crate mod ps;
crate mod reject;
crate mod select; crate mod select;
crate mod take; crate mod take;
crate mod to_array; crate mod to_array;

View File

@ -1,6 +1,6 @@
use crate::errors::ShellError; use crate::errors::ShellError;
use crate::object::process::Process; use crate::object::process::Process;
use crate::object::{DirEntry, ShellObject, Value}; use crate::object::{dir_entry_dict, ShellObject, Value};
use crate::prelude::*; use crate::prelude::*;
use crate::Args; use crate::Args;
use derive_new::new; use derive_new::new;

View File

@ -1,6 +1,6 @@
use crate::errors::ShellError; use crate::errors::ShellError;
use crate::object::process::Process; use crate::object::process::Process;
use crate::object::{DirEntry, ShellObject, Value}; use crate::object::{dir_entry_dict, ShellObject, Value};
use crate::prelude::*; use crate::prelude::*;
use crate::Args; use crate::Args;
use crate::Command; use crate::Command;
@ -37,7 +37,7 @@ impl crate::Command for Ls {
let mut shell_entries = VecDeque::new(); let mut shell_entries = VecDeque::new();
for entry in entries { for entry in entries {
let value = Value::object(DirEntry::new(entry?)?); let value = Value::Object(dir_entry_dict(&entry?)?);
shell_entries.push_back(ReturnValue::Value(value)) shell_entries.push_back(ReturnValue::Value(value))
} }

View File

@ -1,5 +1,5 @@
use crate::errors::ShellError; use crate::errors::ShellError;
use crate::object::process::Process; use crate::object::process::process_dict;
use crate::object::{ShellObject, Value}; use crate::object::{ShellObject, Value};
use crate::prelude::*; use crate::prelude::*;
use crate::Command; use crate::Command;
@ -34,9 +34,7 @@ impl crate::Command for Ps {
let list = list let list = list
.into_iter() .into_iter()
.map(|(_, process)| { .map(|(_, process)| ReturnValue::Value(Value::Object(process_dict(process))))
ReturnValue::Value(Value::Object(Box::new(Process::new(process.clone()))))
})
.collect::<VecDeque<_>>(); .collect::<VecDeque<_>>();
Ok(list) Ok(list)

46
src/commands/reject.rs Normal file
View File

@ -0,0 +1,46 @@
use crate::errors::ShellError;
use crate::object::base::reject;
use crate::object::process::Process;
use crate::object::{dir_entry_dict, ShellObject, Value};
use crate::prelude::*;
use crate::Args;
use derive_new::new;
use std::path::{Path, PathBuf};
use sysinfo::SystemExt;
#[derive(new)]
pub struct RejectBlueprint;
impl crate::CommandBlueprint for RejectBlueprint {
fn create(
&self,
args: Vec<Value>,
host: &dyn Host,
env: &mut Environment,
) -> Result<Box<dyn Command>, ShellError> {
if args.is_empty() {
return Err(ShellError::string("take requires an integer"));
}
let fields: Result<_, _> = args.iter().map(|a| a.as_string()).collect();
Ok(Box::new(Reject { fields: fields? }))
}
}
#[derive(new)]
pub struct Reject {
fields: Vec<String>,
}
impl crate::Command for Reject {
fn run(&mut self, stream: VecDeque<Value>) -> Result<VecDeque<ReturnValue>, ShellError> {
let objects = stream
.iter()
.map(|item| Value::Object(reject(item, &self.fields)))
.map(|item| ReturnValue::Value(item))
.collect();
Ok(objects)
}
}

View File

@ -1,7 +1,7 @@
use crate::errors::ShellError; use crate::errors::ShellError;
use crate::object::base::select; use crate::object::base::select;
use crate::object::process::Process; use crate::object::process::Process;
use crate::object::{DirEntry, ShellObject, Value}; use crate::object::{dir_entry_dict, ShellObject, Value};
use crate::prelude::*; use crate::prelude::*;
use crate::Args; use crate::Args;
use derive_new::new; use derive_new::new;
@ -37,7 +37,7 @@ impl crate::Command for Select {
fn run(&mut self, stream: VecDeque<Value>) -> Result<VecDeque<ReturnValue>, ShellError> { fn run(&mut self, stream: VecDeque<Value>) -> Result<VecDeque<ReturnValue>, ShellError> {
let objects = stream let objects = stream
.iter() .iter()
.map(|item| Value::Object(Box::new(select(item, &self.fields)))) .map(|item| Value::Object(select(item, &self.fields)))
.map(|item| ReturnValue::Value(item)) .map(|item| ReturnValue::Value(item))
.collect(); .collect();

View File

@ -1,6 +1,6 @@
use crate::errors::ShellError; use crate::errors::ShellError;
use crate::object::process::Process; use crate::object::process::Process;
use crate::object::{DirEntry, ShellObject, Value}; use crate::object::{dir_entry_dict, ShellObject, Value};
use crate::prelude::*; use crate::prelude::*;
use crate::Args; use crate::Args;
use derive_new::new; use derive_new::new;

View File

@ -1,6 +1,6 @@
use crate::errors::ShellError; use crate::errors::ShellError;
use crate::object::process::Process; use crate::object::process::Process;
use crate::object::{DirEntry, ShellObject, Value}; use crate::object::{dir_entry_dict, ShellObject, Value};
use crate::prelude::*; use crate::prelude::*;
use crate::Args; use crate::Args;
use derive_new::new; use derive_new::new;

View File

@ -67,6 +67,7 @@ fn main() -> Result<(), Box<Error>> {
("cd", Box::new(cd::CdBlueprint)), ("cd", Box::new(cd::CdBlueprint)),
("take", Box::new(take::TakeBlueprint)), ("take", Box::new(take::TakeBlueprint)),
("select", Box::new(select::SelectBlueprint)), ("select", Box::new(select::SelectBlueprint)),
("reject", Box::new(reject::RejectBlueprint)),
("to-array", Box::new(to_array::ToArrayBlueprint)), ("to-array", Box::new(to_array::ToArrayBlueprint)),
]); ]);
} }

View File

@ -8,4 +8,4 @@ crate mod types;
crate use base::{Primitive, ShellObject, Value}; crate use base::{Primitive, ShellObject, Value};
crate use desc::{DataDescriptor, DataDescriptorInstance}; crate use desc::{DataDescriptor, DataDescriptorInstance};
crate use dict::Dictionary; crate use dict::Dictionary;
crate use files::DirEntry; crate use files::dir_entry_dict;

View File

@ -51,7 +51,7 @@ impl Primitive {
#[derive(Debug)] #[derive(Debug)]
pub enum Value { pub enum Value {
Primitive(Primitive), Primitive(Primitive),
Object(Box<dyn ShellObject>), Object(crate::object::Dictionary),
List(Vec<Value>), List(Vec<Value>),
Error(Box<ShellError>), Error(Box<ShellError>),
} }
@ -87,7 +87,7 @@ impl ShellObject for Value {
fn copy(&self) -> Value { fn copy(&self) -> Value {
match self { match self {
Value::Primitive(p) => Value::Primitive(p.clone()), Value::Primitive(p) => Value::Primitive(p.clone()),
Value::Object(o) => Value::Object(Box::new(o.copy())), Value::Object(o) => Value::Object(o.copy_dict()),
Value::List(l) => { Value::List(l) => {
let list = l.iter().map(|i| i.copy()).collect(); let list = l.iter().map(|i| i.copy()).collect();
Value::List(list) Value::List(list)
@ -179,10 +179,6 @@ impl Value {
crate fn list(values: impl Into<Vec<Value>>) -> Value { crate fn list(values: impl Into<Vec<Value>>) -> Value {
Value::List(values.into()) Value::List(values.into())
} }
crate fn object(value: impl ShellObject + 'static) -> Value {
Value::Object(Box::new(value))
}
} }
pub trait ShellObject: Debug { pub trait ShellObject: Debug {
@ -207,6 +203,22 @@ crate fn select(obj: impl ShellObject, fields: &[String]) -> crate::object::Dict
out out
} }
crate fn reject(obj: impl ShellObject, fields: &[String]) -> crate::object::Dictionary {
let mut out = crate::object::Dictionary::default();
let descs = obj.data_descriptors();
for desc in descs {
if fields.contains(&desc.name) {
continue;
} else {
out.add(desc.name.clone(), obj.get_data(&desc).borrow().copy())
}
}
out
}
pub trait ToEntriesView { pub trait ToEntriesView {
fn to_entries_view(&self) -> EntriesView; fn to_entries_view(&self) -> EntriesView;
} }

View File

@ -50,6 +50,6 @@ impl crate::object::ShellObject for Dictionary {
} }
fn copy(&self) -> Value { fn copy(&self) -> Value {
Value::Object(Box::new(self.copy_dict())) Value::Object(self.copy_dict())
} }
} }

View File

@ -14,59 +14,33 @@ pub enum FileType {
Symlink, Symlink,
} }
impl DirEntry { crate fn dir_entry_dict(entry: &std::fs::DirEntry) -> Result<Dictionary, ShellError> {
crate fn new(inner: std::fs::DirEntry) -> Result<DirEntry, ShellError> { let mut dict = Dictionary::default();
let mut dict = Dictionary::default(); let filename = entry.file_name();
let filename = inner.file_name(); dict.add("file name", Value::string(filename.to_string_lossy()));
dict.add("file name", Value::string(filename.to_string_lossy()));
let metadata = inner.metadata()?; let metadata = entry.metadata()?;
// let file_type = inner.file_type()?; // let file_type = inner.file_type()?;
let kind = if metadata.is_dir() { let kind = if metadata.is_dir() {
FileType::Directory FileType::Directory
} else if metadata.is_file() { } else if metadata.is_file() {
FileType::File FileType::File
} else { } else {
FileType::Symlink FileType::Symlink
}; };
dict.add("file type", Value::string(format!("{:?}", kind))); dict.add("file type", Value::string(format!("{:?}", kind)));
dict.add( dict.add(
"readonly", "readonly",
Value::boolean(metadata.permissions().readonly()), Value::boolean(metadata.permissions().readonly()),
); );
dict.add("size", Value::bytes(metadata.len() as u128)); dict.add("size", Value::bytes(metadata.len() as u128));
dict.add("created", Value::system_date_result(metadata.created())); dict.add("created", Value::system_date_result(metadata.created()));
dict.add("accessed", Value::system_date_result(metadata.accessed())); dict.add("accessed", Value::system_date_result(metadata.accessed()));
dict.add("modified", Value::system_date_result(metadata.modified())); dict.add("modified", Value::system_date_result(metadata.modified()));
// dict.add("created_at", Value::date()) Ok(dict)
Ok(DirEntry { dict })
}
}
impl ShellObject for DirEntry {
fn to_shell_string(&self) -> String {
format!("[object DirEntry]")
}
fn data_descriptors(&self) -> Vec<DataDescriptor> {
self.dict.data_descriptors()
}
fn get_data(&'a self, desc: &DataDescriptor) -> MaybeOwned<'a, Value> {
self.dict.get_data(desc)
}
fn copy(&self) -> Value {
let copy = DirEntry {
dict: self.dict.copy_dict(),
};
Value::Object(Box::new(copy))
}
} }

View File

@ -11,43 +11,21 @@ pub struct Process {
dict: Dictionary, dict: Dictionary,
} }
impl Process { crate fn process_dict(proc: &sysinfo::Process) -> Dictionary {
crate fn new(inner: sysinfo::Process) -> Process { let mut dict = Dictionary::default();
let mut dict = Dictionary::default(); dict.add("name", Value::string(proc.name()));
dict.add("name", Value::string(inner.name()));
let cmd = inner.cmd(); let cmd = proc.cmd();
let cmd_value = if cmd.len() == 0 { let cmd_value = if cmd.len() == 0 {
Value::nothing() Value::nothing()
} else { } else {
Value::string(join(cmd, "")) Value::string(join(cmd, ""))
}; };
dict.add("cmd", cmd_value); dict.add("cmd", cmd_value);
dict.add("pid", Value::int(inner.pid() as i64)); dict.add("pid", Value::int(proc.pid() as i64));
dict.add("status", Value::int(inner.status() as i64)); dict.add("status", Value::int(proc.status() as i64));
Process { dict } dict
}
}
impl ShellObject for Process {
fn to_shell_string(&self) -> String {
format!("Process")
}
fn data_descriptors(&self) -> Vec<DataDescriptor> {
self.dict.data_descriptors()
}
fn get_data(&'a self, desc: &DataDescriptor) -> MaybeOwned<'a, Value> {
self.dict.get_data(desc)
}
fn copy(&self) -> Value {
let copy = self.dict.copy_dict();
Value::Object(Box::new(copy))
}
} }