Admin tools update

server/admin.go

@@ -149,6 +149,49 @@ func adminChangeUsername(w http.ResponseWriter, r *http.Request) {
 	w.Write([]byte("ok"))
 }
 
+func adminTempBanUser(w http.ResponseWriter, r *http.Request) {
+	uuid, _, rank, _, _, _ := getPlayerDataFromToken(r.Header.Get("Authorization"))
+	if rank == 0 {
+		handleError(w, r, "access denied")
+		return
+	}
+
+	targetUuid, duration := r.URL.Query().Get("uuid"), r.URL.Query().Get("duration")
+	if targetUuid == "" {
+		user := r.URL.Query().Get("user")
+		if user == "" {
+			handleError(w, r, "uuid or user not specified")
+			return
+		}
+
+		uuid, err := getUuidFromName(user)
+		if err != nil {
+			handleInternalError(w, r, err)
+			return
+		}
+
+		if uuid == "" {
+			handleError(w, r, "invalid user specified")
+			return
+		}
+
+		targetUuid = uuid
+	}
+
+	if duration = 0 {
+		handleError(w, r, "duration not specified")
+		return
+	}
+
+	duration , err := handleTempBan(uuid, targetUuid)
+	if err != nil {
+		handleInternalError(w, r, err)
+		return
+	}
+
+	w.Write([]byte(duration))
+}
+
 func adminResetPw(w http.ResponseWriter, r *http.Request) {
 	_, _, rank, _, _, _ := getPlayerDataFromToken(r.Header.Get("Authorization"))
 	if rank == 0 {

server/api.go

@@ -90,6 +90,7 @@ func initApi() {
 	http.HandleFunc("/admin/unban", adminBanMute)
 	http.HandleFunc("/admin/unmute", adminBanMute)
 	http.HandleFunc("/admin/changeusername", adminChangeUsername)
+	http.HandleFunc("/admin/temporaryban", adminTempBanUser)
 	http.HandleFunc("/admin/resetpw", adminResetPw)
 	http.HandleFunc("/admin/grantbadge", adminManageBadge)
 	http.HandleFunc("/admin/revokebadge", adminManageBadge)

server/database.go

@@ -196,6 +196,38 @@ func tryChangePlayerUsername(senderUuid string, recipientUuid string, newUsernam
 	return nil
 }
 
+func handleTempBan(uuid string, targetUuid string) (duration int, err error) {
+	var userExists int
+	db.QueryRow("SELECT EXISTS (SELECT * FROM accounts WHERE uuid = ?)", uuid).Scan(&userExists)
+
+	if userExists == 0 {
+		return "", errors.New("user not found")
+	}
+
+	err = tryBanPlayer(uuid, targetUuid)
+
+	if err != nil {
+		handleInternalError(http.ResponseWriter, *http.Request, err)
+		return
+	}
+
+	_, err := db.Exec("INSERT INTO playerModerationAction (uuid, action, expiry) VALUES (?, 'tempban', ?)", uuid, duration)
+	if err != nil {
+		return err
+	}
+
+	tempBan := time.NewTimer(duration * 60 * time.Second)
+
+	<-tempBan.C
+	err = tryUnbanPlayer(uuid, targetUuid)
+	if err != nil {
+		handleInternalError(http.ResponseWriter, *http.Request, err)
+		return
+	}
+
+	w.Write([]byte("ok"))
+}
+
 func getPlayerMedals(uuid string) (medals [5]int) {
 	if client, ok := clients.Load(uuid); ok {
 		return client.medals // return medals from session if client is connected