Oxicloud/src/common/config.rs

use std::time::Duration;
use std::path::PathBuf;
use std::env;

/// Configuración de caché
#[derive(Debug, Clone)]
pub struct CacheConfig {
    /// TTL para entradas de archivos en caché (ms)
    pub file_ttl_ms: u64,
    /// TTL para entradas de directorios en caché (ms)
    pub directory_ttl_ms: u64,
    /// Máximo número de entradas en caché
    pub max_entries: usize,
}

impl Default for CacheConfig {
    fn default() -> Self {
        Self {
            file_ttl_ms: 60_000,     // 1 minuto
            directory_ttl_ms: 120_000, // 2 minutos
            max_entries: 10_000,      // 10,000 entradas
        }
    }
}

/// Configuración de timeouts para diferentes operaciones
#[derive(Debug, Clone)]
pub struct TimeoutConfig {
    /// Timeout para operaciones de archivo (ms)
    pub file_operation_ms: u64,
    /// Timeout para operaciones de directorio (ms)
    pub dir_operation_ms: u64,
    /// Timeout para adquisición de locks (ms)
    pub lock_acquisition_ms: u64,
    /// Timeout para operaciones de red (ms)
    pub network_operation_ms: u64,
}

impl Default for TimeoutConfig {
    fn default() -> Self {
        Self {
            file_operation_ms: 10000,    // 10 segundos
            dir_operation_ms: 30000,     // 30 segundos
            lock_acquisition_ms: 5000,   // 5 segundos
            network_operation_ms: 15000, // 15 segundos
        }
    }
}

impl TimeoutConfig {
    /// Obtiene un Duration para operaciones de archivo
    pub fn file_timeout(&self) -> Duration {
        Duration::from_millis(self.file_operation_ms)
    }

    /// Obtiene un Duration para operaciones de escritura de archivo
    pub fn file_write_timeout(&self) -> Duration {
        Duration::from_millis(self.file_operation_ms)
    }

    /// Obtiene un Duration para operaciones de lectura de archivo
    pub fn file_read_timeout(&self) -> Duration {
        Duration::from_millis(self.file_operation_ms)
    }

    /// Obtiene un Duration para operaciones de eliminación de archivo
    pub fn file_delete_timeout(&self) -> Duration {
        Duration::from_millis(self.file_operation_ms)
    }

    /// Obtiene un Duration para operaciones de directorio
    pub fn dir_timeout(&self) -> Duration {
        Duration::from_millis(self.dir_operation_ms)
    }

    /// Obtiene un Duration para adquisición de locks
    pub fn lock_timeout(&self) -> Duration {
        Duration::from_millis(self.lock_acquisition_ms)
    }

    /// Obtiene un Duration para operaciones de red
    pub fn network_timeout(&self) -> Duration {
        Duration::from_millis(self.network_operation_ms)
    }
}

/// Configuración para manejo de recursos grandes
#[derive(Debug, Clone)]
pub struct ResourceConfig {
    /// Umbral en MB para considerar un archivo como grande
    pub large_file_threshold_mb: u64,
    /// Umbral de entradas para considerar un directorio como grande
    pub large_dir_threshold_entries: usize,
    /// Tamaño de chunk para procesamiento de archivos grandes (bytes)
    pub chunk_size_bytes: usize,
    /// Límite de tamaño de archivo para cargar en memoria (MB)
    pub max_in_memory_file_size_mb: u64,
}

impl Default for ResourceConfig {
    fn default() -> Self {
        Self {
            large_file_threshold_mb: 100,       // 100 MB
            large_dir_threshold_entries: 1000,  // 1000 entradas
            chunk_size_bytes: 1024 * 1024,      // 1 MB
            max_in_memory_file_size_mb: 50,     // 50 MB
        }
    }
}

impl ResourceConfig {
    /// Convierte un tamaño en bytes a MB
    pub fn bytes_to_mb(&self, bytes: u64) -> u64 {
        bytes / (1024 * 1024)
    }

