1#![allow(dead_code)]
2
3#[cfg(forge_backend)]
4use std::collections::HashMap;
5use std::env;
6#[cfg(feature = "java-forge")]
7use std::io::{BufRead, BufReader, BufWriter, Write};
8use std::path::{Path, PathBuf};
9#[cfg(feature = "java-forge")]
10use std::process::{Child, ChildStdin, Command, Stdio};
11use std::sync::atomic::AtomicBool;
12use std::sync::mpsc as std_mpsc;
13#[cfg(feature = "java-forge")]
14use std::sync::mpsc::RecvTimeoutError;
15#[cfg(forge_backend)]
16use std::sync::mpsc::TryRecvError;
17use std::sync::Arc;
18#[cfg(feature = "java-forge")]
19use std::sync::Mutex;
20#[cfg(forge_backend)]
21use std::time::Duration;
22#[cfg(feature = "java-forge")]
23use std::time::Instant;
24
25use manabrew_protocol::deck_dto::{Deck, DeckCardIdentity};
26
27use crate::config::DeckSelection;
28#[cfg(feature = "java-forge")]
29use manabot::{BotAgent, SimpleAi};
30#[cfg(forge_backend)]
31use manabrew_agent_interface::game_view_dto::GameViewDto;
32use manabrew_agent_interface::prompt::{AgentMessage, ClientToServerMessage};
33#[cfg(forge_backend)]
34use manabrew_agent_interface::prompt::{
35 AgentPrompt, ChooseActionOutput, DiceRolledOutput, DirectiveInput, GameOverInput, PromptInput,
36 PromptOutput, ProtocolError, ProtocolErrorCode, ResponseViolation, StateUpdate,
37};
38#[cfg(feature = "java-forge")]
39use manabrew_agent_interface::prompt::{MulliganOutput, MulliganPutBackOutput};
40use serde::Serialize;
41#[cfg(feature = "java-forge")]
42use serde_json::json;
43#[cfg(feature = "java-forge")]
44use serde_json::Value;
45#[cfg(forge_backend)]
46use tracing::warn;
47#[cfg(forge_backend)]
48use tracing::{debug, info};
49
50use super::HostedGameOver;
51use crate::config::workspace_root;
52
53pub fn unsupported_message() -> &'static str {
54 "hosted java-forge backend is unavailable; rebuild self-hosted-node with --features java-forge"
55}
56
57#[cfg(feature = "java-forge")]
58pub fn run_smoke_game(max_prompts: usize) -> Result<(), String> {
59 let config = JavaRuntimeConfig::from_env();
60 let assets_dir = config.assets_dir.to_string_lossy().to_string();
61 let bridge = SubprocessBridge::spawn(&config)?;
62 let mut session = JavaForgeSession::new(bridge);
63 session.initialize(&assets_dir)?;
64
65 let deck_a = smoke_deck("Mountain", "Lightning Bolt");
66 let deck_b = smoke_deck("Forest", "Grizzly Bears");
67 let request = StartGameRequest::new(
68 "self-hosted-java-smoke".to_string(),
69 String::new(),
70 20,
71 42,
72 vec![
73 PlayerConfig::new("Smoke A".to_string(), &deck_a, Vec::new()),
74 PlayerConfig::new("Smoke B".to_string(), &deck_b, Vec::new()),
75 ],
76 );
77 let session_id = session.start_game(&request)?;
78 info!(session_id, "java-forge smoke session started");
79
80 let mut prompts_seen = 0usize;
81 while prompts_seen < max_prompts {
82 let Some(prompt_json) = wait_for_prompt(&mut session, 600)? else {
83 session.end_game()?;
84 return Err("timed out waiting for java-forge smoke prompt".to_string());
85 };
86 let prompt: AgentPrompt = serde_json::from_str(&prompt_json)
87 .map_err(|err| format!("failed to parse java-forge smoke prompt: {err}"))?;
88 let player = player_index(&prompt.deciding_player_id);
89 info!(prompts_seen, player, "java-forge smoke prompt");
90 let pass = PromptOutput::ChooseAction(ChooseActionOutput::Pass {
91 until: None,
92 exhaust_stack: false,
93 });
94 session.submit_action(&serde_json::to_string(&pass).map_err(|err| err.to_string())?)?;
95 prompts_seen += 1;
96 }
97
98 let snapshot_json = session.get_snapshot(Some(0))?;
99 let snapshot: Value = serde_json::from_str(&snapshot_json)
100 .map_err(|err| format!("failed to parse java-forge smoke snapshot: {err}"))?;
101 info!(
102 turn = snapshot
103 .get("turn")
104 .and_then(|value| value.as_i64())
105 .unwrap_or_default(),
106 phase = snapshot
107 .get("phase")
108 .and_then(|value| value.as_str())
109 .unwrap_or("<missing>"),
110 "java-forge smoke snapshot"
111 );
112 session.end_game()?;
113 Ok(())
114}
115
116#[cfg(not(feature = "java-forge"))]
117pub fn run_smoke_game(_max_prompts: usize) -> Result<(), String> {
118 Err(
119 "java-forge smoke requires building self-hosted-node with --features java-forge"
120 .to_string(),
121 )
122}
123
124#[cfg(feature = "graal-forge")]
130pub fn run_graal_smoke() -> Result<(), String> {
131 let config = JavaRuntimeConfig::from_env();
132 let engine = GraalEngineHandle::create(&config.assets_dir)?;
133
134 let deck_a = smoke_deck("Mountain", "Lightning Bolt");
135 let deck_b = smoke_deck("Forest", "Grizzly Bears");
136 let request = StartGameRequest::new(
137 "self-hosted-graal-smoke".to_string(),
138 String::new(),
139 20,
140 42,
141 vec![
142 PlayerConfig::new("Smoke A".to_string(), &deck_a, Vec::new()),
143 PlayerConfig::new("Smoke B".to_string(), &deck_b, Vec::new()),
144 ],
145 );
146
147 let session_id = engine.start_game(&request.to_json().map_err(|err| err.to_string())?)?;
148 info!(session_id, "graal-forge smoke session started");
149 engine.end_game(&session_id)?;
150 Ok(())
151}
152
153#[cfg(not(feature = "graal-forge"))]
154pub fn run_graal_smoke() -> Result<(), String> {
155 Err(
156 "graal-forge smoke requires building self-hosted-node with --features graal-forge"
157 .to_string(),
158 )
159}
160
161#[cfg(feature = "java-forge")]
162pub fn run_scenario(name: &str, max_prompts: usize) -> Result<(), String> {
163 let scenario = JavaScenario::from_name(name)?;
164 let config = JavaRuntimeConfig::from_env();
165 let assets_dir = config.assets_dir.to_string_lossy().to_string();
166 let bridge = SubprocessBridge::spawn(&config)?;
167 let mut session = JavaForgeSession::new(bridge);
168 session.initialize(&assets_dir)?;
169
170 let request = StartGameRequest::new(
171 format!("self-hosted-java-scenario-{}", scenario.name()),
172 String::new(),
173 20,
174 42,
175 vec![
176 PlayerConfig::new(
177 "Scenario A".to_string(),
178 &scenario_deck("Swamp"),
179 Vec::new(),
180 ),
181 PlayerConfig::new(
182 "Scenario B".to_string(),
183 &scenario_deck("Forest"),
184 Vec::new(),
185 ),
186 ],
187 );
188 let session_id = session.start_game(&request)?;
189 info!(
190 session_id,
191 scenario = scenario.name(),
192 "java-forge scenario started"
193 );
194
195 let result = run_scenario_loop(&mut session, scenario, max_prompts);
196 let end_result = session.end_game();
197 result.and(end_result)
198}
199
200#[cfg(not(feature = "java-forge"))]
201pub fn run_scenario(_name: &str, _max_prompts: usize) -> Result<(), String> {
202 Err(
203 "java-forge scenarios require building self-hosted-node with --features java-forge"
204 .to_string(),
205 )
206}
207
208#[cfg(feature = "java-forge")]
209pub fn run_self_play(
210 seats: &[DeckSelection],
211 starting_life: i32,
212 seed: u64,
213 max_prompts: usize,
214 games: usize,
215) -> Result<(), String> {
216 let config = JavaRuntimeConfig::from_env();
217 let assets_dir = config.assets_dir.to_string_lossy().to_string();
218 let bridge = SubprocessBridge::spawn(&config)?;
219 let mut session = JavaForgeSession::new(bridge);
220 session.initialize(&assets_dir)?;
221
222 let mut players = Vec::with_capacity(seats.len());
223 for (i, seat) in seats.iter().enumerate() {
224 let identities = deck_card_identities(&seat.deck);
225 players.push(PlayerConfig::new(
226 format!("Self-Play {}", i + 1),
227 &identities,
228 commander_names_for_java(&seat.deck, seat.commander_name.as_deref()),
229 ));
230 }
231
232 for game_index in 0..games.max(1) {
233 let request = StartGameRequest::new(
234 format!("self-hosted-java-self-play-{game_index}"),
235 String::new(),
236 starting_life,
237 seed.wrapping_add(game_index as u64),
238 players.clone(),
239 );
240 let session_id = session.start_game(&request)?;
241 info!(
242 session_id,
243 game_index,
244 games,
245 players = seats.len(),
246 starting_life,
247 max_prompts,
248 "java-forge self-play game started"
249 );
250 let result = run_self_play_loop(&mut session, max_prompts);
251 let end_result = session.end_game();
252 result.and(end_result)?;
253 }
254 Ok(())
255}
256
257#[cfg(not(feature = "java-forge"))]
258pub fn run_self_play(
259 _seats: &[DeckSelection],
260 _starting_life: i32,
261 _seed: u64,
262 _max_prompts: usize,
263 _games: usize,
264) -> Result<(), String> {
265 Err(
266 "java-forge self-play requires building self-hosted-node with --features java-forge"
267 .to_string(),
268 )
269}
270
271#[cfg(feature = "java-forge")]
272type SharedBridge = Arc<Mutex<SubprocessBridge>>;
273
274#[cfg(feature = "java-forge")]
275struct PoolSlot {
276 bridge: SharedBridge,
277 active: usize,
278}
279
280#[cfg(feature = "java-forge")]
281pub struct JavaEnginePool {
282 config: JavaRuntimeConfig,
283 max_sessions: usize,
284 sessions_per_process: usize,
285 slots: Mutex<Vec<PoolSlot>>,
286 in_use: Mutex<HashMap<String, SharedBridge>>,
287}
288
289#[cfg(feature = "java-forge")]
290#[derive(Clone)]
291pub struct JavaEngineHandle {
292 pool: Arc<JavaEnginePool>,
293}
294
295#[cfg(feature = "java-forge")]
296impl JavaEnginePool {
297 pub fn start(
298 config: &JavaRuntimeConfig,
299 max_sessions: usize,
300 sessions_per_process: usize,
301 ) -> Result<Arc<Self>, String> {
302 let max_sessions = max_sessions.max(1);
303 let sessions_per_process = sessions_per_process.max(1);
304 let processes = max_sessions.div_ceil(sessions_per_process);
305 let mut slots = Vec::with_capacity(processes);
306 for slot in 0..processes {
307 info!(
308 slot,
309 processes, sessions_per_process, "pre-warming java subprocess"
310 );
311 let bridge = SubprocessBridge::spawn(config)?;
312 slots.push(PoolSlot {
313 bridge: Arc::new(Mutex::new(bridge)),
314 active: 0,
315 });
316 }
317 Ok(Arc::new(Self {
318 config: config.clone(),
319 max_sessions,
320 sessions_per_process,
321 slots: Mutex::new(slots),
322 in_use: Mutex::new(HashMap::new()),
323 }))
324 }
325
326 pub fn handle(self: &Arc<Self>) -> JavaEngineHandle {
327 JavaEngineHandle {
328 pool: Arc::clone(self),
329 }
330 }
331}
332
333#[cfg(feature = "java-forge")]
334impl Drop for JavaEnginePool {
335 fn drop(&mut self) {
336 let slots = self.slots.get_mut().map(std::mem::take).unwrap_or_default();
337 for slot in slots {
338 if let Ok(mutex) = Arc::try_unwrap(slot.bridge) {
339 if let Ok(inner) = mutex.into_inner() {
340 inner.shutdown();
341 }
342 }
343 }
344 }
345}
346
347#[cfg(feature = "java-forge")]
348impl JavaEnginePool {
349 fn acquire(&self) -> Result<SharedBridge, String> {
350 let deadline = Instant::now() + Duration::from_secs(60);
351 loop {
352 let claimed = {
353 let mut slots = self
354 .slots
355 .lock()
356 .map_err(|_| "java engine slots poisoned".to_string())?;
357 let mut claimed = None;
358 for slot in slots.iter_mut() {
359 if slot.active < self.sessions_per_process {
360 slot.active += 1;
361 claimed = Some(Arc::clone(&slot.bridge));
362 break;
363 }
364 }
365 claimed
366 };
367 if let Some(bridge) = claimed {
368 let alive = bridge
369 .lock()
370 .ok()
371 .map(|mut guard| guard.is_alive())
372 .unwrap_or(false);
373 if alive {
374 return Ok(bridge);
375 }
376 warn!("discarding dead java subprocess from pool");
377 self.replace_slot(&bridge);
378 continue;
379 }
380 if Instant::now() >= deadline {
381 return Err(format!(
382 "java engine pool exhausted (max_sessions={}); no free session slot after 60s",
383 self.max_sessions
384 ));
385 }
386 std::thread::sleep(Duration::from_millis(50));
387 }
388 }
389
390 fn release(&self, bridge: SharedBridge) {
391 let now_idle = {
392 let mut slots = match self.slots.lock() {
393 Ok(slots) => slots,
394 Err(_) => return,
395 };
396 match slots.iter_mut().find(|s| Arc::ptr_eq(&s.bridge, &bridge)) {
397 Some(slot) => {
398 slot.active = slot.active.saturating_sub(1);
399 slot.active == 0
400 }
401 None => return,
402 }
403 };
404 if !now_idle {
405 return;
406 }
407 let healthy = {
408 let mut guard = match bridge.lock() {
409 Ok(guard) => guard,
410 Err(_) => return,
411 };
412 guard.is_alive() && guard.reset().is_ok()
413 };
414 if !healthy {
415 warn!("java subprocess unhealthy at idle; respawning");
416 self.replace_slot(&bridge);
417 }
418 }
419
420 fn replace_slot(&self, dead: &SharedBridge) {
421 let Ok(mut slots) = self.slots.lock() else {
422 return;
423 };
424 let Some(index) = slots.iter().position(|s| Arc::ptr_eq(&s.bridge, dead)) else {
425 return;
426 };
427 match SubprocessBridge::spawn(&self.config) {
428 Ok(replacement) => {
429 slots[index] = PoolSlot {
430 bridge: Arc::new(Mutex::new(replacement)),
431 active: 0,
432 };
433 }
434 Err(error) => {
435 warn!(%error, "failed to respawn java subprocess; retiring pool slot");
436 slots.remove(index);
437 }
438 }
439 }
440}
441
442#[cfg(feature = "java-forge")]
443impl JavaEngineHandle {
444 fn bridge_for(&self, session_id: &str) -> Result<SharedBridge, String> {
445 let in_use = self
446 .pool
447 .in_use
448 .lock()
449 .map_err(|_| "java engine in_use map poisoned".to_string())?;
450 in_use
451 .get(session_id)
452 .cloned()
453 .ok_or_else(|| format!("unknown java session: {session_id}"))
454 }
455
456 pub fn start_game(&self, request_json: &str) -> Result<String, String> {
457 let bridge = self.pool.acquire()?;
458 let response = {
459 let mut guard = bridge
460 .lock()
461 .map_err(|_| "java subprocess mutex poisoned".to_string())?;
462 guard.start_game_json(request_json)
463 };
464 let response = match response {
465 Ok(response) => response,
466 Err(error) => {
467 self.pool.release(bridge);
468 return Err(error);
469 }
470 };
471 let parsed: StartGameResponse = match serde_json::from_str(&response) {
472 Ok(parsed) => parsed,
473 Err(error) => {
474 self.pool.release(bridge);
475 return Err(format!("malformed startGame response: {error}"));
476 }
477 };
478 let session_id = parsed.session_id.clone();
479 let displaced = {
480 let mut in_use = self
481 .pool
482 .in_use
483 .lock()
484 .map_err(|_| "java engine in_use map poisoned".to_string())?;
485 in_use.insert(session_id.clone(), bridge)
486 };
487 if let Some(displaced) = displaced {
488 warn!(
489 session_id,
490 "session_id collision; releasing displaced java subprocess"
491 );
492 self.pool.release(displaced);
493 }
494 Ok(session_id)
495 }
496
497 pub fn submit_action(&self, session_id: &str, action_json: &str) -> Result<String, String> {
498 let bridge = self.bridge_for(session_id)?;
499 let mut guard = bridge
500 .lock()
501 .map_err(|_| "java subprocess mutex poisoned".to_string())?;
502 guard.submit_action(session_id, action_json)
503 }
504
505 pub fn get_prompt(
506 &self,
507 session_id: &str,
508 player_index: usize,
509 ) -> Result<Option<String>, String> {
510 let bridge = self.bridge_for(session_id)?;
511 let mut guard = bridge
512 .lock()
513 .map_err(|_| "java subprocess mutex poisoned".to_string())?;
514 guard.get_prompt(session_id, player_index)
515 }
516
517 pub fn is_game_over(&self, session_id: &str) -> Result<bool, String> {
518 let bridge = self.bridge_for(session_id)?;
519 let mut guard = bridge
520 .lock()
521 .map_err(|_| "java subprocess mutex poisoned".to_string())?;
522 guard.is_game_over(session_id)
523 }
524
525 pub fn get_snapshot(&self, session_id: &str, viewer: Option<usize>) -> Result<String, String> {
526 let bridge = self.bridge_for(session_id)?;
527 let mut guard = bridge
528 .lock()
529 .map_err(|_| "java subprocess mutex poisoned".to_string())?;
530 guard.get_snapshot(session_id, viewer)
531 }
532
533 pub fn end_game(&self, session_id: &str) -> Result<(), String> {
534 let bridge = {
535 let mut in_use = self
536 .pool
537 .in_use
538 .lock()
539 .map_err(|_| "java engine in_use map poisoned".to_string())?;
540 in_use.remove(session_id)
541 };
542 let Some(bridge) = bridge else {
543 return Ok(());
544 };
545 let result = {
546 let mut guard = bridge
547 .lock()
548 .map_err(|_| "java subprocess mutex poisoned".to_string())?;
549 guard.end_game(session_id)
550 };
551 self.pool.release(bridge);
552 result
553 }
554
555 pub fn abort_game(&self, session_id: &str) -> Result<(), String> {
556 let bridge = {
557 let mut in_use = self
558 .pool
559 .in_use
560 .lock()
561 .map_err(|_| "java engine in_use map poisoned".to_string())?;
562 in_use.remove(session_id)
563 };
564 let Some(bridge) = bridge else {
565 return Ok(());
566 };
567 let result = {
568 let mut guard = bridge
569 .lock()
570 .map_err(|_| "java subprocess mutex poisoned".to_string())?;
571 guard.abort_game(session_id)
572 };
573 self.pool.release(bridge);
574 result
575 }
576}
577
578#[cfg(feature = "java-forge")]
579static JAVA_ENGINE: std::sync::OnceLock<Arc<JavaEnginePool>> = std::sync::OnceLock::new();
580
581#[cfg(feature = "java-forge")]
582pub fn init_engine() -> Result<(), String> {
583 if JAVA_ENGINE.get().is_some() {
584 return Ok(());
585 }
586 let config = JavaRuntimeConfig::from_env();
587 let max_sessions = env::var("SELF_HOSTED_NODE_MAX_GAMES")
593 .ok()
594 .and_then(|value| value.parse::<usize>().ok())
595 .filter(|n| *n >= 1)
596 .unwrap_or(1);
597 let sessions_per_process = env::var("SELF_HOSTED_NODE_GAMES_PER_JVM")
598 .ok()
599 .and_then(|value| value.parse::<usize>().ok())
600 .filter(|n| *n >= 1)
601 .unwrap_or(1);
602 let pool = JavaEnginePool::start(&config, max_sessions, sessions_per_process)?;
603 JAVA_ENGINE
604 .set(pool)
605 .map_err(|_| "java engine already initialized".to_string())
606}
607
608#[cfg(all(feature = "graal-forge", not(feature = "java-forge")))]
609pub fn init_engine() -> Result<(), String> {
610 Ok(())
611}
612
613#[cfg(not(forge_backend))]
614pub fn init_engine() -> Result<(), String> {
615 Err(
616 "forge engine requires building self-hosted-node with --features java-forge or graal-forge"
617 .to_string(),
618 )
619}
620
621#[cfg(feature = "java-forge")]
622fn engine_handle() -> Result<JavaEngineHandle, String> {
623 JAVA_ENGINE
624 .get()
625 .map(JavaEnginePool::handle)
626 .ok_or_else(|| "java engine is not initialized".to_string())
627}
628
629#[cfg(feature = "java-forge")]
630type ForgeEngine = JavaEngineHandle;
631
632#[cfg(feature = "java-forge")]
633fn obtain_engine() -> Result<ForgeEngine, String> {
634 engine_handle()
635}
636
637#[cfg(all(feature = "graal-forge", not(feature = "java-forge")))]
638type ForgeEngine = GraalEngineHandle;
639
640#[cfg(all(feature = "graal-forge", not(feature = "java-forge")))]
641fn obtain_engine() -> Result<ForgeEngine, String> {
642 GraalEngineHandle::create(&JavaRuntimeConfig::from_env().assets_dir)
643}
644
645#[cfg(feature = "graal-forge")]
646mod graal_ffi {
647 use std::os::raw::{c_char, c_int};
648
649 #[allow(non_camel_case_types)]
650 pub type graal_isolate_t = std::ffi::c_void;
651 #[allow(non_camel_case_types)]
652 pub type graal_isolatethread_t = std::ffi::c_void;
653
654 extern "C" {
655 pub fn graal_create_isolate(
656 params: *mut std::ffi::c_void,
657 isolate: *mut *mut graal_isolate_t,
658 thread: *mut *mut graal_isolatethread_t,
659 ) -> c_int;
660 pub fn graal_tear_down_isolate(thread: *mut graal_isolatethread_t) -> c_int;
661 pub fn graal_attach_thread(
662 isolate: *mut graal_isolate_t,
663 thread: *mut *mut graal_isolatethread_t,
664 ) -> c_int;
665 pub fn graal_detach_thread(thread: *mut graal_isolatethread_t) -> c_int;
666 pub fn forge_initialize(
667 thread: *mut graal_isolatethread_t,
668 assets_dir: *const c_char,
669 ) -> *mut c_char;
670 pub fn forge_start_game(
671 thread: *mut graal_isolatethread_t,
672 request_json: *const c_char,
673 ) -> *mut c_char;
674 pub fn forge_submit_action(
675 thread: *mut graal_isolatethread_t,
676 session_id: *const c_char,
677 action_json: *const c_char,
678 ) -> *mut c_char;
679 pub fn forge_get_prompt(
680 thread: *mut graal_isolatethread_t,
681 session_id: *const c_char,
682 player_index: c_int,
683 ) -> *mut c_char;
684 pub fn forge_get_snapshot(
685 thread: *mut graal_isolatethread_t,
686 session_id: *const c_char,
687 viewer: c_int,
688 ) -> *mut c_char;
689 pub fn forge_get_game_over(
690 thread: *mut graal_isolatethread_t,
691 session_id: *const c_char,
692 ) -> *mut c_char;
693 pub fn forge_end_game(
694 thread: *mut graal_isolatethread_t,
695 session_id: *const c_char,
696 ) -> *mut c_char;
697 pub fn forge_abort_game(
698 thread: *mut graal_isolatethread_t,
699 session_id: *const c_char,
700 ) -> *mut c_char;
701 pub fn forge_free_string(thread: *mut graal_isolatethread_t, ptr: *mut c_char);
702 }
703}
704
705#[cfg(feature = "graal-forge")]
706#[derive(serde::Deserialize)]
707struct ForgeReply {
708 ok: bool,
709 #[serde(default)]
710 result: String,
711 #[serde(default)]
712 error: Option<String>,
713}
714
715#[cfg(feature = "graal-forge")]
721struct GraalBridge {
722 thread: *mut graal_ffi::graal_isolatethread_t,
723 attached: bool,
724}
725
726#[cfg(feature = "graal-forge")]
727struct SharedIsolate(*mut graal_ffi::graal_isolate_t);
728#[cfg(feature = "graal-forge")]
729unsafe impl Send for SharedIsolate {}
730
731#[cfg(feature = "graal-forge")]
732static SHARED_GRAAL_ISOLATE: std::sync::Mutex<Option<SharedIsolate>> = std::sync::Mutex::new(None);
733
734#[cfg(feature = "graal-forge")]
735impl GraalBridge {
736 fn create() -> Result<Self, String> {
737 let mut isolate: *mut graal_ffi::graal_isolate_t = std::ptr::null_mut();
738 let mut thread: *mut graal_ffi::graal_isolatethread_t = std::ptr::null_mut();
739 let rc = unsafe {
740 graal_ffi::graal_create_isolate(std::ptr::null_mut(), &mut isolate, &mut thread)
741 };
742 if rc != 0 {
743 return Err(format!("graal_create_isolate failed with code {rc}"));
744 }
745 Ok(Self {
746 thread,
747 attached: false,
748 })
749 }
750
751 fn create_in_shared_isolate(assets_dir: &Path) -> Result<Self, String> {
752 let mut guard = SHARED_GRAAL_ISOLATE
753 .lock()
754 .map_err(|_| "shared graal isolate poisoned".to_string())?;
755 if let Some(shared) = guard.as_ref() {
756 let mut thread: *mut graal_ffi::graal_isolatethread_t = std::ptr::null_mut();
757 let rc = unsafe { graal_ffi::graal_attach_thread(shared.0, &mut thread) };
758 if rc != 0 {
759 return Err(format!("graal_attach_thread failed with code {rc}"));
760 }
761 return Ok(Self {
762 thread,
763 attached: true,
764 });
765 }
766 let mut isolate: *mut graal_ffi::graal_isolate_t = std::ptr::null_mut();
767 let mut thread: *mut graal_ffi::graal_isolatethread_t = std::ptr::null_mut();
768 let rc = unsafe {
769 graal_ffi::graal_create_isolate(std::ptr::null_mut(), &mut isolate, &mut thread)
770 };
771 if rc != 0 {
772 return Err(format!("graal_create_isolate failed with code {rc}"));
773 }
774 let mut bridge = Self {
775 thread,
776 attached: false,
777 };
778 let assets = cstring(&assets_dir.to_string_lossy())?;
779 bridge.decode(unsafe { graal_ffi::forge_initialize(bridge.thread, assets.as_ptr()) })?;
780 bridge.attached = true;
781 *guard = Some(SharedIsolate(isolate));
782 info!("shared graal isolate initialized");
783 Ok(bridge)
784 }
785
786 fn decode(&self, raw: *mut std::os::raw::c_char) -> Result<String, String> {
787 if raw.is_null() {
788 return Err("forge native lib returned null".to_string());
789 }
790 let envelope = unsafe { std::ffi::CStr::from_ptr(raw) }
791 .to_string_lossy()
792 .into_owned();
793 unsafe { graal_ffi::forge_free_string(self.thread, raw) };
794 let reply: ForgeReply = serde_json::from_str(&envelope)
795 .map_err(|err| format!("malformed forge envelope: {err}"))?;
796 if reply.ok {
797 Ok(reply.result)
798 } else {
799 Err(reply
800 .error
801 .unwrap_or_else(|| "unknown forge error".to_string()))
802 }
803 }
804}
805
806#[cfg(feature = "graal-forge")]
807impl Drop for GraalBridge {
808 fn drop(&mut self) {
809 if self.attached {
810 unsafe { graal_ffi::graal_detach_thread(self.thread) };
811 } else {
812 unsafe { graal_ffi::graal_tear_down_isolate(self.thread) };
813 }
814 }
815}
816
817#[cfg(feature = "graal-forge")]
818#[derive(Clone)]
819struct GraalEngineHandle {
820 bridge: std::rc::Rc<GraalBridge>,
821}
822
823#[cfg(feature = "graal-forge")]
824impl GraalEngineHandle {
825 fn create(assets_dir: &Path) -> Result<Self, String> {
826 let shared = env::var("SELF_HOSTED_NODE_SHARED_ISOLATE")
830 .map(|value| value == "1" || value.eq_ignore_ascii_case("true"))
831 .unwrap_or(false);
832 let bridge = if shared {
833 GraalBridge::create_in_shared_isolate(assets_dir)?
834 } else {
835 let bridge = GraalBridge::create()?;
836 let assets = cstring(&assets_dir.to_string_lossy())?;
837 bridge
838 .decode(unsafe { graal_ffi::forge_initialize(bridge.thread, assets.as_ptr()) })?;
839 bridge
840 };
841 Ok(Self {
842 bridge: std::rc::Rc::new(bridge),
843 })
844 }
845
846 fn start_game(&self, request_json: &str) -> Result<String, String> {
847 let request = cstring(request_json)?;
848 let response = self
849 .bridge
850 .decode(unsafe { graal_ffi::forge_start_game(self.bridge.thread, request.as_ptr()) })?;
851 let parsed: StartGameResponse = serde_json::from_str(&response)
852 .map_err(|err| format!("malformed startGame response: {err}"))?;
853 Ok(parsed.session_id)
854 }
855
856 fn submit_action(&self, session_id: &str, action_json: &str) -> Result<String, String> {
857 let session = cstring(session_id)?;
858 let action = cstring(action_json)?;
859 self.bridge.decode(unsafe {
860 graal_ffi::forge_submit_action(self.bridge.thread, session.as_ptr(), action.as_ptr())
861 })
862 }
863
864 fn get_prompt(&self, session_id: &str, player_index: usize) -> Result<Option<String>, String> {
865 let session = cstring(session_id)?;
866 let prompt = self.bridge.decode(unsafe {
867 graal_ffi::forge_get_prompt(
868 self.bridge.thread,
869 session.as_ptr(),
870 player_index as std::os::raw::c_int,
871 )
872 })?;
873 Ok((!prompt.is_empty()).then_some(prompt))
874 }
875
876 fn is_game_over(&self, session_id: &str) -> Result<bool, String> {
877 let session = cstring(session_id)?;
878 let value = self.bridge.decode(unsafe {
879 graal_ffi::forge_get_game_over(self.bridge.thread, session.as_ptr())
880 })?;
881 Ok(value.trim() == "true")
882 }
883
884 fn get_snapshot(&self, session_id: &str, viewer: Option<usize>) -> Result<String, String> {
885 let session = cstring(session_id)?;
886 let viewer = viewer.map_or(-1, |v| v as std::os::raw::c_int);
887 self.bridge.decode(unsafe {
888 graal_ffi::forge_get_snapshot(self.bridge.thread, session.as_ptr(), viewer)
889 })
890 }
891
892 fn end_game(&self, session_id: &str) -> Result<(), String> {
893 let session = cstring(session_id)?;
894 self.bridge
895 .decode(unsafe { graal_ffi::forge_end_game(self.bridge.thread, session.as_ptr()) })
896 .map(|_| ())
897 }
898
899 fn abort_game(&self, session_id: &str) -> Result<(), String> {
900 let session = cstring(session_id)?;
901 self.bridge
902 .decode(unsafe { graal_ffi::forge_abort_game(self.bridge.thread, session.as_ptr()) })
903 .map(|_| ())
904 }
905}
906
907#[cfg(feature = "graal-forge")]
908fn cstring(value: &str) -> Result<std::ffi::CString, String> {
909 std::ffi::CString::new(value).map_err(|_| "string contained interior NUL".to_string())
910}
911
912#[cfg(feature = "java-forge")]
913pub fn run_concurrent_self_play(
914 seats: &[DeckSelection],
915 starting_life: i32,
916 seed: u64,
917 max_prompts: usize,
918 concurrency: usize,
919) -> Result<(), String> {
920 let config = JavaRuntimeConfig::from_env();
921 let games_per_process = env::var("SELF_HOSTED_NODE_GAMES_PER_JVM")
922 .ok()
923 .and_then(|value| value.parse::<usize>().ok())
924 .filter(|n| *n >= 1)
925 .unwrap_or(1);
926 let pool = JavaEnginePool::start(&config, concurrency.max(1), games_per_process)?;
927 info!(
928 concurrency,
929 "java-engine started; launching concurrent games"
930 );
931
932 let mut players = Vec::with_capacity(seats.len());
933 for (i, seat) in seats.iter().enumerate() {
934 let identities = deck_card_identities(&seat.deck);
935 players.push(PlayerConfig::new(
936 format!("Self-Play {}", i + 1),
937 &identities,
938 commander_names_for_java(&seat.deck, seat.commander_name.as_deref()),
939 ));
940 }
941
942 let mut joins = Vec::with_capacity(concurrency.max(1));
943 for game_index in 0..concurrency.max(1) {
944 let handle = pool.handle();
945 let request = StartGameRequest::new(
946 format!("self-hosted-java-concurrent-{game_index}"),
947 String::new(),
948 starting_life,
949 seed.wrapping_add(game_index as u64),
950 players.clone(),
951 );
952 joins.push(std::thread::spawn(move || -> Result<(), String> {
953 let request_json = request.to_json().map_err(|error| error.to_string())?;
954 let session_id = handle.start_game(&request_json)?;
955 info!(session_id, game_index, "concurrent java game started");
956 let result = drive_game_via_handle(&handle, &session_id, max_prompts);
957 let _ = handle.end_game(&session_id);
958 result
959 }));
960 }
961
962 let mut outcome = Ok(());
963 for join in joins {
964 match join.join() {
965 Ok(Ok(())) => {}
966 Ok(Err(error)) => outcome = Err(error),
967 Err(_) => outcome = Err("concurrent game thread panicked".to_string()),
968 }
969 }
970 outcome
971}
972
973#[cfg(not(feature = "java-forge"))]
974pub fn run_concurrent_self_play(
975 _seats: &[DeckSelection],
976 _starting_life: i32,
977 _seed: u64,
978 _max_prompts: usize,
979 _concurrency: usize,
980) -> Result<(), String> {
981 Err(
982 "java-forge concurrent self-play requires building self-hosted-node with --features java-forge"
983 .to_string(),
984 )
985}
986
987#[cfg(feature = "java-forge")]
988fn drive_game_via_handle(
989 handle: &JavaEngineHandle,
990 session_id: &str,
991 max_prompts: usize,
992) -> Result<(), String> {
993 let mut bots: HashMap<usize, SimpleAi> = HashMap::new();
994 let mut last_prompt: Option<String> = None;
995 let mut acted = 0usize;
996 let mut seen_prompt = false;
997 let max_iterations = max_prompts.saturating_mul(200).max(2_000);
998
999 for _ in 0..max_iterations {
1000 if let Some(prompt_json) = handle.get_prompt(session_id, 0)? {
1001 seen_prompt = true;
1002 if last_prompt.as_deref() == Some(prompt_json.as_str()) {
1003 if handle.is_game_over(session_id)? {
1004 return Ok(());
1005 }
1006 std::thread::sleep(Duration::from_millis(20));
1007 continue;
1008 }
1009 let prompt: AgentPrompt = serde_json::from_str(&prompt_json)
1010 .map_err(|error| format!("failed to parse concurrent prompt: {error}"))?;
1011 let player = player_index(&prompt.deciding_player_id);
1012 if let Some(action) = bots.entry(player).or_default().decide(prompt) {
1013 let action_json = serde_json::to_string(&action).map_err(|err| err.to_string())?;
1014 handle.submit_action(session_id, &action_json)?;
1015 acted += 1;
1016 if acted >= max_prompts {
1017 return Err(format!(
1018 "concurrent game {session_id} did not finish within {max_prompts} decisions"
1019 ));
1020 }
1021 }
1022 last_prompt = Some(prompt_json);
1023 continue;
1024 }
1025 if seen_prompt && handle.is_game_over(session_id)? {
1026 return Ok(());
1027 }
1028 std::thread::sleep(Duration::from_millis(20));
1029 }
1030 Err(format!(
1031 "concurrent game {session_id} exceeded its iteration cap"
1032 ))
1033}
1034
1035#[derive(Debug, Clone)]
1036pub struct JavaRuntimeConfig {
1037 pub assets_dir: PathBuf,
1038 pub harness_jar: PathBuf,
1039 pub java_home: Option<PathBuf>,
1040 pub extra_classpath: Vec<PathBuf>,
1041}
1042
1043impl JavaRuntimeConfig {
1044 pub fn from_env() -> Self {
1045 let root = workspace_root();
1046 Self {
1047 assets_dir: env_path("SELF_HOSTED_NODE_FORGE_ASSETS_DIR")
1048 .or_else(|| env_path("MANA_BREW_FORGE_ASSETS_DIR"))
1049 .unwrap_or_else(|| root.join("forge/forge-gui")),
1050 harness_jar: env_path("SELF_HOSTED_NODE_FORGE_HARNESS_JAR")
1051 .or_else(|| env_path("MANA_BREW_FORGE_HARNESS_JAR"))
1052 .unwrap_or_else(|| {
1053 root.join("forge-harness/target/forge-harness-jar-with-dependencies.jar")
1054 }),
1055 java_home: env_path("SELF_HOSTED_NODE_JAVA_HOME")
1056 .or_else(|| env_path("MANA_BREW_JAVA_HOME"))
1057 .or_else(|| env_path("JAVA_HOME")),
1058 extra_classpath: env_classpath("SELF_HOSTED_NODE_FORGE_EXTRA_CLASSPATH")
1059 .into_iter()
1060 .chain(env_classpath("MANA_BREW_FORGE_EXTRA_CLASSPATH"))
1061 .collect(),
1062 }
1063 }
1064
1065 pub fn validate(&self) -> Result<(), String> {
1066 require_dir(&self.assets_dir, "Forge assets directory")?;
1067 require_file(&self.harness_jar, "Forge harness jar")?;
1068 if let Some(java_home) = &self.java_home {
1069 require_dir(java_home, "Java home")?;
1070 }
1071 for entry in &self.extra_classpath {
1072 if !entry.exists() {
1073 return Err(format!(
1074 "Classpath entry does not exist: {}",
1075 entry.display()
1076 ));
1077 }
1078 }
1079 Ok(())
1080 }
1081
1082 pub fn classpath_entries(&self) -> Vec<PathBuf> {
1083 let mut entries = Vec::with_capacity(1 + self.extra_classpath.len());
1084 entries.push(self.harness_jar.clone());
1085 entries.extend(self.extra_classpath.iter().cloned());
1086 entries
1087 }
1088}
1089
1090#[cfg(forge_backend)]
1091#[allow(clippy::too_many_arguments)]
1092pub fn run_hosted_engine_game(
1093 game_id: String,
1094 player_names: Vec<String>,
1095 decks: Vec<Deck>,
1096 commander_names: Vec<Option<String>>,
1097 commander_variant: bool,
1098 game_variant: String,
1099 local_player_index: Option<usize>,
1100 ai_player_indices: Vec<usize>,
1101 starting_life: i32,
1102 remote_prompt_tx: std_mpsc::Sender<(usize, AgentMessage)>,
1103 remote_response_rxs: Vec<(usize, std_mpsc::Receiver<ClientToServerMessage>)>,
1104 game_over_tx: std_mpsc::Sender<HostedGameOver>,
1105 cancel: Arc<AtomicBool>,
1106) -> Result<(), String> {
1107 run_hosted_engine_game_inner(
1108 game_id,
1109 player_names,
1110 decks,
1111 commander_names,
1112 commander_variant,
1113 game_variant,
1114 local_player_index,
1115 ai_player_indices,
1116 starting_life,
1117 remote_prompt_tx,
1118 remote_response_rxs,
1119 game_over_tx,
1120 cancel,
1121 )
1122}
1123
1124#[cfg(not(forge_backend))]
1125#[allow(clippy::too_many_arguments)]
1126pub fn run_hosted_engine_game(
1127 _game_id: String,
1128 _player_names: Vec<String>,
1129 _decks: Vec<Deck>,
1130 _commander_names: Vec<Option<String>>,
1131 _commander_variant: bool,
1132 _game_variant: String,
1133 _local_player_index: Option<usize>,
1134 _ai_player_indices: Vec<usize>,
1135 _starting_life: i32,
1136 _remote_prompt_tx: std_mpsc::Sender<(usize, AgentMessage)>,
1137 _remote_response_rxs: Vec<(usize, std_mpsc::Receiver<ClientToServerMessage>)>,
1138 _game_over_tx: std_mpsc::Sender<HostedGameOver>,
1139 _cancel: Arc<AtomicBool>,
1140) -> Result<(), String> {
1141 Err(unsupported_message().to_string())
1142}
1143
1144#[cfg(forge_backend)]
1145#[allow(clippy::too_many_arguments)]
1146fn run_hosted_engine_game_inner(
1147 game_id: String,
1148 player_names: Vec<String>,
1149 decks: Vec<Deck>,
1150 commander_names: Vec<Option<String>>,
1151 commander_variant: bool,
1152 game_variant: String,
1153 local_player_index: Option<usize>,
1154 ai_player_indices: Vec<usize>,
1155 starting_life: i32,
1156 remote_prompt_tx: std_mpsc::Sender<(usize, AgentMessage)>,
1157 remote_response_rxs: Vec<(usize, std_mpsc::Receiver<ClientToServerMessage>)>,
1158 game_over_tx: std_mpsc::Sender<HostedGameOver>,
1159 cancel: Arc<AtomicBool>,
1160) -> Result<(), String> {
1161 let engine = obtain_engine()?;
1162
1163 let mut players = Vec::with_capacity(player_names.len());
1164 for (index, name) in player_names.iter().enumerate() {
1165 let identities = deck_card_identities(&decks[index]);
1166 let seat_commander_names = if commander_variant {
1167 commander_names_for_java(&decks[index], commander_names[index].as_deref())
1168 } else {
1169 Vec::new()
1170 };
1171 players.push(PlayerConfig::new(
1172 name.clone(),
1173 &identities,
1174 seat_commander_names,
1175 ));
1176 }
1177 for &idx in &ai_player_indices {
1178 if let Some(player) = players.get_mut(idx) {
1179 player.ai = true;
1180 }
1181 }
1182 let request = StartGameRequest::new(
1183 game_id.clone(),
1184 game_variant,
1185 starting_life,
1186 rand::random(),
1187 players,
1188 );
1189 let session_id = engine.start_game(&request.to_json().map_err(|err| err.to_string())?)?;
1190 info!(game_id, session_id, "hosted java-forge session started");
1191
1192 struct SessionGuard {
1193 engine: ForgeEngine,
1194 session_id: String,
1195 armed: std::cell::Cell<bool>,
1196 }
1197 impl Drop for SessionGuard {
1198 fn drop(&mut self) {
1199 if self.armed.get() {
1200 if let Err(error) = self.engine.abort_game(&self.session_id) {
1201 warn!(session_id = %self.session_id, %error, "failed to abort java session; context may leak");
1202 }
1203 }
1204 }
1205 }
1206 let guard = SessionGuard {
1207 engine: engine.clone(),
1208 session_id: session_id.clone(),
1209 armed: std::cell::Cell::new(true),
1210 };
1211
1212 let mut remote_response_rxs: HashMap<usize, std_mpsc::Receiver<ClientToServerMessage>> =
1213 remote_response_rxs.into_iter().collect();
1214 let mut last_prompt: Option<AgentPrompt> = None;
1215 let mut pending_roll_acks: usize = 0;
1216
1217 loop {
1218 if cancel.load(std::sync::atomic::Ordering::Relaxed) {
1219 info!(
1220 session_id,
1221 "hosted java-forge session cancelled; player left the game"
1222 );
1223 return Ok(());
1224 }
1225 for (player_index, rx) in &mut remote_response_rxs {
1226 loop {
1227 match rx.try_recv() {
1228 Ok(ClientToServerMessage::Response {
1229 action: PromptOutput::DiceRolled(DiceRolledOutput::DiceRolledAcknowledged),
1230 ..
1231 }) => {
1232 if pending_roll_acks > 0 {
1233 pending_roll_acks -= 1;
1234 if pending_roll_acks == 0 {
1235 let ack = serde_json::to_string(&PromptOutput::DiceRolled(
1236 DiceRolledOutput::DiceRolledAcknowledged,
1237 ))
1238 .map_err(|err| format!("failed to serialize roll ack: {err}"))?;
1239 engine.submit_action(&session_id, &ack)?;
1240 }
1241 }
1242 }
1243 Ok(ClientToServerMessage::Response { prompt_id, action }) => {
1244 if prompt_id != 0 {
1247 let Some(prompt) =
1248 last_prompt.as_ref().filter(|p| p.prompt_id == prompt_id)
1249 else {
1250 reject_response(
1251 &remote_prompt_tx,
1252 *player_index,
1253 last_prompt.as_ref().filter(|p| {
1254 self::player_index(&p.deciding_player_id) == *player_index
1255 }),
1256 ProtocolErrorCode::StalePrompt,
1257 format!("response for prompt {prompt_id} is not open"),
1258 );
1259 continue;
1260 };
1261 if self::player_index(&prompt.deciding_player_id) != *player_index {
1262 reject_response(
1263 &remote_prompt_tx,
1264 *player_index,
1265 None,
1266 ProtocolErrorCode::WrongPlayer,
1267 format!(
1268 "prompt {prompt_id} is for {}",
1269 prompt.deciding_player_id
1270 ),
1271 );
1272 continue;
1273 }
1274 match prompt.input.validate_response(&action) {
1275 Ok(()) => {}
1276 Err(ResponseViolation::WrongPromptType) => {
1277 reject_response(
1278 &remote_prompt_tx,
1279 *player_index,
1280 Some(prompt),
1281 ProtocolErrorCode::WrongPromptType,
1282 "response output does not match the prompt type"
1283 .to_string(),
1284 );
1285 continue;
1286 }
1287 Err(ResponseViolation::UnknownActionId(id)) => {
1288 reject_response(
1289 &remote_prompt_tx,
1290 *player_index,
1291 Some(prompt),
1292 ProtocolErrorCode::UnknownActionId,
1293 format!(
1294 "action id {id:?} was not advertised by the prompt"
1295 ),
1296 );
1297 continue;
1298 }
1299 Err(ResponseViolation::CancelNotAllowed) => {
1300 reject_response(
1301 &remote_prompt_tx,
1302 *player_index,
1303 Some(prompt),
1304 ProtocolErrorCode::CancelNotAllowed,
1305 "this prompt is not cancellable".to_string(),
1306 );
1307 continue;
1308 }
1309 }
1310 }
1311 let action_json = serde_json::to_string(&action).map_err(|err| {
1312 format!(
1313 "failed to serialize prompt output for player {player_index}: {err}"
1314 )
1315 })?;
1316 debug!(player_index, %action_json, "submitting remote response to java");
1317 engine.submit_action(&session_id, &action_json)?;
1318 }
1319 Ok(ClientToServerMessage::Directive {
1320 directive: DirectiveInput::Concede,
1321 }) => {
1322 let directive_json = directive_concede_json(*player_index);
1326 debug!(player_index, %directive_json, "submitting concede directive to java");
1327 engine.submit_action(&session_id, &directive_json)?;
1328 }
1329 Err(TryRecvError::Empty) => break,
1330 Err(TryRecvError::Disconnected) => {
1331 debug!(player_index, "java-forge response channel disconnected");
1332 break;
1333 }
1334 }
1335 }
1336 }
1337
1338 if let Some(prompt_json) = engine.get_prompt(&session_id, 0)? {
1339 let prompt: AgentPrompt = serde_json::from_str(&prompt_json)
1340 .map_err(|err| format!("failed to parse java prompt: {err}"))?;
1341 if last_prompt.as_ref().map(|p| p.prompt_id) != Some(prompt.prompt_id) {
1342 last_prompt = Some(prompt.clone());
1343 let player = player_index(&prompt.deciding_player_id);
1344 debug!(player, "forwarding java prompt to remote");
1345 if matches!(prompt.input, PromptInput::DiceRolled(_)) {
1346 let prompt_msg = AgentMessage::Prompt(prompt);
1347 for &agent_index in remote_response_rxs.keys() {
1348 let state = AgentMessage::State(state_via_handle(
1349 &engine,
1350 &session_id,
1351 Some(agent_index),
1352 )?);
1353 let _ = remote_prompt_tx.send((agent_index, state));
1354 let _ = remote_prompt_tx.send((agent_index, prompt_msg.clone()));
1355 }
1356 send_observer_state(&engine, &session_id, &remote_prompt_tx);
1357 pending_roll_acks = remote_response_rxs.len();
1358 if pending_roll_acks == 0 {
1359 let ack = serde_json::to_string(&PromptOutput::DiceRolled(
1360 DiceRolledOutput::DiceRolledAcknowledged,
1361 ))
1362 .map_err(|err| format!("failed to serialize roll ack: {err}"))?;
1363 engine.submit_action(&session_id, &ack)?;
1364 }
1365 } else if Some(player) == local_player_index {
1366 if let Some(output) = auto_action(&prompt) {
1367 let action_json = serde_json::to_string(&output)
1368 .map_err(|err| format!("failed to serialize auto action: {err}"))?;
1369 engine.submit_action(&session_id, &action_json)?;
1370 }
1371 } else {
1372 for &agent_index in remote_response_rxs.keys() {
1373 let state = AgentMessage::State(state_via_handle(
1374 &engine,
1375 &session_id,
1376 Some(agent_index),
1377 )?);
1378 if remote_prompt_tx.send((agent_index, state)).is_err() {
1379 return Ok(());
1380 }
1381 }
1382 send_observer_state(&engine, &session_id, &remote_prompt_tx);
1383 let prompt_msg = AgentMessage::Prompt(prompt);
1384 if remote_prompt_tx.send((player, prompt_msg)).is_err() {
1385 return Ok(());
1386 }
1387 }
1388 }
1389 }
1390
1391 if engine.is_game_over(&session_id)? {
1392 info!("hosted java-forge session reached game over");
1393 let mut final_messages = Vec::new();
1394 for &agent_index in remote_response_rxs.keys() {
1395 match state_via_handle(&engine, &session_id, Some(agent_index)) {
1396 Ok(state_update) => {
1397 final_messages.push((agent_index, AgentMessage::State(state_update)));
1398 }
1399 Err(error) => {
1400 warn!(%error, agent_index, "game over: final snapshot unavailable; sending game-over prompt only");
1401 }
1402 }
1403 }
1404 if let Ok(state_update) = state_via_handle(&engine, &session_id, None) {
1405 final_messages.push((
1406 crate::host::OBSERVER_SEAT,
1407 AgentMessage::State(state_update),
1408 ));
1409 }
1410 let game_over = AgentMessage::Prompt(game_over_prompt());
1411 for &agent_index in remote_response_rxs.keys() {
1412 final_messages.push((agent_index, game_over.clone()));
1413 }
1414 let _ = game_over_tx.send(HostedGameOver {
1415 game_id: game_id.clone(),
1416 messages: final_messages,
1417 });
1418 engine.end_game(&session_id)?;
1419 guard.armed.set(false);
1420 return Ok(());
1421 }
1422
1423 std::thread::sleep(Duration::from_millis(50));
1424 }
1425}
1426
1427#[cfg(feature = "java-forge")]
1428fn wait_for_prompt<B: JavaBridge>(
1429 session: &mut JavaForgeSession<B>,
1430 max_polls: usize,
1431) -> Result<Option<String>, String> {
1432 for _ in 0..max_polls {
1433 if let Some(prompt) = session.get_prompt(0)? {
1434 return Ok(Some(prompt));
1435 }
1436 std::thread::sleep(Duration::from_millis(50));
1437 }
1438 Ok(None)
1439}
1440
1441#[cfg(forge_backend)]
1442fn send_observer_state(
1443 engine: &ForgeEngine,
1444 session_id: &str,
1445 remote_prompt_tx: &std_mpsc::Sender<(usize, AgentMessage)>,
1446) {
1447 match state_via_handle(engine, session_id, None) {
1448 Ok(state_update) => {
1449 let _ = remote_prompt_tx.send((
1450 crate::host::OBSERVER_SEAT,
1451 AgentMessage::State(state_update),
1452 ));
1453 }
1454 Err(error) => warn!(%error, "observer snapshot unavailable"),
1455 }
1456}
1457
1458#[cfg(forge_backend)]
1459fn player_index(deciding_player_id: &str) -> usize {
1460 deciding_player_id
1461 .strip_prefix("player-")
1462 .and_then(|n| n.parse().ok())
1463 .unwrap_or(0)
1464}
1465
1466#[cfg(forge_backend)]
1467fn reject_response(
1468 remote_prompt_tx: &std_mpsc::Sender<(usize, AgentMessage)>,
1469 seat: usize,
1470 reopen_prompt: Option<&AgentPrompt>,
1471 code: ProtocolErrorCode,
1472 message: String,
1473) {
1474 let _ = remote_prompt_tx.send((
1475 seat,
1476 AgentMessage::Error(ProtocolError {
1477 code,
1478 message,
1479 prompt_id: reopen_prompt.map(|p| p.prompt_id),
1480 }),
1481 ));
1482 if let Some(prompt) = reopen_prompt {
1483 let _ = remote_prompt_tx.send((seat, AgentMessage::Prompt(prompt.clone())));
1484 }
1485}
1486
1487#[cfg(forge_backend)]
1488fn auto_action(prompt: &AgentPrompt) -> Option<PromptOutput> {
1489 match prompt.input {
1490 PromptInput::ChooseAction(_) => {
1491 Some(PromptOutput::ChooseAction(ChooseActionOutput::Pass {
1492 until: None,
1493 exhaust_stack: false,
1494 }))
1495 }
1496 _ => None,
1497 }
1498}
1499
1500#[cfg(forge_backend)]
1501fn game_over_prompt() -> AgentPrompt {
1502 AgentPrompt {
1503 prompt_id: u32::MAX,
1504 deciding_player_id: "player-0".to_string(),
1505 source_card: None,
1506 input: PromptInput::GameOver(GameOverInput {}),
1507 }
1508}
1509
1510#[cfg(forge_backend)]
1511fn state_via_handle(
1512 engine: &ForgeEngine,
1513 session_id: &str,
1514 viewer: Option<usize>,
1515) -> Result<StateUpdate, String> {
1516 let game_view: GameViewDto = serde_json::from_str(&engine.get_snapshot(session_id, viewer)?)
1517 .map_err(|err| format!("failed to parse java snapshot: {err}"))?;
1518 Ok(StateUpdate { game_view })
1519}
1520
1521#[cfg(forge_backend)]
1522fn deck_card_identities(deck: &Deck) -> Vec<DeckCardIdentity> {
1523 deck.cards
1524 .iter()
1525 .chain(deck.commanders.iter().flatten())
1526 .map(|card| card.identity.clone())
1527 .collect()
1528}
1529
1530#[cfg(forge_backend)]
1531fn commander_names_for_java(deck: &Deck, fallback: Option<&str>) -> Vec<String> {
1532 let names: Vec<String> = deck
1533 .commanders
1534 .iter()
1535 .flatten()
1536 .map(|card| java_card_name(&card.identity.name))
1537 .collect();
1538 if !names.is_empty() {
1539 return names;
1540 }
1541 fallback
1542 .filter(|name| !name.is_empty())
1543 .map(|name| vec![java_card_name(name)])
1544 .unwrap_or_default()
1545}
1546
1547#[cfg(forge_backend)]
1548fn smoke_deck(land_name: &str, spell_name: &str) -> Vec<DeckCardIdentity> {
1549 (0..24)
1550 .map(|_| DeckCardIdentity {
1551 name: land_name.to_string(),
1552 ..Default::default()
1553 })
1554 .chain((0..36).map(|_| DeckCardIdentity {
1555 name: spell_name.to_string(),
1556 ..Default::default()
1557 }))
1558 .collect()
1559}
1560
1561#[cfg(feature = "java-forge")]
1562fn scenario_deck(land_name: &str) -> Vec<DeckCardIdentity> {
1563 (0..60)
1564 .map(|_| DeckCardIdentity {
1565 name: land_name.to_string(),
1566 ..Default::default()
1567 })
1568 .collect()
1569}
1570
1571#[cfg(feature = "java-forge")]
1572enum JavaScenario {
1573 KeepAndPlayLand {
1574 played_land: bool,
1575 },
1576 MulliganOncePlayLand {
1577 mulliganed: bool,
1578 kept_second_hand: bool,
1579 put_back_done: bool,
1580 played_land: bool,
1581 },
1582}
1583
1584#[cfg(feature = "java-forge")]
1585impl JavaScenario {
1586 fn from_name(name: &str) -> Result<Self, String> {
1587 match name {
1588 "keep-and-play-land" => Ok(Self::KeepAndPlayLand { played_land: false }),
1589 "mulligan-once-play-land" => Ok(Self::MulliganOncePlayLand {
1590 mulliganed: false,
1591 kept_second_hand: false,
1592 put_back_done: false,
1593 played_land: false,
1594 }),
1595 _ => Err(format!(
1596 "unknown java-forge scenario '{name}'. Supported scenarios: keep-and-play-land, mulligan-once-play-land"
1597 )),
1598 }
1599 }
1600
1601 fn name(&self) -> &'static str {
1602 match self {
1603 Self::KeepAndPlayLand { .. } => "keep-and-play-land",
1604 Self::MulliganOncePlayLand { .. } => "mulligan-once-play-land",
1605 }
1606 }
1607
1608 fn next_action(
1609 &mut self,
1610 prompt: &Value,
1611 game_view: &GameViewDto,
1612 ) -> Result<Option<PromptOutput>, String> {
1613 match self {
1614 Self::KeepAndPlayLand { played_land } => {
1615 if *played_land && battlefield_contains(game_view, "Swamp") {
1616 return Ok(None);
1617 }
1618 match prompt_type(prompt) {
1619 Some("mulligan") => Ok(Some(PromptOutput::Mulligan(
1620 MulliganOutput::MulliganDecision { keep: true },
1621 ))),
1622 Some("chooseAction") => {
1623 if let Some(action) = play_first_card_action(prompt, "Swamp")? {
1624 *played_land = true;
1625 Ok(Some(action))
1626 } else {
1627 Ok(Some(PromptOutput::ChooseAction(ChooseActionOutput::Pass {
1628 until: None,
1629 exhaust_stack: false,
1630 })))
1631 }
1632 }
1633 other => Err(format!(
1634 "scenario '{}' expected mulligan or chooseAction, got {:?}",
1635 self.name(),
1636 other
1637 )),
1638 }
1639 }
1640 Self::MulliganOncePlayLand {
1641 mulliganed,
1642 kept_second_hand,
1643 put_back_done,
1644 played_land,
1645 } => {
1646 if *played_land && battlefield_contains(game_view, "Swamp") {
1647 return Ok(None);
1648 }
1649 match prompt_type(prompt) {
1650 Some("mulligan") if !*mulliganed => {
1651 *mulliganed = true;
1652 Ok(Some(PromptOutput::Mulligan(MulliganOutput::MulliganDecision { keep: false })))
1653 }
1654 Some("mulligan") if !*kept_second_hand => {
1655 *kept_second_hand = true;
1656 Ok(Some(PromptOutput::Mulligan(MulliganOutput::MulliganDecision { keep: true })))
1657 }
1658 Some("mulliganPutBack") if !*put_back_done => {
1659 let count = prompt
1660 .get("input")
1661 .and_then(|input| input.get("count"))
1662 .and_then(Value::as_u64)
1663 .unwrap_or(1) as usize;
1664 let card_ids = prompt_card_ids(prompt, "handCardIds", count)?;
1665 *put_back_done = true;
1666 Ok(Some(PromptOutput::MulliganPutBack(MulliganPutBackOutput::MulliganPutBackDecision { card_ids })))
1667 }
1668 Some("chooseAction") => {
1669 if let Some(action) = play_first_card_action(prompt, "Swamp")? {
1670 *played_land = true;
1671 Ok(Some(action))
1672 } else {
1673 Ok(Some(PromptOutput::ChooseAction(ChooseActionOutput::Pass { until: None, exhaust_stack: false })))
1674 }
1675 }
1676 other => Err(format!(
1677 "scenario '{}' expected mulligan, mulliganPutBack, or chooseAction, got {:?}",
1678 self.name(),
1679 other
1680 )),
1681 }
1682 }
1683 }
1684 }
1685}
1686
1687#[cfg(feature = "java-forge")]
1688fn run_scenario_loop<B: JavaBridge>(
1689 session: &mut JavaForgeSession<B>,
1690 mut scenario: JavaScenario,
1691 max_prompts: usize,
1692) -> Result<(), String> {
1693 let mut prompts_seen = 0usize;
1694 let mut last_prompt_json: Option<String> = None;
1695 while prompts_seen < max_prompts {
1696 let Some(prompt_json) = wait_for_prompt(session, 600)? else {
1697 return Err(format!(
1698 "timed out waiting for java-forge scenario '{}' prompt",
1699 scenario.name()
1700 ));
1701 };
1702 if last_prompt_json.as_deref() == Some(prompt_json.as_str()) {
1703 std::thread::sleep(Duration::from_millis(50));
1704 continue;
1705 }
1706 last_prompt_json = Some(prompt_json.clone());
1707 prompts_seen += 1;
1708
1709 let prompt: AgentPrompt = serde_json::from_str(&prompt_json)
1710 .map_err(|err| format!("failed to parse java scenario prompt: {err}"))?;
1711 let player = player_index(&prompt.deciding_player_id);
1712 if player != 0 {
1713 if let Some(output) = auto_action(&prompt) {
1714 session.submit_action(
1715 &serde_json::to_string(&output).map_err(|err| err.to_string())?,
1716 )?;
1717 }
1718 continue;
1719 }
1720
1721 let game_view: GameViewDto = serde_json::from_str(&session.get_snapshot(Some(0))?)
1722 .map_err(|err| format!("failed to parse java scenario snapshot: {err}"))?;
1723 let normalized_prompt = serde_json::to_value(&prompt).map_err(|err| err.to_string())?;
1724 info!(
1725 scenario = scenario.name(),
1726 prompts_seen,
1727 prompt_type = prompt_type(&normalized_prompt).unwrap_or("<missing>"),
1728 "java-forge scenario prompt"
1729 );
1730 let Some(action) = scenario.next_action(&normalized_prompt, &game_view)? else {
1731 info!(
1732 scenario = scenario.name(),
1733 prompts_seen, "java-forge scenario assertions satisfied"
1734 );
1735 return Ok(());
1736 };
1737 submit_player_action(session, &action)?;
1738 }
1739 Err(format!(
1740 "java-forge scenario '{}' did not complete within {max_prompts} prompts",
1741 scenario.name()
1742 ))
1743}
1744
1745#[cfg(feature = "java-forge")]
1746pub fn run_concede_smoke() -> Result<(), String> {
1747 let config = JavaRuntimeConfig::from_env();
1748 let assets_dir = config.assets_dir.to_string_lossy().to_string();
1749 let bridge = SubprocessBridge::spawn(&config)?;
1750 let mut session = JavaForgeSession::new(bridge);
1751 session.initialize(&assets_dir)?;
1752 run_concede_game(&mut session, 3, 2, 12)?;
1753 run_concede_game(&mut session, 2, 1, 8)?;
1754 Ok(())
1755}
1756
1757#[cfg(not(feature = "java-forge"))]
1758pub fn run_concede_smoke() -> Result<(), String> {
1759 Err(unsupported_message().to_string())
1760}
1761
1762#[cfg(forge_backend)]
1763fn directive_concede_json(player: usize) -> String {
1764 format!(r#"{{"type":"directive","directive":{{"type":"concede"}},"player":{player}}}"#)
1765}
1766
1767#[cfg(feature = "java-forge")]
1768fn run_concede_game<B: JavaBridge>(
1769 session: &mut JavaForgeSession<B>,
1770 seats: usize,
1771 conceder: usize,
1772 concede_after: usize,
1773) -> Result<(), String> {
1774 const POST_CONCEDE_DECISIONS: usize = 12;
1775 const STALL_REPEATS: usize = 300;
1776
1777 let cards: Vec<manabrew_protocol::deck_dto::DeckCard> = (0..60)
1778 .map(|_| manabrew_protocol::deck_dto::DeckCard {
1779 identity: DeckCardIdentity {
1780 name: "Mountain".to_string(),
1781 set_code: "M20".to_string(),
1782 ..Default::default()
1783 },
1784 ..Default::default()
1785 })
1786 .collect();
1787 let deck = Deck {
1788 name: "concede-smoke".to_string(),
1789 cards,
1790 ..Default::default()
1791 };
1792 let identities = deck_card_identities(&deck);
1793 let players: Vec<PlayerConfig> = (0..seats)
1794 .map(|i| PlayerConfig::new(format!("Concede {}", i + 1), &identities, Vec::new()))
1795 .collect();
1796 let request = StartGameRequest::new(
1797 format!("concede-smoke-{seats}p"),
1798 String::new(),
1799 20,
1800 7,
1801 players,
1802 );
1803 let session_id = session.start_game(&request)?;
1804 info!(
1805 session_id,
1806 seats, conceder, concede_after, "concede smoke game started"
1807 );
1808
1809 let mut bots: HashMap<usize, SimpleAi> = HashMap::new();
1810 let mut last_prompt_json: Option<String> = None;
1811 let mut acted = 0usize;
1812 let mut acted_after_concede = 0usize;
1813 let mut conceded = false;
1814 let mut repeat_count = 0usize;
1815
1816 for _ in 0..40_000 {
1817 if session.is_game_over()? {
1818 if !conceded {
1819 return Err("concede smoke: game ended before the concede fired".to_string());
1820 }
1821 break;
1822 }
1823 let Some(prompt_json) = session.get_prompt(0)? else {
1824 std::thread::sleep(Duration::from_millis(10));
1825 continue;
1826 };
1827 if last_prompt_json.as_deref() == Some(prompt_json.as_str()) {
1828 repeat_count += 1;
1829 if repeat_count > STALL_REPEATS {
1830 return Err(format!(
1831 "concede smoke stalled on the same prompt (seats={seats} acted={acted} conceded={conceded} after={acted_after_concede}): {prompt_json}"
1832 ));
1833 }
1834 std::thread::sleep(Duration::from_millis(10));
1835 continue;
1836 }
1837 repeat_count = 0;
1838
1839 let prompt: AgentPrompt = serde_json::from_str(&prompt_json)
1840 .map_err(|err| format!("concede smoke: bad prompt: {err}"))?;
1841 let player = player_index(&prompt.deciding_player_id);
1842
1843 if !conceded && acted >= concede_after {
1844 info!(conceder, acted, "concede smoke: injecting concede");
1845 session.submit_action(&directive_concede_json(conceder))?;
1846 conceded = true;
1847 last_prompt_json = None;
1848 continue;
1849 }
1850 if conceded && player == conceder {
1851 session.submit_action(&directive_concede_json(conceder))?;
1852 last_prompt_json = Some(prompt_json);
1853 continue;
1854 }
1855
1856 if let Some(action) = bots.entry(player).or_default().decide(prompt) {
1857 submit_player_action(session, &action)?;
1858 acted += 1;
1859 if conceded {
1860 acted_after_concede += 1;
1861 }
1862 }
1863 last_prompt_json = Some(prompt_json);
1864 if conceded && seats > 2 && acted_after_concede >= POST_CONCEDE_DECISIONS {
1865 break;
1866 }
1867 }
1868
1869 if !conceded {
1870 return Err("concede smoke: never reached the injection point".to_string());
1871 }
1872 if seats == 2 {
1873 if !session.is_game_over()? {
1874 return Err("concede smoke: 2p concession did not end the game".to_string());
1875 }
1876 info!(acted, "concede smoke: 2p concession ended the game");
1877 } else {
1878 if acted_after_concede < POST_CONCEDE_DECISIONS && !session.is_game_over()? {
1879 return Err(format!(
1880 "concede smoke: game did not progress after the concession (only {acted_after_concede} decisions)"
1881 ));
1882 }
1883 let snapshot = parse_snapshot(session)?;
1884 let status = snapshot
1885 .pointer(&format!("/players/{conceder}/status"))
1886 .and_then(Value::as_str)
1887 .unwrap_or("");
1888 if status != "conceded" {
1889 return Err(format!(
1890 "concede smoke: seat {conceder} has status '{status}', expected 'conceded'"
1891 ));
1892 }
1893 info!(
1894 acted,
1895 acted_after_concede, "concede smoke: game continued past the concession"
1896 );
1897 }
1898 session.end_game()?;
1899 Ok(())
1900}
1901
1902#[cfg(feature = "java-forge")]
1903fn run_self_play_loop<B: JavaBridge>(
1904 session: &mut JavaForgeSession<B>,
1905 max_prompts: usize,
1906) -> Result<(), String> {
1907 const STALL_REPEATS: usize = 100;
1908
1909 let mut bots: HashMap<usize, SimpleAi> = HashMap::new();
1910 let mut last_prompt_json: Option<String> = None;
1911 let mut acted = 0usize;
1912 let mut repeat_count = 0usize;
1913 let mut seen_prompt = false;
1914 let max_iterations = max_prompts.saturating_mul(200).max(2_000);
1915
1916 for _ in 0..max_iterations {
1917 if let Some(prompt_json) = session.get_prompt(0)? {
1918 seen_prompt = true;
1919 if last_prompt_json.as_deref() == Some(prompt_json.as_str()) {
1920 if session.is_game_over()? {
1921 info!(acted, "java-forge self-play reached game over");
1922 return Ok(());
1923 }
1924 repeat_count += 1;
1925 if repeat_count > STALL_REPEATS {
1926 let raw_value: Value =
1927 serde_json::from_str(&prompt_json).unwrap_or(Value::Null);
1928 dump_stuck(
1929 "java re-emitted the same prompt after the bot acted (stall)",
1930 &raw_value,
1931 None,
1932 session,
1933 );
1934 return Err(
1935 "self-play stalled: java re-emitted the same prompt after the bot's action"
1936 .to_string(),
1937 );
1938 }
1939 std::thread::sleep(Duration::from_millis(20));
1940 continue;
1941 }
1942 repeat_count = 0;
1943
1944 let prompt: AgentPrompt = serde_json::from_str(&prompt_json)
1945 .map_err(|err| format!("failed to parse java self-play prompt: {err}"))?;
1946 let player = player_index(&prompt.deciding_player_id);
1947 let raw_value: Value = serde_json::from_str(&prompt_json).unwrap_or(Value::Null);
1948 let normalized = raw_value.clone();
1949
1950 match bots.entry(player).or_default().decide(prompt) {
1951 Some(action) => {
1952 if let Err(err) = submit_player_action(session, &action) {
1953 dump_stuck(
1954 "java rejected the bot action",
1955 &raw_value,
1956 Some(&normalized),
1957 session,
1958 );
1959 return Err(format!(
1960 "self-play: java rejected action for player {player}: {err}"
1961 ));
1962 }
1963 acted += 1;
1964 if acted >= max_prompts {
1965 dump_stuck(
1966 "did not reach game over within max prompts",
1967 &raw_value,
1968 Some(&normalized),
1969 session,
1970 );
1971 return Err(format!(
1972 "self-play did not reach game over within {max_prompts} decisions"
1973 ));
1974 }
1975 }
1976 None => debug!(
1977 player,
1978 prompt_type = prompt_type(&normalized).unwrap_or("<missing>"),
1979 "self-play: no action for prompt (display-only)"
1980 ),
1981 }
1982 last_prompt_json = Some(prompt_json);
1983 continue;
1984 }
1985
1986 if seen_prompt && session.is_game_over()? {
1987 info!(acted, "java-forge self-play reached game over");
1988 return Ok(());
1989 }
1990 std::thread::sleep(Duration::from_millis(20));
1991 }
1992
1993 dump_stuck(
1994 "self-play exceeded its iteration cap without game over",
1995 &Value::Null,
1996 None,
1997 session,
1998 );
1999 Err("self-play exceeded its iteration cap without reaching game over".to_string())
2000}
2001
2002#[cfg(feature = "java-forge")]
2003fn parse_snapshot<B: JavaBridge>(session: &mut JavaForgeSession<B>) -> Result<Value, String> {
2004 let snapshot_json = session.get_snapshot(Some(0))?;
2005 serde_json::from_str(&snapshot_json)
2006 .map_err(|err| format!("failed to parse java self-play snapshot: {err}"))
2007}
2008
2009#[cfg(feature = "java-forge")]
2010fn dump_stuck<B: JavaBridge>(
2011 reason: &str,
2012 prompt: &Value,
2013 normalized: Option<&Value>,
2014 session: &mut JavaForgeSession<B>,
2015) {
2016 let snapshot = parse_snapshot(session).unwrap_or(Value::Null);
2017 let artifact = json!({
2018 "reason": reason,
2019 "rawPrompt": prompt,
2020 "normalizedPrompt": normalized,
2021 "snapshot": snapshot,
2022 });
2023 let ts = std::time::SystemTime::now()
2024 .duration_since(std::time::UNIX_EPOCH)
2025 .map(|d| d.as_millis())
2026 .unwrap_or(0);
2027 let path = workspace_root().join(format!("target/self-play-stuck-{ts}.json"));
2028 match serde_json::to_string_pretty(&artifact) {
2029 Ok(body) => {
2030 if let Err(error) = std::fs::write(&path, body) {
2031 warn!(%error, reason, "self-play stuck; failed to write artifact");
2032 } else {
2033 warn!(path = %path.display(), reason, "self-play stuck; wrote artifact");
2034 }
2035 }
2036 Err(error) => warn!(%error, reason, "self-play stuck; failed to serialize artifact"),
2037 }
2038}
2039
2040#[cfg(feature = "java-forge")]
2041fn submit_player_action<B: JavaBridge>(
2042 session: &mut JavaForgeSession<B>,
2043 action: &PromptOutput,
2044) -> Result<(), String> {
2045 let action_json = serde_json::to_string(action)
2046 .map_err(|err| format!("failed to serialize scenario action: {err}"))?;
2047 session.submit_action(&action_json)?;
2048 Ok(())
2049}
2050
2051#[cfg(feature = "java-forge")]
2052fn prompt_type(prompt: &Value) -> Option<&str> {
2053 prompt
2054 .get("input")
2055 .and_then(|input| input.get("type"))
2056 .and_then(Value::as_str)
2057}
2058
2059#[cfg(feature = "java-forge")]
2060fn play_first_card_action(prompt: &Value, card_name: &str) -> Result<Option<PromptOutput>, String> {
2061 let Some(action) = prompt
2062 .get("input")
2063 .and_then(|input| input.get("actions"))
2064 .and_then(Value::as_array)
2065 .and_then(|actions| {
2066 actions.iter().find(|action| {
2067 action
2068 .get("modeLabel")
2069 .and_then(Value::as_str)
2070 .is_some_and(|label| label.contains(card_name))
2071 })
2072 })
2073 else {
2074 return Ok(None);
2075 };
2076 let action_id = action
2077 .get("id")
2078 .and_then(Value::as_str)
2079 .ok_or_else(|| format!("playable action for '{card_name}' is missing id"))?;
2080 Ok(Some(PromptOutput::ChooseAction(ChooseActionOutput::Act {
2081 action_id: action_id.to_string(),
2082 })))
2083}
2084
2085#[cfg(feature = "java-forge")]
2086fn prompt_card_ids(prompt: &Value, field: &str, count: usize) -> Result<Vec<String>, String> {
2087 let card_ids = prompt
2088 .get("input")
2089 .and_then(|input| input.get(field))
2090 .and_then(Value::as_array)
2091 .ok_or_else(|| format!("prompt is missing {field}"))?;
2092 if card_ids.len() < count {
2093 return Err(format!(
2094 "prompt field {field} has {} cards, need {count}",
2095 card_ids.len()
2096 ));
2097 }
2098 Ok(card_ids
2099 .iter()
2100 .take(count)
2101 .filter_map(Value::as_str)
2102 .map(str::to_string)
2103 .collect())
2104}
2105
2106#[cfg(feature = "java-forge")]
2107fn battlefield_contains(game_view: &GameViewDto, card_name: &str) -> bool {
2108 use manabrew_agent_interface::game_view_dto::{CardView, ZoneKind};
2109 game_view.zones.iter().any(|zone| {
2110 zone.zone == ZoneKind::Battlefield
2111 && zone.owner_id == "player-0"
2112 && zone.cards.iter().any(|card| match card {
2113 CardView::Visible(dto) => dto.identity.name == card_name,
2114 CardView::Hidden { .. } => false,
2115 })
2116 })
2117}
2118
2119pub trait JavaBridge {
2120 fn initialize(&mut self, assets_dir: &str) -> Result<(), String>;
2121 fn start_game_json(&mut self, request_json: &str) -> Result<String, String>;
2122 fn submit_action(&mut self, session_id: &str, action_json: &str) -> Result<String, String>;
2123 fn get_prompt(
2124 &mut self,
2125 session_id: &str,
2126 player_index: usize,
2127 ) -> Result<Option<String>, String>;
2128 fn get_snapshot(&mut self, session_id: &str, viewer: Option<usize>) -> Result<String, String>;
2129 fn is_game_over(&mut self, session_id: &str) -> Result<bool, String>;
2130 fn end_game(&mut self, session_id: &str) -> Result<(), String>;
2131 fn abort_game(&mut self, session_id: &str) -> Result<(), String>;
2132}
2133
2134pub struct JavaForgeSession<B> {
2135 bridge: B,
2136 session_id: Option<String>,
2137}
2138
2139impl<B: JavaBridge> JavaForgeSession<B> {
2140 pub fn new(bridge: B) -> Self {
2141 Self {
2142 bridge,
2143 session_id: None,
2144 }
2145 }
2146
2147 pub fn initialize(&mut self, assets_dir: &str) -> Result<(), String> {
2148 self.bridge.initialize(assets_dir)
2149 }
2150
2151 pub fn start_game(&mut self, request: &StartGameRequest) -> Result<String, String> {
2152 let request_json = request.to_json().map_err(|err| err.to_string())?;
2153 let response_json = self.bridge.start_game_json(&request_json)?;
2154 let response: StartGameResponse =
2155 serde_json::from_str(&response_json).map_err(|err| err.to_string())?;
2156 self.session_id = Some(response.session_id.clone());
2157 Ok(response.session_id)
2158 }
2159
2160 pub fn submit_action(&mut self, action_json: &str) -> Result<String, String> {
2161 let session_id = self.require_session_id()?.to_string();
2162 self.bridge.submit_action(&session_id, action_json)
2163 }
2164
2165 pub fn get_prompt(&mut self, player_index: usize) -> Result<Option<String>, String> {
2166 let session_id = self.require_session_id()?.to_string();
2167 self.bridge.get_prompt(&session_id, player_index)
2168 }
2169
2170 pub fn get_snapshot(&mut self, viewer: Option<usize>) -> Result<String, String> {
2171 let session_id = self.require_session_id()?.to_string();
2172 self.bridge.get_snapshot(&session_id, viewer)
2173 }
2174
2175 pub fn is_game_over(&mut self) -> Result<bool, String> {
2176 let session_id = self.require_session_id()?.to_string();
2177 self.bridge.is_game_over(&session_id)
2178 }
2179
2180 pub fn end_game(&mut self) -> Result<(), String> {
2181 let Some(session_id) = self.session_id.take() else {
2182 return Ok(());
2183 };
2184 self.bridge.end_game(&session_id)
2185 }
2186
2187 fn require_session_id(&self) -> Result<&str, String> {
2188 self.session_id
2189 .as_deref()
2190 .ok_or_else(|| "java-forge session has not started".to_string())
2191 }
2192}
2193
2194pub struct UnavailableJavaBridge;
2195
2196impl JavaBridge for UnavailableJavaBridge {
2197 fn initialize(&mut self, _assets_dir: &str) -> Result<(), String> {
2198 Err(unsupported_message().to_string())
2199 }
2200
2201 fn start_game_json(&mut self, _request_json: &str) -> Result<String, String> {
2202 Err(unsupported_message().to_string())
2203 }
2204
2205 fn submit_action(&mut self, _session_id: &str, _action_json: &str) -> Result<String, String> {
2206 Err(unsupported_message().to_string())
2207 }
2208
2209 fn get_prompt(
2210 &mut self,
2211 _session_id: &str,
2212 _player_index: usize,
2213 ) -> Result<Option<String>, String> {
2214 Err(unsupported_message().to_string())
2215 }
2216
2217 fn get_snapshot(
2218 &mut self,
2219 _session_id: &str,
2220 _viewer: Option<usize>,
2221 ) -> Result<String, String> {
2222 Err(unsupported_message().to_string())
2223 }
2224
2225 fn is_game_over(&mut self, _session_id: &str) -> Result<bool, String> {
2226 Err(unsupported_message().to_string())
2227 }
2228
2229 fn end_game(&mut self, _session_id: &str) -> Result<(), String> {
2230 Err(unsupported_message().to_string())
2231 }
2232
2233 fn abort_game(&mut self, _session_id: &str) -> Result<(), String> {
2234 Err(unsupported_message().to_string())
2235 }
2236}
2237
2238#[cfg(feature = "java-forge")]
2239#[derive(serde::Deserialize)]
2240struct SubprocessReply {
2241 ok: bool,
2242 #[serde(default)]
2243 result: String,
2244 #[serde(default)]
2245 error: Option<String>,
2246}
2247
2248#[cfg(feature = "java-forge")]
2249const CALL_TIMEOUT: Duration = Duration::from_secs(60);
2250#[cfg(feature = "java-forge")]
2251const SHUTDOWN_GRACE: Duration = Duration::from_secs(5);
2252
2253#[cfg(feature = "java-forge")]
2254pub struct SubprocessBridge {
2255 child: Child,
2256 stdin: BufWriter<ChildStdin>,
2257 stdout_rx: std_mpsc::Receiver<String>,
2258 stdout_handle: Option<std::thread::JoinHandle<()>>,
2259 stderr_handle: Option<std::thread::JoinHandle<()>>,
2260}
2261
2262#[cfg(feature = "java-forge")]
2263impl SubprocessBridge {
2264 fn spawn(config: &JavaRuntimeConfig) -> Result<Self, String> {
2265 config.validate()?;
2266
2267 let java_bin = resolve_java_bin(config);
2268 let mut cmd = Command::new(&java_bin);
2269 cmd.arg("-Dfile.encoding=UTF-8");
2270 cmd.arg("-Dsun.stdout.encoding=UTF-8");
2271 cmd.arg("-Dsun.stderr.encoding=UTF-8");
2272 cmd.arg("-Djava.awt.headless=true");
2273 cmd.arg("-jar").arg(&config.harness_jar);
2274 cmd.arg("--interactive-server");
2275 cmd.arg("--forge-home")
2276 .arg(format!("{}/", config.assets_dir.display()));
2277
2278 let mut child = cmd
2279 .stdin(Stdio::piped())
2280 .stdout(Stdio::piped())
2281 .stderr(Stdio::piped())
2282 .spawn()
2283 .map_err(|err| format!("failed to spawn java subprocess: {err}"))?;
2284
2285 let stdin = child
2286 .stdin
2287 .take()
2288 .ok_or_else(|| "java subprocess has no stdin".to_string())?;
2289 let stdout = child
2290 .stdout
2291 .take()
2292 .ok_or_else(|| "java subprocess has no stdout".to_string())?;
2293 let stderr = child.stderr.take();
2294
2295 let (stdout_tx, stdout_rx) = std_mpsc::sync_channel::<String>(1024);
2298 let stdout_handle = std::thread::spawn(move || {
2299 let reader = BufReader::new(stdout);
2300 for line in reader.lines().map_while(Result::ok) {
2301 if stdout_tx.send(line).is_err() {
2302 break;
2303 }
2304 }
2305 });
2306
2307 let stderr_handle = std::thread::spawn(move || {
2308 if let Some(stderr) = stderr {
2309 let reader = BufReader::new(stderr);
2310 for line in reader.lines().map_while(Result::ok) {
2311 if line.contains("Exception") || line.contains("ERROR") {
2312 warn!(target: "self_hosted_node::java", "[java] {line}");
2313 } else {
2314 debug!(target: "self_hosted_node::java", "[java] {line}");
2315 }
2316 }
2317 }
2318 });
2319
2320 Ok(Self {
2321 child,
2322 stdin: BufWriter::new(stdin),
2323 stdout_rx,
2324 stdout_handle: Some(stdout_handle),
2325 stderr_handle: Some(stderr_handle),
2326 })
2327 }
2328
2329 fn call(&mut self, request_json: &str) -> Result<String, String> {
2330 loop {
2334 match self.stdout_rx.try_recv() {
2335 Ok(stale) => {
2336 debug!(target: "self_hosted_node::java", line = %stale, "discarding stale stdout line");
2337 }
2338 Err(TryRecvError::Empty) => break,
2339 Err(TryRecvError::Disconnected) => {
2340 return Err("java subprocess closed stdout (crashed?)".to_string());
2341 }
2342 }
2343 }
2344
2345 self.stdin
2346 .write_all(request_json.as_bytes())
2347 .map_err(|err| format!("failed to write subprocess stdin: {err}"))?;
2348 self.stdin
2349 .write_all(b"\n")
2350 .map_err(|err| format!("failed to write subprocess newline: {err}"))?;
2351 self.stdin
2352 .flush()
2353 .map_err(|err| format!("failed to flush subprocess stdin: {err}"))?;
2354
2355 let deadline = Instant::now() + CALL_TIMEOUT;
2356 loop {
2357 let remaining = deadline.saturating_duration_since(Instant::now());
2358 if remaining.is_zero() {
2359 return Err(format!(
2360 "java subprocess timed out after {}s",
2361 CALL_TIMEOUT.as_secs()
2362 ));
2363 }
2364 match self.stdout_rx.recv_timeout(remaining) {
2365 Ok(line) => {
2366 let trimmed = line.trim();
2367 if trimmed.is_empty() {
2368 continue;
2369 }
2370 match serde_json::from_str::<SubprocessReply>(trimmed) {
2371 Ok(reply) if reply.ok => return Ok(reply.result),
2372 Ok(reply) => {
2373 return Err(reply.error.unwrap_or_else(|| "unknown java error".into()));
2374 }
2375 Err(_) => {
2376 debug!(target: "self_hosted_node::java", line = trimmed, "non-protocol stdout line");
2377 }
2378 }
2379 }
2380 Err(RecvTimeoutError::Timeout) => {
2381 return Err(format!(
2382 "java subprocess timed out after {}s",
2383 CALL_TIMEOUT.as_secs()
2384 ));
2385 }
2386 Err(RecvTimeoutError::Disconnected) => {
2387 return Err("java subprocess closed stdout (crashed?)".to_string());
2388 }
2389 }
2390 }
2391 }
2392
2393 fn reset(&mut self) -> Result<(), String> {
2394 self.call("{\"command\":\"reset\"}").map(|_| ())
2395 }
2396
2397 fn is_alive(&mut self) -> bool {
2398 matches!(self.child.try_wait(), Ok(None))
2399 }
2400
2401 fn shutdown(mut self) {
2402 let _ = self.stdin.write_all(b"{\"command\":\"quit\"}\n");
2403 let _ = self.stdin.flush();
2404 let deadline = Instant::now() + SHUTDOWN_GRACE;
2405 loop {
2406 match self.child.try_wait() {
2407 Ok(Some(_)) => break,
2408 Ok(None) if Instant::now() >= deadline => {
2409 let _ = self.child.kill();
2410 let _ = self.child.wait();
2411 break;
2412 }
2413 Ok(None) => std::thread::sleep(Duration::from_millis(100)),
2414 Err(_) => break,
2415 }
2416 }
2417 if let Some(handle) = self.stdout_handle.take() {
2418 let _ = handle.join();
2419 }
2420 if let Some(handle) = self.stderr_handle.take() {
2421 let _ = handle.join();
2422 }
2423 }
2424}
2425
2426#[cfg(feature = "java-forge")]
2427impl JavaBridge for SubprocessBridge {
2428 fn initialize(&mut self, _assets_dir: &str) -> Result<(), String> {
2429 Ok(())
2430 }
2431
2432 fn start_game_json(&mut self, request_json: &str) -> Result<String, String> {
2433 let body = json!({ "command": "startGame", "payload": request_json });
2434 self.call(&body.to_string())
2435 }
2436
2437 fn submit_action(&mut self, session_id: &str, action_json: &str) -> Result<String, String> {
2438 let body = json!({
2439 "command": "submitAction",
2440 "sessionId": session_id,
2441 "payload": action_json,
2442 });
2443 self.call(&body.to_string())
2444 }
2445
2446 fn get_prompt(
2447 &mut self,
2448 session_id: &str,
2449 player_index: usize,
2450 ) -> Result<Option<String>, String> {
2451 let body = json!({
2452 "command": "getPrompt",
2453 "sessionId": session_id,
2454 "playerIndex": player_index,
2455 });
2456 let prompt = self.call(&body.to_string())?;
2457 Ok((!prompt.is_empty()).then_some(prompt))
2458 }
2459
2460 fn get_snapshot(&mut self, session_id: &str, viewer: Option<usize>) -> Result<String, String> {
2461 let mut body = json!({ "command": "getSnapshot", "sessionId": session_id });
2462 if let Some(viewer) = viewer {
2463 body["viewer"] = json!(viewer);
2464 }
2465 self.call(&body.to_string())
2466 }
2467
2468 fn is_game_over(&mut self, session_id: &str) -> Result<bool, String> {
2469 let body = json!({ "command": "getGameOver", "sessionId": session_id });
2470 let value = self.call(&body.to_string())?;
2471 Ok(value.trim() == "true")
2472 }
2473
2474 fn end_game(&mut self, session_id: &str) -> Result<(), String> {
2475 let body = json!({ "command": "endGame", "sessionId": session_id });
2476 self.call(&body.to_string()).map(|_| ())
2477 }
2478
2479 fn abort_game(&mut self, session_id: &str) -> Result<(), String> {
2480 let body = json!({ "command": "abortGame", "sessionId": session_id });
2481 self.call(&body.to_string()).map(|_| ())
2482 }
2483}
2484
2485#[cfg(feature = "java-forge")]
2486impl Drop for SubprocessBridge {
2487 fn drop(&mut self) {
2488 let _ = self.child.kill();
2489 let _ = self.child.wait();
2490 if let Some(handle) = self.stdout_handle.take() {
2491 let _ = handle.join();
2492 }
2493 if let Some(handle) = self.stderr_handle.take() {
2494 let _ = handle.join();
2495 }
2496 }
2497}
2498
2499#[cfg(feature = "java-forge")]
2500fn resolve_java_bin(config: &JavaRuntimeConfig) -> String {
2501 if let Some(home) = &config.java_home {
2502 let bin = home.join("bin").join("java");
2503 if bin.is_file() {
2504 return bin.to_string_lossy().to_string();
2505 }
2506 }
2507 if let Ok(home) = env::var("JAVA_HOME") {
2508 let bin = PathBuf::from(home).join("bin").join("java");
2509 if bin.is_file() {
2510 return bin.to_string_lossy().to_string();
2511 }
2512 }
2513 "java".to_string()
2514}
2515
2516#[derive(Debug, Clone, Serialize)]
2517#[serde(rename_all = "camelCase")]
2518pub struct StartGameRequest {
2519 game_id: String,
2520 variant: String,
2521 starting_life: i32,
2522 seed: u64,
2523 players: Vec<PlayerConfig>,
2524}
2525
2526#[derive(Debug, Clone, Serialize)]
2527#[serde(rename_all = "camelCase")]
2528pub struct PlayerConfig {
2529 name: String,
2530 deck: Vec<CardIdentityForJava>,
2531 commander_names: Vec<String>,
2532 ai: bool,
2533}
2534
2535#[derive(Debug, Clone, Serialize)]
2536#[serde(rename_all = "camelCase")]
2537pub struct CardIdentityForJava {
2538 name: String,
2539 set_code: Option<String>,
2540}
2541
2542#[derive(Debug, serde::Deserialize)]
2543#[serde(rename_all = "camelCase")]
2544struct StartGameResponse {
2545 session_id: String,
2546 #[allow(dead_code)]
2547 player_indexes: Vec<usize>,
2548}
2549
2550impl StartGameRequest {
2551 pub fn new(
2552 game_id: String,
2553 variant: String,
2554 starting_life: i32,
2555 seed: u64,
2556 players: Vec<PlayerConfig>,
2557 ) -> Self {
2558 Self {
2559 game_id,
2560 variant,
2561 starting_life,
2562 seed,
2563 players,
2564 }
2565 }
2566
2567 pub fn to_json(&self) -> Result<String, serde_json::Error> {
2568 serde_json::to_string(self)
2569 }
2570}
2571
2572impl PlayerConfig {
2573 pub fn new(name: String, deck: &[DeckCardIdentity], commander_names: Vec<String>) -> Self {
2574 Self {
2575 name,
2576 deck: deck.iter().map(CardIdentityForJava::from).collect(),
2577 commander_names,
2578 ai: false,
2579 }
2580 }
2581}
2582
2583impl From<&DeckCardIdentity> for CardIdentityForJava {
2584 fn from(identity: &DeckCardIdentity) -> Self {
2585 Self {
2586 name: java_card_name(&identity.name),
2587 set_code: (!identity.set_code.is_empty()).then(|| identity.set_code.clone()),
2588 }
2589 }
2590}
2591
2592fn java_card_name(name: &str) -> String {
2593 name.split_once(" // ")
2594 .map(|(front, _)| front.to_string())
2595 .unwrap_or_else(|| name.to_string())
2596}
2597
2598fn env_path(key: &str) -> Option<PathBuf> {
2599 env::var_os(key)
2600 .filter(|value| !value.is_empty())
2601 .map(PathBuf::from)
2602}
2603
2604fn env_classpath(key: &str) -> Vec<PathBuf> {
2605 let Some(value) = env::var_os(key) else {
2606 return Vec::new();
2607 };
2608 env::split_paths(&value).collect()
2609}
2610
2611fn require_dir(path: &Path, label: &str) -> Result<(), String> {
2612 if path.is_dir() {
2613 Ok(())
2614 } else {
2615 Err(format!("{label} does not exist: {}", path.display()))
2616 }
2617}
2618
2619fn require_file(path: &Path, label: &str) -> Result<(), String> {
2620 if path.is_file() {
2621 Ok(())
2622 } else {
2623 Err(format!("{label} does not exist: {}", path.display()))
2624 }
2625}