install_dcv_sdk.py

#!/usr/bin/env python3
"""
Automatically download and install Amazon DCV Web Client SDK
"""

import os
import zipfile
import shutil
from pathlib import Path
import urllib.request
from rich.console import Console
from rich.progress import Progress, SpinnerColumn, TextColumn, BarColumn, DownloadColumn, TransferSpeedColumn

console = Console()

DCV_SDK_URL = "https://d1uj6qtbmh3dt5.cloudfront.net/webclientsdk/nice-dcv-web-client-sdk-1.9.100-952.zip"
DCV_SDK_VERSION = "1.9.100-952"

def download_dcv_sdk(download_dir: Path) -> Path:
    """Download DCV SDK zip file."""
    zip_path = download_dir / f"dcv-sdk-{DCV_SDK_VERSION}.zip"
    
    if zip_path.exists():
        console.print(f"[yellow]⚠️  DCV SDK zip already exists: {zip_path}[/yellow]")
        response = console.input("[dim]Re-download? (y/N): [/dim]")
        if response.lower() != 'y':
            return zip_path
    
    console.print(f"[cyan]📥 Downloading DCV SDK {DCV_SDK_VERSION}...[/cyan]")
    console.print(f"[dim]URL: {DCV_SDK_URL}[/dim]")
    
    try:
        with Progress(
            SpinnerColumn(),
            TextColumn("[progress.description]{task.description}"),
            BarColumn(),
            DownloadColumn(),
            TransferSpeedColumn(),
            console=console
        ) as progress:
            task = progress.add_task("[cyan]Downloading...", total=None)
            
            def update_progress(block_num, block_size, total_size):
                if total_size > 0:
                    progress.update(task, total=total_size, completed=block_num * block_size)
            
            urllib.request.urlretrieve(DCV_SDK_URL, zip_path, reporthook=update_progress)
        
        console.print(f"[green]✅ Downloaded: {zip_path}[/green]")
        return zip_path
    except Exception as e:
        console.print(f"[red]❌ Failed to download DCV SDK: {e}[/red]")
        raise

def extract_dcv_sdk(zip_path: Path, extract_dir: Path) -> Path:
    """Extract DCV SDK zip file."""
    console.print(f"[cyan]📦 Extracting DCV SDK...[/cyan]")
    
    extract_dir.mkdir(parents=True, exist_ok=True)
    
    try:
        with zipfile.ZipFile(zip_path, 'r') as zip_ref:
            # Get total size for progress
            total_size = sum(info.file_size for info in zip_ref.infolist())
            extracted_size = 0
            
            with Progress(
                SpinnerColumn(),
                TextColumn("[progress.description]{task.description}"),
                BarColumn(),
                console=console
            ) as progress:
                task = progress.add_task("[cyan]Extracting...", total=total_size)
                
                for info in zip_ref.infolist():
                    zip_ref.extract(info, extract_dir)
                    extracted_size += info.file_size
                    progress.update(task, completed=extracted_size)
        
        console.print(f"[green]✅ Extracted to: {extract_dir}[/green]")
        
        # Find all candidate directories that contain dcv.js
        candidates = [item.parent for item in extract_dir.rglob("dcv.js")]
        
        preferred_dir = None
        if candidates:
            # Prefer UMD build if available
            for candidate in candidates:
                lower = str(candidate).lower()
                if "dcvjs-umd" in lower or lower.endswith("umd"):
                    preferred_dir = candidate
                    break
            if not preferred_dir:
                preferred_dir = candidates[0]
        
        if preferred_dir:
            if preferred_dir not in candidates[:1]:
                console.print(f"[cyan]ℹ️ Using preferred DCV UMD build at: {preferred_dir}[/cyan]")
            else:
                console.print(f"[green]✅ Found dcv.js in: {preferred_dir}[/green]")
            return preferred_dir
        
        # Fallback: try common directory names explicitly
        possible_dirs = [
            extract_dir / "dcvjs-umd",
            extract_dir / "dcvjs",
            extract_dir / "nice-dcv-web-client-sdk-1.9.100-952" / "dcvjs-umd",
            extract_dir / "nice-dcv-web-client-sdk-1.9.100-952" / "dcvjs",
        ]
        
        for possible_dir in possible_dirs:
            if possible_dir.exists() and (possible_dir / "dcv.js").exists():
                console.print(f"[green]✅ Found dcv.js in: {possible_dir}[/green]")
                return possible_dir
        
        # List extracted contents for debugging
        console.print(f"[yellow]⚠️  Could not find dcv.js. Extracted contents:[/yellow]")
        for item in extract_dir.iterdir():
            if item.is_dir():
                console.print(f"[dim]  Directory: {item.name}[/dim]")
                # List first level subdirectories
                try:
                    for subitem in item.iterdir():
                        if subitem.is_dir():
                            console.print(f"[dim]    └── {subitem.name}/[/dim]")
                        elif subitem.name == "dcv.js":
                            console.print(f"[green]    └── {subitem.name} (FOUND!)[/green]")
                            return subitem.parent
                except:
                    pass
        
        raise FileNotFoundError("dcv.js not found in extracted archive")
        
    except Exception as e:
        console.print(f"[red]❌ Failed to extract DCV SDK: {e}[/red]")
        raise

