186 lines
7.8 KiB
Rust
186 lines
7.8 KiB
Rust
mod core;
|
|
use core::utils::extensions::inference_llamacpp_extension::cleanup::cleanup_processes;
|
|
use core::{
|
|
cmd::get_jan_data_folder_path,
|
|
setup::{self, setup_mcp},
|
|
state::{generate_app_token, AppState},
|
|
utils::download::DownloadManagerState,
|
|
};
|
|
use std::{collections::HashMap, sync::Arc};
|
|
use tauri::{Emitter, Manager, RunEvent};
|
|
use tokio::sync::Mutex;
|
|
|
|
use crate::core::mcp::clean_up_mcp_servers;
|
|
|
|
#[cfg_attr(mobile, tauri::mobile_entry_point)]
|
|
pub fn run() {
|
|
let mut builder = tauri::Builder::default();
|
|
#[cfg(desktop)]
|
|
{
|
|
builder = builder.plugin(tauri_plugin_single_instance::init(|_app, argv, _cwd| {
|
|
println!("a new app instance was opened with {argv:?} and the deep link event was already triggered");
|
|
// when defining deep link schemes at runtime, you must also check `argv` here
|
|
}));
|
|
}
|
|
|
|
let app = builder
|
|
.plugin(tauri_plugin_os::init())
|
|
.plugin(tauri_plugin_deep_link::init())
|
|
.plugin(tauri_plugin_dialog::init())
|
|
.plugin(tauri_plugin_opener::init())
|
|
.plugin(tauri_plugin_http::init())
|
|
.plugin(tauri_plugin_store::Builder::new().build())
|
|
.plugin(tauri_plugin_updater::Builder::new().build())
|
|
.plugin(tauri_plugin_shell::init())
|
|
.invoke_handler(tauri::generate_handler![
|
|
// FS commands - Deperecate soon
|
|
core::fs::join_path,
|
|
core::fs::mkdir,
|
|
core::fs::exists_sync,
|
|
core::fs::readdir_sync,
|
|
core::fs::read_file_sync,
|
|
core::fs::rm,
|
|
core::fs::file_stat,
|
|
core::fs::write_file_sync,
|
|
// App commands
|
|
core::cmd::get_app_configurations,
|
|
core::cmd::get_active_extensions,
|
|
core::cmd::get_user_home_path,
|
|
core::cmd::update_app_configuration,
|
|
core::cmd::get_jan_data_folder_path,
|
|
core::cmd::get_jan_extensions_path,
|
|
core::cmd::relaunch,
|
|
core::cmd::open_app_directory,
|
|
core::cmd::open_file_explorer,
|
|
core::cmd::install_extensions,
|
|
core::cmd::app_token,
|
|
core::cmd::start_server,
|
|
core::cmd::stop_server,
|
|
core::cmd::get_server_status,
|
|
core::cmd::read_logs,
|
|
core::cmd::change_app_data_folder,
|
|
core::cmd::factory_reset,
|
|
// MCP commands
|
|
core::mcp::get_tools,
|
|
core::mcp::call_tool,
|
|
core::mcp::restart_mcp_servers,
|
|
core::mcp::get_connected_servers,
|
|
core::mcp::save_mcp_configs,
|
|
core::mcp::get_mcp_configs,
|
|
core::mcp::activate_mcp_server,
|
|
core::mcp::deactivate_mcp_server,
|
|
core::mcp::reset_mcp_restart_count,
|
|
// Threads
|
|
core::threads::list_threads,
|
|
core::threads::create_thread,
|
|
core::threads::modify_thread,
|
|
core::threads::delete_thread,
|
|
core::threads::list_messages,
|
|
core::threads::create_message,
|
|
core::threads::modify_message,
|
|
core::threads::delete_message,
|
|
core::threads::get_thread_assistant,
|
|
core::threads::create_thread_assistant,
|
|
core::threads::modify_thread_assistant,
|
|
// generic utils
|
|
core::utils::write_yaml,
|
|
core::utils::read_yaml,
|
|
core::utils::decompress,
|
|
core::utils::is_library_available,
|
|
// Download
|
|
core::utils::download::download_files,
|
|
core::utils::download::cancel_download_task,
|
|
// hardware
|
|
core::hardware::get_system_info,
|
|
core::hardware::get_system_usage,
|
|
// llama-cpp extension
|
|
core::utils::extensions::inference_llamacpp_extension::server::load_llama_model,
|
|
core::utils::extensions::inference_llamacpp_extension::server::unload_llama_model,
|
|
core::utils::extensions::inference_llamacpp_extension::server::get_devices,
|
|
core::utils::extensions::inference_llamacpp_extension::server::get_random_port,
|
|
core::utils::extensions::inference_llamacpp_extension::server::find_session_by_model,
|
|
core::utils::extensions::inference_llamacpp_extension::server::get_loaded_models,
|
|
core::utils::extensions::inference_llamacpp_extension::server::generate_api_key,
|
|
core::utils::extensions::inference_llamacpp_extension::server::is_process_running,
|
|
])
|
|
.manage(AppState {
|
|
app_token: Some(generate_app_token()),
|
|
mcp_servers: Arc::new(Mutex::new(HashMap::new())),
|
|
download_manager: Arc::new(Mutex::new(DownloadManagerState::default())),
|
|
mcp_restart_counts: Arc::new(Mutex::new(HashMap::new())),
|
|
mcp_active_servers: Arc::new(Mutex::new(HashMap::new())),
|
|
mcp_successfully_connected: Arc::new(Mutex::new(HashMap::new())),
|
|
server_handle: Arc::new(Mutex::new(None)),
|
|
llama_server_process: Arc::new(Mutex::new(HashMap::new())),
|
|
})
|
|
.setup(|app| {
|
|
app.handle().plugin(
|
|
tauri_plugin_log::Builder::default()
|
|
.level(log::LevelFilter::Debug)
|
|
.targets([
|
|
tauri_plugin_log::Target::new(tauri_plugin_log::TargetKind::Stdout),
|
|
tauri_plugin_log::Target::new(tauri_plugin_log::TargetKind::Webview),
|
|
tauri_plugin_log::Target::new(tauri_plugin_log::TargetKind::Folder {
|
|
path: get_jan_data_folder_path(app.handle().clone()).join("logs"),
|
|
file_name: Some("app".to_string()),
|
|
}),
|
|
])
|
|
.build(),
|
|
)?;
|
|
app.handle()
|
|
.plugin(tauri_plugin_updater::Builder::new().build())?;
|
|
// Install extensions
|
|
if let Err(e) = setup::install_extensions(app.handle().clone(), false) {
|
|
log::error!("Failed to install extensions: {}", e);
|
|
}
|
|
|
|
#[cfg(any(windows, target_os = "linux"))]
|
|
{
|
|
use tauri_plugin_deep_link::DeepLinkExt;
|
|
app.deep_link().register_all()?;
|
|
}
|
|
setup_mcp(app);
|
|
Ok(())
|
|
})
|
|
.on_window_event(|window, event| match event {
|
|
tauri::WindowEvent::CloseRequested { .. } => {
|
|
if window.label() == "main" {
|
|
let state = window.app_handle().state::<AppState>();
|
|
|
|
tauri::async_runtime::block_on(async {
|
|
clean_up_mcp_servers(state.clone()).await;
|
|
cleanup_processes(state).await;
|
|
});
|
|
}
|
|
}
|
|
_ => {}
|
|
})
|
|
.build(tauri::generate_context!())
|
|
.expect("error while running tauri application");
|
|
|
|
// Handle app lifecycle events
|
|
app.run(|app, event| match event {
|
|
RunEvent::Exit => {
|
|
// This is called when the app is actually exiting (e.g., macOS dock quit)
|
|
// We can't prevent this, so run cleanup quickly
|
|
let app_handle = app.clone();
|
|
tokio::task::block_in_place(|| {
|
|
tauri::async_runtime::block_on(async {
|
|
let state = app_handle.state::<AppState>();
|
|
|
|
// Hide window immediately
|
|
if let Some(window) = app_handle.get_webview_window("main") {
|
|
let _ = window.hide();
|
|
let _ = window.emit("kill-mcp-servers", ());
|
|
}
|
|
|
|
// Quick cleanup with shorter timeout
|
|
clean_up_mcp_servers(state.clone()).await;
|
|
cleanup_processes(state).await;
|
|
});
|
|
});
|
|
}
|
|
_ => {}
|
|
});
|
|
}
|