last_light/game/ui/ui_health_bar.go

package ui

import (
	"fmt"
	"math"
	"mvvasilev/last_light/engine"
	"mvvasilev/last_light/game/model"
	"mvvasilev/last_light/game/systems"

	"github.com/gdamore/tcell/v2"
	"github.com/gdamore/tcell/v2/views"
	"github.com/google/uuid"
)

type UIHealthBar struct {
	id     uuid.UUID
	player *model.Player

	window *UIWindow

	style tcell.Style
}

func CreateHealthBar(x, y, w, h int, player *model.Player, style tcell.Style) *UIHealthBar {
	return &UIHealthBar{
		window: CreateWindow(x, y, w, h, "HP", style),
		player: player,
		style:  style,
	}
}

func (uihp *UIHealthBar) MoveTo(x int, y int) {
	uihp.window.MoveTo(x, y)
}

func (uihp *UIHealthBar) Position() engine.Position {
	return uihp.window.Position()
}

func (uihp *UIHealthBar) Size() engine.Size {
	return uihp.window.Size()
}

func (uihp *UIHealthBar) Input(inputAction systems.InputAction) {
}

func (uihp *UIHealthBar) UniqueId() uuid.UUID {
	return uihp.id
}

func (uihp *UIHealthBar) Draw(v views.View) {
	x, y, w, h := uihp.Position().X(), uihp.Position().Y(), uihp.Size().Width(), uihp.Size().Height()

	uihp.window.Draw(v)

	stages := []string{"█", "▓", "▒", "░"} // 0 = 1.0, 1 = 0.75, 2 = 0.5, 3 = 0.25

	percentage := (float64(w) - 2.0) * (float64(uihp.player.HealthData().Health) / float64(uihp.player.HealthData().MaxHealth))

	whole := math.Trunc(percentage)
	last := percentage - whole

	hpBar := ""
	hpStyle := tcell.StyleDefault.Foreground(tcell.ColorIndianRed)

	for range int(whole) {
		hpBar += stages[0]
	}

	lastRune := func() string {
		if last <= 0.0 {
			return ""
		}

		if last <= 0.25 {
			return stages[3]
		}

		if last <= 0.50 {
			return stages[2]
		}

		if last <= 0.75 {
			return stages[1]
		}

		if last <= 1.00 {
			return stages[0]
		}

		return ""
	}

	hpBar += lastRune()

	engine.DrawText(x+1, y+1, hpBar, hpStyle, v)

	hpText := fmt.Sprintf("%v/%v", uihp.player.HealthData().Health, uihp.player.HealthData().MaxHealth)

	engine.DrawText(
		x+w/2-len(hpText)/2,
		y+h-1,
		hpText,
		hpStyle,
		v,
	)
}