func (as *APIServer) requestHost(w http.ResponseWriter, r *http.Request) {
user := MustHaveUser(r)
hostRequest := struct {
Distro string `json:"distro"`
PublicKey string `json:"public_key"`
UserData string `json:"userdata"`
}{}
err := util.ReadJSONInto(r.Body, &hostRequest)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
if hostRequest.Distro == "" {
http.Error(w, "distro may not be blank", http.StatusBadRequest)
return
}
if hostRequest.PublicKey == "" {
http.Error(w, "public key may not be blank", http.StatusBadRequest)
return
}
opts := spawn.Options{
Distro: hostRequest.Distro,
UserName: user.Id,
PublicKey: hostRequest.PublicKey,
UserData: hostRequest.UserData,
}
spawner := spawn.New(&as.Settings)
err = spawner.Validate(opts)
if err != nil {
errCode := http.StatusBadRequest
if _, ok := err.(spawn.BadOptionsErr); !ok {
errCode = http.StatusInternalServerError
}
as.LoggedError(w, r, errCode, fmt.Errorf("Spawn request failed validation: %v", err))
return
}
err = spawner.CreateHost(opts, user)
if err != nil {
evergreen.Logger.Logf(slogger.ERROR, err.Error())
mailErr := notify.TrySendNotificationToUser(opts.UserName, "Spawning failed", err.Error(),
notify.ConstructMailer(as.Settings.Notify))
if mailErr != nil {
evergreen.Logger.Logf(slogger.ERROR, "Failed to send notification: %v", mailErr)
}
return
}
as.WriteJSON(w, http.StatusOK, "")
}
func (uis *UIServer) requestNewHost(w http.ResponseWriter, r *http.Request) {
authedUser := MustHaveUser(r)
putParams := struct {
Distro string `json:"distro"`
KeyName string `json:"key_name"`
PublicKey string `json:"public_key"`
SaveKey bool `json:"save_key"`
UserData string `json:"userdata"`
}{}
if err := util.ReadJSONInto(r.Body, &putParams); err != nil {
http.Error(w, fmt.Sprintf("Bad json in request: %v", err), http.StatusBadRequest)
return
}
opts := spawn.Options{
Distro: putParams.Distro,
UserName: authedUser.Username(),
PublicKey: putParams.PublicKey,
UserData: putParams.UserData,
}
spawner := spawn.New(&uis.Settings)
if err := spawner.Validate(opts); err != nil {
errCode := http.StatusBadRequest
if _, ok := err.(spawn.BadOptionsErr); !ok {
errCode = http.StatusInternalServerError
}
uis.LoggedError(w, r, errCode, err)
return
}
// save the supplied public key if needed
if putParams.SaveKey {
dbuser, err := user.FindOne(user.ById(authedUser.Username()))
if err != nil {
uis.LoggedError(w, r, http.StatusInternalServerError, fmt.Errorf("Error fetching user: %v", err))
return
}
err = model.AddUserPublicKey(dbuser.Id, putParams.KeyName, putParams.PublicKey)
if err != nil {
uis.LoggedError(w, r, http.StatusInternalServerError, fmt.Errorf("Error saving public key: %v", err))
return
}
PushFlash(uis.CookieStore, r, w, NewSuccessFlash("Public key successfully saved."))
}
// Start a background goroutine that handles host creation/setup.
go func() {
host, err := spawner.CreateHost(opts)
if err != nil {
evergreen.Logger.Logf(slogger.ERROR, "error spawning host: %v", err)
mailErr := notify.TrySendNotificationToUser(authedUser.Email(), fmt.Sprintf("Spawning failed"),
err.Error(), notify.ConstructMailer(uis.Settings.Notify))
if mailErr != nil {
evergreen.Logger.Logf(slogger.ERROR, "Failed to send notification: %v", mailErr)
}
if host != nil { // a host was inserted - we need to clean it up
dErr := host.SetDecommissioned()
if err != nil {
evergreen.Logger.Logf(slogger.ERROR, "Failed to set host %v decommissioned: %v", host.Id, dErr)
}
}
return
}
}()
PushFlash(uis.CookieStore, r, w, NewSuccessFlash("Host spawned"))
uis.WriteJSON(w, http.StatusOK, "Host successfully spawned")
return
}