#!/bin/bash

# PharmQ noVNC Setup Script
# Proxmox Host에 noVNC over WebSocket 자동 설치
# 사용법: curl -fsSL https://git.0bin.in/thug0bin/pve9-repo-fix/raw/branch/main/VNC/pharmq-novnc-setup.sh | bash

set -euo pipefail

# 색상 코드 정의
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
PURPLE='\033[0;35m'
CYAN='\033[0;36m'
NC='\033[0m' # No Color

# 로그 함수들
msg_info() {
    echo -e "${BLUE}[INFO]${NC} $1"
}

msg_ok() {
    echo -e "${GREEN}[OK]${NC} $1"
}

msg_error() {
    echo -e "${RED}[ERROR]${NC} $1"
    exit 1
}

msg_warn() {
    echo -e "${YELLOW}[WARN]${NC} $1"
}

# 헤더 출력
print_header() {
    clear
    echo -e "${PURPLE}"
    echo "═══════════════════════════════════════════════════════════════════"
    echo "           PharmQ noVNC Setup Script v1.0"
    echo "═══════════════════════════════════════════════════════════════════"
    echo -e "${NC}"
    echo "이 스크립트는 Proxmox Host에 noVNC over WebSocket을 자동으로 설치합니다."
    echo "각 약국의 VM VNC 화면을 웹 브라우저로 제공합니다."
    echo ""
}

# Proxmox 버전 확인
check_proxmox_version() {
    msg_info "Proxmox VE 버전 확인 중..."

    if ! command -v pveversion > /dev/null 2>&1; then
        msg_warn "pveversion 명령을 찾을 수 없습니다. Proxmox 환경이 아닐 수 있습니다."
        read -p "계속 진행하시겠습니까? [y/N]: " continue_anyway </dev/tty
        case $continue_anyway in
            [yY]|[yY][eE][sS]) ;;
            *) msg_error "설치가 취소되었습니다." ;;
        esac
        return 0
    fi

    local pve_version=$(pveversion 2>/dev/null | head -n1 | grep -oP '\d+\.\d+' | head -n1 | cut -d'.' -f1)

    if [ -z "$pve_version" ]; then
        msg_warn "Proxmox VE 버전을 확인할 수 없습니다. 계속 진행합니다."
        return 0
    fi

    if ! [[ "$pve_version" =~ ^[0-9]+$ ]]; then
        msg_warn "Proxmox VE 버전 형식이 올바르지 않습니다 ($pve_version). 계속 진행합니다."
        return 0
    fi

    msg_ok "Proxmox VE $pve_version.x 버전 확인됨"
}

# 루트 권한 확인
check_root() {
    if [ "$EUID" -ne 0 ]; then
        msg_error "이 스크립트는 root 권한으로 실행해야 합니다. sudo를 사용하세요."
    fi
}

# VM 정보 가져오기
get_vm_list() {
    msg_info "실행 중인 VM 목록 가져오는 중..."

    if ! command -v qm > /dev/null 2>&1; then
        msg_error "qm 명령어를 찾을 수 없습니다. Proxmox VE 환경이 아닙니다."
    fi

    # 실행 중인 VM 목록
    VM_LIST=$(qm list 2>/dev/null | tail -n +2 | awk '{print $1}' | sort -n)

    if [ -z "$VM_LIST" ]; then
        msg_warn "실행 중인 VM이 없습니다."
        return 1
    fi

    msg_ok "VM 목록 가져오기 완료"
    return 0
}

# VNC 포트 가져오기
get_vnc_port() {
    local vmid=$1
    local vnc_port=$(qm config "$vmid" 2>/dev/null | grep -oP 'vnc.*?:59\d+' | grep -oP '59\d+' | head -n1)
    echo "$vnc_port"
}

