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)
|
||||
controllers := map[string]http.Handler{
|
||||
client.EnrollRoute: middlewares.WithLogger(client.NewEnrollController(logger, service), logger),
|
||||
client.ConfigRoute: middlewares.WithLogger(client.NewGetConfigController(logger, service, appConf), logger),
|
||||
}
|
||||
|
||||
m := http.NewServeMux()
|
||||
|
|
|
@ -99,6 +99,21 @@ func (chs *ClientHandlerService) AddClient(client *bootoption.Client) (uuid.UUID
|
|||
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) {
|
||||
clients, err := chs.load()
|
||||
if err != nil {
|
||||
|
|
Loading…
Reference in a new issue