Send the proper MIME type when serving static filesc-rewrite

1 files changed, 26 insertions, 3 deletions

diff --git a/file.c b/file.c
index c40cfdc..7d7abb2 100644
--- a/file.c
+++ b/file.c
@@ -47,21 +47,44 @@ enum serve_method file_method_for_path(const char *filepath, enum http_res_code
 	return SERVE_METHOD_FILE;
 }
 
+static const char *mime_type_for_path(const char *filepath)
+{
+	const char *ext = strrchr(filepath, '.');
+	if (strcmp(ext, ".html") == 0)
+		return "text/html";
+	if (strcmp(ext, ".css") == 0)
+		return "text/css";
+	if (strcmp(ext, ".js") == 0)
+		return "text/javascript";
+	if (strcmp(ext, ".png") == 0)
+		return "image/png";
+	if (strcmp(ext, ".gif") == 0)
+		return "image/gif";
+	if (strcmp(ext, ".jpg") == 0 || strcmp(ext, ".jpeg") == 0)
+		return "image/jpeg";
+	if (strcmp(ext, ".webp") == 0)
+		return "image/webp";
+	return "text/plain";
+}
+
 int file_read(const char *filepath, int sockfd)
 {
 	FILE *fp;
 	char buffer[FILE_READBUF_SIZE];
+	const char *mime_type;
 	size_t bytes_read;
 
-	const char *content_type = "Content-Type: text/html\r\n\r\n";
-
 	fp = fopen(filepath, "rb");
 	if (fp == NULL) {
 		perror("Failed to open file");
 		return 1;
 	}
 
-	write(sockfd, content_type, strlen(content_type));
+	mime_type = mime_type_for_path(filepath);
+	write(sockfd, "Content-Type: ", 14);
+	write(sockfd, mime_type, strlen(mime_type));
+	write(sockfd, "\r\n\r\n", 4);
+
 	while ((bytes_read = fread(buffer, 1, FILE_READBUF_SIZE, fp)) > 0)
 		write(sockfd, buffer, bytes_read);