# VM 선택 및 설정
configure_vms() {
    echo -e "${CYAN}VM 설정을 시작합니다:${NC}"
    echo ""

    get_vm_list || msg_error "VM을 찾을 수 없습니다."

    echo "실행 중인 VM 목록:"
    echo ""

    declare -A VM_INFO
    local idx=1

    for vmid in $VM_LIST; do
        local vm_name=$(qm config "$vmid" 2>/dev/null | grep "^name:" | awk '{print $2}' || echo "VM-$vmid")
        local vnc_port=$(get_vnc_port "$vmid")

        if [ -n "$vnc_port" ]; then
            echo "  $idx) VM $vmid - $vm_name (VNC: $vnc_port)"
            VM_INFO[$idx]="$vmid:$vm_name:$vnc_port"
            ((idx++))
        fi
    done

    if [ ${#VM_INFO[@]} -eq 0 ]; then
        msg_error "VNC가 활성화된 VM이 없습니다."
    fi

    echo ""
    echo "noVNC로 제공할 VM을 최대 2개까지 선택하세요."
    echo ""

    # VM1 선택
    while true; do
        read -p "VM1 선택 (번호 입력): " vm1_choice </dev/tty
        if [[ -n "${VM_INFO[$vm1_choice]}" ]]; then
            IFS=':' read -r VM1_ID VM1_NAME VM1_VNC_PORT <<< "${VM_INFO[$vm1_choice]}"
            break
        else
            msg_warn "잘못된 선택입니다. 다시 입력해주세요."
        fi
    done

    msg_ok "VM1: $VM1_ID - $VM1_NAME (VNC: $VM1_VNC_PORT)"

    # VM2 선택 (선택사항)
    echo ""
    read -p "VM2를 추가하시겠습니까? [y/N]: " add_vm2 </dev/tty

    VM2_ENABLED=false
    case $add_vm2 in
        [yY]|[yY][eE][sS])
            while true; do
                read -p "VM2 선택 (번호 입력): " vm2_choice </dev/tty
                if [[ -n "${VM_INFO[$vm2_choice]}" ]]; then
                    if [ "$vm2_choice" = "$vm1_choice" ]; then
                        msg_warn "VM1과 다른 VM을 선택해주세요."
                        continue
                    fi
                    IFS=':' read -r VM2_ID VM2_NAME VM2_VNC_PORT <<< "${VM_INFO[$vm2_choice]}"
                    VM2_ENABLED=true
                    msg_ok "VM2: $VM2_ID - $VM2_NAME (VNC: $VM2_VNC_PORT)"
                    break
                else
                    msg_warn "잘못된 선택입니다. 다시 입력해주세요."
                fi
            done
            ;;
        *)
            msg_info "VM2 설정을 건너뜁니다."
            ;;
    esac
}

# 약국 정보 입력
get_pharmacy_info() {
    echo ""
    echo -e "${CYAN}약국 정보를 입력해주세요:${NC}"
    echo ""

    # 약국 코드
    while true; do
        read -p "약국 코드 (예: P0014): " PHARMACY_CODE </dev/tty
        if [ -z "$PHARMACY_CODE" ]; then
            msg_warn "약국 코드는 필수입니다."
            continue
        fi
        if [[ ! "$PHARMACY_CODE" =~ ^P[0-9]{4}$ ]]; then
            msg_warn "약국 코드는 P0001 형식이어야 합니다."
            continue
        fi
        break
    done

    # 약국 이름
    while true; do
        read -p "약국 이름 (예: 늘기쁨약국): " PHARMACY_NAME </dev/tty
        if [ -z "$PHARMACY_NAME" ]; then
            msg_warn "약국 이름은 필수입니다."
            continue
        fi
        break
    done

    # VNC 호스트 (로컬 IP)
    local local_ip=$(hostname -I | awk '{print $1}')
    read -p "Proxmox 호스트 IP [$local_ip]: " VNC_HOST </dev/tty
    VNC_HOST=${VNC_HOST:-$local_ip}

    # Proxmox API 정보
    echo ""
    echo "Proxmox API 접속 정보 (마우스 리셋 기능에 필요):"
    read -p "Proxmox 사용자명 [root@pam]: " PVE_USER </dev/tty
    PVE_USER=${PVE_USER:-root@pam}

    echo -n "Proxmox 비밀번호: "
    read -s PVE_PASSWORD </dev/tty
    echo ""

    if [ -z "$PVE_PASSWORD" ]; then
        msg_warn "비밀번호가 입력되지 않았습니다. 마우스 리셋 기능이 작동하지 않을 수 있습니다."
    fi

    # WebSocket 포트
    WS_PORT1=6085
    WS_PORT2=6086
    FLASK_PORT=6000

    echo ""
    echo -e "${YELLOW}입력된 정보:${NC}"
    echo "  약국 코드: $PHARMACY_CODE"
    echo "  약국 이름: $PHARMACY_NAME"
    echo "  Proxmox 호스트: $VNC_HOST"
    echo "  VM1: $VM1_ID - $VM1_NAME (VNC: $VM1_VNC_PORT, WebSocket: $WS_PORT1)"
    if [ "$VM2_ENABLED" = true ]; then
        echo "  VM2: $VM2_ID - $VM2_NAME (VNC: $VM2_VNC_PORT, WebSocket: $WS_PORT2)"
    fi
    echo "  Flask 포트: $FLASK_PORT"
    echo ""

    read -p "이 설정으로 계속하시겠습니까? [y/N]: " confirm </dev/tty
    case $confirm in
        [yY]|[yY][eE][sS])
            return 0
            ;;
        *)
            msg_error "설치가 취소되었습니다."
            ;;
    esac
}

