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 }