Wherry/src/commands.rs

//! Tauri command surface — thin wrappers over domain/fs/protocols/storage/transfer.
//! See docs/TAURI_BACKEND_API.md for the design this implements.
use std::sync::Arc;
use std::sync::atomic::{AtomicU32, Ordering};

use tauri::Manager;
use tauri::State;

use crate::domain::{
    ConnectionParams, FileEntry, Protocol, Site, TaskState, TransferKind, TransferTask,
};
use crate::error::AppError;
use crate::fs::remote::RemoteRegistry;
use crate::protocols::{
    RemoteFs,
    ftp::{FtpClient, FtpsClient},
    sftp::SftpClient,
};
use crate::settings;
use crate::storage::{self, HistoryRow};
use crate::transfers::queue::TransferQueue;
use crate::window::WindowState;

pub struct AppState {
    pub db: Arc<std::sync::Mutex<rusqlite::Connection>>,
    pub registry: Arc<RemoteRegistry>,
    pub queue: TransferQueue,
    pub max_concurrent: Arc<AtomicU32>,
    pub auto_clear_secs: Arc<AtomicU32>,
}

// --- Connections ---------------------------------------------------------

#[tauri::command]
pub async fn connect(
    state: State<'_, AppState>,
    params: ConnectionParams,
) -> Result<String, AppError> {
    let registry = state.registry.clone();
    let db = state.db.clone();

    // Пароль приходит с фронтенда и хранится в БД (в sites.password).
    // Для истории используем канонический conn_id, чтобы повторные
    // подключения не плодили записи.
    let canonical_id = db
        .lock()
        .ok()
        .and_then(|conn| {
            storage::find_history_conn_id(&conn, &params.host, params.port, &params.username)
                .ok()
                .flatten()
        })
        .unwrap_or_else(|| params.id.clone());

    let fs: Arc<dyn RemoteFs> = match params.protocol {
        Protocol::Sftp => {
            if let Some(key_path) = &params.key_path {
                Arc::new(SftpClient::connect_key(
                    &params.host,
                    params.port,
                    &params.username,
                    key_path,
                    params.password.as_deref(),
                )?)
            } else if let Some(password) = &params.password {
                Arc::new(SftpClient::connect_password(
                    &params.host,
                    params.port,
                    &params.username,
                    password,
                )?)
            } else {
                Arc::new(SftpClient::connect_auto(
                    &params.host,
                    params.port,
                    &params.username,
                )?)
            }
        }
        Protocol::Ftp => {
            let password = params.password.clone().ok_or_else(|| {
                AppError::InvalidInput("Password is required for FTP connections.".into())
            })?;
            Arc::new(
                FtpClient::connect(&params.host, params.port, &params.username, &password).await?,
            )
        }
        Protocol::Ftps => {
            let password = params.password.clone().ok_or_else(|| {
                AppError::InvalidInput("Password is required for FTPS connections.".into())
            })?;
            Arc::new(
                FtpsClient::connect(&params.host, params.port, &params.username, &password).await?,
            )
        }
    };

    registry.insert(params.id.clone(), fs);

    if let Ok(conn) = db.lock() {
        let _ = storage::add_history_entry(
            &conn,
            &params.host,
            params.port,
            &params.username,
            &canonical_id,
            &params.protocol,
            params.key_path.as_deref(),
        );
    }

    Ok(params.id)
}

#[tauri::command]
pub fn disconnect(state: State<'_, AppState>, connection_id: String) {
    state.registry.remove(&connection_id);
}

// --- Remote file ops -------------------------------------------------------

fn get_remote(state: &AppState, connection_id: &str) -> Result<Arc<dyn RemoteFs>, AppError> {
    state
        .registry
        .get(connection_id)
        .ok_or_else(|| AppError::NotFound("connection not found".into()))
}

#[tauri::command]
pub async fn remote_list(
    state: State<'_, AppState>,
    connection_id: String,
    path: String,
) -> Result<Vec<FileEntry>, AppError> {
    Ok(get_remote(&state, &connection_id)?.list(&path).await?)
}

#[tauri::command]
pub async fn remote_stat(
    state: State<'_, AppState>,
    connection_id: String,
    path: String,
) -> Result<FileEntry, AppError> {
    Ok(get_remote(&state, &connection_id)?.stat(&path).await?)
}

#[tauri::command]
pub async fn remote_mkdir(
    state: State<'_, AppState>,
    connection_id: String,
    path: String,
) -> Result<(), AppError> {
    Ok(get_remote(&state, &connection_id)?.mkdir(&path).await?)
}

#[tauri::command]
pub async fn remote_rename(
    state: State<'_, AppState>,
    connection_id: String,
    from: String,
    to: String,
) -> Result<(), AppError> {
    Ok(get_remote(&state, &connection_id)?
        .rename(&from, &to)
        .await?)
}

