package fetcher

import (
	"math"

	"github.com/hetznercloud/hcloud-go/v2/hcloud"
)

var _ Fetcher = &serverTraffic{}

// NewServerTraffic creates a new fetcher that will collect pricing information on server traffic.
func NewServerTraffic(pricing *PriceProvider, additionalLabels ...string) Fetcher {
	return &serverTraffic{newBase(pricing, "server_traffic", []string{"location", "type"}, additionalLabels...)}
}

type serverTraffic struct {
	*baseFetcher
}

func (serverTraffic serverTraffic) Run(client *hcloud.Client) error {
	servers, _, err := client.Server.List(ctx, hcloud.ServerListOpts{})
	if err != nil {
		return err
	}

	for _, s := range servers {
		location := s.Datacenter.Location

		labels := append([]string{
			s.Name,
			location.Name,
			s.ServerType.Name,
		},
			parseAdditionalLabels(serverTraffic.additionalLabels, s.Labels)...,
		)

		additionalTraffic := int(s.OutgoingTraffic) - int(s.IncludedTraffic)
		if additionalTraffic < 0 {
			serverTraffic.hourly.WithLabelValues(labels...).Set(0)
			serverTraffic.monthly.WithLabelValues(labels...).Set(0)
			break
		}

		monthlyPrice := math.Ceil(float64(additionalTraffic)/sizeTB) * serverTraffic.pricing.Traffic()
		hourlyPrice := pricingPerHour(monthlyPrice)

		serverTraffic.hourly.WithLabelValues(labels...).Set(hourlyPrice)
		serverTraffic.monthly.WithLabelValues(labels...).Set(monthlyPrice)
	}

	return nil
}