Add browser
This commit is contained in:
68
CHDataManagement/Server/TryFilesMiddleware.swift
Normal file
68
CHDataManagement/Server/TryFilesMiddleware.swift
Normal file
@@ -0,0 +1,68 @@
|
||||
import Vapor
|
||||
|
||||
struct TryFilesMiddleware: AsyncMiddleware {
|
||||
|
||||
let publicDirectory: String
|
||||
|
||||
func respond(to request: Request, chainingTo next: any AsyncResponder) async throws -> Response {
|
||||
let relativePath = request.url.path
|
||||
let potentialPaths = [
|
||||
publicDirectory + relativePath,
|
||||
publicDirectory + relativePath + ".html",
|
||||
publicDirectory + relativePath + "/1.html"
|
||||
]
|
||||
|
||||
for path in potentialPaths {
|
||||
var isDirectory: ObjCBool = false
|
||||
guard FileManager.default.fileExists(atPath: path, isDirectory: &isDirectory),
|
||||
!isDirectory.boolValue else {
|
||||
continue
|
||||
}
|
||||
let url = URL(fileURLWithPath: path)
|
||||
let data = try Data(contentsOf: url)
|
||||
let response = Response(status: .ok, body: .init(data: data))
|
||||
response.headers.replaceOrAdd(name: .contentType, value: url.mimeType)
|
||||
response.headers.replaceOrAdd(name: .contentLength, value: "\(data.count)")
|
||||
return response
|
||||
}
|
||||
return .init(status: .notFound)
|
||||
}
|
||||
}
|
||||
|
||||
private extension URL {
|
||||
|
||||
var mimeType: String {
|
||||
switch pathExtension.lowercased() {
|
||||
case "html": return "text/html"
|
||||
case "css": return "text/css"
|
||||
case "js": return "application/javascript"
|
||||
case "json": return "application/json"
|
||||
case "xml": return "application/xml"
|
||||
case "txt": return "text/plain"
|
||||
|
||||
case "jpg", "jpeg": return "image/jpeg"
|
||||
case "png": return "image/png"
|
||||
case "gif": return "image/gif"
|
||||
case "svg": return "image/svg+xml"
|
||||
case "webp": return "image/webp"
|
||||
case "ico": return "image/x-icon"
|
||||
|
||||
case "pdf": return "application/pdf"
|
||||
case "zip": return "application/zip"
|
||||
case "tar": return "application/x-tar"
|
||||
case "gz": return "application/gzip"
|
||||
case "rar": return "application/vnd.rar"
|
||||
|
||||
case "mp3": return "audio/mpeg"
|
||||
case "wav": return "audio/wav"
|
||||
case "ogg": return "audio/ogg"
|
||||
|
||||
case "mp4": return "video/mp4"
|
||||
case "mov": return "video/quicktime"
|
||||
case "avi": return "video/x-msvideo"
|
||||
case "webm": return "video/webm"
|
||||
|
||||
default: return "application/octet-stream" // Default binary data type
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user