// =====================================================
// SWIFT + RAITON + FUTON RELEASE BOSS AI (CustomNPCs 1.12.2)
// Swift moves -> ShinobiNPCScriptAPI
// Raiton/Futon moves -> direct calls
// 2 phases (above/below 50% HP), weighted picks,
// Rage on spike -> spam Thunderous Cross.
// =====================================================

// ================= CONFIG =================
var BASE_COOLDOWN       = 0;
var WINDUP_TICKS        = 12;
var AGGRESSIVE_HP_RATIO = 0.50;

var RAGE_TICKS      = 20 * 8;  // 8 s
var RAGE_THUNDER_CD = 3;
var RAGE_FAIL_CD    = 4;

// Dodge config
var DODGE_CHANCE        = 0.30;  // 30% chance to dodge an incoming hit
var DODGE_COOLDOWN_MAX  = 60;    // 3 s between dodges
var DODGE_BEHIND_CHANCE = 0.60;  // 60% of dodges teleport behind player, rest just sidestep

var AIM_EVERY_CAST = true;
var AIM_PITCH      = true;

// ---- Normal phase weights ----
var WEIGHTS_BASE = {
    MOMENTUM  : 40,
    THUNDEROUS: 35,
    GIAN      : 25,
    VACWAVE   : 30
};

// ---- Aggressive phase (below 50% HP) ----
var WEIGHTS_AGGRO = {
    MOMENTUM  : 50,
    THUNDEROUS: 30,
    GIAN      : 35,
    VACWAVE   : 20
};

var POWERS = {
    GIAN   : 4.0,
    VACWAVE: 4.0
};

var SWIFT_POWER = 10.0;

// ================= STATE =================
var flightActivated = false;
var flightRefreshCd = 0;
var FLIGHT_REFRESH  = 60;
var lastPhase       = -1;
var rageTicks       = 0;
var rageCd          = 0;
var dodgeCd         = 0;

// ================= IMPORTS =================
var Shinobi        = Java.type("com.leolifeless.shinobiaddon.compat.customnpcs.ShinobiNPCScriptAPI");
var ItemRaiton     = Java.type("net.narutomod.item.ItemRaiton");
var ItemFuton      = Java.type("net.narutomod.item.ItemFuton");
var ItemStack      = Java.type("net.minecraft.item.ItemStack");
var NBTTagCompound = Java.type("net.minecraft.nbt.NBTTagCompound");
var Math           = Java.type("java.lang.Math");

// ================= HELPERS =================
function getStored(npc, key, defVal) {
    var d = npc.getStoreddata();
    if (!d.has(key)) {
        if (defVal !== undefined) d.put(key, defVal);
        return defVal;
    }
    return d.get(key);
}
function setStored(npc, key, val) {
    npc.getStoreddata().put(key, val);
}

function clamp(v, a, b) {
    return v < a ? a : (v > b ? b : v);
}

function copyWeights(obj) {
    var out = {};
    for (var k in obj) out[k] = obj[k];
    return out;
}

function weightedPick(weights) {
    var keys = [], sum = 0;
    for (var k in weights) { keys.push(k); sum += weights[k]; }
    if (!keys.length || sum <= 0) return null;
    var r = Math.random() * sum, acc = 0;
    for (var i = 0; i < keys.length; i++) {
        acc += weights[keys[i]];
        if (r <= acc) return keys[i];
    }
    return keys[keys.length - 1];
}

function getHp(mc) {
    try { if (mc && mc.getHealth) return mc.getHealth(); } catch(e) {}
    try { if (mc && mc.health != null) return mc.health; } catch(e) {}
    return 1;
}
function getMaxHp(mc) {
    try { if (mc && mc.getMaxHealth) return mc.getMaxHealth(); } catch(e) {}
    return 20;
}

function distToTarget(npc, t) {
    if (!t) return 9999;
    var me = npc.getMCEntity();
    var dx = me.posX - t.posX;
    var dy = me.posY - t.posY;
    var dz = me.posZ - t.posZ;
    return Math.sqrt(dx*dx + dy*dy + dz*dz);
}

function faceTarget(me, t) {
    try {
        var dx = t.posX - me.posX;
        var dz = t.posZ - me.posZ;
        var dy = (t.posY + t.height * 0.6) - (me.posY + me.getEyeHeight());
        me.rotationYaw = Math.atan2(dz, dx) * 180 / Math.PI - 90;
        me.rotationYawHead = me.rotationYaw;
        me.renderYawOffset = me.rotationYaw;
        if (AIM_PITCH) {
            me.rotationPitch = -(Math.atan2(dy, Math.sqrt(dx*dx + dz*dz)) * 180 / Math.PI);
        }
    } catch(e) {}
}

