Add beatmap rank requests

app/method.go

@@ -48,6 +48,7 @@ func initialCaretaker(c *gin.Context, f func(md common.MethodData) common.CodeMe
 		RequestData: data,
 		C:           c,
 		Doggo:       doggo,
+		R:           red,
 	}
 	if token != "" {
 		tokenReal, exists := GetTokenFull(token, db)

app/start.go

@@ -13,12 +13,14 @@ import (
 	"github.com/gin-gonic/gin"
 	"github.com/jmoiron/sqlx"
 	"github.com/serenize/snaker"
+	"gopkg.in/redis.v5"
 )
 
 var (
 	db    *sqlx.DB
 	cf    common.Conf
 	doggo *statsd.Client
+	red   *redis.Client
 )
 
 var commonClusterfucks = map[string]string{
@@ -65,6 +67,13 @@ func Start(conf common.Conf, dbO *sqlx.DB) *gin.Engine {
 		doggo.Incr("requests", nil, 1)
 	})
 
+	// redis
+	red = redis.NewClient(&redis.Options{
+		Addr:     conf.RedisAddr,
+		Password: conf.RedisPassword,
+		DB:       conf.RedisDB,
+	})
+
 	api := r.Group("/api")
 	{
 		p := api.Group("/")
@@ -119,7 +128,7 @@ func Start(conf common.Conf, dbO *sqlx.DB) *gin.Engine {
 			gv1.POST("/friends/del", Method(v1.FriendsDelPOST, common.PrivilegeWrite))
 			gv1.POST("/users/self/settings", Method(v1.UsersSelfSettingsPOST, common.PrivilegeWrite))
 			gv1.POST("/users/self/userpage", Method(v1.UserSelfUserpagePOST, common.PrivilegeWrite))
-			//gv1.POST("/beatmaps/rank_requests", Method(v1.BeatmapRankRequestsSubmitPOST, common.PrivilegeWrite))
+			gv1.POST("/beatmaps/rank_requests", Method(v1.BeatmapRankRequestsSubmitPOST, common.PrivilegeWrite))
 
 			// Admin: beatmap
 			gv1.POST("/beatmaps/set_status", Method(v1.BeatmapSetStatusPOST, common.PrivilegeBeatmap))

app/v1/beatmap_requests.go

@@ -5,7 +5,6 @@ import (
 	"strconv"
 	"time"
 
-	"git.zxq.co/ripple/rippleapi/beatmapget"
 	"git.zxq.co/ripple/rippleapi/common"
 	"git.zxq.co/ripple/rippleapi/limit"
 )
@@ -106,26 +105,45 @@ func BeatmapRankRequestsSubmitPOST(md common.MethodData) common.CodeMessager {
 		return common.SimpleResponse(403, "It's not possible to do a rank request at this time.")
 	}
 
-	if d.SetID == 0 {
-		d.SetID, err = beatmapget.Beatmap(d.ID)
-	} else {
-		err = beatmapget.Set(d.SetID)
+	w := common.
+		Where("beatmap_id = ?", strconv.Itoa(d.ID)).Or().
+		Where("beatmapset_id = ?", strconv.Itoa(d.SetID))
+
+	var ranked int
+	err = md.DB.QueryRow("SELECT ranked FROM beatmaps "+w.Clause+" LIMIT 1", w.Params...).Scan(&ranked)
+	if ranked >= 2 {
+		return common.SimpleResponse(406, "That beatmap is already ranked.")
 	}
-	if err == beatmapget.ErrBeatmapNotFound {
-		return common.SimpleResponse(404, "That beatmap could not be found anywhere!")
-	}
-	if err != nil {
+
+	switch err {
+	case nil:
+		// move on
+	case sql.ErrNoRows:
+		if d.SetID != 0 {
+			md.R.Publish("ripple:beatmap_updates:sets", strconv.Itoa(d.SetID))
+		} else {
+			md.R.Publish("ripple:beatmap_updates:single", strconv.Itoa(d.ID))
+		}
+	default:
 		md.Err(err)
 		return Err500
 	}
 
+	// type and value of beatmap rank request
+	t := "b"
+	v := d.ID
+	if d.SetID != 0 {
+		t = "s"
+		v = d.SetID
+	}
 	err = md.DB.QueryRow("SELECT 1 FROM rank_requests WHERE bid = ? AND type = ? AND time > ?",
-		d.SetID, "s", time.Now().Unix()).Scan(new(int))
+		v, t, time.Now().Add(-time.Hour*24).Unix()).Scan(new(int))
+
+	// error handling
 	switch err {
 	case sql.ErrNoRows:
 		break
 	case nil:
-		// TODO: return beatmap
 		// we're returning a success because if the request was already sent in the past 24
 		// hours, it's as if the user submitted it.
 		return common.SimpleResponse(200, "Your request has been submitted.")
@@ -136,12 +154,11 @@ func BeatmapRankRequestsSubmitPOST(md common.MethodData) common.CodeMessager {
 
 	_, err = md.DB.Exec(
 		"INSERT INTO rank_requests (userid, bid, type, time, blacklisted) VALUES (?, ?, ?, ?, 0)",
-		md.ID(), d.SetID, "s", time.Now().Unix())
+		md.ID(), v, t, time.Now().Unix())
 	if err != nil {
 		md.Err(err)
 		return Err500
 	}
 
-	// TODO: return beatmap
 	return common.SimpleResponse(200, "Your request has been submitted.")
 }