#!/bin/bash
# 多语言快速截图脚本

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"

# 语言参数
LANGUAGE=${1:-zh-Hans}

SCREENSHOT_DIR="$SCRIPT_DIR/app-store-screenshots/$LANGUAGE/6.7inch"
mkdir -p "$SCREENSHOT_DIR"

COUNTER=1

# 语言名称映射
declare -A LANG_NAMES=(
    ["zh-Hans"]="简体中文"
    ["zh-Hant"]="繁体中文"
    ["en"]="English"
    ["es"]="Español"
    ["ar"]="العربية"
    ["fr"]="Français"
    ["ja"]="日本語"
    ["ko"]="한국어"
)

echo "📸 快速截图模式"
echo "🌐 语言: ${LANG_NAMES[$LANGUAGE]} ($LANGUAGE)"
echo "📂 保存位置: $SCREENSHOT_DIR"
echo ""
echo "操作说明:"
echo "  1. 在模拟器中导航到要截图的页面"
echo "  2. 按 Enter 键截图"
echo "  3. 输入 'q' 或 'quit' 退出"
echo ""

while true; do
    read -p "[截图 $COUNTER] 按 Enter 截图 (或输入 'q' 退出): " input

    if [[ "$input" == "q" ]] || [[ "$input" == "quit" ]]; then
        echo "✅ 完成! 共截取 $((COUNTER - 1)) 张截图"
        break
    fi

    FILENAME=$(printf "%02d.png" $COUNTER)
    FILEPATH="$SCREENSHOT_DIR/$FILENAME"

    xcrun simctl io booted screenshot "$FILEPATH"

    if [ $? -eq 0 ]; then
        echo "✅ 截图已保存: $FILENAME"
        COUNTER=$((COUNTER + 1))
    else
        echo "❌ 截图失败"
    fi

    echo ""
done

echo ""
echo "📂 所有截图保存在: $SCREENSHOT_DIR"
ls -lh "$SCREENSHOT_DIR"