llamactl/pkg/server/handlers_nodes.go

package server

import (
	"encoding/json"
	"llamactl/pkg/config"
	"net/http"

	"github.com/go-chi/chi/v5"
)

// NodeResponse represents a sanitized node configuration for API responses
type NodeResponse struct {
	Name    string `json:"name"`
	Address string `json:"address"`
}

// ListNodes godoc
// @Summary List all configured nodes
// @Description Returns a list of all nodes configured in the server
// @Tags nodes
// @Security ApiKeyAuth
// @Produces json
// @Success 200 {array} NodeResponse "List of nodes"
// @Failure 500 {string} string "Internal Server Error"
// @Router /nodes [get]
func (h *Handler) ListNodes() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		// Convert to sanitized response format
		nodeResponses := make([]NodeResponse, len(h.cfg.Nodes))
		for i, node := range h.cfg.Nodes {
			nodeResponses[i] = NodeResponse{
				Name:    node.Name,
				Address: node.Address,
			}
		}

		w.Header().Set("Content-Type", "application/json")
		if err := json.NewEncoder(w).Encode(nodeResponses); err != nil {
			http.Error(w, "Failed to encode nodes: "+err.Error(), http.StatusInternalServerError)
			return
		}
	}
}

// GetNode godoc
// @Summary Get details of a specific node
// @Description Returns the details of a specific node by name
// @Tags nodes
// @Security ApiKeyAuth
// @Produces json
// @Param name path string true "Node Name"
// @Success 200 {object} NodeResponse "Node details"
// @Failure 400 {string} string "Invalid name format"
// @Failure 404 {string} string "Node not found"
// @Failure 500 {string} string "Internal Server Error"
// @Router /nodes/{name} [get]
func (h *Handler) GetNode() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		name := chi.URLParam(r, "name")
		if name == "" {
			http.Error(w, "Node name cannot be empty", http.StatusBadRequest)
			return
		}

		var nodeConfig *config.NodeConfig
		for i := range h.cfg.Nodes {
			if h.cfg.Nodes[i].Name == name {
				nodeConfig = &h.cfg.Nodes[i]
				break
			}
		}

		if nodeConfig == nil {
			http.Error(w, "Node not found", http.StatusNotFound)
			return
		}

		// Convert to sanitized response format
		nodeResponse := NodeResponse{
			Name:    nodeConfig.Name,
			Address: nodeConfig.Address,
		}

		w.Header().Set("Content-Type", "application/json")
		if err := json.NewEncoder(w).Encode(nodeResponse); err != nil {
			http.Error(w, "Failed to encode node: "+err.Error(), http.StatusInternalServerError)
			return
		}
	}
}