    /// Determina si un archivo es considerado grande
    pub fn is_large_file(&self, size_bytes: u64) -> bool {
        self.bytes_to_mb(size_bytes) >= self.large_file_threshold_mb
    }
    
    /// Determina si un archivo es suficientemente grande para procesamiento paralelo
    pub fn needs_parallel_processing(&self, size_bytes: u64, config: &ConcurrencyConfig) -> bool {
        self.bytes_to_mb(size_bytes) >= config.min_size_for_parallel_chunks_mb
    }

    /// Determina si un archivo puede cargarse completo en memoria
    pub fn can_load_in_memory(&self, size_bytes: u64) -> bool {
        self.bytes_to_mb(size_bytes) <= self.max_in_memory_file_size_mb
    }

    /// Determina si un directorio es considerado grande
    pub fn is_large_directory(&self, entry_count: usize) -> bool {
        entry_count >= self.large_dir_threshold_entries
    }
    
    /// Calcula el número de chunks para procesamiento paralelo
    pub fn calculate_optimal_chunks(&self, size_bytes: u64, config: &ConcurrencyConfig) -> usize {
        // Si el archivo no es suficientemente grande, retornar 1
        if !self.needs_parallel_processing(size_bytes, config) {
            return 1;
        }
        
        // Calcular el número de chunks basado en el tamaño
        let chunk_count = (size_bytes as usize + config.parallel_chunk_size_bytes - 1) 
                         / config.parallel_chunk_size_bytes;
                         
        // Limitar al máximo de chunks en paralelo
        chunk_count.min(config.max_parallel_chunks)
    }
    
    /// Calcula el tamaño óptimo de cada chunk para procesamiento paralelo
    pub fn calculate_chunk_size(&self, file_size: u64, chunk_count: usize) -> usize {
        if chunk_count <= 1 {
            return file_size as usize;
        }
        
        // Distribuir equitativamente el tamaño entre los chunks
        ((file_size as usize) + chunk_count - 1) / chunk_count
    }
}

/// Configuración para operaciones concurrentes
#[derive(Debug, Clone)]
pub struct ConcurrencyConfig {
    /// Máximo de tareas de archivo concurrentes
    pub max_concurrent_files: usize,
    /// Máximo de tareas de directorio concurrentes
    pub max_concurrent_dirs: usize,
    /// Máximo de operaciones de IO concurrentes
    pub max_concurrent_io: usize,
    /// Máximo de chunks para procesar en paralelo por archivo
    pub max_parallel_chunks: usize,
    /// Tamaño mínimo de archivo (MB) para aplicar procesamiento paralelo de chunks
    pub min_size_for_parallel_chunks_mb: u64,
    /// Tamaño de chunk para procesamiento paralelo (bytes)
    pub parallel_chunk_size_bytes: usize,
}

impl Default for ConcurrencyConfig {
    fn default() -> Self {
        Self {
            max_concurrent_files: 10,
            max_concurrent_dirs: 5,
            max_concurrent_io: 20,
            max_parallel_chunks: 8,
            min_size_for_parallel_chunks_mb: 200, // 200 MB
            parallel_chunk_size_bytes: 8 * 1024 * 1024, // 8 MB
        }
    }
}

/// Configuración de almacenamiento
#[derive(Debug, Clone)]
pub struct StorageConfig {
    /// Directorio raíz para el almacenamiento
    pub root_dir: String,
    /// Tamaño de chunk para procesamiento de archivos
    pub chunk_size: usize,
    /// Umbral para procesamiento paralelo
    pub parallel_threshold: usize,
    /// Días de retención para archivos en la papelera
    pub trash_retention_days: u32,
}

impl Default for StorageConfig {
    fn default() -> Self {
        Self {
            root_dir: "storage".to_string(),
            chunk_size: 1024 * 1024,      // 1 MB
            parallel_threshold: 100 * 1024 * 1024, // 100 MB
            trash_retention_days: 30,     // 30 días
        }
    }
}

