avro-newts.scala

import com.sksamuel.avro4s.SchemaFor
import com.sksamuel.avro4s.refined._
import io.estatico.newtype.Coercible
import io.chrisdavenport.fuuid.FUUID
import java.util.UUID
import scala.concurrent.duration.FiniteDuration
import io.circe.{Json, JsonObject}
import com.comcast.ip4s.Ipv4Address

object avro extends SchemaForCoercible {
  final case class MapOf[K, V](key: K, value: V)

  // Avro only supports Map[String, V] so any other map should be encoded as an array
  implicit def _mapSchemaFor[K: SchemaFor: =!=[*, String], V: SchemaFor]: SchemaFor[Map[K, V]] =
    SchemaFor[List[MapOf[K, V]]].forType[Map[K, V]]

  implicit val _schemaForFuuid: SchemaFor[FUUID] =
    SchemaFor[UUID].forType[FUUID]

  implicit val _schemaForDuration: SchemaFor[FiniteDuration] =
    SchemaFor[String].forType[FiniteDuration]

  implicit val _schemaForJsonObject: SchemaFor[JsonObject] =
    SchemaFor[String].forType[JsonObject]

  implicit val _schemaForJson: SchemaFor[Json] =
    SchemaFor[String].forType[Json]

  implicit val _schemaForIpv4Address: SchemaFor[Ipv4Address] =
    SchemaFor[String].forType[Ipv4Address]
}

trait SchemaForCoercible {
  implicit def _schemaForCoercible[A: Coercible[B, *], B: SchemaFor]: SchemaFor[A] =
    SchemaFor[B].forType[A]
}