// Teleport behind the target and face them
function teleportBehindTarget(me, t) {
    try {
        var angle = Math.toRadians(t.rotationYaw);
        var offsetX = -Math.sin(angle) * 2.0;
        var offsetZ =  Math.cos(angle) * 2.0;
        me.setPositionAndUpdate(
            t.posX + offsetX,
            t.posY,
            t.posZ + offsetZ
        );
        faceTarget(me, t);
    } catch(e) {}
}

// Sidestep dodge — teleport to a random side
function teleportSidestep(me, t) {
    try {
        var angle = Math.toRadians(t.rotationYaw) + (Math.PI / 2) * (Math.random() < 0.5 ? 1 : -1);
        var offsetX = -Math.sin(angle) * 3.0;
        var offsetZ =  Math.cos(angle) * 3.0;
        me.setPositionAndUpdate(
            t.posX + offsetX,
            t.posY,
            t.posZ + offsetZ
        );
        faceTarget(me, t);
    } catch(e) {}
}

function makeRaitonStack() {
    var s = new ItemStack(ItemRaiton.block);
    try {
        if (!s.hasTagCompound()) s.setTagCompound(new NBTTagCompound());
    } catch(e) {}
    return s;
}

function makeFutonStack() {
    var s = new ItemStack(ItemFuton.block);
    try {
        if (!s.hasTagCompound()) s.setTagCompound(new NBTTagCompound());
    } catch(e) {}
    return s;
}

function callJutsuSafe(jutsuEnum, stack, entity, power) {
    try {
        return !!jutsuEnum.jutsu.createJutsu(stack, entity, power);
    } catch(e) { return false; }
}

function castPick(pick, npc, me) {
    if (pick === "THUNDEROUS") return Shinobi.cast(npc, "swift_thunderous_cross", SWIFT_POWER, true);
    if (pick === "MOMENTUM")   return Shinobi.cast(npc, "swift_momentum_breaker",  SWIFT_POWER, true);
    if (pick === "GIAN")       return callJutsuSafe(ItemRaiton.GIAN,   makeRaitonStack(), me, POWERS.GIAN);
    if (pick === "VACWAVE")    return callJutsuSafe(ItemFuton.VACWAVE,  makeFutonStack(),  me, POWERS.VACWAVE);
    return false;
}

var NEEDS_WINDUP = { GIAN: true };