/// Configuración de base de datos
#[derive(Debug, Clone)]
pub struct DatabaseConfig {
    pub connection_string: String,
    pub max_connections: u32,
    pub min_connections: u32,
    pub connect_timeout_secs: u64,
    pub idle_timeout_secs: u64,
    pub max_lifetime_secs: u64,
}

impl Default for DatabaseConfig {
    fn default() -> Self {
        Self {
            // Updated connection string with default credentials that PostgreSQL often uses
            connection_string: "postgres://postgres:postgres@localhost:5432/oxicloud".to_string(),
            max_connections: 20,
            min_connections: 5,
            connect_timeout_secs: 10,
            idle_timeout_secs: 300,
            max_lifetime_secs: 1800,
        }
    }
}

/// Configuración de autenticación
#[derive(Debug, Clone)]
pub struct AuthConfig {
    pub jwt_secret: String,
    pub access_token_expiry_secs: i64,
    pub refresh_token_expiry_secs: i64,
    pub hash_memory_cost: u32,
    pub hash_time_cost: u32,
}

impl Default for AuthConfig {
    fn default() -> Self {
        Self {
            jwt_secret: "ox1cl0ud-sup3r-s3cr3t-k3y-f0r-t0k3n-s1gn1ng".to_string(),
            access_token_expiry_secs: 3600, // 1 hora
            refresh_token_expiry_secs: 2592000, // 30 días
            hash_memory_cost: 65536, // 64MB
            hash_time_cost: 3,
        }
    }
}

/// Configuración de funcionalidades (feature flags)
#[derive(Debug, Clone)]
pub struct FeaturesConfig {
    pub enable_auth: bool,
    pub enable_user_storage_quotas: bool,
    pub enable_file_sharing: bool,
    pub enable_trash: bool,
    pub enable_search: bool,
}

impl Default for FeaturesConfig {
    fn default() -> Self {
        Self {
            enable_auth: true,  // Enable authentication by default
            enable_user_storage_quotas: false,
            enable_file_sharing: true,  // Enable file sharing by default
            enable_trash: true,  // Enable trash feature
            enable_search: true, // Enable search feature
        }
    }
}

/// Configuración global de la aplicación
#[derive(Debug, Clone)]
pub struct AppConfig {
    /// Ruta del directorio de almacenamiento
    pub storage_path: PathBuf,
    /// Ruta del directorio de archivos estáticos
    pub static_path: PathBuf,
    /// Puerto del servidor
    pub server_port: u16,
    /// Host del servidor
    pub server_host: String,
    /// Configuración de caché
    pub cache: CacheConfig,
    /// Configuración de timeouts
    pub timeouts: TimeoutConfig,
    /// Configuración de recursos
    pub resources: ResourceConfig,
    /// Configuración de concurrencia
    pub concurrency: ConcurrencyConfig,
    /// Configuración de almacenamiento
    pub storage: StorageConfig,
    /// Configuración de base de datos
    pub database: DatabaseConfig,
    /// Configuración de autenticación
    pub auth: AuthConfig,
    /// Configuración de funcionalidades
    pub features: FeaturesConfig,
}

impl Default for AppConfig {
    fn default() -> Self {
        Self {
            storage_path: PathBuf::from("./storage"),
            static_path: PathBuf::from("./static"),
            server_port: 8085,
            server_host: "127.0.0.1".to_string(),
            cache: CacheConfig::default(),
            timeouts: TimeoutConfig::default(),
            resources: ResourceConfig::default(),
            concurrency: ConcurrencyConfig::default(),
            storage: StorageConfig::default(),
            database: DatabaseConfig::default(),
            auth: AuthConfig::default(),
            features: FeaturesConfig::default(),
        }
    }
}

