1
0
mirror of https://github.com/bitwarden/server.git synced 2025-06-30 15:42:48 -05:00

refactor for cipher details, folders, favorites

This commit is contained in:
Kyle Spearrin
2017-03-18 11:58:02 -04:00
parent 2b71420818
commit 588f6c7c2c
15 changed files with 139 additions and 138 deletions

View File

@ -38,7 +38,7 @@ namespace Bit.Api.Controllers
throw new NotFoundException(); throw new NotFoundException();
} }
return new CipherResponseModel(cipher, userId); return new CipherResponseModel(cipher);
} }
[HttpGet("")] [HttpGet("")]
@ -46,25 +46,25 @@ namespace Bit.Api.Controllers
{ {
var userId = _userService.GetProperUserId(User).Value; var userId = _userService.GetProperUserId(User).Value;
var ciphers = await _cipherRepository.GetManyByUserIdAsync(userId); var ciphers = await _cipherRepository.GetManyByUserIdAsync(userId);
var responses = ciphers.Select(c => new CipherResponseModel(c, userId)); var responses = ciphers.Select(c => new CipherResponseModel(c));
return new ListResponseModel<CipherResponseModel>(responses); return new ListResponseModel<CipherResponseModel>(responses);
} }
[Obsolete] //[Obsolete]
[HttpGet("history")] //[HttpGet("history")]
public async Task<CipherHistoryResponseModel> Get(DateTime since) //public async Task<CipherHistoryResponseModel> Get(DateTime since)
{ //{
var userId = _userService.GetProperUserId(User).Value; // var userId = _userService.GetProperUserId(User).Value;
var history = await _cipherRepository.GetManySinceRevisionDateAndUserIdWithDeleteHistoryAsync( // var history = await _cipherRepository.GetManySinceRevisionDateAndUserIdWithDeleteHistoryAsync(
since, userId); // since, userId);
return new CipherHistoryResponseModel(history.Item1, history.Item2, userId); // return new CipherHistoryResponseModel(history.Item1, history.Item2, userId);
} //}
[HttpPost("import")] [HttpPost("import")]
public async Task PostImport([FromBody]ImportRequestModel model) public async Task PostImport([FromBody]ImportRequestModel model)
{ {
var userId = _userService.GetProperUserId(User).Value; var userId = _userService.GetProperUserId(User).Value;
var folderCiphers = model.Folders.Select(f => f.ToCipher(userId)).ToList(); var folderCiphers = model.Folders.Select(f => f.ToFolder(userId)).ToList();
var otherCiphers = model.Logins.Select(s => s.ToCipher(userId)).ToList(); var otherCiphers = model.Logins.Select(s => s.ToCipher(userId)).ToList();
await _cipherService.ImportCiphersAsync( await _cipherService.ImportCiphersAsync(
@ -73,20 +73,20 @@ namespace Bit.Api.Controllers
model.FolderRelationships); model.FolderRelationships);
} }
[HttpPut("{id}/favorite")] //[HttpPut("{id}/favorite")]
[HttpPost("{id}/favorite")] //[HttpPost("{id}/favorite")]
public async Task Favorite(string id) //public async Task Favorite(string id)
{ //{
var cipher = await _cipherRepository.GetByIdAsync(new Guid(id), _userService.GetProperUserId(User).Value); // var cipher = await _cipherRepository.GetByIdAsync(new Guid(id), _userService.GetProperUserId(User).Value);
if(cipher == null) // if(cipher == null)
{ // {
throw new NotFoundException(); // throw new NotFoundException();
} // }
cipher.Favorite = !cipher.Favorite; // cipher.Favorite = !cipher.Favorite;
await _cipherService.SaveAsync(cipher); // await _cipherService.SaveAsync(cipher);
} //}
[HttpDelete("{id}")] [HttpDelete("{id}")]
[HttpPost("{id}/delete")] [HttpPost("{id}/delete")]

View File

