NotFound middleware returns prettier error

2019-06-15 21:15:30 +02:00
parent 9e99dc104f
commit 132f17e0d5
1 changed files with 6 additions and 5 deletions
--- a/src/middleware/notfound.cpp
+++ b/src/middleware/notfound.cpp
@@ -1,3 +1,4 @@
+#include "../http/mime.hpp"
 #include "notfound.hpp"
 #include <sstream>

@@ -11,17 +12,17 @@ namespace Middleware
 		}

 		response.code = HttpResponse::Code::NOT_FOUND;
+		response.contentType = Http::GetMimeType(Http::FileType::HTML);
 		
 		std::stringstream ss;
-		ss << "404 - file not found\n";
-		ss << "File: ";
-		ss << request.url.GetPath() << '\n';
+		ss << "<!DOCTYPE html><html><head></head><body>";
+		ss << "<h1>404 - File Not Found</h1>";
+		ss << "<p>File: " << request.url.GetPath() << "<p>";
+		ss << "</body></html>";

 		auto responseContent = ss.str();
 		response.content.insert(response.content.begin(),
 			responseContent.begin(),
 			responseContent.end());
-		
-		response.contentType = "text/plain";
 	}
 }