Skip to content

Commit 2d7125e

Browse files
Apply suggestions from code review
1 parent cb47d43 commit 2d7125e

File tree

5 files changed

+27
-34
lines changed

5 files changed

+27
-34
lines changed

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

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,12 @@
77

88
<h3>@ViewData["Title"]</h3>
99

10-
<div class="row">
11-
<div class="col-md-6">
12-
<p>
13-
You have <strong>@Model.ReferralCount</strong> referrals.
14-
</p>
15-
</div>
10+
<div>
11+
<p>
12+
You have <strong>@Model.ReferralCount</strong> referrals.
13+
</p>
14+
</div>
1615

17-
@section Scripts {
18-
<partial name="_ValidationScriptsPartial" />
19-
}
16+
@section Scripts {
17+
<partial name="_ValidationScriptsPartial" />
18+
}

EssentialCSharp.Web/Areas/Identity/Pages/Account/Manage/Referrals.cshtml.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ public class ReferralsDataModel : PageModel
1010
private readonly UserManager<EssentialCSharpWebUser> _UserManager;
1111
private readonly ILogger<ReferralsDataModel> _Logger;
1212

13-
public int ReferralCount { get; set; }
13+
public int ReferralCount { get; private set; }
1414

1515
public ReferralsDataModel(
1616
UserManager<EssentialCSharpWebUser> userManager,
Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,17 @@
11
using System.Security.Claims;
22

3-
namespace EssentialCSharp.Web.Extensions
3+
namespace EssentialCSharp.Web.Extensions;
4+
5+
public static class ClaimsExtensions
46
{
5-
public static class ClaimsExtensions
7+
public static string? GetReferrerId(this ClaimsPrincipal claimsPrincipal)
68
{
7-
public static string? GetReferrerId(this ClaimsPrincipal claimsPrincipal)
8-
{
9-
return claimsPrincipal.FindFirstValue(ReferrerIdClaimType);
10-
}
9+
return claimsPrincipal.FindFirstValue(ReferrerIdClaimType);
10+
}
1111

12-
public static string? GetReferrerId(this IList<Claim> claims)
13-
{
14-
return claims.FirstOrDefault(claim => claim.Type == ReferrerIdClaimType)?.Value;
15-
}
16-
public const string ReferrerIdClaimType = "ReferrerId";
12+
public static string? GetReferrerId(this IList<Claim> claims)
13+
{
14+
return claims.FirstOrDefault(claim => claim.Type == ReferrerIdClaimType)?.Value;
1715
}
16+
public const string ReferrerIdClaimType = "ReferrerId";
1817
}

EssentialCSharp.Web/Middleware/ReferralTrackingMiddleware.cs

Lines changed: 7 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -20,23 +20,18 @@ public async Task InvokeAsync(HttpContext context, IReferralService referralServ
2020
// Retrieve current referral Id for processing
2121
System.Collections.Specialized.NameValueCollection query = HttpUtility.ParseQueryString(context.Request.QueryString.Value!);
2222
string? referralId = query["rid"];
23-
if (context.User is { Identity.IsAuthenticated: true } claimsUser)
23+
if (string.IsNullOrWhiteSpace(referralId))
2424
{
25-
TrackReferralAsync(referralService, referralId, claimsUser);
25+
await _Next(context);
26+
return;
2627
}
27-
else
28+
if (context.User is { Identity.IsAuthenticated: true } claimsUser)
2829
{
29-
TrackReferralAsync(referralService, referralId, null);
30+
referralService.TrackReferralAsync(referralId, claimsUser);
3031
}
31-
32-
await _Next(context);
33-
34-
static void TrackReferralAsync(IReferralService referralService, string? referralId, ClaimsPrincipal? claimsUser)
32+
else
3533
{
36-
if (!string.IsNullOrWhiteSpace(referralId))
37-
{
38-
referralService.TrackReferralAsync(referralId, claimsUser);
39-
}
34+
referralService.TrackReferralAsync(referralId, null);
4035
}
4136
}
4237
}

EssentialCSharp.Web/Services/Referrals/ReferralService.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ public void TrackReferralAsync(string referralId, ClaimsPrincipal? user)
7373
TrackReferral(dbContext, referralId);
7474
}
7575

76-
static void TrackReferral(EssentialCSharpWebContext dbContext, string referralId)
76+
private static void TrackReferral(EssentialCSharpWebContext dbContext, string referralId)
7777
{
7878
var userClaim = dbContext.UserClaims.FirstOrDefault(claim => claim.ClaimType == ClaimsExtensions.ReferrerIdClaimType && claim.ClaimValue == referralId);
7979
if (userClaim is null)

0 commit comments

Comments
 (0)