fix(ui): register other supported tar types to render previews

Co-authored-by: Sola-ris <190788035+Sola-ris@users.noreply.github.com>

Author: CyanVoxel

docs/preview-support.md

@@ -105,12 +105,17 @@ Preview support for office documents or well-known project file formats varies b
 
 Archive thumbnails will display the first image from the archive within the Preview Panel.
 
-| Filetype | Extensions                        |
-| -------- | --------------------------------- |
-| 7-Zip    | `.7z`, `.s7z`                     |
-| RAR      | `.rar`                            |
-| Tar      | `.tar`, `.tar.gz`, `.taz`, `.tgz` |
-| Zip      | `.zip`                            |
+| Filetype        | Extensions                                             |
+| --------------- | ------------------------------------------------------ |
+| 7-Zip           | `.7z`, `.s7z`                                          |
+| RAR             | `.rar`                                                 |
+| Tar             | `.tar`                                                 |
+| Tar (bzip2)     | `.tar.bz2`, `.tar.bz`, `.tbz2`, `.tbz`, `.tb2`, `.tz2` |
+| Tar (gzip)      | `.tar.gz`, `.tgz`, `.taz`                              |
+| Tar (LZMA)      | `.tar.lzma`, `.tlz`                                    |
+| Tar (XZ)        | `.tar.xz`, `.txz`                                      |
+| Tar (Zstandard) | `.tar.zst`, `.tzst`                                    |
+| Zip             | `.zip`                                                 |
 
 ### :material-book: eBooks
 

src/tagstudio/previews/renderers/archive.py

@@ -37,7 +37,12 @@ def register_types(cls) -> None:
         MediaTypes.register("archive", ".s7z", RENDER)
         MediaTypes.register("archive", ".tar", RENDER)
         MediaTypes.register("archive", ".zip", RENDER)
+        MediaTypes.register("archive", ".tar.bz", RENDER)
+        MediaTypes.register("archive", ".tar.bz2", RENDER)
         MediaTypes.register("archive", ".tar.gz", RENDER)
+        MediaTypes.register("archive", ".tar.lzma", RENDER)
+        MediaTypes.register("archive", ".tar.xz", RENDER)
+        MediaTypes.register("archive", ".tar.zst", RENDER)
 
     @override
     @classmethod
@@ -105,7 +110,16 @@ def open_archive(filepath: Path) -> Archive:
         archiver = SevenZipFile
     elif exts & {".cbr", ".rar"}:
         archiver = rarfile.RarFile
-    elif exts & {".cbt", ".tar", ".tar.gz"}:
+    elif exts & {
+        ".cbt",
+        ".tar",
+        ".tar.bz",
+        ".tar.bz2",
+        ".tar.gz",
+        ".tar.lzma",
+        ".tar.xz",
+        ".tar.zst",
+    }:
         archiver = TarFile
     return archiver(filepath, "r")
 

tests/previews/test_file_renderer.py

@@ -4,7 +4,9 @@
 # pyright: reportPrivateUsage=false
 
 
-from compression import gzip, lzma
+import tarfile
+from collections.abc import Callable
+from compression import bz2, gzip, lzma, zstd
 from io import BytesIO
 from pathlib import Path
 
@@ -13,6 +15,7 @@
 
 from tagstudio.core.enums import Theme
 from tagstudio.previews.file_renderer import FileRenderer
+from tagstudio.previews.renderers.archive import ArchivePreview
 
 
 def _example_png_bytes() -> bytes:
@@ -37,6 +40,44 @@ def test_render_decompressed(tmp_path: Path, filename: str):
     assert image.size == (40, 20)
 
 
+def _example_tar_bytes() -> bytes:
+    data = BytesIO()
+    with tarfile.open(fileobj=data, mode="w") as tar:
+        png = _example_png_bytes()
+        info = tarfile.TarInfo("images/photo.png")
+        info.size = len(png)
+        tar.addfile(info, BytesIO(png))
+    return data.getvalue()
+
+
+@pytest.mark.parametrize(
+    ["filename", "compress"],
+    [
+        ("bundle.tar.bz2", bz2.compress),
+        ("bundle.tbz2", bz2.compress),
+        ("bundle.tar.lzma", lambda data: lzma.compress(data, format=lzma.FORMAT_ALONE)),
+        ("bundle.tlz", lambda data: lzma.compress(data, format=lzma.FORMAT_ALONE)),
+        ("my.cool.bundle.tar.xz", lzma.compress),
+        ("BUNDLE.TXZ", lzma.compress),
+        ("bundle.tar.zst", zstd.compress),
+        ("bundle.tzst", zstd.compress),
+    ],
+)
+def test_render_compressed_tarball(
+    tmp_path: Path, filename: str, compress: Callable[[bytes], bytes]
+):
+    source = tmp_path / filename
+    source.write_bytes(compress(_example_tar_bytes()))
+
+    assert FileRenderer._find_preview(source) is ArchivePreview
+    image = ArchivePreview.render(
+        source, is_small=True, theme=Theme.DARK, size=(128, 128), dpi_scale=1.0
+    )
+
+    assert image is not None
+    assert image.size == (40, 20)
+
+
 def test_render_decompressed_without_inner_extension(tmp_path: Path):
     source = tmp_path / "log.gz"
     source.write_bytes(gzip.compress(b"plain text"))