path: root/src/main/scala/xyz/driver/pdsuicommon/parsers/DimensionsParser.scala

package xyz.driver.server.parsers

import play.api.libs.json._
import play.api.routing.sird._
import xyz.driver.pdsuicommon.utils.WritesUtils

import scala.util.{Failure, Success, Try}

class Dimensions(private val xs: Set[String] = Set.empty) {
  def contains(x: String): Boolean = xs.isEmpty || xs.contains(x)
}

object DimensionsParser {

  private class DimensionsWrapper[T](dimensions: Dimensions)(implicit orig: Writes[T]) extends Writes[T] {
    private val filteredWrites         = WritesUtils.filterKeys[T](dimensions.contains)
    override def writes(o: T): JsValue = filteredWrites.writes(o)
  }

  def tryParse(queryString: QueryString): Try[Dimensions] = {
    val rawDimensions = queryString.getOrElse("dimensions", Seq.empty)
    rawDimensions match {
      case Nil => Success(new Dimensions())

      case x +: Nil =>
        val raw: Set[String] = x.split(",").view.map(_.trim).filter(_.nonEmpty).to[Set]
        Success(new Dimensions(raw))

      case xs =>
        Failure(new IllegalArgumentException(s"Dimensions are specified ${xs.size} times"))
    }
  }
}