OpenIdOAuthProvider.cs

using System;
using System.Collections.Generic;
using DotNetOpenAuth.Messaging;
using DotNetOpenAuth.OpenId.Extensions.AttributeExchange;
using DotNetOpenAuth.OpenId.Extensions.SimpleRegistration;
using DotNetOpenAuth.OpenId.RelyingParty;
using ServiceStack.Common;
using ServiceStack.Common.Web;
using ServiceStack.Configuration;
using ServiceStack.ServiceClient.Web;
using ServiceStack.ServiceHost;
using ServiceStack.ServiceInterface;
using ServiceStack.ServiceInterface.Auth;
using ServiceStack.Text;

namespace ServiceStack.Authentication.OpenId
{
    public class OpenIdOAuthProvider : OAuthProvider
    {
        public const string DefaultName = "OpenId";

        public OpenIdOAuthProvider(IResourceManager appSettings, string name = DefaultName, string realm = null)
            : base(appSettings, realm, name) { }

        public virtual ClaimsRequest CreateClaimsRequest(IHttpRequest httpReq)
        {
            return new ClaimsRequest {
                Country = DemandLevel.Request,
                Email = DemandLevel.Request,
                Gender = DemandLevel.Require,
                PostalCode = DemandLevel.Require,
                TimeZone = DemandLevel.Require,
            };
        }

        public override object Authenticate(IServiceBase authService, IAuthSession session, Auth request)
        {
            var tokens = Init(authService, ref session, request);

            var httpReq = authService.RequestContext.Get<IHttpRequest>();
            var httpMethod = httpReq.HttpMethod;
            if (httpMethod == HttpMethod.Post)
            {
                var openIdUrl = httpReq.GetParam("OpenIdUrl") ?? base.AuthRealm;
                if (openIdUrl.IsNullOrEmpty())
                    throw new ArgumentNullException("'OpenIdUrl' is required a required field");

                try
                {
                    using (var openid = new OpenIdRelyingParty())
                    {
                        var openIdRequest = openid.CreateRequest(openIdUrl);

                        AddAttributeExchangeExtensions(openIdRequest);

                        // This is where you would add any OpenID extensions you wanted
                        // to include in the authentication request.
                        openIdRequest.AddExtension(CreateClaimsRequest(httpReq));

                        // Send your visitor to their Provider for authentication.
                        var openIdResponse = openIdRequest.RedirectingResponse;
                        var contentType = openIdResponse.Headers[HttpHeaders.ContentType];
                        var httpResult = new HttpResult(openIdResponse.ResponseStream, contentType) {
                            StatusCode = openIdResponse.Status,
                            StatusDescription = "Moved Temporarily",
                        };
                        foreach (string header in openIdResponse.Headers)
                        {
                            httpResult.Headers[header] = openIdResponse.Headers[header];
                        }
                        return httpResult;
                    }
                }
                catch (ProtocolException ex)
                {
                    Log.Error("Falied to login to {0}".Fmt(openIdUrl), ex);
                    return authService.Redirect(session.ReferrerUrl.AddHashParam("f", "Unknown"));
                }
            }

            if (httpMethod == HttpMethod.Get)
            {
                using (var openid = new OpenIdRelyingParty())
                {
                    var response = openid.GetResponse();
                    if (response != null)
                    {
                        switch (response.Status)
                        {
                            case AuthenticationStatus.Authenticated:
                                // This is where you would look for any OpenID extension responses included
                                // in the authentication assertion.
                                var claimsResponse = response.GetExtension<ClaimsResponse>();
                                var authInfo = claimsResponse.ToDictionary();

                                authInfo["user_id"] = response.ClaimedIdentifier; //a url

                                // Store off the "friendly" username to display -- NOT for username lookup
                                authInfo["openid_ref"] = response.FriendlyIdentifierForDisplay;

                                var provided = GetAttributeEx(response);
                                foreach (var entry in provided)
                                {
                                    authInfo[entry.Key] = entry.Value;
                                }

                                // Use FormsAuthentication to tell ASP.NET that the user is now logged in,
                                // with the OpenID Claimed Identifier as their username.

                                session.IsAuthenticated = true;
                                authService.SaveSession(session, SessionExpiry);
                                OnAuthenticated(authService, session, tokens, authInfo);

                                //Haz access!
                                return authService.Redirect(session.ReferrerUrl.AddHashParam("s", "1"));

                            case AuthenticationStatus.Canceled:
                                return authService.Redirect(session.ReferrerUrl.AddHashParam("f", "ProviderCancelled"));

                            case AuthenticationStatus.Failed:
                                return authService.Redirect(session.ReferrerUrl.AddHashParam("f", "Unknown"));
                        }
                    }
                }
            }

            //Shouldn't get here
            return authService.Redirect(session.ReferrerUrl.AddHashParam("f", "Unknown"));
        }

