zio.http.Boundary.scala Maven / Gradle / Ivy
The newest version!
/*
* 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}----")
}
}