rusty-stack 0.3.1

use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Stage {
    Welcome,
    HardwareDetect,
    Preflight,
    ComponentSelect,
    Configuration,
    Confirm,
    Installing,
    Complete,
    Benchmarks,
    Recovery,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Category {
    Foundation,
    Core,
    UiUx,
    Extension,
    Environment,
    Verification,
    Performance,
}

#[derive(Debug, Clone)]
pub struct Component {
    pub id: String,
    pub name: String,
    pub description: String,
    /// Script path relative to scripts_dir, or empty string for native Rust modules.
    ///
    /// For native Rust components, this field is empty and the installer
    /// dispatches to the corresponding Rust module via
    /// `installers::components::is_native_component()`.
    ///
    /// # Validation Assertions
    ///
    /// - **VAL-INSTALL-037**: Component.script no longer holds .sh filenames for ported components
    /// - **VAL-INSTALL-038**: state.rs supports native module routing
    /// - **VAL-INSTALL-040**: Verification/performance components use native routing
    pub script: String,
    pub category: Category,
    pub required: bool,
    pub selected: bool,
    pub installed: bool,
    pub progress: f32,
    pub estimate: String,
    /// Whether this component requires sudo to install
    pub needs_sudo: bool,
}

impl Component {
    /// Returns `true` if this component uses a native Rust installer
    /// (no shell script involved).
    ///
    /// # Validation Assertions
    ///
    /// - **VAL-INSTALL-038**: state.rs supports native module routing
    pub fn is_native(&self) -> bool {
        crate::installers::components::is_native_component(&self.id)
    }
}

#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct GPUInfo {
    pub model: String,
    pub driver: String,
    pub architecture: String,
    pub rocm_version: String,
    pub gpu_count: usize,
    pub memory_gb: f32,
    pub temperature_c: Option<f32>,
    pub power_watts: Option<f32>,
}

#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct SystemInfo {
    pub os: String,
    pub distribution: String,
    pub kernel: String,
    pub cpu_model: String,
    pub memory_gb: f32,
    pub storage_gb: f32,
    pub storage_available_gb: f32,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum PreflightStatus {
    Passed,
    Warning,
    Failed,
}

impl PreflightStatus {
    pub fn label(self) -> &'static str {
        match self {
            PreflightStatus::Passed => "passed",
            PreflightStatus::Warning => "warning",
            PreflightStatus::Failed => "failed",
        }
    }
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum PreflightType {
    Critical,
    Warning,
    Info,
}

#[derive(Debug, Clone)]
pub struct PreflightCheck {
    pub name: String,
    pub status: PreflightStatus,
    pub check_type: PreflightType,
    pub message: String,
    pub details: String,
    pub score: i32,
}

#[derive(Debug, Clone, Default)]
pub struct PreflightResult {
    pub passed: bool,
    pub summary: String,
    pub checks: Vec<PreflightCheck>,
    pub passed_count: usize,
    pub failed_count: usize,
    pub warning_count: usize,
    pub total_score: i32,
    pub can_continue: bool,
}

#[derive(Debug, Clone, Default)]
pub struct HardwareState {
    pub gpu: GPUInfo,
    pub system: SystemInfo,
    pub status: String,
    pub progress: f32,
}

#[derive(Debug, Clone, Default)]
pub struct InstallStatus {
    pub progress: f32,
    pub message: String,
    pub completed: bool,
}

