KvpUserInfoHandler.cs

// Copyright (c) Source Tree Solutions, LLC. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
// Author:					Joe Audette
// Created:					2017-07-10
// Last Modified:			2017-07-14
//

using cloudscribe.Core.Models;
using cloudscribe.Core.Web.ExtensionPoints;
using cloudscribe.Core.Web.ViewModels.SiteUser;
using cloudscribe.UserProperties.Models;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc.ModelBinding;
using Microsoft.AspNetCore.Mvc.ViewFeatures;
using Microsoft.Extensions.Logging;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;

namespace cloudscribe.UserProperties.Kvp
{
    public class KvpUserInfoHandler : IHandleCustomUserInfo
    {
        public KvpUserInfoHandler(
            IProfileOptionsResolver customPropsResolver,
            IUserPropertyService userPropertyService,
            IUserPropertyValidator userPropertyValidator,
            ILogger<KvpUserInfoHandler> logger
            )
        {
            _customPropsResolver = customPropsResolver;
            _log = logger;
            _userPropertyValidator = userPropertyValidator;
            _userPropertyService = userPropertyService;
        }

        protected IProfileOptionsResolver _customPropsResolver;
        protected ILogger _log;
        protected UserPropertySet _props = null;
        protected IUserPropertyService _userPropertyService;
        protected IUserPropertyValidator _userPropertyValidator;

        private async Task EnsureProps()
        {
            if (_props == null)
            {
                _props = await _customPropsResolver.GetProfileProps();
            }
        }

        public virtual Task<string> GetUserInfoViewName(
            ISiteContext site,
            SiteUser siteUser,
            HttpContext httpContext)
        {
            return Task.FromResult("UserInfo"); // this is just returning the default view name.
        }

        public virtual async Task HandleUserInfoGet(
            ISiteContext site,
            SiteUser siteUser,
            UserInfoViewModel viewModel,
            HttpContext httpContext,
            ViewDataDictionary viewData,
            CancellationToken cancellationToken = default(CancellationToken)
            )
        {
            await EnsureProps();
            // add user props to viewData to pass them to the view
            var userProps = await _userPropertyService.FetchByUser(site.Id.ToString(), siteUser.Id.ToString());

            foreach (var p in _props.Properties)
            {
                if (p.VisibleToUserOnProfile)
                {
                    if(_userPropertyService.IsNativeUserProperty(p.Key))
                    {
                        viewData[p.Key] = _userPropertyService.GetNativeUserProperty(siteUser, p.Key);
                    }
                    else
                    {
                        var found = userProps.Where(x => x.Key == p.Key).FirstOrDefault();
                        if (found != null)
                        {
                            viewData[p.Key] = found.Value;
                        }
                    }
                    
                }
            }  
        }

        public virtual async Task<bool> HandleUserInfoValidation(
            ISiteContext site,
            SiteUser siteUser,
            UserInfoViewModel viewModel,
            HttpContext httpContext,
            ViewDataDictionary viewData,
            ModelStateDictionary modelState,
            CancellationToken cancellationToken = default(CancellationToken)
            )
        {
            await EnsureProps();
            var result = true;

            foreach (var p in _props.Properties)
            {
                if (p.EditableByUserOnProfile)
                {
                    var postedValue = httpContext.Request.Form[p.Key];
                    if (_userPropertyValidator.IsValid(p, postedValue, modelState))
                    {
                        // if valid keep the field populated in case some other model validation failed and the form is re-displayed
                        viewData[p.Key] = postedValue;
                    }
                    else
                    {
                        result = false;
                    }

                }
            }

            return result;
        }

        public virtual async Task HandleUserInfoPostSuccess(
            ISiteContext site,
            SiteUser siteUser,
            UserInfoViewModel viewModel,
            HttpContext httpContext,
            CancellationToken cancellationToken = default(CancellationToken)
            )
        {
            await EnsureProps();
            // we "could" re-validate here but
            // the method above gets called just before this in the same postback
            // so we know there were no validation errors or this method would not be invoked
            if (siteUser != null)
            {
                foreach (var p in _props.Properties)
                {
                    if (p.EditableByUserOnProfile)
                    {
                        var postedValue = httpContext.Request.Form[p.Key];
                        if (_userPropertyService.IsNativeUserProperty(p.Key))
                        {
                            _userPropertyService.UpdateNativeUserProperty(siteUser, p.Key, postedValue);
                        }
                        else
                        {
                            // persist to kvp storage
                            await _userPropertyService.CreateOrUpdate(
                                site.Id.ToString(),
                                siteUser.Id.ToString(),
                                p.Key,
                                postedValue);
                        }   
                    }
                }
            }
            else
            {
                _log.LogError("user was null in HandleUserInfoPostSuccess, unable to update user with custom data");
            }
        }


    }
}