@ -1,5 +1,4 @@
using System; using System;
using System.Collections.Generic;
using System.Linq; using System.Linq;
using System.Threading.Tasks; using System.Threading.Tasks;
using Microsoft.AspNetCore.Mvc; using Microsoft.AspNetCore.Mvc;
@ -7,7 +6,6 @@ using Bit.Core.Repositories;
using Microsoft.AspNetCore.Authorization; using Microsoft.AspNetCore.Authorization;
using Bit.Core.Models.Api; using Bit.Core.Models.Api;
using Bit.Core.Exceptions; using Bit.Core.Exceptions;
using Bit.Core.Models.Table;
using Bit.Core.Services; using Bit.Core.Services;
namespace Bit.Api.Controllers namespace Bit.Api.Controllers
@ -33,7 +31,7 @@ namespace Bit.Api.Controllers
} }
[HttpGet("{id}")] [HttpGet("{id}")]
public async Task<LoginResponseModel> Get(string id, string[] expand = null) public async Task<LoginResponseModel> Get(string id)
{ {
var userId = _userService.GetProperUserId(User).Value; var userId = _userService.GetProperUserId(User).Value;
var login = await _cipherRepository.GetByIdAsync(new Guid(id), userId); var login = await _cipherRepository.GetByIdAsync(new Guid(id), userId);
@ -42,37 +40,34 @@ namespace Bit.Api.Controllers
throw new NotFoundException(); throw new NotFoundException();
} }
var response = new LoginResponseModel(login, userId); var response = new LoginResponseModel(login);
await ExpandAsync(login, response, expand, null, userId);
return response; return response;
} }
[HttpGet("")] [HttpGet("")]
public async Task<ListResponseModel<LoginResponseModel>> Get(string[] expand = null) public async Task<ListResponseModel<LoginResponseModel>> Get()
{ {
var userId = _userService.GetProperUserId(User).Value; var userId = _userService.GetProperUserId(User).Value;
var logins = await _cipherRepository.GetManyByTypeAndUserIdAsync(Core.Enums.CipherType.Login, var logins = await _cipherRepository.GetManyByTypeAndUserIdAsync(Core.Enums.CipherType.Login,
userId); userId);
var responses = logins.Select(s => new LoginResponseModel(s, userId)).ToList(); var responses = logins.Select(s => new LoginResponseModel(s)).ToList();
await ExpandManyAsync(logins, responses, expand, null, userId);
return new ListResponseModel<LoginResponseModel>(responses); return new ListResponseModel<LoginResponseModel>(responses);
} }
[HttpPost("")] [HttpPost("")]
public async Task<LoginResponseModel> Post([FromBody]LoginRequestModel model, string[] expand = null) public async Task<LoginResponseModel> Post([FromBody]LoginRequestModel model)
{ {
var userId = _userService.GetProperUserId(User).Value; var userId = _userService.GetProperUserId(User).Value;
var login = model.ToCipher(userId); var login = model.ToCipher(userId);
await _cipherService.SaveAsync(login); await _cipherService.SaveAsync(login);
var response = new LoginResponseModel(login, userId); var response = new LoginResponseModel(login);
await ExpandAsync(login, response, expand, null, userId);
return response; return response;
} }
[HttpPut("{id}")] [HttpPut("{id}")]
[HttpPost("{id}")] [HttpPost("{id}")]
public async Task<LoginResponseModel> Put(string id, [FromBody]LoginRequestModel model, string[] expand = null) public async Task<LoginResponseModel> Put(string id, [FromBody]LoginRequestModel model)
{ {
var userId = _userService.GetProperUserId(User).Value; var userId = _userService.GetProperUserId(User).Value;
var login = await _cipherRepository.GetByIdAsync(new Guid(id), _userService.GetProperUserId(User).Value); var login = await _cipherRepository.GetByIdAsync(new Guid(id), _userService.GetProperUserId(User).Value);
@ -83,8 +78,7 @@ namespace Bit.Api.Controllers
await _cipherService.SaveAsync(model.ToCipher(login)); await _cipherService.SaveAsync(model.ToCipher(login));
var response = new LoginResponseModel(login, userId); var response = new LoginResponseModel(login);
await ExpandAsync(login, response, expand, null, userId);
return response; return response;
} }
@ -100,61 +94,5 @@ namespace Bit.Api.Controllers
await _cipherService.DeleteAsync(login); await _cipherService.DeleteAsync(login);
} }
private async Task ExpandAsync(Cipher login, LoginResponseModel response, string[] expand, Cipher folder, Guid userId)
{
if(expand == null || expand.Count() == 0)
{
return;
}
if(expand.Any(e => e.ToLower() == "folder") && login.FolderId.HasValue)
{
if(folder == null)
{
folder = await _cipherRepository.GetByIdAsync(login.FolderId.Value);
}
response.Folder = new FolderResponseModel(folder, userId);
}
}
private async Task ExpandManyAsync(IEnumerable<Cipher> logins, ICollection<LoginResponseModel> responses,
string[] expand, IEnumerable<Cipher> folders, Guid userId)
{
if(expand == null || expand.Count() == 0)
{
return;
}
if(expand.Any(e => e.ToLower() == "folder"))
{
if(folders == null)
{
folders = await _cipherRepository.GetManyByTypeAndUserIdAsync(Core.Enums.CipherType.Folder,
_userService.GetProperUserId(User).Value);
}
if(folders != null && folders.Count() > 0)
{
foreach(var response in responses)
{
var login = logins.SingleOrDefault(s => s.Id.ToString() == response.Id);
if(login == null)
{
continue;
}
var folder = folders.SingleOrDefault(f => f.Id == login.FolderId);
if(folder == null)
{
continue;
}
response.Folder = new FolderResponseModel(folder, userId);
}
}
}
}
} }
} }