#[tauri::command]
pub async fn remote_delete(
    state: State<'_, AppState>,
    connection_id: String,
    path: String,
) -> Result<(), AppError> {
    Ok(get_remote(&state, &connection_id)?.delete(&path).await?)
}

// --- Local file ops --------------------------------------------------------

#[tauri::command]
pub fn local_list(path: String) -> Result<Vec<FileEntry>, AppError> {
    Ok(crate::fs::local::list(&path)?)
}

#[tauri::command]
pub fn local_home_dir() -> String {
    crate::fs::local::home_dir()
}

#[tauri::command]
pub fn local_mkdir(path: String) -> Result<(), AppError> {
    Ok(crate::fs::local::mkdir(&path)?)
}

#[tauri::command]
pub fn local_rename(from: String, to: String) -> Result<(), AppError> {
    Ok(crate::fs::local::rename(&from, &to)?)
}

#[tauri::command]
pub fn local_delete(path: String) -> Result<(), AppError> {
    Ok(crate::fs::local::delete(&path)?)
}

#[tauri::command]
pub fn local_open(path: String) -> Result<(), AppError> {
    Ok(crate::fs::local::open(&path)?)
}

#[tauri::command]
pub fn local_move_into(src_path: String, dest_dir: String) -> Result<(), AppError> {
    Ok(crate::fs::local::move_into(&src_path, &dest_dir)?)
}

// --- Transfers --------------------------------------------------------------

#[tauri::command]
#[allow(clippy::too_many_arguments)]
pub fn enqueue_transfer(
    state: State<'_, AppState>,
    kind: TransferKind,
    connection_id: String,
    local_path: String,
    remote_path: String,
    file_name: String,
    total_bytes: u64,
) -> String {
    let task = TransferTask::new(
        kind,
        connection_id,
        local_path,
        remote_path,
        file_name,
        total_bytes,
    );
    let id = task.id.clone();
    state.queue.push(task);
    id
}

#[tauri::command]
pub fn list_tasks(state: State<'_, AppState>) -> Vec<TransferTask> {
    state.queue.all()
}

#[tauri::command]
pub fn pause_task(state: State<'_, AppState>, id: String) {
    state.queue.update_state(&id, TaskState::Paused);
}

#[tauri::command]
pub fn resume_task(state: State<'_, AppState>, id: String) {
    state.queue.update_state(&id, TaskState::Queued);
}

#[tauri::command]
pub fn cancel_task(state: State<'_, AppState>, id: String) {
    state.queue.update_state(&id, TaskState::Cancelled);
}

#[tauri::command]
pub fn remove_task(state: State<'_, AppState>, id: String) {
    state.queue.remove(&id);
}

#[tauri::command]
pub fn set_max_concurrent(state: State<'_, AppState>, n: u32) {
    state.max_concurrent.store(n.max(1), Ordering::Relaxed);
    if let Ok(conn) = state.db.lock() {
        settings::set_u32(&conn, "max_concurrent_transfers", n.max(1));
    }
}

// --- Sites / bookmarks / history / settings ---------------------------------

#[tauri::command]
pub fn list_sites(state: State<'_, AppState>) -> Result<Vec<Site>, AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    Ok(storage::get_sites(&conn)?)
}

#[tauri::command]
pub fn save_site(state: State<'_, AppState>, site: Site) -> Result<(), AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    Ok(storage::save_site(&conn, &site)?)
}

#[tauri::command]
pub fn delete_site(state: State<'_, AppState>, id: String) -> Result<(), AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    Ok(storage::delete_site(&conn, &id)?)
}

#[tauri::command]
pub fn list_bookmarks(state: State<'_, AppState>) -> Result<Vec<(i64, String, String)>, AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    Ok(settings::get_bookmarks(&conn)?)
}

#[tauri::command]
pub fn add_bookmark(
    state: State<'_, AppState>,
    name: String,
    path: String,
) -> Result<i64, AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    Ok(settings::add_bookmark(&conn, &name, &path)?)
}

#[tauri::command]
pub fn remove_bookmark(state: State<'_, AppState>, id: i64) -> Result<(), AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    Ok(settings::remove_bookmark(&conn, id)?)
}

#[tauri::command]
pub fn list_history(state: State<'_, AppState>) -> Result<Vec<HistoryRow>, AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    Ok(storage::get_history(&conn)?)
}

#[tauri::command]
pub fn clear_history(state: State<'_, AppState>) -> Result<(), AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    Ok(storage::clear_history(&conn)?)
}

#[tauri::command]
pub fn find_history_conn_id(
    state: State<'_, AppState>,
    host: String,
    port: u16,
    username: String,
) -> Result<Option<String>, AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    Ok(storage::find_history_conn_id(
        &conn, &host, port, &username,
    )?)
}

#[tauri::command]
pub fn get_pref(state: State<'_, AppState>, key: String) -> Result<Option<String>, AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    Ok(settings::get_setting(&conn, &key))
}

