package informatis.draws;

import arc.graphics.Color;
import arc.graphics.g2d.Draw;
import arc.graphics.g2d.Fill;
import mindustry.Vars;
import mindustry.gen.Building;
import mindustry.graphics.Pal;
import mindustry.world.Tile;
import mindustry.world.blocks.ConstructBlock;
import mindustry.world.blocks.defense.turrets.Turret;
import mindustry.world.blocks.units.Reconstructor;
import mindustry.world.blocks.units.UnitFactory;

import static arc.Core.settings;
import static informatis.SUtils.isInCamera;

public class BlockBarDraw extends OverDraw {
    public BlockBarDraw() {
        super("blockBar", OverDrawCategory.Block);
    }

    @Override
    public void onTile(Tile tile) {
        if(!isInCamera(tile.worldx(), tile.worldy(), 8) || tile.build == null) return;

        Building b = tile.build;

        if(settings.getBool("blockBar")) {
            drawBar(b, 0, -(b.block.size * 4 - 2), b.healthf(), Pal.health);

            if (b instanceof Turret.TurretBuild turretBuild) {
                drawBar(b, 0, b.block.size * 4 - 2, turretBuild.reloadCounter / ((Turret) b.block).reload, Pal.ammo);
            }
            if(b instanceof ConstructBlock.ConstructBuild constructBuild)
                drawBar(b, 0, b.block.size * 4 - 2, constructBuild.progress(), b.team.color);
            if(b instanceof Reconstructor.ReconstructorBuild reconstructorBuild)
                drawBar(b, 0, b.block.size * 4 - 2, reconstructorBuild.fraction(), b.team.color);
            if(b instanceof UnitFactory.UnitFactoryBuild factoryBuild)
                drawBar(b, 0, b.block.size * 4 - 2, factoryBuild.fraction(), b.team.color);
        }
    }

    void drawBar(Building b, float offsetX, float offsetY, float progress, Color color) {
        float bx = b.x + offsetX, by = b.y + offsetY;
        float width = b.block.size * 7.5f, height = 2;
        Draw.color(Pal.gray);
        Fill.quad(
                bx - width/2, by + height/2,
                bx - width/2, by - height/2,
                bx + width/2, by - height/2,
                bx + width/2, by + height/2);
        Draw.color(color);
        width = b.block.size * 7.5f - 0.5f; height = 2 - 0.5f;
        Fill.quad(
                bx - width/2, by + height/2,
                bx - width/2, by - height/2,
                bx - width/2 + width * progress, by - height/2,
                bx - width/2 + width * progress, by + height/2);
    }
}