# 네트워크 연결 확인
check_network() {
    msg_info "네트워크 연결 확인 중..."

    if ! ping -c 1 -W 5 8.8.8.8 > /dev/null 2>&1; then
        msg_error "인터넷 연결을 확인할 수 없습니다. 네트워크 설정을 확인해주세요."
    fi

    msg_ok "네트워크 연결 확인됨"
}

# 필수 패키지 설치
install_packages() {
    msg_info "필수 패키지 설치 중..."

    msg_info "  - 패키지 목록 업데이트 중..."
    if ! apt update > /dev/null 2>&1; then
        msg_error "패키지 목록 업데이트에 실패했습니다."
    fi

    local packages="python3 python3-venv python3-pip git curl websockify"

    for package in $packages; do
        if dpkg -l | grep -q "^ii  $package "; then
            msg_ok "    $package 이미 설치됨"
            continue
        fi

        msg_info "  - $package 설치 중..."
        if apt install -y "$package" > /dev/null 2>&1; then
            msg_ok "    $package 설치 완료"
        else
            msg_error "$package 설치에 실패했습니다."
        fi
    done

    msg_ok "모든 패키지 설치 완료"
}

# pharmq-novnc 다운로드 및 설치
install_pharmq_novnc() {
    msg_info "pharmq-novnc 애플리케이션 설치 중..."

    local INSTALL_DIR="/srv/pharmq-novnc"
    local REPO_URL="https://git.0bin.in/thug0bin/pharmq-novnc.git"

    # 기존 설치 확인
    if [ -d "$INSTALL_DIR" ]; then
        msg_warn "기존 설치 디렉토리가 발견되었습니다: $INSTALL_DIR"
        read -p "기존 설치를 덮어쓰시겠습니까? [y/N]: " overwrite </dev/tty
        case $overwrite in
            [yY]|[yY][eE][sS])
                msg_info "기존 설치 백업 중..."
                mv "$INSTALL_DIR" "${INSTALL_DIR}.backup.$(date +%Y%m%d-%H%M%S)"
                ;;
            *)
                msg_error "설치가 취소되었습니다."
                ;;
        esac
    fi

    # Git clone
    msg_info "pharmq-novnc 리포지토리 다운로드 중..."
    if git clone "$REPO_URL" "$INSTALL_DIR" > /dev/null 2>&1; then
        msg_ok "다운로드 완료"
    else
        msg_error "리포지토리 다운로드에 실패했습니다."
    fi

    # Python 가상환경 생성
    msg_info "Python 가상환경 생성 중..."
    cd "$INSTALL_DIR"
    python3 -m venv venv
    source venv/bin/activate
    pip install --upgrade pip > /dev/null 2>&1
    pip install -r requirements.txt > /dev/null 2>&1
    deactivate
    msg_ok "Python 가상환경 설정 완료"
}

