package data.scripts.plugins;

import com.fs.starfarer.api.Global;
import com.fs.starfarer.api.combat.CombatEngineAPI;
import com.fs.starfarer.api.combat.CombatEntityAPI;
import com.fs.starfarer.api.combat.DamageType;
import com.fs.starfarer.api.combat.DamagingProjectileAPI;
import com.fs.starfarer.api.combat.EveryFrameCombatPlugin;
import com.fs.starfarer.api.combat.ShipAPI;
import com.fs.starfarer.api.input.InputEventAPI;

import java.util.*;
import org.lwjgl.util.vector.Vector2f;

import org.dark.shaders.distortion.DistortionShader;
import org.dark.shaders.distortion.RippleDistortion;

public class ShieldRipple implements EveryFrameCombatPlugin {

    public CombatEngineAPI engine;
    public static final Set<DamagingProjectileAPI> projectiles = new LinkedHashSet<>();

    @Override
    public void init(CombatEngineAPI engine) {
    }

    @Override
    public void advance(float amount, List<InputEventAPI> events) {
        if (engine != Global.getCombatEngine()) {
            projectiles.clear();
            this.engine = Global.getCombatEngine();
        }

        if (engine.isPaused()) {
            return;
        }

        List<DamagingProjectileAPI> activeProjectiles = engine.getProjectiles();
        for (DamagingProjectileAPI projectile : activeProjectiles) {
            if (projectile.didDamage() || projectile.isFading()) {
                continue;
            }

            if (!projectiles.contains(projectile)) {
                projectiles.add(projectile);
            }
        }

        Iterator<DamagingProjectileAPI> iter = projectiles.iterator();
        while (iter.hasNext()) {
            DamagingProjectileAPI projectile = iter.next();
            if (projectile.didDamage()) {
                CombatEntityAPI target = projectile.getDamageTarget();

                if (target instanceof ShipAPI) {
                    if (target.getShield() != null && target.getShield().isOn() && target.getShield().isWithinArc(Vector2f.add(projectile.getLocation(), (Vector2f) projectile.getVelocity().scale(amount), null))) {
                        onHit(projectile.getLocation(), projectile.getDamageAmount(), projectile.getDamageType());
                    }
                }

                iter.remove();
            } else if (!engine.isEntityInPlay(projectile)) {
                iter.remove();
            }
        }
    }

    private void onHit(Vector2f location, float damage, DamageType type) {
        if (type == DamageType.FRAGMENTATION) {
            damage *= 0.25f;
        }
        if (type == DamageType.HIGH_EXPLOSIVE) {
            damage *= 0.5f;
        }
        if (type == DamageType.KINETIC) {
            damage *= 2f;
        }

        if (damage < 100f) {
            return;
        }

        RippleDistortion ripple = new RippleDistortion();
        float fadeTime = (float) Math.sqrt(Math.sqrt(damage)) * 0.1f;
        float intensity = (float) Math.sqrt(damage) * 5f;
        ripple.setSize(intensity);
        ripple.setIntensity(intensity * 0.2f);
        ripple.setLocation(location);
        ripple.setFrameRate(60f / fadeTime);
        ripple.fadeInSize(fadeTime * 1.2f);
        ripple.fadeOutIntensity(fadeTime);
        ripple.setSize(intensity * 0.2f);
        DistortionShader.addDistortion(ripple);
    }
}