nft-faucet/NftFaucet/Services/StateRepository.cs

using NftFaucet.Models.Dto;
using NftFaucet.Models.State;
using NftFaucet.Plugins;
using NftFaucet.Plugins.ProviderPlugins;
using NftFaucet.Plugins.UploadPlugins;
using TG.Blazor.IndexedDB;

namespace NftFaucet.Services;

public class StateRepository
{
    private readonly IndexedDBManager _dbManager;
    private readonly Mapper _mapper;
    private const string AppStateStoreName = "AppState";
    private const string TokensStoreName = "Tokens";
    private const string UploadLocationsStoreName = "UploadLocations";
    private const string ProviderStatesStoreName = "ProviderStates";
    private const string UploaderStatesStoreName = "UploaderStates";

    public StateRepository(IndexedDBManager dbManager, Mapper mapper)
    {
        _dbManager = dbManager;
        _mapper = mapper;
    }

    public async Task SaveAppState(ScopedAppState appState)
    {
        var appStateDto = _mapper.ToDto(appState) ?? new AppStateDto();
        var record = new StoreRecord<AppStateDto>
        {
            Storename = AppStateStoreName,
            Data = appStateDto,
        };

        var existingAppStateDto = await GetFirst<AppStateDto>(AppStateStoreName);
        if (existingAppStateDto == null)
        {
            await _dbManager.AddRecord(record);
        }
        else
        {
            await _dbManager.UpdateRecord(record);
        }
    }

    public async Task LoadAppState(ScopedAppState appState)
    {
        var appStateDto = await GetFirst<AppStateDto>(AppStateStoreName) ?? new AppStateDto();
        var loadedAppState = _mapper.ToDomain(appStateDto);
        appState.LoadUserStorage(loadedAppState.UserStorage);
    }

    public async Task SaveToken(IToken token)
    {
        var tokenDto = _mapper.ToDto(token) ?? new TokenDto();
        var record = new StoreRecord<TokenDto>
        {
            Storename = TokensStoreName,
            Data = tokenDto,
        };

        var existingTokenDto = await _dbManager.GetRecordById<Guid, TokenDto>(TokensStoreName, tokenDto.Id);
        if (existingTokenDto == null)
        {
            await _dbManager.AddRecord(record);
        }
        else
        {
            await _dbManager.UpdateRecord(record);
        }
    }

    public async Task<IToken[]> LoadTokens()
    {
        var existingTokens = await _dbManager.GetRecords<TokenDto>(TokensStoreName);
        if (existingTokens == null || existingTokens.Count == 0)
            return Array.Empty<IToken>();

        return existingTokens.Select(_mapper.ToDomain).Where(x => x != null).ToArray();
    }

    public async Task SaveUploadLocation(ITokenUploadLocation uploadLocation)
    {
        var uploadLocationDto = _mapper.ToDto(uploadLocation) ?? new UploadLocationDto();
        var record = new StoreRecord<UploadLocationDto>
        {
            Storename = UploadLocationsStoreName,
            Data = uploadLocationDto,
        };

        var existingUploadLocationDto = await _dbManager.GetRecordById<Guid, TokenDto>(UploadLocationsStoreName, uploadLocationDto.Id);
        if (existingUploadLocationDto == null)
        {
            await _dbManager.AddRecord(record);
        }
        else
        {
            await _dbManager.UpdateRecord(record);
        }
    }

    public async Task<ITokenUploadLocation[]> LoadUploadLocations()
    {
        var existingUploadLocations = await _dbManager.GetRecords<UploadLocationDto>(UploadLocationsStoreName);
        if (existingUploadLocations == null || existingUploadLocations.Count == 0)
            return Array.Empty<ITokenUploadLocation>();

        return existingUploadLocations.Select(_mapper.ToDomain).ToArray();
    }

    public async Task SaveProviderState(IProvider provider)
    {
        var state = await provider.GetState();
        var stateDto = new ProviderStateDto
        {
            Id = provider.Id,
            State = state,
        };
        var record = new StoreRecord<ProviderStateDto>
        {
            Storename = ProviderStatesStoreName,
            Data = stateDto,
        };

        var existingStateDto = await _dbManager.GetRecordById<Guid, ProviderStateDto>(ProviderStatesStoreName, stateDto.Id);
        if (existingStateDto == null)
        {
            await _dbManager.AddRecord(record);
        }
        else
        {
            await _dbManager.UpdateRecord(record);
        }
    }

    public async Task<UploaderStateDto[]> LoadUploaderStates()
    {
        var existingUploaderStates = await _dbManager.GetRecords<UploaderStateDto>(UploaderStatesStoreName);
        if (existingUploaderStates == null || existingUploaderStates.Count == 0)
            return Array.Empty<UploaderStateDto>();

        return existingUploaderStates.ToArray();
    }
    
    public async Task SaveUploaderState(IUploader uploader)
    {
        var state = await uploader.GetState();
        var stateDto = new UploaderStateDto
        {
            Id = uploader.Id,
            State = state,
        };
        var record = new StoreRecord<UploaderStateDto>
        {
            Storename = UploaderStatesStoreName,
            Data = stateDto,
        };

        var existingStateDto = await _dbManager.GetRecordById<Guid, UploaderStateDto>(UploaderStatesStoreName, stateDto.Id);
        if (existingStateDto == null)
        {
            await _dbManager.AddRecord(record);
        }
        else
        {
            await _dbManager.UpdateRecord(record);
        }
    }

    public async Task<ProviderStateDto[]> LoadProviderStates()
    {
        var existingProviderStates = await _dbManager.GetRecords<ProviderStateDto>(ProviderStatesStoreName);
        if (existingProviderStates == null || existingProviderStates.Count == 0)
            return Array.Empty<ProviderStateDto>();

        return existingProviderStates.ToArray();
    }

    private async Task<T> GetFirst<T>(string storeName)
    {
        var existingRecords = await _dbManager.GetRecords<T>(storeName);
        if (existingRecords == null || existingRecords.Count == 0)
            return default;

        return existingRecords.FirstOrDefault();
    }
}