TaikoLocalServer/TaikoWebUI/Services/GameDataService.cs

using System.Collections.Immutable;
using MudBlazor.Utilities;
using Swan.Mapping;
using TaikoWebUI.Shared.Models;

namespace TaikoWebUI.Services;

public class GameDataService : IGameDataService
{
    private readonly HttpClient client;

    private readonly Dictionary<uint, MusicDetail> musicMap = new();

    private ImmutableDictionary<uint, DanData> danMap = null!;

    public const int COSTUME_HEAD_MAX = 140;
    public const int COSTUME_FACE_MAX = 58;
    public const int COSTUME_BODY_MAX = 156;
    public const int COSTUME_KIGURUMI_MAX = 154;
    public const int COSTUME_PUCHI_MAX = 129;
    public const int COSTUME_COLOR_MAX = 63;
    public const int PLAYER_TITLE_MAX = 750;

    public static string[] headMap = new string[COSTUME_HEAD_MAX];
    public static string[] faceMap = new string[COSTUME_FACE_MAX];
    public static string[] bodyMap = new string[COSTUME_BODY_MAX];
    public static string[] kigurumiMap = new string[COSTUME_KIGURUMI_MAX];
    public static string[] puchiMap = new string[COSTUME_PUCHI_MAX];

    public static Dictionary<int, string> titleMap = new();

    public GameDataService(HttpClient client)
    {
        this.client = client;
    }

    public async Task InitializeAsync(string dataBaseUrl)
    {
        var musicInfo = await client.GetFromJsonAsync<MusicInfo>($"{dataBaseUrl}/data/musicinfo.json");
        var wordList = await client.GetFromJsonAsync<WordList>($"{dataBaseUrl}/data/wordlist.json");
        var musicOrder = await client.GetFromJsonAsync<MusicOrder>($"{dataBaseUrl}/data/music_order.json");
        var danData = await client.GetFromJsonAsync<List<DanData>>($"{dataBaseUrl}/data/dan_data.json");

        musicInfo.ThrowIfNull();
        wordList.ThrowIfNull();
        musicOrder.ThrowIfNull();
        danData.ThrowIfNull();

        danMap = danData.ToImmutableDictionary(data => data.DanId);
        
        // To prevent duplicate entries in wordlist
        var dict = wordList.WordListEntries.GroupBy(entry => entry.Key)
            .ToImmutableDictionary(group => group.Key, group => group.First());
        foreach (var music in musicInfo.Items)
        {
            var songNameKey = $"song_{music.Id}";
            var songArtistKey = $"song_sub_{music.Id}";

            var musicName = dict.GetValueOrDefault(songNameKey, new WordListEntry());
            var musicArtist = dict.GetValueOrDefault(songArtistKey, new WordListEntry());

            var musicSongId = music.SongId;
            var musicDetail = music.CopyPropertiesToNew<MusicDetail>();
            musicDetail.SongName = musicName.JapaneseText;
            musicDetail.ArtistName = musicArtist.JapaneseText;

            musicMap.TryAdd(musicSongId, musicDetail);
        }

        for (var index = 0; index < musicOrder.Order.Count; index++)
        {
            var musicOrderEntry = musicOrder.Order[index];
            var songId = musicOrderEntry.SongId;
            if (musicMap.ContainsKey(songId))
            {
                musicMap[songId].Index = index;
            }
        }

        for (var i = 0; i < COSTUME_HEAD_MAX; i++)
        {
            var index = i;
            var key = $"costume_head_{index}";

            var costumeWordlistItem = dict.GetValueOrDefault(key, new WordListEntry());
            headMap[index] = costumeWordlistItem.JapaneseText;
        }

        for (var i = 0; i < COSTUME_FACE_MAX; i++)
        {
            var index = i;
            var key = $"costume_face_{index}";

            var costumeWordlistItem = dict.GetValueOrDefault(key, new WordListEntry());
            faceMap[index] = costumeWordlistItem.JapaneseText;
        }

        for (var i = 0; i < COSTUME_BODY_MAX; i++)
        {
            var index = i;
            var key = $"costume_body_{index}";

            var costumeWordlistItem = dict.GetValueOrDefault(key, new WordListEntry());
            bodyMap[index] = costumeWordlistItem.JapaneseText;
        }

        for (var i = 0; i < COSTUME_KIGURUMI_MAX; i++)
        {
            var index = i;
            var key = $"costume_kigurumi_{index}";

            var costumeWordlistItem = dict.GetValueOrDefault(key, new WordListEntry());
            kigurumiMap[index] = costumeWordlistItem.JapaneseText;
        }

        for (var i = 0; i < COSTUME_PUCHI_MAX; i++)
        {
            var index = i;
            var key = $"costume_puchi_{index}";

            var costumeWordlistItem = dict.GetValueOrDefault(key, new WordListEntry());
            puchiMap[index] = costumeWordlistItem.JapaneseText;
        }

        for (var i = 1; i < PLAYER_TITLE_MAX; i++)
        {
            var index = i;
            var key = $"syougou_{index}";

            var titleWordlistItem = dict.GetValueOrDefault(key, new WordListEntry());

            // prevent duplicate values with different keys
            if (titleMap.ContainsValue(titleWordlistItem.JapaneseText))
            {
                continue;
            }

            titleMap.TryAdd(index, titleWordlistItem.JapaneseText);
        }
    }

    public string GetMusicNameBySongId(uint songId)
    {
       return musicMap.TryGetValue(songId, out var musicDetail) ? musicDetail.SongName : string.Empty;
    }

    public string GetMusicArtistBySongId(uint songId)
    {
        return musicMap.TryGetValue(songId, out var musicDetail) ? musicDetail.ArtistName : string.Empty;
    }
    public SongGenre GetMusicGenreBySongId(uint songId)
    {
        return musicMap.TryGetValue(songId, out var musicDetail) ? musicDetail.Genre : SongGenre.Variety;
    }

    public int GetMusicIndexBySongId(uint songId)
    {
        return musicMap.TryGetValue(songId, out var musicDetail) ? musicDetail.Index : int.MaxValue;
    }
    public DanData GetDanDataById(uint danId)
    {
        return danMap.GetValueOrDefault(danId, new DanData());
    }

    public int GetMusicStarLevel(uint songId, Difficulty difficulty)
    {
        var success = musicMap.TryGetValue(songId, out var musicDetail);
        return difficulty switch
        {
            Difficulty.None => throw new ArgumentException("Difficulty cannot be none"),
            Difficulty.Easy => success ? musicDetail!.StarEasy : 0,
            Difficulty.Normal => success ? musicDetail!.StarNormal : 0,
            Difficulty.Hard => success ? musicDetail!.StarHard : 0,
            Difficulty.Oni => success ? musicDetail!.StarOni : 0,
            Difficulty.UraOni => success ? musicDetail!.StarUra : 0,
            _ => throw new ArgumentOutOfRangeException(nameof(difficulty), difficulty, null)
        };
    }
}