Umbraco-CMS/src/Umbraco.Web.BackOffice/Controllers/PublishedSnapshotCacheStatusController.cs

using Microsoft.AspNetCore.Mvc;
using Umbraco.Cms.Core;
using Umbraco.Cms.Core.Cache;
using Umbraco.Cms.Core.PublishedCache;
using Umbraco.Cms.Web.Common.Attributes;
using Umbraco.Extensions;

namespace Umbraco.Cms.Web.BackOffice.Controllers;

[PluginController(Constants.Web.Mvc.BackOfficeApiArea)]
public class PublishedSnapshotCacheStatusController : UmbracoAuthorizedApiController
{
    private readonly DistributedCache _distributedCache;
    private readonly IPublishedSnapshotService _publishedSnapshotService;
    private readonly IPublishedSnapshotStatus _publishedSnapshotStatus;

    /// <summary>
    ///     Initializes a new instance of the <see cref="PublishedSnapshotCacheStatusController" /> class.
    /// </summary>
    public PublishedSnapshotCacheStatusController(
        IPublishedSnapshotService publishedSnapshotService,
        IPublishedSnapshotStatus publishedSnapshotStatus,
        DistributedCache distributedCache)
    {
        _publishedSnapshotService = publishedSnapshotService ??
                                    throw new ArgumentNullException(nameof(publishedSnapshotService));
        _publishedSnapshotStatus = publishedSnapshotStatus;
        _distributedCache = distributedCache;
    }

    /// <summary>
    ///     Rebuilds the Database cache
    /// </summary>
    [HttpPost]
    public string RebuildDbCache()
    {
        _publishedSnapshotService.Rebuild();
        return _publishedSnapshotStatus.GetStatus();
    }

    /// <summary>
    ///     Gets a status report
    /// </summary>
    [HttpGet]
    public string GetStatus() => _publishedSnapshotStatus.GetStatus();

    /// <summary>
    ///     Cleans up unused snapshots
    /// </summary>
    [HttpGet]
    public async Task<string> Collect()
    {
        GC.Collect();
        await _publishedSnapshotService.CollectAsync();
        return _publishedSnapshotStatus.GetStatus();
    }

    [HttpPost]
    public void ReloadCache() => _distributedCache.RefreshAllPublishedSnapshot();
}