#!/bin/bash

# 색상 설정
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

echo -e "${BLUE}=== FFmpeg 설치 스크립트 ===${NC}"

# FFmpeg 확인
if command -v ffmpeg &> /dev/null; then
    echo -e "${GREEN}FFmpeg가 이미 설치되어 있습니다.${NC}"
    ffmpeg -version | head -n 1
    exit 0
fi

# 시스템 확인
echo -e "\n${YELLOW}시스템 확인 중...${NC}"
OS="$(uname -s)"
case "${OS}" in
    Linux*)     
        OS_TYPE="Linux"
        # 패키지 관리자 확인
        if command -v apt-get &> /dev/null; then
            PM="apt"
        elif command -v yum &> /dev/null; then
            PM="yum"
        else
            PM="unknown"
        fi
        ;;
    Darwin*)    
        OS_TYPE="Mac"
        # Homebrew 확인
        if command -v brew &> /dev/null; then
            PM="brew"
        else
            PM="unknown"
        fi
        ;;
    CYGWIN*|MINGW*)     
        OS_TYPE="Windows"
        PM="unknown"
        ;;
    *)          
        OS_TYPE="Unknown"
        PM="unknown"
        ;;
esac

echo -e "감지된 운영 체제: ${BLUE}${OS_TYPE}${NC}"

# FFmpeg 설치
echo -e "\n${YELLOW}FFmpeg 설치 준비 중...${NC}"

case "${PM}" in
    brew)
        echo -e "macOS용 FFmpeg를 Homebrew로 설치합니다."
        read -p "계속하시겠습니까? (y/n) " -n 1 -r
        echo
        if [[ $REPLY =~ ^[Yy]$ ]]; then
            echo -e "${BLUE}FFmpeg 설치 중...${NC}"
            brew install ffmpeg
            if [ $? -ne 0 ]; then
                echo -e "${RED}FFmpeg 설치 실패${NC}"
                exit 1
            fi
            echo -e "${GREEN}FFmpeg 설치 완료${NC}"
        else
            echo -e "${YELLOW}설치가 취소되었습니다.${NC}"
            exit 1
        fi
        ;;
    apt)
        echo -e "Ubuntu/Debian용 FFmpeg를 apt로 설치합니다."
        read -p "계속하시겠습니까? (y/n) " -n 1 -r
        echo
        if [[ $REPLY =~ ^[Yy]$ ]]; then
            echo -e "${BLUE}FFmpeg 설치 중...${NC}"
            sudo apt-get update && sudo apt-get install -y ffmpeg
            if [ $? -ne 0 ]; then
                echo -e "${RED}FFmpeg 설치 실패${NC}"
                exit 1
            fi
            echo -e "${GREEN}FFmpeg 설치 완료${NC}"
        else
            echo -e "${YELLOW}설치가 취소되었습니다.${NC}"
            exit 1
        fi
        ;;
    yum)
        echo -e "CentOS/RHEL용 FFmpeg를 yum으로 설치합니다."
        read -p "계속하시겠습니까? (y/n) " -n 1 -r
        echo
        if [[ $REPLY =~ ^[Yy]$ ]]; then
            echo -e "${BLUE}FFmpeg 설치 중...${NC}"
            sudo yum install -y ffmpeg
            if [ $? -ne 0 ]; then
                echo -e "${RED}FFmpeg 설치 실패${NC}"
                exit 1
            fi
            echo -e "${GREEN}FFmpeg 설치 완료${NC}"
        else
            echo -e "${YELLOW}설치가 취소되었습니다.${NC}"
            exit 1
        fi
        ;;
    *)
        echo -e "${RED}자동 설치를 지원하지 않는 시스템입니다.${NC}"
        echo -e "다음 방법 중 하나로 FFmpeg를 설치해 주세요:"
        echo -e "- macOS: ${YELLOW}brew install ffmpeg${NC}"
        echo -e "- Ubuntu/Debian: ${YELLOW}sudo apt-get update && sudo apt-get install ffmpeg${NC}"
        echo -e "- CentOS/RHEL: ${YELLOW}sudo yum install ffmpeg${NC}"
        echo -e "- Windows: ${YELLOW}https://ffmpeg.org/download.html${NC} 에서 다운로드"
        exit 1
        ;;
esac

# 설치 확인
echo -e "\n${YELLOW}FFmpeg 설치 확인 중...${NC}"
if command -v ffmpeg &> /dev/null; then
    echo -e "${GREEN}FFmpeg가 성공적으로 설치되었습니다.${NC}"
    ffmpeg -version | head -n 1
else
    echo -e "${RED}FFmpeg 설치가 확인되지 않습니다. 수동으로 설치해 주세요.${NC}"
    exit 1
fi

echo -e "\n${GREEN}FFmpeg 설치가 완료되었습니다.${NC}"