83 lines
3.3 KiB
C#
83 lines
3.3 KiB
C#
using System.Linq;
|
|
using Umbraco.Core;
|
|
using Umbraco.Core.Mapping;
|
|
using Umbraco.Core.Models;
|
|
using Umbraco.Core.Services;
|
|
using Umbraco.Web.Models.ContentEditing;
|
|
|
|
namespace Umbraco.Web.Models.Mapping
|
|
{
|
|
internal class RelationMapDefinition : IMapDefinition
|
|
{
|
|
private readonly IEntityService _entityService;
|
|
private readonly IRelationService _relationService;
|
|
|
|
public RelationMapDefinition(IEntityService entityService, IRelationService relationService)
|
|
{
|
|
_entityService = entityService;
|
|
_relationService = relationService;
|
|
}
|
|
|
|
public void DefineMaps(UmbracoMapper mapper)
|
|
{
|
|
mapper.Define<IRelationType, RelationTypeDisplay>((source, context) => new RelationTypeDisplay(), Map);
|
|
mapper.Define<IRelation, RelationDisplay>((source, context) => new RelationDisplay(), Map);
|
|
mapper.Define<RelationTypeSave, IRelationType>(Map);
|
|
}
|
|
|
|
// Umbraco.Code.MapAll -Icon -Trashed -AdditionalData
|
|
// Umbraco.Code.MapAll -ParentId -Notifications
|
|
private void Map(IRelationType source, RelationTypeDisplay target, MapperContext context)
|
|
{
|
|
target.ChildObjectType = source.ChildObjectType;
|
|
target.Id = source.Id;
|
|
target.IsBidirectional = source.IsBidirectional;
|
|
target.Key = source.Key;
|
|
target.Name = source.Name;
|
|
target.Alias = source.Alias;
|
|
target.ParentObjectType = source.ParentObjectType;
|
|
target.Udi = Udi.Create(Constants.UdiEntityType.RelationType, source.Key);
|
|
target.Path = "-1," + source.Id;
|
|
|
|
// Set the "friendly" and entity names for the parent and child object types
|
|
if (source.ParentObjectType.HasValue)
|
|
{
|
|
var objType = ObjectTypes.GetUmbracoObjectType(source.ParentObjectType.Value);
|
|
target.ParentObjectTypeName = objType.GetFriendlyName();
|
|
}
|
|
|
|
if (source.ChildObjectType.HasValue)
|
|
{
|
|
var objType = ObjectTypes.GetUmbracoObjectType(source.ChildObjectType.Value);
|
|
target.ChildObjectTypeName = objType.GetFriendlyName();
|
|
}
|
|
}
|
|
|
|
// Umbraco.Code.MapAll -ParentName -ChildName
|
|
private void Map(IRelation source, RelationDisplay target, MapperContext context)
|
|
{
|
|
target.ChildId = source.ChildId;
|
|
target.Comment = source.Comment;
|
|
target.CreateDate = source.CreateDate;
|
|
target.ParentId = source.ParentId;
|
|
|
|
var entities = _relationService.GetEntitiesFromRelation(source);
|
|
|
|
target.ParentName = entities.Item1.Name;
|
|
target.ChildName = entities.Item2.Name;
|
|
}
|
|
|
|
// Umbraco.Code.MapAll -CreateDate -UpdateDate -DeleteDate
|
|
private static void Map(RelationTypeSave source, IRelationType target, MapperContext context)
|
|
{
|
|
target.Alias = source.Alias;
|
|
target.ChildObjectType = source.ChildObjectType;
|
|
target.Id = source.Id.TryConvertTo<int>().Result;
|
|
target.IsBidirectional = source.IsBidirectional;
|
|
target.Key = source.Key;
|
|
target.Name = source.Name;
|
|
target.ParentObjectType = source.ParentObjectType;
|
|
}
|
|
}
|
|
}
|