zio.http.Boundary.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2021 - 2023 Sporta Technologies PVT LTD & the ZIO HTTP contributors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * 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 zio.http

import java.nio.charset.Charset

import zio.stacktracer.TracingImplicits.disableAutoTrace
import zio.{Chunk, Trace}

/**
 * A multipart boundary, which consists of both the boundary and its charset.
 */
final case class Boundary(id: String, charset: Charset) { self =>
  val closingBoundary: String = s"--$id--"

  def contentTypeHeader: Headers = Headers(
    Header.ContentType(MediaType.multipart.`form-data`, Some(self)),
  )

  val encapsulationBoundary: String = s"--$id"

  def isClosing(bytes: Chunk[Byte]): Boolean = bytes == closingBoundaryBytes

  def isEncapsulating(bytes: Chunk[Byte]): Boolean = bytes == encapsulationBoundaryBytes

  private[http] val encapsulationBoundaryBytes = Chunk.fromArray(encapsulationBoundary.getBytes(charset))

  private[http] val closingBoundaryBytes = Chunk.fromArray(closingBoundary.getBytes(charset))
}

object Boundary {
  def apply(boundary: String): Boundary = Boundary(boundary, Charsets.Utf8)

  def fromContent(content: Chunk[Byte], charset: Charset = Charsets.Utf8): Option[Boundary] = {
    var i = 0
    var j = 0

    val boundaryBytes = content.foldWhile(Chunk.empty[Byte])(_ => i < 2) {
      case (buffer, byte) if byte == '\r' || byte == '\n' =>
        i = i + 1
        buffer
      case (buffer, byte) if byte == '-' && j < 2         =>
        j = j + 1
        buffer
      case (buffer, byte)                                 =>
        buffer :+ byte
    }
    if (i == 2 && j == 2)
      Some(Boundary(new String(boundaryBytes.toArray, charset)))
    else Option.empty
  }

  def fromHeaders(headers: Headers): Option[Boundary] =
    for {
      contentType <- headers.header(Header.ContentType)
      boundary    <- contentType.boundary
    } yield boundary

  def fromString(content: String, charset: Charset): Option[Boundary] =
    fromContent(Chunk.fromArray(content.getBytes(charset)), charset)

  def randomUUID(implicit trace: Trace): zio.UIO[Boundary] =
    zio.Random.nextUUID.map { id =>
      Boundary(s"----${id.toString}----")
    }
}