using System.Buffers;
using System.Web;
using Dpz.Core.Backup;
using Dpz.Core.WebApi.Models.EventRequest;
using Dpz.Core.WebApi.Models.Request;
using MediatR;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Primitives;

namespace Dpz.Core.WebApi.Controllers;

/// <summary>
/// 系统设置
/// </summary>
[ApiController, Route("api/[controller]")]
public class SysController(
    IBackupRecordService backupRecordService,
    IRestoreService restoreService,
    ISafeFileService safeFileService,
    ILogger<SysController> logger,
    IMediator mediator
) : ControllerBase
{
    /// <summary>
    /// 还原最新数据
    /// </summary>
    /// <returns></returns>
    [HttpPost("restore"), Authorize(Policy = ApiPermission.System)]
    [ProducesResponseType(StatusCodes.Status204NoContent)]
    [ProducesResponseType(StatusCodes.Status401Unauthorized)]
    public async Task<IActionResult> Restore([FromBody] RestoreRequest request)
    {
        try
        {
            var backupRecord = await backupRecordService.GetLastNewBackupRecordAsync(
                request.Database
            );
            restoreService.DownloadBackupFileAsync = DownloadBackupFileAsync;
            await restoreService.RestoreAsync(request.ConnectionString, backupRecord);
        }
        catch (Exception e)
        {
            logger.LogError(e, "restore fail");
            return BadRequest();
        }

        return NoContent();
    }

    private async Task<string> DownloadBackupFileAsync(string backupUrl)
    {
        var stream = await safeFileService.DownloadAsync(backupUrl);
        var restorePath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "restore");
        if (!Directory.Exists(restorePath))
        {
            Directory.CreateDirectory(restorePath);
        }

        var restoreFilePath = Path.Combine(restorePath, $"{DateTime.Now:yyyyMMdd}.zip");
        await using var fileStream = new FileStream(
            restoreFilePath,
            FileMode.Create,
            FileAccess.Write,
            FileShare.Write,
            1 << 12
        );
        await stream.CopyToAsync(fileStream);
        await fileStream.FlushAsync();

        return restoreFilePath;
    }

    /// <summary>
    /// 接收预处理回调
    /// </summary>
    /// <returns></returns>
    [HttpPost("receive/upyun/notify")]
    [ProducesResponseType(StatusCodes.Status204NoContent)]
    [ProducesResponseType(StatusCodes.Status400BadRequest)]
    [ProducesResponseType(StatusCodes.Status401Unauthorized)]
    public async Task<IActionResult> ReceiveUpyunNotification()
    {
        return await mediator.Send(new UpyunNotificationRequest { Request = Request });
    }
}