1
0
mirror of https://github.com/bitwarden/mobile synced 2025-12-14 23:33:34 +00:00

domains and cipherview

This commit is contained in:
Kyle Spearrin
2019-04-13 22:53:20 -04:00
parent d136eee224
commit f228758fb7
8 changed files with 270 additions and 11 deletions

View File

@@ -1,14 +1,186 @@
using System;
using Bit.Core.Models.Data;
using Bit.Core.Models.View;
using System;
using System.Collections.Generic;
using System.Text;
using System.Linq;
using System.Threading.Tasks;
namespace Bit.Core.Models.Domain
{
public class Cipher : Domain
{
public Cipher() { }
public Cipher(CipherData obj, bool alreadyEncrypted = false, Dictionary<string, object> localData = null)
{
BuildDomainModel(this, obj, new HashSet<string>
{
"Id",
"UserId",
"OrganizationId",
"FolderId",
"Name",
"Notes"
}, alreadyEncrypted, new HashSet<string> { "Id", "UserId", "OrganizationId", "FolderId" });
Type = obj.Type;
Favorite = obj.Favorite;
OrganizationUseTotp = obj.OrganizationUseTotp;
Edit = obj.Edit;
RevisionDate = obj.RevisionDate;
CollectionIds = obj.CollectionIds;
LocalData = localData;
switch(Type)
{
case Enums.CipherType.Login:
Login = new Login(obj.Login, alreadyEncrypted);
break;
case Enums.CipherType.SecureNote:
SecureNote = new SecureNote(obj.SecureNote, alreadyEncrypted);
break;
case Enums.CipherType.Card:
Card = new Card(obj.Card, alreadyEncrypted);
break;
case Enums.CipherType.Identity:
Identity = new Identity(obj.Identity, alreadyEncrypted);
break;
default:
break;
}
Attachments = obj.Attachments?.Select(a => new Attachment(a, alreadyEncrypted)).ToList();
Fields = obj.Fields?.Select(f => new Field(f, alreadyEncrypted)).ToList();
PasswordHistory = obj.PasswordHistory?.Select(ph => new PasswordHistory(ph, alreadyEncrypted)).ToList();
}
public string Id { get; set; }
public string OrganizationId { get; set; }
public string FolderId { get; set; }
public CipherString Name { get; set; }
public CipherString Notes { get; set; }
public Enums.CipherType Type { get; set; }
public bool Favorite { get; set; }
public bool OrganizationUseTotp { get; set; }
public bool Edit { get; set; }
public DateTime RevisionDate { get; set; }
public Dictionary<string, object> LocalData { get; set; }
public Login Login { get; set; }
public Identity Identity { get; set; }
public Card Card { get; set; }
public SecureNote SecureNote { get; set; }
public List<Attachment> Attachments { get; set; }
public List<Field> Fields { get; set; }
public List<PasswordHistory> PasswordHistory { get; set; }
public List<string> CollectionIds { get; set; }
public async Task<CipherView> DecryptAsync(string orgId)
{
var model = new CipherView(this);
await DecryptObjAsync(model, this, new HashSet<string>
{
"Name",
"Notes"
}, orgId);
switch(Type)
{
case Enums.CipherType.Login:
model.Login = await Login.DecryptAsync(orgId);
break;
case Enums.CipherType.SecureNote:
model.SecureNote = await SecureNote.DecryptAsync(orgId);
break;
case Enums.CipherType.Card:
model.Card = await Card.DecryptAsync(orgId);
break;
case Enums.CipherType.Identity:
model.Identity = await Identity.DecryptAsync(orgId);
break;
default:
break;
}
if(Attachments?.Any() ?? false)
{
model.Attachments = new List<AttachmentView>();
var tasks = new List<Task>();
foreach(var attachment in Attachments)
{
var t = attachment.DecryptAsync(orgId)
.ContinueWith(async decAttachment => model.Attachments.Add(await decAttachment));
tasks.Add(t);
}
await Task.WhenAll(tasks);
}
if(Fields?.Any() ?? false)
{
model.Fields = new List<FieldView>();
var tasks = new List<Task>();
foreach(var field in Fields)
{
var t = field.DecryptAsync(orgId)
.ContinueWith(async decField => model.Fields.Add(await decField));
tasks.Add(t);
}
await Task.WhenAll(tasks);
}
if(PasswordHistory?.Any() ?? false)
{
model.PasswordHistory = new List<PasswordHistoryView>();
var tasks = new List<Task>();
foreach(var ph in PasswordHistory)
{
var t = ph.DecryptAsync(orgId)
.ContinueWith(async decPh => model.PasswordHistory.Add(await decPh));
tasks.Add(t);
}
await Task.WhenAll(tasks);
}
return model;
}
public CipherData ToCipherData(string userId)
{
var c = new CipherData
{
Id = Id,
OrganizationId = OrganizationId,
FolderId = FolderId,
UserId = OrganizationId != null ? userId : null,
Edit = Edit,
OrganizationUseTotp = OrganizationUseTotp,
Favorite = Favorite,
RevisionDate = RevisionDate,
Type = Type,
CollectionIds = CollectionIds
};
BuildDataModel(this, c, new HashSet<string>
{
"Name",
"Notes"
});
switch(c.Type)
{
case Enums.CipherType.Login:
c.Login = Login.ToLoginData();
break;
case Enums.CipherType.SecureNote:
c.SecureNote = SecureNote.ToSecureNoteData();
break;
case Enums.CipherType.Card:
c.Card = Card.ToCardData();
break;
case Enums.CipherType.Identity:
c.Identity = Identity.ToIdentityData();
break;
default:
break;
}
c.Fields = Fields?.Select(f => f.ToFieldData()).ToList();
c.Attachments = Attachments?.Select(a => a.ToAttachmentData()).ToList();
c.PasswordHistory = PasswordHistory?.Select(ph => ph.ToPasswordHistoryData()).ToList();
return c;
}
}
}