Dhananjaya99
55 lines
2.5 KiB
C#
55 lines
2.5 KiB
C#
using Fitness_core_BE.Infra.Tenancy;
|
|
using Microsoft.AspNetCore.Mvc.Filters;
|
|
|
|
namespace Fitness_core_BE.Helpers
|
|
{
|
|
// Resolves the request's TenantId into the scoped ITenantContext, so downstream
|
|
// repos/services can read it instead of re-deriving ClientId ad hoc. Must run AFTER
|
|
// CustomMiddlewareAttribute (order 0, populates HttpContext.Items["UserId"]) and BEFORE
|
|
// RequiresPermissionAttribute (order 10) — Order = 5 guarantees this regardless of
|
|
// attribute declaration order on the method, same technique RequiresPermissionAttribute
|
|
// itself uses.
|
|
//
|
|
// Single-club deployment: everyone (Client owner, Staff, Customer) resolves to the same
|
|
// one Client row. GYM's split served many gyms, so a Customer's active gym was picked
|
|
// per-request from a caller-supplied X-Tenant-Id header, independently verified against
|
|
// the DB. There's only one gym here, so that header/verification step is gone — a
|
|
// Customer always resolves to the one club, the same way Client/Staff already did.
|
|
// Admin: platform-wide, no tenant.
|
|
[AttributeUsage(AttributeTargets.Method | AttributeTargets.Class)]
|
|
public class TenantResolutionAttribute : Attribute, IAsyncActionFilter, IOrderedFilter
|
|
{
|
|
public int Order => 5;
|
|
|
|
public async Task OnActionExecutionAsync(ActionExecutingContext context, ActionExecutionDelegate next)
|
|
{
|
|
var httpContext = context.HttpContext;
|
|
var userRoleHelper = httpContext.RequestServices.GetRequiredService<UserRoleHelper>();
|
|
var tenantContext = httpContext.RequestServices.GetRequiredService<ITenantContext>();
|
|
|
|
var userId = DataExtractor.GetUserIdFromContext(httpContext);
|
|
var userType = await userRoleHelper.GetUserType(userId);
|
|
|
|
if (userType == "Admin")
|
|
{
|
|
tenantContext.SetTenant(null, isPlatformAdmin: true);
|
|
await next();
|
|
return;
|
|
}
|
|
|
|
if (userType == "Client" || userType == "Staff")
|
|
{
|
|
var clientId = await userRoleHelper.GetClientId(userId);
|
|
tenantContext.SetTenant(clientId, isPlatformAdmin: false);
|
|
await next();
|
|
return;
|
|
}
|
|
|
|
// Customer -- resolves to the one club, no header needed.
|
|
var defaultClientId = await userRoleHelper.GetDefaultClientId();
|
|
tenantContext.SetTenant(defaultClientId, isPlatformAdmin: false);
|
|
await next();
|
|
}
|
|
}
|
|
}
|