btcpayserver/BTCPayServer/Forms/UIFormsController.cs

#nullable enable
using System;
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
using System.Threading.Tasks;
using BTCPayServer.Abstractions.Constants;
using BTCPayServer.Abstractions.Extensions;
using BTCPayServer.Abstractions.Form;
using BTCPayServer.Abstractions.Models;
using BTCPayServer.Client;
using BTCPayServer.Client.Models;
using BTCPayServer.Controllers;
using BTCPayServer.Data;
using BTCPayServer.Data.Data;
using BTCPayServer.Forms.Models;
using BTCPayServer.Models;
using BTCPayServer.Services.Stores;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;

namespace BTCPayServer.Forms;

[Authorize(Policy = Policies.CanModifyStoreSettings, AuthenticationSchemes = AuthenticationSchemes.Cookie)]
public class UIFormsController : Controller
{
    [AllowAnonymous]
    [HttpGet("~/forms/{formId}")]
    [HttpPost("~/forms")]
    public IActionResult ViewPublicForm(string? formId, string? redirectUrl)
    {
        FormData? formData = string.IsNullOrEmpty(formId) ? null : GetFormData(formId);
        if (formData == null)
        {
            return string.IsNullOrEmpty(redirectUrl)
                ? NotFound()
                : Redirect(redirectUrl);
        }
        
        return View("View", new FormViewModel() { FormData = formData, RedirectUrl = redirectUrl });
    }

    [AllowAnonymous]
    [HttpPost("~/forms/{formId}")]
    public IActionResult SubmitForm(
        string formId, string? redirectUrl,
        [FromServices] StoreRepository storeRepository,  
        [FromServices] UIInvoiceController invoiceController)
    {
        var formData = GetFormData(formId);
        if (formData?.Config is null)
        {
            return NotFound();
        }

        var dbForm = Form.Parse(formData.Config);
        dbForm.ApplyValuesFromForm(Request.Form);
        Dictionary<string, object> data = dbForm.GetValues();
        
        // With redirect, the form comes from another entity that we need to send the data back to
        if (!string.IsNullOrEmpty(redirectUrl))
        {
            return View("PostRedirect", new PostRedirectViewModel
            {
                FormUrl = redirectUrl,
                FormParameters =
                {
                    { "formId", formId },
                    { "formData", JsonConvert.SerializeObject(data) }
                }
            });
        }

        return NotFound();
    }

    private FormData? GetFormData(string id)
    {
        FormData? form = id switch
        {
            { } formId when formId == GenericFormOption.Address.ToString() => new FormData
            {
                Config = FormDataService.StaticFormAddress.ToString(),
                Id = GenericFormOption.Address.ToString(),
                Name = "Provide your address",
            },
            { } formId when formId == GenericFormOption.Email.ToString() => new FormData
            {
                Config = FormDataService.StaticFormEmail.ToString(),
                Id = GenericFormOption.Email.ToString(),
                Name = "Provide your email address",
            },
            _ => null
        };
        return form;
    }
}