diff --git a/VNC/pharmq-novnc-setup.sh b/VNC/pharmq-novnc-setup.sh index eb69822..c8ace74 100755 --- a/VNC/pharmq-novnc-setup.sh +++ b/VNC/pharmq-novnc-setup.sh @@ -1,7 +1,7 @@ #!/bin/bash # PharmQ noVNC Setup Script -# Proxmox Host에 noVNC over WebSocket 자동 설치 +# Ubuntu VM에서 실행하여 Proxmox Host의 VM VNC를 noVNC로 제공 # 사용법: curl -fsSL https://git.0bin.in/thug0bin/pve9-repo-fix/raw/branch/main/VNC/pharmq-novnc-setup.sh | bash set -euo pipefail @@ -41,38 +41,28 @@ print_header() { echo " PharmQ noVNC Setup Script v1.0" echo "═══════════════════════════════════════════════════════════════════" echo -e "${NC}" - echo "이 스크립트는 Proxmox Host에 noVNC over WebSocket을 자동으로 설치합니다." + echo "이 스크립트는 Ubuntu VM에서 Proxmox Host의 VM VNC를 웹으로 제공합니다." echo "각 약국의 VM VNC 화면을 웹 브라우저로 제공합니다." echo "" } -# Proxmox 버전 확인 -check_proxmox_version() { - msg_info "Proxmox VE 버전 확인 중..." +# 환경 감지 (Proxmox Host vs Ubuntu VM) +detect_environment() { + msg_info "실행 환경 감지 중..." - if ! command -v pveversion > /dev/null 2>&1; then - msg_warn "pveversion 명령을 찾을 수 없습니다. Proxmox 환경이 아닐 수 있습니다." + if command -v pveversion > /dev/null 2>&1; then + INSTALL_ENV="proxmox" + msg_warn "Proxmox VE Host에서 실행 중입니다." + msg_warn "이 스크립트는 Proxmox 내부의 Ubuntu VM에서 실행되어야 합니다." read -p "계속 진행하시겠습니까? [y/N]: " continue_anyway /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 버전 확인됨" } # 루트 권한 확인 @@ -82,16 +72,56 @@ check_root() { fi } -# VM 정보 가져오기 -get_vm_list() { - msg_info "실행 중인 VM 목록 가져오는 중..." +# Proxmox API를 통해 VM 목록 가져오기 +get_vm_list_from_api() { + local pve_host=$1 + local pve_user=$2 + local pve_password=$3 - if ! command -v qm > /dev/null 2>&1; then - msg_error "qm 명령어를 찾을 수 없습니다. Proxmox VE 환경이 아닙니다." + msg_info "Proxmox API를 통해 VM 목록 가져오는 중..." + + # API 로그인 + local ticket_response=$(curl -k -s -X POST \ + "https://${pve_host}:8006/api2/json/access/ticket" \ + -d "username=${pve_user}&password=${pve_password}") + + if [ -z "$ticket_response" ]; then + msg_error "Proxmox API 로그인 실패: 응답 없음" fi - # 실행 중인 VM 목록 - VM_LIST=$(qm list 2>/dev/null | tail -n +2 | awk '{print $1}' | sort -n) + local ticket=$(echo "$ticket_response" | python3 -c "import sys, json; print(json.load(sys.stdin)['data']['ticket'])" 2>/dev/null || echo "") + local csrf_token=$(echo "$ticket_response" | python3 -c "import sys, json; print(json.load(sys.stdin)['data']['CSRFPreventionToken'])" 2>/dev/null || echo "") + + if [ -z "$ticket" ]; then + msg_error "Proxmox API 로그인 실패: 티켓을 가져올 수 없습니다." + fi + + # 노드 목록 가져오기 + local nodes_response=$(curl -k -s -X GET \ + "https://${pve_host}:8006/api2/json/nodes" \ + --cookie "PVEAuthCookie=${ticket}") + + local node_name=$(echo "$nodes_response" | python3 -c "import sys, json; print(json.load(sys.stdin)['data'][0]['node'])" 2>/dev/null || echo "") + + if [ -z "$node_name" ]; then + msg_error "Proxmox 노드를 찾을 수 없습니다." + fi + + msg_ok "Proxmox 노드: $node_name" + + # VM 목록 가져오기 + local vms_response=$(curl -k -s -X GET \ + "https://${pve_host}:8006/api2/json/nodes/${node_name}/qemu" \ + --cookie "PVEAuthCookie=${ticket}") + + # VM 목록 파싱 (running 상태만) + VM_LIST=$(echo "$vms_response" | python3 -c " +import sys, json +vms = json.load(sys.stdin)['data'] +running_vms = [vm for vm in vms if vm.get('status') == 'running'] +for vm in running_vms: + print(f\"{vm['vmid']}:{vm.get('name', 'VM-' + str(vm['vmid']))}\") +" 2>/dev/null || echo "") if [ -z "$VM_LIST" ]; then msg_warn "실행 중인 VM이 없습니다." @@ -102,39 +132,100 @@ get_vm_list() { 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) +# VNC 포트 가져오기 (Proxmox API) +get_vnc_port_from_api() { + local pve_host=$1 + local pve_user=$2 + local pve_password=$3 + local vmid=$4 + + # API 로그인 + local ticket_response=$(curl -k -s -X POST \ + "https://${pve_host}:8006/api2/json/access/ticket" \ + -d "username=${pve_user}&password=${pve_password}") + + local ticket=$(echo "$ticket_response" | python3 -c "import sys, json; print(json.load(sys.stdin)['data']['ticket'])" 2>/dev/null || echo "") + + # 노드 가져오기 + local nodes_response=$(curl -k -s -X GET \ + "https://${pve_host}:8006/api2/json/nodes" \ + --cookie "PVEAuthCookie=${ticket}") + + local node_name=$(echo "$nodes_response" | python3 -c "import sys, json; print(json.load(sys.stdin)['data'][0]['node'])" 2>/dev/null || echo "") + + # VM 설정 가져오기 + local vm_config=$(curl -k -s -X GET \ + "https://${pve_host}:8006/api2/json/nodes/${node_name}/qemu/${vmid}/config" \ + --cookie "PVEAuthCookie=${ticket}") + + # VNC 포트 추출 + local vnc_port=$(echo "$vm_config" | python3 -c " +import sys, json +config = json.load(sys.stdin)['data'] +# VNC display 번호로부터 포트 계산: 5900 + display +for key, value in config.items(): + if 'vnc' in key or 'vga' in key: + if 'vnc' in str(value): + # vnc 설정에서 display 번호 추출 (예: 'vnc,:1') + parts = str(value).split(',') + for part in parts: + if part.startswith(':'): + display_num = part[1:] + if display_num.isdigit(): + print(5900 + int(display_num)) + sys.exit(0) +# VNC display를 찾지 못한 경우 기본값 +print(5900 + $vmid % 100) +" 2>/dev/null || echo "5900") + echo "$vnc_port" } # VM 선택 및 설정 configure_vms() { - echo -e "${CYAN}VM 설정을 시작합니다:${NC}" + echo -e "${CYAN}Proxmox API 접속 정보 입력:${NC}" echo "" - get_vm_list || msg_error "VM을 찾을 수 없습니다." + # Proxmox 호스트 IP (기본값: 192.168.0.200) + read -p "Proxmox 호스트 IP [192.168.0.200]: " PVE_HOST /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 + while IFS=: read -r vmid vm_name; do + echo " $idx) VM $vmid - $vm_name" + VM_INFO[$idx]="$vmid:$vm_name" + ((idx++)) + done <<< "$VM_LIST" if [ ${#VM_INFO[@]} -eq 0 ]; then - msg_error "VNC가 활성화된 VM이 없습니다." + msg_error "실행 중인 VM이 없습니다." fi echo "" @@ -145,13 +236,15 @@ configure_vms() { while true; do read -p "VM1 선택 (번호 입력): " vm1_choice /dev/null; then + if ! timeout 10 bash -c " /dev/null 2>&1; then + msg_warn "Proxmox 서버에 연결할 수 없습니다. 설정을 확인해주세요." + read -p "계속 진행하시겠습니까? [y/N]: " continue_anyway /dev/null | 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 + if ss -tlnp 2>/dev/null | 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 + if ss -tlnp 2>/dev/null | grep -q ":$WS_PORT2 "; then msg_ok "WebSocket 2가 포트 $WS_PORT2에서 실행 중입니다." else msg_warn "WebSocket 2가 포트 $WS_PORT2에서 실행되지 않습니다." @@ -535,7 +627,7 @@ test_installation() { if curl -s -f "http://localhost:$FLASK_PORT/health" > /dev/null 2>&1; then msg_ok "헬스체크 성공" else - msg_warn "헬스체크 실패" + msg_warn "헬스체크 실패 (서비스 시작 대기 중일 수 있습니다)" fi } @@ -550,6 +642,7 @@ print_completion() { echo " 약국 코드: $PHARMACY_CODE" echo " 약국 이름: $PHARMACY_NAME" echo " 설치 경로: /srv/pharmq-novnc" + echo " Proxmox 호스트: $PVE_HOST" echo "" echo -e "${CYAN}VM 설정:${NC}" echo " VM1: $VM1_ID - $VM1_NAME" @@ -595,7 +688,7 @@ main() { # 사전 검사 check_root - check_proxmox_version + detect_environment # 기존 설치 확인 if [ -f "/srv/pharmq-novnc/config.json" ] && \