1#![deny(missing_docs)]
2#![cfg_attr(not(debug_assertions), allow(unused_imports, dead_code))]
7pub mod bridge;
46#[cfg(feature = "sqlite")]
48pub mod database;
49pub mod error;
50pub mod js_bridge;
52pub mod mcp;
54mod memory;
55pub mod privacy;
57
58pub mod filmstrip;
60pub mod redaction;
62pub mod screencast;
63pub(crate) mod screenshot;
64mod tools;
65
66pub mod auth;
68pub mod introspection;
70
71use std::collections::{HashMap, HashSet};
72use std::sync::Arc;
73use std::sync::atomic::{AtomicBool, AtomicU16, AtomicU64};
74use tauri::plugin::{Builder, TauriPlugin};
75use tauri::{Listener, Manager, RunEvent, Runtime};
76use tokio::sync::{Mutex, oneshot, watch};
77use victauri_core::{CommandRegistry, EventLog, EventRecorder};
78
79pub use error::BuilderError;
80pub use privacy::PrivacyProfile;
81
82pub use victauri_core::CommandInfo;
83pub use victauri_macros::inspectable;
84
85#[macro_export]
106macro_rules! register_commands {
107 ($app:expr, $($schema_call:expr),+ $(,)?) => {{
108 if let Some(state) = $app.try_state::<std::sync::Arc<$crate::VictauriState>>() {
112 $(
113 state.registry.register($schema_call);
114 )+
115 }
116 }};
117}
118
119const DEFAULT_PORT: u16 = 7373;
120const DEFAULT_EVENT_CAPACITY: usize = 10_000;
121const DEFAULT_RECORDER_CAPACITY: usize = 50_000;
122const DEFAULT_EVAL_TIMEOUT: std::time::Duration = std::time::Duration::from_secs(30);
123const MAX_EVENT_CAPACITY: usize = 1_000_000;
124const MAX_RECORDER_CAPACITY: usize = 1_000_000;
125const MAX_EVAL_TIMEOUT_SECS: u64 = 300;
126
127pub type PendingCallbacks = Arc<Mutex<HashMap<String, oneshot::Sender<String>>>>;
130
131pub struct VictauriState {
133 pub event_log: EventLog,
135 pub registry: CommandRegistry,
137 pub port: AtomicU16,
139 pub pending_evals: PendingCallbacks,
141 pub recorder: EventRecorder,
143 pub privacy: privacy::PrivacyConfig,
145 pub eval_timeout: std::time::Duration,
147 pub shutdown_tx: watch::Sender<bool>,
149 pub started_at: std::time::Instant,
151 pub tool_invocations: AtomicU64,
153 pub allow_file_navigation: bool,
157 pub command_timings: introspection::CommandTimings,
159 pub fault_registry: introspection::FaultRegistry,
161 pub contract_store: introspection::ContractStore,
163 pub startup_timeline: introspection::StartupTimeline,
165 pub event_bus: introspection::EventBusMonitor,
167 pub task_tracker: introspection::TaskTracker,
169 pub bridge_ready: AtomicBool,
171 pub bridge_notify: tokio::sync::Notify,
173 pub screencast: Arc<screencast::Screencast>,
175 pub db_search_paths: Vec<std::path::PathBuf>,
182 pub probes: introspection::AppStateProbes,
185}
186
187pub struct VictauriBuilder {
199 port: Option<u16>,
200 event_capacity: usize,
201 recorder_capacity: usize,
202 eval_timeout: std::time::Duration,
203 auth_token: Option<String>,
204 auth_explicitly_enabled: bool,
205 auth_explicitly_disabled: bool,
206 disabled_tools: Vec<String>,
207 command_allowlist: Option<Vec<String>>,
208 command_blocklist: Vec<String>,
209 storage_key_blocklist: Vec<String>,
210 redaction_patterns: Vec<String>,
211 redaction_enabled: bool,
212 strict_privacy: bool,
213 privacy_profile: Option<privacy::PrivacyProfile>,
214 bridge_capacities: js_bridge::BridgeCapacities,
215 on_ready: Option<Box<dyn FnOnce(u16) + Send + 'static>>,
216 commands: Vec<victauri_core::CommandInfo>,
217 allow_file_navigation: bool,
218 listen_events: Vec<String>,
219 db_search_paths: Vec<std::path::PathBuf>,
220 probes: Vec<(String, std::sync::Arc<introspection::ProbeFn>)>,
221}
222
223impl Default for VictauriBuilder {
224 fn default() -> Self {
225 Self {
226 port: None,
227 event_capacity: DEFAULT_EVENT_CAPACITY,
228 recorder_capacity: DEFAULT_RECORDER_CAPACITY,
229 eval_timeout: DEFAULT_EVAL_TIMEOUT,
230 auth_token: None,
231 auth_explicitly_enabled: false,
232 auth_explicitly_disabled: false,
233 disabled_tools: Vec::new(),
234 command_allowlist: None,
235 command_blocklist: Vec::new(),
236 storage_key_blocklist: Vec::new(),
237 redaction_patterns: Vec::new(),
238 redaction_enabled: false,
239 strict_privacy: false,
240 privacy_profile: None,
241 bridge_capacities: js_bridge::BridgeCapacities::default(),
242 on_ready: None,
243 commands: Vec::new(),
244 allow_file_navigation: false,
245 listen_events: Vec::new(),
246 db_search_paths: Vec::new(),
247 probes: Vec::new(),
248 }
249 }
250}
251
252impl VictauriBuilder {
253 #[must_use]
255 pub fn new() -> Self {
256 Self::default()
257 }
258
259 #[must_use]
261 pub fn port(mut self, port: u16) -> Self {
262 self.port = Some(port);
263 self
264 }
265
266 #[must_use]
268 pub fn event_capacity(mut self, capacity: usize) -> Self {
269 self.event_capacity = capacity;
270 self
271 }
272
273 #[must_use]
275 pub fn recorder_capacity(mut self, capacity: usize) -> Self {
276 self.recorder_capacity = capacity;
277 self
278 }
279
280 #[must_use]
282 pub fn eval_timeout(mut self, timeout: std::time::Duration) -> Self {
283 self.eval_timeout = timeout;
284 self
285 }
286
287 #[must_use]
291 pub fn auth_token(mut self, token: impl Into<String>) -> Self {
292 self.auth_token = Some(token.into());
293 self
294 }
295
296 #[must_use]
304 pub fn auth_enabled(mut self) -> Self {
305 self.auth_explicitly_enabled = true;
306 self
307 }
308
309 #[must_use]
311 pub fn generate_auth_token(mut self) -> Self {
312 self.auth_explicitly_enabled = true;
313 self
314 }
315
316 #[must_use]
326 pub fn auth_disabled(mut self) -> Self {
327 self.auth_explicitly_disabled = true;
328 self
329 }
330
331 #[must_use]
333 pub fn disable_tools(mut self, tools: &[&str]) -> Self {
334 self.disabled_tools = tools.iter().map(std::string::ToString::to_string).collect();
335 self
336 }
337
338 #[must_use]
340 pub fn command_allowlist(mut self, commands: &[&str]) -> Self {
341 self.command_allowlist = Some(
342 commands
343 .iter()
344 .map(std::string::ToString::to_string)
345 .collect(),
346 );
347 self
348 }
349
350 #[must_use]
352 pub fn command_blocklist(mut self, commands: &[&str]) -> Self {
353 self.command_blocklist = commands
354 .iter()
355 .map(std::string::ToString::to_string)
356 .collect();
357 self
358 }
359
360 #[must_use]
365 pub fn storage_key_blocklist(mut self, keys: &[&str]) -> Self {
366 self.storage_key_blocklist = keys.iter().map(std::string::ToString::to_string).collect();
367 self
368 }
369
370 #[must_use]
372 pub fn add_redaction_pattern(mut self, pattern: impl Into<String>) -> Self {
373 self.redaction_patterns.push(pattern.into());
374 self
375 }
376
377 #[must_use]
379 pub fn enable_redaction(mut self) -> Self {
380 self.redaction_enabled = true;
381 self
382 }
383
384 #[must_use]
391 pub fn strict_privacy_mode(mut self) -> Self {
392 self.strict_privacy = true;
393 self.privacy_profile = Some(privacy::PrivacyProfile::Observe);
394 self
395 }
396
397 #[must_use]
406 pub fn privacy_profile(mut self, profile: privacy::PrivacyProfile) -> Self {
407 self.privacy_profile = Some(profile);
408 if matches!(
409 profile,
410 privacy::PrivacyProfile::Observe | privacy::PrivacyProfile::Test
411 ) {
412 self.redaction_enabled = true;
413 }
414 self
415 }
416
417 #[must_use]
419 pub fn console_log_capacity(mut self, capacity: usize) -> Self {
420 self.bridge_capacities.console_logs = capacity;
421 self
422 }
423
424 #[must_use]
426 pub fn network_log_capacity(mut self, capacity: usize) -> Self {
427 self.bridge_capacities.network_log = capacity;
428 self
429 }
430
431 #[must_use]
433 pub fn navigation_log_capacity(mut self, capacity: usize) -> Self {
434 self.bridge_capacities.navigation_log = capacity;
435 self
436 }
437
438 #[must_use]
453 pub fn commands(mut self, schemas: &[victauri_core::CommandInfo]) -> Self {
454 self.commands = schemas.to_vec();
455 self
456 }
457
458 #[must_use]
474 pub fn register_command_names(mut self, names: &[&str]) -> Self {
475 self.commands
476 .extend(names.iter().map(|n| victauri_core::CommandInfo::new(*n)));
477 self
478 }
479
480 #[must_use]
496 pub fn auto_discover(mut self) -> Self {
497 self.commands
498 .extend(victauri_core::auto_discovered_commands());
499 self
500 }
501
502 #[must_use]
513 pub fn listen_events(mut self, events: &[&str]) -> Self {
514 self.listen_events = events
515 .iter()
516 .map(std::string::ToString::to_string)
517 .collect();
518 self
519 }
520
521 #[must_use]
530 pub fn allow_file_navigation(mut self) -> Self {
531 self.allow_file_navigation = true;
532 self
533 }
534
535 #[must_use]
548 pub fn db_search_paths<I, P>(mut self, paths: I) -> Self
549 where
550 I: IntoIterator<Item = P>,
551 P: Into<std::path::PathBuf>,
552 {
553 self.db_search_paths
554 .extend(paths.into_iter().map(Into::into));
555 self
556 }
557
558 #[must_use]
582 pub fn probe<F>(mut self, name: impl Into<String>, probe: F) -> Self
583 where
584 F: Fn() -> serde_json::Value + Send + Sync + 'static,
585 {
586 self.probes.push((name.into(), std::sync::Arc::new(probe)));
587 self
588 }
589
590 #[must_use]
593 pub fn on_ready(mut self, f: impl FnOnce(u16) + Send + 'static) -> Self {
594 self.on_ready = Some(Box::new(f));
595 self
596 }
597
598 fn resolve_port(&self) -> u16 {
599 self.port
600 .or_else(|| std::env::var("VICTAURI_PORT").ok()?.parse().ok())
601 .unwrap_or(DEFAULT_PORT)
602 }
603
604 fn resolve_auth_token(&self) -> Option<String> {
605 if self.auth_explicitly_disabled {
606 return None;
607 }
608 if let Some(ref token) = self.auth_token {
609 return Some(token.clone());
610 }
611 if let Ok(token) = std::env::var("VICTAURI_AUTH_TOKEN") {
612 return Some(token);
613 }
614 Some(auth::generate_token())
615 }
616
617 fn resolve_eval_timeout(&self) -> std::time::Duration {
618 std::env::var("VICTAURI_EVAL_TIMEOUT")
619 .ok()
620 .and_then(|s| s.parse::<u64>().ok())
621 .map_or(self.eval_timeout, std::time::Duration::from_secs)
622 }
623
624 fn build_privacy_config(&self) -> privacy::PrivacyConfig {
625 let profile = self
626 .privacy_profile
627 .unwrap_or(privacy::PrivacyProfile::FullControl);
628
629 let redaction_enabled = self.redaction_enabled
630 || self.strict_privacy
631 || matches!(
632 profile,
633 privacy::PrivacyProfile::Observe | privacy::PrivacyProfile::Test
634 );
635
636 privacy::PrivacyConfig {
637 profile,
638 command_allowlist: self
639 .command_allowlist
640 .as_ref()
641 .map(|v| v.iter().cloned().collect::<HashSet<String>>()),
642 command_blocklist: self.command_blocklist.iter().cloned().collect(),
643 disabled_tools: self.disabled_tools.iter().cloned().collect(),
644 storage_key_blocklist: self.storage_key_blocklist.iter().cloned().collect(),
645 redactor: redaction::Redactor::new(&self.redaction_patterns),
646 redaction_enabled,
647 }
648 }
649
650 fn validate(&self) -> Result<(), BuilderError> {
651 let port = self.resolve_port();
652 if port == 0 {
653 return Err(BuilderError::InvalidPort {
654 port,
655 reason: "port 0 is reserved".to_string(),
656 });
657 }
658
659 if self.event_capacity == 0 || self.event_capacity > MAX_EVENT_CAPACITY {
660 return Err(BuilderError::InvalidEventCapacity {
661 capacity: self.event_capacity,
662 reason: format!("must be between 1 and {MAX_EVENT_CAPACITY}"),
663 });
664 }
665
666 if self.recorder_capacity == 0 || self.recorder_capacity > MAX_RECORDER_CAPACITY {
667 return Err(BuilderError::InvalidRecorderCapacity {
668 capacity: self.recorder_capacity,
669 reason: format!("must be between 1 and {MAX_RECORDER_CAPACITY}"),
670 });
671 }
672
673 let timeout = self.resolve_eval_timeout();
674 if timeout.as_secs() == 0 || timeout.as_secs() > MAX_EVAL_TIMEOUT_SECS {
675 return Err(BuilderError::InvalidEvalTimeout {
676 timeout_secs: timeout.as_secs(),
677 reason: format!("must be between 1 and {MAX_EVAL_TIMEOUT_SECS} seconds"),
678 });
679 }
680
681 Ok(())
682 }
683
684 pub fn build<R: Runtime>(self) -> Result<TauriPlugin<R>, BuilderError> {
694 #[cfg(not(debug_assertions))]
695 {
696 Ok(Builder::new("victauri").build())
697 }
698
699 #[cfg(debug_assertions)]
700 {
701 if env_truthy("VICTAURI_DISABLE") {
705 tracing::info!("Victauri disabled via VICTAURI_DISABLE — returning no-op plugin");
706 return Ok(Builder::new("victauri").build());
707 }
708
709 self.validate()?;
710
711 let port = self.resolve_port();
712 let event_capacity = self.event_capacity;
713 let recorder_capacity = self.recorder_capacity;
714 let eval_timeout = self.resolve_eval_timeout();
715 let auth_token = self.resolve_auth_token();
716 let privacy_config = self.build_privacy_config();
717 let allow_file_navigation = self.allow_file_navigation;
718 let db_search_paths = self.db_search_paths;
719 let on_ready = self.on_ready;
720 let commands = self.commands;
721 let listen_events = self.listen_events;
722 let probes = self.probes;
723 let js_init = js_bridge::init_script(&self.bridge_capacities);
724
725 Ok(Builder::new("victauri")
726 .setup(move |app, _api| {
727 let startup_timeline = introspection::StartupTimeline::new();
728 let event_log = EventLog::new(event_capacity);
729 startup_timeline.mark("event_log_created");
730 let registry = CommandRegistry::new();
731 startup_timeline.mark("registry_created");
732 let (shutdown_tx, shutdown_rx) = watch::channel(false);
733
734 let state = Arc::new(VictauriState {
735 event_log,
736 registry,
737 port: AtomicU16::new(port),
738 pending_evals: Arc::new(Mutex::new(HashMap::new())),
739 recorder: EventRecorder::new(recorder_capacity),
740 privacy: privacy_config,
741 eval_timeout,
742 shutdown_tx,
743 started_at: std::time::Instant::now(),
744 tool_invocations: AtomicU64::new(0),
745 allow_file_navigation,
746 command_timings: introspection::CommandTimings::new(),
747 fault_registry: introspection::FaultRegistry::new(),
748 contract_store: introspection::ContractStore::new(),
749 startup_timeline,
750 event_bus: introspection::EventBusMonitor::default(),
751 task_tracker: introspection::TaskTracker::new(),
752 bridge_ready: AtomicBool::new(false),
753 bridge_notify: tokio::sync::Notify::new(),
754 screencast: Arc::new(screencast::Screencast::default()),
755 db_search_paths,
756 probes: introspection::AppStateProbes::default(),
757 });
758 state.startup_timeline.mark("state_created");
759
760 for (name, probe) in probes {
761 state.probes.register(name, probe);
762 }
763
764 app.manage(state.clone());
765
766 for cmd in commands {
767 state.registry.register(cmd);
768 }
769 state.startup_timeline.mark("commands_registered");
770
771 for event_name in &listen_events {
773 let bus = state.event_bus.clone();
774 let name = event_name.clone();
775 app.listen_any(event_name.clone(), move |event| {
776 let payload =
777 serde_json::from_str::<serde_json::Value>(event.payload())
778 .map_or_else(
779 |_| event.payload().to_string(),
780 |v| v.to_string(),
781 );
782 bus.push(introspection::CapturedTauriEvent {
783 name: name.clone(),
784 payload,
785 timestamp: chrono::Utc::now()
786 .to_rfc3339_opts(chrono::SecondsFormat::Millis, true),
787 });
788 });
789 }
790 state
791 .startup_timeline
792 .mark("event_bus_listeners_registered");
793
794 if let Some(ref token) = auth_token {
795 let prefix_len = token.len().min(8);
796 let suffix_start = token.len().saturating_sub(4);
797 tracing::info!(
798 "Victauri MCP server auth enabled — token: {}…{}",
799 &token[..prefix_len],
800 &token[suffix_start..]
801 );
802 } else {
803 tracing::warn!(
804 "Victauri MCP server running WITHOUT auth — any localhost process can \
805 access all tools. Use VictauriBuilder::auth_enabled() or set \
806 VICTAURI_AUTH_TOKEN for shared/CI environments."
807 );
808 }
809
810 state.startup_timeline.mark("server_spawning");
811 let app_handle = app.clone();
812 let ready_state = state.clone();
813 let server_finished = state.task_tracker.track("mcp_server");
814 tauri::async_runtime::spawn(async move {
815 match mcp::start_server_with_options(
816 app_handle,
817 state,
818 port,
819 auth_token,
820 shutdown_rx,
821 )
822 .await
823 {
824 Ok(()) => {
825 tracing::info!("Victauri MCP server stopped");
826 }
827 Err(e) => {
828 tracing::error!("Victauri MCP server failed: {e}");
829 }
830 }
831 server_finished.store(true, std::sync::atomic::Ordering::Relaxed);
832 });
833
834 if let Some(cb) = on_ready {
835 let ready_finished = ready_state.task_tracker.track("on_ready_probe");
836 tauri::async_runtime::spawn(async move {
837 for _ in 0..50 {
838 tokio::time::sleep(std::time::Duration::from_millis(100)).await;
839 let actual_port =
840 ready_state.port.load(std::sync::atomic::Ordering::Relaxed);
841 if tokio::net::TcpStream::connect(format!(
842 "127.0.0.1:{actual_port}"
843 ))
844 .await
845 .is_ok()
846 {
847 cb(actual_port);
848 ready_finished
849 .store(true, std::sync::atomic::Ordering::Relaxed);
850 return;
851 }
852 }
853 let actual_port =
854 ready_state.port.load(std::sync::atomic::Ordering::Relaxed);
855 tracing::warn!(
856 "Victauri on_ready: server did not become ready within 5s"
857 );
858 cb(actual_port);
859 ready_finished.store(true, std::sync::atomic::Ordering::Relaxed);
860 });
861 }
862
863 emit_security_banner(port);
864 Ok(())
865 })
866 .on_event(|app, event| {
867 let Some(state) = app.try_state::<Arc<VictauriState>>() else {
868 return;
869 };
870 match event {
871 RunEvent::Exit => {
872 let _ = state.shutdown_tx.send(true);
873 tracing::info!("Victauri shutdown signal sent");
874 }
875 RunEvent::ExitRequested { .. } => {
876 state.event_bus.push(introspection::CapturedTauriEvent {
877 name: "tauri://exit-requested".to_string(),
878 payload: String::new(),
879 timestamp: chrono::Utc::now()
880 .to_rfc3339_opts(chrono::SecondsFormat::Millis, true),
881 });
882 }
883 RunEvent::WindowEvent {
884 label,
885 event: win_event,
886 ..
887 } => {
888 let (name, payload) = format_window_event(label, win_event);
889 state.event_bus.push(introspection::CapturedTauriEvent {
890 name,
891 payload,
892 timestamp: chrono::Utc::now()
893 .to_rfc3339_opts(chrono::SecondsFormat::Millis, true),
894 });
895 }
896 _ => {}
897 }
898 })
899 .js_init_script(js_init)
900 .invoke_handler(tauri::generate_handler![
901 tools::victauri_eval_js,
902 tools::victauri_eval_callback,
903 tools::victauri_get_window_state,
904 tools::victauri_list_windows,
905 tools::victauri_get_ipc_log,
906 tools::victauri_get_registry,
907 tools::victauri_get_memory_stats,
908 tools::victauri_dom_snapshot,
909 tools::victauri_verify_state,
910 tools::victauri_detect_ghost_commands,
911 tools::victauri_check_ipc_integrity,
912 ])
913 .build())
914 }
915 }
916}
917
918#[cfg(debug_assertions)]
919fn format_window_event(label: &str, event: &tauri::WindowEvent) -> (String, String) {
920 match event {
921 tauri::WindowEvent::Resized(size) => (
922 format!("window:{label}:resized"),
923 serde_json::json!({"width": size.width, "height": size.height}).to_string(),
924 ),
925 tauri::WindowEvent::Moved(pos) => (
926 format!("window:{label}:moved"),
927 serde_json::json!({"x": pos.x, "y": pos.y}).to_string(),
928 ),
929 tauri::WindowEvent::CloseRequested { .. } => {
930 (format!("window:{label}:close-requested"), String::new())
931 }
932 tauri::WindowEvent::Destroyed => (format!("window:{label}:destroyed"), String::new()),
933 tauri::WindowEvent::Focused(focused) => (
934 format!("window:{label}:focused"),
935 serde_json::json!({"focused": focused}).to_string(),
936 ),
937 tauri::WindowEvent::ScaleFactorChanged { scale_factor, .. } => (
938 format!("window:{label}:scale-factor-changed"),
939 serde_json::json!({"scale_factor": scale_factor}).to_string(),
940 ),
941 tauri::WindowEvent::ThemeChanged(theme) => (
942 format!("window:{label}:theme-changed"),
943 serde_json::json!({"theme": format!("{theme:?}")}).to_string(),
944 ),
945 tauri::WindowEvent::DragDrop(drag_event) => (
946 format!("window:{label}:drag-drop"),
947 format!("{drag_event:?}"),
948 ),
949 _ => (format!("window:{label}:other"), format!("{event:?}")),
950 }
951}
952
953#[cfg(debug_assertions)]
957fn env_truthy(name: &str) -> bool {
958 std::env::var(name).is_ok_and(|v| {
959 matches!(
960 v.trim().to_ascii_lowercase().as_str(),
961 "1" | "true" | "yes" | "on"
962 )
963 })
964}
965
966#[cfg(debug_assertions)]
975fn emit_security_banner(port: u16) {
976 tracing::warn!(
977 "┌─ VICTAURI INTROSPECTION SERVER ACTIVE ─────────────────────────────\n\
978 │ Listening on http://127.0.0.1:{port} — exposes JS eval, IPC, the\n\
979 │ filesystem, and SQLite to local clients. DEBUG-ONLY developer tool;\n\
980 │ it must never reach end users.\n\
981 │ Seeing this in a shipped/release build means your release profile has\n\
982 │ `debug-assertions = true`. Turn that off, or hard-disable Victauri\n\
983 │ with the VICTAURI_DISABLE=1 environment variable.\n\
984 └────────────────────────────────────────────────────────────────────"
985 );
986}
987
988#[must_use]
1008pub fn init<R: Runtime>() -> TauriPlugin<R> {
1009 VictauriBuilder::new()
1010 .build()
1011 .expect("default Victauri configuration is always valid")
1012}
1013
1014#[must_use]
1023pub fn init_auto_discover<R: Runtime>() -> TauriPlugin<R> {
1024 VictauriBuilder::new()
1025 .auto_discover()
1026 .build()
1027 .expect("default Victauri configuration is always valid")
1028}
1029
1030#[cfg(test)]
1031mod tests {
1032 use super::*;
1033
1034 #[test]
1035 fn builder_default_values() {
1036 let builder = VictauriBuilder::new();
1037 assert_eq!(builder.event_capacity, DEFAULT_EVENT_CAPACITY);
1038 assert_eq!(builder.recorder_capacity, DEFAULT_RECORDER_CAPACITY);
1039 assert!(builder.auth_token.is_none());
1040 assert!(!builder.auth_explicitly_enabled);
1041 assert!(!builder.auth_explicitly_disabled);
1042 let resolved = builder.resolve_auth_token();
1043 assert!(
1044 resolved.is_some(),
1045 "auth should be enabled by default (auto-generated token)"
1046 );
1047 assert_eq!(
1048 resolved.unwrap().len(),
1049 36,
1050 "auto-generated token should be UUID v4"
1051 );
1052 assert!(builder.disabled_tools.is_empty());
1053 assert!(builder.command_allowlist.is_none());
1054 assert!(builder.command_blocklist.is_empty());
1055 assert!(!builder.redaction_enabled);
1056 assert!(!builder.strict_privacy);
1057 }
1058
1059 #[test]
1060 fn builder_port_override() {
1061 let builder = VictauriBuilder::new().port(9090);
1062 assert_eq!(builder.resolve_port(), 9090);
1063 }
1064
1065 #[test]
1066 #[allow(unsafe_code)]
1067 fn builder_default_port() {
1068 let builder = VictauriBuilder::new();
1069 unsafe { std::env::remove_var("VICTAURI_PORT") };
1071 assert_eq!(builder.resolve_port(), DEFAULT_PORT);
1072 }
1073
1074 #[test]
1075 fn builder_auth_token_explicit() {
1076 let builder = VictauriBuilder::new().auth_token("my-secret");
1077 assert_eq!(builder.resolve_auth_token(), Some("my-secret".to_string()));
1078 }
1079
1080 #[cfg(debug_assertions)]
1084 #[test]
1085 #[allow(unsafe_code)]
1086 fn env_truthy_recognizes_kill_switch_values() {
1087 let key = "VICTAURI_TEST_KILL_SWITCH";
1089 unsafe { std::env::remove_var(key) };
1091 assert!(!env_truthy(key), "unset should be false");
1092
1093 for v in ["1", "true", "TRUE", " yes ", "On"] {
1094 unsafe { std::env::set_var(key, v) };
1096 assert!(env_truthy(key), "{v:?} should be truthy");
1097 }
1098 for v in ["0", "false", "", "nope"] {
1099 unsafe { std::env::set_var(key, v) };
1101 assert!(!env_truthy(key), "{v:?} should be falsy");
1102 }
1103 unsafe { std::env::remove_var(key) };
1105 }
1106
1107 #[test]
1108 fn builder_auth_enabled() {
1109 let builder = VictauriBuilder::new().auth_enabled();
1110 assert!(builder.auth_explicitly_enabled);
1111 let token = builder.resolve_auth_token().unwrap();
1112 assert_eq!(token.len(), 36, "auto-generated token should be a UUID");
1113 }
1114
1115 #[test]
1116 fn builder_auth_generate_token() {
1117 let builder = VictauriBuilder::new().generate_auth_token();
1118 let token = builder.resolve_auth_token().unwrap();
1119 assert_eq!(token.len(), 36);
1120 }
1121
1122 #[test]
1123 fn builder_auth_disabled_suppresses_default_token() {
1124 let builder = VictauriBuilder::new().auth_disabled();
1125 assert!(
1126 builder.resolve_auth_token().is_none(),
1127 "auth_disabled must suppress the default auto-generated token (auth is ON by default)"
1128 );
1129 }
1130
1131 #[test]
1132 fn builder_auth_disabled_returns_none() {
1133 let builder = VictauriBuilder::new().auth_disabled();
1134 assert!(
1135 builder.resolve_auth_token().is_none(),
1136 "auth_disabled should suppress auto-generated token"
1137 );
1138 }
1139
1140 #[test]
1141 fn builder_auth_disabled_overrides_explicit_token() {
1142 let builder = VictauriBuilder::new()
1143 .auth_token("my-secret")
1144 .auth_disabled();
1145 assert!(
1146 builder.resolve_auth_token().is_none(),
1147 "auth_disabled should override explicit token"
1148 );
1149 }
1150
1151 #[test]
1152 fn builder_capacities() {
1153 let builder = VictauriBuilder::new()
1154 .event_capacity(500)
1155 .recorder_capacity(2000);
1156 assert_eq!(builder.event_capacity, 500);
1157 assert_eq!(builder.recorder_capacity, 2000);
1158 }
1159
1160 #[test]
1161 fn builder_disable_tools() {
1162 let builder = VictauriBuilder::new().disable_tools(&["eval_js", "screenshot"]);
1163 assert_eq!(builder.disabled_tools.len(), 2);
1164 assert!(builder.disabled_tools.contains(&"eval_js".to_string()));
1165 }
1166
1167 #[test]
1168 fn builder_command_allowlist() {
1169 let builder = VictauriBuilder::new().command_allowlist(&["greet", "increment"]);
1170 assert!(builder.command_allowlist.is_some());
1171 assert_eq!(builder.command_allowlist.as_ref().unwrap().len(), 2);
1172 }
1173
1174 #[test]
1175 fn builder_command_blocklist() {
1176 let builder = VictauriBuilder::new().command_blocklist(&["dangerous_cmd"]);
1177 assert_eq!(builder.command_blocklist.len(), 1);
1178 }
1179
1180 #[test]
1181 fn builder_redaction() {
1182 let builder = VictauriBuilder::new()
1183 .add_redaction_pattern(r"SECRET_\w+")
1184 .enable_redaction();
1185 assert!(builder.redaction_enabled);
1186 assert_eq!(builder.redaction_patterns.len(), 1);
1187 }
1188
1189 #[test]
1190 fn builder_strict_privacy_config() {
1191 let builder = VictauriBuilder::new().strict_privacy_mode();
1192 let config = builder.build_privacy_config();
1193 assert!(config.redaction_enabled);
1194 assert_eq!(config.profile, crate::privacy::PrivacyProfile::Observe);
1195 assert!(!config.is_tool_enabled("eval_js"));
1196 assert!(!config.is_tool_enabled("screenshot"));
1197 assert!(!config.is_tool_enabled("interact"));
1198 assert!(config.is_tool_enabled("dom_snapshot"));
1199 }
1200
1201 #[test]
1202 fn builder_normal_privacy_config() {
1203 let builder = VictauriBuilder::new()
1204 .command_blocklist(&["secret_cmd"])
1205 .disable_tools(&["eval_js"]);
1206 let config = builder.build_privacy_config();
1207 assert!(config.command_blocklist.contains("secret_cmd"));
1208 assert!(!config.is_tool_enabled("eval_js"));
1209 assert!(!config.redaction_enabled);
1210 }
1211
1212 #[test]
1213 fn builder_strict_with_extra_blocklist() {
1214 let builder = VictauriBuilder::new()
1215 .strict_privacy_mode()
1216 .command_blocklist(&["extra_dangerous"]);
1217 let config = builder.build_privacy_config();
1218 assert!(config.command_blocklist.contains("extra_dangerous"));
1219 assert!(!config.is_tool_enabled("eval_js"));
1220 }
1221
1222 #[test]
1223 fn builder_test_profile() {
1224 let builder = VictauriBuilder::new().privacy_profile(crate::privacy::PrivacyProfile::Test);
1225 let config = builder.build_privacy_config();
1226 assert_eq!(config.profile, crate::privacy::PrivacyProfile::Test);
1227 assert!(config.redaction_enabled);
1228 assert!(config.is_tool_enabled("interact"));
1229 assert!(config.is_tool_enabled("fill"));
1230 assert!(config.is_tool_enabled("recording"));
1231 assert!(!config.is_tool_enabled("eval_js"));
1232 assert!(!config.is_tool_enabled("screenshot"));
1233 assert!(!config.is_tool_enabled("navigate.go_to"));
1236 }
1237
1238 #[test]
1239 fn builder_profile_with_extra_disables() {
1240 let builder = VictauriBuilder::new()
1241 .privacy_profile(crate::privacy::PrivacyProfile::Test)
1242 .disable_tools(&["interact"]);
1243 let config = builder.build_privacy_config();
1244 assert!(!config.is_tool_enabled("interact"));
1245 assert!(config.is_tool_enabled("fill"));
1246 }
1247
1248 #[test]
1249 fn builder_bridge_capacities() {
1250 let builder = VictauriBuilder::new()
1251 .console_log_capacity(5000)
1252 .network_log_capacity(2000)
1253 .navigation_log_capacity(500);
1254 assert_eq!(builder.bridge_capacities.console_logs, 5000);
1255 assert_eq!(builder.bridge_capacities.network_log, 2000);
1256 assert_eq!(builder.bridge_capacities.navigation_log, 500);
1257 assert_eq!(builder.bridge_capacities.mutation_log, 500);
1258 assert_eq!(builder.bridge_capacities.dialog_log, 100);
1259 }
1260
1261 #[test]
1262 fn builder_on_ready_sets_callback() {
1263 let builder = VictauriBuilder::new().on_ready(|_port| {});
1264 assert!(builder.on_ready.is_some());
1265 }
1266
1267 #[test]
1268 fn builder_file_navigation_disabled_by_default() {
1269 let builder = VictauriBuilder::new();
1270 assert!(
1271 !builder.allow_file_navigation,
1272 "file navigation should be disabled by default"
1273 );
1274 }
1275
1276 #[test]
1277 fn builder_allow_file_navigation() {
1278 let builder = VictauriBuilder::new().allow_file_navigation();
1279 assert!(builder.allow_file_navigation);
1280 }
1281
1282 #[test]
1283 fn builder_listen_events() {
1284 let builder =
1285 VictauriBuilder::new().listen_events(&["notification-added", "settings-changed"]);
1286 assert_eq!(builder.listen_events.len(), 2);
1287 assert!(
1288 builder
1289 .listen_events
1290 .contains(&"notification-added".to_string())
1291 );
1292 assert!(
1293 builder
1294 .listen_events
1295 .contains(&"settings-changed".to_string())
1296 );
1297 }
1298
1299 #[test]
1300 fn builder_listen_events_empty_by_default() {
1301 let builder = VictauriBuilder::new();
1302 assert!(builder.listen_events.is_empty());
1303 }
1304
1305 #[test]
1306 fn init_script_contains_custom_capacities() {
1307 let caps = js_bridge::BridgeCapacities {
1308 console_logs: 3000,
1309 mutation_log: 750,
1310 network_log: 5000,
1311 navigation_log: 400,
1312 dialog_log: 250,
1313 long_tasks: 200,
1314 };
1315 let script = js_bridge::init_script(&caps);
1316 assert!(script.contains("CAP_CONSOLE = 3000"));
1317 assert!(script.contains("CAP_MUTATION = 750"));
1318 assert!(script.contains("CAP_NETWORK = 5000"));
1319 assert!(script.contains("CAP_NAVIGATION = 400"));
1320 assert!(script.contains("CAP_DIALOG = 250"));
1321 assert!(script.contains("CAP_LONG_TASKS = 200"));
1322 }
1323
1324 #[test]
1325 fn init_script_default_contains_standard_capacities() {
1326 let caps = js_bridge::BridgeCapacities::default();
1327 let script = js_bridge::init_script(&caps);
1328 assert!(script.contains("CAP_CONSOLE = 1000"));
1329 assert!(script.contains("CAP_NETWORK = 1000"));
1330 assert!(script.contains("window.__VICTAURI__"));
1331 }
1332
1333 #[test]
1334 fn builder_validates_defaults() {
1335 let builder = VictauriBuilder::new();
1336 assert!(builder.validate().is_ok());
1337 }
1338
1339 #[test]
1340 fn builder_rejects_zero_port() {
1341 let builder = VictauriBuilder::new().port(0);
1342 let err = builder.validate().unwrap_err();
1343 assert!(matches!(err, BuilderError::InvalidPort { port: 0, .. }));
1344 }
1345
1346 #[test]
1347 fn builder_rejects_zero_event_capacity() {
1348 let builder = VictauriBuilder::new().event_capacity(0);
1349 let err = builder.validate().unwrap_err();
1350 assert!(matches!(
1351 err,
1352 BuilderError::InvalidEventCapacity { capacity: 0, .. }
1353 ));
1354 }
1355
1356 #[test]
1357 fn builder_rejects_excessive_event_capacity() {
1358 let builder = VictauriBuilder::new().event_capacity(2_000_000);
1359 assert!(builder.validate().is_err());
1360 }
1361
1362 #[test]
1363 fn builder_rejects_zero_recorder_capacity() {
1364 let builder = VictauriBuilder::new().recorder_capacity(0);
1365 assert!(builder.validate().is_err());
1366 }
1367
1368 #[test]
1369 fn builder_rejects_zero_eval_timeout() {
1370 let builder = VictauriBuilder::new().eval_timeout(std::time::Duration::from_secs(0));
1371 assert!(builder.validate().is_err());
1372 }
1373
1374 #[test]
1375 fn builder_rejects_excessive_eval_timeout() {
1376 let builder = VictauriBuilder::new().eval_timeout(std::time::Duration::from_secs(600));
1377 assert!(builder.validate().is_err());
1378 }
1379
1380 #[test]
1381 fn builder_accepts_edge_values() {
1382 let builder = VictauriBuilder::new()
1383 .port(1)
1384 .event_capacity(1)
1385 .recorder_capacity(1)
1386 .eval_timeout(std::time::Duration::from_secs(1));
1387 assert!(builder.validate().is_ok());
1388
1389 let builder = VictauriBuilder::new()
1390 .port(65535)
1391 .event_capacity(MAX_EVENT_CAPACITY)
1392 .recorder_capacity(MAX_RECORDER_CAPACITY)
1393 .eval_timeout(std::time::Duration::from_secs(MAX_EVAL_TIMEOUT_SECS));
1394 assert!(builder.validate().is_ok());
1395 }
1396}