Fetch Campaign Progress from the leaderboards

CustomCampaigns/Managers/CustomCampaignUIManager.cs

@@ -854,7 +854,7 @@ public void LoadErrors(List<GameplayModifierParamsSO> modifierParams)
 
         private void AddCustomTab()
         {
-            _gameplaySetupManager.Setup(CustomCampaignFlowCoordinator.CustomCampaignManager.Campaign);
+            _gameplaySetupManager.Setup(CustomCampaignFlowCoordinator.CustomCampaignManager.Campaign, _currentCampaignNodes);
         }
 
         private void OnViewActivated()

CustomCampaigns/Managers/GameplaySetupManager.cs

@@ -33,7 +33,7 @@ public GameplaySetupManager(GameplaySetupViewController gameplaySetupViewControl
             _toolsHandler = toolsHandler;
         }
 
-        public void Setup(Campaign.Campaign campaign)
+        public void Setup(Campaign.Campaign campaign, MissionNode[] campaignNodes)
         {
             if (_mainTab == null)
             {
@@ -54,7 +54,7 @@ public void Setup(Campaign.Campaign campaign)
             _mainTab.gameObject.SetActive(false);
             _mainTab.IsVisible = true;
 
-            _toolsHandler.SetCampaign(campaign);
+            _toolsHandler.SetCampaign(campaign, campaignNodes);
         }
 
         public void CampaignExit()

CustomCampaigns/UI/GameplaySetupUI/ToolsHandler.cs

@@ -1,12 +1,21 @@
 using BeatSaberMarkupLanguage;
 using BeatSaberMarkupLanguage.Attributes;
+using CustomCampaignLeaderboardLibrary;
+using CustomCampaigns.Campaign;
 using CustomCampaigns.Campaign.Missions;
 using CustomCampaigns.Managers;
 using CustomCampaigns.UI.FlowCoordinators;
+using CustomCampaigns.Utils;
+using HarmonyLib;
+using IPA.Utilities;
 using System;
 using System.Collections.Generic;
 using System.ComponentModel;
 using System.IO;
+using System.Linq;
+using System.Reflection;
+using System.Threading.Tasks;
+using UnityEngine;
 using UnityEngine.UI;
 
 namespace CustomCampaigns.UI.GameplaySetupUI
@@ -22,6 +31,12 @@ public class ToolsHandler : INotifyPropertyChanged
 
         private Campaign.Campaign _campaign;
 
+        private CampaignProgressModel _campaignProgressModel;
+        private List<MissionNode> _campaignNodes;
+        private MissionMapAnimationController _missionMapAnimationController;
+
+        private HashSet<MissionNode> _checkedMissions = new HashSet<MissionNode>();
+
         private int songsDownloaded;
         private int downloadsFailed;
 
@@ -34,25 +49,32 @@ public class ToolsHandler : INotifyPropertyChanged
         [UIValue("playlists-visible")]
         public bool PlaylistsVisible { get; private set; }
 
+        [UIComponent("fetch-progress-button")]
+        Button fetchProgressButton;
+
         [UIComponent("download-button")]
         Button downloadButton;
 
         [UIComponent("playlist-button")]
         Button playlistButton;
 
-        public ToolsHandler(MenuTransitionsHelper menuTransitionsHelper, CreditsManager creditsManager, DownloadManager downloadManager, Config config)
+        public ToolsHandler(MenuTransitionsHelper menuTransitionsHelper, CreditsManager creditsManager, DownloadManager downloadManager, CampaignFlowCoordinator campaignFlowCoordinator, MissionSelectionMapViewController missionSelectionMapViewController, Config config)
         {
             _menuTransitionsHelper = menuTransitionsHelper;
             _creditsManager = creditsManager;
             _downloadManager = downloadManager;
             _config = config;
 
+            _campaignProgressModel = campaignFlowCoordinator.GetField<CampaignProgressModel, CampaignFlowCoordinator>("_campaignProgressModel");
+            _missionMapAnimationController = missionSelectionMapViewController.GetField<MissionMapAnimationController, MissionSelectionMapViewController>("_missionMapAnimationController");
+
             PlaylistsVisible = Plugin.isPlaylistLibInstalled;
         }
 
-        internal void SetCampaign(Campaign.Campaign campaign)
+        internal void SetCampaign(Campaign.Campaign campaign, MissionNode[] campaignNodes)
         {
             _campaign = campaign;
+            _campaignNodes = campaignNodes.ToList();
 
             var creditsVisible = File.Exists(campaign.campaignPath + "/" + "credits.json");
             CreditsVisibleUnViewed = creditsVisible && !_config.creditsViewed.Contains(_campaign.info.name);
@@ -61,14 +83,109 @@ internal void SetCampaign(Campaign.Campaign campaign)
             PropertyChanged?.Invoke(this, new PropertyChangedEventArgs(nameof(CreditsVisibleUnViewed)));
             PropertyChanged?.Invoke(this, new PropertyChangedEventArgs(nameof(CreditsVisibleViewed)));
 
-            downloadButton.interactable = true;
-            downloadButton.SetButtonText("Download Missing Songs");
+            if (_campaign.missions.Count > 0)
+            {
+                fetchProgressButton.interactable = true;
+                fetchProgressButton.SetButtonText("Fetch Campaign Progress");
 
-            playlistButton.interactable = true;
+                downloadButton.interactable = true;
+                downloadButton.SetButtonText("Download Missing Songs");
+            }
 
+            playlistButton.interactable = true;
             playlistButton.SetButtonText(GetDoesPlaylistExist(_campaign) ? "Update Playlist" : "Create Playlist");
         }
 
+        #region Progress Button
+        private Mission GetMatchingMissionFromMissionNode(MissionNode node)
+        {
+            foreach (var mission in _campaign.missions)
+            {
+                MissionDataSO missionDataSO = mission.TryGetMissionData();
+                if (missionDataSO != null && missionDataSO == node.missionData)
+                {
+                    return mission;
+                }
+            }
+
+            return null;
+        }
+
+        private async Task<bool> GetIsMissionNodeOnLeaderboard(MissionNode node)
+        {
+            Mission mission = GetMatchingMissionFromMissionNode(node);
+            if (mission == null)
+            {
+                Plugin.logger.Debug($"Could not find matching mission - {node.name}");
+                return false;
+            }
+
+            LeaderboardResponse response;
+            if (_campaign.info.customMissionLeaderboard == "")
+            {
+                var hash = CustomCampaignLeaderboardLibraryUtils.GetHash(mission);
+                response = await CustomCampaignLeaderboard.LoadLeaderboards(UserInfoManager.UserInfo.platformUserId, hash);
+            }
+
+            else
+            {
+                string url = CustomCampaignLeaderboardLibraryUtils.GetURL(mission, _campaign.info.customMissionLeaderboard);
+                response = await CustomCampaignLeaderboard.LoadLeaderboards(url);
+            }
+
+            return response != null && response.you.position > 0;
+        }
+
+        [UIAction("fetch-progress-click")]
+        public async void FetchCampaignProgress()
+        {
+            fetchProgressButton.interactable = false;
+            fetchProgressButton.SetButtonText("Fetching Progress...");
+
+            Queue<MissionNode> missions = new Queue<MissionNode>();
+            missions.Enqueue(_campaignNodes[0]);
+            int i = 0;
+            while (missions.Count > 0)
+            {
+                MissionNode node = missions.Dequeue();
+                if (_checkedMissions.Contains(node))
+                {
+                    continue;
+                }
+
+                i++;
+                fetchProgressButton.SetButtonText($"Fetching {i} / {_campaignNodes.Count}...");
+                _checkedMissions.Add(node);
+
+                bool isCleared = false;
+
+                if (_campaignProgressModel.IsMissionCleared(node.missionId))
+                {
+                    isCleared = true;
+                }
+                else if (await GetIsMissionNodeOnLeaderboard(node))
+                {
+                    _campaignProgressModel.SetMissionCleared(node.missionId);
+                    isCleared = true;
+                }
+
+                if (isCleared)
+                {
+                    foreach (var child in node.childNodes)
+                    {
+                        if (!_checkedMissions.Contains(child))
+                        {
+                            missions.Enqueue(child);
+                        }
+                    }
+                }
+            }
+
+            _missionMapAnimationController.UpdateMissionMapAfterMissionWasCleared(false, null);
+            fetchProgressButton.SetButtonText("Campaign Progress Fetched");
+        }
+        #endregion
+
         #region Download Buton
         [UIAction("download-click")]
         public void DownloadMissingSongs()

CustomCampaigns/UI/ViewControllers/CampaignMissionLeaderboardViewController.cs

@@ -78,15 +78,13 @@ private IEnumerator UpdateLeaderboardsCoroutine()
                 Task<LeaderboardResponse> task;
                 if (customURL == "")
                 {
-                    Plugin.logger.Debug("normal leaderboard");
                     var hash = CustomCampaignLeaderboardLibraryUtils.GetHash(mission);
                     task = CustomCampaignLeaderboard.LoadLeaderboards(UserInfoManager.UserInfo.platformUserId, hash);
                 }
 
                 else
                 {
-                    Plugin.logger.Debug("custom leaderboard");
-                    string url = GetURL(mission, customURL);
+                    string url = CustomCampaignLeaderboardLibraryUtils.GetURL(mission, customURL);
                     task = CustomCampaignLeaderboard.LoadLeaderboards(url);
                 }
 
@@ -99,16 +97,6 @@ private IEnumerator UpdateLeaderboardsCoroutine()
             }
         }
 
