// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using Microsoft.AspNetCore.Components;
using Microsoft.AspNetCore.Components.Web;
 
namespace BasicTestApp;
 
public class PreserveStateService : IDisposable
{
    private readonly PersistentComponentState _componentApplicationState;
    private PersistingComponentStateSubscription _persistingSubscription;
 
    private ServiceState _state = new();
 
    public PreserveStateService(PersistentComponentState componentApplicationState)
    {
        _componentApplicationState = componentApplicationState;
        _persistingSubscription = _componentApplicationState.RegisterOnPersisting(PersistState, RenderMode.InteractiveAuto);
        TryRestoreState();
    }
 
    public Guid Guid => _state.TheState;
 
    private void TryRestoreState()
    {
        if (_componentApplicationState.TryTakeFromJson<ServiceState>("Service", out var state))
        {
            _state = state;
        }
        else
        {
            _state = new ServiceState { TheState = Guid.NewGuid() };
        }
    }
 
    public void NewState() => _state = new ServiceState { TheState = Guid.NewGuid() };
 
    private Task PersistState()
    {
        _componentApplicationState.PersistAsJson("Service", _state);
        return Task.CompletedTask;
    }
 
    public void Dispose() => _persistingSubscription.Dispose();
 
    private class ServiceState
    {
        public Guid TheState { get; set; }
    }
}