refactor: media types rework, dynamically load preview renderers

REUSE.toml

@@ -55,3 +55,14 @@ path = [
 ]
 SPDX-FileCopyrightText = "(c) github:google/material-design-icons Contributors"
 SPDX-License-Identifier = "Apache-2.0"
+
+[[annotations]]
+path = ["src/tagstudio/resources/qt/fonts/Oxanium-Bold.ttf"]
+SPDX-FileCopyrightText = "(c) 2019 The Oxanium Project Authors (https://github.com/sevmeyer/oxanium)"
+SPDX-License-Identifier = "OFL-1.1"
+
+
+[[annotations]]
+path = ["src/tagstudio/resources/fonts/JetBrainsMono/**"]
+SPDX-FileCopyrightText = "(c) 2020 The JetBrains Mono Project Authors (https://github.com/JetBrains/JetBrainsMono)"
+SPDX-License-Identifier = "OFL-1.1"

pyproject.toml

@@ -21,15 +21,18 @@ dependencies = [
     "mutagen~=1.47",
     "numpy~=2.2",
     "opencv_python~=4.11",
-    "Pillow>=10.2,<12",
     "pillow-heif~=1.5.0",
     "pillow-jxl-plugin~=1.3",
+    "Pillow>=10.2,<12",
     "py7zr~=1.1.3",
     "pydantic~=2.10",
     "pydub~=0.25",
+    "Pygments~=2.21",
     "PySide6==6.11.2",
     "rarfile==4.2",
     "rawpy~=0.27",
+    "requests~=2.31.0",
+    "semver~=3.0.4",
     "Send2Trash>=1.8,<3",
     "SQLAlchemy~=2.0",
     "srctools~=2.6",
@@ -38,8 +41,6 @@ dependencies = [
     "typing_extensions~=4.13",
     "ujson~=5.10",
     "wcmatch==10.*",
-    "requests~=2.31.0",
-    "semver~=3.0.4",
 ]
 
 [project.gui-scripts]

src/tagstudio/core/driver.py

@@ -10,6 +10,7 @@
 from tagstudio.core.constants import TS_FOLDER_NAME
 from tagstudio.core.enums import AppCacheItems
 from tagstudio.core.library.alchemy.library import LibraryStatus
+from tagstudio.core.query_lang.file_groups import register_types
 from tagstudio.qt.app_settings import AppSettings
 
 logger = structlog.get_logger(__name__)
@@ -21,6 +22,8 @@ class DriverMixin:
     # TODO: AppSettings is Qt-specific and should not be in a base driver class.
     settings: AppSettings
 
+    register_types()  # Register all filetypes for the SEARCH context.
+
     def evaluate_path(self, open_path: str | None) -> LibraryStatus:
         """Check if the path of library is valid."""
         library_path: Path | None = None

src/tagstudio/core/enums.py

@@ -31,7 +31,7 @@ class TagClickActionOption(enum.IntEnum):
     DEFAULT = OPEN_EDIT
 
 
-class Theme(enum.StrEnum):
+class ThemePalette(enum.StrEnum):
     COLOR_BG_DARK = "#65000000"
     COLOR_BG_LIGHT = "#22000000"
     COLOR_DARK_LABEL = "#DD000000"
@@ -44,6 +44,13 @@ class Theme(enum.StrEnum):
     COLOR_FORBIDDEN_BG = "#65440D12"
 
 
+class Theme(enum.IntEnum):
+    DARK = 0
+    LIGHT = 1
+    SYSTEM = 2
+    DEFAULT = SYSTEM
+
+
 class OpenStatus(enum.IntEnum):
     NOT_FOUND = 0
     SUCCESS = 1

src/tagstudio/core/library/alchemy/visitors.py

@@ -13,7 +13,7 @@
 from tagstudio.core.library.alchemy.constants import TAG_CHILDREN_ID_QUERY
 from tagstudio.core.library.alchemy.joins import TagEntry
 from tagstudio.core.library.alchemy.models import Entry, Tag, TagAlias
-from tagstudio.core.media_types import FILETYPE_EQUIVALENTS, MediaCategories
+from tagstudio.core.media_types import MediaTypeGroup, MediaTypes
 from tagstudio.core.query_lang.ast import (
     AST,
     ANDList,
@@ -24,6 +24,7 @@
     ORList,
     Property,
 )
+from tagstudio.core.query_lang.file_groups import SEARCH
 
 # Only import for type checking/autocompletion, will not be imported at runtime.
 if TYPE_CHECKING:
@@ -34,13 +35,6 @@
 logger = structlog.get_logger(__name__)
 
 
-def get_filetype_equivalency_list(item: str) -> list[str] | set[str]:
-    for s in FILETYPE_EQUIVALENTS:
-        if item in s:
-            return s
-    return [item]
-
-
 class SQLBoolExpressionBuilder(BaseVisitor[ColumnElement[bool]]):
     def __init__(self, lib: Library) -> None:
         super().__init__()
@@ -95,15 +89,27 @@ def visit_constraint(self, node: Constraint) -> ColumnElement[bool]:
                 )
                 return Entry.path.regexp_match(re.escape(node.value))
         elif node.type == ConstraintType.MediaType:
-            extensions: set[str] = set[str]()
-            for media_cat in MediaCategories.ALL_CATEGORIES:
-                if node.value == media_cat.name:
-                    extensions = extensions | media_cat.extensions
-                    break
+            key = (
+                MediaTypes.get_group_key_from_name(
+                    node.value, case_sensitive=False, ignore_whitespace=True
+                )
+                or node.value
+            )
+
+            media_type: MediaTypeGroup | None = getattr(MediaTypes, key, None)
+            extensions: set[str] = (
+                media_type.context_sets.get(SEARCH, set()) if media_type else set()
+            )
             return Entry.suffix.in_(map(lambda x: x.replace(".", ""), extensions))
+
         elif node.type == ConstraintType.FileType:
+            # NOTE: Entries store their suffix without a leading dot, the MediaTypes system includes
+            # the leading dot (if any), and this search system should take in either variant.
             return or_(
-                *[Entry.suffix.ilike(ft) for ft in get_filetype_equivalency_list(node.value)]
+                *[
+                    Entry.suffix.ilike(ft.removeprefix("."))
+                    for ft in MediaTypes.get_equivalent_exts(f".{node.value.removeprefix('.')}")
+                ]
             )
         elif node.type == ConstraintType.Special:  # noqa: SIM102 unnecessary once there is a second special constraint
             if node.value.lower() == "untagged":

src/tagstudio/core/library/refresh.py

@@ -105,7 +105,14 @@ def __get_dir_list(self, library_dir: Path, ignore_patterns: list[str]) -> list[
                 shell=True,
                 encoding="UTF-8",
             )
-            compiled_ignore_path.unlink()
+            try:
+                compiled_ignore_path.unlink()
+            except Exception as e:
+                logger.error(
+                    "[Refresh] Could not remove compiled ignore path",
+                    path=compiled_ignore_path,
+                    error=e,
+                )
 
             if result.stderr:
                 logger.error(result.stderr)