Umbraco-CMS/src/Umbraco.Infrastructure/Models/Mapping/TemplateMapDefinition.cs

using Umbraco.Core.Mapping;
using Umbraco.Core.Models;
using Umbraco.Core.Strings;
using Umbraco.Web.Models.ContentEditing;

namespace Umbraco.Web.Models.Mapping
{
    public class TemplateMapDefinition : IMapDefinition
    {
        private readonly IShortStringHelper _shortStringHelper;

        public TemplateMapDefinition(IShortStringHelper shortStringHelper)
        {
            _shortStringHelper = shortStringHelper;
        }

        public void DefineMaps(UmbracoMapper mapper)
        {
            mapper.Define<ITemplate, TemplateDisplay>((source, context) => new TemplateDisplay(), Map);
            mapper.Define<TemplateDisplay, ITemplate>((source, context) => new Template(_shortStringHelper, source.Name, source.Alias), Map);
        }

        // Umbraco.Code.MapAll
        private static void Map(ITemplate source, TemplateDisplay target, MapperContext context)
        {
            target.Id = source.Id;
            target.Name = source.Name;
            target.Alias = source.Alias;
            target.Key = source.Key;
            target.Content = source.Content;
            target.Path = source.Path;
            target.VirtualPath = source.VirtualPath;
            target.MasterTemplateAlias = source.MasterTemplateAlias;
            target.IsMasterTemplate = source.IsMasterTemplate;
        }

        // Umbraco.Code.MapAll -CreateDate -UpdateDate -DeleteDate
        // Umbraco.Code.MapAll -Path -VirtualPath -MasterTemplateId -IsMasterTemplate
        // Umbraco.Code.MapAll -GetFileContent
        private static void Map(TemplateDisplay source, ITemplate target, MapperContext context)
        {
            // don't need to worry about mapping MasterTemplateAlias here;
            // the template controller handles any changes made to the master template
            target.Name = source.Name;
            target.Alias = source.Alias;
            target.Content = source.Content;
            target.Id = source.Id;
            target.Key = source.Key;
        }
    }
}