Add the possibility to enroll a new client
This commit is contained in:
parent
a173e1bf6d
commit
639f4e60ea
2 changed files with 116 additions and 2 deletions
|
@ -19,6 +19,7 @@ type action int
|
|||
const (
|
||||
actionList action = iota
|
||||
actionGetRemote
|
||||
actionEnroll
|
||||
actionUnknown
|
||||
)
|
||||
|
||||
|
@ -29,6 +30,7 @@ type cliArgs struct {
|
|||
id uuid.UUID
|
||||
remoteAddr string
|
||||
prettyPrint bool
|
||||
name string
|
||||
}
|
||||
|
||||
var defaultArgs cliArgs = cliArgs{
|
||||
|
@ -45,9 +47,13 @@ func parseArgs() (cliArgs, error) {
|
|||
|
||||
getRemoteFlagSet := flag.NewFlagSet("get-remote", flag.ExitOnError)
|
||||
uuidFlag := getRemoteFlagSet.String("uuid", "", "Client UUID")
|
||||
remoteFlag := getRemoteFlagSet.String("remote-host", "http://localhost:5000", "Address for the remote boot server")
|
||||
hostGetRemoteFlag := getRemoteFlagSet.String("remote-host", "http://localhost:5000", "Address for the remote boot server")
|
||||
jsonFlag := getRemoteFlagSet.Bool("json", false, "Display the result in JSON format")
|
||||
|
||||
enrollFlagSet := flag.NewFlagSet("enroll", flag.ExitOnError)
|
||||
hostEnrollFlag := enrollFlagSet.String("remote-host", "http://localhost:5000", "Address for the remote boot server")
|
||||
nameFlag := enrollFlagSet.String("name", "default", "Name for the client on the remote boot server")
|
||||
|
||||
debugFlag := flag.Bool("debug", false, "Display debug logs")
|
||||
noColourFlag := flag.Bool("no-colour", false, "Disable colour logs")
|
||||
|
||||
|
@ -57,6 +63,8 @@ func parseArgs() (cliArgs, error) {
|
|||
args.action = actionList
|
||||
case "get-remote":
|
||||
args.action = actionGetRemote
|
||||
case "enroll":
|
||||
args.action = actionEnroll
|
||||
default:
|
||||
continue
|
||||
}
|
||||
|
@ -73,8 +81,12 @@ func parseArgs() (cliArgs, error) {
|
|||
return args, fmt.Errorf("invalid format for uuid %q", *uuidFlag)
|
||||
}
|
||||
args.id = parsedID
|
||||
args.remoteAddr = *remoteFlag
|
||||
args.remoteAddr = *hostGetRemoteFlag
|
||||
args.prettyPrint = !*jsonFlag
|
||||
case actionEnroll:
|
||||
enrollFlagSet.Parse(os.Args[firstArg:])
|
||||
args.remoteAddr = *hostEnrollFlag
|
||||
args.name = *nameFlag
|
||||
default:
|
||||
flag.Parse()
|
||||
return cliArgs{}, errors.New("missing an action")
|
||||
|
@ -119,6 +131,13 @@ func getRemoteConfig(l *logger.SimpleLogger, host string, id uuid.UUID, pretty b
|
|||
}
|
||||
}
|
||||
|
||||
func enroll(l *logger.SimpleLogger, host, name string) {
|
||||
l.Info("Enrolling client")
|
||||
if err := remote.Enroll(context.Background(), host, name, l); err != nil {
|
||||
l.Fatal(err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
args, err := parseArgs()
|
||||
if err != nil {
|
||||
|
@ -133,6 +152,8 @@ func main() {
|
|||
displayAppList(l)
|
||||
case actionGetRemote:
|
||||
getRemoteConfig(l, args.remoteAddr, args.id, args.prettyPrint)
|
||||
case actionEnroll:
|
||||
enroll(l, args.remoteAddr, args.name)
|
||||
default:
|
||||
l.Fatal("Unknown action")
|
||||
}
|
||||
|
|
93
config/remote/enroll.go
Normal file
93
config/remote/enroll.go
Normal file
|
@ -0,0 +1,93 @@
|
|||
package remote
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
|
||||
"git.faercol.me/faercol/http-boot-config/config/logger"
|
||||
"git.faercol.me/faercol/http-boot-config/config/prober"
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
const enrollURL = "/enroll"
|
||||
|
||||
type enrollEFIOption struct {
|
||||
Name string `json:"name"`
|
||||
Path string `json:"path"`
|
||||
}
|
||||
|
||||
type enrollPayload struct {
|
||||
Name string `json:"name"`
|
||||
Options map[string]enrollEFIOption `json:"options"`
|
||||
SelectedOption string `json:"selected_option"`
|
||||
}
|
||||
|
||||
type enrollRespPayload struct {
|
||||
ID string `json:"ID"`
|
||||
}
|
||||
|
||||
func enrollToServer(ctx context.Context, host string, name string, apps []prober.EfiApp) (uuid.UUID, error) {
|
||||
payload := enrollPayload{
|
||||
Name: name,
|
||||
Options: make(map[string]enrollEFIOption),
|
||||
}
|
||||
for _, a := range apps {
|
||||
appID := uuid.New()
|
||||
payload.Options[appID.String()] = enrollEFIOption{Name: a.Name, Path: a.Path}
|
||||
if a.Active {
|
||||
payload.SelectedOption = appID.String()
|
||||
}
|
||||
}
|
||||
|
||||
dat, err := json.Marshal(payload)
|
||||
if err != nil {
|
||||
return uuid.Nil, fmt.Errorf("failed to serialize payload: %w", err)
|
||||
}
|
||||
|
||||
subCtx, cancel := context.WithTimeout(ctx, reqTimeout)
|
||||
defer cancel()
|
||||
req, err := http.NewRequestWithContext(subCtx, http.MethodPost, host+enrollURL, bytes.NewBuffer(dat))
|
||||
if err != nil {
|
||||
return uuid.Nil, fmt.Errorf("failed to initialize request: %w", err)
|
||||
}
|
||||
|
||||
resp, err := http.DefaultClient.Do(req)
|
||||
if err != nil {
|
||||
return uuid.Nil, fmt.Errorf("failed to do query to remote boot server: %w", err)
|
||||
}
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return uuid.Nil, fmt.Errorf("unexpected returncode %d", resp.StatusCode)
|
||||
}
|
||||
|
||||
var respPayload enrollRespPayload
|
||||
respDat, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return uuid.Nil, fmt.Errorf("failed to read response from server: %w", err)
|
||||
}
|
||||
if err := json.Unmarshal(respDat, &respPayload); err != nil {
|
||||
return uuid.Nil, fmt.Errorf("failed to deserialize data from server: %w", err)
|
||||
}
|
||||
newID, err := uuid.Parse(respPayload.ID)
|
||||
if err != nil {
|
||||
return uuid.Nil, fmt.Errorf("invalid UUID %q from server: %w", respPayload.ID, err)
|
||||
}
|
||||
return newID, nil
|
||||
}
|
||||
|
||||
func Enroll(ctx context.Context, host, name string, l *logger.SimpleLogger) error {
|
||||
apps, err := prober.GetEFIApps(l)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get list of available EFI applications: %w", err)
|
||||
}
|
||||
|
||||
newID, err := enrollToServer(ctx, host, name, apps)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to enroll client to remote host: %w", err)
|
||||
}
|
||||
l.Infof("Successfully enrolled new client, ID is %q", newID.String())
|
||||
return nil
|
||||
}
|
Loading…
Reference in a new issue