216 lines
7.0 KiB
Bash
216 lines
7.0 KiB
Bash
#!/usr/bin/env bash
|
||
|
||
# ==============================================================================
|
||
# 【脚本功能说明】
|
||
# 1. 结合 Fastfetch，在终端启动时展示随机二次元图片 (支持 SFW / NSFW 模式)。
|
||
# 2. 具备静默后台异步下载机制，库存不足时自动补货，绝不阻塞前台终端的启动。
|
||
# 3. 具备智能缓存管理机制，自动控制待展示区与已使用区 (used) 的图片数量上限。
|
||
# 4. 具备极致的网络环境容错处理，无网或弱网时自动降级 fallback，避免死等。
|
||
# 5. 具备自动清理 Fastfetch 内部生成的图片转换缓存功能，防止磁盘空间无感膨胀。
|
||
# ==============================================================================
|
||
|
||
# 启用严格模式：遇到错误退出、未定义变量退出、管道中任何命令失败则失败
|
||
set -euo pipefail
|
||
|
||
# ================= 配置区域 =================
|
||
|
||
# [开关] 阅后即焚模式 (针对 Fastfetch 内部缓存)
|
||
# true = 运行后强力清空 ~/.cache/fastfetch/images/ (防止转码缓存膨胀)
|
||
# false = 保留缓存
|
||
clean_cache_mode=true
|
||
|
||
# 每次补货下载多少张
|
||
download_batch_size=10
|
||
# 最大库存上限 (待展示区)
|
||
max_cache_limit=100
|
||
# 库存少于多少张时开始补货
|
||
min_trigger_limit=60
|
||
# used 目录最大存放数量 (超过此数量将按照时间顺序删除最旧的文件)
|
||
max_used_limit=50
|
||
|
||
# ===========================================
|
||
|
||
# --- 0. 参数解析与模式设置 ---
|
||
|
||
nsfw_mode=false
|
||
if [[ "${NSFW:-}" == "1" ]]; then
|
||
nsfw_mode=true
|
||
fi
|
||
|
||
args_for_fastfetch=()
|
||
for arg in "$@"; do
|
||
if [[ "$arg" == "--nsfw" ]]; then
|
||
nsfw_mode=true
|
||
else
|
||
args_for_fastfetch+=("$arg")
|
||
fi
|
||
done
|
||
|
||
# --- 1. 目录配置 ---
|
||
|
||
if [[ "$nsfw_mode" == true ]]; then
|
||
cache_dir="$HOME/.cache/fastfetch_waifu_nsfw"
|
||
lock_file="/tmp/fastfetch_waifu_nsfw.lock"
|
||
else
|
||
cache_dir="$HOME/.cache/fastfetch_waifu"
|
||
lock_file="/tmp/fastfetch_waifu.lock"
|
||
fi
|
||
|
||
used_dir="$cache_dir/used"
|
||
mkdir -p "$cache_dir" "$used_dir"
|
||
|
||
# --- 2. 核心函数定义 ---
|
||
|
||
check_network() {
|
||
curl -sI --connect-timeout 2 "http://captive.apple.com/hotspot-detect.html" >/dev/null 2>&1
|
||
}
|
||
|
||
get_random_url() {
|
||
local timeout="--connect-timeout 5 --max-time 15"
|
||
local rand=$(( RANDOM % 3 + 1 ))
|
||
|
||
# 管道中使用 || true 确保 jq 解析失败时不会触发 pipefail 导致脚本崩溃
|
||
if [[ "$nsfw_mode" == true ]]; then
|
||
case $rand in
|
||
1) curl -s $timeout "https://api.waifu.im/images?IncludedTags=waifu&IsNsfw=true" | jq -r '.images[0].url' 2>/dev/null || true ;;
|
||
2) curl -s $timeout "https://api.waifu.pics/nsfw/waifu" | jq -r '.url' 2>/dev/null || true ;;
|
||
3) curl -s $timeout "https://api.waifu.pics/nsfw/neko" | jq -r '.url' 2>/dev/null || true ;;
|
||
esac
|
||
else
|
||
case $rand in
|
||
1) curl -s $timeout "https://api.waifu.im/images?IncludedTags=waifu&IsNsfw=false" | jq -r '.images[0].url' 2>/dev/null || true ;;
|
||
2) curl -s $timeout "https://nekos.best/api/v2/waifu" | jq -r '.results[0].url' 2>/dev/null || true ;;
|
||
3) curl -s $timeout "https://api.waifu.pics/sfw/waifu" | jq -r '.url' 2>/dev/null || true ;;
|
||
esac
|
||
fi
|
||
}
|
||
|
||
download_one_image() {
|
||
local url
|
||
url=$(get_random_url || true)
|
||
|
||
if [[ -n "$url" && "$url" =~ ^http ]]; then
|
||
local filename="waifu_$(date +%s%N)_$RANDOM.jpg"
|
||
local target_path="$cache_dir/$filename"
|
||
|
||
# 使用 || true 防止 curl 意外崩溃
|
||
curl -s -L --connect-timeout 5 --max-time 15 -o "$target_path" "$url" || true
|
||
|
||
if [[ -s "$target_path" ]]; then
|
||
if command -v file >/dev/null 2>&1; then
|
||
if ! file --mime-type "$target_path" | grep -q "image/"; then
|
||
rm -f "$target_path"
|
||
fi
|
||
fi
|
||
else
|
||
rm -f "$target_path"
|
||
fi
|
||
fi
|
||
}
|
||
|
||
background_job() {
|
||
(
|
||
# 1. 彻底切断标准输入输出，变成纯粹的后台幽灵进程
|
||
exec </dev/null >/dev/null 2>&1
|
||
|
||
# 2. 忽略终端挂断信号
|
||
trap '' HUP
|
||
|
||
# 3. 局部关闭严格模式，确保后台尽力而为，不因单次网络错误退出
|
||
set +e
|
||
|
||
# 4. 获取文件描述符 200 的排他锁，防止并发下载
|
||
exec 200>"$lock_file"
|
||
flock -n 200 || exit 0
|
||
|
||
# 网络检查，没网就悄悄退出
|
||
if ! check_network; then
|
||
exit 0
|
||
fi
|
||
|
||
# 开启空 glob 扩展
|
||
shopt -s nullglob
|
||
|
||
# --- 补货逻辑 ---
|
||
local current_files=("$cache_dir"/*.jpg)
|
||
if (( ${#current_files[@]} < min_trigger_limit )); then
|
||
for ((i=0; i<download_batch_size; i++)); do
|
||
download_one_image
|
||
sleep 0.5
|
||
done
|
||
fi
|
||
|
||
# --- 清理过多库存逻辑 ---
|
||
local final_files=("$cache_dir"/*.jpg)
|
||
if (( ${#final_files[@]} > max_cache_limit )); then
|
||
local skip_lines=$(( max_cache_limit + 1 ))
|
||
local cache_to_delete
|
||
# 只有明确文件存在时才执行 ls，避免 nullglob 导致的当前目录误删
|
||
mapfile -t cache_to_delete < <(ls -tp "$cache_dir"/*.jpg 2>/dev/null | tail -n +"$skip_lines")
|
||
if (( ${#cache_to_delete[@]} > 0 )); then
|
||
rm -f -- "${cache_to_delete[@]}"
|
||
fi
|
||
fi
|
||
) &
|
||
}
|
||
|
||
# --- 3. 主程序逻辑 ---
|
||
|
||
# 开启空 glob 扩展，安全获取文件列表
|
||
shopt -s nullglob
|
||
files=("$cache_dir"/*.jpg)
|
||
num_files=${#files[@]}
|
||
selected_img=""
|
||
|
||
if (( num_files > 0 )); then
|
||
rand_index=$(( RANDOM % num_files ))
|
||
selected_img="${files[$rand_index]}"
|
||
|
||
background_job
|
||
else
|
||
echo "库存不够啦！正在去搬运新的图片，请稍等哦..."
|
||
|
||
if check_network; then
|
||
# 前台下载时暂时关闭严格模式，防止网络异常导致脚本闪退
|
||
set +e
|
||
download_one_image
|
||
set -e
|
||
else
|
||
echo "网络好像不太通畅，无法下载新图片 QAQ"
|
||
fi
|
||
|
||
# 重新获取文件列表
|
||
files=("$cache_dir"/*.jpg)
|
||
if (( ${#files[@]} > 0 )); then
|
||
selected_img="${files[0]}"
|
||
background_job
|
||
fi
|
||
fi
|
||
|
||
if [[ -n "$selected_img" && -f "$selected_img" ]]; then
|
||
# 显示图片
|
||
fastfetch --logo "$selected_img" --logo-preserve-aspect-ratio true "${args_for_fastfetch[@]}"
|
||
|
||
# === 移动到 used 目录 ===
|
||
mv "$selected_img" "$used_dir/"
|
||
|
||
# === 检查 used 目录并清理旧图 ===
|
||
used_files=("$used_dir"/*.jpg)
|
||
if (( ${#used_files[@]} > max_used_limit )); then
|
||
skip_lines=$(( max_used_limit + 1 ))
|
||
|
||
mapfile -t files_to_delete < <(ls -tp "$used_dir"/*.jpg 2>/dev/null | tail -n +"$skip_lines")
|
||
if (( ${#files_to_delete[@]} > 0 )); then
|
||
rm -f -- "${files_to_delete[@]}"
|
||
fi
|
||
fi
|
||
|
||
# 清理 Fastfetch 内部缓存
|
||
if [[ "$clean_cache_mode" == true ]]; then
|
||
rm -rf "$HOME/.cache/fastfetch/images"
|
||
fi
|
||
else
|
||
echo "图片获取失败了，这次只能先显示默认的 Logo 啦 QAQ"
|
||
fastfetch "${args_for_fastfetch[@]}"
|
||
fi
|