36 lines
1.2 KiB
C#
36 lines
1.2 KiB
C#
using Microsoft.AspNetCore.Mvc;
|
|
using System.Security.Claims;
|
|
using YABA.Common.Extensions;
|
|
using YABA.Common.Lookups;
|
|
|
|
namespace YABA.API.Extensions
|
|
{
|
|
public static class ControllerExtensions
|
|
{
|
|
public static string GetAuthProviderId(this ControllerBase controller)
|
|
{
|
|
return GetCustomClaim(controller, ClaimsLookup.AuthProviderId);
|
|
}
|
|
|
|
public static int GetUserId(this ControllerBase controller)
|
|
{
|
|
var isValidUserId = int.TryParse(GetCustomClaim(controller, ClaimsLookup.UserId), out int userId);
|
|
return isValidUserId ? userId : 0;
|
|
}
|
|
|
|
public static string GetCustomClaim(this ControllerBase controller, ClaimsLookup claim)
|
|
{
|
|
var claimsIdentity = controller.User.Identity as ClaimsIdentity;
|
|
return claimsIdentity.FindFirst(claim.GetClaimName())?.Value.ToString();
|
|
}
|
|
|
|
public static string GetIpAddress(this ControllerBase controller)
|
|
{
|
|
if (controller.Request.Headers.ContainsKey("X-Forwarded-For"))
|
|
return controller.Request.Headers["X-Forwarded-For"];
|
|
|
|
return controller.HttpContext.Connection.RemoteIpAddress.MapToIPv4().ToString();
|
|
}
|
|
}
|
|
}
|