spacedrive/core/tests/cross_device_copy_test.rs

//! Cross-device copy test using the action system
//!
//! This test demonstrates the copy system's routing capabilities by having Alice
//! create files and then dispatch copy actions where the source SdPath is on
//! Alice's device and the destination is on Bob's device.

use sd_core::{
	domain::addressing::{SdPath, SdPathBatch},
	ops::files::copy::{action::FileCopyAction, CopyOptions},
	testing::CargoTestRunner,
	Core,
};
use std::{env, path::PathBuf, time::Duration};
use tokio::time::timeout;

/// Alice's cross-device copy scenario - sender role
#[tokio::test]
#[ignore] // Only run when explicitly called via subprocess
async fn alice_cross_device_copy_scenario() {
	// Exit early if not running as Alice
	if env::var("TEST_ROLE").unwrap_or_default() != "alice" {
		return;
	}

	// Set test directory for file-based discovery
	env::set_var(
		"SPACEDRIVE_TEST_DIR",
		"/tmp/spacedrive-cross-device-copy-test",
	);

	let data_dir = PathBuf::from("/tmp/spacedrive-cross-device-copy-test/alice");
	let device_name = "Alice's Test Device";

	println!("Alice: Starting Core cross-device copy test (sender)");
	println!("Alice: Data dir: {:?}", data_dir);

	// Initialize Core
	println!("Alice: Initializing Core...");
	let mut core = timeout(Duration::from_secs(10), Core::new(data_dir.clone()))
		.await
		.unwrap()
		.unwrap();
	println!("Alice: Core initialized successfully");

	// Set device name
	println!("Alice: Setting device name for testing...");
	core.device.set_name(device_name.to_string()).unwrap();

	// Initialize networking
	println!("Alice: Initializing networking...");
	timeout(Duration::from_secs(10), core.init_networking())
		.await
		.unwrap()
		.unwrap();

	// Wait longer for networking to fully initialize
	tokio::time::sleep(Duration::from_secs(3)).await;
	println!("Alice: Networking initialized successfully");

	// Create a library for job dispatch
	println!("Alice: Creating library for copy operations...");
	let library = core
		.libraries
		.create_library("Alice Copy Library", None, core.context.clone())
		.await
		.unwrap();
	let library_id = library.id();
	println!("Alice: Library created successfully (ID: {})", library_id);

	// Start pairing as initiator
	println!("Alice: Starting pairing as initiator...");
	let (pairing_code, expires_in) = if let Some(networking) = core.networking() {
		timeout(
			Duration::from_secs(15),
			networking.start_pairing_as_initiator(false),
		)
		.await
		.unwrap()
		.unwrap()
	} else {
		panic!("Networking not initialized");
	};

	let short_code = pairing_code
		.split_whitespace()
		.take(3)
		.collect::<Vec<_>>()
		.join(" ");
	println!(
		"Alice: Pairing code generated: {}... (expires in {}s)",
		short_code, expires_in
	);

	// Write pairing code to shared location for Bob to read
	std::fs::create_dir_all("/tmp/spacedrive-cross-device-copy-test").unwrap();
	std::fs::write(
		"/tmp/spacedrive-cross-device-copy-test/pairing_code.txt",
		&pairing_code,
	)
	.unwrap();
	println!(
		"Alice: Pairing code written to /tmp/spacedrive-cross-device-copy-test/pairing_code.txt"
	);

	// Wait for pairing completion
	println!("Alice: Waiting for Bob to connect...");
	let mut bob_device_id = None;
	let mut attempts = 0;
	let max_attempts = 45; // 45 seconds

	loop {
		tokio::time::sleep(Duration::from_secs(1)).await;

		let connected_devices = core
			.services
			.device
			.get_connected_devices_info()
			.await
			.unwrap();
		if !connected_devices.is_empty() {
			bob_device_id = Some(connected_devices[0].device_id);
			println!(
				"Alice: Bob connected! Device ID: {}",
				connected_devices[0].device_id
			);
			println!(
				"Alice: Connected device: {} ({})",
				connected_devices[0].device_name, connected_devices[0].device_id
			);

			// Wait for session keys to be established
			println!("Alice: Allowing extra time for session key establishment...");
			tokio::time::sleep(Duration::from_secs(2)).await;
			break;
		}

		attempts += 1;
		if attempts >= max_attempts {
			panic!("Alice: Pairing timeout - Bob not connected");
		}

		if attempts % 5 == 0 {
			println!("Alice: Pairing status check {} - waiting", attempts / 5);
		}
	}

	let bob_id = bob_device_id.unwrap();

	// Create test files to copy
	println!("Alice: Creating test files for cross-device copy...");
	let test_files_dir = data_dir.join("test_files");
	std::fs::create_dir_all(&test_files_dir).unwrap();

	let test_files = vec![
		("test1.txt", "Hello from Alice's device - file 1!"),
		("test2.txt", "Cross-device copy test - file 2"),
		(
			"test3.json",
			r#"{"test": "cross-device-copy", "from": "alice", "to": "bob"}"#,
		),
	];

	let mut source_paths = Vec::new();
	for (filename, content) in &test_files {
		let file_path = test_files_dir.join(filename);
		std::fs::write(&file_path, content).unwrap();
		println!("  Created: {} ({} bytes)", filename, content.len());
		source_paths.push(file_path);
	}

	// Write file list for Bob to expect
	let file_list: Vec<String> = test_files
		.iter()
		.map(|(name, content)| format!("{}:{}", name, content.len()))
		.collect();
	std::fs::write(
		"/tmp/spacedrive-cross-device-copy-test/expected_files.txt",
		file_list.join("\n"),
	)
	.unwrap();

	// Get Alice's device ID
	let alice_device_id = core.device.device_id().unwrap();
	println!("Alice: My device ID is {}", alice_device_id);

	// Prepare copy operations using the action system
	println!("Alice: Dispatching cross-device copy actions...");

	// Get the action manager from context
	let action_manager = core
		.context
		.get_action_manager()
		.await
		.expect("Action manager not initialized");

	// Copy each file individually to test the routing
	for (i, (source_path, (filename, _))) in source_paths.iter().zip(&test_files).enumerate() {
		println!("Alice: Preparing copy action {} for {}", i + 1, filename);

		// Create source SdPath (on Alice's device)
		let source_sdpath = SdPath::physical("alice-device".to_string(), source_path);

		// Create destination SdPath (on Bob's device)
		let dest_path = PathBuf::from("/tmp/received_files").join(filename);
		let dest_sdpath = SdPath::physical("bob-device".to_string(), &dest_path);

		println!(
			"  Source: {} (device: {})",
			source_path.display(),
			alice_device_id
		);
		println!(
			"  Destination: {} (device: {})",
			dest_path.display(),
			bob_id
		);

		// Build the copy action directly with SdPath
		let copy_action = FileCopyAction {
			sources: SdPathBatch::new(vec![source_sdpath]),
			destination: dest_sdpath,
			options: CopyOptions {
				overwrite: true,
				verify_checksum: true,
				preserve_timestamps: true,
				..Default::default()
			},
			on_conflict: None,
		};

		// Dispatch the action
		match action_manager
			.dispatch_library(Some(library_id), copy_action)
			.await
		{
			Ok(output) => {
				println!("Alice: Copy action {} dispatched successfully", i + 1);
				println!("  Output: {:?}", output);
			}
			Err(e) => {
				println!("Alice: Copy action {} failed: {}", i + 1, e);
				panic!("Failed to dispatch copy action: {}", e);
			}
		}

		// Small delay between operations
		tokio::time::sleep(Duration::from_millis(500)).await;
	}

	// Wait for Bob to confirm receipt
	println!("Alice: Waiting for Bob to confirm file receipt...");
	let mut bob_confirmed = false;
	for attempt in 1..=60 {
		if std::fs::read_to_string("/tmp/spacedrive-cross-device-copy-test/bob_verified.txt")
			.map(|content| content.starts_with("verified:"))
			.unwrap_or(false)
		{
			println!("Alice: Bob confirmed file receipt and verification!");
			bob_confirmed = true;
			break;
		}

		if attempt % 10 == 0 {
			println!(
				"Alice: Still waiting for Bob's confirmation... ({}s)",
				attempt
			);
		}
		tokio::time::sleep(Duration::from_secs(1)).await;
	}

	if bob_confirmed {
		println!("CROSS_DEVICE_COPY_SUCCESS: Alice successfully dispatched copy actions");
		std::fs::write(
			"/tmp/spacedrive-cross-device-copy-test/alice_success.txt",
			"success",
		)
		.unwrap();
	} else {
		panic!("Alice: Bob did not confirm file receipt within timeout");
	}

	println!("Alice: Cross-device copy test completed");
}

