using System; using System.Collections.Generic; using System.Drawing; using System.Linq; using AutoMapper; using ModernKeePass.Application.Common.Interfaces; using ModernKeePass.Application.Common.Mappings; using ModernKeePass.Domain.Entities; using ModernKeePass.Domain.Enums; namespace ModernKeePass.Application.Entry.Models { public class EntryVm: IEntityVm, IMapFrom { public string ParentGroupId { get; set; } public string ParentGroupName { get; set; } public string Id { get; set; } public FieldVm Title { get; set; } public FieldVm Username { get; set; } public FieldVm Password { get; set; } public FieldVm Notes { get; set; } public FieldVm Url { get; set; } public bool HasUrl => !string.IsNullOrEmpty(Url.Value); public List AdditionalFields { get; set; } public List History { get; set; } public Icon Icon { get; set; } public Color ForegroundColor { get; set; } public Color BackgroundColor { get; set; } public bool HasExpirationDate { get; set; } public DateTimeOffset ExpirationDate { get; set; } public DateTimeOffset ModificationDate { get; set; } public Dictionary Attachments { get; set; } public override string ToString() { return ModificationDate.ToString("g"); } public void Mapping(Profile profile) { profile.CreateMap() .ForMember(d => d.ParentGroupId, opts => opts.MapFrom(s => s.ParentId)) .ForMember(d => d.ParentGroupName, opts => opts.MapFrom(s => s.ParentName)) .ForMember(d => d.Id, opts => opts.MapFrom(s => s.Id)) .ForMember(d => d.Title, opts => opts.MapFrom(s => s.Fields.FirstOrDefault(f => f.Name.Equals(EntryFieldName.Title, StringComparison.OrdinalIgnoreCase)) ?? new FieldEntity { Name = EntryFieldName.Title, IsProtected = true } )) .ForMember(d => d.Username, opts => opts.MapFrom(s => s.Fields.FirstOrDefault(f => f.Name.Equals(EntryFieldName.UserName, StringComparison.OrdinalIgnoreCase)) ?? new FieldEntity { Name = EntryFieldName.UserName, IsProtected = true } )) .ForMember(d => d.Password, opts => opts.MapFrom(s => s.Fields.FirstOrDefault(f => f.Name.Equals(EntryFieldName.Password, StringComparison.OrdinalIgnoreCase)) ?? new FieldEntity { Name = EntryFieldName.Password, IsProtected = true } )) .ForMember(d => d.Url, opts => opts.MapFrom(s => s.Fields.FirstOrDefault(f => f.Name.Equals(EntryFieldName.Url, StringComparison.OrdinalIgnoreCase)) ?? new FieldEntity { Name = EntryFieldName.Url, IsProtected = true } )) .ForMember(d => d.Notes, opts => opts.MapFrom(s => s.Fields.FirstOrDefault(f => f.Name.Equals(EntryFieldName.Notes, StringComparison.OrdinalIgnoreCase)) ?? new FieldEntity { Name = EntryFieldName.Notes, IsProtected = true } )) .ForMember(d => d.AdditionalFields, opts => opts.MapFrom(s => s.Fields.Where(f => !EntryFieldName.StandardFieldNames.Contains(f.Name, StringComparer.OrdinalIgnoreCase)))) .ForMember(d => d.History, opts => opts.MapFrom(s => s.History.Reverse())) .ForMember(d => d.HasExpirationDate, opts => opts.MapFrom(s => s.HasExpirationDate)) .ForMember(d => d.ExpirationDate, opts => opts.MapFrom(s => s.ExpirationDate)) .ForMember(d => d.ModificationDate, opts => opts.MapFrom(s => s.LastModificationDate)) .ForMember(d => d.Icon, opts => opts.MapFrom(s => s.HasExpirationDate && s.ExpirationDate < DateTimeOffset.Now ? Icon.ReportHacked : s.Icon)) .ForMember(d => d.ForegroundColor, opts => opts.MapFrom(s => s.ForegroundColor)) .ForMember(d => d.BackgroundColor, opts => opts.MapFrom(s => s.BackgroundColor)) .ForMember(d => d.Attachments, opts => opts.MapFrom(s => s.Attachments)); } } }