package docker

import (
	"encoding/json"
	"fmt"
	"strconv"

	"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape/discoveryutils"
	"github.com/VictoriaMetrics/VictoriaMetrics/lib/promutils"
)

// See https://docs.docker.com/engine/api/v1.40/#tag/Network
type network struct {
	ID       string
	Name     string
	Scope    string
	Internal bool
	Ingress  bool
	Labels   map[string]string
}

func getNetworksLabelsByNetworkID(cfg *apiConfig) (map[string]*promutils.Labels, error) {
	networks, err := getNetworks(cfg)
	if err != nil {
		return nil, err
	}
	return getNetworkLabelsByNetworkID(networks), nil
}

func getNetworks(cfg *apiConfig) ([]network, error) {
	resp, err := cfg.getAPIResponse("/networks")
	if err != nil {
		return nil, fmt.Errorf("cannot query dockerswarm api for networks: %w", err)
	}
	return parseNetworks(resp)
}

func parseNetworks(data []byte) ([]network, error) {
	var networks []network
	if err := json.Unmarshal(data, &networks); err != nil {
		return nil, fmt.Errorf("cannot parse networks: %w", err)
	}
	return networks, nil
}

func getNetworkLabelsByNetworkID(networks []network) map[string]*promutils.Labels {
	ms := make(map[string]*promutils.Labels)
	for _, network := range networks {
		m := promutils.NewLabels(8)
		m.Add("__meta_docker_network_id", network.ID)
		m.Add("__meta_docker_network_name", network.Name)
		m.Add("__meta_docker_network_internal", strconv.FormatBool(network.Internal))
		m.Add("__meta_docker_network_ingress", strconv.FormatBool(network.Ingress))
		m.Add("__meta_docker_network_scope", network.Scope)
		for k, v := range network.Labels {
			m.Add(discoveryutils.SanitizeLabelName("__meta_docker_network_label_"+k), v)
		}
		ms[network.ID] = m
	}
	return ms
}