Object.transformFields/transformValues

CHANGELOG

@@ -1,6 +1,7 @@
 v0.0.8 (..)
 -----------
 
+* Object.transformFields/transformValues
 * Url.open
 * Url.withQueryParams, http.XXX --queryParams={}
 * os.delete deletes paths bottom up

src/main/scala/com/github/mdr/mash/ns/core/ObjectClass.scala

@@ -16,6 +16,8 @@ object ObjectClass extends MashClass("core.Object") {
     HasFieldMethod,
     HoistMethod,
     MapMethod,
+    TransformFieldsMethod,
+    TransformValuesMethod,
     UnblessMethod,
     WithFieldMethod,
     WhereMethod,

src/main/scala/com/github/mdr/mash/ns/core/objectClass/TransformFieldsMethod.scala

@@ -0,0 +1,44 @@
+package com.github.mdr.mash.ns.core.objectClass
+
+import com.github.mdr.mash.evaluator.ToStringifier
+import com.github.mdr.mash.functions.{ BoundParams, MashMethod, Parameter, ParameterModel }
+import com.github.mdr.mash.ns.core.ObjectClass
+import com.github.mdr.mash.runtime.{ MashObject, MashString, MashValue }
+
+object TransformFieldsMethod extends MashMethod("transformFields") {
+
+  object Params {
+    val F = Parameter(
+      nameOpt = Some("f"),
+      summaryOpt = Some("Function used to transform field names of the object"),
+      descriptionOpt = Some(
+        """The function may take up to two arguments. If it takes only one argument, the field name is provided.
+          |  If takes two, the field name and value are provided.""".stripMargin))
+  }
+
+  import Params._
+
+  val params = ParameterModel(Seq(F))
+
+  def apply(target: MashValue, boundParams: BoundParams): MashObject = {
+    val obj = target.asInstanceOf[MashObject]
+    boundParams.validateFunction1Or2(F) match {
+      case Left(f)  ⇒
+        MashObject.of(obj.immutableFields.map { case (name, value) ⇒
+          ToStringifier.stringify(f(MashString(name))) -> value
+        })
+      case Right(f) ⇒
+        MashObject.of(obj.immutableFields.map { case (name, value) ⇒
+          ToStringifier.stringify(f(MashString(name), value)) -> value
+        })
+    }
+  }
+
+  override def typeInferenceStrategy = ObjectClass
+
+  override def summaryOpt: Option[String] = Some("Return a new object with the field names transformed")
+
+  override def descriptionOpt = Some(
+    """Examples:
+      |  { foo: 3, bar: 4 }.transformFields (.toUpper) # { FOO: 3, BAR: 4 }""".stripMargin)
+}

src/main/scala/com/github/mdr/mash/ns/core/objectClass/TransformValuesMethod.scala

@@ -0,0 +1,37 @@
+package com.github.mdr.mash.ns.core.objectClass
+
+import com.github.mdr.mash.functions.{ BoundParams, MashMethod, Parameter, ParameterModel }
+import com.github.mdr.mash.ns.core.ObjectClass
+import com.github.mdr.mash.runtime.{ MashNumber, MashObject, MashString, MashValue }
+
+object TransformValuesMethod extends MashMethod("transformValues") {
+
+  object Params {
+    val F = Parameter(
+      nameOpt = Some("f"),
+      summaryOpt = Some("Function used to transform values of the object"),
+      descriptionOpt = Some(
+        """The function may take up to two arguments. If it takes only one argument, the value is provided.
+          |  If takes two, the field name and value are provided.""".stripMargin))
+  }
+
+  import Params._
+
+  val params = ParameterModel(Seq(F))
+
+  def apply(target: MashValue, boundParams: BoundParams): MashObject = {
+    val obj = target.asInstanceOf[MashObject]
+    boundParams.validateFunction1Or2(F) match {
+      case Left(f)  ⇒ MashObject.of(obj.immutableFields.mapValues(f))
+      case Right(f) ⇒ MashObject.of(obj.immutableFields.map { case (name, value) ⇒ name -> f(MashString(name), value) })
+    }
+  }
+
+  override def typeInferenceStrategy = ObjectClass
+
+  override def summaryOpt: Option[String] = Some("Return a new object with the values transformed")
+
+  override def descriptionOpt = Some(
+    """Examples:
+      |  { foo: 3, bar: 4 }.transformValues (n => n * n) # { foo: 9, bar: 16 }""".stripMargin)
+}

src/test/scala/com/github/mdr/mash/evaluator/EvaluatorTest.scala

@@ -1086,6 +1086,14 @@ class EvaluatorTest extends AbstractEvaluatorTest {
   "{ apple: 1, bob: 2, cat: 3 } | map (f v => { (f.toUpper): v * v })" ==> "{ APPLE: 1, BOB: 4, CAT: 9 }"
   "{ apple: 1 } | map (f v => { (f): v, (f.reverse): v })" ==> "{ apple: 1, elppa: 1 }"
 
+  // Object.transformValues
+  "{ foo: 3, bar: 4 }.transformValues (n => n * n)" ==> "{ foo: 9, bar: 16 }"
+  """{ foo: 3, bar: 4 }.transformValues (f v => "$f$v")""" ==> "{ foo: 'foo3', bar: 'bar4' }"
+
+  // Object.transformFields
+  "{ foo: 3, bar: 4 }.transformFields (.toUpper)" ==> "{ FOO: 3, BAR: 4 }"
+  "{ foo: 3, bar: 4 }.transformFields (f v => f.toUpper + v)" ==> "{ FOO3: 3, BAR4: 4 }"
+
   // Object.grep
   "{ foo: 'wibble', bar: 'wobble', wibble: 'baz' }.grep 'wibble'" ==> "{ foo: 'wibble', wibble: 'baz' }"
   "{ foo: 'wibble', bar: 'wobble', wibble: 'baz' } | grep 'wibble'" ==> "{ foo: 'wibble', wibble: 'baz' }"
@@ -1106,4 +1114,5 @@ class EvaluatorTest extends AbstractEvaluatorTest {
 
   // Url.host
   "net.url 'http://example.com' | .host" ==> "'example.com'"
+
 }