From 41946c362e7e3f0c1af568603df46ce1c22b8e46 Mon Sep 17 00:00:00 2001
From: AiAe <me@aiae.dev>
Date: Mon, 18 Nov 2024 15:58:21 +0200
Subject: [PATCH 1/2] Mapset search sort by

---
 db/elasticsearch_mapsets.go | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/db/elasticsearch_mapsets.go b/db/elasticsearch_mapsets.go
index 1563988..2f1161c 100644
--- a/db/elasticsearch_mapsets.go
+++ b/db/elasticsearch_mapsets.go
@@ -25,6 +25,8 @@ type ElasticMapsetSearchOptions struct {
 	ModeArray         []enums.GameMode     `form:"mode[]" json:"mode[]"`
 	Page              int                  `form:"page" json:"page"`
 	Limit             int                  `form:"limit" json:"limit"`
+	SortBy            string               `form:"sort_by" json:"sort_by"`
+	SortOrder         string               `form:"sort_order" json:"sort_order"`
 
 	MinDifficultyRating float64 `form:"min_difficulty_rating" json:"min_difficulty_rating"`
 	MaxDifficultyRating float64 `form:"max_difficulty_rating" json:"max_difficulty_rating"`
@@ -67,6 +69,8 @@ func NewElasticMapsetSearchOptions() *ElasticMapsetSearchOptions {
 		MinLastUpdated:      0,
 		MaxLastUpdated:      math.MaxInt64,
 		Explicit:            false,
+		SortBy:              "date_last_updated",
+		SortOrder:           "desc",
 	}
 }
 
@@ -349,7 +353,25 @@ func SearchElasticMapsets(options *ElasticMapsetSearchOptions) ([]*Mapset, int,
 		boolQuery.BoolQuery.Must = append(boolQuery.BoolQuery.Must, explicitTerm)
 	}
 
+	validSortFields := map[string]bool{
+		"length":            true,
+		"difficulty_rating": true,
+		"max_combo":         true,
+		"play_count":        true,
+		"date_submitted":    true,
+		"date_last_updated": true,
+	}
+
 	sort := "date_last_updated"
+	sortOrder := "desc"
+
+	if options.SortBy != "" && validSortFields[options.SortBy] {
+		sort = options.SortBy
+	}
+
+	if options.SortOrder == "asc" || options.SortOrder == "desc" {
+		sortOrder = options.SortOrder
+	}
 
 	if options.IsClanRanked {
 		clanRankedTerm := TermCustom{}
@@ -377,7 +399,7 @@ func SearchElasticMapsets(options *ElasticMapsetSearchOptions) ([]*Mapset, int,
 		Query: boolQuery,
 		Sort: []map[string]SortOrder{
 			{"_score": {Order: "desc"}},
-			{sort: {Order: "desc"}},
+			{sort: {Order: sortOrder}},
 		},
 		Aggs: map[string]interface{}{
 			"distinct_mapset_ids": map[string]interface{}{

From d72a379aee4026262441fc2972d642b13d126e37 Mon Sep 17 00:00:00 2001
From: AiAe <me@aiae.dev>
Date: Mon, 18 Nov 2024 16:04:05 +0200
Subject: [PATCH 2/2] Add bpm and ln%

---
 db/elasticsearch_mapsets.go | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/db/elasticsearch_mapsets.go b/db/elasticsearch_mapsets.go
index 2f1161c..e4a4ed6 100644
--- a/db/elasticsearch_mapsets.go
+++ b/db/elasticsearch_mapsets.go
@@ -354,12 +354,14 @@ func SearchElasticMapsets(options *ElasticMapsetSearchOptions) ([]*Mapset, int,
 	}
 
 	validSortFields := map[string]bool{
-		"length":            true,
-		"difficulty_rating": true,
-		"max_combo":         true,
-		"play_count":        true,
-		"date_submitted":    true,
-		"date_last_updated": true,
+		"length":               true,
+		"difficulty_rating":    true,
+		"max_combo":            true,
+		"play_count":           true,
+		"bpm":                  true,
+		"long_note_percentage": true,
+		"date_submitted":       true,
+		"date_last_updated":    true,
 	}
 
 	sort := "date_last_updated"