1 files changed, 97 insertions, 0 deletions
diff --git a/src/libnetdata/http/content_type.c b/src/libnetdata/http/content_type.c
new file mode 100644
index 000000000..3e388a1da
--- /dev/null
+++ b/src/libnetdata/http/content_type.c
@@ -0,0 +1,97 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#include "content_type.h"
+
+
+static struct {
+    const char *format;
+    HTTP_CONTENT_TYPE content_type;
+    bool needs_charset;
+    const char *options;
+} content_types[] = {
+    // primary - preferred during id-to-string conversions
+    { .format = "text/html",                    CT_TEXT_HTML, true },
+    { .format = "text/plain",                   CT_TEXT_PLAIN, true },
+    { .format = "text/css",                     CT_TEXT_CSS, true },
+    { .format = "text/yaml",                    CT_TEXT_YAML, true },
+    { .format = "application/yaml",             CT_APPLICATION_YAML, true },
+    { .format = "text/xml",                     CT_TEXT_XML, true },
+    { .format = "text/xsl",                     CT_TEXT_XSL, true },
+    { .format = "application/json",             CT_APPLICATION_JSON, true },
+    { .format = "application/xml",              CT_APPLICATION_XML, true },
+    { .format = "application/javascript",       CT_APPLICATION_X_JAVASCRIPT, true },
+    { .format = "application/octet-stream",     CT_APPLICATION_OCTET_STREAM, false },
+    { .format = "image/svg+xml",                CT_IMAGE_SVG_XML, false },
+    { .format = "application/x-font-truetype",  CT_APPLICATION_X_FONT_TRUETYPE, false },
+    { .format = "application/x-font-opentype",  CT_APPLICATION_X_FONT_OPENTYPE, false },
+    { .format = "application/font-woff",        CT_APPLICATION_FONT_WOFF, false },
+    { .format = "application/font-woff2",       CT_APPLICATION_FONT_WOFF2, false },
+    { .format = "application/vnd.ms-fontobject",CT_APPLICATION_VND_MS_FONTOBJ, false },
+    { .format = "image/png",                    CT_IMAGE_PNG, false },
+    { .format = "image/jpeg",                   CT_IMAGE_JPG, false },
+    { .format = "image/gif",                    CT_IMAGE_GIF, false },
+    { .format = "image/x-icon",                 CT_IMAGE_XICON, false },
+    { .format = "image/bmp",                    CT_IMAGE_BMP, false },
+    { .format = "image/icns",                   CT_IMAGE_ICNS, false },
+    { .format = "audio/mpeg",                   CT_AUDIO_MPEG, false },
+    { .format = "audio/ogg",                    CT_AUDIO_OGG, false },
+    { .format = "video/mp4",                    CT_VIDEO_MP4, false },
+    { .format = "application/pdf",              CT_APPLICATION_PDF, false },
+    { .format = "application/zip",              CT_APPLICATION_ZIP, false },
+    { .format = "image/png",                    CT_IMAGE_PNG, false },
+
+    // secondary - overlapping with primary
+
+    { .format = "text/plain",                   CT_PROMETHEUS, false, "version=0.0.4" },
+    { .format = "prometheus",                   CT_PROMETHEUS },
+    { .format = "text",                         CT_TEXT_PLAIN },
+    { .format = "txt",                          CT_TEXT_PLAIN },
+    { .format = "json",                         CT_APPLICATION_JSON },
+    { .format = "html",                         CT_TEXT_HTML },
+    { .format = "xml",                          CT_APPLICATION_XML },
+
+    // terminator
+    { .format = NULL,                           CT_TEXT_PLAIN },
+};
+
+HTTP_CONTENT_TYPE content_type_string2id(const char *format) {
+    if(format && *format) {
+        for (int i = 0; content_types[i].format; i++)
+            if (strcmp(content_types[i].format, format) == 0)
+                return content_types[i].content_type;
+    }
+
+    return CT_TEXT_PLAIN;
+}
+
+const char *content_type_id2string(HTTP_CONTENT_TYPE content_type) {
+    for (int i = 0; content_types[i].format; i++)
+        if (content_types[i].content_type == content_type)
+            return content_types[i].format;
+
+    return "text/plain";
+}
+
+void http_header_content_type(BUFFER *wb, HTTP_CONTENT_TYPE content_type) {
+    buffer_strcat(wb, "Content-Type: ");
+
+    for (int i = 0; content_types[i].format; i++) {
+        if (content_types[i].content_type == content_type) {
+            buffer_strcat(wb, content_types[i].format);
+
+            if(content_types[i].needs_charset) {
+                buffer_strcat(wb, "; charset=utf-8");
+            }
+            if(content_types[i].options) {
+                buffer_strcat(wb, "; ");
+                buffer_strcat(wb, content_types[i].options);
+            }
+
+            buffer_strcat(wb, "\r\n");
+
+            return;
+        }
+    }
+
+    buffer_strcat(wb, "text/plain; charset=utf-8\r\n");
+}