mirror of
https://git.0x0.st/mia/0x0.git
synced 2024-11-13 21:58:57 -05:00
126 lines
4 KiB
Python
126 lines
4 KiB
Python
from enum import Enum
|
|
from textual import log
|
|
|
|
mimemoji = {
|
|
"audio" : "🔈",
|
|
"video" : "🎞",
|
|
"text" : "📄",
|
|
"image" : "🖼",
|
|
"application/zip" : "🗜️",
|
|
"application/x-zip-compressed" : "🗜️",
|
|
"application/x-tar" : "🗄",
|
|
"application/x-cpio" : "🗄",
|
|
"application/x-xz" : "🗜️",
|
|
"application/x-7z-compressed" : "🗜️",
|
|
"application/gzip" : "🗜️",
|
|
"application/zstd" : "🗜️",
|
|
"application/x-rar" : "🗜️",
|
|
"application/x-rar-compressed" : "🗜️",
|
|
"application/vnd.ms-cab-compressed" : "🗜️",
|
|
"application/x-bzip2" : "🗜️",
|
|
"application/x-lzip" : "🗜️",
|
|
"application/x-iso9660-image" : "💿",
|
|
"application/pdf" : "📕",
|
|
"application/epub+zip" : "📕",
|
|
"application/mxf" : "🎞",
|
|
"application/vnd.android.package-archive" : "📦",
|
|
"application/vnd.debian.binary-package" : "📦",
|
|
"application/x-rpm" : "📦",
|
|
"application/x-dosexec" : "⚙",
|
|
"application/x-execuftable" : "⚙",
|
|
"application/x-sharedlib" : "⚙",
|
|
"application/java-archive" : "☕",
|
|
"application/x-qemu-disk" : "🖴",
|
|
"application/pgp-encrypted" : "🔏",
|
|
}
|
|
|
|
MIMECategory = Enum("MIMECategory",
|
|
["Archive", "Text", "AV", "Document", "Fallback"]
|
|
)
|
|
|
|
class MIMEHandler:
|
|
def __init__(self):
|
|
self.handlers = {
|
|
MIMECategory.Archive : [[
|
|
"application/zip",
|
|
"application/x-zip-compressed",
|
|
"application/x-tar",
|
|
"application/x-cpio",
|
|
"application/x-xz",
|
|
"application/x-7z-compressed",
|
|
"application/gzip",
|
|
"application/zstd",
|
|
"application/x-rar",
|
|
"application/x-rar-compressed",
|
|
"application/vnd.ms-cab-compressed",
|
|
"application/x-bzip2",
|
|
"application/x-lzip",
|
|
"application/x-iso9660-image",
|
|
"application/vnd.android.package-archive",
|
|
"application/vnd.debian.binary-package",
|
|
"application/x-rpm",
|
|
"application/java-archive",
|
|
"application/vnd.openxmlformats"
|
|
], []],
|
|
MIMECategory.Text : [[
|
|
"text",
|
|
"application/json",
|
|
"application/xml",
|
|
], []],
|
|
MIMECategory.AV : [[
|
|
"audio", "video", "image",
|
|
"application/mxf"
|
|
], []],
|
|
MIMECategory.Document : [[
|
|
"application/pdf",
|
|
"application/epub",
|
|
"application/x-mobipocket-ebook",
|
|
], []],
|
|
MIMECategory.Fallback : [[], []]
|
|
}
|
|
|
|
self.exceptions = {
|
|
MIMECategory.Archive : {
|
|
".cbz" : MIMECategory.Document,
|
|
".xps" : MIMECategory.Document,
|
|
".epub" : MIMECategory.Document,
|
|
},
|
|
MIMECategory.Text : {
|
|
".fb2" : MIMECategory.Document,
|
|
}
|
|
}
|
|
|
|
def register(self, category, handler):
|
|
self.handlers[category][1].append(handler)
|
|
|
|
def handle(self, mime, ext):
|
|
def getcat(s):
|
|
cat = MIMECategory.Fallback
|
|
for k, v in self.handlers.items():
|
|
s = s.split(";")[0]
|
|
if s in v[0] or s.split("/")[0] in v[0]:
|
|
cat = k
|
|
break
|
|
|
|
for x in v[0]:
|
|
if s.startswith(x):
|
|
cat = k
|
|
break
|
|
|
|
if cat in self.exceptions:
|
|
cat = self.exceptions[cat].get(ext) or cat
|
|
|
|
return cat
|
|
|
|
cat = getcat(mime)
|
|
for handler in self.handlers[cat][1]:
|
|
try:
|
|
if handler(cat): return
|
|
except: pass
|
|
|
|
for handler in self.handlers[MIMECategory.Fallback][1]:
|
|
try:
|
|
if handler(None): return
|
|
except: pass
|
|
|
|
raise RuntimeError(f"Unhandled MIME type category: {cat}")
|