impl AppConfig {
    pub fn from_env() -> Self {
        let mut config = Self::default();
        
        // Usar variables de entorno para sobrescribir valores por defecto
        if let Ok(storage_path) = env::var("OXICLOUD_STORAGE_PATH") {
            config.storage_path = PathBuf::from(storage_path);
        }
            
        if let Ok(static_path) = env::var("OXICLOUD_STATIC_PATH") {
            config.static_path = PathBuf::from(static_path);
        }
            
        if let Ok(server_port) = env::var("OXICLOUD_SERVER_PORT") {
            if let Ok(port) = server_port.parse::<u16>() {
                config.server_port = port;
            }
        }
            
        if let Ok(server_host) = env::var("OXICLOUD_SERVER_HOST") {
            config.server_host = server_host;
        }
        
        // Configuración de Database
        if let Ok(connection_string) = env::var("OXICLOUD_DB_CONNECTION_STRING") {
            config.database.connection_string = connection_string;
        }
            
        if let Ok(max_connections) = env::var("OXICLOUD_DB_MAX_CONNECTIONS")
            .map(|v| v.parse::<u32>()) {
            if let Ok(val) = max_connections {
                config.database.max_connections = val;
            }
        }
            
        if let Ok(min_connections) = env::var("OXICLOUD_DB_MIN_CONNECTIONS")
            .map(|v| v.parse::<u32>()) {
            if let Ok(val) = min_connections {
                config.database.min_connections = val;
            }
        }
        
        // Configuración Auth
        if let Ok(jwt_secret) = env::var("OXICLOUD_JWT_SECRET") {
            config.auth.jwt_secret = jwt_secret;
        }
            
        if let Ok(access_token_expiry) = env::var("OXICLOUD_ACCESS_TOKEN_EXPIRY_SECS")
            .map(|v| v.parse::<i64>()) {
            if let Ok(val) = access_token_expiry {
                config.auth.access_token_expiry_secs = val;
            }
        }
            
        if let Ok(refresh_token_expiry) = env::var("OXICLOUD_REFRESH_TOKEN_EXPIRY_SECS")
            .map(|v| v.parse::<i64>()) {
            if let Ok(val) = refresh_token_expiry {
                config.auth.refresh_token_expiry_secs = val;
            }
        }
        
        // Feature flags
        if let Ok(enable_auth) = env::var("OXICLOUD_ENABLE_AUTH")
            .map(|v| v.parse::<bool>()) {
            if let Ok(val) = enable_auth {
                config.features.enable_auth = val;
            }
        }
        
        if let Ok(enable_user_storage_quotas) = env::var("OXICLOUD_ENABLE_USER_STORAGE_QUOTAS")
            .map(|v| v.parse::<bool>()) {
            if let Ok(val) = enable_user_storage_quotas {
                config.features.enable_user_storage_quotas = val;
            }
        }
        
        if let Ok(enable_file_sharing) = env::var("OXICLOUD_ENABLE_FILE_SHARING")
            .map(|v| v.parse::<bool>()) {
            if let Ok(val) = enable_file_sharing {
                config.features.enable_file_sharing = val;
            }
        }
        
        if let Ok(enable_trash) = env::var("OXICLOUD_ENABLE_TRASH")
            .map(|v| v.parse::<bool>()) {
            if let Ok(val) = enable_trash {
                config.features.enable_trash = val;
            }
        }
        
        if let Ok(enable_search) = env::var("OXICLOUD_ENABLE_SEARCH")
            .map(|v| v.parse::<bool>()) {
            if let Ok(val) = enable_search {
                config.features.enable_search = val;
            }
        }
        
        config
    }
    
    pub fn with_features(mut self, features: FeaturesConfig) -> Self {
        self.features = features;
        self
    }
    
    pub fn db_enabled(&self) -> bool {
        self.features.enable_auth
    }
    
    pub fn auth_enabled(&self) -> bool {
        self.features.enable_auth
    }
}

/// Obtenemos una configuración global por defecto
pub fn default_config() -> AppConfig {
    AppConfig::default()
}