You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
1.5 KiB
72 lines
1.5 KiB
2 years ago
|
CUSTOM_TYPES = {
|
||
|
"example": "text/example"
|
||
|
}
|
||
|
|
||
|
import mimetypes
|
||
|
mimetypes.init()
|
||
|
|
||
|
types = mimetypes.types_map
|
||
|
|
||
|
for k,v in CUSTOM_TYPES.items():
|
||
|
types[k] = v
|
||
|
|
||
|
genString = "if (0) {} "
|
||
|
|
||
|
for k,v in types.items():
|
||
|
if "text" in v or "application/json" in v or "application/x-sh" in v or "application/xhtml+xml" in v or "application/xml" in v or "application/javascript" in v:
|
||
|
genString += f"""
|
||
|
else if (!strcmp(dot, "{k}")) {{
|
||
|
*mime = "{v}";
|
||
|
}}"""
|
||
|
else:
|
||
|
genString += f"""
|
||
|
else if (!strcmp(dot, "{k}")) {{
|
||
|
*mime = "{v}";
|
||
|
return 1;
|
||
|
}}"""
|
||
|
|
||
|
fullText = f"""#pragma once
|
||
|
#include <string.h>
|
||
|
#ifdef NO_GC
|
||
|
#include <gc.h>
|
||
|
#define malloc(n) GC_malloc(n)
|
||
|
#else
|
||
|
#include <stdlib.h>
|
||
|
#endif
|
||
|
|
||
|
#include "log.h"
|
||
|
|
||
|
int hasTriedIndex;
|
||
|
|
||
|
int getMime(char** mime, char* filename) {{
|
||
|
char* dot = strrchr(filename, '.');
|
||
|
|
||
|
if (!dot || dot == filename) {{
|
||
|
if (!hasTriedIndex) {{
|
||
|
hasTriedIndex++;
|
||
|
strcat(filename, "/index.html");
|
||
|
info("trying to upgrade directory to index.html");
|
||
|
return getMime(mime, filename);
|
||
|
}} else {{
|
||
|
warn("file extension does not exist or is the same as request's path.\\n");
|
||
|
hasTriedIndex = 1;
|
||
|
*mime = "text/plain";
|
||
|
return 0;
|
||
|
}}
|
||
|
}} else {{ {genString} else {{
|
||
|
warn("mimetype for %s not found\\n", filename);
|
||
|
*mime = "text/plain";
|
||
|
}}
|
||
|
if (hasTriedIndex == 1) {{
|
||
|
hasTriedIndex = 2;
|
||
|
}} else {{
|
||
|
hasTriedIndex = 0;
|
||
|
}}
|
||
|
}}
|
||
|
return 0;
|
||
|
}}
|
||
|
"""
|
||
|
|
||
|
with open("lib/mime.h", "w") as f:
|
||
|
f.write(fullText)
|