From b0bd41a4f0d292abdb9bd86a0dea8fd597c95a27 Mon Sep 17 00:00:00 2001
From: Lilian <lilian@blahaj.space>
Date: Wed, 11 Sep 2024 03:20:13 +0200
Subject: [PATCH] [frontend/components] Add state reconstruction to note search

---
 .../Components/SearchComponent.razor          | 56 ++++++++++++++++---
 .../Components/SearchComponent.razor.js       |  7 +++
 2 files changed, 55 insertions(+), 8 deletions(-)
 create mode 100644 Iceshrimp.Frontend/Components/SearchComponent.razor.js

diff --git a/Iceshrimp.Frontend/Components/SearchComponent.razor b/Iceshrimp.Frontend/Components/SearchComponent.razor
index c38c4b8e..6e3dd47f 100644
--- a/Iceshrimp.Frontend/Components/SearchComponent.razor
+++ b/Iceshrimp.Frontend/Components/SearchComponent.razor
@@ -6,9 +6,12 @@
 @using Iceshrimp.Frontend.Localization
 @using Iceshrimp.Shared.Schemas.Web
 @using Microsoft.Extensions.Localization
+@using Iceshrimp.Frontend.Core.Services.StateServicePatterns
 @inject IStringLocalizer<Localization> Loc;
 @inject ApiService                     Api;
 @inject NavigationManager              Navigation;
+@inject StateService                   StateService;
+@inject IJSRuntime                     Js;
 
 <div class="search">
     <input @bind="SearchString" class="input"/>
@@ -23,7 +26,7 @@
 
 @if (_resultType == ResultType.Notes)
 {
-    <div class="note-results">
+    <div class="note-results" @ref="Scroller">
         @if (NoteSearchInstance?.NoteResponses != null)
         {
             foreach (var note in NoteSearchInstance?.NoteResponses!)
@@ -61,11 +64,37 @@
 }
 
 @code {
-    private string         SearchString       { get; set; } = "";
-    private SearchTypeEnum SearchType         { get; set; } = SearchTypeEnum.Note;
-    private NoteSearch?    NoteSearchInstance { get; set; }
-    private UserSearch?    UserSearchInstance { get; set; }
-    private ResultType     _resultType;
+    private string                      SearchString       { get; set; } = "";
+    private SearchTypeEnum              SearchType         { get; set; } = SearchTypeEnum.Note;
+    private NoteSearch?                 NoteSearchInstance { get; set; }
+    private UserSearch?                 UserSearchInstance { get; set; }
+    private IJSInProcessObjectReference Module             { get; set; } = null!;
+    private ElementReference            Scroller           { get; set; }
+    private ResultType                  _resultType;
+    private bool                        _setScroll;
+    private float                       _scrollTop;
+
+    protected override async Task OnInitializedAsync()
+    {
+        Module = (IJSInProcessObjectReference)await Js.InvokeAsync<IJSObjectReference>("import", "/Components/SearchComponent.razor.js");
+        var state = StateService.Search.GetState();
+        if (state != null)
+        {
+            NoteSearchInstance = new NoteSearch(state.SearchString, Api, state.SearchResults);
+            _resultType        = ResultType.Notes;
+            _scrollTop         = state.ScrollTop;
+            _setScroll         = true;
+        }
+    }
+
+    protected override void OnAfterRender(bool firstRender)
+    {
+        if (_setScroll)
+        {
+            SetScrollY(_scrollTop);
+            _setScroll = false;
+        }
+    }
 
     private async Task Search()
     {
@@ -118,6 +147,7 @@
 
     private void OpenNote(string id)
     {
+        StateService.Search.SetState(new SearchState { SearchResults = NoteSearchInstance!.NoteResponses, ScrollTop = GetScrollY(), SearchString = NoteSearchInstance!.SearchString });
         Navigation.NavigateTo($"/notes/{id}");
     }
 
@@ -128,6 +158,16 @@
         Navigation.NavigateTo($"/{username}");
     }
 
+    private float GetScrollY()
+    {
+        return Module.Invoke<float>("GetScrollY");
+    }
+
+    private void SetScrollY(float scrollTop)
+    {
+        Module.InvokeVoid("SetScrollY", scrollTop);
+    }
+
     private enum ResultType
     {
         Default,
@@ -174,7 +214,7 @@
     {
         internal required string             MinId        = noteResponses.Last().Id;
         internal required string             MaxId        = noteResponses.First().Id;
-        internal required string             SearchString = searchString;
+        public required   string             SearchString = searchString;
         internal required ApiService         Api          = api;
         public required   List<NoteResponse> NoteResponses { get; init; } = noteResponses;
         public            bool               SearchComplete = false;
@@ -195,4 +235,4 @@
             }
         }
     }
-}
\ No newline at end of file
+}
diff --git a/Iceshrimp.Frontend/Components/SearchComponent.razor.js b/Iceshrimp.Frontend/Components/SearchComponent.razor.js
new file mode 100644
index 00000000..5ddcf9cf
--- /dev/null
+++ b/Iceshrimp.Frontend/Components/SearchComponent.razor.js
@@ -0,0 +1,7 @@
+export function GetScrollY(){
+    return window.scrollY;
+}
+
+export function SetScrollY(number){
+    window.scroll(window.scrollX, number);
+}
\ No newline at end of file