        protected override void LoadUserAuthInfo(AuthUserSession userSession, IOAuthTokens tokens, Dictionary<string, string> authInfo)
        {
            if (authInfo.ContainsKey("user_id"))
                tokens.UserId = authInfo.GetValueOrDefault("user_id");

            if (authInfo.ContainsKey("name"))
                tokens.DisplayName = authInfo.GetValueOrDefault("name");

            if (authInfo.ContainsKey("FullName"))
            {
                tokens.FullName = authInfo.GetValueOrDefault("FullName");
                if (tokens.DisplayName.IsNullOrEmpty())
                    tokens.DisplayName = tokens.FullName;
            }

            if (authInfo.ContainsKey("Email"))
                tokens.Email = authInfo.GetValueOrDefault("Email");

            if (authInfo.ContainsKey("BirthDate"))
                tokens.BirthDate = authInfo.GetValueOrDefault("BirthDate").FromJsv<DateTime?>();

            if (authInfo.ContainsKey("BirthDateRaw"))
                tokens.BirthDateRaw = authInfo.GetValueOrDefault("BirthDateRaw");

            if (authInfo.ContainsKey("Country"))
                tokens.Country = authInfo.GetValueOrDefault("Country");

            if (authInfo.ContainsKey("Culture"))
                tokens.Culture = authInfo.GetValueOrDefault("Culture");

            if (authInfo.ContainsKey("Gender"))
                tokens.Gender = authInfo.GetValueOrDefault("Gender");

            if (authInfo.ContainsKey("MailAddress"))
                tokens.MailAddress = authInfo.GetValueOrDefault("MailAddress");

            if (authInfo.ContainsKey("Nickname"))
                tokens.Nickname = authInfo.GetValueOrDefault("Nickname");

            if (authInfo.ContainsKey("PostalCode"))
                tokens.PostalCode = authInfo.GetValueOrDefault("PostalCode");

            if (authInfo.ContainsKey("TimeZone"))
                tokens.TimeZone = authInfo.GetValueOrDefault("TimeZone");

            LoadUserOAuthProvider(userSession, tokens);
        }

        public override void LoadUserOAuthProvider(IAuthSession authSession, IOAuthTokens tokens)
        {
            var userSession = authSession as AuthUserSession;
            if (userSession == null) return;
        }

        private void AddAttributeExchangeExtensions(IAuthenticationRequest auth)
        {
            // Try to use OpenId 2.0's attribute exchange
            var fetch = new FetchRequest();
            //Technically, http://axschema.org/... are "standard", but we'll still find these in the wild
            fetch.Attributes.Add(new AttributeRequest("http://schema.openid.net/namePerson", false));
            fetch.Attributes.Add(new AttributeRequest("http://schema.openid.net/contact/email", false));

            fetch.Attributes.AddRequired("http://axschema.org/contact/country/home");
            fetch.Attributes.AddRequired("http://axschema.org/namePerson/first");
            fetch.Attributes.AddRequired("http://axschema.org/namePerson/last");
            fetch.Attributes.AddRequired("http://axschema.org/pref/language");
            fetch.Attributes.AddRequired("http://schemas.openid.net/ax/api/user_id");

            //Standard compliant AX schema
            fetch.Attributes.Add(new AttributeRequest(WellKnownAttributes.Name.FullName, false));

            //For... no good reason, really, google OpenId requires you "require" an e-mail address to get it
            bool requireEmail = auth.Provider.Uri.AbsoluteUri.Contains(".google.com");
            fetch.Attributes.Add(new AttributeRequest(WellKnownAttributes.Contact.Email, requireEmail));

            auth.AddExtension(fetch);
        }

