lib.rs

src-tauri/src/lib.rs

1. //! Cortex, the desktop shell: windows, tray, shortcut and the commands the
2. //! interface calls. The logic lives in cortex-core.
4. mod commands;
5. mod state;
6. mod windows;
8. use std::collections::HashMap;
9. use std::sync::Mutex;
11. use cortex_core::secrets::{KeyCache, SecretStore};
12. use cortex_core::storage::Db;
13. use tauri::Manager;
14. use tauri_plugin_global_shortcut::ShortcutState;
16. use state::AppState;
18. fn secret_store() -> Box<dyn SecretStore> {
19. #[cfg(all(feature = "keychain", not(feature = "e2e")))]
20. {
21. Box::new(cortex_core::secrets::KeychainStore)
22. }
23. #[cfg(not(all(feature = "keychain", not(feature = "e2e"))))]
24. {
25. // Tests and builds without the keychain: keys last for the session only.
26. Box::new(cortex_core::secrets::MemoryStore::default())
27. }
28. }
30. /// End-to-end runs only: point Cortex at a stand-in server and a scratch folder.
31. #[cfg(feature = "e2e")]
32. fn e2e_overrides(state: &AppState) {
33. use cortex_core::api::Settings;
34. let db = state.db();
35. if let Ok(url) = std::env::var("CORTEX_SEED_URL") {
36. let mut p = db.get_provider(cortex_core::storage::DEEPINFRA_ID).unwrap();
37. p.base_url = url;
38. db.upsert_provider(&p).unwrap();
39. state.keys.set(cortex_core::storage::DEEPINFRA_ID, "e2e-key").unwrap();
40. let mut s: Settings = db.get_setting(commands::SETTINGS_KEY).ok().flatten().unwrap_or_default();
41. s.default_model_id = "standin/echo".into();
42. s.quick_model_id = "standin/echo".into();
43. s.monthly_budget_usd = Some(1.0);
44. if let Ok(dir) = std::env::var("CORTEX_EXPORT_DIR") {
45. s.vault_dir = Some(dir);
46. }
47. db.set_setting(commands::SETTINGS_KEY, &s).unwrap();
48. }
49. }
51. #[cfg_attr(mobile, tauri::mobile_entry_point)]
52. pub fn run() {
53. let mut builder = tauri::Builder::default();
54. // First, so a second launch just brings the running one forward. Not in
55. // test builds: a test run must never hand itself to an app left running.
56. if !cfg!(feature = "e2e") {
57. builder = builder.plugin(tauri_plugin_single_instance::init(|app, _argv, _cwd| windows::show_main(app)));
58. }
59. builder = builder
60. .plugin(
61. tauri_plugin_global_shortcut::Builder::new()
62. .with_handler(|app, _shortcut, event| {
63. if event.state() == ShortcutState::Pressed {
64. windows::toggle_quick(app);
65. }
66. })
67. .build(),
68. )
69. .plugin(tauri_plugin_dialog::init())
70. .plugin(tauri_plugin_opener::init());
71. if std::env::var_os("CORTEX_DATA_DIR").is_none() {
72. builder = builder.plugin(tauri_plugin_window_state::Builder::default().with_denylist(&["quick"]).build());
73. }
74. builder
75. .setup(|app| {
76. let data_dir = match std::env::var_os("CORTEX_DATA_DIR").filter(|_| cfg!(feature = "e2e")) {
77. Some(d) => d.into(),
78. None => app.path().app_data_dir()?,
79. };
80. std::fs::create_dir_all(&data_dir)?;
81. let db = Db::open(&data_dir.join("cortex.db")).map_err(|e| e.to_string())?;
82. let state = AppState {
83. db: Mutex::new(db),
84. keys: KeyCache::new(secret_store()),
85. streams: Mutex::new(HashMap::new()),
86. data_dir,
87. quick_shown_at: Mutex::new(None),
88. dialog_open: Mutex::new(false),
89. tray_spend: Mutex::new(None),
90. };
91. // Attachments added to the composer and never sent.
92. if let Ok(paths) = state.db().delete_stale_drafts(commands::DAY_MS) {
93. commands::remove_unused_files(&state, &paths);
94. }
95. #[cfg(feature = "e2e")]
96. e2e_overrides(&state);
97. app.manage(state);
99. let handle = app.handle().clone();
100. windows::create_windows(&handle)?;
101. let settings = commands::load_settings(&handle.state::<AppState>());
102. windows::apply_theme(&handle, &settings.theme);
103. if settings.material != "sidebar" {
104. windows::apply_material(&handle, &settings.material);
105. }
106. if let Err(err) = windows::create_tray(&handle) {
107. eprintln!("cortex: no tray: {err}"); // e.g. a Linux desktop without a tray host
108. }
109. windows::register_initial_shortcut(&handle);
110. tauri::async_runtime::spawn(commands::refresh_stale(handle));
111. Ok(())
112. })
113. .invoke_handler(tauri::generate_handler![
114. commands::app_info,
115. commands::window_ready,
116. commands::show_main,
117. commands::list_providers,
118. commands::save_provider,
119. commands::delete_provider,
120. commands::set_api_key,
121. commands::clear_api_key,
122. commands::has_api_key,
123. commands::list_models,
124. commands::refresh_models,
125. commands::list_conversations,
126. commands::get_conversation,
127. commands::create_conversation,
128. commands::rename_conversation,
129. commands::set_pinned,
130. commands::set_archived,
131. commands::delete_conversation,
132. commands::set_conversation_model,
133. commands::set_system_prompt,
134. commands::send_message,
135. commands::cancel_stream,
136. commands::truncate_from,
137. commands::estimate_prompt,
138. commands::search,
139. commands::spend_summary,
140. commands::usage_report,
141. commands::list_presets,
142. commands::save_preset,
143. commands::delete_preset,
144. commands::get_settings,
145. commands::save_settings,
146. commands::pick_attachments,
147. commands::attach_bytes,
148. commands::attachment_bytes,
149. commands::discard_attachment,
150. commands::export_conversation,
151. commands::choose_vault_folder,
152. commands::quick_open_in_main,
153. commands::hide_quick,
154. commands::resize_quick,
155. ])
156. .build(tauri::generate_context!())
157. .expect("error while building Cortex")
158. .run(|app, event| {
159. #[cfg(target_os = "macos")]
160. if let tauri::RunEvent::Reopen { .. } = event {
161. windows::show_main(app);
162. }
163. let _ = (app, event);
164. });
165. }