package metrics

import (
	"regexp"
	"strings"

	"github.com/czerwonk/bird_exporter/protocol"
)

// DefaultLabelStrategy defines the labels to add to an metric and its data retrieval method
type DefaultLabelStrategy struct {
	descriptionLabels      bool
	descriptionLabelsRegex *regexp.Regexp
}

func NewDefaultLabelStrategy(descriptionLabels bool, descriptionLabelsRegex string) *DefaultLabelStrategy {
	return &DefaultLabelStrategy{
		descriptionLabels:      descriptionLabels,
		descriptionLabelsRegex: regexp.MustCompile(descriptionLabelsRegex),
	}
}

// LabelNames returns the list of label names
func (d *DefaultLabelStrategy) LabelNames(p *protocol.Protocol) []string {
	res := []string{"name", "proto", "ip_version", "import_filter", "export_filter"}
	if d.descriptionLabels && p.Description != "" {
		res = append(res, labelKeysFromDescription(p.Description, d)...)
	}

	return res
}

// LabelValues returns the values for a protocol
func (d *DefaultLabelStrategy) LabelValues(p *protocol.Protocol) []string {
	res := []string{p.Name, protoString(p), p.IPVersion, p.ImportFilter, p.ExportFilter}
	if d.descriptionLabels && p.Description != "" {
		res = append(res, labelValuesFromDescription(p.Description, d)...)
	}

	return res
}

func labelKeysFromDescription(desc string, d *DefaultLabelStrategy) []string {
	res := []string{}

	matches := d.descriptionLabelsRegex.FindAllStringSubmatch(desc, -1)
	for _, submatch := range matches {
		res = append(res, strings.TrimSpace(submatch[1]))
	}

	return res
}

func labelValuesFromDescription(desc string, d *DefaultLabelStrategy) []string {
	res := []string{}

	matches := d.descriptionLabelsRegex.FindAllStringSubmatch(desc, -1)
	for _, submatch := range matches {
		res = append(res, strings.TrimSpace(submatch[2]))
	}

	return res
}

func protoString(p *protocol.Protocol) string {
	switch p.Proto {
	case protocol.BGP:
		return "BGP"
	case protocol.OSPF:
		if p.IPVersion == "4" {
			return "OSPF"
		}
		return "OSPFv3"
	case protocol.Static:
		return "Static"
	case protocol.Kernel:
		return "Kernel"
	case protocol.Direct:
		return "Direct"
	case protocol.Babel:
		return "Babel"
	case protocol.RPKI:
		return "RPKI"
	case protocol.BFD:
		return "BFD"
	}

	return ""
}