osu/osu.Game/Online/API/Requests/SearchBeatmapSetsRequest.cs

// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.

using osu.Framework.IO.Network;
using osu.Game.Overlays;
using osu.Game.Overlays.BeatmapListing;
using osu.Game.Rulesets;
using Newtonsoft.Json;

namespace osu.Game.Online.API.Requests
{
    public class SearchBeatmapSetsRequest : APIRequest<SearchBeatmapSetsResponse>
    {
        public class Cursor
        {
            [JsonProperty("title.raw")]
            public string Title;

            [JsonProperty("artist.raw")]
            public string Artist;

            [JsonProperty("beatmaps.difficultyrating")]
            public string Difficulty;

            [JsonProperty("approved_date")]
            public string Ranked;

            [JsonProperty("rating")]
            public string Rating;

            [JsonProperty("play_count")]
            public string Plays;

            [JsonProperty("favourite_count")]
            public string Favourites;

            [JsonProperty("_score")]
            public string Relevance;

            [JsonProperty("_id")]
            public string Id;
        }

        public SearchCategory SearchCategory { get; set; }

        public SortCriteria SortCriteria { get; set; }

        public SortDirection SortDirection { get; set; }

        public SearchGenre Genre { get; set; }

        public SearchLanguage Language { get; set; }

        private readonly string query;
        private readonly RulesetInfo ruleset;
        private readonly Cursor cursor;

        private string directionString => SortDirection == SortDirection.Descending ? @"desc" : @"asc";

        public SearchBeatmapSetsRequest(string query, RulesetInfo ruleset, Cursor cursor = null,
            SearchCategory searchCategory = SearchCategory.Any, SortCriteria sortCriteria = SortCriteria.Ranked, SortDirection sortDirection = SortDirection.Descending)
        {
            this.query = string.IsNullOrEmpty(query) ? string.Empty : System.Uri.EscapeDataString(query);
            this.ruleset = ruleset;
            this.cursor = cursor;

            SearchCategory = searchCategory;
            SortCriteria = sortCriteria;
            SortDirection = sortDirection;
            Genre = SearchGenre.Any;
            Language = SearchLanguage.Any;
        }

        protected override WebRequest CreateWebRequest()
        {
            var req = base.CreateWebRequest();
            req.AddParameter("q", query);

            if (ruleset.ID.HasValue)
                req.AddParameter("m", ruleset.ID.Value.ToString());

            req.AddParameter("s", SearchCategory.ToString().ToLowerInvariant());

            if (Genre != SearchGenre.Any)
                req.AddParameter("g", ((int)Genre).ToString());

            if (Language != SearchLanguage.Any)
                req.AddParameter("l", ((int)Language).ToString());

            req.AddParameter("sort", $"{SortCriteria.ToString().ToLowerInvariant()}_{directionString}");

            if (cursor != null)
            {
                if (cursor.Title != null)
                    req.AddParameter("cursor[title.raw]", cursor.Title);
                if (cursor.Artist != null)
                    req.AddParameter("cursor[artist.raw]", cursor.Artist);
                if (cursor.Difficulty != null)
                    req.AddParameter("cursor[beatmaps.difficultyrating]", cursor.Difficulty);
                if (cursor.Ranked != null)
                    req.AddParameter("cursor[approved_date]", cursor.Ranked);
                if (cursor.Rating != null)
                    req.AddParameter("cursor[rating]", cursor.Rating);
                if (cursor.Plays != null)
                    req.AddParameter("cursor[play_count]", cursor.Plays);
                if (cursor.Favourites != null)
                    req.AddParameter("cursor[favourite_count]", cursor.Favourites);
                if (cursor.Relevance != null)
                    req.AddParameter("cursor[_score]", cursor.Relevance);

                req.AddParameter("cursor[_id]", cursor.Id);
            }

            return req;
        }

        protected override string Target => @"beatmapsets/search";
    }
}