add controller to get current client config
All checks were successful
continuous-integration/drone/push Build is passing
All checks were successful
continuous-integration/drone/push Build is passing
This commit is contained in:
parent
a42d3e5643
commit
99f67a7e79
3 changed files with 104 additions and 0 deletions
88
bootserver/controllers/client/getconfig.go
Normal file
88
bootserver/controllers/client/getconfig.go
Normal 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)
|
||||||
|
}
|
|
@ -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()
|
||||||
|
|
|
@ -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 {
|
||||||
|
|
Loading…
Reference in a new issue