def install_dcv_sdk(target_dir: Path, source_dir: Path):
    """Install DCV SDK files to target directory."""
    console.print(f"[cyan]📋 Installing DCV SDK to: {target_dir}[/cyan]")
    console.print(f"[dim]Source directory: {source_dir}[/dim]")
    
    target_dir.mkdir(parents=True, exist_ok=True)
    
    # Required files and directories
    required_files = [
        "dcv.js",
        "dcv/broadwayh264decoder-worker.js",
        "dcv/jsmpegdecoder-worker.js",
        "dcv/lz4decoder-worker.js",
        "dcv/microphoneprocessor.js"
    ]
    
    # Copy dcv.js
    dcv_js_source = source_dir / "dcv.js"
    if not dcv_js_source.exists():
        console.print(f"[red]❌ dcv.js not found in {source_dir}[/red]")
        console.print(f"[yellow]Listing contents of {source_dir}:[/yellow]")
        try:
            for item in source_dir.iterdir():
                if item.is_file():
                    console.print(f"[dim]  File: {item.name}[/dim]")
                elif item.is_dir():
                    console.print(f"[dim]  Directory: {item.name}/[/dim]")
        except Exception as e:
            console.print(f"[dim]  Error listing: {e}[/dim]")
        raise FileNotFoundError(f"dcv.js not found in {source_dir}")
    
    shutil.copy2(dcv_js_source, target_dir / "dcv.js")
    console.print(f"[green]✅ Copied dcv.js[/green]")
    
    # Copy dcv directory
    dcv_dir_source = source_dir / "dcv"
    dcv_dir_target = target_dir / "dcv"
    
    if dcv_dir_source.exists():
        if dcv_dir_target.exists():
            shutil.rmtree(dcv_dir_target)
        shutil.copytree(dcv_dir_source, dcv_dir_target)
        console.print(f"[green]✅ Copied dcv/ directory[/green]")
    else:
        console.print(f"[yellow]⚠️  dcv/ directory not found in {source_dir}[/yellow]")
        console.print(f"[dim]Checking for dcv directory in subdirectories...[/dim]")
        # Try to find dcv directory recursively
        found_dcv = False
        for item in source_dir.rglob("dcv"):
            if item.is_dir() and item.name == "dcv":
                console.print(f"[green]✅ Found dcv/ directory at: {item}[/green]")
                if dcv_dir_target.exists():
                    shutil.rmtree(dcv_dir_target)
                shutil.copytree(item, dcv_dir_target)
                console.print(f"[green]✅ Copied dcv/ directory[/green]")
                found_dcv = True
                break
        if not found_dcv:
            console.print(f"[yellow]⚠️  dcv/ directory not found, creating empty directory...[/yellow]")
            dcv_dir_target.mkdir(parents=True, exist_ok=True)
    
    # Copy lib directory if it exists
    lib_dir_source = source_dir / "lib"
    lib_dir_target = target_dir / "lib"
    
    if lib_dir_source.exists():
        if lib_dir_target.exists():
            shutil.rmtree(lib_dir_target)
        shutil.copytree(lib_dir_source, lib_dir_target)
        console.print(f"[green]✅ Copied lib/ directory[/green]")
    else:
        # Try to find lib directory recursively
        for item in source_dir.rglob("lib"):
            if item.is_dir() and item.name == "lib":
                console.print(f"[green]✅ Found lib/ directory at: {item}[/green]")
                if lib_dir_target.exists():
                    shutil.rmtree(lib_dir_target)
                shutil.copytree(item, lib_dir_target)
                console.print(f"[green]✅ Copied lib/ directory[/green]")
                break
    
    # Verify installation
    missing_files = []
    for file_path in required_files:
        full_path = target_dir / file_path
        if not full_path.exists():
            missing_files.append(file_path)
    
    if missing_files:
        console.print(f"[yellow]⚠️  Some files are missing: {', '.join(missing_files)}[/yellow]")
        console.print("[dim]The viewer may still work, but some features might not be available.[/dim]")
    else:
        console.print(f"[green]✅ All required files installed successfully![/green]")
    
    # Check file size
    dcv_js_size = (target_dir / "dcv.js").stat().st_size
    if dcv_js_size < 10000:
        console.print(f"[yellow]⚠️  dcv.js seems too small ({dcv_js_size} bytes)[/yellow]")
        console.print("[dim]Expected size > 100KB. Please verify the installation.[/dim]")
    else:
        console.print(f"[green]✅ dcv.js size: {dcv_js_size:,} bytes[/green]")

def main():
    """Main installation function."""
    console.print("\n[bold cyan]Amazon DCV Web Client SDK Installer[/bold cyan]\n")
    
    # Determine target directory (same directory as browser_viewer.py)
    script_dir = Path(__file__).parent
    target_dir = script_dir / "static" / "dcvjs"
    
    # Create temp directory for download and extraction
    temp_dir = script_dir / ".dcv_temp"
    temp_dir.mkdir(exist_ok=True)
    
    try:
        # Download
        zip_path = download_dcv_sdk(temp_dir)
        
        # Extract
        extracted_dir = extract_dcv_sdk(zip_path, temp_dir / "extracted")
        
        # Install
        install_dcv_sdk(target_dir, extracted_dir)
        
        console.print(f"\n[bold green]🎉 DCV SDK installation complete![/bold green]")
        console.print(f"[dim]Installed to: {target_dir}[/dim]\n")
        
        # Cleanup
        console.print("[cyan]🧹 Cleaning up temporary files...[/cyan]")
        if temp_dir.exists():
            shutil.rmtree(temp_dir)
        console.print("[green]✅ Cleanup complete[/green]\n")
        
    except Exception as e:
        console.print(f"\n[red]❌ Installation failed: {e}[/red]\n")
        console.print("[yellow]You can manually download and install DCV SDK:[/yellow]")
        console.print(f"[dim]1. Download from: {DCV_SDK_URL}[/dim]")
        console.print(f"[dim]2. Extract and copy dcvjs-umd/* to: {target_dir}[/dim]\n")
        return 1
    
    return 0

if __name__ == "__main__":
    import sys
    sys.exit(main())