// ================= MAIN TICK =================
function tick(event) {
    var npc = event.npc;
    if (!npc) return;

    var me;
    try { me = npc.getMCEntity(); } catch(e) { return; }

    var tgt = npc.getAttackTarget();
    if (!tgt) return;
    var t = tgt.getMCEntity ? tgt.getMCEntity() : tgt;
    if (!t) return;

    // Init storeddata keys
    if (!npc.getStoreddata().has("sw_cd"))     setStored(npc, "sw_cd", 0);
    if (!npc.getStoreddata().has("sw_wind"))   setStored(npc, "sw_wind", 0);
    if (!npc.getStoreddata().has("sw_casting")) setStored(npc, "sw_casting", "");

    // Keep Shadowless Flight active
    if (!flightActivated || flightRefreshCd <= 0) {
        Shinobi.cast(npc, "swift_shadowless_flight", SWIFT_POWER, true);
        flightActivated = true;
        flightRefreshCd = FLIGHT_REFRESH;
    } else {
        flightRefreshCd--;
    }

    // Tick dodge cooldown
    if (dodgeCd > 0) dodgeCd--;

    // Phase update
    var phase = 0;
    try {
        var ratio = clamp(getHp(me) / getMaxHp(me), 0.0, 1.0);
        phase = ratio <= AGGRESSIVE_HP_RATIO ? 1 : 0;
    } catch(e) {}

    if (phase !== lastPhase) {
        announcePhase(npc, phase);
        lastPhase = phase;
    }

    // Rage decay
    if (rageTicks > 0) rageTicks--;

    // ---- RAGE MODE: spam Thunderous Cross ----
    if (rageTicks > 0) {
        if (rageCd > 0) { rageCd--; return; }
        if (AIM_EVERY_CAST) faceTarget(me, t);
        var ok = Shinobi.cast(npc, "swift_thunderous_cross", SWIFT_POWER, true);
        rageCd = ok ? RAGE_THUNDER_CD : RAGE_FAIL_CD;
        return;
    }

    // ---- WINDUP ----
    var wind    = getStored(npc, "sw_wind", 0);
    var casting = getStored(npc, "sw_casting", "");

    if (wind > 0) {
        if (wind % 3 === 0) {
            try {
                npc.spawnParticle("SPELL_WITCH",
                    npc.getX(), npc.getY() + 1, npc.getZ(),
                    0.3, 0.3, 0.3, 8, 0);
            } catch(e) {}
        }
        wind--;
        setStored(npc, "sw_wind", wind);

        if (wind === 0 && casting) {
            if (AIM_EVERY_CAST) faceTarget(me, t);
            var used = castPick(casting, npc, me);
            setStored(npc, "sw_casting", "");
            setStored(npc, "sw_cd", used ? BASE_COOLDOWN : 8);
        }
        return;
    }

    // ---- COOLDOWN ----
    var cd = getStored(npc, "sw_cd", 0);
    if (cd > 0) { setStored(npc, "sw_cd", cd - 1); return; }

    // ---- PICK ATTACK ----
    var weights = copyWeights(phase === 1 ? WEIGHTS_AGGRO : WEIGHTS_BASE);

    var d = distToTarget(npc, t);
    if (d > 14) {
        weights.GIAN       += 20;
        weights.VACWAVE    += 20;
        weights.THUNDEROUS += 10;
    } else if (d < 5) {
        weights.MOMENTUM   += 20;
    }

    var pick = weightedPick(weights);
    if (!pick) return;

    if (NEEDS_WINDUP[pick]) {
        setStored(npc, "sw_wind", WINDUP_TICKS);
        setStored(npc, "sw_casting", pick);
        return;
    }

    if (AIM_EVERY_CAST) faceTarget(me, t);
    var used2 = castPick(pick, npc, me);
    setStored(npc, "sw_cd", used2 ? BASE_COOLDOWN : 8);
}

// ================= EVENTS =================
function target(event) {
    var npc = event.npc;
    if (!npc) return;
    setStored(npc, "sw_cd", 0);
    try {
        var me = npc.getMCEntity();
        callJutsuSafe(ItemFuton.VACWAVE, makeFutonStack(), me, POWERS.VACWAVE);
        setStored(npc, "sw_cd", 10);
    } catch(e) {}
}

function damaged(event) {
    var npc = event.npc;
    if (!npc) return;

    // Spike -> rage
    var spiked = false;
    try {
        var src = event.source;
        if (src != null) {
            var imm = src.getImmediateSource ? src.getImmediateSource() : null;
            if (imm != null) {
                var cn = "" + imm.getClass().getName();
                if (cn.toLowerCase().indexOf("spike") >= 0)    spiked = true;
                if (cn.toLowerCase().indexOf("icespike") >= 0) spiked = true;
            }
        }
    } catch(e) {}

    if (spiked) {
        rageTicks = RAGE_TICKS;
        rageCd    = 0;
        npc.say("§e§l...You just made a mistake.");
        return;
    }

    // ---- DODGE on incoming hit ----
    if (dodgeCd <= 0 && Math.random() < DODGE_CHANCE) {
        try {
            var me = npc.getMCEntity();
            var tgt = npc.getAttackTarget();
            if (tgt != null) {
                var t = tgt.getMCEntity ? tgt.getMCEntity() : tgt;
                if (t != null) {
                    // Cancel the damage
                    event.setCanceled(true);

                    if (Math.random() < DODGE_BEHIND_CHANCE) {
                        // Appear behind player then sometimes follow up with Momentum
                        teleportBehindTarget(me, t);
                        if (Math.random() < 0.50) {
                            Shinobi.cast(npc, "swift_momentum_breaker", SWIFT_POWER, true);
                        }
                    } else {
                        // Sidestep
                        teleportSidestep(me, t);
                    }

                    dodgeCd = DODGE_COOLDOWN_MAX;
                }
            }
        } catch(e) {}
    }

    // Non-spike, non-dodge: small chance to shed cooldown
    try {
        if (event.damage > 5 && Math.random() < 0.35) {
            var cd = getStored(npc, "sw_cd", 0);
            setStored(npc, "sw_cd", Math.max(0, cd - 15));
        }
    } catch(e) {}
}

function announcePhase(npc, phase) {
    if (phase === 0) {
        npc.say("§b§oSpeed beyond sight...");
    } else {
        npc.say("§e§lYou cannot keep up with true lightning!");
    }
}