# config.json 생성
create_config() {
    msg_info "설정 파일 생성 중..."

    local INSTALL_DIR="/srv/pharmq-novnc"
    local CONFIG_FILE="$INSTALL_DIR/config.json"

    if [ "$VM2_ENABLED" = true ]; then
        cat > "$CONFIG_FILE" << EOF
{
  "pharmacy_id": "$PHARMACY_CODE",
  "pharmacy_name": "$PHARMACY_NAME",

  "vnc_host": "$VNC_HOST",

  "proxmox": {
    "host": "$VNC_HOST",
    "port": 8006,
    "username": "$PVE_USER",
    "password": "$PVE_PASSWORD",
    "verify_ssl": false
  },

  "vms": [
    {
      "id": $VM1_ID,
      "name": "$VM1_NAME",
      "vnc_port": $VM1_VNC_PORT,
      "websockify_port": $WS_PORT1
    },
    {
      "id": $VM2_ID,
      "name": "$VM2_NAME",
      "vnc_port": $VM2_VNC_PORT,
      "websockify_port": $WS_PORT2
    }
  ],

  "flask": {
    "host": "0.0.0.0",
    "port": $FLASK_PORT
  },

  "notes": "vnc_host는 Proxmox VNC 서버 주소입니다. Flask는 Ubuntu VM에서 실행되며 Proxmox API를 통해 원격 VM을 제어합니다."
}
EOF
    else
        cat > "$CONFIG_FILE" << EOF
{
  "pharmacy_id": "$PHARMACY_CODE",
  "pharmacy_name": "$PHARMACY_NAME",

  "vnc_host": "$VNC_HOST",

  "proxmox": {
    "host": "$VNC_HOST",
    "port": 8006,
    "username": "$PVE_USER",
    "password": "$PVE_PASSWORD",
    "verify_ssl": false
  },

  "vms": [
    {
      "id": $VM1_ID,
      "name": "$VM1_NAME",
      "vnc_port": $VM1_VNC_PORT,
      "websockify_port": $WS_PORT1
    }
  ],

  "flask": {
    "host": "0.0.0.0",
    "port": $FLASK_PORT
  },

  "notes": "vnc_host는 Proxmox VNC 서버 주소입니다. Flask는 Ubuntu VM에서 실행되며 Proxmox API를 통해 원격 VM을 제어합니다."
}
EOF
    fi

    msg_ok "설정 파일 생성 완료: $CONFIG_FILE"
}

# systemd 서비스 설치
setup_systemd_services() {
    msg_info "systemd 서비스 설치 중..."

    local INSTALL_DIR="/srv/pharmq-novnc"

    # 서비스 파일 복사
    cp "$INSTALL_DIR/systemd/"*.service /etc/systemd/system/

    # systemd 리로드
    systemctl daemon-reload

    msg_ok "systemd 서비스 등록 완료"
}

# 서비스 시작
start_services() {
    msg_info "서비스 시작 중..."

    local services=("pharmq-websockify-vnc1" "pharmq-vnc-app")

    if [ "$VM2_ENABLED" = true ]; then
        services+=("pharmq-websockify-vnc2")
    fi

    for service in "${services[@]}"; do
        msg_info "  - $service.service 시작 중..."
        systemctl enable "$service.service" > /dev/null 2>&1
        systemctl restart "$service.service"

        sleep 2

        if systemctl is-active "$service.service" > /dev/null 2>&1; then
            msg_ok "    $service: 실행 중"
        else
            msg_warn "    $service: 시작 실패"
            msg_info "    로그 확인: journalctl -xeu $service.service"
        fi
    done

    msg_ok "모든 서비스 시작 완료"
}

