1 0fbe79b3 2021-01-18 op /*
  2 0fbe79b3 2021-01-18 op  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
  3 0fbe79b3 2021-01-18 op  *
  4 0fbe79b3 2021-01-18 op  * Permission to use, copy, modify, and distribute this software for any
  5 0fbe79b3 2021-01-18 op  * purpose with or without fee is hereby granted, provided that the above
  6 0fbe79b3 2021-01-18 op  * copyright notice and this permission notice appear in all copies.
  7 0fbe79b3 2021-01-18 op  *
  8 0fbe79b3 2021-01-18 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 0fbe79b3 2021-01-18 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 0fbe79b3 2021-01-18 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 0fbe79b3 2021-01-18 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 0fbe79b3 2021-01-18 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 0fbe79b3 2021-01-18 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 0fbe79b3 2021-01-18 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 0fbe79b3 2021-01-18 op  */
 16 0fbe79b3 2021-01-18 op 
 17 0fbe79b3 2021-01-18 op #include <errno.h>
 18 0fbe79b3 2021-01-18 op #include <stdlib.h>
 19 0fbe79b3 2021-01-18 op #include <string.h>
 20 0fbe79b3 2021-01-18 op 
 21 0fbe79b3 2021-01-18 op #include "gmid.h"
 22 0fbe79b3 2021-01-18 op 
 23 0fbe79b3 2021-01-18 op void
 24 0fbe79b3 2021-01-18 op init_mime(void)
 25 0fbe79b3 2021-01-18 op {
 26 a010b0dd 2021-01-18 op 	conf.mimes.len = 0;
 27 a010b0dd 2021-01-18 op 	conf.mimes.cap = 2;
 28 0fbe79b3 2021-01-18 op 
 29 a010b0dd 2021-01-18 op 	conf.mimes.t = calloc(conf.mimes.cap, sizeof(struct etm));
 30 a010b0dd 2021-01-18 op 	if (conf.mimes.t == NULL)
 31 0fbe79b3 2021-01-18 op 		fatal("calloc: %s", strerror(errno));
 32 982069a1 2021-01-18 op 
 33 a010b0dd 2021-01-18 op 	conf.mimes.def = strdup("application/octet-stream");
 34 a010b0dd 2021-01-18 op 	if (conf.mimes.def == NULL)
 35 982069a1 2021-01-18 op 		fatal("strdup: %s", strerror(errno));
 36 982069a1 2021-01-18 op 
 37 0fbe79b3 2021-01-18 op }
 38 0fbe79b3 2021-01-18 op 
 39 982069a1 2021-01-18 op void
 40 982069a1 2021-01-18 op set_default_mime(const char *m)
 41 982069a1 2021-01-18 op {
 42 a010b0dd 2021-01-18 op 	free(conf.mimes.def);
 43 a010b0dd 2021-01-18 op 	if ((conf.mimes.def = strdup(m)) == NULL)
 44 982069a1 2021-01-18 op 		fatal("strdup: %s", strerror(errno));
 45 982069a1 2021-01-18 op }
 46 982069a1 2021-01-18 op 
 47 0fbe79b3 2021-01-18 op /* register mime for the given extension */
 48 0fbe79b3 2021-01-18 op void
 49 0fbe79b3 2021-01-18 op add_mime(const char *mime, const char *ext)
 50 0fbe79b3 2021-01-18 op {
 51 a010b0dd 2021-01-18 op 	if (conf.mimes.len == conf.mimes.cap) {
 52 a010b0dd 2021-01-18 op 		conf.mimes.cap *= 1.5;
 53 a010b0dd 2021-01-18 op 		conf.mimes.t = realloc(conf.mimes.t,
 54 a010b0dd 2021-01-18 op 		    conf.mimes.cap * sizeof(struct etm));
 55 a010b0dd 2021-01-18 op 		if (conf.mimes.t == NULL)
 56 0fbe79b3 2021-01-18 op 			fatal("realloc: %s", strerror(errno));
 57 0fbe79b3 2021-01-18 op 	}
 58 0fbe79b3 2021-01-18 op 
 59 a010b0dd 2021-01-18 op 	conf.mimes.t[conf.mimes.len].mime = mime;
 60 a010b0dd 2021-01-18 op 	conf.mimes.t[conf.mimes.len].ext  = ext;
 61 a010b0dd 2021-01-18 op 	conf.mimes.len++;
 62 0fbe79b3 2021-01-18 op }
 63 0fbe79b3 2021-01-18 op 
 64 0fbe79b3 2021-01-18 op /* load a default set of common mime-extension associations */
 65 0fbe79b3 2021-01-18 op void
 66 0fbe79b3 2021-01-18 op load_default_mime()
 67 0fbe79b3 2021-01-18 op {
 68 0fbe79b3 2021-01-18 op 	struct etm *i, m[] = {
 69 0fbe79b3 2021-01-18 op 		{"application/pdf",	"pdf"},
 70 0fbe79b3 2021-01-18 op 		{"image/gif",		"gif"},
 71 0fbe79b3 2021-01-18 op 		{"image/jpeg",		"jpg"},
 72 0fbe79b3 2021-01-18 op 		{"image/jpeg",		"jpeg"},
 73 0fbe79b3 2021-01-18 op 		{"image/png",		"png"},
 74 0fbe79b3 2021-01-18 op 		{"image/svg+xml",	"svg"},
 75 0fbe79b3 2021-01-18 op 		{"text/gemini",		"gemini"},
 76 0fbe79b3 2021-01-18 op 		{"text/gemini",		"gmi"},
 77 0fbe79b3 2021-01-18 op 		{"text/markdown",	"markdown"},
 78 0fbe79b3 2021-01-18 op 		{"text/markdown",	"md"},
 79 0fbe79b3 2021-01-18 op 		{"text/plain",		"txt"},
 80 0fbe79b3 2021-01-18 op 		{"text/xml",		"xml"},
 81 0fbe79b3 2021-01-18 op 		{NULL, NULL}
 82 0fbe79b3 2021-01-18 op 	};
 83 0fbe79b3 2021-01-18 op 
 84 0fbe79b3 2021-01-18 op 	for (i = m; i->mime != NULL; ++i)
 85 0fbe79b3 2021-01-18 op 		add_mime(i->mime, i->ext);
 86 0fbe79b3 2021-01-18 op }
 87 0fbe79b3 2021-01-18 op 
 88 0fbe79b3 2021-01-18 op static const char *
 89 0fbe79b3 2021-01-18 op path_ext(const char *path)
 90 0fbe79b3 2021-01-18 op {
 91 0fbe79b3 2021-01-18 op 	const char *end;
 92 0fbe79b3 2021-01-18 op 
 93 0fbe79b3 2021-01-18 op 	end = path + strlen(path)-1;
 94 0fbe79b3 2021-01-18 op 	for (; end != path; --end) {
 95 0fbe79b3 2021-01-18 op 		if (*end == '.')
 96 0fbe79b3 2021-01-18 op 			return end+1;
 97 0fbe79b3 2021-01-18 op 		if (*end == '/')
 98 0fbe79b3 2021-01-18 op 			break;
 99 0fbe79b3 2021-01-18 op 	}
100 0fbe79b3 2021-01-18 op 
101 0fbe79b3 2021-01-18 op 	return NULL;
102 0fbe79b3 2021-01-18 op }
103 0fbe79b3 2021-01-18 op 
104 0fbe79b3 2021-01-18 op const char *
105 0fbe79b3 2021-01-18 op mime(const char *path)
106 0fbe79b3 2021-01-18 op {
107 982069a1 2021-01-18 op 	const char *ext;
108 0fbe79b3 2021-01-18 op 	struct etm *t;
109 0fbe79b3 2021-01-18 op 
110 0fbe79b3 2021-01-18 op 	if ((ext = path_ext(path)) == NULL)
111 a010b0dd 2021-01-18 op 		return conf.mimes.def;
112 0fbe79b3 2021-01-18 op 
113 a010b0dd 2021-01-18 op 	for (t = conf.mimes.t; t->mime != NULL; ++t)
114 0fbe79b3 2021-01-18 op 		if (!strcmp(ext, t->ext))
115 0fbe79b3 2021-01-18 op 			return t->mime;
116 0fbe79b3 2021-01-18 op 
117 a010b0dd 2021-01-18 op 	return conf.mimes.def;
118 0fbe79b3 2021-01-18 op }