fiatjaf
134 lines
3.9 KiB
Go
134 lines
3.9 KiB
Go
package blossom
|
|
|
|
import (
|
|
"mime"
|
|
"strings"
|
|
)
|
|
|
|
var commonMimeExtensions = map[string]string{
|
|
"application/json": ".json",
|
|
"application/pdf": ".pdf",
|
|
"application/vnd.android.package-archive": ".apk",
|
|
"application/vnd.sqlite3": ".sqlite3",
|
|
"application/xml": ".xml",
|
|
"audio/aac": ".aac",
|
|
"audio/flac": ".flac",
|
|
"audio/midi": ".midi",
|
|
"audio/mp3": ".mp3",
|
|
"audio/mpeg": ".mp3",
|
|
"audio/mp4": ".m4a",
|
|
"audio/ogg": ".ogg",
|
|
"audio/wav": ".wav",
|
|
"audio/webm": ".weba",
|
|
"audio/x-aiff": ".aiff",
|
|
"audio/x-m4a": ".m4a",
|
|
"image/avif": ".avif",
|
|
"image/gif": ".gif",
|
|
"image/jpeg": ".jpg",
|
|
"image/png": ".png",
|
|
"image/svg+xml": ".svg",
|
|
"image/webp": ".webp",
|
|
"text/css": ".css",
|
|
"text/csv": ".csv",
|
|
"text/html": ".html",
|
|
"text/javascript": ".js",
|
|
"text/markdown": ".md",
|
|
"text/plain": ".txt",
|
|
"text/xml": ".xml",
|
|
"video/mp2t": ".ts",
|
|
"video/mp4": ".mp4",
|
|
"video/ogg": ".ogv",
|
|
"video/quicktime": ".mov",
|
|
"video/webm": ".webm",
|
|
"video/x-matroska": ".mkv",
|
|
}
|
|
|
|
var commonExtensionMimes = map[string]string{
|
|
".aac": "audio/aac",
|
|
".aiff": "audio/x-aiff",
|
|
".apk": "application/vnd.android.package-archive",
|
|
".avif": "image/avif",
|
|
".css": "text/css; charset=utf-8",
|
|
".csv": "text/csv; charset=utf-8",
|
|
".flac": "audio/flac",
|
|
".gif": "image/gif",
|
|
".html": "text/html; charset=utf-8",
|
|
".jpeg": "image/jpeg",
|
|
".jpg": "image/jpeg",
|
|
".js": "text/javascript; charset=utf-8",
|
|
".json": "application/json",
|
|
".m4a": "audio/mp4",
|
|
".md": "text/markdown; charset=utf-8",
|
|
".midi": "audio/midi",
|
|
".mkv": "video/x-matroska",
|
|
".mov": "video/quicktime",
|
|
".mp3": "audio/mpeg",
|
|
".mp4": "video/mp4",
|
|
".oga": "audio/ogg",
|
|
".ogg": "audio/ogg",
|
|
".ogv": "video/ogg",
|
|
".pdf": "application/pdf",
|
|
".png": "image/png",
|
|
".sqlite3": "application/vnd.sqlite3",
|
|
".svg": "image/svg+xml",
|
|
".ts": "video/mp2t",
|
|
".txt": "text/plain; charset=utf-8",
|
|
".wav": "audio/wav",
|
|
".weba": "audio/webm",
|
|
".webm": "video/webm",
|
|
".webp": "image/webp",
|
|
".xml": "application/xml",
|
|
}
|
|
|
|
func normalizeMIMEType(mimetype string) string {
|
|
if mimetype == "" {
|
|
return ""
|
|
}
|
|
|
|
base, _, err := mime.ParseMediaType(mimetype)
|
|
if err == nil {
|
|
return strings.ToLower(base)
|
|
}
|
|
|
|
if idx := strings.IndexByte(mimetype, ';'); idx >= 0 {
|
|
mimetype = mimetype[:idx]
|
|
}
|
|
|
|
return strings.ToLower(strings.TrimSpace(mimetype))
|
|
}
|
|
|
|
func GetExtension(mimetype string) string {
|
|
mimetype = normalizeMIMEType(mimetype)
|
|
if mimetype == "" {
|
|
return ""
|
|
}
|
|
|
|
if ext, ok := commonMimeExtensions[mimetype]; ok {
|
|
return ext
|
|
}
|
|
|
|
exts, _ := mime.ExtensionsByType(mimetype)
|
|
if len(exts) > 0 {
|
|
return exts[0]
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
func GetMIMEType(ext string) string {
|
|
if ext == "" {
|
|
return ""
|
|
}
|
|
|
|
ext = strings.ToLower(strings.TrimSpace(ext))
|
|
if ext != "" && ext[0] != '.' {
|
|
ext = "." + ext
|
|
}
|
|
|
|
if mimetype, ok := commonExtensionMimes[ext]; ok {
|
|
return mimetype
|
|
}
|
|
|
|
return mime.TypeByExtension(ext)
|
|
}
|