package checker

import (
	"encoding/json"
	"time"

	sdk "git.happydns.org/checker-sdk-go/checker"
)

func Provider() sdk.ObservationProvider { return &blacklistProvider{} }

type blacklistProvider struct{}

func (p *blacklistProvider) Key() sdk.ObservationKey      { return ObservationKeyBlacklist }
func (p *blacklistProvider) Definition() *sdk.CheckerDefinition { return Definition() }

// ExtractMetrics turns the (now uniform) per-source results into a
// small set of generic gauges. Source-specific metrics (VT engine
// counts, URLhaus URL count, …) live in SourceResult.Details and are
// rendered in the HTML report; metrics here stay coarse so the
// scheduler / Prometheus side does not have to know which sources are
// installed.
func (p *blacklistProvider) ExtractMetrics(ctx sdk.ReportContext, collectedAt time.Time) ([]sdk.CheckMetric, error) {
	var data BlacklistData
	if err := json.Unmarshal(ctx.Data(), &data); err != nil {
		return nil, err
	}

	metrics := []sdk.CheckMetric{
		{
			Name: "blacklist_total_hits", Value: float64(data.TotalHits()),
			Unit: "results", Timestamp: collectedAt,
		},
	}
	for _, r := range data.Results {
		if !r.Enabled {
			continue
		}
		v := 0.0
		if r.Listed {
			v = 1
		}
		metrics = append(metrics, sdk.CheckMetric{
			Name:      "blacklist_source_listed",
			Value:     v,
			Unit:      "bool",
			Labels:    map[string]string{"source": r.SourceID, "subject": r.Subject},
			Timestamp: collectedAt,
		})
	}
	return metrics, nil
}