mirror of
https://github.com/Benexl/FastAnime.git
synced 2025-12-13 00:00:01 -08:00
feat(cli): include the fzf preview script inside the project for convinience
This commit is contained in:
Benex254
@@ -1,111 +1,23 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import shutil
|
||||
|
||||
from rich import print
|
||||
from rich.prompt import Prompt
|
||||
|
||||
from ... import APP_CACHE_DIR, USER_CONFIG_PATH
|
||||
from ... import USER_CONFIG_PATH
|
||||
from ...libs.anilist.anilist import AniList
|
||||
from ...libs.anilist.anilist_data_schema import AnilistBaseMediaDataSchema
|
||||
from ...libs.anime_provider.allanime.api import anime_provider
|
||||
from ...libs.anime_provider.types import Anime, SearchResult, Server
|
||||
from ...libs.fzf import fzf
|
||||
from ...Utility import anilist_data_helper
|
||||
from ...Utility.data import anime_normalizer
|
||||
from ...Utility.utils import (
|
||||
anime_title_percentage_match,
|
||||
remove_html_tags,
|
||||
sanitize_filename,
|
||||
)
|
||||
from ...Utility.utils import anime_title_percentage_match, sanitize_filename
|
||||
from ..config import Config
|
||||
from ..utils.mpv import mpv
|
||||
from ..utils.tools import QueryDict, exit_app
|
||||
from ..utils.utils import clear, fuzzy_inquirer
|
||||
|
||||
SEARCH_RESULTS_CACHE = os.path.join(APP_CACHE_DIR, "search_results")
|
||||
|
||||
|
||||
def write_search_results(
|
||||
search_results: list[AnilistBaseMediaDataSchema], config: Config
|
||||
):
|
||||
import textwrap
|
||||
|
||||
import requests
|
||||
|
||||
for anime in search_results:
|
||||
if not os.path.exists(SEARCH_RESULTS_CACHE):
|
||||
os.mkdir(SEARCH_RESULTS_CACHE)
|
||||
anime_title = (
|
||||
anime["title"][config.preferred_language] or anime["title"]["romaji"]
|
||||
)
|
||||
anime_title = sanitize_filename(anime_title)
|
||||
ANIME_CACHE = os.path.join(SEARCH_RESULTS_CACHE, anime_title)
|
||||
if not os.path.exists(ANIME_CACHE):
|
||||
os.mkdir(ANIME_CACHE)
|
||||
with open(
|
||||
f"{ANIME_CACHE}/image",
|
||||
"wb",
|
||||
) as f:
|
||||
try:
|
||||
image = requests.get(anime["coverImage"]["large"], timeout=5)
|
||||
f.write(image.content)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
with open(f"{ANIME_CACHE}/data", "w") as f:
|
||||
# data = json.dumps(anime, sort_keys=True, indent=2, separators=(',', ': '))
|
||||
template = f"""
|
||||
{"-"*40}
|
||||
Anime Title(jp): {anime['title']['romaji']}
|
||||
Anime Title(eng): {anime['title']['english']}
|
||||
{"-"*40}
|
||||
Popularity: {anime['popularity']}
|
||||
Favourites: {anime['favourites']}
|
||||
Status: {anime['status']}
|
||||
Episodes: {anime['episodes']}
|
||||
Genres: {anilist_data_helper.format_list_data_with_comma(anime['genres'])}
|
||||
Next Episode: {anilist_data_helper.extract_next_airing_episode(anime['nextAiringEpisode'])}
|
||||
Start Date: {anilist_data_helper.format_anilist_date_object(anime['startDate'])}
|
||||
End Date: {anilist_data_helper.format_anilist_date_object(anime['endDate'])}
|
||||
{"-"*40}
|
||||
Description:
|
||||
"""
|
||||
template = textwrap.dedent(template)
|
||||
template = f"""
|
||||
{template}
|
||||
{textwrap.fill(remove_html_tags(str(anime['description'])),width=45)}
|
||||
"""
|
||||
f.write(template)
|
||||
|
||||
|
||||
def get_preview(search_results: list[AnilistBaseMediaDataSchema], config: Config):
|
||||
from threading import Thread
|
||||
|
||||
background_worker = Thread(
|
||||
target=write_search_results, args=(search_results, config)
|
||||
)
|
||||
background_worker.daemon = True
|
||||
background_worker.start()
|
||||
|
||||
os.environ["SHELL"] = shutil.which("bash") or "bash"
|
||||
preview = """
|
||||
if [ -s %s/{}/image ]; then fzf-preview %s/{}/image
|
||||
else echo Loading...
|
||||
fi
|
||||
if [ -s %s/{}/data ]; then cat %s/{}/data
|
||||
else echo Loading...
|
||||
fi
|
||||
""" % (
|
||||
SEARCH_RESULTS_CACHE,
|
||||
SEARCH_RESULTS_CACHE,
|
||||
SEARCH_RESULTS_CACHE,
|
||||
SEARCH_RESULTS_CACHE,
|
||||
)
|
||||
# preview.replace("\n", ";")
|
||||
return preview
|
||||
|
||||
|
||||
def player_controls(config: Config, anilist_config: QueryDict):
|
||||
# user config
|
||||
@@ -581,6 +493,8 @@ def select_anime(config: Config, anilist_config: QueryDict):
|
||||
choices = [*anime_data.keys(), "Back"]
|
||||
if config.use_fzf:
|
||||
if config.preview:
|
||||
from .utils import get_preview
|
||||
|
||||
preview = get_preview(search_results, config)
|
||||
selected_anime_title = fzf.run(
|
||||
choices,
|
||||
|
||||
170
fastanime/cli/interfaces/utils.py
Normal file
170
fastanime/cli/interfaces/utils.py
Normal file
@@ -0,0 +1,170 @@
|
||||
import os
|
||||
import shutil
|
||||
|
||||
from ... import APP_CACHE_DIR
|
||||
from ...libs.anilist.anilist_data_schema import AnilistBaseMediaDataSchema
|
||||
from ...Utility import anilist_data_helper
|
||||
from ...Utility.utils import remove_html_tags, sanitize_filename
|
||||
from ..config import Config
|
||||
|
||||
fzf_preview = """
|
||||
#
|
||||
# The purpose of this script is to demonstrate how to preview a file or an
|
||||
# image in the preview window of fzf.
|
||||
#
|
||||
# Dependencies:
|
||||
# - https://github.com/sharkdp/bat
|
||||
# - https://github.com/hpjansson/chafa
|
||||
# - https://iterm2.com/utilities/imgcat
|
||||
fzf-preview(){
|
||||
if [[ $# -ne 1 ]]; then
|
||||
>&2 echo "usage: $0 FILENAME"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
file=${1/#\~\//$HOME/}
|
||||
type=$(file --dereference --mime -- "$file")
|
||||
|
||||
if [[ ! $type =~ image/ ]]; then
|
||||
if [[ $type =~ =binary ]]; then
|
||||
file "$1"
|
||||
exit
|
||||
fi
|
||||
|
||||
# Sometimes bat is installed as batcat.
|
||||
if command -v batcat > /dev/null; then
|
||||
batname="batcat"
|
||||
elif command -v bat > /dev/null; then
|
||||
batname="bat"
|
||||
else
|
||||
cat "$1"
|
||||
exit
|
||||
fi
|
||||
|
||||
${batname} --style="${BAT_STYLE:-numbers}" --color=always --pager=never -- "$file"
|
||||
exit
|
||||
fi
|
||||
|
||||
dim=${FZF_PREVIEW_COLUMNS}x${FZF_PREVIEW_LINES}
|
||||
if [[ $dim = x ]]; then
|
||||
dim=$(stty size < /dev/tty | awk '{print $2 "x" $1}')
|
||||
elif ! [[ $KITTY_WINDOW_ID ]] && (( FZF_PREVIEW_TOP + FZF_PREVIEW_LINES == $(stty size < /dev/tty | awk '{print $1}') )); then
|
||||
# Avoid scrolling issue when the Sixel image touches the bottom of the screen
|
||||
# * https://github.com/junegunn/fzf/issues/2544
|
||||
dim=${FZF_PREVIEW_COLUMNS}x$((FZF_PREVIEW_LINES - 1))
|
||||
fi
|
||||
|
||||
# 1. Use kitty icat on kitty terminal
|
||||
if [[ $KITTY_WINDOW_ID ]]; then
|
||||
# 1. 'memory' is the fastest option but if you want the image to be scrollable,
|
||||
# you have to use 'stream'.
|
||||
#
|
||||
# 2. The last line of the output is the ANSI reset code without newline.
|
||||
# This confuses fzf and makes it render scroll offset indicator.
|
||||
# So we remove the last line and append the reset code to its previous line.
|
||||
kitty icat --clear --transfer-mode=memory --unicode-placeholder --stdin=no --place="$dim@0x0" "$file" | sed '$d' | sed $'$s/$/\e[m/'
|
||||
|
||||
# 2. Use chafa with Sixel output
|
||||
elif command -v chafa > /dev/null; then
|
||||
chafa -f sixel -s "$dim" "$file"
|
||||
# Add a new line character so that fzf can display multiple images in the preview window
|
||||
echo
|
||||
|
||||
# 3. If chafa is not found but imgcat is available, use it on iTerm2
|
||||
elif command -v imgcat > /dev/null; then
|
||||
# NOTE: We should use https://iterm2.com/utilities/it2check to check if the
|
||||
# user is running iTerm2. But for the sake of simplicity, we just assume
|
||||
# that's the case here.
|
||||
imgcat -W "${dim%%x*}" -H "${dim##*x}" "$file"
|
||||
|
||||
# 4. Cannot find any suitable method to preview the image
|
||||
else
|
||||
file "$file"
|
||||
fi
|
||||
}
|
||||
"""
|
||||
|
||||
|
||||
SEARCH_RESULTS_CACHE = os.path.join(APP_CACHE_DIR, "search_results")
|
||||
|
||||
|
||||
def write_search_results(
|
||||
search_results: list[AnilistBaseMediaDataSchema], config: Config
|
||||
):
|
||||
import textwrap
|
||||
|
||||
import requests
|
||||
|
||||
for anime in search_results:
|
||||
if not os.path.exists(SEARCH_RESULTS_CACHE):
|
||||
os.mkdir(SEARCH_RESULTS_CACHE)
|
||||
anime_title = (
|
||||
anime["title"][config.preferred_language] or anime["title"]["romaji"]
|
||||
)
|
||||
anime_title = sanitize_filename(anime_title)
|
||||
ANIME_CACHE = os.path.join(SEARCH_RESULTS_CACHE, anime_title)
|
||||
if not os.path.exists(ANIME_CACHE):
|
||||
os.mkdir(ANIME_CACHE)
|
||||
with open(
|
||||
f"{ANIME_CACHE}/image",
|
||||
"wb",
|
||||
) as f:
|
||||
try:
|
||||
image = requests.get(anime["coverImage"]["large"], timeout=5)
|
||||
f.write(image.content)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
with open(f"{ANIME_CACHE}/data", "w") as f:
|
||||
# data = json.dumps(anime, sort_keys=True, indent=2, separators=(',', ': '))
|
||||
template = f"""
|
||||
{"-"*40}
|
||||
Anime Title(jp): {anime['title']['romaji']}
|
||||
Anime Title(eng): {anime['title']['english']}
|
||||
{"-"*40}
|
||||
Popularity: {anime['popularity']}
|
||||
Favourites: {anime['favourites']}
|
||||
Status: {anime['status']}
|
||||
Episodes: {anime['episodes']}
|
||||
Genres: {anilist_data_helper.format_list_data_with_comma(anime['genres'])}
|
||||
Next Episode: {anilist_data_helper.extract_next_airing_episode(anime['nextAiringEpisode'])}
|
||||
Start Date: {anilist_data_helper.format_anilist_date_object(anime['startDate'])}
|
||||
End Date: {anilist_data_helper.format_anilist_date_object(anime['endDate'])}
|
||||
{"-"*40}
|
||||
Description:
|
||||
"""
|
||||
template = textwrap.dedent(template)
|
||||
template = f"""
|
||||
{template}
|
||||
{textwrap.fill(remove_html_tags(str(anime['description'])),width=45)}
|
||||
"""
|
||||
f.write(template)
|
||||
|
||||
|
||||
def get_preview(search_results: list[AnilistBaseMediaDataSchema], config: Config):
|
||||
from threading import Thread
|
||||
|
||||
background_worker = Thread(
|
||||
target=write_search_results, args=(search_results, config)
|
||||
)
|
||||
background_worker.daemon = True
|
||||
background_worker.start()
|
||||
|
||||
os.environ["SHELL"] = shutil.which("bash") or "bash"
|
||||
preview = """
|
||||
%s
|
||||
if [ -s %s/{}/image ]; then fzf-preview %s/{}/image
|
||||
else echo Loading...
|
||||
fi
|
||||
if [ -s %s/{}/data ]; then cat %s/{}/data
|
||||
else echo Loading...
|
||||
fi
|
||||
""" % (
|
||||
fzf_preview,
|
||||
SEARCH_RESULTS_CACHE,
|
||||
SEARCH_RESULTS_CACHE,
|
||||
SEARCH_RESULTS_CACHE,
|
||||
SEARCH_RESULTS_CACHE,
|
||||
)
|
||||
# preview.replace("\n", ";")
|
||||
return preview
|
||||
Reference in New Issue
Block a user