subs

Plugins/BTCPayServer.Plugins.Subscriptions/SubscriptionController.cs

@@ -0,0 +1,182 @@
+using System;
+using System.Collections.Generic;
+using System.Threading.Tasks;
+using BTCPayServer;
+using BTCPayServer.Abstractions.Constants;
+using BTCPayServer.Abstractions.Extensions;
+using BTCPayServer.Client;
+using BTCPayServer.Client.Models;
+using BTCPayServer.Data;
+using BTCPayServer.Models;
+using BTCPayServer.Plugins.Subscriptions;
+using BTCPayServer.Services.Apps;
+using BTCPayServer.Services.PaymentRequests;
+using Microsoft.AspNetCore.Authorization;
+using Microsoft.AspNetCore.Mvc;
+using Newtonsoft.Json.Linq;
+using PaymentRequestData = BTCPayServer.Data.PaymentRequestData;
+
+namespace BTCPayServer.Plugins.Subscriptions;
+
+[Authorize(AuthenticationSchemes = AuthenticationSchemes.Cookie)]
+[Authorize(Policy = Policies.CanModifyStoreSettings, AuthenticationSchemes = AuthenticationSchemes.Cookie)]
+public class SubscriptionController : Controller
+{
+    private readonly AppService _appService;
+    private readonly PaymentRequestRepository _paymentRequestRepository;
+    private readonly SubscriptionService _subscriptionService;
+
+    public SubscriptionController(AppService appService,
+        PaymentRequestRepository paymentRequestRepository, SubscriptionService subscriptionService)
+    {
+        _appService = appService;
+        _paymentRequestRepository = paymentRequestRepository;
+        _subscriptionService = subscriptionService;
+    }
+
+    [AllowAnonymous]
+    [HttpGet("~/plugins/subscription/{appId}")]
+    public async Task<IActionResult> View(string appId)
+    {
+        var app = await _appService.GetApp(appId, SubscriptionApp.AppType, true, false);
+
+        if (app == null)
+            return NotFound();
+        var ss = app.GetSettings<SubscriptionAppSettings>();
+        ss.SubscriptionName = app.Name;
+        ViewData["StoreBranding"] = new StoreBrandingViewModel(app.StoreData.GetStoreBlob());
+        return View(ss);
+    }
+
+    [AllowAnonymous]
+    [HttpGet("~/plugins/subscription/{appId}/{id}")]
+    public async Task<IActionResult> ViewSubscription(string appId, string id)
+    {
+        var app = await _appService.GetApp(appId, SubscriptionApp.AppType, true, false);
+
+        if (app == null)
+            return NotFound();
+        var ss = app.GetSettings<SubscriptionAppSettings>();
+        ss.SubscriptionName = app.Name;
+        if (!ss.Subscriptions.TryGetValue(id, out _))
+        {
+            return NotFound();
+        }
+
+        ViewData["StoreBranding"] = new StoreBrandingViewModel(app.StoreData.GetStoreBlob());
+
+        return View(ss);
+    }
+
+    [AllowAnonymous]
+    [HttpGet("~/plugins/subscription/{appId}/{id}/reactivate")]
+    public async Task<IActionResult> Reactivate(string appId, string id)
+    {
+        var pr = await _subscriptionService.ReactivateSubscription(appId, id);
+        if (pr == null)
+            return NotFound();
+        return RedirectToAction("ViewPaymentRequest", "UIPaymentRequest", new {payReqId = pr.Id});
+    }
+
+
+    [AllowAnonymous]
+    [HttpGet("~/plugins/subscription/{appId}/subscribe")]
+    public async Task<IActionResult> Subscribe(string appId)
+    {
+        var app = await _appService.GetApp(appId, SubscriptionApp.AppType, false, false);
+
+        if (app == null)
+            return NotFound();
+        var ss = app.GetSettings<SubscriptionAppSettings>();
+        ss.SubscriptionName = app.Name;
+
+        var pr = new PaymentRequestData()
+        {
+            StoreDataId = app.StoreDataId,
+            Archived = false,
+            Status = Client.Models.PaymentRequestData.PaymentRequestStatus.Pending
+        };
+        pr.SetBlob(new CreatePaymentRequestRequest()
+        {
+            Amount = ss.Price,
+            Currency = ss.Currency,
+            ExpiryDate = DateTimeOffset.UtcNow.AddDays(1),
+            Description = ss.Description,
+            Title = ss.SubscriptionName,
+            FormId = ss.FormId,
+            AllowCustomPaymentAmounts = false,
+            AdditionalData = new Dictionary<string, JToken>()
+            {
+                {"source", JToken.FromObject("subscription")},
+                {"appId", JToken.FromObject(appId)},
+                {"url", HttpContext.Request.GetAbsoluteRoot()}
+            },
+        });
+
+        pr = await _paymentRequestRepository.CreateOrUpdatePaymentRequest(pr);
+
+        return RedirectToAction("ViewPaymentRequest", "UIPaymentRequest", new {payReqId = pr.Id});
+    }
+
+
+    [HttpGet("~/plugins/subscription/{appId}/update")]
+    public async Task<IActionResult> Update(string appId)
+    {
+        var app = await _appService.GetApp(appId, SubscriptionApp.AppType, false, true);
+
+        if (app == null)
+            return NotFound();
+        ViewData["archived"] = app.Archived;
+        var ss = app.GetSettings<SubscriptionAppSettings>();
+        ss.SubscriptionName = app.Name;
+
+        return View(ss);
+    }
+
+    [HttpPost("~/plugins/subscription/{appId}/update")]
+    public async Task<IActionResult> Update(string appId, SubscriptionAppSettings vm)
+    {
+        var app = await _appService.GetApp(appId, SubscriptionApp.AppType, true, true);
+
+        if (string.IsNullOrEmpty(vm.Currency))
+        {
+            vm.Currency = app.StoreData.GetStoreBlob().DefaultCurrency;
+            ModelState.Remove(nameof(vm.Currency));
+        }
+
+        if (string.IsNullOrEmpty(vm.Currency))
+        {
+            ModelState.AddModelError(nameof(vm.Currency), "Currency is required");
+        }
+
+        if (string.IsNullOrEmpty(vm.SubscriptionName))
+        {
+            ModelState.AddModelError(nameof(vm.SubscriptionName), "Subscription name is required");
+        }
+
+        if (vm.Price <= 0)
+        {
+            ModelState.AddModelError(nameof(vm.Price), "Price must be greater than 0");
+        }
+
+        if (vm.DurationDays <= 0)
+        {
+            ModelState.AddModelError(nameof(vm.DurationDays), "Duration must be greater than 0");
+        }
+
+
+        ViewData["archived"] = app.Archived;
+        if (!ModelState.IsValid)
+        {
+            return View(vm);
+        }
+
+        var old = app.GetSettings<SubscriptionAppSettings>();
+        vm.Subscriptions = old.Subscriptions;
+        app.SetSettings(vm);
+        app.Name = vm.SubscriptionName;
+        await _appService.UpdateOrCreateApp(app);
+        TempData["SuccessMessage"] = "Subscription settings modified";
+        return RedirectToAction(nameof(Update), new {appId});
+    }
+}