Latest PDS UI utils

1 files changed, 21 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/pdsuicommon/utils/WritesUtils.scala b/src/main/scala/xyz/driver/pdsuicommon/utils/WritesUtils.scala
new file mode 100644
index 0000000..fa05e96
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuicommon/utils/WritesUtils.scala
@@ -0,0 +1,21 @@
+package xyz.driver.pdsuicommon.utils
+
+import play.api.libs.json._
+
+object WritesUtils {
+
+  def filterKeys[T](p: String => Boolean)(implicit w: Writes[T]): Writes[T] = {
+    filter {
+      case (key, _) => p(key)
+    }
+  }
+
+  def filter[T](p: (String, JsValue) => Boolean)(implicit w: Writes[T]): Writes[T] = {
+    w.transform { input: JsValue =>
+      input match {
+        case JsObject(map) => JsObject(map.filter(Function.tupled(p)))
+        case x => x
+      }
+    }
+  }
+}