Fix for 1.36.0

SongPlayHistory/Model/LevelMapKey.cs

@@ -19,8 +19,8 @@ internal LevelMapKey(string levelId, string characteristicName, BeatmapDifficult
             Difficulty = difficulty;
         }
 
-        public LevelMapKey(IDifficultyBeatmap beatmap)
-            : this(beatmap.level.levelID, beatmap.parentDifficultyBeatmapSet.beatmapCharacteristic.serializedName, beatmap.difficulty)
+        public LevelMapKey(BeatmapKey beatmap)
+            : this(beatmap.levelId, beatmap.beatmapCharacteristic.serializedName, beatmap.difficulty)
         {
         }
 

SongPlayHistory/Patches/DiTailsVotePatch.cs

@@ -25,12 +25,12 @@ private static bool Prepare()
         }
 
         [HarmonyPrefix]
-        public static void Prefix(bool upvote, IDifficultyBeatmap? ____activeBeatmap)
+        public static void Prefix(bool upvote, BeatmapLevel? ____activeBeatmap)  //TODO: update when DiTails Update
         {
             if (____activeBeatmap == null) return;
             var vote = upvote ? VoteType.Upvote : VoteType.Downvote;
-            Plugin.Log.Debug($"DiTails voted {vote} to {____activeBeatmap.level.levelID}");
-            InMenuVoteTrackingHelper.Instance?.Vote(____activeBeatmap.level, vote);
+            Plugin.Log.Debug($"DiTails voted {vote} to {____activeBeatmap.levelID}");
+            InMenuVoteTrackingHelper.Instance?.Vote(____activeBeatmap, vote);
         }
     }
 }

SongPlayHistory/Patches/HarmonyPatches.cs

@@ -30,7 +30,7 @@ public static bool Prepare()
         }
 
         [HarmonyAfter("com.kyle1413.BeatSaber.SongCore")]
-        public static void Postfix(LevelListTableCell __instance, IPreviewBeatmapLevel? level, bool isFavorite, 
+        public static void Postfix(LevelListTableCell __instance, BeatmapLevel? level, bool isFavorite, 
             Image ____favoritesBadgeImage, TextMeshProUGUI? ____songBpmText)
         {
             if (!PluginConfig.Instance.ShowVotes) return;

SongPlayHistory/SongPlayData/IRecordManager.cs

@@ -5,7 +5,7 @@ namespace SongPlayHistory.SongPlayData;
 
 public interface IRecordManager
 {
-    public IList<ISongPlayRecord> GetRecords(IDifficultyBeatmap beatmap);
+    public IList<ISongPlayRecord> GetRecords(BeatmapKey beatmap);
 
     public IList<ISongPlayRecord> GetRecords(LevelMapKey key);
 

SongPlayHistory/SongPlayData/IScoringCacheManager.cs

@@ -6,5 +6,5 @@ namespace SongPlayHistory.SongPlayData;
 
 public interface IScoringCacheManager
 {
-    public Task<LevelScoringCache> GetScoringInfo(IDifficultyBeatmap beatmap, CancellationToken cancellationToken = new());
+    public Task<LevelScoringCache> GetScoringInfo(BeatmapKey beatmapKey, BeatmapLevel? beatmapLevel = null, CancellationToken cancellationToken = new());
 }

SongPlayHistory/SongPlayData/RecordsManager.cs

@@ -125,7 +125,7 @@ private void OnLevelFinished(object scene, LevelFinishedEventArgs eventArgs)
 
             if (eventArgs.LevelType == LevelType.Multiplayer)
             {
-                var beatmap = ((MultiplayerLevelScenesTransitionSetupDataSO)scene).difficultyBeatmap;
+                var beatmap = ((MultiplayerLevelScenesTransitionSetupDataSO)scene).beatmapKey;
                 SaveRecord(beatmap, result, true, energyDidReach0, failRecord);
             }
             else
@@ -136,12 +136,12 @@ private void OnLevelFinished(object scene, LevelFinishedEventArgs eventArgs)
                     _logger.Info("It was in practice or party mode, ignored.");
                     return;
                 }
-                var beatmap = ((StandardLevelScenesTransitionSetupDataSO)scene).difficultyBeatmap;
+                var beatmap = ((StandardLevelScenesTransitionSetupDataSO)scene).beatmapKey;
                 SaveRecord(beatmap, result, false, energyDidReach0, failRecord);
             }
         }
 