View File

@ -2,7 +2,7 @@
{ {
public enum CipherType : byte public enum CipherType : byte
{ {
Folder = 0, //Folder = 0,
Login = 1 Login = 1
} }
} }

View File

@ -1,6 +1,7 @@
using System; using System;
using Bit.Core.Models.Table; using Bit.Core.Models.Table;
using Newtonsoft.Json; using Newtonsoft.Json;
using Core.Models.Data;
namespace Bit.Core.Models.Api namespace Bit.Core.Models.Api
{ {
@ -28,7 +29,23 @@ namespace Bit.Core.Models.Api
public LoginDataModel(Cipher cipher) public LoginDataModel(Cipher cipher)
{ {
if(cipher.Type != Core.Enums.CipherType.Login) if(cipher.Type != Enums.CipherType.Login)
{
throw new ArgumentException("Cipher is not correct type.");
}
var data = JsonConvert.DeserializeObject<LoginDataModel>(cipher.Data);
Name = data.Name;
Uri = data.Uri;
Username = data.Username;
Password = data.Password;
Notes = data.Notes;
}
public LoginDataModel(CipherDetails cipher)
{
if(cipher.Type != Enums.CipherType.Login)
{ {
throw new ArgumentException("Cipher is not correct type."); throw new ArgumentException("Cipher is not correct type.");
} }

View File

@ -39,15 +39,12 @@ namespace Bit.Core.Models.Api
{ {
Id = new Guid(Id), Id = new Guid(Id),
UserId = userId, UserId = userId,
FolderId = string.IsNullOrWhiteSpace(FolderId) ? null : (Guid?)new Guid(FolderId), //FolderId = string.IsNullOrWhiteSpace(FolderId) ? null : (Guid?)new Guid(FolderId),
Type = Type Type = Type
}; };
switch(Type) switch(Type)
{ {
case CipherType.Folder:
cipher.Data = JsonConvert.SerializeObject(new FolderDataModel(this), new JsonSerializerSettings { NullValueHandling = NullValueHandling.Ignore });
break;
case CipherType.Login: case CipherType.Login:
cipher.Data = JsonConvert.SerializeObject(new LoginDataModel(this), new JsonSerializerSettings { NullValueHandling = NullValueHandling.Ignore }); cipher.Data = JsonConvert.SerializeObject(new LoginDataModel(this), new JsonSerializerSettings { NullValueHandling = NullValueHandling.Ignore });
break; break;

View File

@ -3,6 +3,7 @@ using System.ComponentModel.DataAnnotations;
using Bit.Core.Utilities; using Bit.Core.Utilities;
using Bit.Core.Models.Table; using Bit.Core.Models.Table;
using Newtonsoft.Json; using Newtonsoft.Json;
using Core.Models.Data;
namespace Bit.Core.Models.Api namespace Bit.Core.Models.Api
{ {
@ -28,21 +29,22 @@ namespace Bit.Core.Models.Api
[StringLength(10000)] [StringLength(10000)]
public string Notes { get; set; } public string Notes { get; set; }
public Cipher ToCipher(Guid userId) public CipherDetails ToCipher(Guid userId)
{ {
return ToCipher(new Cipher return ToCipher(new CipherDetails
{ {
UserId = userId UserId = userId
}); });
} }
public Cipher ToCipher(Cipher existingLogin) public CipherDetails ToCipher(CipherDetails existingLogin)
{ {
existingLogin.FolderId = string.IsNullOrWhiteSpace(FolderId) ? null : (Guid?)new Guid(FolderId); existingLogin.FolderId = string.IsNullOrWhiteSpace(FolderId) ? null : (Guid?)new Guid(FolderId);
existingLogin.Favorite = Favorite; existingLogin.Favorite = Favorite;
existingLogin.Data = JsonConvert.SerializeObject(new LoginDataModel(this), existingLogin.Data = JsonConvert.SerializeObject(new LoginDataModel(this),
new JsonSerializerSettings { NullValueHandling = NullValueHandling.Ignore }); new JsonSerializerSettings { NullValueHandling = NullValueHandling.Ignore });
existingLogin.Type = Core.Enums.CipherType.Login; existingLogin.Type = Enums.CipherType.Login;
return existingLogin; return existingLogin;
} }

View File

@ -20,7 +20,7 @@ namespace Bit.Core.Models.Api
throw new ArgumentNullException(nameof(deletedIds)); throw new ArgumentNullException(nameof(deletedIds));
} }
Revised = revisedCiphers.Select(c => new CipherResponseModel(c, userId)); Revised = revisedCiphers.Select(c => new CipherResponseModel(c));
Deleted = deletedIds.Select(id => id.ToString()); Deleted = deletedIds.Select(id => id.ToString());
} }

View File

@ -1,11 +1,12 @@
using System; using System;
using Bit.Core.Models.Table; using Bit.Core.Models.Table;
using Core.Models.Data;
namespace Bit.Core.Models.Api namespace Bit.Core.Models.Api
{ {
public class CipherResponseModel : ResponseModel public class CipherResponseModel : ResponseModel
{ {
public CipherResponseModel(Cipher cipher, Guid userId, string obj = "cipher") public CipherResponseModel(Cipher cipher, string obj = "cipher")
: base(obj) : base(obj)
{ {
if(cipher == null) if(cipher == null)
@ -14,17 +15,33 @@ namespace Bit.Core.Models.Api
} }
Id = cipher.Id.ToString(); Id = cipher.Id.ToString();
FolderId = cipher.FolderId?.ToString();
Type = cipher.Type; Type = cipher.Type;
Favorite = cipher.Favorite;
RevisionDate = cipher.RevisionDate; RevisionDate = cipher.RevisionDate;
switch(cipher.Type) switch(cipher.Type)
{ {
case Core.Enums.CipherType.Folder: case Enums.CipherType.Login:
Data = new FolderDataModel(cipher); Data = new LoginDataModel(cipher);
break; break;
case Core.Enums.CipherType.Login: default:
throw new ArgumentException("Unsupported " + nameof(Type) + ".");
}
}
public CipherResponseModel(CipherDetails cipher, string obj = "cipher")
: base(obj)
{
if(cipher == null)
{
throw new ArgumentNullException(nameof(cipher));
}
Id = cipher.Id.ToString();
Type = cipher.Type;
RevisionDate = cipher.RevisionDate;
switch(cipher.Type)
{
case Enums.CipherType.Login:
Data = new LoginDataModel(cipher); Data = new LoginDataModel(cipher);
break; break;
default: default:
@ -34,7 +51,7 @@ namespace Bit.Core.Models.Api
public string Id { get; set; } public string Id { get; set; }
public string FolderId { get; set; } public string FolderId { get; set; }
public Core.Enums.CipherType Type { get; set; } public Enums.CipherType Type { get; set; }
public bool Favorite { get; set; } public bool Favorite { get; set; }
public dynamic Data { get; set; } public dynamic Data { get; set; }
public DateTime RevisionDate { get; set; } public DateTime RevisionDate { get; set; }

View File

@ -1,11 +1,12 @@
using System; using System;
using Core.Models.Data;
using Bit.Core.Models.Table; using Bit.Core.Models.Table;
namespace Bit.Core.Models.Api namespace Bit.Core.Models.Api
{ {
public class LoginResponseModel : ResponseModel public class LoginResponseModel : ResponseModel
{ {
public LoginResponseModel(Cipher cipher, Guid userId, string obj = "login") public LoginResponseModel(CipherDetails cipher, string obj = "login")
: base(obj) : base(obj)
{ {
if(cipher == null) if(cipher == null)
@ -13,7 +14,7 @@ namespace Bit.Core.Models.Api
throw new ArgumentNullException(nameof(cipher)); throw new ArgumentNullException(nameof(cipher));
} }
if(cipher.Type != Core.Enums.CipherType.Login) if(cipher.Type != Enums.CipherType.Login)
{ {
throw new ArgumentException(nameof(cipher.Type)); throw new ArgumentException(nameof(cipher.Type));
} }

View File

@ -1,18 +1,11 @@
using Bit.Core.Enums; using Bit.Core.Models.Table;
using System; using System;
namespace Core.Models.Data namespace Core.Models.Data
{ {
public class CipherDetails public class CipherDetails : Cipher
{ {
public Guid Id { get; set; }
public Guid? UserId { get; set; }
public Guid? OrganizationId { get; set; }
public Guid? FolderId { get; set; } public Guid? FolderId { get; set; }
public CipherType Type { get; set; }
public bool Favorite { get; set; } public bool Favorite { get; set; }
public string Data { get; set; }
public DateTime CreationDate { get; set; }
public DateTime RevisionDate { get; set; }
} }
} }

View File

@ -8,7 +8,7 @@ namespace Bit.Core.Repositories
{ {
public interface ICipherRepository : IRepository<Cipher, Guid> public interface ICipherRepository : IRepository<Cipher, Guid>
{ {
Task<Cipher> GetByIdAsync(Guid id, Guid userId); Task<CipherDetails> GetByIdAsync(Guid id, Guid userId);
Task<ICollection<CipherDetails>> GetManyByUserIdAsync(Guid userId); Task<ICollection<CipherDetails>> GetManyByUserIdAsync(Guid userId);
Task<ICollection<CipherDetails>> GetManyByTypeAndUserIdAsync(Enums.CipherType type, Guid userId); Task<ICollection<CipherDetails>> GetManyByTypeAndUserIdAsync(Enums.CipherType type, Guid userId);
Task<Tuple<ICollection<CipherDetails>, ICollection<Guid>>> GetManySinceRevisionDateAndUserIdWithDeleteHistoryAsync( Task<Tuple<ICollection<CipherDetails>, ICollection<Guid>>> GetManySinceRevisionDateAndUserIdWithDeleteHistoryAsync(

View File

@ -21,15 +21,17 @@ namespace Bit.Core.Repositories.SqlServer
: base(connectionString) : base(connectionString)
{ } { }
public async Task<Cipher> GetByIdAsync(Guid id, Guid userId) public async Task<CipherDetails> GetByIdAsync(Guid id, Guid userId)
{ {
var cipher = await GetByIdAsync(id); using(var connection = new SqlConnection(ConnectionString))
if(cipher == null || cipher.UserId != userId)
{ {
return null; var results = await connection.QueryAsync<CipherDetails>(
} $"[{Schema}].[CipherDetails_ReadById]",
new { Id = id },
commandType: CommandType.StoredProcedure);
return cipher; return results.FirstOrDefault(c => c.UserId == userId);
}
} }
public async Task<ICollection<CipherDetails>> GetManyByUserIdAsync(Guid userId) public async Task<ICollection<CipherDetails>> GetManyByUserIdAsync(Guid userId)

View File

@ -1,6 +1,7 @@
using System.Collections.Generic; using System.Collections.Generic;
using System.Threading.Tasks; using System.Threading.Tasks;
using Bit.Core.Models.Table; using Bit.Core.Models.Table;
using Core.Models.Data;
namespace Bit.Core.Services namespace Bit.Core.Services
{ {
@ -8,7 +9,9 @@ namespace Bit.Core.Services
{ {
Task SaveAsync(Cipher cipher); Task SaveAsync(Cipher cipher);
Task DeleteAsync(Cipher cipher); Task DeleteAsync(Cipher cipher);
Task ImportCiphersAsync(List<Cipher> folders, List<Cipher> ciphers, Task SaveFolderAsync(Folder folder);
Task DeleteFolderAsync(Folder folder);
Task ImportCiphersAsync(List<Folder> folders, List<CipherDetails> ciphers,
IEnumerable<KeyValuePair<int, int>> folderRelationships); IEnumerable<KeyValuePair<int, int>> folderRelationships);
} }
} }

View File

@ -4,21 +4,25 @@ using System.Collections.Generic;
using System.Threading.Tasks; using System.Threading.Tasks;
using Bit.Core.Models.Table; using Bit.Core.Models.Table;
using Bit.Core.Repositories; using Bit.Core.Repositories;
using Core.Models.Data;
namespace Bit.Core.Services namespace Bit.Core.Services
{ {
public class CipherService : ICipherService public class CipherService : ICipherService
{ {
private readonly ICipherRepository _cipherRepository; private readonly ICipherRepository _cipherRepository;
private readonly IFolderRepository _folderRepository;
private readonly IUserRepository _userRepository; private readonly IUserRepository _userRepository;
private readonly IPushService _pushService; private readonly IPushService _pushService;
public CipherService( public CipherService(
ICipherRepository cipherRepository, ICipherRepository cipherRepository,
IFolderRepository folderRepository,
IUserRepository userRepository, IUserRepository userRepository,
IPushService pushService) IPushService pushService)
{ {
_cipherRepository = cipherRepository; _cipherRepository = cipherRepository;
_folderRepository = folderRepository;
_userRepository = userRepository; _userRepository = userRepository;
_pushService = pushService; _pushService = pushService;
} }
@ -50,16 +54,43 @@ namespace Bit.Core.Services
await _pushService.PushSyncCipherDeleteAsync(cipher); await _pushService.PushSyncCipherDeleteAsync(cipher);
} }
public async Task SaveFolderAsync(Folder folder)
{
if(folder.Id == default(Guid))
{
await _folderRepository.CreateAsync(folder);
// push
//await _pushService.PushSyncCipherCreateAsync(cipher);
}
else
{
folder.RevisionDate = DateTime.UtcNow;
await _folderRepository.UpsertAsync(folder);
// push
//await _pushService.PushSyncCipherUpdateAsync(cipher);
}
}
public async Task DeleteFolderAsync(Folder folder)
{
await _folderRepository.DeleteAsync(folder);
// push
//await _pushService.PushSyncCipherDeleteAsync(cipher);
}
public async Task ImportCiphersAsync( public async Task ImportCiphersAsync(
List<Cipher> folders, List<Folder> folders,
List<Cipher> ciphers, List<CipherDetails> ciphers,
IEnumerable<KeyValuePair<int, int>> folderRelationships) IEnumerable<KeyValuePair<int, int>> folderRelationships)
{ {
// create all the folders // create all the folders
var folderTasks = new List<Task>(); var folderTasks = new List<Task>();
foreach(var folder in folders) foreach(var folder in folders)
{ {
folderTasks.Add(_cipherRepository.CreateAsync(folder)); folderTasks.Add(_folderRepository.CreateAsync(folder));
} }
await Task.WhenAll(folderTasks); await Task.WhenAll(folderTasks);
@ -74,7 +105,7 @@ namespace Bit.Core.Services
continue; continue;
} }
cipher.FolderId = folder.Id; //cipher.FolderId = folder.Id;
} }
// create all the ciphers // create all the ciphers

View File

@ -54,9 +54,9 @@ namespace Bit.Core.Services
{ {
switch(cipher.Type) switch(cipher.Type)
{ {
case CipherType.Folder: //case CipherType.Folder:
await PushCipherAsync(cipher, PushType.SyncFolderDelete); // await PushCipherAsync(cipher, PushType.SyncFolderDelete);
break; // break;
case CipherType.Login: case CipherType.Login:
await PushCipherAsync(cipher, PushType.SyncLoginDelete); await PushCipherAsync(cipher, PushType.SyncLoginDelete);
break; break;