package pigcart.cosycritters.particle;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.Minecraft;
import net.minecraft.client.multiplayer.ClientLevel;
import net.minecraft.client.particle.*;
import net.minecraft.core.particles.SimpleParticleType;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.util.Mth;
import net.minecraft.world.phys.Vec3;
import pigcart.cosycritters.CosyCritters;

public class BirdParticle extends TextureSheetParticle {

    int spawnAnimationLength = 40;
    int spawnAnimationTime = spawnAnimationLength;
    Vec3 spawnAnimationStart;
    Vec3 spawnAnimationEnd;
    boolean spawnAnimation = true;
    boolean flyUpAwayToTheSun = false;
    Vec3 facing;

    private BirdParticle(ClientLevel level, double x, double y, double z, double landAtX, double landAtY, double landAtZ) {
        super(level, x, y, z);
        this.setSprite(Minecraft.getInstance().particleEngine.textureAtlas.getSprite(ResourceLocation.fromNamespaceAndPath(CosyCritters.MOD_ID, random.nextBoolean() ? "crow_flying_left" : "crow_flying_right")));
        this.quadSize = 0;
        this.lifetime = 6000;
        this.facing = new Vec3((this.random.nextFloat() - 0.5), this.random.nextFloat(), (this.random.nextFloat() - 0.5)).normalize().multiply(0.5, 0.5, 0.5);
        this.spawnAnimationStart = new Vec3(x, y, z);
        this.spawnAnimationEnd = new Vec3(landAtX, landAtY, landAtZ);
        CosyCritters.birdCount++;
    }

    @Override
    public void remove() {
        if (!this.removed) {
            CosyCritters.birdCount = Math.max(0, CosyCritters.birdCount - 1);
        }
        super.remove();
    }

    @Override
    public void tick() {
        super.tick();
        if (spawnAnimation) {
            if (spawnAnimationTime != 0) {
                spawnAnimationTime--;
                this.x = Mth.lerp((double) spawnAnimationTime / spawnAnimationLength, spawnAnimationEnd.x, spawnAnimationStart.x);
                this.y = Mth.lerp((double) spawnAnimationTime / spawnAnimationLength, spawnAnimationEnd.y, spawnAnimationStart.y);
                this.z = Mth.lerp((double) spawnAnimationTime / spawnAnimationLength, spawnAnimationEnd.z, spawnAnimationStart.z);
                this.quadSize = Mth.lerp((float) spawnAnimationTime / spawnAnimationLength, 0.5F, 0);
            } else {
                spawnAnimation = false;
                this.setSprite(Minecraft.getInstance().particleEngine.textureAtlas.getSprite(ResourceLocation.fromNamespaceAndPath(CosyCritters.MOD_ID, random.nextBoolean() ? "crow_left" : "crow_right")));
            }
        }
        else if (flyUpAwayToTheSun) {
            this.quadSize = Mth.lerp((float) age / lifetime, 0.5F, 0);
            this.xd = this.facing.x;
            this.yd = this.facing.y;
            this.zd = this.facing.z;
        }
        else if (this.age % 20 == 0) {
            Vec3 birdPos = new Vec3(this.x, this.y, this.z);
            if (Minecraft.getInstance().cameraEntity.position().distanceTo(birdPos) < 10 && !flyUpAwayToTheSun) {
                flyUpAwayToTheSun = true;
                this.setSprite(Minecraft.getInstance().particleEngine.textureAtlas.getSprite(ResourceLocation.fromNamespaceAndPath(CosyCritters.MOD_ID, random.nextBoolean() ? "crow_flying_left" : "crow_flying_right")));
                this.lifetime = 100;
                this.age = 0;
                // for some reason updating the velocity after this sends the bird to its spawn position????
                this.setPos(spawnAnimationEnd.x, spawnAnimationEnd.y, spawnAnimationEnd.z);
            }
            else if (!Minecraft.getInstance().cameraEntity.position().closerThan(new Vec3(x, y, z), 64)) {
                this.remove();
            }
        }
    }

    @Override
    public ParticleRenderType getRenderType() {
        return ParticleRenderType.PARTICLE_SHEET_OPAQUE;
    }

    @Environment(EnvType.CLIENT)
    public static class Provider implements ParticleProvider<SimpleParticleType> {

        public Provider(SpriteSet spriteSet) {
        }

        @Override
        public Particle createParticle(SimpleParticleType parameters, ClientLevel level, double x, double y, double z, double landAtX, double landAtY, double landAtZ) {
            return new BirdParticle(level, x, y, z, landAtX, landAtY, landAtZ);
        }
    }
}