#[tauri::command]
pub fn set_pref(state: State<'_, AppState>, key: String, value: String) -> Result<(), AppError> {
    if key == "auto_clear_completed_secs"
        && let Ok(secs) = value.parse::<u32>()
    {
        state.auto_clear_secs.store(secs, Ordering::Relaxed);
    }
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    Ok(settings::set_setting(&conn, &key, &value)?)
}

/// Удалить пароль из сохранённого сайта (очистить поле password в БД).
#[tauri::command]
pub fn delete_password(state: State<'_, AppState>, site_id: String) -> Result<(), AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    conn.execute(
        "UPDATE sites SET password = NULL WHERE id = ?1",
        rusqlite::params![site_id],
    )
    .map_err(|e| AppError::Internal(e.to_string()))?;
    Ok(())
}

/// The folder holding the sqlite db (Settings → About "reveal in folder").
#[tauri::command]
pub fn app_data_dir() -> String {
    dirs::data_dir()
        .unwrap_or_else(|| std::path::PathBuf::from("."))
        .join("wherry")
        .to_string_lossy()
        .to_string()
}

#[derive(serde::Serialize)]
#[serde(rename_all = "camelCase")]
pub struct PlatformInfo {
    pub os: &'static str,
    /// Linux only: XDG_CURRENT_DESKTOP, lowercased ("kde", "gnome", "hyprland", …).
    pub desktop: Option<String>,
    /// Linux only: "wayland" | "x11".
    pub session: Option<String>,
}

/// Из webview на Linux не различить KDE/GNOME/Hyprland — окружение видно
/// только процессу; фронтенд вешает результат как data-атрибуты на <html>.
#[tauri::command]
pub fn platform_info() -> PlatformInfo {
    let os = std::env::consts::OS;
    let (desktop, session) = if os == "linux" {
        (
            std::env::var("XDG_CURRENT_DESKTOP")
                .ok()
                .map(|d| d.to_lowercase()),
            std::env::var("XDG_SESSION_TYPE").ok(),
        )
    } else {
        (None, None)
    };
    PlatformInfo {
        os,
        desktop,
        session,
    }
}

// --- Multi-window support ---------------------------------------------------

#[tauri::command]
pub fn save_window_state(
    state: State<'_, AppState>,
    window_state: WindowState,
) -> Result<(), AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    let key = format!("window_state_{}", window_state.label);
    let json =
        serde_json::to_string(&window_state).map_err(|e| AppError::Internal(e.to_string()))?;
    settings::set_setting(&conn, &key, &json)?;
    Ok(())
}

#[tauri::command]
pub fn load_window_state(
    state: State<'_, AppState>,
    label: String,
) -> Result<Option<WindowState>, AppError> {
    let conn = state
        .db
        .lock()
        .map_err(|e| AppError::Internal(e.to_string()))?;
    let key = format!("window_state_{}", label);
    match settings::get_setting(&conn, &key) {
        Some(json) => {
            let ws: WindowState =
                serde_json::from_str(&json).map_err(|e| AppError::Internal(e.to_string()))?;
            Ok(Some(ws))
        }
        None => Ok(None),
    }
}

#[tauri::command]
pub async fn new_window(
    app_handle: tauri::AppHandle,
    state: State<'_, AppState>,
) -> Result<(), String> {
    let label = format!("browser-{}", uuid::Uuid::new_v4());
    let window =
        tauri::WebviewWindowBuilder::new(&app_handle, &label, tauri::WebviewUrl::App("/".into()))
            .title("Wherry")
            .inner_size(1280.0, 800.0)
            .min_inner_size(900.0, 600.0)
            .center()
            .build()
            .map_err(|e| e.to_string())?;

    let db = state.db.clone();
    let app_handle = app_handle.clone();
    let label_clone = label.clone();
    window.on_window_event(move |event| {
        if let tauri::WindowEvent::CloseRequested { .. } = event
            && let Some(win) = app_handle.get_webview_window(&label_clone)
        {
            save_window_state_internal(&win, &db);
        }
    });

    Ok(())
}

pub(crate) fn save_window_state_internal(
    window: &tauri::WebviewWindow,
    db: &Arc<std::sync::Mutex<rusqlite::Connection>>,
) {
    let label = window.label().to_string();
    let pos = window.outer_position().ok();
    let size = window.outer_size().unwrap_or(tauri::PhysicalSize {
        width: 1280,
        height: 800,
    });
    let maximized = window.is_maximized().unwrap_or(false);

    let ws = WindowState {
        label: label.clone(),
        x: pos.map(|p| p.x),
        y: pos.map(|p| p.y),
        width: size.width,
        height: size.height,
        maximized,
    };

    if let Ok(conn) = db.lock() {
        let key = format!("window_state_{}", label);
        if let Ok(json) = serde_json::to_string(&ws) {
            let _ = settings::set_setting(&conn, &key, &json);
        }
    }
}