smithy4s.http.internals.SchemaVisitorPathEncoder.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of smithy4s-core_3 Show documentation
core
There is a newer version: 0.19.0-41-91762fb
/*
 *  Copyright 2021-2024 Disney Streaming
 *
 *  Licensed under the Tomorrow Open Source Technology License, Version 1.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *     https://disneystreaming.github.io/TOST-1.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

package smithy4s.http.internals

import smithy4s.schema._
import PathEncode.MaybePathEncode
import smithy.api.TimestampFormat
import smithy4s.Bijection
import smithy4s.http.PathSegment
import smithy4s.http.PathSegment.{GreedySegment, LabelSegment, StaticSegment}
import smithy4s.{Hints, Lazy, Refinement, ShapeId}
import smithy.api.Http

object SchemaVisitorPathEncoder
    extends SchemaVisitor[MaybePathEncode]
    with SchemaVisitor.Default[MaybePathEncode] {
  self =>

  def default[A]: MaybePathEncode[A] = None

  override def primitive[P](
      shapeId: ShapeId,
      hints: Hints,
      tag: Primitive[P]
  ): MaybePathEncode[P] = {
    tag match {
      case Primitive.PShort      => PathEncode.fromToString
      case Primitive.PInt        => PathEncode.fromToString
      case Primitive.PFloat      => PathEncode.fromToString
      case Primitive.PLong       => PathEncode.fromToString
      case Primitive.PDouble     => PathEncode.fromToString
      case Primitive.PBigInt     => PathEncode.fromToString
      case Primitive.PBigDecimal => PathEncode.fromToString
      case Primitive.PBoolean    => PathEncode.fromToString
      case Primitive.PString     => PathEncode.fromToString
      case Primitive.PUUID       => PathEncode.fromToString
      case Primitive.PByte       => PathEncode.fromToString
      case Primitive.PBlob       => default
      case Primitive.PDocument   => default
      case Primitive.PTimestamp =>
        val fmt =
          hints.get(TimestampFormat).getOrElse(TimestampFormat.DATE_TIME)
        Some(PathEncode.raw(_.format(fmt)))
    }
  }

  override def enumeration[E](
      shapeId: ShapeId,
      hints: Hints,
      tag: EnumTag[E],
      values: List[EnumValue[E]],
      total: E => EnumValue[E]
  ): MaybePathEncode[E] =
    tag match {
      case EnumTag.IntEnum() =>
        PathEncode.from(e => total(e).intValue.toString)
      case _ =>
        PathEncode.from(e => total(e).stringValue)
    }

  override def struct[S](
      shapeId: ShapeId,
      hints: Hints,
      fields: Vector[Field[S, _]],
      make: IndexedSeq[Any] => S
  ): MaybePathEncode[S] = {
    type Writer = S => List[String]

    def toPathEncoder[A](
        field: Field[S, A],
        greedy: Boolean
    ): Option[Writer] = {
      val writer =
        self(field.schema).map(_.contramap(field.get))
      if (greedy) writer.map(_.encodeGreedy)
      else writer.map(_.encode)
    }
    def compile1(path: PathSegment): Option[Writer] = path match {
      case StaticSegment(value) => Some(Function.const(List(value)))
      case LabelSegment(value) =>
        fields
          .find(_.label == value)
          .flatMap(field => toPathEncoder(field, greedy = false))
      case GreedySegment(value) =>
        fields
          .find(_.label == value)
          .flatMap(field => toPathEncoder(field, greedy = true))
    }

    def compilePath(path: Vector[PathSegment]): Option[Vector[Writer]] =
      path.traverse(compile1(_))
    for {
      httpHint <- hints.get[Http]
      path <- pathSegments(httpHint.uri.value)
      writers <- compilePath(path)
    } yield new PathEncode[S] {
      def encode(s: S): List[String] = writers.flatMap(_.apply(s)).toList
      def encodeGreedy(s: S): List[String] = Nil
    }
  }

  override def biject[A, B](
      schema: Schema[A],
      bijection: Bijection[A, B]
  ): MaybePathEncode[B] = {
    self(schema).map(_.contramap(bijection.from))
  }

  override def refine[A, B](
      schema: Schema[A],
      refinement: Refinement[A, B]
  ): MaybePathEncode[B] = {
    self(schema).map(_.contramap(refinement.from))
  }

  override def lazily[A](suspend: Lazy[Schema[A]]): MaybePathEncode[A] = {
    // "safe" because the `structure` implementation will not exercise any recursion
    // due to the fact that httpLabel can only be applied on members targeting
    // simple shapes.
    suspend.map(this.apply(_)).value
  }
}