gizmo-scripting 0.10.0

//! Fighter API — Lua'ya sunulan dövüş sistemi fonksiyonları
//!
//! Lua scriptlerinden kombo sorgulama, hitstop/hitstun uygulama ve saldırı başlatma için kullanılır.

use crate::commands::{CommandQueue, ScriptCommand};
use gizmo_core::World;
use mlua::prelude::*;
use std::sync::Arc;

pub fn register_fighter_api(lua: &Lua, command_queue: Arc<CommandQueue>) -> Result<(), LuaError> {
    crate::api_table::register_protected(lua, "fighter", |fighter_table| {

    // Oku-Yaz tablosu
    fighter_table.raw_set("_buffers", lua.create_table()?)?;
    fighter_table.raw_set("_is_locked", lua.create_table()?)?;

    // === SET FIGHTER MOVE ===
    {
        let cq = command_queue.clone();
        fighter_table.raw_set(
            "set_move",
            lua.create_function(
                move |_, (id, name, startup, active, recovery, damage): (u32, String, u32, u32, u32, f32)| {
                    cq.push(ScriptCommand::SetFighterMove {
                        id,
                        name,
                        startup,
                        active,
                        recovery,
                        damage,
                    });
                    Ok(())
                },
            )?,
        )?;
    }

    // === APPLY HITSTOP ===
    {
        let cq = command_queue.clone();
        fighter_table.raw_set(
            "apply_hitstop",
            lua.create_function(move |_, (id, frames): (u32, u32)| {
                cq.push(ScriptCommand::ApplyHitstop(id, frames));
                Ok(())
            })?,
        )?;
    }

    // === APPLY HITSTUN ===
    {
        let cq = command_queue.clone();
        fighter_table.raw_set(
            "apply_hitstun",
            lua.create_function(move |_, (id, frames): (u32, u32)| {
                cq.push(ScriptCommand::ApplyHitstun(id, frames));
                Ok(())
            })?,
        )?;
    }


    // Lua tarafında kombo kontrol eden yardımcı fonksiyon
    lua.load(
        r#"
        function fighter.is_locked(id)
            return fighter._is_locked[id] or false
        end

        function fighter.check_combo(id, combo, max_gap)
            local buffer = fighter._buffers[id]
            if not buffer then return false end

            local combo_idx = #combo
            if combo_idx == 0 then return false end

            local gap_counter = 0
            
            for i = 1, #buffer do
                local frame = buffer[i]
                local target_input = combo[combo_idx]

                if frame.just_pressed[target_input] then
                    combo_idx = combo_idx - 1
                    gap_counter = 0
                    if combo_idx == 0 then
                        return true
                    end
                elseif gap_counter >= max_gap then
                    return false
                else
                    gap_counter = gap_counter + 1
                end
            end
            
            return false
        end
    "#,
    )
    .exec()?;

        Ok(())
    })
}

