add controller to get current client config
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
Melora Hugues 2023-08-13 20:40:22 +02:00
parent a42d3e5643
commit 99f67a7e79
3 changed files with 104 additions and 0 deletions

View file

@ -0,0 +1,88 @@
package client
import (
"encoding/json"
"errors"
"net/http"
"git.faercol.me/faercol/http-boot-server/bootserver/bootoption"
"git.faercol.me/faercol/http-boot-server/bootserver/config"
"git.faercol.me/faercol/http-boot-server/bootserver/helpers"
"git.faercol.me/faercol/http-boot-server/bootserver/services"
"github.com/google/uuid"
"github.com/sirupsen/logrus"
)
const ConfigRoute = "/config"
type clientNetConfig struct {
MulticastGroup string `json:"multicast_group"`
Port int `json:"port"`
}
type clientConfig struct {
EFIConfig *bootoption.Client `json:"efi_config"`
NetConfig *clientNetConfig `json:"net_config"`
}
type GetConfigController struct {
clientService *services.ClientHandlerService
config *config.AppConfig
l *logrus.Logger
}
func NewGetConfigController(l *logrus.Logger, service *services.ClientHandlerService, config *config.AppConfig) *GetConfigController {
return &GetConfigController{
clientService: service,
l: l,
config: config,
}
}
func (cc *GetConfigController) getConfig(w http.ResponseWriter, r *http.Request) (int, []byte, error) {
if r.Method != http.MethodGet {
return http.StatusMethodNotAllowed, nil, nil
}
id := r.URL.Query().Get("id")
parsedId, err := uuid.Parse(id)
if err != nil {
cc.l.Errorf("Invalid uuid %q", id)
return http.StatusBadRequest, []byte("invalid uuid format"), errors.New("invalid uuid format")
}
efiConfig, err := cc.clientService.GetClientConfig(parsedId)
if err != nil {
if errors.Is(err, services.ErrUnknownClient) {
cc.l.Errorf("Unknown client %q", id)
return http.StatusNotFound, []byte("unknown client"), err
} else {
cc.l.Errorf("Unexpected error getting config for client: %s", err.Error())
return http.StatusInternalServerError, nil, err
}
}
cltConfig := clientConfig{
EFIConfig: efiConfig,
NetConfig: &clientNetConfig{
MulticastGroup: cc.config.UPDMcastGroup,
Port: cc.config.UDPPort,
},
}
jsonDat, err := json.Marshal(cltConfig)
if err != nil {
cc.l.Errorf("Failed to serialize EFI config: %s", err.Error())
return http.StatusInternalServerError, nil, err
}
w.Header().Add("Content-Type", "application/json")
return http.StatusOK, jsonDat, nil
}
func (cc *GetConfigController) ServeHTTP(w http.ResponseWriter, r *http.Request) {
returncode, content, err := cc.getConfig(w, r)
if err != nil {
cc.l.Errorf("Error getting client config: %s", err.Error())
}
helpers.HandleResponse(w, r, returncode, content, cc.l)
}

View file

@ -67,6 +67,7 @@ func New(appConf *config.AppConfig, logger *logrus.Logger) (*Server, error) {
service := services.NewClientHandlerService(appConf.DataFilepath, logger) service := services.NewClientHandlerService(appConf.DataFilepath, logger)
controllers := map[string]http.Handler{ controllers := map[string]http.Handler{
client.EnrollRoute: middlewares.WithLogger(client.NewEnrollController(logger, service), logger), client.EnrollRoute: middlewares.WithLogger(client.NewEnrollController(logger, service), logger),
client.ConfigRoute: middlewares.WithLogger(client.NewGetConfigController(logger, service, appConf), logger),
} }
m := http.NewServeMux() m := http.NewServeMux()

View file

@ -99,6 +99,21 @@ func (chs *ClientHandlerService) AddClient(client *bootoption.Client) (uuid.UUID
return client.ID, nil return client.ID, nil
} }
func (chs *ClientHandlerService) GetClientConfig(client uuid.UUID) (*bootoption.Client, error) {
clients, err := chs.load()
if err != nil {
return nil, fmt.Errorf("failed to load current config %w", err)
}
defer chs.unloadNoCommmit(clients)
clientDetails, ok := clients[client]
if !ok {
return nil, ErrUnknownClient
}
clientDetails.ID = client
return clientDetails, nil
}
func (chs *ClientHandlerService) GetClientSelectedBootOption(client uuid.UUID) (*bootoption.EFIApp, error) { func (chs *ClientHandlerService) GetClientSelectedBootOption(client uuid.UUID) (*bootoption.EFIApp, error) {
clients, err := chs.load() clients, err := chs.load()
if err != nil { if err != nil {