-        private string GetURL(Mission mission, string customURL)
-        {
-            string url = customURL.Replace("{missionHash}", CustomCampaignLeaderboardLibraryUtils.GetHash(mission))
-                                  .Replace("{mapHash}", mission.hash)
-                                  .Replace("{characteristic}", mission.characteristic)
-                                  .Replace("{difficulty}", ((int) mission.difficulty).ToString())
-                                  .Replace("{userID}", UserInfoManager.UserInfo.platformUserId);
-            return url;
-        }
-
         private async void UpdateScores(LeaderboardResponse response)
         {
             List<ScoreData> scores = new List<ScoreData>();

CustomCampaigns/UI/Views/tools.bsml

@@ -1,4 +1,5 @@
 <vertical spacing="5" vertical-fit="PreferredSize" xmlns:xsi='http://www.w3.org/2001/XMLSchema-instance' xsi:schemaLocation='https://monkeymanboy.github.io/BSML-Docs/ https://raw.githubusercontent.com/monkeymanboy/BSML-Docs/gh-pages/BSMLSchema.xsd'>
+  <button id='fetch-progress-button' text='Fetch Campaign Progress' hover-hint='Automatically unlocks any locked levels where you have a score on the custom campaigns leaderboard.' on-click='fetch-progress-click' pref-height='7.5'/>
   <button id='download-button' text='Download Missing Songs' on-click='download-click' pref-height='7.5'/>
   <button id='playlist-button' text='Export to Playlist' on-click='playlist-click' active='~playlists-visible' pref-height='7.5'/>
   <action-button text="View Campaign Credits" on-click='credits-click' id='credits-button' active='~credits-visible-unviewed' pref-height='7.5'/>

CustomCampaigns/Utils/CustomCampaignLeaderboardLibraryUtils.cs

@@ -46,6 +46,16 @@ public static string GetHash(string s)
             return sb.ToString();
         }
 
+        public static string GetURL(Mission mission, string customURL)
+        {
+            string url = customURL.Replace("{missionHash}", CustomCampaignLeaderboardLibraryUtils.GetHash(mission))
+                                  .Replace("{mapHash}", mission.hash)
+                                  .Replace("{characteristic}", mission.characteristic)
+                                  .Replace("{difficulty}", ((int) mission.difficulty).ToString())
+                                  .Replace("{userID}", UserInfoManager.UserInfo.platformUserId);
+            return url;
+        }
+
         public static string GetSpecialName(string id, string name)
         {
             if (IsRainbowName(id))