replace zxq.co/ripple/hanayo

vendor/zxq.co/ripple/rippleapi/beatmapget/fullset.go

@@ -0,0 +1,81 @@
+package beatmapget
+
+import (
+	"database/sql"
+	"errors"
+	"time"
+
+	"zxq.co/ripple/rippleapi/common"
+	"gopkg.in/thehowl/go-osuapi.v1"
+)
+
+// Set checks if an update is required for all beatmaps in a set.
+func Set(s int) error {
+	var (
+		lastUpdated common.UnixTimestamp
+		ranked      int
+	)
+	err := DB.QueryRow("SELECT latest_update, ranked FROM beatmaps WHERE beatmapset_id = ? LIMIT 1", s).
+		Scan(&lastUpdated, &ranked)
+	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) (int, error) {
+	var (
+		setID       int
+		lastUpdated common.UnixTimestamp
+		ranked      int
+	)
+	err := DB.QueryRow("SELECT beatmapset_id, latest_update, ranked FROM beatmaps WHERE beatmap_id = ? LIMIT 1", b).
+		Scan(&setID, &lastUpdated, &ranked)
+	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
+	}
+}
+
+func set(s int, updated common.UnixTimestamp, ranked int) error {
+	expire := Expire
+	if ranked == 2 {
+		expire *= 6
+	}
+	if time.Now().Before(time.Time(updated).Add(expire)) {
+		return nil
+	}
+	beatmaps, err := Client.GetBeatmaps(osuapi.GetBeatmapsOpts{
+		BeatmapSetID: s,
+	})
+	if err != nil {
+		return err
+	}
+	for _, beatmap := range beatmaps {
+		err := UpdateIfRequired(BeatmapDefiningQuality{
+			ID: beatmap.BeatmapID,
+		})
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}