feat: initial implementation of container management platform

2026-06-03 20:12:58 +00:00 · 2026-02-16 10:18:05 +01:00
commit ffa5489dc1
167 changed files with 55910 additions and 0 deletions
@@ -0,0 +1,334 @@
+package proxmox
+
+import (
+	"crypto/tls"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+)
+
+// Client represents a Proxmox API client
+type Client struct {
+	baseURL    string
+	username   string
+	password   string
+	tokenID    string
+	token      string
+	httpClient *http.Client
+	ticket     string
+	csrfToken  string
+}
+
+// NewClient creates a new Proxmox API client
+func NewClient(baseURL, username, password string) *Client {
+	return &Client{
+		baseURL:  strings.TrimSuffix(baseURL, "/"),
+		username: username,
+		password: password,
+		httpClient: &http.Client{
+			Timeout: 30 * time.Second,
+			Transport: &http.Transport{
+				TLSClientConfig: &tls.Config{
+					InsecureSkipVerify: true, // Proxmox typically uses self-signed certs
+				},
+			},
+		},
+	}
+}
+
+// NewClientWithToken creates a new Proxmox API client using API token
+func NewClientWithToken(baseURL, tokenID, token string) *Client {
+	return &Client{
+		baseURL:  strings.TrimSuffix(baseURL, "/"),
+		tokenID:  tokenID,
+		token:    token,
+		httpClient: &http.Client{
+			Timeout: 30 * time.Second,
+			Transport: &http.Transport{
+				TLSClientConfig: &tls.Config{
+					InsecureSkipVerify: true,
+				},
+			},
+		},
+	}
+}
+
+// Login authenticates with Proxmox and stores session tokens
+func (c *Client) Login() error {
+	data := url.Values{}
+	data.Set("username", c.username)
+	data.Set("password", c.password)
+	
+	resp, err := c.httpClient.PostForm(c.baseURL+"/api2/json/access/ticket", data)
+	if err != nil {
+		return fmt.Errorf("failed to login: %w", err)
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("login failed with status: %s", resp.Status)
+	}
+
+	var result struct {
+		Data struct {
+			Ticket     string `json:"ticket"`
+			CSRFPreventionToken string `json:"CSRFPreventionToken"`
+			Username   string `json:"username"`
+		} `json:"data"`
+	}
+
+	if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
+		return fmt.Errorf("failed to decode login response: %w", err)
+	}
+
+	c.ticket = result.Data.Ticket
+	c.csrfToken = result.Data.CSRFPreventionToken
+
+	return nil
+}
+
+// makeRequest makes an authenticated request to Proxmox API
+func (c *Client) makeRequest(method, path string, body io.Reader) (*http.Response, error) {
+	req, err := http.NewRequest(method, c.baseURL+path, body)
+	if err != nil {
+		return nil, err
+	}
+
+	// Use token authentication if available
+	if c.tokenID != "" && c.token != "" {
+		req.Header.Set("Authorization", "PVEAPIToken="+c.tokenID+"="+c.token)
+	} else {
+		// Use session ticket authentication
+		if c.ticket == "" {
+			if err := c.Login(); err != nil {
+				return nil, fmt.Errorf("failed to authenticate: %w", err)
+			}
+		}
+		req.AddCookie(&http.Cookie{
+			Name:  "PVEAuthCookie",
+			Value: c.ticket,
+		})
+	}
+
+	req.Header.Set("Content-Type", "application/json")
+	
+	// Add CSRF token for state-changing requests
+	if method != "GET" && method != "HEAD" && c.csrfToken != "" {
+		req.Header.Set("CSRFPreventionToken", c.csrfToken)
+	}
+
+	return c.httpClient.Do(req)
+}
+
+// GetNodes retrieves all nodes in the Proxmox cluster
+func (c *Client) GetNodes() ([]Node, error) {
+	resp, err := c.makeRequest("GET", "/api2/json/nodes", nil)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	var result struct {
+		Data []Node `json:"data"`
+	}
+
+	if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
+		return nil, fmt.Errorf("failed to decode nodes response: %w", err)
+	}
+
+	return result.Data, nil
+}
+
+// GetVMs retrieves all VMs/LXCs on a specific node
+func (c *Client) GetVMs(node string) ([]VM, error) {
+	resp, err := c.makeRequest("GET", fmt.Sprintf("/api2/json/nodes/%s/qemu", node), nil)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	var result struct {
+		Data []VM `json:"data"`
+	}
+
+	if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
+		return nil, fmt.Errorf("failed to decode VMs response: %w", err)
+	}
+
+	return result.Data, nil
+}
+
+// GetContainers retrieves all LXC containers on a specific node
+func (c *Client) GetContainers(node string) ([]Container, error) {
+	resp, err := c.makeRequest("GET", fmt.Sprintf("/api2/json/nodes/%s/lxc", node), nil)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	var result struct {
+		Data []Container `json:"data"`
+	}
+
+	if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
+		return nil, fmt.Errorf("failed to decode containers response: %w", err)
+	}
+
+	return result.Data, nil
+}
+
+// CreateVM creates a new VM on the specified node
+func (c *Client) CreateVM(node string, config VMConfig) (string, error) {
+	data := url.Values{}
+	
+	// Basic VM configuration
+	data.Set("vmid", fmt.Sprintf("%d", config.VMID))
+	data.Set("name", config.Name)
+	data.Set("memory", fmt.Sprintf("%d", config.Memory))
+	data.Set("cores", fmt.Sprintf("%d", config.Cores))
+	
+	if config.Template != "" {
+		data.Set("template", config.Template)
+	}
+	
+	if config.Storage != "" {
+		data.Set("scsi0", fmt.Sprintf("%s:%d", config.Storage, config.DiskSize))
+	}
+	
+	if config.NetworkBridge != "" {
+		data.Set("net0", fmt.Sprintf("model=virtio,bridge=%s", config.NetworkBridge))
+	}
+
+	resp, err := c.makeRequest("POST", fmt.Sprintf("/api2/json/nodes/%s/qemu", node), strings.NewReader(data.Encode()))
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		body, _ := io.ReadAll(resp.Body)
+		return "", fmt.Errorf("failed to create VM: %s - %s", resp.Status, string(body))
+	}
+
+	var result struct {
+		Data string `json:"data"`
+	}
+
+	if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
+		return "", fmt.Errorf("failed to decode create VM response: %w", err)
+	}
+
+	return result.Data, nil
+}
+
+// CreateContainer creates a new LXC container on the specified node
+func (c *Client) CreateContainer(node string, config ContainerConfig) (string, error) {
+	data := url.Values{}
+	
+	// Basic container configuration
+	data.Set("vmid", fmt.Sprintf("%d", config.VMID))
+	data.Set("hostname", config.Hostname)
+	data.Set("memory", fmt.Sprintf("%d", config.Memory))
+	data.Set("cores", fmt.Sprintf("%d", config.Cores))
+	
+	if config.Template != "" {
+		data.Set("ostemplate", config.Template)
+	}
+	
+	if config.Storage != "" {
+		data.Set("rootfs", fmt.Sprintf("%s:%d", config.Storage, config.DiskSize))
+	}
+	
+	if config.NetworkBridge != "" {
+		data.Set("net0", fmt.Sprintf("name=eth0,bridge=%s", config.NetworkBridge))
+	}
+
+	resp, err := c.makeRequest("POST", fmt.Sprintf("/api2/json/nodes/%s/lxc", node), strings.NewReader(data.Encode()))
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		body, _ := io.ReadAll(resp.Body)
+		return "", fmt.Errorf("failed to create container: %s - %s", resp.Status, string(body))
+	}
+
+	var result struct {
+		Data string `json:"data"`
+	}
+
+	if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
+		return "", fmt.Errorf("failed to decode create container response: %w", err)
+	}
+
+	return result.Data, nil
+}
+
+// StartVM starts a VM
+func (c *Client) StartVM(node string, vmid int) error {
+	resp, err := c.makeRequest("POST", fmt.Sprintf("/api2/json/nodes/%s/qemu/%d/status/start", node, vmid), nil)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("failed to start VM: %s", resp.Status)
+	}
+
+	return nil
+}
+
+// StopVM stops a VM
+func (c *Client) StopVM(node string, vmid int) error {
+	resp, err := c.makeRequest("POST", fmt.Sprintf("/api2/json/nodes/%s/qemu/%d/status/stop", node, vmid), nil)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("failed to stop VM: %s", resp.Status)
+	}
+
+	return nil
+}
+
+// DeleteVM deletes a VM
+func (c *Client) DeleteVM(node string, vmid int) error {
+	resp, err := c.makeRequest("DELETE", fmt.Sprintf("/api2/json/nodes/%s/qemu/%d", node, vmid), nil)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("failed to delete VM: %s", resp.Status)
+	}
+
+	return nil
+}
+
+// GetVMStatus retrieves the status of a VM
+func (c *Client) GetVMStatus(node string, vmid int) (*VMStatus, error) {
+	resp, err := c.makeRequest("GET", fmt.Sprintf("/api2/json/nodes/%s/qemu/%d/status/current", node, vmid), nil)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	var result struct {
+		Data VMStatus `json:"data"`
+	}
+
+	if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
+		return nil, fmt.Errorf("failed to decode VM status response: %w", err)
+	}
+
+	return &result.Data, nil
+}