# 방화벽 설정 (선택사항)
configure_firewall() {
    echo ""
    echo -e "${CYAN}방화벽 설정 (선택사항)${NC}"
    echo ""
    echo "Tailscale VPN으로만 접근하도록 방화벽을 설정하시겠습니까?"
    echo "이 설정은 외부에서의 직접 접근을 차단하고 VPN을 통한 접근만 허용합니다."
    echo ""
    read -p "방화벽 설정을 진행하시겠습니까? [y/N]: " setup_fw </dev/tty

    case $setup_fw in
        [yY]|[yY][eE][sS])
            msg_info "방화벽 설정 중..."

            if ! command -v ufw > /dev/null 2>&1; then
                msg_info "ufw 설치 중..."
                apt install -y ufw > /dev/null 2>&1
            fi

            # Tailscale 인터페이스 확인
            if ip link show tailscale0 > /dev/null 2>&1; then
                ufw allow in on tailscale0 to any port $FLASK_PORT,$WS_PORT1,$WS_PORT2 proto tcp
                ufw deny $FLASK_PORT/tcp
                ufw deny $WS_PORT1/tcp
                ufw deny $WS_PORT2/tcp

                msg_ok "방화벽 설정 완료 (Tailscale VPN만 허용)"
            else
                msg_warn "Tailscale 인터페이스를 찾을 수 없습니다. 방화벽 설정을 건너뜁니다."
            fi
            ;;
        *)
            msg_info "방화벽 설정을 건너뜁니다."
            ;;
    esac
}

# 설치 테스트
test_installation() {
    msg_info "설치 테스트 중..."

    # 포트 리스닝 확인
    sleep 3

    if ss -tlnp | grep -q ":$FLASK_PORT "; then
        msg_ok "Flask 앱이 포트 $FLASK_PORT에서 실행 중입니다."
    else
        msg_warn "Flask 앱이 포트 $FLASK_PORT에서 실행되지 않습니다."
    fi

    if ss -tlnp | grep -q ":$WS_PORT1 "; then
        msg_ok "WebSocket 1이 포트 $WS_PORT1에서 실행 중입니다."
    else
        msg_warn "WebSocket 1이 포트 $WS_PORT1에서 실행되지 않습니다."
    fi

    if [ "$VM2_ENABLED" = true ]; then
        if ss -tlnp | grep -q ":$WS_PORT2 "; then
            msg_ok "WebSocket 2가 포트 $WS_PORT2에서 실행 중입니다."
        else
            msg_warn "WebSocket 2가 포트 $WS_PORT2에서 실행되지 않습니다."
        fi
    fi

    # 헬스체크
    if curl -s -f "http://localhost:$FLASK_PORT/health" > /dev/null 2>&1; then
        msg_ok "헬스체크 성공"
    else
        msg_warn "헬스체크 실패"
    fi
}

