112 lines
2.8 KiB
Go
112 lines
2.8 KiB
Go
package agentapi
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"strings"
|
|
|
|
dispatcher "git.g3e.fr/syonad/two/internal/dispatcher/agent"
|
|
"git.g3e.fr/syonad/two/pkg/db/kv"
|
|
)
|
|
|
|
func (s *Server) VmsHandler(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
switch r.Method {
|
|
case http.MethodGet:
|
|
s.listVMs(w, r)
|
|
case http.MethodPost:
|
|
s.startVM(w, r)
|
|
default:
|
|
w.WriteHeader(http.StatusMethodNotAllowed)
|
|
json.NewEncoder(w).Encode(ErrorResponse{Error: "method not allowed"})
|
|
}
|
|
}
|
|
|
|
func (s *Server) listVMs(w http.ResponseWriter, _ *http.Request) {
|
|
entries, err := kv.ListByPrefix(s.db, "vm/")
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
json.NewEncoder(w).Encode(ErrorResponse{Error: "failed to list vms"})
|
|
return
|
|
}
|
|
|
|
names := map[string]struct{}{}
|
|
for key := range entries {
|
|
parts := strings.Split(key, "/")
|
|
if len(parts) >= 2 {
|
|
names[parts[1]] = struct{}{}
|
|
}
|
|
}
|
|
|
|
result := make([]VM, 0, len(names))
|
|
for name := range names {
|
|
vm, err := vmFromDB(name, entries)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
result = append(result, vm)
|
|
}
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(result)
|
|
}
|
|
|
|
func (s *Server) startVM(w http.ResponseWriter, r *http.Request) {
|
|
var req VMCreateRequest
|
|
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
json.NewEncoder(w).Encode(ErrorResponse{Error: "invalid request body"})
|
|
return
|
|
}
|
|
if req.Name == "" || len(req.Interfaces) == 0 || len(req.Storage) == 0 {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
json.NewEncoder(w).Encode(ErrorResponse{Error: "name, interfaces and storage are required"})
|
|
return
|
|
}
|
|
|
|
var primary *VMInterface
|
|
for i := range req.Interfaces {
|
|
if req.Interfaces[i].Primary {
|
|
primary = &req.Interfaces[i]
|
|
break
|
|
}
|
|
}
|
|
if primary == nil {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
json.NewEncoder(w).Encode(ErrorResponse{Error: "one interface must be primary"})
|
|
return
|
|
}
|
|
|
|
cmd := dispatcher.StartVMCommand{
|
|
Name: req.Name,
|
|
Subnet: primary.Subnet,
|
|
IP: primary.IP,
|
|
VolumePath: req.Storage[0].Path,
|
|
Memory: req.Memory,
|
|
CPUs: req.CPUs,
|
|
UEFI: req.UEFI,
|
|
Password: req.Password,
|
|
SSHKey: req.SSHKey,
|
|
}
|
|
|
|
if err := s.dispatcher.Prepare(cmd); err != nil {
|
|
if _, dbErr := kv.GetFromDB(s.db, "vm/"+req.Name+"/state"); dbErr == nil {
|
|
w.WriteHeader(http.StatusConflict)
|
|
} else {
|
|
w.WriteHeader(http.StatusUnprocessableEntity)
|
|
}
|
|
json.NewEncoder(w).Encode(ErrorResponse{Error: err.Error()})
|
|
return
|
|
}
|
|
s.dispatcher.Dispatch(cmd)
|
|
|
|
entries, _ := kv.ListByPrefix(s.db, "vm/"+req.Name+"/")
|
|
vm, err := vmFromDB(req.Name, entries)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
json.NewEncoder(w).Encode(ErrorResponse{Error: "failed to read vm state"})
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusAccepted)
|
|
json.NewEncoder(w).Encode(vm)
|
|
}
|