/// Bob's cross-device copy scenario - receiver role
#[tokio::test]
#[ignore] // Only run when explicitly called via subprocess
async fn bob_cross_device_copy_scenario() {
	// Exit early if not running as Bob
	if env::var("TEST_ROLE").unwrap_or_default() != "bob" {
		return;
	}

	// Set test directory for file-based discovery
	env::set_var(
		"SPACEDRIVE_TEST_DIR",
		"/tmp/spacedrive-cross-device-copy-test",
	);

	let data_dir = PathBuf::from("/tmp/spacedrive-cross-device-copy-test/bob");
	let device_name = "Bob's Test Device";

	println!("Bob: Starting Core cross-device copy test (receiver)");
	println!("Bob: Data dir: {:?}", data_dir);

	// Initialize Core
	println!("Bob: Initializing Core...");
	let mut core = timeout(Duration::from_secs(10), Core::new(data_dir))
		.await
		.unwrap()
		.unwrap();
	println!("Bob: Core initialized successfully");

	// Set device name
	println!("Bob: Setting device name for testing...");
	core.device.set_name(device_name.to_string()).unwrap();

	// Initialize networking
	println!("Bob: Initializing networking...");
	timeout(Duration::from_secs(10), core.init_networking())
		.await
		.unwrap()
		.unwrap();

	// Wait longer for networking to fully initialize
	tokio::time::sleep(Duration::from_secs(3)).await;
	println!("Bob: Networking initialized successfully");

	// Create a library for job dispatch
	println!("Bob: Creating library for copy operations...");
	let _library = core
		.libraries
		.create_library("Bob Copy Library", None, core.context.clone())
		.await
		.unwrap();
	println!("Bob: Library created successfully");

	// Wait for Alice to create pairing code
	println!("Bob: Looking for pairing code from Alice...");
	let pairing_code = loop {
		if let Ok(code) =
			std::fs::read_to_string("/tmp/spacedrive-cross-device-copy-test/pairing_code.txt")
		{
			break code.trim().to_string();
		}
		tokio::time::sleep(Duration::from_millis(500)).await;
	};
	println!("Bob: Found pairing code");

	// Join pairing session
	println!("Bob: Joining pairing with Alice...");
	if let Some(networking) = core.networking() {
		timeout(
			Duration::from_secs(15),
			networking.start_pairing_as_joiner(&pairing_code, false),
		)
		.await
		.unwrap()
		.unwrap();
	} else {
		panic!("Networking not initialized");
	}
	println!("Bob: Successfully joined pairing");

	// Wait for pairing completion
	println!("Bob: Waiting for pairing to complete...");
	let mut attempts = 0;
	let max_attempts = 30;

	loop {
		tokio::time::sleep(Duration::from_secs(1)).await;

		let connected_devices = core
			.services
			.device
			.get_connected_devices_info()
			.await
			.unwrap();
		if !connected_devices.is_empty() {
			println!("Bob: Pairing completed successfully!");
			println!(
				"Bob: Connected to {} ({})",
				connected_devices[0].device_name, connected_devices[0].device_id
			);

			// Wait for session keys
			println!("Bob: Allowing extra time for session key establishment...");
			tokio::time::sleep(Duration::from_secs(2)).await;
			break;
		}

		attempts += 1;
		if attempts >= max_attempts {
			panic!("Bob: Pairing timeout - no devices connected");
		}

		if attempts % 5 == 0 {
			println!("Bob: Pairing status check {} - waiting", attempts / 5);
		}
	}

	// Create directory for received files
	let received_dir = std::path::Path::new("/tmp/received_files");
	std::fs::create_dir_all(received_dir).unwrap();
	println!(
		"Bob: Created directory for received files: {:?}",
		received_dir
	);

	// Load expected files
	println!("Bob: Loading expected file list...");
	let expected_files = loop {
		if let Ok(content) =
			std::fs::read_to_string("/tmp/spacedrive-cross-device-copy-test/expected_files.txt")
		{
			break content
				.lines()
				.map(|line| {
					let parts: Vec<&str> = line.split(':').collect();
					(parts[0].to_string(), parts[1].parse::<usize>().unwrap_or(0))
				})
				.collect::<Vec<(String, usize)>>();
		}
		tokio::time::sleep(Duration::from_millis(500)).await;
	};

	println!(
		"Bob: Expecting {} files via cross-device copy",
		expected_files.len()
	);
	for (filename, size) in &expected_files {
		println!("  Expecting: {} ({} bytes)", filename, size);
	}

	// Monitor for received files
	println!("Bob: Waiting for files to arrive via action system...");
	let mut received_files = Vec::new();
	let start_time = std::time::Instant::now();
	let timeout_duration = Duration::from_secs(60);

	while received_files.len() < expected_files.len() && start_time.elapsed() < timeout_duration {
		tokio::time::sleep(Duration::from_secs(1)).await;

		// Check for new files in received directory
		if let Ok(entries) = std::fs::read_dir(received_dir) {
			for entry in entries {
				if let Ok(entry) = entry {
					let filename = entry.file_name().to_string_lossy().to_string();
					if !received_files.contains(&filename) {
						if let Ok(metadata) = entry.metadata() {
							received_files.push(filename.clone());
							println!(
								"Bob: Received file: {} ({} bytes)",
								filename,
								metadata.len()
							);

							// Verify file size
							if let Some((_, expected_size)) =
								expected_files.iter().find(|(name, _)| name == &filename)
							{
								if metadata.len() == *expected_size as u64 {
									println!("  Size verified: {} bytes", metadata.len());
								} else {
									println!(
										"  Size mismatch: expected {}, got {}",
										expected_size,
										metadata.len()
									);
								}
							}
						}
					}
				}
			}
		}

		let elapsed = start_time.elapsed().as_secs();
		if elapsed > 0 && elapsed % 10 == 0 && received_files.is_empty() {
			println!("Bob: Still waiting for files... ({}s elapsed)", elapsed);
		}
	}

	// Verify all expected files were received
	if received_files.len() == expected_files.len() {
		println!("Bob: All expected files received successfully!");

		// Write verification confirmation
		std::fs::write(
			"/tmp/spacedrive-cross-device-copy-test/bob_verified.txt",
			format!("verified:{}", chrono::Utc::now().timestamp()),
		)
		.unwrap();

		// Write success marker
		std::fs::write(
			"/tmp/spacedrive-cross-device-copy-test/bob_success.txt",
			"success",
		)
		.unwrap();

		println!("CROSS_DEVICE_COPY_SUCCESS: Bob verified all received files");
	} else {
		println!(
			"Bob: Only received {}/{} expected files",
			received_files.len(),
			expected_files.len()
		);
		panic!("Bob: Not all files were received");
	}

	println!("Bob: Cross-device copy test completed");
}

