package data.scripts.weapons;

import com.fs.starfarer.api.Global;
import com.fs.starfarer.api.combat.CombatEngineAPI;
import com.fs.starfarer.api.combat.DamageType;
import com.fs.starfarer.api.combat.EveryFrameWeaponEffectPlugin;
import com.fs.starfarer.api.combat.ShipAPI;
import com.fs.starfarer.api.combat.WeaponAPI;

import java.awt.Color;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import org.lwjgl.util.vector.Vector2f;

import org.dark.shaders.light.LightShader;
import org.dark.shaders.light.StandardLight;
import org.lazywizard.lazylib.MathUtils;
import org.lazywizard.lazylib.VectorUtils;
import org.lazywizard.lazylib.combat.CombatUtils;
import org.lazywizard.lazylib.combat.entities.SimpleEntity;

public class EMPBomb implements EveryFrameWeaponEffectPlugin {

    private boolean used = false;
    private CombatEngineAPI currentEngine;
    private static final List<MegaArc> arcs = new ArrayList<>();
    private static final List<MegaArc> newarcs = new ArrayList<>();

    private static final Map<String, Float> bossShips = new HashMap<>();

    static {
        bossShips.put("ssp_superhyperion", 4f);
        bossShips.put("ssp_oberon", 2f);
        bossShips.put("ssp_ultron", 3f);
        bossShips.put("ssp_zeus", 2f);
        bossShips.put("ssp_ezekiel", 4f);
        bossShips.put("ssp_cristarium", 2f);
        bossShips.put("ssp_zero", 4f);
    }

    @Override
    public void advance(float amount, CombatEngineAPI engine, WeaponAPI weapon) {
        if (engine != currentEngine) {
            arcs.clear();
            newarcs.clear();
            currentEngine = engine;
            return;
        }

        if (weapon.getChargeLevel() > 0f & !used) {
            Vector2f point = MathUtils.getRandomPointOnCircumference(weapon.getShip().getLocation(), weapon.getShip().getCollisionRadius() * 5f);
            Vector2f vel = VectorUtils.getDirectionalVector(point, weapon.getShip().getLocation());
            vel.scale((float) Math.random() * 125f + 150f);
            engine.addSmoothParticle(point, vel, (float) Math.random() * 10f + 10f, 0.5f, 1f, new Color(255, 255, 255));
            Global.getSoundPlayer().playLoop("high_intensity_laser_loop", weapon.getShip(), 1f + weapon.getChargeLevel() * 4f, 3f, weapon.getShip().getLocation(), weapon.getShip().getVelocity());
        }

        if (weapon.getCooldownRemaining() > 0f) {
            if (!used) {
                used = true;

                StandardLight light = new StandardLight();
                light.setLocation(weapon.getShip().getLocation());
                light.setIntensity(3f);
                light.setSize(3500f);
                light.setColor(new Color(255, 255, 255));
                light.fadeOut(2f);
                LightShader.addLight(light);

                for (int i = 0; i < 40; i++) {
                    Vector2f direction = new Vector2f(1f, 0f);
                    VectorUtils.rotate(direction, (float) Math.random() * 360f, direction);
                    arcs.add(new MegaArc(weapon.getShip(), weapon.getShip().getLocation(), direction, (float) Math.random() * 1250f + 1250f, (float) Math.random() * 20f + 15f, (float) Math.random() * 1.5f + 0.5f));
                }

                engine.addSmoothParticle(weapon.getShip().getLocation(), new Vector2f(), 1250f, 2f, 2f, new Color(255, 255, 255));
                engine.addHitParticle(weapon.getShip().getLocation(), new Vector2f(), 750f, 2f, 2f, new Color(255, 255, 255));
                Global.getSoundPlayer().playSound("vortex_blast", 1f, 1f, weapon.getShip().getLocation(), new Vector2f());
                Global.getSoundPlayer().playSound("shieldbreaker_blast", 1f, 0.75f, weapon.getShip().getLocation(), new Vector2f());

                List<ShipAPI> ships = CombatUtils.getShipsWithinRange(weapon.getShip().getLocation(), 3000f);
                for (ShipAPI ship : ships) {
                    if (ship != weapon.getShip() && !bossShips.containsKey(ship.getHullSpec().getHullId())) {
                        ship.getFluxTracker().forceOverload(10f + (3000f - MathUtils.getDistance(weapon.getShip(), ship)) / 250f);
                    }
                }
            }
        }

        if (weapon.getCooldownRemaining() <= 0f) {
            if (used) {
                used = false;
            }
        }

        Iterator<MegaArc> iter = arcs.iterator();
        while (iter.hasNext()) {
            MegaArc arc = iter.next();
            if (Math.random() >= 0.33f) {
                if (arc.advance(amount * 3f)) {
                    iter.remove();
                }
            }
        }

        iter = newarcs.iterator();
        while (iter.hasNext()) {
            MegaArc arc = iter.next();
            arcs.add(arc);
            iter.remove();
        }
    }

    private static final class MegaArc {

        private final ShipAPI source;
        private final Vector2f location = new Vector2f();
        private final Vector2f direction = new Vector2f();
        private final float speed;
        private final float thickness;
        private float remaining;

        private MegaArc(ShipAPI source, Vector2f location, Vector2f direction, float speed, float thickness, float lifetime) {
            this.source = source;
            this.location.set(location);
            this.direction.set(direction);
            this.speed = speed;
            this.thickness = thickness;
            this.remaining = lifetime;
        }

        private boolean advance(float amount) {
            Vector2f newLocation = new Vector2f(direction);
            newLocation.scale(speed * amount);
            Vector2f.add(newLocation, location, newLocation);
            Global.getCombatEngine().spawnEmpArc(source, location, new SimpleEntity(newLocation), new SimpleEntity(newLocation), DamageType.ENERGY, 0f, 0f, 10000f, null, thickness, new Color(200, 225, 255, 150), new Color(255, 255, 255, 255));
            location.set(newLocation);
            VectorUtils.rotate(direction, (float) Math.random() * 60f - 30f, direction);

            if (Math.random() > 0.95) {
                Global.getSoundPlayer().playSound("vortex_absorb", 1f, 0.5f, location, new Vector2f());
                Vector2f newDirection = new Vector2f(1f, 0f);
                VectorUtils.rotate(newDirection, (float) Math.random() * 360f, newDirection);
                newarcs.add(new MegaArc(source, location, newDirection, speed + (float) Math.random() * 200f - 100f, Math.max(thickness - (float) Math.random() * 5f, 2.5f), Math.max(remaining + (float) Math.random() * 0.3f - 0.15f, 0.01f)));
            }

            remaining -= amount;
            return remaining <= 0f;
        }
    }
}