merge: main branch

internal/proxmox/config.go

@@ -0,0 +1,69 @@
+package proxmox
+
+import (
+	"context"
+	"crypto/tls"
+	"errors"
+	"net/http"
+	"strings"
+	"time"
+
+	"github.com/luthermonson/go-proxmox"
+	"github.com/yusing/go-proxy/internal/gperr"
+	"github.com/yusing/go-proxy/internal/net/gphttp"
+)
+
+type Config struct {
+	URL string `json:"url" validate:"required,url"`
+
+	TokenID string `json:"token_id" validate:"required"`
+	Secret  string `json:"secret" validate:"required"`
+
+	NoTLSVerify bool `json:"no_tls_verify" yaml:"no_tls_verify,omitempty"`
+
+	client *Client
+}
+
+func (c *Config) Client() *Client {
+	if c.client == nil {
+		panic("proxmox client accessed before init")
+	}
+	return c.client
+}
+
+func (c *Config) Init() gperr.Error {
+	var tr *http.Transport
+	if c.NoTLSVerify {
+		tr = gphttp.NewTransportWithTLSConfig(&tls.Config{
+			InsecureSkipVerify: true,
+		})
+	} else {
+		tr = gphttp.NewTransport()
+	}
+
+	if strings.HasSuffix(c.URL, "/") {
+		c.URL = c.URL[:len(c.URL)-1]
+	}
+	if !strings.HasSuffix(c.URL, "/api2/json") {
+		c.URL += "/api2/json"
+	}
+
+	opts := []proxmox.Option{
+		proxmox.WithAPIToken(c.TokenID, c.Secret),
+		proxmox.WithHTTPClient(&http.Client{
+			Transport: tr,
+		}),
+	}
+	c.client = NewClient(c.URL, opts...)
+
+	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
+	defer cancel()
+
+	if err := c.client.UpdateClusterInfo(ctx); err != nil {
+		if errors.Is(err, context.DeadlineExceeded) {
+			return gperr.New("timeout fetching proxmox cluster info")
+		}
+		return gperr.New("failed to fetch proxmox cluster info").With(err)
+	}
+	return nil
+}