+42

README.rst

··· 48 48 from this git repository, run ``FLASK_APP=fhost flask db upgrade``. 49 49 50 50 51 51 + Moderation UI 52 52 + ------------- 53 53 + 54 54 + 0x0 features a TUI program for file moderation. With it, you can view a list 55 55 + of uploaded files, as well as extended information on them. It allows you to 56 56 + take actions like removing files temporarily or permanently, as well as 57 57 + blocking IP addresses and associated files. 58 58 + 59 59 + If a sufficiently recent version of python-mpv with libmpv is present and 60 60 + your terminal supports it, you also get graphical file previews, including 61 61 + video playback. Upstream mpv currently supports sixel graphics, but there is 62 62 + `an open pull request <https://github.com/mpv-player/mpv/pull/11002>`_ that 63 63 + adds support for the `kitty graphics protocol <https://sw.kovidgoyal.net/kitty/graphics-protocol/>`_. 64 64 + For this to work, set the ``MOD_PREVIEW_PROTO`` option in ``instance/config.py``. 65 65 + 66 66 + Requirements: 67 67 + 68 68 + * `Textual <https://textual.textualize.io/>`_ 69 69 + 70 70 + Optional: 71 71 + 72 72 + * `python-mpv <https://github.com/jaseg/python-mpv>`_ 73 73 + (graphical previews) 74 74 + * `PyAV <https://github.com/PyAV-Org/PyAV>`_ 75 75 + (information on multimedia files) 76 76 + * `PyMuPDF <https://github.com/pymupdf/PyMuPDF>`_ 77 77 + (previews and file information for PDF, XPS, EPUB, MOBI and FB2) 78 78 + * `libarchive-c <https://github.com/Changaco/python-libarchive-c>`_ 79 79 + (archive content listing) 80 80 + 81 81 + .. note:: 82 82 + `Mosh <https://mosh.org/>`_ currently does not support sixels or kitty graphics. 83 83 + 84 84 + .. hint:: 85 85 + You may need to set the ``COLORTERM`` environment variable to 86 86 + ``truecolor``. 87 87 + 88 88 + .. tip:: 89 89 + Using compression with SSH (``-C`` option) can significantly 90 90 + reduce the bandwidth requirements for graphics. 91 91 + 92 92 + 51 93 NSFW Detection 52 94 -------------- 53 95

+11

instance/config.example.py

··· 58 58 FHOST_MAX_EXPIRATION = 365 * 24 * 60 * 60 * 1000 59 59 60 60 61 61 + # This should be detected automatically when running behind a reverse proxy, but needs 62 62 + # to be set for URL resolution to work in e.g. the moderation UI. 63 63 + # SERVER_NAME = "example.com" 64 64 + 65 65 + 66 66 + # Specifies which graphics protocol to use for the media previews in the moderation UI. 67 67 + # Requires pympv with libmpv >= 0.36.0 and terminal support. 68 68 + # Available choices are "sixel" and "kitty". 69 69 + # MOD_PREVIEW_PROTO = "sixel" 70 70 + 71 71 + 61 72 # Use the X-SENDFILE header to speed up serving files w/ compatible webservers 62 73 # 63 74 # Some webservers can be configured use the X-Sendfile header to handle sending

+56

mod.css

··· 1 1 + #ftable { 2 2 + width: 1fr; 3 3 + } 4 4 + 5 5 + #infopane { 6 6 + width: 50%; 7 7 + outline-top: hkey $primary; 8 8 + background: $panel; 9 9 + } 10 10 + 11 11 + #finfo { 12 12 + background: $boost; 13 13 + height: 12; 14 14 + width: 1fr; 15 15 + box-sizing: content-box; 16 16 + } 17 17 + 18 18 + #mpv { 19 19 + display: none; 20 20 + height: 20%; 21 21 + width: 1fr; 22 22 + content-align: center middle; 23 23 + } 24 24 + 25 25 + #ftextlog { 26 26 + height: 1fr; 27 27 + width: 1fr; 28 28 + } 29 29 + 30 30 + #filter_container { 31 31 + height: auto; 32 32 + display: none; 33 33 + } 34 34 + 35 35 + #filter_label { 36 36 + content-align: right middle; 37 37 + height: 1fr; 38 38 + width: 20%; 39 39 + margin: 0 1 0 2; 40 40 + } 41 41 + 42 42 + #filter_input { 43 43 + width: 1fr; 44 44 + } 45 45 + 46 46 + Notification { 47 47 + dock: bottom; 48 48 + layer: notification; 49 49 + width: auto; 50 50 + margin: 2 4; 51 51 + padding: 1 2; 52 52 + background: $background; 53 53 + color: $text; 54 54 + height: auto; 55 55 + 56 56 + }

