package system

import (
	"code.rocketnine.space/tslocum/citylimits/component"
	"code.rocketnine.space/tslocum/citylimits/world"
	"code.rocketnine.space/tslocum/gohan"
	"github.com/hajimehoshi/ebiten/v2"
)

type TaxSystem struct {
}

func NewTaxSystem() *TaxSystem {
	s := &TaxSystem{}

	return s
}

func (s *TaxSystem) Needs() []gohan.ComponentID {
	return []gohan.ComponentID{
		component.PositionComponentID,
		component.VelocityComponentID,
		component.WeaponComponentID,
	}
}

func (s *TaxSystem) Uses() []gohan.ComponentID {
	return nil
}

func (s *TaxSystem) Update(_ *gohan.Context) error {
	if world.World.Paused {
		return nil
	}

	if world.World.Ticks%world.YearTicks != 0 {
		return nil
	}

	taxCollectionAmount := 777.77
	for _, zone := range world.World.Zones {
		if zone.Population == 0 {
			continue
		}

		taxRate := world.World.TaxR
		if zone.Type == world.StructureCommercialZone {
			taxRate = world.World.TaxC
		} else if zone.Type == world.StructureIndustrialZone {
			taxRate = world.World.TaxI
		}

		world.World.Funds += int(taxCollectionAmount * taxRate)
	}
	return nil
}

func (s *TaxSystem) Draw(_ *gohan.Context, _ *ebiten.Image) error {
	return gohan.ErrSystemWithoutDraw
}