add method type parameter ERROR_OUTPUT for csrf

Author: fupelaqu

example/src/main/scala/com/softwaremill/example/session/SetSessionTapir.scala

@@ -10,9 +10,9 @@ import com.softwaremill.session._
 import com.typesafe.scalalogging.StrictLogging
 import sttp.model.headers.WWWAuthenticateChallenge
 import sttp.tapir.EndpointInput.AuthType
-import sttp.tapir.{EndpointInput, auth, endpoint, stringBody}
+import sttp.tapir.{EndpointInput, auth, stringBody}
 import sttp.tapir.model.UsernamePassword
-import sttp.tapir.server.{PartialServerEndpointWithSecurityOutput, ServerEndpoint}
+import sttp.tapir.server.ServerEndpoint
 import sttp.tapir.server.akkahttp.AkkaHttpServerInterpreter
 import sttp.tapir.swagger.bundle.SwaggerInterpreter
 

tapir/src/main/scala/com/softwaremill/session/CsrfEndpoints.scala

@@ -9,14 +9,14 @@ import scala.concurrent.{ExecutionContext, Future}
 
 trait CsrfEndpoints {
 
-  def hmacTokenCsrfProtection[T, SECURITY_INPUT, PRINCIPAL, SECURITY_OUTPUT](
+  def hmacTokenCsrfProtection[T, SECURITY_INPUT, PRINCIPAL, ERROR_OUTPUT, SECURITY_OUTPUT](
       checkMode: TapirCsrfCheckMode[T]
   )(
       body: => PartialServerEndpointWithSecurityOutput[
         SECURITY_INPUT,
         PRINCIPAL,
         Unit,
-        Unit,
+        ERROR_OUTPUT,
         SECURITY_OUTPUT,
         Unit,
         Any,
@@ -26,7 +26,7 @@ trait CsrfEndpoints {
     (SECURITY_INPUT, Option[String], Method, Option[String]),
     PRINCIPAL,
     Unit,
-    Unit,
+    _,
     (SECURITY_OUTPUT, Option[CookieValueWithMeta]),
     Unit,
     Any,

tapir/src/main/scala/com/softwaremill/session/TapirCsrf.scala

@@ -113,13 +113,14 @@ private[session] trait TapirCsrf[T] { _: CsrfCheck =>
   def hmacTokenCsrfProtection[
       SECURITY_INPUT,
       PRINCIPAL,
+      ERROR_OUTPUT,
       SECURITY_OUTPUT
   ](
       body: => PartialServerEndpointWithSecurityOutput[
         SECURITY_INPUT,
         PRINCIPAL,
         Unit,
-        Unit,
+        ERROR_OUTPUT,
         SECURITY_OUTPUT,
         Unit,
         Any,
@@ -129,7 +130,7 @@ private[session] trait TapirCsrf[T] { _: CsrfCheck =>
     (SECURITY_INPUT, Option[String], Method, Option[String]),
     PRINCIPAL,
     Unit,
-    Unit,
+    _,
     (SECURITY_OUTPUT, Option[CookieValueWithMeta]),
     Unit,
     Any,

tapir/src/main/scala/com/softwaremill/session/TapirEndpoints.scala

@@ -17,7 +17,7 @@ trait TapirEndpoints extends SessionEndpoints with CsrfEndpoints {
     (Seq[Option[String]], Option[String], Method, Option[String]),
     T,
     Unit,
-    Unit,
+    _,
     (Seq[Option[String]], Option[CookieValueWithMeta]),
     Unit,
     Any,
@@ -35,7 +35,7 @@ trait TapirEndpoints extends SessionEndpoints with CsrfEndpoints {
     (Seq[Option[String]], Option[String], Method, Option[String]),
     Option[T],
     Unit,
-    Unit,
+    _,
     (Seq[Option[String]], Option[CookieValueWithMeta]),
     Unit,
     Any,