diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/main.cpp | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/src/main.cpp b/src/main.cpp index ee7c673..c0c9bb3 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -11,6 +11,21 @@ #include "HttpResponse.h" #include "ServerConnection.h" +static std::string content_type_for_extension(const std::filesystem::path& extension) +{ + if (extension == ".html") + return "text/html"; + if (extension == ".css") + return "text/css"; + if (extension == ".gif") + return "image/gif"; + if (extension == ".png") + return "image/png"; + if (extension == ".webp") + return "image/webp"; + return "text/plain"; +} + static HttpResponse serve_from_filesystem(const HttpRequest& request) { namespace fs = std::filesystem; @@ -40,7 +55,7 @@ static HttpResponse serve_from_filesystem(const HttpRequest& request) std::string file_contents((std::istreambuf_iterator<char>(input_file)), std::istreambuf_iterator<char>()); response.set_status_code(HttpStatusCode::OK); - response.add_header("Content-Type", "text/plain"); + response.add_header("Content-Type", content_type_for_extension(request_path.extension())); response.set_content(file_contents); } else { response.set_status_code(HttpStatusCode::NotFound); |