1 files changed, 56 insertions, 0 deletions
diff --git a/src/main/scala/xyz/driver/restquery/query/Sorting.scala b/src/main/scala/xyz/driver/restquery/query/Sorting.scala
new file mode 100644
index 0000000..e2642ad
--- /dev/null
+++ b/src/main/scala/xyz/driver/restquery/query/Sorting.scala
@@ -0,0 +1,56 @@
+package xyz.driver.restquery.domain
+
+import scala.collection.generic.CanBuildFrom
+
+sealed trait SortingOrder
+object SortingOrder {
+
+  case object Ascending  extends SortingOrder
+  case object Descending extends SortingOrder
+
+}
+
+sealed trait Sorting
+
+object Sorting {
+
+  val Empty = Sequential(Seq.empty)
+
+  /**
+    * @param tableName None if the table is default (same)
+    * @param name      Dimension name
+    * @param order     Order
+    */
+  final case class Dimension(tableName: Option[String], name: String, order: SortingOrder) extends Sorting {
+    def isForeign: Boolean = tableName.isDefined
+  }
+
+  final case class Sequential(sorting: Seq[Dimension]) extends Sorting {
+    override def toString: String = if (isEmpty(this)) "Empty" else super.toString
+  }
+
+  def isEmpty(input: Sorting): Boolean = {
+    input match {
+      case Sequential(Seq()) => true
+      case _                 => false
+    }
+  }
+
+  def filter(sorting: Sorting, p: Dimension => Boolean): Seq[Dimension] = sorting match {
+    case x: Dimension if p(x) => Seq(x)
+    case _: Dimension         => Seq.empty
+    case Sequential(xs)       => xs.filter(p)
+  }
+
+  def collect[B, That](sorting: Sorting)(f: PartialFunction[Dimension, B])(
+          implicit bf: CanBuildFrom[Seq[Dimension], B, That]): That = sorting match {
+    case x: Dimension if f.isDefinedAt(x) =>
+      val r = bf.apply()
+      r += f(x)
+      r.result()
+
+    case _: Dimension   => bf.apply().result()
+    case Sequential(xs) => xs.collect(f)
+  }
+
+}