package metrics

import (
	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promauto"
	"lostak.dev/pve-exporter/proxmox"
)

// PVE SDN state collector.
type PveSdnCollector struct {
	apiClient *proxmox.PveApiClient // PVE API client instance.

	state *prometheus.GaugeVec // SDN state prometheus gauge.
}

// Create new instance of PVE SDN collector.
func NewPveSdnCollector(apiClient *proxmox.PveApiClient) *PveSdnCollector {
	c := PveSdnCollector{apiClient: apiClient}

	// SDN Up state.
	c.state = promauto.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "pve_sdn_state",
			Help: "Node software defined network state.",
		},
		[]string{"cluster", "node", "sdn", "sdn_id"},
	)

	return &c
}

// PveMetricsCollector interface implementation.
func (c *PveSdnCollector) CollectMetrics() error {
	cluster, err := c.apiClient.GetClusterStatus()
	if err != nil {
		return err
	}

	resources, err := c.apiClient.GetClusterResources()
	if err != nil {
		return err
	}

	c.state.Reset()

	for _, node := range cluster.NodeStatuses {
		sdns := resources.FindNodeSDN(node.Name)
		if len(*sdns) > 0 {
			for _, sdn := range *sdns {
				labels := prometheus.Labels{
					"cluster": cluster.GetClusterName(),
					"node":    node.Name,
					"sdn":     sdn.SDN,
					"sdn_id":  sdn.ID,
				}

				c.state.With(labels).Set(sdn.GetStatusNumeric())
			}
		}
	}
	return nil
}

// PveMetricsCollector interface implementation.
func (c *PveSdnCollector) GetName() string {
	return "SDN"
}