rank requests stuff, temporarily disabled

beatmapget/beatmapget.go

@@ -119,8 +119,7 @@ func UpdateRequired(b *BeatmapDefiningQuality) (bool, error) {
 		expire *= 6
 	}
 
-	if expire != 0 && time.Now().After(time.Time(data.LatestUpdate).Add(expire)) &&
-		!data.Frozen {
+	if expire != 0 && time.Now().After(time.Time(data.LatestUpdate).Add(expire)) && !data.Frozen {
 		return true, nil
 	}
 	return false, nil

beatmapget/fullset.go

@@ -1,6 +1,8 @@
 package beatmapget
 
 import (
+	"database/sql"
+	"errors"
 	"time"
 
 	"git.zxq.co/ripple/rippleapi/common"
@@ -13,28 +15,44 @@ func Set(s int) error {
 		lastUpdated common.UnixTimestamp
 		ranked      int
 	)
-	err := DB.QueryRow("SELECT last_updated, ranked FROM beatmaps WHERE beatmapset_id = ? LIMIT 1", s).
+	err := DB.QueryRow("SELECT latest_update, ranked FROM beatmaps WHERE beatmapset_id = ? LIMIT 1", s).
 		Scan(&lastUpdated, &ranked)
-	if err != nil {
+	if err != nil && err != sql.ErrNoRows {
 		return err
 	}
 	return set(s, lastUpdated, ranked)
 }
 
+// ErrBeatmapNotFound is returned by Beatmap if a beatmap could not be found.
+var ErrBeatmapNotFound = errors.New("beatmapget: beatmap not found")
+
 // Beatmap check if an update is required for all beatmaps in the set
 // containing this beatmap.
-func Beatmap(b int) error {
+func Beatmap(b int) (int, error) {
 	var (
 		setID       int
 		lastUpdated common.UnixTimestamp
 		ranked      int
 	)
-	err := DB.QueryRow("SELECT beatmapset_id, last_updated, ranked FROM beatmaps WHERE beatmap_id = ? LIMIT 1", b).
+	err := DB.QueryRow("SELECT beatmapset_id, latest_update, ranked FROM beatmaps WHERE beatmap_id = ? LIMIT 1", b).
 		Scan(&setID, &lastUpdated, &ranked)
-	if err != nil {
-		return err
+	switch err {
+	case nil:
+		return setID, set(setID, lastUpdated, ranked)
+	case sql.ErrNoRows:
+		beatmaps, err := Client.GetBeatmaps(osuapi.GetBeatmapsOpts{
+			BeatmapID: b,
+		})
+		if err != nil {
+			return 0, err
+		}
+		if len(beatmaps) == 0 {
+			return 0, ErrBeatmapNotFound
+		}
+		return beatmaps[0].BeatmapSetID, set(beatmaps[0].BeatmapSetID, common.UnixTimestamp(time.Time{}), 0)
+	default:
+		return setID, err
 	}
-	return set(setID, lastUpdated, ranked)
 }
 
 func set(s int, updated common.UnixTimestamp, ranked int) error {