        /// <summary>
        /// Extracts an Attribute Exchange response, if one exists
        /// </summary>
        private Dictionary<string, string> GetAttributeEx(IAuthenticationResponse response)
        {
            var ret = new Dictionary<string, string>();

            var fetchResponse = response.GetExtension<FetchResponse>();

            if (fetchResponse == null) return ret;

            var names = new List<string>();
            var emails = new List<string>();

            if (fetchResponse.Attributes.Contains("http://schema.openid.net/namePerson"))
                names.AddRange(fetchResponse.Attributes["http://schema.openid.net/namePerson"].Values);

            if (fetchResponse.Attributes.Contains(WellKnownAttributes.Name.FullName))
                names.AddRange(fetchResponse.Attributes[WellKnownAttributes.Name.FullName].Values);

            if (fetchResponse.Attributes.Contains(WellKnownAttributes.Name.Alias))
                names.AddRange(fetchResponse.Attributes[WellKnownAttributes.Name.Alias].Values);

            if (fetchResponse.Attributes.Contains(WellKnownAttributes.Name.First))
                names.AddRange(fetchResponse.Attributes[WellKnownAttributes.Name.First].Values);

            if (fetchResponse.Attributes.Contains(WellKnownAttributes.Name.Last))
                names.AddRange(fetchResponse.Attributes[WellKnownAttributes.Name.Last].Values);

            if (fetchResponse.Attributes.Contains("http://schema.openid.net/contact/email"))
                emails.AddRange(fetchResponse.Attributes["http://schema.openid.net/contact/email"].Values);

            if (fetchResponse.Attributes.Contains(WellKnownAttributes.Contact.Email))
                emails.AddRange(fetchResponse.Attributes[WellKnownAttributes.Contact.Email].Values);

            if (names.Count > 0)
                ret["FullName"] = names[0];

            if (emails.Count > 0)
                ret["Email"] = emails[0];

            return ret;
        }
    }

    public static class OpenIdExtensions
    {
        public static Dictionary<string, string> ToDictionary(this ClaimsResponse response)
        {
            var map = new Dictionary<string, string>();
            if (response == null) return map;

            if (response.BirthDate.HasValue)
                map["BirthDate"] = response.BirthDate.Value.ToJsv();
            if (!response.BirthDateRaw.IsNullOrEmpty())
                map["BirthDateRaw"] = response.BirthDateRaw;
            if (!response.Country.IsNullOrEmpty())
                map["Country"] = response.Country;
            if (response.Culture != null)
                map["Culture"] = response.Culture.TwoLetterISOLanguageName;
            if (!response.Email.IsNullOrEmpty())
                map["Email"] = response.Email;
            if (!response.FullName.IsNullOrEmpty())
                map["FullName"] = response.FullName;
            if (response.Gender.HasValue)
                map["Gender"] = response.Gender.Value.ToString();
            if (!response.Language.IsNullOrEmpty())
                map["Language"] = response.Language;
            if (response.MailAddress != null)
                map["MailAddress"] = response.MailAddress.ToJsv();
            if (!response.Nickname.IsNullOrEmpty())
                map["Nickname"] = response.Nickname;
            if (!response.PostalCode.IsNullOrEmpty())
                map["PostalCode"] = response.PostalCode;
            if (!response.TimeZone.IsNullOrEmpty())
                map["TimeZone"] = response.TimeZone;

            return map;
        }
    }
}