+279

mod.py

··· 1 1 + #!/usr/bin/env python3 2 2 + 3 3 + from itertools import zip_longest 4 4 + from sys import stdout 5 5 + import time 6 6 + 7 7 + from textual.app import App, ComposeResult 8 8 + from textual.widgets import DataTable, Header, Footer, TextLog, Static, Input 9 9 + from textual.containers import Horizontal, Vertical 10 10 + from textual.screen import Screen 11 11 + from textual import log 12 12 + from rich.text import Text 13 13 + from jinja2.filters import do_filesizeformat 14 14 + 15 15 + from fhost import db, File, su, app as fhost_app, in_upload_bl 16 16 + from modui import * 17 17 + 18 18 + fhost_app.app_context().push() 19 19 + 20 20 + class NullptrMod(Screen): 21 21 + BINDINGS = [ 22 22 + ("q", "quit_app", "Quit"), 23 23 + ("f1", "filter(1, 'Lookup name:')", "Lookup name"), 24 24 + ("f2", "filter(2, 'Filter IP address:')", "Filter IP"), 25 25 + ("f3", "filter(3, 'Filter MIME Type:')", "Filter MIME"), 26 26 + ("f4", "filter(4, 'Filter extension:')", "Filter Ext."), 27 27 + ("f5", "refresh", "Refresh"), 28 28 + ("f6", "filter_clear", "Clear filter"), 29 29 + ("r", "remove_file(False)", "Remove file"), 30 30 + ("ctrl+r", "remove_file(True)", "Ban file"), 31 31 + ("p", "ban_ip(False)", "Ban IP"), 32 32 + ("ctrl+p", "ban_ip(True)", "Nuke IP"), 33 33 + ] 34 34 + 35 35 + async def action_quit_app(self): 36 36 + self.mpvw.shutdown() 37 37 + await self.app.action_quit() 38 38 + 39 39 + def action_refresh(self): 40 40 + ftable = self.query_one("#ftable") 41 41 + ftable.watch_query(None, None) 42 42 + 43 43 + def action_filter_clear(self): 44 44 + self.query_one("#filter_container").display = False 45 45 + ftable = self.query_one("#ftable") 46 46 + ftable.focus() 47 47 + ftable.query = ftable.base_query 48 48 + 49 49 + def action_filter(self, fcol: int, label: str): 50 50 + self.query_one("#filter_label").update(label) 51 51 + finput = self.query_one("#filter_input") 52 52 + self.filter_col = fcol 53 53 + self.query_one("#filter_container").display = True 54 54 + finput.focus() 55 55 + self._refresh_layout() 56 56 + 57 57 + if self.current_file: 58 58 + match fcol: 59 59 + case 1: finput.value = "" 60 60 + case 2: finput.value = self.current_file.addr 61 61 + case 3: finput.value = self.current_file.mime 62 62 + case 4: finput.value = self.current_file.ext 63 63 + 64 64 + def on_input_submitted(self, message: Input.Submitted) -> None: 65 65 + self.query_one("#filter_container").display = False 66 66 + ftable = self.query_one("#ftable") 67 67 + ftable.focus() 68 68 + 69 69 + if len(message.value): 70 70 + match self.filter_col: 71 71 + case 1: 72 72 + try: ftable.query = ftable.base_query.filter(File.id == su.debase(message.value)) 73 73 + except ValueError: pass 74 74 + case 2: ftable.query = ftable.base_query.filter(File.addr == message.value) 75 75 + case 3: ftable.query = ftable.base_query.filter(File.mime.like(message.value)) 76 76 + case 4: ftable.query = ftable.base_query.filter(File.ext.like(message.value)) 77 77 + else: 78 78 + ftable.query = ftable.base_query 79 79 + 80 80 + def action_remove_file(self, permanent: bool) -> None: 81 81 + if self.current_file: 82 82 + self.current_file.delete(permanent) 83 83 + db.session.commit() 84 84 + self.mount(Notification(f"{'Banned' if permanent else 'Removed'} file {self.current_file.getname()}")) 85 85 + self.action_refresh() 86 86 + 87 87 + def action_ban_ip(self, nuke: bool) -> None: 88 88 + if self.current_file: 89 89 + if not fhost_app.config["FHOST_UPLOAD_BLACKLIST"]: 90 90 + self.mount(Notification("Failed: FHOST_UPLOAD_BLACKLIST not set!")) 91 91 + return 92 92 + else: 93 93 + if in_upload_bl(self.current_file.addr): 94 94 + txt = f"{self.current_file.addr} is already banned" 95 95 + else: 96 96 + with fhost_app.open_instance_resource(fhost_app.config["FHOST_UPLOAD_BLACKLIST"], "a") as bl: 97 97 + print(self.current_file.addr.lstrip("::ffff:"), file=bl) 98 98 + txt = f"Banned {self.current_file.addr}" 99 99 + 100 100 + if nuke: 101 101 + tsize = 0 102 102 + trm = 0 103 103 + for f in File.query.filter(File.addr == self.current_file.addr): 104 104 + if f.getpath().is_file(): 105 105 + tsize += f.size or f.getpath().stat().st_size 106 106 + trm += 1 107 107 + f.delete(True) 108 108 + db.session.commit() 109 109 + txt += f", removed {trm} {'files' if trm != 1 else 'file'} totaling {jinja2.filters.do_filesizeformat(tsize, True)}" 110 110 + self.mount(Notification(txt)) 111 111 + self._refresh_layout() 112 112 + ftable = self.query_one("#ftable") 113 113 + ftable.watch_query(None, None) 114 114 + 115 115 + def on_update(self) -> None: 116 116 + stdout.write("\033[?25l") 117 117 + stdout.flush() 118 118 + 119 119 + def compose(self) -> ComposeResult: 120 120 + yield Header() 121 121 + yield Horizontal( 122 122 + FileTable(id="ftable", zebra_stripes=True), 123 123 + Vertical( 124 124 + DataTable(id="finfo", show_header=False), 125 125 + MpvWidget(id="mpv"), 126 126 + TextLog(id="ftextlog"), 127 127 + id="infopane")) 128 128 + yield Horizontal(Static("Filter:", id="filter_label"), Input(id="filter_input"), id="filter_container") 129 129 + yield Footer() 130 130 + 131 131 + def on_mount(self) -> None: 132 132 + self.current_file = None 133 133 + 134 134 + self.ftable = self.query_one("#ftable") 135 135 + self.ftable.focus() 136 136 + 137 137 + self.finfo = self.query_one("#finfo") 138 138 + self.finfo.add_columns("key", "value") 139 139 + 140 140 + self.mpvw = self.query_one("#mpv") 141 141 + self.ftlog = self.query_one("#ftextlog") 142 142 + 143 143 + self.mimehandler = mime.MIMEHandler() 144 144 + self.mimehandler.register(mime.MIMECategory.Archive, self.handle_libarchive) 145 145 + self.mimehandler.register(mime.MIMECategory.Text, self.handle_text) 146 146 + self.mimehandler.register(mime.MIMECategory.AV, self.handle_mpv) 147 147 + self.mimehandler.register(mime.MIMECategory.Document, self.handle_mupdf) 148 148 + self.mimehandler.register(mime.MIMECategory.Fallback, self.handle_libarchive) 149 149 + self.mimehandler.register(mime.MIMECategory.Fallback, self.handle_mpv) 150 150 + self.mimehandler.register(mime.MIMECategory.Fallback, self.handle_raw) 151 151 + 152 152 + def handle_libarchive(self, cat): 153 153 + import libarchive 154 154 + with libarchive.file_reader(str(self.current_file.getpath())) as a: 155 155 + self.ftlog.write("\n".join(e.path for e in a)) 156 156 + return True 157 157 + 158 158 + def handle_text(self, cat): 159 159 + with open(self.current_file.getpath(), "r") as sf: 160 160 + data = sf.read(1000000).replace("\033","") 161 161 + self.ftlog.write(data) 162 162 + return True 163 163 + 164 164 + def handle_mupdf(self, cat): 165 165 + import fitz 166 166 + with fitz.open(self.current_file.getpath(), 167 167 + filetype=self.current_file.ext.lstrip(".")) as doc: 168 168 + p = doc.load_page(0) 169 169 + pix = p.get_pixmap(dpi=72) 170 170 + imgdata = pix.tobytes("ppm").hex() 171 171 + 172 172 + self.mpvw.styles.height = "40%" 173 173 + self.mpvw.start_mpv("hex://" + imgdata, 0) 174 174 + 175 175 + self.ftlog.write(Text.from_markup(f"[bold]Pages:[/bold] {doc.page_count}")) 176 176 + self.ftlog.write(Text.from_markup("[bold]Metadata:[/bold]")) 177 177 + for k, v in doc.metadata.items(): 178 178 + self.ftlog.write(Text.from_markup(f" [bold]{k}:[/bold] {v}")) 179 179 + toc = doc.get_toc() 180 180 + if len(toc): 181 181 + self.ftlog.write(Text.from_markup("[bold]TOC:[/bold]")) 182 182 + for lvl, title, page in toc: 183 183 + self.ftlog.write(f"{' ' * lvl} {page}: {title}") 184 184 + return True 185 185 + 186 186 + def handle_mpv(self, cat): 187 187 + if cat == mime.MIMECategory.AV or self.current_file.nsfw_score >= 0: 188 188 + self.mpvw.styles.height = "20%" 189 189 + self.mpvw.start_mpv(str(self.current_file.getpath()), 0) 190 190 + 191 191 + import av 192 192 + with av.open(str(self.current_file.getpath())) as c: 193 193 + self.ftlog.write(Text("Format:", style="bold")) 194 194 + self.ftlog.write(f" {c.format.long_name}") 195 195 + if len(c.metadata): 196 196 + self.ftlog.write(Text("Metadata:", style="bold")) 197 197 + for k, v in c.metadata.items(): 198 198 + self.ftlog.write(f" {k}: {v}") 199 199 + for s in c.streams: 200 200 + self.ftlog.write(Text(f"Stream {s.index}:", style="bold")) 201 201 + self.ftlog.write(f" Type: {s.type}") 202 202 + if s.base_rate: 203 203 + self.ftlog.write(f" Frame rate: {s.base_rate}") 204 204 + if len(s.metadata): 205 205 + self.ftlog.write(Text(" Metadata:", style="bold")) 206 206 + for k, v in s.metadata.items(): 207 207 + self.ftlog.write(f" {k}: {v}") 208 208 + return True 209 209 + return False 210 210 + 211 211 + def handle_raw(self, cat): 212 212 + def hexdump(binf, length): 213 213 + def fmt(s): 214 214 + if isinstance(s, str): 215 215 + c = chr(int(s, 16)) 216 216 + else: 217 217 + c = chr(s) 218 218 + s = c 219 219 + if c.isalpha(): return f"\0[chartreuse1]{s}\0[/chartreuse1]" 220 220 + if c.isdigit(): return f"\0[gold1]{s}\0[/gold1]" 221 221 + if not c.isprintable(): 222 222 + g = "grey50" if c == "\0" else "cadet_blue" 223 223 + return f"\0[{g}]{s if len(s) == 2 else '.'}\0[/{g}]" 224 224 + return s 225 225 + return Text.from_markup("\n".join(f"{' '.join(map(fmt, map(''.join, zip(*[iter(c.hex())] * 2))))}" 226 226 + f"{' ' * (16 - len(c))}" 227 227 + f" {''.join(map(fmt, c))}" 228 228 + for c in map(lambda x: bytes([n for n in x if n != None]), 229 229 + zip_longest(*[iter(binf.read(min(length, 16 * 10)))] * 16)))) 230 230 + 231 231 + with open(self.current_file.getpath(), "rb") as binf: 232 232 + self.ftlog.write(hexdump(binf, self.current_file.size)) 233 233 + if self.current_file.size > 16*10*2: 234 234 + binf.seek(self.current_file.size-16*10) 235 235 + self.ftlog.write(" [...] ".center(64, '─')) 236 236 + self.ftlog.write(hexdump(binf, self.current_file.size - binf.tell())) 237 237 + 238 238 + return True 239 239 + 240 240 + def on_file_table_selected(self, message: FileTable.Selected) -> None: 241 241 + f = message.file 242 242 + self.current_file = f 243 243 + self.finfo.clear() 244 244 + self.finfo.add_rows([ 245 245 + ("ID:", str(f.id)), 246 246 + ("File name:", f.getname()), 247 247 + ("URL:", f.geturl() if fhost_app.config["SERVER_NAME"] else "⚠ Set SERVER_NAME in config.py to display"), 248 248 + ("File size:", do_filesizeformat(f.size, True)), 249 249 + ("MIME type:", f.mime), 250 250 + ("SHA256 checksum:", f.sha256), 251 251 + ("Uploaded by:", Text(f.addr)), 252 252 + ("Management token:", f.mgmt_token), 253 253 + ("Secret:", f.secret), 254 254 + ("Is NSFW:", ("Yes" if f.is_nsfw else "No") + f" (Score: {f.nsfw_score:0.4f})"), 255 255 + ("Is banned:", "Yes" if f.removed else "No"), 256 256 + ("Expires:", time.strftime("%Y-%m-%d %H:%M:%S", time.gmtime(File.get_expiration(f.expiration, f.size)/1000))) 257 257 + ]) 258 258 + 259 259 + self.mpvw.stop_mpv(True) 260 260 + self.ftlog.remove() 261 261 + self.query_one("#infopane").mount(TextLog(id="ftextlog")) 262 262 + self.ftlog = self.query_one("#ftextlog") 263 263 + 264 264 + if f.getpath().is_file(): 265 265 + self.mimehandler.handle(f.mime, f.ext) 266 266 + self.ftlog.scroll_home(animate=False) 267 267 + 268 268 + class NullptrModApp(App): 269 269 + CSS_PATH = "mod.css" 270 270 + 271 271 + def on_mount(self) -> None: 272 272 + self.title = "0x0 File Moderation Interface" 273 273 + self.main_screen = NullptrMod() 274 274 + self.install_screen(self.main_screen, name="main") 275 275 + self.push_screen("main") 276 276 + 277 277 + if __name__ == "__main__": 278 278 + app = NullptrModApp() 279 279 + app.run()