# 완료 메시지
print_completion() {
    echo ""
    echo -e "${GREEN}═══════════════════════════════════════════════════════════════════${NC}"
    echo -e "${GREEN}                    설치가 완료되었습니다!${NC}"
    echo -e "${GREEN}═══════════════════════════════════════════════════════════════════${NC}"
    echo ""
    echo -e "${CYAN}설정 요약:${NC}"
    echo "  약국 코드: $PHARMACY_CODE"
    echo "  약국 이름: $PHARMACY_NAME"
    echo "  설치 경로: /srv/pharmq-novnc"
    echo ""
    echo -e "${CYAN}VM 설정:${NC}"
    echo "  VM1: $VM1_ID - $VM1_NAME"
    echo "    - VNC 포트: $VM1_VNC_PORT"
    echo "    - WebSocket 포트: $WS_PORT1"
    if [ "$VM2_ENABLED" = true ]; then
        echo "  VM2: $VM2_ID - $VM2_NAME"
        echo "    - VNC 포트: $VM2_VNC_PORT"
        echo "    - WebSocket 포트: $WS_PORT2"
    fi
    echo ""
    echo -e "${CYAN}접속 URL:${NC}"
    echo "  로컬 헬스체크: http://localhost:$FLASK_PORT/health"
    echo "  VM1 noVNC: http://localhost:$FLASK_PORT/"
    if [ "$VM2_ENABLED" = true ]; then
        echo "  VM2 noVNC: http://localhost:$FLASK_PORT/vnc2"
    fi
    echo ""
    echo -e "${CYAN}Gateway를 통한 접속 (외부):${NC}"
    echo "  https://gateway.pharmq.kr/api/novnc/$PHARMACY_CODE/vnc1"
    if [ "$VM2_ENABLED" = true ]; then
        echo "  https://gateway.pharmq.kr/api/novnc/$PHARMACY_CODE/vnc2"
    fi
    echo ""
    echo -e "${CYAN}관리 명령어:${NC}"
    echo "  상태 확인: /srv/pharmq-novnc/scripts/check-status.sh"
    echo "  전체 재시작: systemctl restart pharmq-websockify-vnc1 pharmq-vnc-app"
    if [ "$VM2_ENABLED" = true ]; then
        echo "                systemctl restart pharmq-websockify-vnc2"
    fi
    echo "  로그 확인: journalctl -u pharmq-vnc-app.service -f"
    echo ""
    echo -e "${YELLOW}다음 단계:${NC}"
    echo "  1. 헬스체크 테스트: curl http://localhost:$FLASK_PORT/health"
    echo "  2. Gateway에서 접속 테스트"
    echo "  3. 브라우저에서 noVNC 화면 확인"
    echo ""
}

# 메인 함수
main() {
    print_header

    # 사전 검사
    check_root
    check_proxmox_version

    # 기존 설치 확인
    if [ -f "/srv/pharmq-novnc/config.json" ] && \
       systemctl is-active --quiet pharmq-vnc-app.service; then
        echo ""
        echo -e "${GREEN}=========================================="
        echo "✅ PharmQ noVNC가 이미 설치되어 있습니다!"
        echo -e "==========================================${NC}"
        echo ""

        CURRENT_PHARMACY_CODE=$(python3 -c "import json; print(json.load(open('/srv/pharmq-novnc/config.json'))['pharmacy_id'])" 2>/dev/null || echo "")
        CURRENT_PHARMACY_NAME=$(python3 -c "import json; print(json.load(open('/srv/pharmq-novnc/config.json'))['pharmacy_name'])" 2>/dev/null || echo "")

        echo -e "${CYAN}현재 설정:${NC}"
        echo "  약국 코드: ${CURRENT_PHARMACY_CODE:-'설정되지 않음'}"
        echo "  약국 이름: ${CURRENT_PHARMACY_NAME:-'설정되지 않음'}"
        echo ""
        echo "다음 중 선택하세요:"
        echo "  1) 상태 확인"
        echo "  2) 재설치 (기존 설정 백업 후 새로 설치)"
        echo "  3) 종료"
        echo ""
        echo -n "선택 [1/2/3]: "
        read -r choice </dev/tty

        case "$choice" in
            1)
                echo ""
                msg_info "서비스 상태 확인 중..."
                if [ -x "/srv/pharmq-novnc/scripts/check-status.sh" ]; then
                    /srv/pharmq-novnc/scripts/check-status.sh
                else
                    systemctl status pharmq-vnc-app.service pharmq-websockify-vnc1.service
                fi
                exit 0
                ;;
            2)
                msg_warn "재설치를 시작합니다..."
                echo ""
                ;;
            3)
                msg_ok "종료합니다."
                exit 0
                ;;
            *)
                msg_warn "잘못된 선택입니다. 종료합니다."
                exit 1
                ;;
        esac
    fi

    # VM 설정
    configure_vms

    # 약국 정보 입력
    get_pharmacy_info

    # 네트워크 확인
    check_network

    # 설치 진행
    install_packages
    install_pharmq_novnc
    create_config
    setup_systemd_services
    start_services

    # 방화벽 설정 (선택사항)
    configure_firewall

    # 테스트 및 완료
    test_installation
    print_completion
}

# 에러 핸들링
trap 'msg_error "스크립트 실행 중 오류가 발생했습니다."' ERR

# 스크립트 실행
main "$@"