Skip to main content

self_hosted_node/engine_backend/
java_backend.rs

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// Unlike run_smoke_game, which spawns the JAR in a subprocess, this drives the
125// GraalVM native library in-process: isolate creation, forge_initialize (which
126// loads the card database) and a real game start. CI runs it after restoring
127// forge-harness/native/build from cache, where nothing else would notice a
128// stale or broken libforgeharness.
129#[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    // SELF_HOSTED_NODE_MAX_GAMES is the concurrent-session ceiling for this node.
588    // SELF_HOSTED_NODE_GAMES_PER_JVM multiplexes that many sessions into each
589    // subprocess (the engine is concurrency-safe since the endstep patches), so
590    // the pool spawns ceil(max_games / games_per_jvm) processes. Default 1 keeps
591    // the historical one-subprocess-per-game shape.
592    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// A GraalVM isolate hosts the in-process Forge engine. The `thread` handle is
716// bound to the hosted-engine thread that created or attached it (isolate
717// threads are not portable), so the handle is Rc/!Send by design. In shared
718// mode one isolate serves all rooms: the first creator initializes Forge, later
719// rooms attach their own thread to it and sessions coexist in the adapter.
720#[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        // SELF_HOSTED_NODE_SHARED_ISOLATE=1 hosts every room's game in one
827        // isolate (safe since the endstep concurrency patches), so the card db
828        // loads once. Default keeps the historical isolate-per-game shape.
829        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                        // prompt_id 0 is transport-synthesized: the
1245                        // absent-player default, exempt from validation.
1246                        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                        // A directive can arrive while another player's prompt
1323                        // is open (this loop drains every seat), so it names
1324                        // its seat.
1325                        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        // Bounded so a chatty Java side can't grow the queue without bound.
2296        // Cap is generous — protocol replies are one line per request.
2297        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        // Drain anything still queued from a prior request — a previous call()
2331        // that timed out may have left its reply in the channel, and consuming
2332        // it now would shift every subsequent call off-by-one.
2333        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}