+3

modui/__init__.py

··· 1 1 + from .filetable import FileTable 2 2 + from .notification import Notification 3 3 + from .mpvwidget import MpvWidget

+72

modui/filetable.py

··· 1 1 + from textual.widgets import DataTable, Static 2 2 + from textual.reactive import Reactive 3 3 + from textual.message import Message, MessageTarget 4 4 + from textual import events, log 5 5 + from jinja2.filters import do_filesizeformat 6 6 + 7 7 + from fhost import File 8 8 + from modui import mime 9 9 + 10 10 + class FileTable(DataTable): 11 11 + query = Reactive(None) 12 12 + order_col = Reactive(0) 13 13 + order_desc = Reactive(True) 14 14 + limit = 10000 15 15 + colmap = [File.id, File.removed, File.nsfw_score, None, File.ext, File.size, File.mime] 16 16 + 17 17 + def __init__(self, **kwargs): 18 18 + super().__init__(**kwargs) 19 19 + self.add_columns("#", "☣️", "🔞", "📂", "name", "size", "mime") 20 20 + self.base_query = File.query.filter(File.size != None) 21 21 + self.query = self.base_query 22 22 + 23 23 + class Selected(Message): 24 24 + def __init__(self, sender: MessageTarget, f: File) -> None: 25 25 + self.file = f 26 26 + super().__init__(sender) 27 27 + 28 28 + def watch_order_col(self, old, value) -> None: 29 29 + self.watch_query(None, None) 30 30 + 31 31 + def watch_order_desc(self, old, value) -> None: 32 32 + self.watch_query(None, None) 33 33 + 34 34 + def watch_query(self, old, value) -> None: 35 35 + def fmt_file(f: File) -> tuple: 36 36 + return ( 37 37 + str(f.id), 38 38 + "🔴" if f.removed else " ", 39 39 + "🚩" if f.is_nsfw else " ", 40 40 + "👻" if not f.getpath().is_file() else " ", 41 41 + f.getname(), 42 42 + do_filesizeformat(f.size, True), 43 43 + f"{mime.mimemoji.get(f.mime.split('/')[0], mime.mimemoji.get(f.mime)) or ' '} " + f.mime, 44 44 + ) 45 45 + 46 46 + if (self.query): 47 47 + self.clear() 48 48 + order = FileTable.colmap[self.order_col] 49 49 + q = self.query 50 50 + if order: q = q.order_by(order.desc() if self.order_desc else order, File.id) 51 51 + self.add_rows(map(fmt_file, q.limit(self.limit))) 52 52 + 53 53 + def _scroll_cursor_in_to_view(self, animate: bool = False) -> None: 54 54 + region = self._get_cell_region(self.cursor_row, 0) 55 55 + spacing = self._get_cell_border() 56 56 + self.scroll_to_region(region, animate=animate, spacing=spacing) 57 57 + 58 58 + async def watch_cursor_cell(self, old, value) -> None: 59 59 + super().watch_cursor_cell(old, value) 60 60 + if value[0] < len(self.data) and value[0] >= 0: 61 61 + f = File.query.get(int(self.data[value[0]][0])) 62 62 + await self.emit(self.Selected(self, f)) 63 63 + 64 64 + def on_click(self, event: events.Click) -> None: 65 65 + super().on_click(event) 66 66 + meta = self.get_style_at(event.x, event.y).meta 67 67 + if meta: 68 68 + if meta["row"] == -1: 69 69 + qi = FileTable.colmap[meta["column"]] 70 70 + if meta["column"] == self.order_col: 71 71 + self.order_desc = not self.order_desc 72 72 + self.order_col = meta["column"]

