Merge branch 'master' into size

This commit is contained in:
Jonathan Turner 2019-05-28 14:06:35 +12:00 committed by GitHub
commit d35a9d17ad
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
10 changed files with 96 additions and 6 deletions

3
Cargo.lock generated
View File

@ -1012,6 +1012,9 @@ dependencies = [
"prettytable-rs 0.8.0 (registry+https://github.com/rust-lang/crates.io-index)",
"regex 1.1.6 (registry+https://github.com/rust-lang/crates.io-index)",
"rustyline 4.1.0 (registry+https://github.com/rust-lang/crates.io-index)",
"serde 1.0.91 (registry+https://github.com/rust-lang/crates.io-index)",
"serde_derive 1.0.91 (registry+https://github.com/rust-lang/crates.io-index)",
"serde_json 1.0.39 (registry+https://github.com/rust-lang/crates.io-index)",
"subprocess 0.1.18 (registry+https://github.com/rust-lang/crates.io-index)",
"sysinfo 0.8.4 (registry+https://github.com/rust-lang/crates.io-index)",
"term 0.5.2 (registry+https://github.com/rust-lang/crates.io-index)",

View File

@ -37,6 +37,9 @@ log = "0.4.6"
pretty_env_logger = "0.3.0"
lalrpop-util = "0.17.0"
regex = "1.1.6"
serde = "1.0.91"
serde_json = "1.0.39"
serde_derive = "1.0.91"
[dependencies.pancurses]
version = "0.16"

View File

@ -48,11 +48,14 @@ pub async fn cli() -> Result<(), Box<Error>> {
("view", Arc::new(view::view)),
("skip", Arc::new(skip::skip)),
("first", Arc::new(take::take)),
("select", Arc::new(select::select)),
("size", Arc::new(size::size)),
("from-json", Arc::new(from_json::from_json)),
("open", Arc::new(open::open)),
("column", Arc::new(select::select)),
("split", Arc::new(split::split)),
("reject", Arc::new(reject::reject)),
("to-array", Arc::new(to_array::to_array)),
("to-json", Arc::new(to_json::to_json)),
("where", Arc::new(where_::r#where)),
("sort-by", Arc::new(sort_by::sort_by)),
]);

View File

@ -2,7 +2,9 @@ crate mod args;
crate mod cd;
crate mod classified;
crate mod command;
crate mod from_json;
crate mod ls;
crate mod open;
crate mod ps;
crate mod reject;
crate mod select;
@ -12,6 +14,7 @@ crate mod sort_by;
crate mod split;
crate mod take;
crate mod to_array;
crate mod to_json;
crate mod view;
crate mod where_;

View File

@ -19,7 +19,22 @@ pub fn split(args: CommandArgs) -> Result<OutputStream, ShellError> {
debug!("split result = {:?}", split_result);
if split_result.len() == (args.len() - 1) {
// If they didn't provide column names, make up our own
if (args.len() - 1) == 0 {
let mut gen_columns = vec![];
for i in 0..split_result.len() {
gen_columns.push(format!("Column{}", i + 1));
}
let mut dict = crate::object::Dictionary::default();
for (k, v) in split_result.iter().zip(gen_columns.iter()) {
dict.add(
v.clone(),
Value::Primitive(Primitive::String(k.to_string())),
);
}
ReturnValue::Value(Value::Object(dict))
} else if split_result.len() == (args.len() - 1) {
let mut dict = crate::object::Dictionary::default();
for (k, v) in split_result.iter().zip(args.iter().skip(1)) {
dict.add(

View File

@ -1,9 +1,9 @@
#[allow(unused)]
use crate::prelude::*;
use serde_derive::Serialize;
use derive_new::new;
#[derive(Debug, Ord, PartialOrd, Eq, PartialEq, new, Clone)]
#[derive(Debug, Ord, PartialOrd, Eq, PartialEq, new, Clone, Serialize)]
pub struct ShellError {
title: String,
error: Value,

View File

@ -9,6 +9,9 @@ use derive_new::new;
use ordered_float::OrderedFloat;
use std::time::SystemTime;
use serde::{Serialize, Serializer};
use serde_derive::Serialize;
type OF64 = OrderedFloat<f64>;
#[derive(Debug, Clone, Ord, PartialOrd, Eq, PartialEq)]
@ -23,6 +26,23 @@ pub enum Primitive {
Date(DateTime<Utc>),
}
impl Serialize for Primitive {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
match self {
Primitive::Nothing => serializer.serialize_i32(0),
Primitive::Int(i) => serializer.serialize_i64(*i),
Primitive::Float(f) => serializer.serialize_f64(f.into_inner()),
Primitive::Bytes(b) => serializer.serialize_u128(*b),
Primitive::String(ref s) => serializer.serialize_str(s),
Primitive::Boolean(b) => serializer.serialize_bool(*b),
Primitive::Date(d) => serializer.serialize_str(&d.to_string()),
}
}
}
impl Primitive {
crate fn format(&self, field_name: Option<&str>) -> String {
match self {
@ -55,7 +75,7 @@ impl Primitive {
}
}
#[derive(Debug, Ord, PartialOrd, Eq, PartialEq, Clone, new)]
#[derive(Debug, Ord, PartialOrd, Eq, PartialEq, Clone, new, Serialize)]
pub struct Operation {
crate left: Value,
crate operator: Operator,
@ -73,6 +93,21 @@ pub enum Value {
Error(Box<ShellError>),
}
impl Serialize for Value {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
match self {
Value::Primitive(p) => p.serialize(serializer),
Value::Object(o) => o.serialize(serializer),
Value::List(l) => l.serialize(serializer),
Value::Operation(o) => o.serialize(serializer),
Value::Error(e) => e.serialize(serializer),
}
}
}
impl Value {
crate fn from_leaf(leaf: &tokens::Leaf) -> Value {
use tokens::*;

View File

@ -1,5 +1,6 @@
use crate::object::types::{AnyShell, Type};
use derive_new::new;
use serde::{Serialize, Serializer};
#[derive(Debug, Clone, Eq, PartialEq, Ord, PartialOrd, Hash)]
pub enum DescriptorName {
@ -37,6 +38,18 @@ pub struct DataDescriptor {
crate ty: Box<dyn Type>,
}
impl Serialize for DataDescriptor {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
match self.name {
DescriptorName::String(ref s) => serializer.serialize_str(s),
DescriptorName::ValueOf => serializer.serialize_str("value_of")
}
}
}
impl From<&str> for DataDescriptor {
fn from(input: &str) -> DataDescriptor {
DataDescriptor {

View File

@ -3,6 +3,7 @@ use crate::prelude::*;
use crate::object::DataDescriptor;
use crate::object::{Primitive, Value};
use indexmap::IndexMap;
use serde::ser::{Serialize, Serializer, SerializeMap};
use std::cmp::{Ordering, PartialOrd};
#[derive(Debug, Default, Eq, PartialEq, Clone)]
@ -10,6 +11,19 @@ pub struct Dictionary {
entries: IndexMap<DataDescriptor, Value>,
}
impl Serialize for Dictionary {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
let mut map = serializer.serialize_map(Some(self.entries.len()))?;
for (k, v) in &self.entries {
map.serialize_entry(&k, &v)?;
}
map.end()
}
}
impl PartialOrd for Dictionary {
// TODO: FIXME
fn partial_cmp(&self, _other: &Dictionary) -> Option<Ordering> {

View File

@ -1,7 +1,8 @@
use derive_new::new;
use std::str::FromStr;
use serde_derive::{Deserialize, Serialize};
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Deserialize, Serialize)]
pub enum Operator {
Equal,
NotEqual,