VictoriaMetrics/lib/promscrape/discovery/vultr/api.go

package vultr

import (
	"fmt"
	"net/url"

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

// apiConfig contains config for API server.
type apiConfig struct {
	c    *discoveryutils.Client
	port int

	listQueryParams string
}

// getAPIConfig get or create API config from configMap.
func getAPIConfig(sdc *SDConfig, baseDir string) (*apiConfig, error) {
	v, err := configMap.Get(sdc, func() (any, error) { return newAPIConfig(sdc, baseDir) })
	if err != nil {
		return nil, err
	}
	return v.(*apiConfig), nil
}

// newAPIConfig create API Config.
func newAPIConfig(sdc *SDConfig, baseDir string) (*apiConfig, error) {
	port := sdc.Port
	if port == 0 {
		port = 80
	}

	// See: https://www.vultr.com/api/
	apiServer := "https://api.vultr.com"

	if sdc.HTTPClientConfig.BearerToken == nil {
		return nil, fmt.Errorf("missing `bearer_token` option")
	}

	ac, err := sdc.HTTPClientConfig.NewConfig(baseDir)
	if err != nil {
		return nil, fmt.Errorf("cannot parse auth config: %w", err)
	}
	proxyAC, err := sdc.ProxyClientConfig.NewConfig(baseDir)
	if err != nil {
		return nil, fmt.Errorf("cannot parse proxy auth config: %w", err)
	}

	c, err := discoveryutils.NewClient(apiServer, ac, sdc.ProxyURL, proxyAC, &sdc.HTTPClientConfig)
	if err != nil {
		return nil, fmt.Errorf("cannot create client for %q: %w", apiServer, err)
	}

	// Prepare additional query params for list instance API.
	// See https://www.vultr.com/api/#tag/instances/operation/list-instances
	var qp url.Values
	if sdc.Label != "" {
		qp.Set("label", sdc.Label)
	}
	if sdc.MainIP != "" {
		qp.Set("main_ip", sdc.MainIP)
	}
	if sdc.Region != "" {
		qp.Set("region", sdc.Region)
	}
	if sdc.FirewallGroupID != "" {
		qp.Set("firewall_group_id", sdc.FirewallGroupID)
	}
	if sdc.Hostname != "" {
		qp.Set("hostname", sdc.Hostname)
	}

	cfg := &apiConfig{
		c:    c,
		port: port,

		listQueryParams: qp.Encode(),
	}
	return cfg, nil
}