Disco/Disco.Services/Authorization/DiscoAuthorizeAttribute.cs

using Disco.Services.Users;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Web.Mvc;

namespace Disco.Services.Authorization
{
    public class DiscoAuthorizeAttribute : DiscoAuthorizeBaseAttribute
    {
        string authorizedClaim;

        public DiscoAuthorizeAttribute() { }

        public DiscoAuthorizeAttribute(string AuthorisedClaim)
        {
            this.authorizedClaim = AuthorisedClaim;
        }

        public override bool IsAuthorized(System.Web.HttpContextBase httpContext)
        {
            if (Token == null)
                return false; // No Current User

            if (authorizedClaim == null)
                return Token.RoleTokens.Count > 0; // Just Authenticate - no Authorization (but require at least 1 role)
            else
                return Token.Has(authorizedClaim);
        }

        public override string HandleUnauthorizedMessage()
        {
            string resultMessage;

            if (UserService.CurrentAuthorization == null)
                resultMessage = AuthorizationToken.RequireAuthenticationMessage;
            else
                if (string.IsNullOrEmpty(authorizedClaim))
                    resultMessage = AuthorizationToken.RequireDiscoAuthorizationMessage;
                else
                    resultMessage = AuthorizationToken.BuildRequireMessage(authorizedClaim);

            return resultMessage;
        }
    }
}