#[tracing::instrument(skip_all, name = "script_fighter_read")]
pub fn update_fighter_read_api(lua: &Lua, world: &World) -> Result<(), LuaError> {
    // The real table, not the global: the global is a read-only proxy so a script cannot
    // rewrite the API (see `api_table`), and the engine's per-frame writes go behind it.
    let fighter_table = crate::api_table::raw(lua, "fighter")?;

    let buffers = lua.create_table()?;
    let is_locked = lua.create_table()?;

    let controllers = world.borrow::<gizmo_physics_core::components::FighterController>();
    for (eid, _) in controllers.iter() {
        if let Some(fighter) = controllers.get(eid) {
            is_locked.set(eid, fighter.is_locked())?;

            let frames_table = lua.create_table()?;
            for (i, frame) in fighter.input_buffer.frames.iter().enumerate() {
                let frame_table = lua.create_table()?;
                
                let jp_table = lua.create_table()?;
                for k in &frame.just_pressed {
                    jp_table.set(k.clone(), true)?;
                }
                
                let p_table = lua.create_table()?;
                for k in &frame.pressed {
                    p_table.set(k.clone(), true)?;
                }
                
                frame_table.set("just_pressed", jp_table)?;
                frame_table.set("pressed", p_table)?;
                
                frames_table.set(i + 1, frame_table)?;
            }
            buffers.set(eid, frames_table)?;
        }
    }

    fighter_table.raw_set("_buffers", buffers)?;
    fighter_table.raw_set("_is_locked", is_locked)?;

    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::commands::CommandQueue;

    /// Belirtilen frame indekslerinde `input` tuşunu just_pressed olarak
    /// işaretleyen bir buffer'ı Lua'ya kuran yardımcı ve check_combo sonucu.
    fn run_combo(setup: &str) -> bool {
        let lua = Lua::new();
        register_fighter_api(&lua, Arc::new(CommandQueue::new())).unwrap();
        lua.load(setup).exec().unwrap();
        lua.load("return fighter.check_combo(1, combo, max_gap)")
            .eval()
            .unwrap()
    }

    /// Regression: max_gap=2 tam olarak 2 frame boşluğa izin vermeli.
    /// 'b' bulunduktan sonra 2 boş frame + 'a' => kabul.
    /// 3 boş frame => ret.
    #[test]
    fn combo_gap_boundary_is_exact() {
        // Buffer ileri taranır; önce combo'nun son elemanı ('b') aranır.
        // frame1='b', frame2/3 boş, frame4='a' -> 2 boşluk, kabul edilmeli.
        let accepted = run_combo(
            r#"
            combo = { "a", "b" }
            max_gap = 2
            local function f(k) return { just_pressed = k and { [k] = true } or {} } end
            fighter._buffers[1] = { f("b"), f(nil), f(nil), f("a") }
            "#,
        );
        assert!(accepted, "2 frame boşluk max_gap=2 için kabul edilmeli");

        // frame1='b', frame2/3/4 boş, frame5='a' -> 3 boşluk, reddedilmeli.
        let rejected = run_combo(
            r#"
            combo = { "a", "b" }
            max_gap = 2
            local function f(k) return { just_pressed = k and { [k] = true } or {} } end
            fighter._buffers[1] = { f("b"), f(nil), f(nil), f(nil), f("a") }
            "#,
        );
        assert!(
            !rejected,
            "3 frame boşluk max_gap=2 için REDDEDİLMELİ (off-by-one)"
        );
    }

    /// Bitişik (boşluksuz) tam kombo tanınmalı; combo tersten (son eleman önce) taranır.
    #[test]
    fn adjacent_full_combo_is_recognized() {
        let accepted = run_combo(
            r#"
            combo = { "a", "b", "c" }
            max_gap = 1
            local function f(k) return { just_pressed = { [k] = true } } end
            -- Buffer ileri taranır, önce combo'nun sonu ("c") aranır: c, b, a
            fighter._buffers[1] = { f("c"), f("b"), f("a") }
            "#,
        );
        assert!(accepted, "bitişik c-b-a dizisi a,b,c kombosunu tamamlamalı");
    }

    /// Boş kombo listesi asla eşleşmemeli (combo_idx == 0 erken çıkış).
    #[test]
    fn empty_combo_never_matches() {
        let matched = run_combo(
            r#"
            combo = {}
            max_gap = 5
            local function f(k) return { just_pressed = { [k] = true } } end
            fighter._buffers[1] = { f("a"), f("b") }
            "#,
        );
        assert!(!matched, "boş kombo false dönmeli");
    }

    /// Entity için buffer yoksa check_combo güvenle false dönmeli (nil buffer koruması).
    #[test]
    fn missing_buffer_returns_false() {
        let matched = run_combo(
            r#"
            combo = { "a" }
            max_gap = 5
            -- fighter._buffers[1] hiç ayarlanmadı
            "#,
        );
        assert!(!matched, "buffer yoksa false dönmeli");
    }

    /// Kombo tam tamamlanmazsa (yalnız son eleman var, ilki yok) false dönmeli.
    #[test]
    fn partially_matched_combo_is_rejected() {
        let matched = run_combo(
            r#"
            combo = { "a", "b" }
            max_gap = 5
            local function f(k) return { just_pressed = { [k] = true } } end
            -- Sadece "b" var, "a" hiç basılmadı → kombo tamamlanmaz.
            fighter._buffers[1] = { f("b"), f("x"), f("y") }
            "#,
        );
        assert!(!matched, "eksik kombo tamamlanmamış sayılmalı");
    }

    /// is_locked: _is_locked tablosunda giriş yoksa false, true ise true.
    #[test]
    fn is_locked_reads_table_with_false_default() {
        let lua = Lua::new();
        register_fighter_api(&lua, Arc::new(CommandQueue::new())).unwrap();
        lua.load(
            r#"
            assert(fighter.is_locked(1) == false, "giriş yoksa varsayılan false")
            fighter._is_locked[1] = true
            assert(fighter.is_locked(1) == true, "true set edilince true")
            "#,
        )
        .exec()
        .unwrap();
    }

    /// set_move / apply_hitstop / apply_hitstun doğru komutları (frame verileri dahil) kuyruğa yazmalı.
    #[test]
    fn fighter_write_calls_push_expected_commands() {
        let lua = Lua::new();
        let cq = Arc::new(CommandQueue::new());
        register_fighter_api(&lua, cq.clone()).unwrap();

        lua.load(
            r#"
            fighter.set_move(1, "jab", 3, 2, 8, 5.5)
            fighter.apply_hitstop(1, 6)
            fighter.apply_hitstun(2, 20)
            "#,
        )
        .exec()
        .unwrap();

        let cmds = cq.drain();
        assert_eq!(cmds.len(), 3);
        match &cmds[0] {
            ScriptCommand::SetFighterMove { id, name, startup, active, recovery, damage } => {
                assert_eq!(*id, 1);
                assert_eq!(name, "jab");
                assert_eq!((*startup, *active, *recovery), (3, 2, 8));
                assert!((damage - 5.5).abs() < 1e-6);
            }
            other => panic!("beklenen SetFighterMove, gelen {other:?}"),
        }
        assert!(matches!(cmds[1], ScriptCommand::ApplyHitstop(1, 6)));
        assert!(matches!(cmds[2], ScriptCommand::ApplyHitstun(2, 20)));
    }
}