#!/usr/bin/env bash
# Thư mục chứa script
PROJECT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
cd "$PROJECT_DIR"

clear
echo "=========================================================="
echo "      TypeWhisper-Nhi Auto Updater"
echo "=========================================================="
echo ""

# 1. Pull latest code from Git
echo "1. Đang tải mã nguồn mới từ GitHub..."
if ! git pull; then
    echo "❌ Lỗi: Không thể kéo mã nguồn từ GitHub. Vui lòng kiểm tra kết nối mạng."
    read -p "Nhấn Enter để thoát..."
    exit 1
fi

# 2. Update python dependencies
echo ""
echo "2. Đang cập nhật các thư viện Python..."
./venv/bin/pip install --upgrade pip
./venv/bin/pip install -r requirements.txt

# 3. Check for new models
echo ""
echo "3. Đang kiểm tra và tải mô hình AI mới (nếu có)..."
./venv/bin/python3 download_models.py

# 4. Restart background service if it is running
echo ""
echo "4. Đang khởi động lại dịch vụ chạy ngầm..."
PLIST_PATH="$HOME/Library/LaunchAgents/com.typewhisper.nhi.plist"
if [ -f "$PLIST_PATH" ]; then
    launchctl unload "$PLIST_PATH" 2>/dev/null
    launchctl load "$PLIST_PATH"
    echo "✅ Đã khởi động lại dịch vụ chính com.typewhisper.nhi"
fi

echo ""
echo "=========================================================="
echo "🎉 CẬP NHẬT HOÀN TẤT!"
echo "=========================================================="
osascript -e 'display notification "Đã cập nhật lên phiên bản mới nhất thành công!" with title "TypeWhisper Updater"'
sleep 2