mirror of
https://github.com/Benexl/FastAnime.git
synced 2026-01-07 02:03:49 -08:00
feat: stabilize the interactive workflow
This commit is contained in:
87
fastanime/cli/utils/image.py
Normal file
87
fastanime/cli/utils/image.py
Normal file
@@ -0,0 +1,87 @@
|
||||
# fastanime/cli/utils/image.py
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
import shutil
|
||||
import subprocess
|
||||
from typing import Optional
|
||||
|
||||
import click
|
||||
import httpx
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
def render_image(url: str, capture: bool = False, size: str = "30x30") -> Optional[str]:
|
||||
"""
|
||||
Renders an image from a URL in the terminal using icat or chafa.
|
||||
|
||||
This function automatically detects the best available tool.
|
||||
|
||||
Args:
|
||||
url: The URL of the image to render.
|
||||
capture: If True, returns the terminal-formatted image as a string
|
||||
instead of printing it. Defaults to False.
|
||||
size: The size parameter to pass to the rendering tool (e.g., "WxH").
|
||||
|
||||
Returns:
|
||||
If capture is True, returns the image data as a string.
|
||||
If capture is False, prints directly to the terminal and returns None.
|
||||
Returns None on any failure.
|
||||
"""
|
||||
# --- Common subprocess arguments ---
|
||||
subprocess_kwargs = {
|
||||
"check": False, # We will handle errors manually
|
||||
"capture_output": capture,
|
||||
"text": capture, # Decode stdout/stderr as text if capturing
|
||||
}
|
||||
|
||||
# --- Try icat (Kitty terminal) first ---
|
||||
if icat_executable := shutil.which("icat"):
|
||||
process = subprocess.run(
|
||||
[icat_executable, "--align", "left", url], **subprocess_kwargs
|
||||
)
|
||||
if process.returncode == 0:
|
||||
return process.stdout if capture else None
|
||||
logger.warning(f"icat failed for URL {url} with code {process.returncode}")
|
||||
|
||||
# --- Fallback to chafa ---
|
||||
if chafa_executable := shutil.which("chafa"):
|
||||
try:
|
||||
# Chafa requires downloading the image data first
|
||||
with httpx.Client() as client:
|
||||
response = client.get(url, follow_redirects=True, timeout=20)
|
||||
response.raise_for_status()
|
||||
img_bytes = response.content
|
||||
|
||||
# Add stdin input to the subprocess arguments
|
||||
subprocess_kwargs["input"] = img_bytes
|
||||
|
||||
process = subprocess.run(
|
||||
[chafa_executable, f"--size={size}", "-"], **subprocess_kwargs
|
||||
)
|
||||
if process.returncode == 0:
|
||||
return process.stdout if capture else None
|
||||
logger.warning(f"chafa failed for URL {url} with code {process.returncode}")
|
||||
|
||||
except httpx.HTTPStatusError as e:
|
||||
logger.error(
|
||||
f"HTTP error fetching image for chafa: {e.response.status_code}"
|
||||
)
|
||||
click.echo(
|
||||
f"[dim]Error fetching image: {e.response.status_code}[/dim]", err=True
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"An exception occurred while running chafa: {e}")
|
||||
|
||||
return None
|
||||
|
||||
# --- Final fallback if no tool is found ---
|
||||
if not capture:
|
||||
# Only show this message if the user expected to see something.
|
||||
click.echo(
|
||||
"[dim](Image preview skipped: icat or chafa not found)[/dim]", err=True
|
||||
)
|
||||
|
||||
return None
|
||||
Reference in New Issue
Block a user