[APIV4] Implement PUT /users/{user_id}/status for apiV4 (#5928)

1 files changed, 33 insertions, 1 deletions

diff --git a/api4/status.go b/api4/status.go
index 142f5ce5a..46d4f963b 100644
--- a/api4/status.go
+++ b/api4/status.go
@@ -18,7 +18,7 @@ func InitStatus() {
 
 	BaseRoutes.User.Handle("/status", ApiHandler(getUserStatus)).Methods("GET")
 	BaseRoutes.Users.Handle("/status/ids", ApiHandler(getUserStatusesByIds)).Methods("POST")
-
+	BaseRoutes.User.Handle("/status", ApiHandler(updateUserStatus)).Methods("PUT")
 }
 
 func getUserStatus(c *Context, w http.ResponseWriter, r *http.Request) {
@@ -59,3 +59,35 @@ func getUserStatusesByIds(c *Context, w http.ResponseWriter, r *http.Request) {
 		w.Write([]byte(model.StatusListToJson(statusMap)))
 	}
 }
+
+func updateUserStatus(c *Context, w http.ResponseWriter, r *http.Request) {
+	c.RequireUserId()
+	if c.Err != nil {
+		return
+	}
+
+	status := model.StatusFromJson(r.Body)
+	if status == nil {
+		c.SetInvalidParam("status")
+		return
+	}
+
+	if !app.SessionHasPermissionToUser(c.Session, c.Params.UserId) {
+		c.SetPermissionError(model.PERMISSION_EDIT_OTHER_USERS)
+		return
+	}
+
+	switch status.Status {
+	case "online":
+		app.SetStatusOnline(c.Params.UserId, "", true)
+	case "offline":
+		app.SetStatusOffline(c.Params.UserId, true)
+	case "away":
+		app.SetStatusAwayIfNeeded(c.Params.UserId, true)
+	default:
+		c.SetInvalidParam("status")
+		return
+	}
+
+	getUserStatus(c, w, r)
+}