From 59dec999b83b55050f965bfdbc04ea08513d75af Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9s=20N=2E=20Robalino?= <andres@androbtech.com>
Date: Sun, 28 Jul 2019 02:01:32 -0500
Subject: [PATCH] string utils plugin baseline.

---
 Cargo.toml            |   4 ++
 src/plugins/str.rs    | 125 ++++++++++++++++++++++++++++++++++++++++++
 tests/filters_test.rs |  22 ++++++++
 3 files changed, 151 insertions(+)
 create mode 100644 src/plugins/str.rs

diff --git a/Cargo.toml b/Cargo.toml
index 84bc8ff71..39aa0682f 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -104,6 +104,10 @@ path = "src/plugins/add.rs"
 name = "nu_plugin_edit"
 path = "src/plugins/edit.rs"
 
+[[bin]]
+name = "nu_plugin_str"
+path = "src/plugins/str.rs"
+
 [[bin]]
 name = "nu_plugin_skip"
 path = "src/plugins/skip.rs"
diff --git a/src/plugins/str.rs b/src/plugins/str.rs
new file mode 100644
index 000000000..7fdfcb21e
--- /dev/null
+++ b/src/plugins/str.rs
@@ -0,0 +1,125 @@
+use indexmap::IndexMap;
+use nu::{
+    serve_plugin, CallInfo, CommandConfig, NamedType, Plugin, PositionalType, Primitive,
+    ReturnSuccess, ReturnValue, ShellError, Spanned, SpannedItem, Value,
+};
+
+struct Str {
+    field: Option<String>,
+    downcase: bool,
+    upcase: bool,
+}
+
+impl Str {
+    fn new() -> Str {
+        Str {
+            field: None,
+            downcase: false,
+            upcase: false,
+        }
+    }
+
+    fn strutils(
+        &self,
+        value: Spanned<Value>,
+        field: &Option<String>,
+    ) -> Result<Spanned<Value>, ShellError> {
+        match value.item {
+            Value::Primitive(Primitive::String(s)) => {
+                let applied = if self.downcase {
+                    Value::string(s.to_ascii_lowercase())
+                } else if self.upcase {
+                    Value::string(s.to_ascii_uppercase())
+                } else {
+                    Value::string(s)
+                };
+
+                Ok(Spanned {
+                    item: applied,
+                    span: value.span,
+                })
+            }
+            Value::Object(_) => match field {
+                Some(f) => {
+                    let replacement = match value.item.get_data_by_path(value.span, f) {
+                        Some(result) => self.strutils(result.map(|x| x.clone()), &None)?,
+                        None => {
+                            return Err(ShellError::string("str could not find field to replace"))
+                        }
+                    };
+                    match value
+                        .item
+                        .replace_data_at_path(value.span, f, replacement.item.clone())
+                    {
+                        Some(v) => return Ok(v),
+                        None => {
+                            return Err(ShellError::string("str could not find field to replace"))
+                        }
+                    }
+                }
+                None => Err(ShellError::string(
+                    "str needs a field when applying it to a value in an object",
+                )),
+            },
+            x => Err(ShellError::string(format!(
+                "Unrecognized type in stream: {:?}",
+                x
+            ))),
+        }
+    }
+}
+
+impl Plugin for Str {
+    fn config(&mut self) -> Result<CommandConfig, ShellError> {
+        let mut named = IndexMap::new();
+        named.insert("downcase".to_string(), NamedType::Switch);
+        named.insert("upcase".to_string(), NamedType::Switch);
+
+        Ok(CommandConfig {
+            name: "str".to_string(),
+            positional: vec![PositionalType::optional_any("Field")],
+            is_filter: true,
+            is_sink: false,
+            named,
+            rest_positional: true,
+        })
+    }
+    fn begin_filter(&mut self, call_info: CallInfo) -> Result<Vec<ReturnValue>, ShellError> {
+        if call_info.args.has("downcase") {
+            self.downcase = true;
+        } else if call_info.args.has("upcase") {
+            self.upcase = true;
+        }
+
+        if let Some(args) = call_info.args.positional {
+            for arg in args {
+                match arg {
+                    Spanned {
+                        item: Value::Primitive(Primitive::String(s)),
+                        ..
+                    } => {
+                        self.field = Some(s);
+                    }
+                    _ => {
+                        return Err(ShellError::string(format!(
+                            "Unrecognized type in params: {:?}",
+                            arg
+                        )))
+                    }
+                }
+            }
+        }
+
+        Ok(vec![])
+    }
+
+    fn filter(&mut self, input: Spanned<Value>) -> Result<Vec<ReturnValue>, ShellError> {
+        Ok(vec![ReturnSuccess::value(
+            self.strutils(input, &self.field)?,
+        )])
+    }
+}
+
+fn main() {
+    serve_plugin(&mut Str::new());
+}
diff --git a/tests/filters_test.rs b/tests/filters_test.rs
index bf07898a2..5660fbaee 100644
--- a/tests/filters_test.rs
+++ b/tests/filters_test.rs
@@ -65,6 +65,28 @@ fn can_split_by_column() {
     assert_eq!(output, "name");
 }
 
+#[test]
+fn str_downcases() {
+    nu!(
+        output,
+        cwd("tests/fixtures/formats"),
+        "open caco3_plastics.csv | first 1 | str origin --downcase | get origin | echo $it"
+    );
+
+    assert_eq!(output, "spain");
+}
+
+#[test]
+fn str_upcases() {
+    nu!(
+        output,
+        cwd("tests/fixtures/formats"),
+        "open appveyor.yml | str environment.global.PROJECT_NAME --upcase | get environment.global.PROJECT_NAME | echo $it"
+    );
+
+    assert_eq!(output, "NUSHELL");
+}
+
 #[test]
 fn can_inc_version() {
     nu!(