/// Main test orchestrator - spawns cargo test subprocesses
#[tokio::test]
async fn test_cross_device_copy() {
	// Clean up any old test files
	let _ = std::fs::remove_dir_all("/tmp/spacedrive-cross-device-copy-test");
	let _ = std::fs::remove_dir_all("/tmp/received_files");
	std::fs::create_dir_all("/tmp/spacedrive-cross-device-copy-test").unwrap();

	println!("Testing cross-device copy with action system routing");

	let mut runner = CargoTestRunner::for_test_file("cross_device_copy_test")
		.with_timeout(Duration::from_secs(180))
		.add_subprocess("alice", "alice_cross_device_copy_scenario")
		.add_subprocess("bob", "bob_cross_device_copy_scenario");

	// Spawn Alice first
	println!("Starting Alice as copy action dispatcher...");
	runner
		.spawn_single_process("alice")
		.await
		.expect("Failed to spawn Alice");

	// Wait for Alice to initialize
	tokio::time::sleep(Duration::from_secs(8)).await;

	// Start Bob as receiver
	println!("Starting Bob as copy receiver...");
	runner
		.spawn_single_process("bob")
		.await
		.expect("Failed to spawn Bob");

	// Run until both complete successfully
	let result = runner
		.wait_for_success(|_outputs| {
			let alice_success =
				std::fs::read_to_string("/tmp/spacedrive-cross-device-copy-test/alice_success.txt")
					.map(|content| content.trim() == "success")
					.unwrap_or(false);
			let bob_success =
				std::fs::read_to_string("/tmp/spacedrive-cross-device-copy-test/bob_success.txt")
					.map(|content| content.trim() == "success")
					.unwrap_or(false);

			alice_success && bob_success
		})
		.await;

	match result {
		Ok(_) => {
			println!("Cross-device copy test successful! Action system routing works correctly.");
		}
		Err(e) => {
			println!("Cross-device copy test failed: {}", e);
			for (name, output) in runner.get_all_outputs() {
				println!("\n{} output:\n{}", name, output);
			}
			panic!("Cross-device copy test failed");
		}
	}
}