+122

modui/mime.py

··· 1 1 + from enum import Enum 2 2 + from textual import log 3 3 + 4 4 + mimemoji = { 5 5 + "audio" : "🔈", 6 6 + "video" : "🎞", 7 7 + "text" : "📄", 8 8 + "image" : "🖼", 9 9 + "application/zip" : "🗜️", 10 10 + "application/x-zip-compressed" : "🗜️", 11 11 + "application/x-tar" : "🗄", 12 12 + "application/x-cpio" : "🗄", 13 13 + "application/x-xz" : "🗜️", 14 14 + "application/x-7z-compressed" : "🗜️", 15 15 + "application/gzip" : "🗜️", 16 16 + "application/zstd" : "🗜️", 17 17 + "application/x-rar" : "🗜️", 18 18 + "application/x-rar-compressed" : "🗜️", 19 19 + "application/vnd.ms-cab-compressed" : "🗜️", 20 20 + "application/x-bzip2" : "🗜️", 21 21 + "application/x-lzip" : "🗜️", 22 22 + "application/x-iso9660-image" : "💿", 23 23 + "application/pdf" : "📕", 24 24 + "application/epub+zip" : "📕", 25 25 + "application/mxf" : "🎞", 26 26 + "application/vnd.android.package-archive" : "📦", 27 27 + "application/vnd.debian.binary-package" : "📦", 28 28 + "application/x-rpm" : "📦", 29 29 + "application/x-dosexec" : "⚙", 30 30 + "application/x-execuftable" : "⚙", 31 31 + "application/x-sharedlib" : "⚙", 32 32 + "application/java-archive" : "☕", 33 33 + "application/x-qemu-disk" : "🖴", 34 34 + "application/pgp-encrypted" : "🔏", 35 35 + } 36 36 + 37 37 + MIMECategory = Enum("MIMECategory", 38 38 + ["Archive", "Text", "AV", "Document", "Fallback"] 39 39 + ) 40 40 + 41 41 + class MIMEHandler: 42 42 + def __init__(self): 43 43 + self.handlers = { 44 44 + MIMECategory.Archive : [[ 45 45 + "application/zip", 46 46 + "application/x-zip-compressed", 47 47 + "application/x-tar", 48 48 + "application/x-cpio", 49 49 + "application/x-xz", 50 50 + "application/x-7z-compressed", 51 51 + "application/gzip", 52 52 + "application/zstd", 53 53 + "application/x-rar", 54 54 + "application/x-rar-compressed", 55 55 + "application/vnd.ms-cab-compressed", 56 56 + "application/x-bzip2", 57 57 + "application/x-lzip", 58 58 + "application/x-iso9660-image", 59 59 + "application/vnd.android.package-archive", 60 60 + "application/vnd.debian.binary-package", 61 61 + "application/x-rpm", 62 62 + "application/java-archive", 63 63 + "application/vnd.openxmlformats" 64 64 + ], []], 65 65 + MIMECategory.Text : [["text"], []], 66 66 + MIMECategory.AV : [[ 67 67 + "audio", "video", "image", 68 68 + "application/mxf" 69 69 + ], []], 70 70 + MIMECategory.Document : [[ 71 71 + "application/pdf", 72 72 + "application/epub", 73 73 + "application/x-mobipocket-ebook", 74 74 + ], []], 75 75 + MIMECategory.Fallback : [[], []] 76 76 + } 77 77 + 78 78 + self.exceptions = { 79 79 + MIMECategory.Archive : { 80 80 + ".cbz" : MIMECategory.Document, 81 81 + ".xps" : MIMECategory.Document, 82 82 + ".epub" : MIMECategory.Document, 83 83 + }, 84 84 + MIMECategory.Text : { 85 85 + ".fb2" : MIMECategory.Document, 86 86 + } 87 87 + } 88 88 + 89 89 + def register(self, category, handler): 90 90 + self.handlers[category][1].append(handler) 91 91 + 92 92 + def handle(self, mime, ext): 93 93 + def getcat(s): 94 94 + cat = MIMECategory.Fallback 95 95 + for k, v in self.handlers.items(): 96 96 + s = s.split(";")[0] 97 97 + if s in v[0] or s.split("/")[0] in v[0]: 98 98 + cat = k 99 99 + break 100 100 + 101 101 + for x in v[0]: 102 102 + if s.startswith(x): 103 103 + cat = k 104 104 + break 105 105 + 106 106 + if cat in self.exceptions: 107 107 + cat = self.exceptions[cat].get(ext) or cat 108 108 + 109 109 + return cat 110 110 + 111 111 + cat = getcat(mime) 112 112 + for handler in self.handlers[cat][1]: 113 113 + try: 114 114 + if handler(cat): return 115 115 + except: pass 116 116 + 117 117 + for handler in self.handlers[MIMECategory.Fallback][1]: 118 118 + try: 119 119 + if handler(None): return 120 120 + except: pass 121 121 + 122 122 + raise RuntimeError(f"Unhandled MIME type category: {cat}")

