GoDoxy/internal/proxmox/config.go

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" yaml:"url" validate:"required,url"`

	TokenID string `json:"token_id" yaml:"token_id" validate:"required"`
	Secret  string `json:"secret" yaml:"token_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
}