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

69 lines
1.7 KiB
Go
Raw Normal View History

package hetzner
import (
"fmt"
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape/discoveryutils"
)
var configMap = discoveryutils.NewConfigMap()
type apiConfig struct {
client *discoveryutils.Client
role string
port int
}
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
}
func newAPIConfig(sdc *SDConfig, baseDir string) (*apiConfig, error) {
hcc := sdc.HTTPClientConfig
var apiServer string
switch sdc.Role {
case "robot":
// See https://robot.hetzner.com/doc/webservice/en.html
apiServer = "https://robot-ws.your-server.de"
if hcc.BasicAuth == nil {
return nil, fmt.Errorf("basic_auth must be set when role is `robot`")
}
case "hcloud":
// See https://docs.hetzner.cloud/
apiServer = "https://api.hetzner.cloud"
if hcc.Authorization == nil {
return nil, fmt.Errorf("authorization must be set when role is `hcloud`")
}
default:
return nil, fmt.Errorf("unexpected role=%q; must be one of `robot` or `hcloud`", sdc.Role)
}
ac, err := hcc.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)
}
client, err := discoveryutils.NewClient(apiServer, ac, sdc.ProxyURL, proxyAC, &sdc.HTTPClientConfig)
if err != nil {
return nil, fmt.Errorf("cannot create HTTP client for %q: %w", apiServer, err)
}
port := 80
if sdc.Port != nil {
port = *sdc.Port
}
cfg := &apiConfig{
client: client,
role: sdc.Role,
port: port,
}
return cfg, nil
}