pub fn default_components() -> Vec<Component> {
    use Category::*;

    vec![
        // Environment first
        Component {
            id: "permanent-env".into(),
            name: "Permanent ROCm Env".into(),
            description: "Unified permanent environment for Python 3.12".into(),
            script: String::new(), // Native Rust installer
            category: Environment,
            required: false,
            selected: true,
            installed: false,
            progress: 0.0,
            estimate: "1-2 min".into(),
            needs_sudo: true,
        },
        // Foundation components
        Component {
            id: "rocm".into(),
            name: "ROCm Platform".into(),
            description: "AMD ROCm GPU computing platform".into(),
            script: String::new(), // Native Rust installer
            category: Foundation,
            required: true,
            selected: true,
            installed: false,
            progress: 0.0,
            estimate: "30-45 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "pytorch".into(),
            name: "PyTorch with ROCm".into(),
            description: "PyTorch optimized for AMD GPUs".into(),
            script: String::new(), // Native Rust installer
            category: Foundation,
            required: true,
            selected: true,
            installed: false,
            progress: 0.0,
            estimate: "10-15 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "triton".into(),
            name: "Triton".into(),
            description: "Compiler for parallel programming".into(),
            script: String::new(), // Native Rust installer
            category: Foundation,
            required: true,
            selected: true,
            installed: false,
            progress: 0.0,
            estimate: "5-10 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "mpi4py".into(),
            name: "MPI4Py".into(),
            description: "MPI bindings for Python".into(),
            script: String::new(), // Native Rust installer
            category: Foundation,
            required: true,
            selected: true,
            installed: false,
            progress: 0.0,
            estimate: "3-5 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "deepspeed".into(),
            name: "DeepSpeed".into(),
            description: "Deep learning optimization library".into(),
            script: String::new(), // Native Rust installer
            category: Foundation,
            required: true,
            selected: true,
            installed: false,
            progress: 0.0,
            estimate: "8-12 min".into(),
            needs_sudo: true,
        },
        // Core components
        Component {
            id: "ml-stack-core".into(),
            name: "ML Stack Core".into(),
            description: "Core ML Stack components".into(),
            script: String::new(), // Native Rust installer
            category: Core,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "15-20 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "flash-attn".into(),
            name: "Flash Attention".into(),
            description: "Efficient attention computation".into(),
            script: String::new(), // Native Rust installer
            category: Core,
            required: false,
            selected: true,
            installed: false,
            progress: 0.0,
            estimate: "20-30 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "repair-stack".into(),
            name: "Repair ML Stack".into(),
            description: "Repair ML Stack installation".into(),
            script: String::new(), // Native Rust installer
            category: Core,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "5-10 min".into(),
            needs_sudo: true,
        },
        // Extension components
        Component {
            id: "megatron".into(),
            name: "Megatron-LM".into(),
            description: "Large-scale training framework".into(),
            script: String::new(), // Native Rust installer
            category: Extension,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "25-40 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "vllm".into(),
            name: "vLLM".into(),
            description: "High-throughput inference engine".into(),
            script: String::new(), // Native Rust installer
            category: Extension,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "15-25 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "aiter".into(),
            name: "AITER".into(),
            description: "AMD AITER optimization tooling".into(),
            script: String::new(), // Native Rust installer
            category: Extension,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "10-20 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "vllm-studio".into(),
            name: "vLLM Studio".into(),
            description: "Model lifecycle manager for vLLM/SGLang".into(),
            script: String::new(), // Native Rust installer
            category: UiUx,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "10-20 min".into(),
            needs_sudo: false,
        },
        Component {
            id: "comfyui".into(),
            name: "ComfyUI".into(),
            description: "Node-based AI image generation UI with ROCm support".into(),
            script: String::new(), // Native Rust installer
            category: UiUx,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "5-10 min".into(),
            needs_sudo: false,
        },
        Component {
            id: "textgen".into(),
            name: "text-generation-webui".into(),
            description: "LLM chat/inference web UI with ROCm support (oobabooga)".into(),
            script: String::new(), // Native Rust installer
            category: UiUx,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "10-20 min".into(),
            needs_sudo: false,
        },
        Component {
            id: "onnx".into(),
            name: "ONNX Runtime".into(),
            description: "Cross-platform inference accelerator".into(),
            script: String::new(), // Native Rust installer
            category: Extension,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "2-5 min".into(),
            needs_sudo: false,
        },
        Component {
            id: "bitsandbytes".into(),
            name: "BITSANDBYTES".into(),
            description: "Efficient quantization for deep learning".into(),
            script: String::new(), // Native Rust installer
            category: Extension,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "3-5 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "rocm-smi".into(),
            name: "ROCm SMI".into(),
            description: "System monitoring for AMD GPUs".into(),
            script: String::new(), // Native Rust installer
            category: Extension,
            required: false,
            selected: true,
            installed: false,
            progress: 0.0,
            estimate: "2-3 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "migraphx".into(),
            name: "MIGraphX".into(),
            description: "AMD graph inference engine".into(),
            script: String::new(), // Native Rust installer
            category: Extension,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "10-15 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "pytorch-profiler".into(),
            name: "PyTorch Profiler".into(),
            description: "Performance analysis for PyTorch".into(),
            script: String::new(), // Native Rust installer
            category: Extension,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "3-5 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "wandb".into(),
            name: "Weights & Biases".into(),
            description: "Experiment tracking and visualization".into(),
            script: String::new(), // Native Rust installer
            category: Extension,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "5-8 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "llama-cpp".into(),
            name: "llama.cpp (HIP/ROCm)".into(),
            description: "llama.cpp with HIP/ROCm GPU acceleration".into(),
            script: String::new(), // Native Rust installer
            category: Extension,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "10-15 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "fastvideo".into(),
            name: "FastVideo".into(),
            description: "Video generation framework with ROCm gfx11 optimizations".into(),
            script: String::new(), // Native Rust installer
            category: Extension,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "10-20 min".into(),
            needs_sudo: false,
        },
        // Verification components (native Rust via verification module)
        Component {
            id: "verify-basic".into(),
            name: "Verify Installation".into(),
            description: "Basic verification".into(),
            script: String::new(), // Native Rust verification
            category: Verification,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "2-5 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "verify-enhanced".into(),
            name: "Enhanced Verify Installation".into(),
            description: "Advanced verification".into(),
            script: String::new(), // Native Rust verification
            category: Verification,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "5-10 min".into(),
            needs_sudo: true,
        },
        Component {
            id: "verify-build".into(),
            name: "Verify and Build".into(),
            description: "Verify + build components".into(),
            script: String::new(), // Native Rust verification
            category: Verification,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "10-15 min".into(),
            needs_sudo: true,
        },
        // Performance components (native Rust via benchmark_runners module)
        Component {
            id: "mlperf-inference".into(),
            name: "MLPerf Inference".into(),
            description: "MLPerf benchmark suite for inference performance".into(),
            script: String::new(), // Native Rust benchmark
            category: Performance,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "30-60 min".into(),
            needs_sudo: false,
        },
        Component {
            id: "rocm-benchmarks".into(),
            name: "ROCm Benchmarks".into(),
            description: "ROCm-specific performance benchmarks".into(),
            script: String::new(), // Native Rust benchmark
            category: Performance,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "20-40 min".into(),
            needs_sudo: false,
        },
        Component {
            id: "gpu-memory-bandwidth".into(),
            name: "GPU Memory Bandwidth".into(),
            description: "Memory bandwidth performance testing".into(),
            script: String::new(), // Native Rust benchmark
            category: Performance,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "10-15 min".into(),
            needs_sudo: false,
        },
        Component {
            id: "rocm-smi-bench".into(),
            name: "ROCm SMI Benchmarks".into(),
            description: "ROCm SMI performance monitoring".into(),
            script: String::new(), // Native Rust benchmark
            category: Performance,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "15-25 min".into(),
            needs_sudo: false,
        },
        Component {
            id: "vllm-performance".into(),
            name: "vLLM Performance".into(),
            description: "High-throughput vLLM inference benchmark".into(),
            script: String::new(), // Native Rust benchmark
            category: Performance,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "15-25 min".into(),
            needs_sudo: false,
        },
        Component {
            id: "deepspeed-performance".into(),
            name: "DeepSpeed Performance".into(),
            description: "DeepSpeed ZeRO optimization throughput benchmark".into(),
            script: String::new(), // Native Rust benchmark
            category: Performance,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "10-20 min".into(),
            needs_sudo: false,
        },
        Component {
            id: "megatron-performance".into(),
            name: "Megatron-LM Performance".into(),
            description: "Megatron-LM import and throughput benchmark".into(),
            script: String::new(), // Native Rust benchmark
            category: Performance,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "10-20 min".into(),
            needs_sudo: false,
        },
        Component {
            id: "all-benchmarks".into(),
            name: "Full Suite Benchmark".into(),
            description: "Run all post-installation performance tests".into(),
            script: String::new(), // Native Rust benchmark
            category: Performance,
            required: false,
            selected: false,
            installed: false,
            progress: 0.0,
            estimate: "45-90 min".into(),
            needs_sudo: false,
        },
    ]
}