SongBrowserUI.cs 54 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367
  1. using UnityEngine;
  2. using System.Linq;
  3. using System;
  4. using System.Collections.Generic;
  5. using UnityEngine.UI;
  6. using HMUI;
  7. using VRUI;
  8. using SongBrowser.DataAccess;
  9. using TMPro;
  10. using Logger = SongBrowser.Logging.Logger;
  11. using SongBrowser.DataAccess.BeatSaverApi;
  12. using System.Collections;
  13. using SongCore.Utilities;
  14. using SongBrowser.Internals;
  15. using CustomUI.BeatSaber;
  16. namespace SongBrowser.UI
  17. {
  18. public enum UIState
  19. {
  20. Disabled,
  21. Main,
  22. SortBy,
  23. FilterBy
  24. }
  25. /// <summary>
  26. /// Hijack the flow coordinator. Have access to all StandardLevel easily.
  27. /// </summary>
  28. public class SongBrowserUI : MonoBehaviour
  29. {
  30. // Logging
  31. public const String Name = "SongBrowserUI";
  32. private const float SEGMENT_PERCENT = 0.1f;
  33. private const int LIST_ITEMS_VISIBLE_AT_ONCE = 6;
  34. // BeatSaber Internal UI structures
  35. DataAccess.BeatSaberUIController _beatUi;
  36. // New UI Elements
  37. private List<SongSortButton> _sortButtonGroup;
  38. private List<SongFilterButton> _filterButtonGroup;
  39. private Button _sortByButton;
  40. private Button _sortByDisplay;
  41. private Button _filterByButton;
  42. private Button _filterByDisplay;
  43. private Button _randomButton;
  44. private Button _clearSortFilterButton;
  45. private Button _addFavoriteButton;
  46. private SimpleDialogPromptViewController _deleteDialog;
  47. private Button _deleteButton;
  48. private Button _pageUpFastButton;
  49. private Button _pageDownFastButton;
  50. private SearchKeyboardViewController _searchViewController;
  51. private PlaylistFlowCoordinator _playListFlowCoordinator;
  52. private RectTransform _ppStatButton;
  53. private RectTransform _starStatButton;
  54. private RectTransform _njsStatButton;
  55. private Sprite _currentAddFavoriteButtonSprite;
  56. // Model
  57. private SongBrowserModel _model;
  58. public SongBrowserModel Model
  59. {
  60. get
  61. {
  62. return _model;
  63. }
  64. }
  65. // UI Created
  66. private bool _uiCreated = false;
  67. /// <summary>
  68. /// Constructor
  69. /// </summary>
  70. public SongBrowserUI() : base()
  71. {
  72. if (_model == null)
  73. {
  74. _model = new SongBrowserModel();
  75. }
  76. _model.Init();
  77. }
  78. /// <summary>
  79. /// Builds the UI for this plugin.
  80. /// </summary>
  81. public void CreateUI(MainMenuViewController.MenuButton mode)
  82. {
  83. Logger.Trace("CreateUI()");
  84. // Determine the flow controller to use
  85. FlowCoordinator flowCoordinator = null;
  86. if (mode == MainMenuViewController.MenuButton.SoloFreePlay)
  87. {
  88. Logger.Debug("Entering SOLO mode...");
  89. flowCoordinator = Resources.FindObjectsOfTypeAll<SoloFreePlayFlowCoordinator>().First();
  90. }
  91. else if (mode == MainMenuViewController.MenuButton.Party)
  92. {
  93. Logger.Debug("Entering PARTY mode...");
  94. flowCoordinator = Resources.FindObjectsOfTypeAll<PartyFreePlayFlowCoordinator>().First();
  95. }
  96. else
  97. {
  98. Logger.Debug("Entering SOLO CAMPAIGN mode...");
  99. flowCoordinator = Resources.FindObjectsOfTypeAll<CampaignFlowCoordinator>().First();
  100. }
  101. _beatUi = new DataAccess.BeatSaberUIController(flowCoordinator);
  102. // returning to the menu and switching modes could trigger this.
  103. if (_uiCreated)
  104. {
  105. return;
  106. }
  107. try
  108. {
  109. if (_playListFlowCoordinator == null)
  110. {
  111. _playListFlowCoordinator = UIBuilder.CreateFlowCoordinator<PlaylistFlowCoordinator>("PlaylistFlowCoordinator");
  112. _playListFlowCoordinator.didFinishEvent += HandleDidSelectPlaylist;
  113. }
  114. // delete dialog
  115. this._deleteDialog = UnityEngine.Object.Instantiate<SimpleDialogPromptViewController>(_beatUi.SimpleDialogPromptViewControllerPrefab);
  116. this._deleteDialog.name = "DeleteDialogPromptViewController";
  117. this._deleteDialog.gameObject.SetActive(false);
  118. // create song browser main ui
  119. CreateOuterUi();
  120. CreateSortButtons();
  121. CreateFilterButtons();
  122. CreateAddFavoritesButton();
  123. CreateDeleteButton();
  124. CreateFastPageButtons();
  125. RefreshSortButtonUI();
  126. this.InstallHandlers();
  127. this.ModifySongStatsPanel();
  128. this.ResizeSongUI();
  129. _uiCreated = true;
  130. Logger.Debug("Done Creating UI...");
  131. }
  132. catch (Exception e)
  133. {
  134. Logger.Exception("Exception during CreateUI: ", e);
  135. }
  136. }
  137. /// <summary>
  138. /// Create the outer ui.
  139. /// </summary>
  140. private void CreateOuterUi()
  141. {
  142. Logger.Debug("Creating outer UI...");
  143. float clearButtonX = -32.5f;
  144. float clearButtonY = 34.5f;
  145. float buttonY = 37f;
  146. float buttonHeight = 5.0f;
  147. float sortByButtonX = -22.5f + buttonHeight;
  148. float outerButtonFontSize = 3.0f;
  149. float displayButtonFontSize = 2.5f;
  150. float outerButtonWidth = 24.0f;
  151. // clear button
  152. _clearSortFilterButton = _beatUi.LevelPackLevelsViewController.CreateUIButton("HowToPlayButton", new Vector2(clearButtonX, clearButtonY), new Vector2(buttonHeight, buttonHeight), () =>
  153. {
  154. OnClearButtonClickEvent();
  155. }, "",
  156. Base64Sprites.XIcon);
  157. _clearSortFilterButton.GetComponentsInChildren<HorizontalLayoutGroup>().First(btn => btn.name == "Content").padding = new RectOffset(1, 1, 0, 0);
  158. RectTransform textRect = _clearSortFilterButton.GetComponentsInChildren<RectTransform>(true).FirstOrDefault(c => c.name == "Text");
  159. if (textRect != null)
  160. {
  161. UnityEngine.Object.Destroy(textRect.gameObject);
  162. }
  163. // create SortBy button and its display
  164. float curX = sortByButtonX;
  165. _sortByButton = _beatUi.LevelPackLevelsViewController.CreateUIButton("CreditsButton", new Vector2(curX, buttonY), new Vector2(outerButtonWidth, buttonHeight), () =>
  166. {
  167. RefreshOuterUIState(UIState.SortBy);
  168. }, "Sort By");
  169. _sortByButton.SetButtonTextSize(outerButtonFontSize);
  170. curX += outerButtonWidth;
  171. _sortByDisplay = _beatUi.LevelPackLevelsViewController.CreateUIButton("ApplyButton", new Vector2(curX, buttonY), new Vector2(outerButtonWidth, buttonHeight), () =>
  172. {
  173. this.Model.ToggleInverting();
  174. ProcessSongList();
  175. RefreshSongUI();
  176. }, "");
  177. _sortByDisplay.SetButtonTextSize(displayButtonFontSize);
  178. _sortByDisplay.ToggleWordWrapping(false);
  179. curX += outerButtonWidth;
  180. // create FilterBy button and its display
  181. _filterByButton = _beatUi.LevelPackLevelsViewController.CreateUIButton("CreditsButton", new Vector2(curX, buttonY), new Vector2(outerButtonWidth, buttonHeight), () =>
  182. {
  183. RefreshOuterUIState(UIState.FilterBy);
  184. }, "Filter By");
  185. _filterByButton.SetButtonTextSize(outerButtonFontSize);
  186. curX += outerButtonWidth;
  187. _filterByDisplay = _beatUi.LevelPackLevelsViewController.CreateUIButton("ApplyButton", new Vector2(curX, buttonY), new Vector2(outerButtonWidth, buttonHeight), () =>
  188. {
  189. CancelFilter();
  190. RefreshSongUI();
  191. }, "");
  192. _filterByDisplay.SetButtonTextSize(displayButtonFontSize);
  193. _filterByDisplay.ToggleWordWrapping(false);
  194. // random button
  195. _randomButton = _beatUi.LevelPackLevelsViewController.CreateUIButton("HowToPlayButton", new Vector2(curX + (outerButtonWidth / 2.0f) + (buttonHeight), clearButtonY), new Vector2(buttonHeight, buttonHeight), () =>
  196. {
  197. OnSortButtonClickEvent(SongSortMode.Random);
  198. }, "",
  199. Base64Sprites.RandomIcon);
  200. _randomButton.GetComponentsInChildren<HorizontalLayoutGroup>().First(btn => btn.name == "Content").padding = new RectOffset(0, 0, 0, 0);
  201. textRect = _randomButton.GetComponentsInChildren<RectTransform>(true).FirstOrDefault(c => c.name == "Text");
  202. if (textRect != null)
  203. {
  204. UnityEngine.Object.Destroy(textRect.gameObject);
  205. }
  206. }
  207. /// <summary>
  208. /// Create the sort button ribbon
  209. /// </summary>
  210. private void CreateSortButtons()
  211. {
  212. Logger.Debug("Create sort buttons...");
  213. float sortButtonFontSize = 2.25f;
  214. float sortButtonX = -24.5f;
  215. float sortButtonWidth = 12.25f;
  216. float buttonSpacing = 0.5f;
  217. float buttonY = 37f;
  218. float buttonHeight = 5.0f;
  219. string[] sortButtonNames = new string[]
  220. {
  221. "Song", "Author", "Newest", "YourPlays", "PP", "Difficult", "UpVotes", "PlayCount", "Rating"
  222. };
  223. SongSortMode[] sortModes = new SongSortMode[]
  224. {
  225. SongSortMode.Default, SongSortMode.Author, SongSortMode.Newest, SongSortMode.YourPlayCount, SongSortMode.PP, SongSortMode.Difficulty, SongSortMode.UpVotes, SongSortMode.PlayCount, SongSortMode.Rating
  226. };
  227. _sortButtonGroup = new List<SongSortButton>();
  228. for (int i = 0; i < sortButtonNames.Length; i++)
  229. {
  230. float curButtonX = sortButtonX + (sortButtonWidth * i) + (buttonSpacing * i);
  231. SongSortButton sortButton = new SongSortButton();
  232. sortButton.SortMode = sortModes[i];
  233. sortButton.Button = _beatUi.LevelPackLevelsViewController.CreateUIButton("ApplyButton",
  234. new Vector2(curButtonX, buttonY), new Vector2(sortButtonWidth, buttonHeight),
  235. () =>
  236. {
  237. OnSortButtonClickEvent(sortButton.SortMode);
  238. RefreshOuterUIState(UIState.Main);
  239. },
  240. sortButtonNames[i]);
  241. sortButton.Button.SetButtonTextSize(sortButtonFontSize);
  242. sortButton.Button.GetComponentsInChildren<HorizontalLayoutGroup>().First(btn => btn.name == "Content").padding = new RectOffset(3, 3, 3, 3);
  243. sortButton.Button.ToggleWordWrapping(false);
  244. sortButton.Button.name = "Sort" + sortModes[i].ToString() + "Button";
  245. _sortButtonGroup.Add(sortButton);
  246. }
  247. }
  248. /// <summary>
  249. /// Create the filter by buttons
  250. /// </summary>
  251. private void CreateFilterButtons()
  252. {
  253. Logger.Debug("Creating filter buttons...");
  254. float sortButtonFontSize = 2.25f;
  255. float sortButtonX = -24.5f;
  256. float sortButtonWidth = 12.25f;
  257. float buttonSpacing = 0.5f;
  258. float buttonY = 37f;
  259. float buttonHeight = 5.0f;
  260. List<Tuple<SongFilterMode, UnityEngine.Events.UnityAction, Sprite>> filterButtonSetup = new List<Tuple<SongFilterMode, UnityEngine.Events.UnityAction, Sprite>>()
  261. {
  262. Tuple.Create<SongFilterMode, UnityEngine.Events.UnityAction, Sprite>(SongFilterMode.Favorites, OnFavoriteFilterButtonClickEvent, Base64Sprites.StarFullIcon),
  263. Tuple.Create<SongFilterMode, UnityEngine.Events.UnityAction, Sprite>(SongFilterMode.Playlist, OnPlaylistButtonClickEvent, Base64Sprites.PlaylistIcon),
  264. Tuple.Create<SongFilterMode, UnityEngine.Events.UnityAction, Sprite>(SongFilterMode.Search, OnSearchButtonClickEvent, Base64Sprites.SearchIcon),
  265. };
  266. _filterButtonGroup = new List<SongFilterButton>();
  267. for (int i = 0; i < filterButtonSetup.Count; i++)
  268. {
  269. Tuple<SongFilterMode, UnityEngine.Events.UnityAction, Sprite> t = filterButtonSetup[i];
  270. float curButtonX = sortButtonX + (sortButtonWidth * i) + (buttonSpacing * i);
  271. SongFilterButton filterButton = new SongFilterButton();
  272. filterButton.FilterMode = t.Item1;
  273. filterButton.Button = _beatUi.LevelPackLevelsViewController.CreateUIButton("ApplyButton",
  274. new Vector2(curButtonX, buttonY), new Vector2(sortButtonWidth, buttonHeight),
  275. t.Item2,
  276. t.Item1.ToString());
  277. filterButton.Button.SetButtonTextSize(sortButtonFontSize);
  278. filterButton.Button.GetComponentsInChildren<HorizontalLayoutGroup>().First(btn => btn.name == "Content").padding = new RectOffset(3, 3, 3, 3);
  279. filterButton.Button.ToggleWordWrapping(false);
  280. filterButton.Button.onClick.AddListener(() =>
  281. {
  282. RefreshOuterUIState(UIState.Main);
  283. });
  284. filterButton.Button.name = "Filter" + t.Item1.ToString() + "Button";
  285. _filterButtonGroup.Add(filterButton);
  286. }
  287. }
  288. /// <summary>
  289. /// Create the fast page up and down buttons
  290. /// </summary>
  291. private void CreateFastPageButtons()
  292. {
  293. Logger.Debug("Creating fast scroll button...");
  294. _pageUpFastButton = Instantiate(_beatUi.TableViewPageUpButton, _beatUi.LevelPackLevelsTableViewRectTransform, false);
  295. (_pageUpFastButton.transform as RectTransform).anchorMin = new Vector2(0.5f, 1f);
  296. (_pageUpFastButton.transform as RectTransform).anchorMax = new Vector2(0.5f, 1f);
  297. (_pageUpFastButton.transform as RectTransform).anchoredPosition = new Vector2(-26f, 0.25f);
  298. (_pageUpFastButton.transform as RectTransform).sizeDelta = new Vector2(8f, 6f);
  299. _pageUpFastButton.GetComponentsInChildren<RectTransform>().First(x => x.name == "BG").sizeDelta = new Vector2(8f, 6f);
  300. _pageUpFastButton.GetComponentsInChildren<UnityEngine.UI.Image>().First(x => x.name == "Arrow").sprite = Base64Sprites.DoubleArrow;
  301. _pageUpFastButton.onClick.AddListener(delegate ()
  302. {
  303. this.JumpSongList(-1, SEGMENT_PERCENT);
  304. });
  305. _pageDownFastButton = Instantiate(_beatUi.TableViewPageDownButton, _beatUi.LevelPackLevelsTableViewRectTransform, false);
  306. (_pageDownFastButton.transform as RectTransform).anchorMin = new Vector2(0.5f, 0f);
  307. (_pageDownFastButton.transform as RectTransform).anchorMax = new Vector2(0.5f, 0f);
  308. (_pageDownFastButton.transform as RectTransform).anchoredPosition = new Vector2(-26f, -1f);
  309. (_pageDownFastButton.transform as RectTransform).sizeDelta = new Vector2(8f, 6f);
  310. _pageDownFastButton.GetComponentsInChildren<RectTransform>().First(x => x.name == "BG").sizeDelta = new Vector2(8f, 6f);
  311. _pageDownFastButton.GetComponentsInChildren<UnityEngine.UI.Image>().First(x => x.name == "Arrow").sprite = Base64Sprites.DoubleArrow;
  312. _pageDownFastButton.onClick.AddListener(delegate ()
  313. {
  314. this.JumpSongList(1, SEGMENT_PERCENT);
  315. });
  316. }
  317. /// <summary>
  318. /// Create the +/- favorite button in the play button container.
  319. /// </summary>
  320. private void CreateAddFavoritesButton()
  321. {
  322. // Create add favorite button
  323. Logger.Debug("Creating Add to favorites button...");
  324. _addFavoriteButton = UIBuilder.CreateIconButton(_beatUi.PlayButtons, _beatUi.PracticeButton, Base64Sprites.AddToFavoritesIcon);
  325. _addFavoriteButton.onClick.AddListener(delegate () {
  326. ToggleSongInPlaylist();
  327. });
  328. }
  329. /// <summary>
  330. /// Create the delete button in the play button container
  331. /// </summary>
  332. private void CreateDeleteButton()
  333. {
  334. // Create delete button
  335. Logger.Debug("Creating delete button...");
  336. _deleteButton = UIBuilder.CreateIconButton(_beatUi.PlayButtons, _beatUi.PracticeButton, Base64Sprites.DeleteIcon);
  337. _deleteButton.onClick.AddListener(delegate () {
  338. HandleDeleteSelectedLevel();
  339. });
  340. }
  341. /// <summary>
  342. /// Resize the stats panel to fit more stats.
  343. /// </summary>
  344. private void ModifySongStatsPanel()
  345. {
  346. // modify details view
  347. Logger.Debug("Resizing Stats Panel...");
  348. var statsPanel = _beatUi.StandardLevelDetailView.GetPrivateField<LevelParamsPanel>("_levelParamsPanel");
  349. var statTransforms = statsPanel.GetComponentsInChildren<RectTransform>();
  350. var valueTexts = statsPanel.GetComponentsInChildren<TextMeshProUGUI>().Where(x => x.name == "ValueText").ToList();
  351. RectTransform panelRect = (statsPanel.transform as RectTransform);
  352. panelRect.sizeDelta = new Vector2(panelRect.sizeDelta.x * 1.2f, panelRect.sizeDelta.y * 1.2f);
  353. for (int i = 0; i < statTransforms.Length; i++)
  354. {
  355. var r = statTransforms[i];
  356. if (r.name == "Separator")
  357. {
  358. continue;
  359. }
  360. r.sizeDelta = new Vector2(r.sizeDelta.x * 0.75f, r.sizeDelta.y * 0.75f);
  361. }
  362. for (int i = 0; i < valueTexts.Count; i++)
  363. {
  364. var text = valueTexts[i];
  365. text.fontSize = 3.25f;
  366. }
  367. // inject our components
  368. _ppStatButton = UnityEngine.Object.Instantiate(statTransforms[1], statsPanel.transform, false);
  369. UIBuilder.SetStatButtonIcon(_ppStatButton, Base64Sprites.GraphIcon);
  370. _starStatButton = UnityEngine.Object.Instantiate(statTransforms[1], statsPanel.transform, false);
  371. UIBuilder.SetStatButtonIcon(_starStatButton, Base64Sprites.StarFullIcon);
  372. _njsStatButton = UnityEngine.Object.Instantiate(statTransforms[1], statsPanel.transform, false);
  373. UIBuilder.SetStatButtonIcon(_njsStatButton, Base64Sprites.SpeedIcon);
  374. // shrink title
  375. var titleText = _beatUi.LevelDetailViewController.GetComponentsInChildren<TextMeshProUGUI>(true).First(x => x.name == "SongNameText");
  376. titleText.fontSize = 5.0f;
  377. }
  378. /// <summary>
  379. /// Resize some of the song table elements.
  380. /// </summary>
  381. public void ResizeSongUI()
  382. {
  383. // Reposition the table view a bit
  384. _beatUi.LevelPackLevelsTableViewRectTransform.anchoredPosition = new Vector2(0f, -2.5f);
  385. // Move the page up/down buttons a bit
  386. TableView tableView = ReflectionUtil.GetPrivateField<TableView>(_beatUi.LevelPackLevelsTableView, "_tableView");
  387. RectTransform pageUpButton = _beatUi.TableViewPageUpButton.transform as RectTransform;
  388. RectTransform pageDownButton = _beatUi.TableViewPageDownButton.transform as RectTransform;
  389. pageUpButton.anchoredPosition = new Vector2(pageUpButton.anchoredPosition.x, pageUpButton.anchoredPosition.y - 1f);
  390. pageDownButton.anchoredPosition = new Vector2(pageDownButton.anchoredPosition.x, pageDownButton.anchoredPosition.y + 1f);
  391. // shrink play button container
  392. RectTransform playContainerRect = _beatUi.StandardLevelDetailView.GetComponentsInChildren<RectTransform>().First(x => x.name == "PlayContainer");
  393. RectTransform playButtonsRect = playContainerRect.GetComponentsInChildren<RectTransform>().First(x => x.name == "PlayButtons");
  394. playButtonsRect.localScale = new Vector3(0.825f, 0.825f, 0.825f);
  395. }
  396. /// <summary>
  397. /// Add our handlers into BeatSaber.
  398. /// </summary>
  399. private void InstallHandlers()
  400. {
  401. // level pack, level, difficulty handlers, characteristics
  402. TableView tableView = ReflectionUtil.GetPrivateField<TableView>(_beatUi.LevelPackLevelsTableView, "_tableView");
  403. _beatUi.LevelPackLevelsViewController.didSelectLevelEvent -= OnDidSelectLevelEvent;
  404. _beatUi.LevelPackLevelsViewController.didSelectLevelEvent += OnDidSelectLevelEvent;
  405. _beatUi.LevelDifficultyViewController.didSelectDifficultyEvent -= OnDidSelectDifficultyEvent;
  406. _beatUi.LevelDifficultyViewController.didSelectDifficultyEvent += OnDidSelectDifficultyEvent;
  407. _beatUi.LevelPacksTableView.didSelectPackEvent -= _levelPacksTableView_didSelectPackEvent;
  408. _beatUi.LevelPacksTableView.didSelectPackEvent += _levelPacksTableView_didSelectPackEvent;
  409. _beatUi.LevelPackViewController.didSelectPackEvent -= _levelPackViewController_didSelectPackEvent;
  410. _beatUi.LevelPackViewController.didSelectPackEvent += _levelPackViewController_didSelectPackEvent;
  411. _beatUi.BeatmapCharacteristicSelectionViewController.didSelectBeatmapCharacteristicEvent -= OnDidSelectBeatmapCharacteristic;
  412. _beatUi.BeatmapCharacteristicSelectionViewController.didSelectBeatmapCharacteristicEvent += OnDidSelectBeatmapCharacteristic;
  413. // make sure the quick scroll buttons don't desync with regular scrolling
  414. _beatUi.TableViewPageDownButton.onClick.AddListener(delegate ()
  415. {
  416. this.RefreshQuickScrollButtons();
  417. });
  418. _beatUi.TableViewPageUpButton.onClick.AddListener(delegate ()
  419. {
  420. this.RefreshQuickScrollButtons();
  421. });
  422. // finished level
  423. ResultsViewController resultsViewController = _beatUi.LevelSelectionFlowCoordinator.GetPrivateField<ResultsViewController>("_resultsViewController");
  424. resultsViewController.continueButtonPressedEvent += ResultsViewController_continueButtonPressedEvent;
  425. }
  426. /// <summary>
  427. /// Helper to reduce code duplication...
  428. /// </summary>
  429. private void RefreshSongUI(bool scrollToLevel=true)
  430. {
  431. RefreshSongList(scrollToLevel);
  432. RefreshSortButtonUI();
  433. if (!scrollToLevel)
  434. {
  435. _beatUi.ScrollToLevelByRow(0);
  436. }
  437. RefreshQuickScrollButtons();
  438. RefreshCurrentSelectionDisplay();
  439. }
  440. /// <summary>
  441. /// External helper.
  442. /// </summary>
  443. public void ProcessSongList()
  444. {
  445. if (!_uiCreated)
  446. {
  447. return;
  448. }
  449. this._model.ProcessSongList(_beatUi.GetCurrentSelectedLevelPack());
  450. }
  451. /// <summary>
  452. /// Helper for common filter cancellation logic.
  453. /// </summary>
  454. public void CancelFilter()
  455. {
  456. _model.Settings.filterMode = SongFilterMode.None;
  457. SongCore.Loader.Instance.RefreshLevelPacks();
  458. }
  459. /// <summary>
  460. /// Handle updating the level pack selection after returning from a song.
  461. /// </summary>
  462. /// <param name="obj"></param>
  463. private void ResultsViewController_continueButtonPressedEvent(ResultsViewController obj)
  464. {
  465. StartCoroutine(this.UpdateLevelPackSelectionEndOfFrame());
  466. }
  467. /// <summary>
  468. /// TODO - evaluate this sillyness...
  469. /// </summary>
  470. /// <returns></returns>
  471. public IEnumerator UpdateLevelPackSelectionEndOfFrame()
  472. {
  473. yield return new WaitForEndOfFrame();
  474. try
  475. {
  476. bool didUpdateLevelPack = this.UpdateLevelPackSelection();
  477. if (!didUpdateLevelPack)
  478. {
  479. _model.ProcessSongList(_beatUi.GetCurrentSelectedLevelPack());
  480. }
  481. _beatUi.SelectAndScrollToLevel(_beatUi.LevelPackLevelsTableView, _model.LastSelectedLevelId);
  482. RefreshQuickScrollButtons();
  483. }
  484. catch (Exception e)
  485. {
  486. Logger.Exception("Exception:", e);
  487. }
  488. }
  489. /// <summary>
  490. /// Handler for level pack selection.
  491. /// </summary>
  492. /// <param name="arg1"></param>
  493. /// <param name="arg2"></param>
  494. private void _levelPacksTableView_didSelectPackEvent(LevelPacksTableView arg1, IBeatmapLevelPack arg2)
  495. {
  496. Logger.Trace("_levelPacksTableView_didSelectPackEvent(arg2={0})", arg2);
  497. try
  498. {
  499. RefreshSortButtonUI();
  500. RefreshQuickScrollButtons();
  501. }
  502. catch (Exception e)
  503. {
  504. Logger.Exception("Exception handling didSelectPackEvent...", e);
  505. }
  506. }
  507. /// <summary>
  508. /// Handler for level pack selection, controller.
  509. /// Sets the current level pack into the model and updates.
  510. /// </summary>
  511. /// <param name="arg1"></param>
  512. /// <param name="arg2"></param>
  513. private void _levelPackViewController_didSelectPackEvent(LevelPacksViewController arg1, IBeatmapLevelPack arg2)
  514. {
  515. Logger.Trace("_levelPackViewController_didSelectPackEvent(arg2={0})", arg2);
  516. try
  517. {
  518. // reset filter mode always here
  519. if (this._model.Settings.currentLevelPackId != arg2.packID)
  520. {
  521. this._model.Settings.filterMode = SongFilterMode.None;
  522. }
  523. // save level pack
  524. this._model.Settings.currentLevelPackId = arg2.packID;
  525. this._model.Settings.Save();
  526. this._model.ProcessSongList(arg2);
  527. // trickery to handle Downloader playlist level packs
  528. // We need to avoid scrolling to a level and then select the header
  529. bool scrollToLevel = true;
  530. if (arg2.packID.Contains("Playlist_"))
  531. {
  532. scrollToLevel = false;
  533. }
  534. RefreshSongUI(scrollToLevel);
  535. }
  536. catch (Exception e)
  537. {
  538. Logger.Exception("Exception handling didSelectPackEvent...", e);
  539. }
  540. }
  541. /// <summary>
  542. /// Remove all filters, update song list, save.
  543. /// </summary>
  544. private void OnClearButtonClickEvent()
  545. {
  546. Logger.Debug("Clearing all sorts and filters.");
  547. _model.Settings.sortMode = SongSortMode.Original;
  548. _model.Settings.invertSortResults = false;
  549. CancelFilter();
  550. _model.Settings.Save();
  551. ProcessSongList();
  552. RefreshSongUI();
  553. }
  554. /// <summary>
  555. /// Sort button clicked.
  556. /// </summary>
  557. private void OnSortButtonClickEvent(SongSortMode sortMode)
  558. {
  559. Logger.Debug("Sort button - {0} - pressed.", sortMode.ToString());
  560. _model.LastSelectedLevelId = null;
  561. if (_model.Settings.sortMode == sortMode)
  562. {
  563. _model.ToggleInverting();
  564. }
  565. _model.Settings.sortMode = sortMode;
  566. // update the seed
  567. if (_model.Settings.sortMode == SongSortMode.Random)
  568. {
  569. _model.Settings.randomSongSeed = Guid.NewGuid().GetHashCode();
  570. }
  571. _model.Settings.Save();
  572. ProcessSongList();
  573. RefreshSongUI();
  574. //Scroll to start of the list
  575. TableView listTableView = _beatUi.LevelPackLevelsTableView.GetPrivateField<TableView>("_tableView");
  576. listTableView.ScrollToCellWithIdx(0, TableView.ScrollPositionType.Beginning, false);
  577. // update the display
  578. _sortByDisplay.SetButtonText(_model.Settings.sortMode.ToString());
  579. }
  580. /// <summary>
  581. /// Filter by favorites.
  582. /// </summary>
  583. private void OnFavoriteFilterButtonClickEvent()
  584. {
  585. Logger.Debug("Filter button - {0} - pressed.", SongFilterMode.Favorites.ToString());
  586. if (_model.Settings.filterMode != SongFilterMode.Favorites)
  587. {
  588. _model.Settings.filterMode = SongFilterMode.Favorites;
  589. _beatUi.SelectLevelPack(PluginConfig.CUSTOM_SONG_LEVEL_PACK_ID);
  590. }
  591. else
  592. {
  593. CancelFilter();
  594. }
  595. _model.Settings.Save();
  596. ProcessSongList();
  597. RefreshSongUI();
  598. }
  599. /// <summary>
  600. /// Filter button clicked.
  601. /// </summary>
  602. /// <param name="sortMode"></param>
  603. private void OnSearchButtonClickEvent()
  604. {
  605. Logger.Debug("Filter button - {0} - pressed.", SongFilterMode.Search.ToString());
  606. if (_model.Settings.filterMode != SongFilterMode.Search)
  607. {
  608. _beatUi.SelectLevelPack(PluginConfig.CUSTOM_SONG_LEVEL_PACK_ID);
  609. this.ShowSearchKeyboard();
  610. }
  611. else
  612. {
  613. CancelFilter();
  614. ProcessSongList();
  615. RefreshSongUI();
  616. _model.Settings.Save();
  617. }
  618. }
  619. /// <summary>
  620. /// Display the playlist selector.
  621. /// </summary>
  622. /// <param name="sortMode"></param>
  623. private void OnPlaylistButtonClickEvent()
  624. {
  625. Logger.Debug("Filter button - {0} - pressed.", SongFilterMode.Playlist.ToString());
  626. _model.LastSelectedLevelId = null;
  627. if (_model.Settings.filterMode != SongFilterMode.Playlist)
  628. {
  629. _beatUi.SelectLevelPack(PluginConfig.CUSTOM_SONG_LEVEL_PACK_ID);
  630. _playListFlowCoordinator.parentFlowCoordinator = _beatUi.LevelSelectionFlowCoordinator;
  631. _beatUi.LevelSelectionFlowCoordinator.InvokePrivateMethod("PresentFlowCoordinator", new object[] { _playListFlowCoordinator, null, false, false });
  632. }
  633. else
  634. {
  635. CancelFilter();
  636. ProcessSongList();
  637. RefreshSongUI();
  638. _model.Settings.Save();
  639. }
  640. }
  641. /// <summary>
  642. /// Adjust UI based on level selected.
  643. /// Various ways of detecting if a level is not properly selected. Seems most hit the first one.
  644. /// </summary>
  645. private void OnDidSelectLevelEvent(LevelPackLevelsViewController view, IPreviewBeatmapLevel level)
  646. {
  647. try
  648. {
  649. Logger.Trace("OnDidSelectLevelEvent()");
  650. if (level == null)
  651. {
  652. Logger.Debug("No level selected?");
  653. return;
  654. }
  655. if (_model.Settings == null)
  656. {
  657. Logger.Debug("Settings not instantiated yet?");
  658. return;
  659. }
  660. _model.LastSelectedLevelId = level.levelID;
  661. HandleDidSelectLevelRow(level);
  662. }
  663. catch (Exception e)
  664. {
  665. Logger.Exception("Exception selecting song:", e);
  666. }
  667. }
  668. /// <summary>
  669. /// Switching one-saber modes for example.
  670. /// </summary>
  671. /// <param name="view"></param>
  672. /// <param name="bc"></param>
  673. private void OnDidSelectBeatmapCharacteristic(BeatmapCharacteristicSegmentedControlController view, BeatmapCharacteristicSO bc)
  674. {
  675. Logger.Trace("OnDidSelectBeatmapCharacteristic({0}", bc.name);
  676. _model.CurrentBeatmapCharacteristicSO = bc;
  677. _model.UpdateLevelRecords();
  678. this.RefreshSongList();
  679. }
  680. /// <summary>
  681. /// Handle difficulty level selection.
  682. /// </summary>
  683. private void OnDidSelectDifficultyEvent(BeatmapDifficultySegmentedControlController view, BeatmapDifficulty beatmap)
  684. {
  685. Logger.Trace("OnDidSelectDifficultyEvent({0})", beatmap);
  686. _deleteButton.interactable = (_beatUi.LevelDetailViewController.selectedDifficultyBeatmap.level.levelID.Length >= 32);
  687. this.RefreshScoreSaberData(_beatUi.LevelDetailViewController.selectedDifficultyBeatmap.level);
  688. this.RefreshNoteJumpSpeed(beatmap);
  689. }
  690. /// <summary>
  691. /// Refresh stats panel.
  692. /// </summary>
  693. /// <param name="level"></param>
  694. private void HandleDidSelectLevelRow(IPreviewBeatmapLevel level)
  695. {
  696. Logger.Trace("HandleDidSelectLevelRow({0})", level);
  697. _deleteButton.interactable = (level.levelID.Length >= 32);
  698. RefreshScoreSaberData(level);
  699. RefreshQuickScrollButtons();
  700. RefreshAddFavoriteButton(level.levelID);
  701. }
  702. /// <summary>
  703. /// Call Downloader delete.
  704. /// </summary>
  705. private void CallDownloaderDelete()
  706. {
  707. BeatSaverDownloader.UI.SongListTweaks.Instance.DeletePressed();
  708. }
  709. /// <summary>
  710. /// Pop up a delete dialog.
  711. /// </summary>
  712. private void HandleDeleteSelectedLevel()
  713. {
  714. bool DownloaderInstalled = CustomHelpers.IsModInstalled("BeatSaverDownloader");
  715. if (DownloaderInstalled)
  716. {
  717. CallDownloaderDelete();
  718. return;
  719. }
  720. IBeatmapLevel level = _beatUi.LevelDetailViewController.selectedDifficultyBeatmap.level;
  721. _deleteDialog.Init("Delete song", $"Do you really want to delete \"{ level.songName} {level.songSubName}\"?", "Delete", "Cancel",
  722. (selectedButton) =>
  723. {
  724. _beatUi.LevelSelectionFlowCoordinator.InvokePrivateMethod("DismissViewController", new object[] { _deleteDialog, null, false });
  725. if (selectedButton == 0)
  726. {
  727. try
  728. {
  729. // determine the index we are deleting so we can keep the cursor near the same spot after
  730. // the header counts as an index, so if the index came from the level array we have to add 1.
  731. var levelsTableView = _beatUi.LevelPackLevelsViewController.GetPrivateField<LevelPackLevelsTableView>("_levelPackLevelsTableView");
  732. List<IPreviewBeatmapLevel> levels = levelsTableView.GetPrivateField<IBeatmapLevelPack>("_pack").beatmapLevelCollection.beatmapLevels.ToList();
  733. int selectedIndex = levels.FindIndex(x => x.levelID == _beatUi.StandardLevelDetailView.selectedDifficultyBeatmap.level.levelID);
  734. if (selectedIndex > -1)
  735. {
  736. var song = new Song(SongCore.Loader.CustomLevels.First(x => x.Value.levelID == _beatUi.LevelDetailViewController.selectedDifficultyBeatmap.level.levelID).Value);
  737. SongCore.Loader.Instance.DeleteSong(song.path);
  738. this._model.RemoveSongFromLevelPack(_beatUi.GetCurrentSelectedLevelPack(), _beatUi.LevelDetailViewController.selectedDifficultyBeatmap.level.levelID);
  739. this.UpdateLevelDataModel();
  740. this.RefreshSongList();
  741. int removedLevels = levels.RemoveAll(x => x.levelID == _beatUi.StandardLevelDetailView.selectedDifficultyBeatmap.level.levelID);
  742. Logger.Info("Removed " + removedLevels + " level(s) from song list!");
  743. TableView listTableView = levelsTableView.GetPrivateField<TableView>("_tableView");
  744. listTableView.ScrollToCellWithIdx(selectedIndex, TableView.ScrollPositionType.Beginning, false);
  745. levelsTableView.SetPrivateField("_selectedRow", selectedIndex);
  746. listTableView.SelectCellWithIdx(selectedIndex, true);
  747. }
  748. }
  749. catch (Exception e)
  750. {
  751. Logger.Error("Unable to delete song! Exception: " + e);
  752. }
  753. }
  754. });
  755. _beatUi.LevelSelectionFlowCoordinator.InvokePrivateMethod("PresentViewController", new object[] { _deleteDialog, null, false });
  756. }
  757. /// <summary>
  758. /// Handle selection of a playlist. Show just the songs in the playlist.
  759. /// </summary>
  760. /// <param name="p"></param>
  761. private void HandleDidSelectPlaylist(Playlist p)
  762. {
  763. if (p != null)
  764. {
  765. Logger.Debug("Showing songs for playlist: {0}", p.playlistTitle);
  766. _model.Settings.filterMode = SongFilterMode.Playlist;
  767. _model.CurrentPlaylist = p;
  768. _model.Settings.Save();
  769. ProcessSongList();
  770. RefreshSongUI();
  771. }
  772. else
  773. {
  774. Logger.Debug("No playlist selected");
  775. }
  776. }
  777. /// <summary>
  778. /// Display the search keyboard
  779. /// </summary>
  780. void ShowSearchKeyboard()
  781. {
  782. if (_searchViewController == null)
  783. {
  784. _searchViewController = UIBuilder.CreateViewController<SearchKeyboardViewController>("SearchKeyboardViewController");
  785. _searchViewController.searchButtonPressed += SearchViewControllerSearchButtonPressed;
  786. _searchViewController.backButtonPressed += SearchViewControllerbackButtonPressed;
  787. }
  788. Logger.Debug("Presenting search keyboard");
  789. _beatUi.LevelSelectionFlowCoordinator.InvokePrivateMethod("PresentViewController", new object[] { _searchViewController, null, false });
  790. }
  791. /// <summary>
  792. /// Handle back button event from search keyboard.
  793. /// </summary>
  794. private void SearchViewControllerbackButtonPressed()
  795. {
  796. _beatUi.LevelSelectionFlowCoordinator.InvokePrivateMethod("DismissViewController", new object[] { _searchViewController, null, false });
  797. this._model.Settings.filterMode = SongFilterMode.None;
  798. this._model.Settings.Save();
  799. RefreshSongUI();
  800. }
  801. /// <summary>
  802. /// Handle search.
  803. /// </summary>
  804. /// <param name="searchFor"></param>
  805. private void SearchViewControllerSearchButtonPressed(string searchFor)
  806. {
  807. _beatUi.LevelSelectionFlowCoordinator.InvokePrivateMethod("DismissViewController", new object[] { _searchViewController, null, false });
  808. Logger.Debug("Searching for \"{0}\"...", searchFor);
  809. _model.Settings.filterMode = SongFilterMode.Search;
  810. _model.Settings.searchTerms.Insert(0, searchFor);
  811. _model.Settings.Save();
  812. _model.LastSelectedLevelId = null;
  813. ProcessSongList();
  814. RefreshSongUI();
  815. }
  816. /// <summary>
  817. /// Make big jumps in the song list.
  818. /// </summary>
  819. /// <param name="numJumps"></param>
  820. private void JumpSongList(int numJumps, float segmentPercent)
  821. {
  822. var levels = _beatUi.GetCurrentLevelPackLevels();
  823. if (levels == null)
  824. {
  825. return;
  826. }
  827. int totalSize = _beatUi.GetLevelPackLevelCount();
  828. int segmentSize = (int)(totalSize * segmentPercent);
  829. // Jump at least one scree size.
  830. if (segmentSize < LIST_ITEMS_VISIBLE_AT_ONCE)
  831. {
  832. segmentSize = LIST_ITEMS_VISIBLE_AT_ONCE;
  833. }
  834. TableView tableView = ReflectionUtil.GetPrivateField<TableView>(_beatUi.LevelPackLevelsTableView, "_tableView");
  835. int currentRow = _beatUi.LevelPackLevelsTableView.GetPrivateField<int>("_selectedRow");
  836. int jumpDirection = Math.Sign(numJumps);
  837. int newRow = currentRow + (jumpDirection * segmentSize);
  838. if (newRow <= 0)
  839. {
  840. newRow = 0;
  841. }
  842. else if (newRow >= totalSize)
  843. {
  844. newRow = totalSize - 1;
  845. }
  846. Logger.Debug("jumpDirection: {0}, newRow: {1}", jumpDirection, newRow);
  847. _beatUi.SelectAndScrollToLevel(_beatUi.LevelPackLevelsTableView, levels[newRow].levelID);
  848. RefreshQuickScrollButtons();
  849. }
  850. /// <summary>
  851. /// Add/Remove song from favorites depending on if it already exists.
  852. /// </summary>
  853. private void ToggleSongInPlaylist()
  854. {
  855. IBeatmapLevel songInfo = _beatUi.LevelDetailViewController.selectedDifficultyBeatmap.level;
  856. if (_model.CurrentEditingPlaylist != null)
  857. {
  858. if (_model.CurrentEditingPlaylistLevelIds.Contains(songInfo.levelID))
  859. {
  860. Logger.Info("Remove {0} from editing playlist", songInfo.songName);
  861. _model.RemoveSongFromEditingPlaylist(songInfo);
  862. if (_model.Settings.filterMode == SongFilterMode.Favorites)
  863. {
  864. this._model.ProcessSongList(_beatUi.GetCurrentSelectedLevelPack());
  865. this.RefreshSongList();
  866. }
  867. }
  868. else
  869. {
  870. Logger.Info("Add {0} to editing playlist", songInfo.songName);
  871. _model.AddSongToEditingPlaylist(songInfo);
  872. }
  873. }
  874. RefreshAddFavoriteButton(songInfo.levelID);
  875. _model.Settings.Save();
  876. }
  877. /// <summary>
  878. /// Update GUI elements that show score saber data.
  879. /// </summary>
  880. public void RefreshScoreSaberData(IPreviewBeatmapLevel level)
  881. {
  882. Logger.Trace("RefreshScoreSaberData({0})", level.levelID);
  883. if (ScoreSaberDatabaseDownloader.ScoreSaberDataFile == null)
  884. {
  885. return;
  886. }
  887. BeatmapDifficulty difficulty = _beatUi.LevelDifficultyViewController.selectedDifficulty;
  888. string difficultyString = difficulty.ToString();
  889. if (difficultyString.Equals("ExpertPlus"))
  890. {
  891. difficultyString = "Expert+";
  892. }
  893. Logger.Debug(difficultyString);
  894. // Check for PP
  895. Logger.Debug("Checking if have info for song {0}", level.songName);
  896. var hash = CustomHelpers.GetSongHash(level.levelID);
  897. if (ScoreSaberDatabaseDownloader.ScoreSaberDataFile.SongHashToScoreSaberData.ContainsKey(hash))
  898. {
  899. Logger.Debug("Checking if have difficulty for song {0} difficulty {1}", level.songName, difficultyString);
  900. ScoreSaberSong scoreSaberSong = ScoreSaberDatabaseDownloader.ScoreSaberDataFile.SongHashToScoreSaberData[hash];
  901. ScoreSaberSongDifficultyStats scoreSaberSongDifficulty = scoreSaberSong.diffs.FirstOrDefault(x => String.Equals(x.diff, difficultyString));
  902. if (scoreSaberSongDifficulty != null)
  903. {
  904. Logger.Debug("Display pp for song.");
  905. double pp = scoreSaberSongDifficulty.pp;
  906. double star = scoreSaberSongDifficulty.star;
  907. UIBuilder.SetStatButtonText(_ppStatButton, String.Format("{0:0.#}", pp));
  908. UIBuilder.SetStatButtonText(_starStatButton, String.Format("{0:0.#}", star));
  909. }
  910. else
  911. {
  912. UIBuilder.SetStatButtonText(_ppStatButton, "NA");
  913. UIBuilder.SetStatButtonText(_starStatButton, "NA");
  914. }
  915. }
  916. else
  917. {
  918. UIBuilder.SetStatButtonText(_ppStatButton, "NA");
  919. UIBuilder.SetStatButtonText(_starStatButton, "NA");
  920. }
  921. Logger.Debug("Done refreshing score saber stats.");
  922. }
  923. /// <summary>
  924. /// Helper to refresh the NJS widget.
  925. /// </summary>
  926. /// <param name="beatmap"></param>
  927. private void RefreshNoteJumpSpeed(BeatmapDifficulty beatmap)
  928. {
  929. UIBuilder.SetStatButtonText(_njsStatButton, String.Format("{0}", beatmap.NoteJumpMovementSpeed()));
  930. }
  931. /// <summary>
  932. /// Update interactive state of the quick scroll buttons.
  933. /// </summary>
  934. private void RefreshQuickScrollButtons()
  935. {
  936. if (!_uiCreated)
  937. {
  938. return;
  939. }
  940. _pageUpFastButton.interactable = _beatUi.TableViewPageUpButton.interactable;
  941. _pageUpFastButton.gameObject.SetActive(_beatUi.TableViewPageUpButton.IsActive());
  942. _pageDownFastButton.interactable = _beatUi.TableViewPageDownButton.interactable;
  943. _pageDownFastButton.gameObject.SetActive(_beatUi.TableViewPageDownButton.IsActive());
  944. }
  945. /// <summary>
  946. /// Show the UI.
  947. /// </summary>
  948. public void Show()
  949. {
  950. Logger.Trace("Show SongBrowserUI()");
  951. this.SetVisibility(true);
  952. }
  953. /// <summary>
  954. /// Hide the UI.
  955. /// </summary>
  956. public void Hide()
  957. {
  958. Logger.Trace("Hide SongBrowserUI()");
  959. this.SetVisibility(false);
  960. }
  961. /// <summary>
  962. /// Handle showing or hiding UI logic.
  963. /// </summary>
  964. /// <param name="visible"></param>
  965. private void SetVisibility(bool visible)
  966. {
  967. // UI not created, nothing visible to hide...
  968. if (!_uiCreated)
  969. {
  970. return;
  971. }
  972. _ppStatButton.gameObject.SetActive(visible);
  973. _starStatButton.gameObject.SetActive(visible);
  974. _njsStatButton.gameObject.SetActive(visible);
  975. RefreshOuterUIState(visible == true ? UIState.Main : UIState.Disabled);
  976. _addFavoriteButton.gameObject.SetActive(visible);
  977. _deleteButton.gameObject.SetActive(visible);
  978. _pageUpFastButton.gameObject.SetActive(visible);
  979. _pageDownFastButton.gameObject.SetActive(visible);
  980. }
  981. /// <summary>
  982. /// Update the top UI state.
  983. /// Hides the outer ui, sort, and filter buttons depending on the state.
  984. /// </summary>
  985. private void RefreshOuterUIState(UIState state)
  986. {
  987. bool sortButtons = false;
  988. bool filterButtons = false;
  989. bool outerButtons = false;
  990. if (state == UIState.SortBy)
  991. {
  992. sortButtons = true;
  993. }
  994. else if (state == UIState.FilterBy)
  995. {
  996. filterButtons = true;
  997. }
  998. else if (state == UIState.Main)
  999. {
  1000. outerButtons = true;
  1001. }
  1002. _sortButtonGroup.ForEach(x => x.Button.gameObject.SetActive(sortButtons));
  1003. _filterButtonGroup.ForEach(x => x.Button.gameObject.SetActive(filterButtons));
  1004. _sortByButton.gameObject.SetActive(outerButtons);
  1005. _sortByDisplay.gameObject.SetActive(outerButtons);
  1006. _filterByButton.gameObject.SetActive(outerButtons);
  1007. _filterByDisplay.gameObject.SetActive(outerButtons);
  1008. _clearSortFilterButton.gameObject.SetActive(outerButtons);
  1009. _randomButton.gameObject.SetActive(outerButtons);
  1010. RefreshCurrentSelectionDisplay();
  1011. }
  1012. /// <summary>
  1013. /// Adjust the text field of the sort by and filter by displays.
  1014. /// </summary>
  1015. private void RefreshCurrentSelectionDisplay()
  1016. {
  1017. string sortByDisplay = null;
  1018. if (_model.Settings.sortMode == SongSortMode.Default)
  1019. {
  1020. sortByDisplay = "Title";
  1021. }
  1022. else
  1023. {
  1024. sortByDisplay = _model.Settings.sortMode.ToString();
  1025. }
  1026. _sortByDisplay.SetButtonText(sortByDisplay);
  1027. _filterByDisplay.SetButtonText(_model.Settings.filterMode.ToString());
  1028. }
  1029. /// <summary>
  1030. /// Helper to quickly refresh add to favorites button
  1031. /// </summary>
  1032. /// <param name="levelId"></param>
  1033. private void RefreshAddFavoriteButton(String levelId)
  1034. {
  1035. if (levelId == null)
  1036. {
  1037. _currentAddFavoriteButtonSprite = null;
  1038. }
  1039. else
  1040. {
  1041. if (_model.CurrentEditingPlaylistLevelIds.Contains(levelId))
  1042. {
  1043. _currentAddFavoriteButtonSprite = Base64Sprites.RemoveFromFavoritesIcon;
  1044. }
  1045. else
  1046. {
  1047. _currentAddFavoriteButtonSprite = Base64Sprites.AddToFavoritesIcon;
  1048. }
  1049. }
  1050. _addFavoriteButton.SetButtonIcon(_currentAddFavoriteButtonSprite);
  1051. }
  1052. /// <summary>
  1053. /// Adjust the UI colors.
  1054. /// </summary>
  1055. public void RefreshSortButtonUI()
  1056. {
  1057. // So far all we need to refresh is the sort buttons.
  1058. foreach (SongSortButton sortButton in _sortButtonGroup)
  1059. {
  1060. UIBuilder.SetButtonBorder(sortButton.Button, Color.white);
  1061. if (sortButton.SortMode == _model.Settings.sortMode)
  1062. {
  1063. if (this._model.Settings.invertSortResults)
  1064. {
  1065. UIBuilder.SetButtonBorder(sortButton.Button, Color.red);
  1066. }
  1067. else
  1068. {
  1069. UIBuilder.SetButtonBorder(sortButton.Button, Color.green);
  1070. }
  1071. }
  1072. }
  1073. // refresh filter buttons
  1074. foreach (SongFilterButton filterButton in _filterButtonGroup)
  1075. {
  1076. UIBuilder.SetButtonBorder(filterButton.Button, Color.white);
  1077. if (filterButton.FilterMode == _model.Settings.filterMode)
  1078. {
  1079. UIBuilder.SetButtonBorder(filterButton.Button, Color.green);
  1080. }
  1081. }
  1082. if (this._model.Settings.invertSortResults)
  1083. {
  1084. UIBuilder.SetButtonBorder(_sortByDisplay, Color.red);
  1085. }
  1086. else
  1087. {
  1088. UIBuilder.SetButtonBorder(_sortByDisplay, Color.green);
  1089. }
  1090. if (this._model.Settings.filterMode != SongFilterMode.None)
  1091. {
  1092. UIBuilder.SetButtonBorder(_filterByDisplay, Color.green);
  1093. }
  1094. else
  1095. {
  1096. UIBuilder.SetButtonBorder(_filterByDisplay, Color.white);
  1097. }
  1098. }
  1099. /// <summary>
  1100. ///
  1101. /// </summary>
  1102. public void RefreshSongList(bool scrollToLevel = true)
  1103. {
  1104. if (!_uiCreated)
  1105. {
  1106. return;
  1107. }
  1108. _beatUi.RefreshSongList(_model.LastSelectedLevelId);
  1109. }
  1110. /// <summary>
  1111. /// Helper for updating the model (which updates the song list)
  1112. /// </summary>
  1113. public void UpdateLevelDataModel()
  1114. {
  1115. try
  1116. {
  1117. Logger.Trace("UpdateLevelDataModel()");
  1118. // get a current beatmap characteristic...
  1119. if (_model.CurrentBeatmapCharacteristicSO == null && _uiCreated)
  1120. {
  1121. _model.CurrentBeatmapCharacteristicSO = _beatUi.BeatmapCharacteristicSelectionViewController.GetPrivateField<BeatmapCharacteristicSO>("_selectedBeatmapCharacteristic");
  1122. }
  1123. _model.UpdateLevelRecords();
  1124. bool didUpdateLevelPack = UpdateLevelPackSelection();
  1125. if (didUpdateLevelPack)
  1126. {
  1127. ProcessSongList();
  1128. }
  1129. }
  1130. catch (Exception e)
  1131. {
  1132. Logger.Exception("SongBrowser UI crashed trying to update the internal song lists: ", e);
  1133. }
  1134. }
  1135. /// <summary>
  1136. /// Logic for fixing BeatSaber's level pack selection bugs.
  1137. /// </summary>
  1138. public bool UpdateLevelPackSelection()
  1139. {
  1140. if (_uiCreated)
  1141. {
  1142. IBeatmapLevelPack currentSelected = _beatUi.GetCurrentSelectedLevelPack();
  1143. Logger.Debug("Current selected level pack: {0}", currentSelected);
  1144. if (String.IsNullOrEmpty(_model.Settings.currentLevelPackId))
  1145. {
  1146. if (currentSelected == null)
  1147. {
  1148. Logger.Debug("No level pack selected, acquiring the first available...");
  1149. var levelPackCollection = _beatUi.LevelPackViewController.GetPrivateField<IBeatmapLevelPackCollection>("_levelPackCollection");
  1150. currentSelected = levelPackCollection.beatmapLevelPacks[0];
  1151. }
  1152. }
  1153. else if (currentSelected == null || (currentSelected.packID != _model.Settings.currentLevelPackId))
  1154. {
  1155. Logger.Debug("Automatically selecting level pack: {0}", _model.Settings.currentLevelPackId);
  1156. // HACK - BeatSaber seems to always go back to OST1 internally.
  1157. // - Lets force it to the last pack id but not have SongBrowser functions fire.
  1158. // Turn off our event processing
  1159. _beatUi.LevelPackViewController.didSelectPackEvent -= _levelPackViewController_didSelectPackEvent;
  1160. _beatUi.LevelPacksTableView.didSelectPackEvent -= _levelPacksTableView_didSelectPackEvent;
  1161. var levelPack = _beatUi.GetLevelPackByPackId(_model.Settings.currentLevelPackId);
  1162. _beatUi.SelectLevelPack(_model.Settings.currentLevelPackId);
  1163. ProcessSongList();
  1164. _beatUi.LevelPackViewController.didSelectPackEvent += _levelPackViewController_didSelectPackEvent;
  1165. _beatUi.LevelPacksTableView.didSelectPackEvent += _levelPacksTableView_didSelectPackEvent;
  1166. return true;
  1167. }
  1168. }
  1169. return false;
  1170. }
  1171. }
  1172. }