+88

modui/mpvwidget.py

··· 1 1 + import time 2 2 + import fcntl, struct, termios 3 3 + from sys import stdout 4 4 + 5 5 + from textual import events, log 6 6 + from textual.widgets import Static 7 7 + 8 8 + from fhost import app as fhost_app 9 9 + 10 10 + class MpvWidget(Static): 11 11 + def __init__(self, **kwargs): 12 12 + super().__init__(**kwargs) 13 13 + 14 14 + self.mpv = None 15 15 + self.vo = fhost_app.config.get("MOD_PREVIEW_PROTO") 16 16 + 17 17 + if not self.vo in ["sixel", "kitty"]: 18 18 + self.update("⚠ Previews not enabled. \n\nSet MOD_PREVIEW_PROTO to 'sixel' or 'kitty' in config.py,\nwhichever is supported by your terminal.") 19 19 + else: 20 20 + try: 21 21 + import mpv 22 22 + self.mpv = mpv.MPV() 23 23 + self.mpv.profile = "sw-fast" 24 24 + self.mpv["vo"] = self.vo 25 25 + self.mpv[f"vo-{self.vo}-config-clear"] = False 26 26 + self.mpv[f"vo-{self.vo}-alt-screen"] = False 27 27 + self.mpv[f"vo-sixel-buffered"] = True 28 28 + self.mpv["audio"] = False 29 29 + self.mpv["loop-file"] = "inf" 30 30 + self.mpv["image-display-duration"] = 0.5 if self.vo == "sixel" else "inf" 31 31 + except Exception as e: 32 32 + self.mpv = None 33 33 + self.update(f"⚠ Previews require python-mpv with libmpv 0.36.0 or later \n\nError was:\n{type(e).__name__}: {e}") 34 34 + 35 35 + def start_mpv(self, f: str|None = None, pos: float|str|None = None) -> None: 36 36 + self.display = True 37 37 + self.screen._refresh_layout() 38 38 + 39 39 + if self.mpv: 40 40 + if self.content_region.x: 41 41 + r, c, w, h = struct.unpack('hhhh', fcntl.ioctl(0, termios.TIOCGWINSZ, '12345678')) 42 42 + width = int((w / c) * self.content_region.width) 43 43 + height = int((h / r) * (self.content_region.height + (1 if self.vo == "sixel" else 0))) 44 44 + self.mpv[f"vo-{self.vo}-left"] = self.content_region.x + 1 45 45 + self.mpv[f"vo-{self.vo}-top"] = self.content_region.y + 1 46 46 + self.mpv[f"vo-{self.vo}-rows"] = self.content_region.height + (1 if self.vo == "sixel" else 0) 47 47 + self.mpv[f"vo-{self.vo}-cols"] = self.content_region.width 48 48 + self.mpv[f"vo-{self.vo}-width"] = width 49 49 + self.mpv[f"vo-{self.vo}-height"] = height 50 50 + 51 51 + if pos != None: 52 52 + self.mpv["start"] = pos 53 53 + 54 54 + if f: 55 55 + self.mpv.loadfile(f) 56 56 + else: 57 57 + self.mpv.playlist_play_index(0) 58 58 + 59 59 + def stop_mpv(self, wait: bool = False) -> None: 60 60 + if self.mpv: 61 61 + if not self.mpv.idle_active: 62 62 + self.mpv.stop(True) 63 63 + if wait: 64 64 + time.sleep(0.1) 65 65 + self.clear_mpv() 66 66 + self.display = False 67 67 + 68 68 + def on_resize(self, size) -> None: 69 69 + if self.mpv: 70 70 + if not self.mpv.idle_active: 71 71 + t = self.mpv.time_pos 72 72 + self.stop_mpv() 73 73 + if t: 74 74 + self.mpv["start"] = t 75 75 + self.start_mpv() 76 76 + 77 77 + def clear_mpv(self) -> None: 78 78 + if self.vo == "kitty": 79 79 + stdout.write("\033_Ga=d;\033\\") 80 80 + stdout.flush() 81 81 + 82 82 + def shutdown(self) -> None: 83 83 + if self.mpv: 84 84 + self.mpv.stop() 85 85 + del self.mpv 86 86 + if self.vo == "kitty": 87 87 + stdout.write("\033_Ga=d;\033\\\033[?25l") 88 88 + stdout.flush()

+8

modui/notification.py

··· 1 1 + from textual.widgets import Static 2 2 + 3 3 + class Notification(Static): 4 4 + def on_mount(self) -> None: 5 5 + self.set_timer(3, self.remove) 6 6 + 7 7 + def on_click(self) -> None: 8 8 + self.remove()