fix: Content-Disposition

LotuxPunk · Jul 14, 2024 · 251fa54 · 251fa54
1 parent fb144e6
commit 251fa54
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/src/main/kotlin/be/vandeas/plugins/Routing.kt b/src/main/kotlin/be/vandeas/plugins/Routing.kt
@@ -66,18 +66,20 @@ fun Application.configureRouting() {
                         is FileReadResult.Failure -> call.respond(HttpStatusCode.InternalServerError, result.message)
                         is FileReadResult.NotFound -> call.respond(HttpStatusCode.NotFound, FileNameWithPath(path = path, fileName = fileName))
                         is FileReadResult.Success -> {
-                            call.response.header(
-                                HttpHeaders.ContentDisposition,
-                                ContentDisposition.Attachment.withParameter(ContentDisposition.Parameters.FileName, fileName)
-                                    .toString()
-                            )
-                            call.respondFile(result.file) {
-                                headers {
-                                    if (downloadFileName.isNotBlank()) {
-                                        append(HttpHeaders.ContentDisposition, ContentDisposition.Attachment.withParameter(ContentDisposition.Parameters.FileName, downloadFileName).toString())
-                                    }
-                                }
+                            if (downloadFileName.isNotBlank()) {
+                                call.response.header(
+                                    HttpHeaders.ContentDisposition,
+                                    ContentDisposition.Attachment.withParameter(ContentDisposition.Parameters.FileName, downloadFileName)
+                                        .toString()
+                                )
+                            } else {
+                                call.response.header(
+                                    HttpHeaders.ContentDisposition,
+                                    ContentDisposition.Attachment.withParameter(ContentDisposition.Parameters.FileName, fileName)
+                                        .toString()
+                                )
                             }
+                            call.respondFile(result.file)
                         }
                     }
                 }