nyaterm/src-tauri/src/lib.rs

#![recursion_limit = "256"]

//! Tauri app entry point: state construction, plugin registration, and command routing.

mod app;
mod cmd;
mod config;
mod core;
mod error;
mod external_open;
mod observability;
mod platform;
mod portable_updater;
mod runtime;
mod storage;
mod tray;
mod utils;
mod window_state;

use std::sync::Arc;
use tauri::Manager;

use crate::cmd::app::AppLockState;
use crate::cmd::docker::DockerSudoManager;
use crate::core::ai::AgentApprovalManager;
use crate::core::mcp::McpManager;
use crate::core::monitoring::stats::RemoteStatsSampler;
use crate::core::sftp::TransferDuplicateManager;
use crate::core::ssh::{
    HostKeyVerifyManager, PendingAuthManager, PendingSshAgentAuthManager, PendingSshAuthManager,
    TunnelManager,
};
use crate::core::{
    CloudSyncManager, QuickCommandsStore, RdpSessionManager, RecordingManager, SessionManager,
    VncSessionManager,
};

#[cfg_attr(mobile, tauri::mobile_entry_point)]
pub fn run() {
    portable_updater::schedule_cleanup_from_environment();
    let runtime = runtime::resolve().expect("failed to resolve runtime paths");
    runtime::prepare_webview_environment(&runtime);

    let session_manager = Arc::new(SessionManager::new());
    let rdp_session_manager = Arc::new(RdpSessionManager::new());
    let vnc_session_manager = Arc::new(VncSessionManager::new());
    let tunnel_manager = Arc::new(TunnelManager::new());
    let recording_manager = Arc::new(RecordingManager::new());
    let pending_auth_manager = Arc::new(PendingAuthManager::new());
    let pending_ssh_auth_manager = Arc::new(PendingSshAuthManager::new());
    let pending_ssh_agent_auth_manager = Arc::new(PendingSshAgentAuthManager::new());
    let host_key_verify_manager = Arc::new(HostKeyVerifyManager::new());
    let quick_commands_store = Arc::new(QuickCommandsStore::new());
    let cloud_sync_manager = Arc::new(CloudSyncManager::new());
    let agent_approval_manager = Arc::new(AgentApprovalManager::new());
    let codex_app_server_manager = Arc::new(core::ai::CodexAppServerManager::new());
    let claude_code_runtime = Arc::new(core::ai::ClaudeCodeRuntime::new());
    let transfer_duplicate_manager = Arc::new(TransferDuplicateManager::new());
    let docker_sudo_manager = Arc::new(DockerSudoManager::new());
    let remote_stats_sampler = Arc::new(RemoteStatsSampler::default());
    let app_lock_state = AppLockState::default();
    let external_open_state = external_open::ExternalOpenState::default();
    let portable_update_state = portable_updater::PortableUpdateState::default();
    let mcp_manager = McpManager::new(
        session_manager.clone(),
        runtime.config_dir().to_path_buf(),
        runtime.executable_dir().to_path_buf(),
    );

    let builder = tauri::Builder::default();
    #[cfg(not(any(target_os = "android", target_os = "ios")))]
    let builder = builder.plugin(tauri_plugin_single_instance::init(|app, args, _cwd| {
        if external_open::handle_external_open_args(
            app,
            args,
            external_open::ExternalOpenSource::SecondInstance,
        ) {
            return;
        }

        if let Err(error) = app::create_additional_main_window(app) {
            tracing::warn!("Failed to create additional main window: {}", error);
            app::show_main_window(app);
        }
    }));
    #[cfg(not(any(target_os = "android", target_os = "ios")))]
    let builder = builder.plugin(tauri_plugin_deep_link::init());
    #[cfg(not(any(target_os = "android", target_os = "ios", target_vendor = "win7")))]
    let builder = builder.plugin(tauri_plugin_updater::Builder::new().build());

    let runtime_for_setup = runtime.clone();
    let mut context = tauri::generate_context!();
    runtime::apply_to_context(&mut context, &runtime);

    builder
        .plugin(tauri_plugin_process::init())
        .plugin(tauri_plugin_opener::init())
        .plugin(tauri_plugin_dialog::init())
        .manage(session_manager.clone())
        .manage(rdp_session_manager.clone())
        .manage(vnc_session_manager.clone())
        .manage(tunnel_manager.clone())
        .manage(recording_manager.clone())
        .manage(pending_auth_manager.clone())
        .manage(pending_ssh_auth_manager.clone())
        .manage(pending_ssh_agent_auth_manager.clone())
        .manage(host_key_verify_manager.clone())
        .manage(quick_commands_store.clone())
        .manage(cloud_sync_manager.clone())
        .manage(agent_approval_manager.clone())
        .manage(codex_app_server_manager.clone())
        .manage(claude_code_runtime.clone())
        .manage(transfer_duplicate_manager.clone())
        .manage(docker_sudo_manager.clone())
        .manage(remote_stats_sampler.clone())
        .manage(mcp_manager.clone())
        .manage(app_lock_state)
        .manage(external_open_state)
        .manage(portable_update_state)
        .on_menu_event(cmd::macos_menu::handle_menu_event)
        .setup(move |a| {
            app::setup(
                a,
                session_manager,
                recording_manager,
                quick_commands_store,
                cloud_sync_manager,
                runtime_for_setup.clone(),
            )?;
            let manager = mcp_manager.clone();
            let app_handle = a.handle().clone();
            if let Err(error) = tauri::async_runtime::block_on(manager.initialize(app_handle)) {
                tracing::error!("Failed to initialize External MCP bridge: {error}");
                manager.record_startup_error(&error);
            }
            Ok(())
        })
        .on_window_event(app::on_window_event)
        .invoke_handler(tauri::generate_handler![
            cmd::app::quit_application,
            cmd::app::hide_main_window,
            cmd::app::open_download_dir,
            cmd::app::open_log_dir,
            cmd::app::get_app_runtime_info,
            cmd::app::get_support_info,
            cmd::app::get_app_lock_state,
            cmd::app::set_app_lock_state,
            cmd::app::open_child_window,
            cmd::app::open_transfer_target_directory,
            cmd::app::resolve_local_drop_paths,
            cmd::app::read_background_image_data_url,
            cmd::macos_menu::set_macos_app_menu,
            cmd::external_open::claim_external_open_requests,
            cmd::updater::check_portable_update,
            cmd::updater::download_portable_update,
            cmd::updater::apply_portable_update,
            cmd::ai::start_ai_chat_stream,
            cmd::ai::list_ai_model_names,
            cmd::ai::refresh_ai_model_settings,
            cmd::ai::cancel_ai_chat_stream,
            cmd::ai::detect_codex_cli,
            cmd::ai::get_codex_account_status,
            cmd::ai::start_codex_login,
            cmd::ai::cancel_codex_login,
            cmd::ai::logout_codex,
            cmd::mcp::get_external_mcp_status,
            cmd::mcp::notify_mcp_session_restore_complete,
            cmd::mcp::set_external_mcp_enabled,
            cmd::mcp::respond_external_mcp_approval,
            cmd::mcp::report_mcp_active_session,
            cmd::mcp::respond_mcp_session_open,
            cmd::mcp::get_external_mcp_client_configs,
            cmd::ai::detect_claude_code_cli,
            cmd::ai::get_claude_code_account_status,
            cmd::ai::respond_agent_step,
            cmd::ai::get_ai_sessions,
            cmd::ai::get_ai_messages,
            cmd::ai::clear_ai_history,
            cmd::ai::delete_ai_session,
            cmd::ai::rebind_ai_session,
            cmd::ai::append_ai_audit,
            cmd::ai::get_ai_audit_logs,
            cmd::clipboard::read_clipboard_text,
            cmd::clipboard::write_clipboard_text,
            cmd::clipboard::read_clipboard_path_payload,
            cmd::clipboard::upload_clipboard_image_to_ssh,
            cmd::log::append_frontend_logs,
            cmd::log::export_diagnostics,
            cmd::note::list_note_tree,
            cmd::note::get_note,
            cmd::note::create_note_folder,
            cmd::note::create_note,
            cmd::note::update_note,
            cmd::note::rename_note_node,
            cmd::note::move_note_node,
            cmd::note::delete_note_node,
            cmd::settings::get_system_fonts,
            cmd::settings::get_system_font_infos,
            cmd::cloud_sync::test_cloud_sync_connection,
            cmd::cloud_sync::get_cloud_sync_status,
            cmd::cloud_sync::sync_push_now,
            cmd::cloud_sync::sync_pull_now,
            cmd::cloud_sync::resolve_cloud_sync_conflict,
            cmd::cloud_sync::list_cloud_sync_history,
            cmd::cloud_sync::begin_github_gist_device_flow,
            cmd::cloud_sync::poll_github_gist_device_flow,
            cmd::cloud_sync::cancel_github_gist_device_flow,
            cmd::session::create_ssh_session,
            cmd::session::create_temporary_ssh_session,
            cmd::session::create_multiplexed_ssh_session,
            cmd::session::create_local_session,
            cmd::session::create_telnet_session,
            cmd::session::create_serial_session,
            cmd::rdp::create_rdp_session,
            cmd::rdp::rdp_attach_frame_channel,
            cmd::rdp::rdp_input_batch,
            cmd::rdp::rdp_set_keyboard_capture,
            cmd::rdp::rdp_resize,
            cmd::rdp::rdp_set_clipboard_text,
            cmd::rdp::rdp_reconnect,
            cmd::rdp::close_rdp_session,
            cmd::rdp::respond_rdp_certificate,
            cmd::vnc::create_vnc_session,
            cmd::vnc::vnc_attach_frame_channel,
            cmd::vnc::vnc_input_batch,
            cmd::vnc::vnc_set_clipboard_text,
            cmd::vnc::vnc_reconnect,
            cmd::vnc::close_vnc_session,
            cmd::session::cancel_session_creation,
            cmd::session::list_serial_ports,
            cmd::session::write_to_session,
            cmd::session::set_session_output_paused,
            cmd::session::ack_session_output,
            cmd::session::resize_session,
            cmd::session::attach_session,
            cmd::session::detach_session_renderer,
            cmd::session::close_session,
            cmd::session::list_sessions,
            cmd::session::get_session_cwd_presentation,
            cmd::session::add_command_history,
            cmd::session::register_command_submission,
            cmd::session::register_command_confirmation_candidate,
            cmd::session::get_command_history,
            cmd::session::delete_command_history,
            cmd::session::fuzzy_search_history,
            cmd::session::fuzzy_search_commands,
            cmd::session::fuzzy_search_candidates,
            cmd::session::start_recording,
            cmd::session::stop_recording,
            cmd::session::is_recording,
            cmd::session::save_session_transcript,
            cmd::session::terminal_history_search,
            cmd::session::list_recording_sessions,
            cmd::session::get_recording_status,
            cmd::session::list_recording_statuses,
            cmd::session::open_recording_file,
            cmd::session::show_recording_in_folder,
            cmd::session::set_recording_memory_limit,
            cmd::session::submit_otp_response,
            cmd::session::cancel_otp_request,
            cmd::session::submit_ssh_auth_response,
            cmd::session::cancel_ssh_auth_request,
            cmd::session::respond_ssh_agent_auth,
            cmd::session::cancel_ssh_agent_auth,
            cmd::session::respond_host_key_verify,
            cmd::session::zmodem_accept_download,
            cmd::session::zmodem_accept_upload,
            cmd::session::zmodem_cancel,
            cmd::sftp::get_home_dir,
            cmd::sftp::list_remote_dir,
            cmd::sftp::list_remote_child_directories,
            cmd::sftp::delete_remote_file,
            cmd::sftp::rename_remote_file,
            cmd::sftp::sanitize_download_file_name,
            cmd::sftp::download_remote_file,
            cmd::sftp::upload_local_file,
            cmd::sftp::get_file_properties,
            cmd::sftp::read_remote_file_text,
            cmd::sftp::open_remote_file_text,
            cmd::sftp::read_remote_file_bytes,
            cmd::sftp::write_remote_file_text,
            cmd::sftp::create_remote_file,
            cmd::sftp::create_remote_dir,
            cmd::sftp::create_remote_symlink,
            cmd::sftp::update_remote_symlink_target,
            cmd::sftp::chmod_remote_file,
            cmd::sftp::update_remote_file_attributes,
            cmd::sftp::download_remote_directory,
            cmd::sftp::upload_local_directory,
            cmd::sftp::copy_file_entry,
            cmd::sftp::pause_transfer,
            cmd::sftp::resume_transfer,
            cmd::sftp::cancel_transfer,
            cmd::sftp::respond_transfer_duplicate,
            cmd::local_fs::get_local_home_dir,
            cmd::local_fs::list_local_dir,
            cmd::local_fs::list_local_child_directories,
            cmd::local_fs::create_local_file,
            cmd::local_fs::create_local_dir,
            cmd::local_fs::rename_local_file,
            cmd::local_fs::delete_local_file,
            cmd::local_fs::get_local_file_properties,
            cmd::local_fs::read_local_file_text,
            cmd::local_fs::open_local_file_text,
            cmd::local_fs::read_local_file_bytes,
            cmd::local_fs::write_local_file_text,
            cmd::connection::get_saved_connections,
            cmd::connection::get_supported_ssh_algorithms,
            cmd::connection::get_ssh_agent_forwarding_identities,
            cmd::connection::import_connection_icon,
            cmd::connection::get_connection_custom_icons,
            cmd::connection::delete_connection_custom_icon,
            cmd::connection::save_connection,
            cmd::connection::update_connection_icon,
            cmd::connection::update_connection_asset_from_monitoring,
            cmd::connection::delete_connection,
            cmd::connection::get_connection_password_value,
            cmd::connection::reorder_items,
            cmd::connection::get_ssh_keys,
            cmd::connection::get_ssh_key_passphrase,
            cmd::connection::get_ssh_key_private_key,
            cmd::connection::save_ssh_key,
            cmd::connection::delete_ssh_key,
            cmd::connection::get_groups,
            cmd::connection::save_group,
            cmd::connection::delete_group,
            cmd::connection::clear_all_connections,
            cmd::connection::get_quick_commands,
            cmd::connection::export_quick_commands,
            cmd::connection::save_quick_commands,
            cmd::connection::upsert_quick_command,
            cmd::connection::increment_quick_command_use_count,
            cmd::connection::import_quick_commands,
            cmd::connection::get_saved_passwords,
            cmd::connection::get_saved_password_value,
            cmd::connection::save_password,
            cmd::connection::delete_password,
            cmd::credential::get_saved_credentials,
            cmd::credential::get_saved_credential_password,
            cmd::credential::save_credential,
            cmd::credential::delete_credential,
            cmd::credential::reorder_credentials,
            cmd::settings::get_app_settings,
            cmd::settings::save_app_settings,
            cmd::settings::save_app_language,
            cmd::settings::import_keyword_highlight_rules,
            cmd::settings::read_theme_file,
            cmd::settings::write_theme_file,
            cmd::settings::save_app_ui_settings,
            cmd::settings::verify_master_password,
            cmd::watcher::start_file_watch,
            cmd::watcher::stop_file_watch,
            cmd::translate::translate_text,
            cmd::importer::import_sessions,
            cmd::importer::import_termius_sessions,
            cmd::ssh_config::list_ssh_config_hosts,
            cmd::ssh_config::get_ssh_config,
            cmd::ssh_config::resolve_ssh_host,
            cmd::ssh_config::import_ssh_config_hosts,
            cmd::backup::export_config,
            cmd::backup::import_config,
            cmd::stats::get_remote_stats,
            cmd::stats::get_terminal_cwd,
            cmd::stats::try_get_terminal_cwd,
            cmd::process::get_remote_processes,
            cmd::process::signal_remote_process,
            cmd::gpu::get_remote_gpu_overview,
            cmd::ascend_npu::get_remote_ascend_npu_overview,
            cmd::docker::get_remote_docker_overview,
            cmd::docker::get_remote_docker_images,
            cmd::docker::get_remote_docker_volumes,
            cmd::docker::get_remote_docker_networks,
            cmd::docker::get_remote_docker_compose_projects,
            cmd::docker::get_docker_container_details,
            cmd::docker::get_docker_container_stats,
            cmd::docker::docker_container_action,
            cmd::docker::docker_image_remove,
            cmd::docker::docker_volume_remove,
            cmd::docker::docker_network_remove,
            cmd::docker::docker_system_prune,
            cmd::docker::get_docker_container_logs,
            cmd::docker::docker_compose_action,
            cmd::docker::get_docker_compose_services,
            cmd::docker::docker_compose_service_action,
            cmd::docker::submit_docker_sudo_password,
            cmd::docker::cancel_docker_sudo_password,
            cmd::docker::prepare_docker_container_logs_command,
            cmd::docker::prepare_docker_container_shell_command,
            cmd::docker::prepare_docker_compose_service_logs_command,
            cmd::tunnel::get_tunnels,
            cmd::tunnel::get_tunnel_runtime_states,
            cmd::tunnel::get_tunnel_groups,
            cmd::tunnel::save_tunnel,
            cmd::tunnel::save_tunnel_group,
            cmd::tunnel::set_tunnel_group,
            cmd::tunnel::delete_tunnel,
            cmd::tunnel::delete_tunnel_group,
            cmd::tunnel::open_tunnel,
            cmd::tunnel::close_tunnel,
            cmd::tunnel::mark_tunnels_reconnecting_for_connection,
            cmd::tunnel::mark_tunnels_disconnected_for_connection,
            cmd::proxy::get_proxies,
            cmd::proxy::get_proxy_groups,
            cmd::proxy::save_proxy,
            cmd::proxy::save_proxy_group,
            cmd::proxy::set_proxy_group,
            cmd::proxy::delete_proxy,
            cmd::proxy::delete_proxy_group,
            cmd::proxy::get_proxy_password,
            cmd::otp::get_otp_entries,
            cmd::otp::get_otp_secret_value,
            cmd::otp::save_otp_entry,
            cmd::otp::delete_otp_entry,
            cmd::otp::generate_otp_code,
            cmd::otp::import_otp_from_qr,
        ])
        .build(context)
        .expect("error while building tauri application")
        .run(|_app, _event| {
            if matches!(
                _event,
                tauri::RunEvent::Exit | tauri::RunEvent::ExitRequested { .. }
            ) {
                if let Some(manager) = _app.try_state::<Arc<McpManager>>() {
                    manager.shutdown_cleanup();
                }
            }
            #[cfg(target_os = "macos")]
            if matches!(_event, tauri::RunEvent::Reopen { .. }) {
                app::show_main_window(_app);
            }
        });
}

pub fn run_portable_update_helper_if_requested() -> bool {
    portable_updater::run_helper_if_requested()
}

pub fn run_cloud_snapshot_decode_helper_if_requested() -> bool {
    core::cloud_sync::run_snapshot_decode_helper_if_requested()
}