Fix implementation

Author: notxcain

zio-http/shared/src/main/scala/zio/http/Header.scala

@@ -87,10 +87,7 @@ object Header {
     private val errorConstructor =
       new ErrorConstructor {
         override def missing(fieldName: String): HttpCodecError =
-          if (fieldName == Header.Authorization.name)
-            HttpCodecError.MissingAuthorizationHeader
-          else
-            HttpCodecError.MissingHeader(fieldName)
+          HttpCodecError.MissingHeader(fieldName)
 
         override def missingAll(fieldNames: Chunk[String]): HttpCodecError =
           HttpCodecError.MissingHeaders(fieldNames)

zio-http/shared/src/main/scala/zio/http/codec/HttpCodecError.scala

@@ -23,7 +23,6 @@ import zio.{Cause, Chunk}
 import zio.schema.codec.DecodeError
 import zio.schema.validation.ValidationError
 
-import zio.http.Header.HeaderType
 import zio.http.{Path, Status}
 
 sealed trait HttpCodecError extends Exception with NoStackTrace with Product with Serializable {
@@ -34,9 +33,6 @@ object HttpCodecError {
   final case class MissingHeader(headerName: String)                                           extends HttpCodecError {
     def message = s"Missing header $headerName"
   }
-  case object MissingAuthorizationHeader                                                       extends HttpCodecError {
-    def message = "Missing header Authorization"
-  }
   final case class MissingHeaders(headerNames: Chunk[String])                                  extends HttpCodecError {
     def message = s"Missing headers ${headerNames.mkString(", ")}"
   }

zio-http/shared/src/main/scala/zio/http/endpoint/Endpoint.scala

@@ -27,7 +27,7 @@ import zio.schema.Schema
 
 import zio.http.Header.Accept.MediaTypeWithQFactor
 import zio.http._
-import zio.http.codec.{StatusCodec, _}
+import zio.http.codec._
 import zio.http.endpoint.Endpoint.{OutErrors, defaultMediaTypes}
 
 /**
@@ -340,9 +340,12 @@ final case class Endpoint[PathInput, Input, Err, Output, Auth <: AuthType](
               case Some(HttpCodecError.CustomError("SchemaTransformationFailure", message))
                   if maybeUnauthedResponse.isDefined && message.endsWith(" auth required") =>
                 maybeUnauthedResponse.get
-              case Some(HttpCodecError.MissingAuthorizationHeader) =>
+              case Some(HttpCodecError.MissingHeader(headerName)) if headerName == Header.Authorization.name =>
                 Handler.succeed(Response.unauthorized)
-              case Some(error)                                     =>
+              case Some(HttpCodecError.MissingHeaders(headerNames))
+                  if headerNames.contains(Header.Authorization.name) =>
+                Handler.succeed(Response.unauthorized)
+              case Some(error)                                                                               =>
                 Handler.fromFunctionZIO { (request: zio.http.Request) =>
                   val response = {
                     val outputMediaTypes =
@@ -356,7 +359,7 @@ final case class Endpoint[PathInput, Input, Err, Output, Auth <: AuthType](
                   }
                   ZIO.succeed(response)
                 }
-              case None                                            =>
+              case None                                                                                      =>
                 Handler.failCause(cause)
             }
           }