-        public IList<ISongPlayRecord> GetRecords(IDifficultyBeatmap beatmap)
+        public IList<ISongPlayRecord> GetRecords(BeatmapKey beatmap)
         {
             var key = new LevelMapKey(beatmap);
             return GetRecords(key);
@@ -160,7 +160,7 @@ public IList<ISongPlayRecord> GetRecords(LevelMapKey key)
             return new List<ISongPlayRecord>();
         }
 
-        private void SaveRecord(IDifficultyBeatmap? beatmap, LevelCompletionResults? result, bool isMultiplayer, bool energyDidReach0, ScoreRecord? failRecord)
+        private void SaveRecord(BeatmapKey? beatmap, LevelCompletionResults? result, bool isMultiplayer, bool energyDidReach0, ScoreRecord? failRecord)
         {
             if (beatmap == null || result == null)
             {
@@ -235,7 +235,7 @@ private void SaveRecord(IDifficultyBeatmap? beatmap, LevelCompletionResults? res
 
             _logger.Info($"Saving result. Record: {record}");
 
-            var key = new LevelMapKey(beatmap).ToOldKey();
+            var key = new LevelMapKey(beatmap.Value).ToOldKey();
 
             if (!Records.ContainsKey(key))
             {

SongPlayHistory/SongPlayData/ScoringCacheManager.cs

@@ -14,16 +14,25 @@ internal class ScoringCacheManager: IScoringCacheManager
     [Inject]
     private readonly PlayerDataModel _playerDataModel = null!;
 
+    [Inject]
+    private readonly BeatmapLevelsModel _beatmapLevelsModel = null!;
+
+    [Inject]
+    private readonly BeatmapDataLoader _beatmapDataLoader = null!;
+
+    [Inject]
+    private readonly EnvironmentsListModel _environmentListModel = null!;
+
     [Inject]
     private readonly SiraLog _logger = null!;
 
     //TODO use persistent storage cache if needed
     private readonly Dictionary<LevelMapKey, LevelScoringCache> _cache = new Dictionary<LevelMapKey, LevelScoringCache>();
 
-    public async Task<LevelScoringCache> GetScoringInfo(IDifficultyBeatmap beatmap, CancellationToken cancellationToken = new())
+    public async Task<LevelScoringCache> GetScoringInfo(BeatmapKey beatmapKey, BeatmapLevel? beatmapLevel = null, CancellationToken cancellationToken = new())
     {
         _logger.Debug($"Get scoring cache from Thread {Environment.CurrentManagedThreadId}");
-        var cacheKey = new LevelMapKey(beatmap);
+        var cacheKey = new LevelMapKey(beatmapKey);
 
         lock (_cache)
         {
@@ -34,14 +43,49 @@ internal class ScoringCacheManager: IScoringCacheManager
         }
 
         cancellationToken.ThrowIfCancellationRequested();
-        // await Task.Delay(15000, cancellationToken); // simulate am extreme loading time
+        // await Task.Delay(15000, cancellationToken); // simulate an extreme loading time
+
+        if (beatmapLevel == null)
+        {
+            _logger.Warn("BeatmapLevel is null, getting from BeatmapLevelsModel.");
+            beatmapLevel = _beatmapLevelsModel.GetBeatmapLevel(beatmapKey.levelId);
+            if (beatmapLevel == null)
+            {
+                _logger.Error("Failed to get BeatmapLevel.");
+                throw new Exception("Failed to get BeatmapLevel.");
+            }
+        }
+
+        _logger.Debug("Loading beat map level data from BeatmapLevelsModel.");
+        var loadResult = await _beatmapLevelsModel.LoadBeatmapLevelDataAsync(beatmapKey.levelId, cancellationToken);
+        cancellationToken.ThrowIfCancellationRequested();
+        if (loadResult.isError)
+        {
+            _logger.Error("Failed to get BeatmapLevelData.");
+            throw new Exception("Failed to load beat map level data.");
+        }
+
+        var beatmapLevelData = loadResult.beatmapLevelData!;
 
-        var beatmapData = await beatmap.GetBeatmapDataAsync(beatmap.GetEnvironmentInfo(), _playerDataModel.playerData.playerSpecificSettings);
+        var basicBeatmapData = beatmapLevel.GetDifficultyBeatmapData(beatmapKey.beatmapCharacteristic, beatmapKey.difficulty);
+        var envName = basicBeatmapData.environmentName;
+        var envInfo = _environmentListModel.GetEnvironmentInfoBySerializedNameSafe(envName);
+        var beatmapData = await _beatmapDataLoader.LoadBeatmapDataAsync(
+                beatmapLevelData, 
+                beatmapKey, 
+                beatmapLevel.beatsPerMinute, 
+                false, //environmentName == this.environmentInfo.serializedName, 
+                envInfo, // do we need to consider player env override?
+                _playerDataModel.playerData.gameplayModifiers,
+                _playerDataModel.playerData.playerSpecificSettings,
+                true);
+
         cancellationToken.ThrowIfCancellationRequested();
-        var notesCount = beatmapData.cuttableNotesCount;
+
+        var notesCount = basicBeatmapData.notesCount;
         var fullMaxScore = ScoreModel.ComputeMaxMultipliedScoreForBeatmap(beatmapData);
         // we can use the original v2 scoring method to calculate the adjusted max score if there is no slider or burst
-        var isV2Score = !beatmapData.GetBeatmapDataItems<SliderData>(0).Any();
+        var isV2Score = !beatmapData!.GetBeatmapDataItems<SliderData>(0).Any();
         cancellationToken.ThrowIfCancellationRequested();
 
         var newCache = new LevelScoringCache

SongPlayHistory/SongPlayHistory.csproj

@@ -17,6 +17,10 @@
     <Reference Include="0Harmony" HintPath="$(BeatSaberDir)\Libs\0Harmony.dll" />
     <Reference Include="BeatmapCore" HintPath="$(BeatSaberDir)\Beat Saber_Data\Managed\BeatmapCore.dll" />
     <Reference Include="BeatSaverVoting" HintPath="$(BeatSaberDir)\Plugins\BeatSaverVoting.dll" Publicize="true" />
+    <Reference Include="BGLib.AppFlow" HintPath="$(BeatSaberDir)\Beat Saber_Data\Managed\BGLib.AppFlow.dll" />
+    <Reference Include="BGLib.Polyglot" HintPath="$(BeatSaberDir)\Beat Saber_Data\Managed\BGLib.Polyglot.dll" />
+    <Reference Include="BGLib.UnityExtension" HintPath="$(BeatSaberDir)\Beat Saber_Data\Managed\BGLib.UnityExtension.dll" />
+    <Reference Include="DataModels" HintPath="$(BeatSaberDir)\Beat Saber_Data\Managed\DataModels.dll" />
     <Reference Include="Hive.Versioning" HintPath="$(BeatSaberDir)\Libs\Hive.Versioning.dll" />
     <Reference Include="Main" HintPath="$(BeatSaberDir)\Beat Saber_Data\Managed\Main.dll" />
     <Reference Include="Newtonsoft.Json" HintPath="$(BeatSaberDir)\Libs\Newtonsoft.Json.dll" />
@@ -26,7 +30,6 @@
     <Reference Include="HMUI" HintPath="$(BeatSaberDir)\Beat Saber_Data\Managed\HMUI.dll" />
     <Reference Include="HMLib" HintPath="$(BeatSaberDir)\Beat Saber_Data\Managed\HMLib.dll" />
     <Reference Include="IPA.Loader" HintPath="$(BeatSaberDir)\Beat Saber_Data\Managed\IPA.Loader.dll" />
-    <Reference Include="Polyglot" HintPath="$(BeatSaberDir)\Beat Saber_Data\Managed\Polyglot.dll" />
     <Reference Include="SiraUtil" HintPath="$(BeatSaberDir)\Plugins\SiraUtil.dll" />
     <Reference Include="SongCore" HintPath="$(BeatSaberDir)\Plugins\SongCore.dll" />
     <Reference Include="Unity.TextMeshPro" HintPath="$(BeatSaberDir)\Beat Saber_Data\Managed\Unity.TextMeshPro.dll" />

SongPlayHistory/UI/SPHUI.cs

@@ -6,7 +6,7 @@
 using System.Threading.Tasks;
 using HMUI;
 using IPA.Utilities;
-using Polyglot;
+using BGLib.Polyglot;
 using SiraUtil.Logging;
 using SongPlayHistory.Configuration;
 using SongPlayHistory.Model;
@@ -136,31 +136,31 @@ public void Dispose()
             _cts = null;
         }
 
-        private void OnDifficultyChanged(StandardLevelDetailViewController _, IDifficultyBeatmap beatmap)
+        private void OnDifficultyChanged(StandardLevelDetailViewController controller)
         {
-            UpdateUI(beatmap);
+            UpdateUI(controller.beatmapKey, controller.beatmapLevel);
         }
 
-        private void OnContentChanged(StandardLevelDetailViewController _, StandardLevelDetailViewController.ContentType contentType)
+        private void OnContentChanged(StandardLevelDetailViewController controller, StandardLevelDetailViewController.ContentType contentType)
         {
             if (contentType == StandardLevelDetailViewController.ContentType.OwnedAndReady)
             {
-                UpdateUI(_levelDetailViewController.selectedDifficultyBeatmap);
+                UpdateUI(controller.beatmapKey, controller.beatmapLevel);
             }
         }
 
         private void OnPlayResultDismiss(ResultsViewController _)
         {
-            UpdateUI(_levelDetailViewController.selectedDifficultyBeatmap);
+            UpdateUI(_levelDetailViewController.beatmapKey, _levelDetailViewController.beatmapLevel);
         }
 
-        private void UpdateUI(IDifficultyBeatmap? beatmap)
+        private void UpdateUI(BeatmapKey beatmapKey, BeatmapLevel? beatmap)
         {
             if (beatmap == null) return;
             _logger.Info("Updating SPH UI");
-            _logger.Debug($"{beatmap.level.songName} {beatmap.parentDifficultyBeatmapSet.beatmapCharacteristic.serializedName} {beatmap.difficulty}");
+            _logger.Debug($"{beatmap.songName} {beatmapKey.beatmapCharacteristic.serializedName} {beatmapKey.difficulty}");
 
-            SetStats(beatmap);
+            SetStats(beatmapKey);
 
             _cts?.Cancel();
             _cts?.Dispose();
@@ -170,11 +170,11 @@ private void UpdateUI(IDifficultyBeatmap? beatmap)
             {
                 try
                 {
-                    await SetRecords(beatmap, token);
+                    await SetRecords(beatmapKey, beatmap, token);
                 }
                 catch (OperationCanceledException e) when (e.CancellationToken == token)
                 {
-                    _logger.Debug($"Update cancelled: {beatmap.level.songName}");
+                    _logger.Debug($"Update cancelled: {beatmap.songName}");
                 }
                 catch (Exception ex)
                 {
@@ -184,14 +184,14 @@ private void UpdateUI(IDifficultyBeatmap? beatmap)
             }, token);
         }
 
-        private async Task SetRecords(IDifficultyBeatmap beatmap, CancellationToken cancellationToken)
+        private async Task SetRecords(BeatmapKey beatmapKey, BeatmapLevel beatmap, CancellationToken cancellationToken)
         {
             _logger.Debug($"Setting records from Thread {Environment.CurrentManagedThreadId}");
 
-            var task = _scoringCacheManager.GetScoringInfo(beatmap, cancellationToken);  // let it run in the background first
+            var task = _scoringCacheManager.GetScoringInfo(beatmapKey, beatmap, cancellationToken);  // let it run in the background first
 
             var config = PluginConfig.Instance;
-            var key = new LevelMapKey(beatmap);
+            var key = new LevelMapKey(beatmapKey);
             var records =
                 from record in _recordsManager.GetRecords(key)
                 where config.ShowFailed || record.LevelEnd == LevelEndType.Cleared
@@ -295,10 +295,12 @@ from record in _recordsManager.GetRecords(key)
             _hoverHint.text = builder.ToString();
         }
 
-        private void SetStats(IDifficultyBeatmap beatmap)
+        private void SetStats(BeatmapKey beatmap)
         {
-            var stats = _playerDataModel.playerData.GetPlayerLevelStatsData(beatmap.level.levelID, beatmap.difficulty, beatmap.parentDifficultyBeatmapSet.beatmapCharacteristic);
-            _playCount.text = stats.playCount.ToString();
+            var playCount = _playerDataModel.playerData.levelsStatsData.TryGetValue(beatmap, out var data)
+                ? data.playCount
+                : 0;
+            _playCount.text = playCount.ToString();
         }
     }
 }

SongPlayHistory/Utils/Utils.cs

@@ -32,9 +32,9 @@ internal static bool IsInReplay()
         }
         #endregion
 
-        internal static string GetLowerCaseCustomLevelHash(CustomPreviewBeatmapLevel level)
+        internal static string? GetLowerCaseCustomLevelHash(BeatmapLevel level)
         {
-            return Hashing.GetCustomLevelHash(level).ToLower();
+            return level.levelID.StartsWith("custom_level_") ? Hashing.GetCustomLevelHash(level).ToLower() : null;
         }
 
         internal static IList<ISongPlayRecord> Copy(this IEnumerable<Record> records)

SongPlayHistory/VoteTracker/BeatSaverVotingTracker.cs

@@ -12,20 +12,25 @@ internal class BeatSaverVotingTracker: IVoteTracker
 
         [Inject]
         private readonly SiraLog _logger = null!;
-        
-        public void Vote(IPreviewBeatmapLevel level, VoteType voteType)
+
+        public void Vote(BeatmapLevel level, VoteType voteType)
         {
-            if (!(level is CustomPreviewBeatmapLevel customLevel)) return;
             try
             {
                 _logger.Debug($"Voting for {level.songName}, {level.levelID}, {voteType}");
+                var hash = Utils.Utils.GetLowerCaseCustomLevelHash(level);
+                if (hash == null)
+                {
+                    _logger.Debug("Not custom level");
+                    return;
+                }
+
                 if (BSVPlugin.votedSongs == null)
                 {
                     _logger.Debug("BeatSaverVoting dictionary is null");
                     return;
                 }
 
-                var hash = Utils.Utils.GetLowerCaseCustomLevelHash(customLevel);
                 var bsvType = voteType == VoteType.Upvote ? BSVType.Upvote : BSVType.Downvote;
                 if (BSVPlugin.votedSongs.TryGetValue(hash, out var songVote) && songVote.voteType == bsvType) return;
 
@@ -39,14 +44,18 @@ public void Vote(IPreviewBeatmapLevel level, VoteType voteType)
             }
         }
 
-        public bool TryGetVote(IPreviewBeatmapLevel level, out VoteType voteType)
+        public bool TryGetVote(BeatmapLevel level, out VoteType voteType)
         {
             voteType = VoteType.Downvote;
-            if (!(level is CustomPreviewBeatmapLevel customLevel)) return false;
             try
             {
                 _logger.Debug($"Getting vote data for {level.songName}, {level.levelID}");
-                var hash = Utils.Utils.GetLowerCaseCustomLevelHash(customLevel);
+                var hash = Utils.Utils.GetLowerCaseCustomLevelHash(level);
+                if (hash == null)
+                {
+                    _logger.Debug("Not custom level");
+                    return false;
+                }
 
                 var voteStatus = BSVPlugin.CurrentVoteStatus(hash);
                 if (voteStatus == null)

SongPlayHistory/VoteTracker/IVoteTracker.cs

@@ -4,9 +4,9 @@ namespace SongPlayHistory.VoteTracker
 {
     public interface IVoteTracker
     {
-        public bool TryGetVote(IPreviewBeatmapLevel level, out VoteType voteType);
+        public bool TryGetVote(BeatmapLevel level, out VoteType voteType);
 
-        public void Vote(IPreviewBeatmapLevel level, VoteType voteType);
+        public void Vote(BeatmapLevel level, VoteType voteType);
 
     }
 }

SongPlayHistory/VoteTracker/InMenuVoteTrackingHelper.cs

@@ -47,14 +47,14 @@ private void OnPlayResultDismiss(ResultsViewController _)
             _tableView.RefreshCellsContent();
         }
 
-        internal void Vote(IPreviewBeatmapLevel level, VoteType voteType)
+        internal void Vote(BeatmapLevel level, VoteType voteType)
         {
             _voteTracker.Vote(level, voteType);
             _logger.Debug("Refreshing cells content");
             _tableView.RefreshCellsContent();
         }
 
-        internal bool TryGetVote(IPreviewBeatmapLevel level, out VoteType voteType)
+        internal bool TryGetVote(BeatmapLevel level, out VoteType voteType)
         {
             return _voteTracker.TryGetVote(level, out voteType);
         }