Add referral Id's into hyperlinks

Author: BenjaminMichaelis

EssentialCSharp.Web/Areas/Identity/Data/EssentialCSharpWebContext.cs

@@ -4,7 +4,7 @@
 
 namespace EssentialCSharp.Web.Data;
 
-public class EssentialCSharpWebContext(DbContextOptions<EssentialCSharpWebContext> options) : IdentityDbContext<EssentialCSharpWebUser>(options)
+public class EssentialCSharpWebContext(DbContextOptions options) : IdentityDbContext<EssentialCSharpWebUser>(options)
 {
     protected override void OnModelCreating(ModelBuilder builder)
     {

EssentialCSharp.Web/Areas/Identity/Pages/Account/Login.cshtml.cs

@@ -1,13 +1,14 @@
 using System.ComponentModel.DataAnnotations;
 using EssentialCSharp.Web.Areas.Identity.Data;
+using EssentialCSharp.Web.Services.Referrals;
 using Microsoft.AspNetCore.Authentication;
 using Microsoft.AspNetCore.Identity;
 using Microsoft.AspNetCore.Mvc;
 using Microsoft.AspNetCore.Mvc.RazorPages;
 
 namespace EssentialCSharp.Web.Areas.Identity.Pages.Account;
 
-public class LoginModel(SignInManager<EssentialCSharpWebUser> signInManager, UserManager<EssentialCSharpWebUser> userManager, ILogger<LoginModel> logger) : PageModel
+public class LoginModel(SignInManager<EssentialCSharpWebUser> signInManager, UserManager<EssentialCSharpWebUser> userManager, ILogger<LoginModel> logger, IReferralService referralService) : PageModel
 {
     private InputModel? _Input;
     [BindProperty]
@@ -77,6 +78,8 @@ public async Task<IActionResult> OnPostAsync(string? returnUrl = null)
             if (foundUser is not null)
             {
                 result = await signInManager.PasswordSignInAsync(foundUser, Input.Password, Input.RememberMe, lockoutOnFailure: true);
+                // Call the referral service to get the referral ID and set it onto the user claim
+                _ = await referralService.GetReferralIdAsync(foundUser);
             }
             else
             {

EssentialCSharp.Web/Areas/Identity/Pages/Account/Manage/Index.cshtml

@@ -32,7 +32,7 @@ ViewData["ActivePage"] = ManageNavPages.Index;
                 <input asp-for="Input.PhoneNumber" class="form-control" placeholder="Please enter your phone number." />
                 <label asp-for="Input.PhoneNumber" class="form-label"></label>
                 <span asp-validation-for="Input.PhoneNumber" class="text-danger"></span>
-            </div> 
+            </div>
             <button id="update-profile-button" type="submit" class="w-100 btn btn-lg btn-primary">Save</button>
         </form>
     </div>

EssentialCSharp.Web/Controllers/HomeController.cs

@@ -7,33 +7,20 @@
 
 namespace EssentialCSharp.Web.Controllers;
 
-public class HomeController : Controller
+public class HomeController(ILogger<HomeController> logger, IWebHostEnvironment hostingEnvironment, ISiteMappingService siteMappingService, IHttpContextAccessor httpContextAccessor) : Controller
 {
-    private readonly IConfiguration _Configuration;
-    private readonly IWebHostEnvironment _HostingEnvironment;
-    private readonly ISiteMappingService _SiteMappingService;
-    private readonly ILogger<HomeController> _Logger;
-
-    public HomeController(ILogger<HomeController> logger, IWebHostEnvironment hostingEnvironment, ISiteMappingService siteMappingService, IConfiguration configuration)
-    {
-        _Logger = logger;
-        _HostingEnvironment = hostingEnvironment;
-        _SiteMappingService = siteMappingService;
-        _Configuration = configuration;
-    }
-
     public IActionResult Index(string key)
     {
         // if no key (default case), then load up home page
-        SiteMapping? siteMapping = _SiteMappingService.SiteMappings.Find(key);
+        SiteMapping? siteMapping = siteMappingService.SiteMappings.Find(key);
 
         if (string.IsNullOrEmpty(key))
         {
             return RedirectToAction(nameof(Home));
         }
         else if (siteMapping is not null)
         {
-            string filePath = Path.Combine(_HostingEnvironment.ContentRootPath, Path.Combine(siteMapping.PagePath));
+            string filePath = Path.Combine(hostingEnvironment.ContentRootPath, Path.Combine(siteMapping.PagePath));
             HtmlDocument doc = new();
             doc.Load(filePath);
             string headHtml = doc.DocumentNode.Element("html").Element("head").InnerHtml;
@@ -44,6 +31,8 @@ public IActionResult Index(string key)
             ViewBag.PreviousPage = FlipPage(siteMapping.ChapterNumber, siteMapping.PageNumber, false);
             ViewBag.HeadContents = headHtml;
             ViewBag.Contents = html;
+            // Set the referral Id for use in the front end if available
+            ViewBag.ReferralId = httpContextAccessor.HttpContext?.User?.Claims?.FirstOrDefault(f => f.Type == "ReferrerId")?.Value;
             return View();
         }
         else
@@ -84,19 +73,19 @@ public IActionResult Home()
     public IActionResult Guidelines()
     {
         ViewBag.PageTitle = "Coding Guidelines";
-        FileInfo fileInfo = new(Path.Combine(_HostingEnvironment.ContentRootPath, "Guidelines", "guidelines.json"));
+        FileInfo fileInfo = new(Path.Combine(hostingEnvironment.ContentRootPath, "Guidelines", "guidelines.json"));
         if (!fileInfo.Exists)
         {
             return RedirectToAction(nameof(Error), new { errorMessage = "Guidelines could not be found", statusCode = 404 });
         }
-        ViewBag.Guidelines = fileInfo.ReadGuidelineJsonFromInputDirectory(_Logger);
+        ViewBag.Guidelines = fileInfo.ReadGuidelineJsonFromInputDirectory(logger);
         ViewBag.GuidelinesUrl = Request.Path.Value;
         return View();
     }
 
     private string FlipPage(int currentChapter, int currentPage, bool next)
     {
-        if (_SiteMappingService.SiteMappings.Count == 0)
+        if (siteMappingService.SiteMappings.Count == 0)
         {
             return "";
         }
@@ -107,18 +96,18 @@ private string FlipPage(int currentChapter, int currentPage, bool next)
             page = 1;
         }
 
-        SiteMapping? siteMap = _SiteMappingService.SiteMappings.FirstOrDefault(f => f.ChapterNumber == currentChapter && f.PageNumber == currentPage + page);
+        SiteMapping? siteMap = siteMappingService.SiteMappings.FirstOrDefault(f => f.ChapterNumber == currentChapter && f.PageNumber == currentPage + page);
 
         if (siteMap is null)
         {
             if (next)
             {
-                siteMap = _SiteMappingService.SiteMappings.FirstOrDefault(f => f.ChapterNumber == currentChapter + 1 && f.PageNumber == 1);
+                siteMap = siteMappingService.SiteMappings.FirstOrDefault(f => f.ChapterNumber == currentChapter + 1 && f.PageNumber == 1);
             }
             else
             {
-                int? previousPage = _SiteMappingService.SiteMappings.LastOrDefault(f => f.ChapterNumber == currentChapter - 1)?.PageNumber;
-                siteMap = _SiteMappingService.SiteMappings.FirstOrDefault(f => f.ChapterNumber == currentChapter - 1 && f.PageNumber == previousPage);
+                int? previousPage = siteMappingService.SiteMappings.LastOrDefault(f => f.ChapterNumber == currentChapter - 1)?.PageNumber;
+                siteMap = siteMappingService.SiteMappings.FirstOrDefault(f => f.ChapterNumber == currentChapter - 1 && f.PageNumber == previousPage);
             }
             if (siteMap is null)
             {

EssentialCSharp.Web/Middleware/ReferralTrackingMiddleware.cs

@@ -2,7 +2,6 @@
 using System.Web;
 using EssentialCSharp.Web.Areas.Identity.Data;
 using EssentialCSharp.Web.Services.Referrals;
-using Microsoft.AspNetCore.Http.Extensions;
 using Microsoft.AspNetCore.Identity;
 
 namespace EssentialCSharp.Web.Middleware;
@@ -21,48 +20,13 @@ public async Task InvokeAsync(HttpContext context, IReferralService referralServ
         // Retrieve current referral Id for processing
         System.Collections.Specialized.NameValueCollection query = HttpUtility.ParseQueryString(context.Request.QueryString.Value!);
         string? referralId = query["rid"];
-        string? userReferralId;
-
         if (context.User is { Identity.IsAuthenticated: true } claimsUser)
         {
-            if (!string.IsNullOrWhiteSpace(referralId))
-            {
-                await TrackReferralAsync(referralService, referralId, claimsUser);
-            }
-
-            // Add the referralId to the request context if it exists on a user
-            EssentialCSharpWebUser? user = await userManager.GetUserAsync(claimsUser);
-            if (user is not null)
-            {
-                userReferralId = await referralService.GetReferralIdAsync(user.Id);
-
-                if (!string.IsNullOrWhiteSpace(userReferralId) && (string.IsNullOrWhiteSpace(query["rid"]) || query["rid"] != userReferralId))
-                {
-                    query.Remove("rid");
-                    query.Add("rid", userReferralId);
-                    var builder = new UriBuilder(context.Request.GetEncodedUrl())
-                    {
-                        Query = query.ToString()
-                    };
-                    context.Response.Redirect(builder.ToString());
-                    return;
-                }
-            }
+            await TrackReferralAsync(referralService, referralId, claimsUser);
         }
         else
         {
-
-            if (!string.IsNullOrWhiteSpace(referralId))
-            {
-                await TrackReferralAsync(referralService, referralId, null);
-                query.Remove("rid");
-                var builder = new UriBuilder(context.Request.GetEncodedUrl())
-                {
-                    Query = query.ToString()
-                };
-                context.Response.Redirect(builder.ToString());
-                return;
-            }
+            await TrackReferralAsync(referralService, referralId, null);
         }
 
         await _Next(context);

EssentialCSharp.Web/Services/Referrals/IReferralService.cs

@@ -1,9 +1,11 @@
 using System.Security.Claims;
+using EssentialCSharp.Web.Areas.Identity.Data;
 
 namespace EssentialCSharp.Web.Services.Referrals;
 
 public interface IReferralService
 {
     Task TrackReferralAsync(string referralId, ClaimsPrincipal? user);
     Task<string?> GetReferralIdAsync(string userId);
+    Task<string?> GetReferralIdAsync(EssentialCSharpWebUser? user);
 }

EssentialCSharp.Web/Services/Referrals/ReferralService.cs

@@ -12,6 +12,11 @@ public class ReferralService(EssentialCSharpWebContext dbContext, SqidsEncoder<i
     public async Task<string?> GetReferralIdAsync(string userId)
     {
         EssentialCSharpWebUser? user = await userManager.FindByIdAsync(userId);
+        return await GetReferralIdAsync(user);
+    }
+
+    public async Task<string?> GetReferralIdAsync(EssentialCSharpWebUser? user)
+    {
         if (user is null)
         {
             return null;

EssentialCSharp.Web/Views/Shared/_Layout.cshtml

@@ -273,6 +273,7 @@
             PREVIOUS_PAGE = @Json.Serialize(ViewBag.PreviousPage)
             NEXT_PAGE = @Json.Serialize(ViewBag.NextPage)
             TOC_DATA = @Json.Serialize(tocData)
+            REFERRAL_ID = @Json.Serialize(ViewBag.ReferralId)
     </script>
 
     @* Recursive vue component template for rendering the table of contents. *@

EssentialCSharp.Web/wwwroot/css/styles.css

@@ -821,9 +821,9 @@ details > summary::-webkit-details-marker {
 
 .nav-pills .nav-link.active,
 .nav-pills .show > .nav-link {
-    color: #fff;
-    background-color: #1b6ec2;
-    border-color: #1861ac;
+  color: #fff;
+  background-color: #1b6ec2;
+  border-color: #1861ac;
 }
 
 button.accept-policy {

EssentialCSharp.Web/wwwroot/js/site.js

@@ -125,8 +125,13 @@ const app = createApp({
         const snackbarColor = ref();
 
         function copyToClipboard(copyText) {
+            let url = window.location.origin + "/" + copyText;
+            let referralId = REFERRAL_ID;
+            if (referralId && referralId.trim()) {
+                url = addQueryParam(url, 'rid', referralId);
+            }
             navigator.clipboard
-                .writeText(window.location.origin + "/" + copyText)
+                .writeText(url)
                 .then(
                     function () {
                         /* Success */
@@ -151,6 +156,11 @@ const app = createApp({
             );
         }
 
+        function addQueryParam(url, key, value) {
+            var separator = url.includes('?') ? '&' : '?';
+            return url + separator + key + '=' + encodeURIComponent(value);
+        }
+
         function goToPrevious() {
             let previousPage = PREVIOUS_PAGE;
             if (previousPage !== null) {