1
0
mirror of https://github.com/bitwarden/server.git synced 2025-02-09 00:41:37 +01:00

SelectionReadOnly MERGE to CollectionGroup

This commit is contained in:
Kyle Spearrin 2017-05-11 11:41:13 -04:00
parent f0d7dc8023
commit 2b8db4d1ed
18 changed files with 129 additions and 73 deletions

View File

@ -93,7 +93,7 @@ namespace Bit.Api.Controllers
} }
var group = model.ToGroup(orgIdGuid); var group = model.ToGroup(orgIdGuid);
await _groupService.SaveAsync(group, model.CollectionIds?.Select(c => new Guid(c))); await _groupService.SaveAsync(group, model.Collections?.Select(c => c.ToSelectionReadOnly()));
return new GroupResponseModel(group); return new GroupResponseModel(group);
} }
@ -107,7 +107,7 @@ namespace Bit.Api.Controllers
throw new NotFoundException(); throw new NotFoundException();
} }
await _groupService.SaveAsync(model.ToGroup(group), model.CollectionIds?.Select(c => new Guid(c))); await _groupService.SaveAsync(model.ToGroup(group), model.Collections?.Select(c => c.ToSelectionReadOnly()));
return new GroupResponseModel(group); return new GroupResponseModel(group);
} }

View File

@ -13,7 +13,7 @@ namespace Bit.Core.Models.Api
public string Name { get; set; } public string Name { get; set; }
[Required] [Required]
public bool? AccessAll { get; set; } public bool? AccessAll { get; set; }
public IEnumerable<string> CollectionIds { get; set; } public IEnumerable<SelectionReadOnlyRequestModel> Collections { get; set; }
public Group ToGroup(Guid orgId) public Group ToGroup(Guid orgId)
{ {

View File

@ -1,5 +1,4 @@
using Bit.Core.Models.Table; using Bit.Core.Models.Table;
using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.ComponentModel.DataAnnotations; using System.ComponentModel.DataAnnotations;
@ -13,7 +12,7 @@ namespace Bit.Core.Models.Api
[Required] [Required]
public Enums.OrganizationUserType? Type { get; set; } public Enums.OrganizationUserType? Type { get; set; }
public bool AccessAll { get; set; } public bool AccessAll { get; set; }
public IEnumerable<OrganizationUserCollectionRequestModel> Collections { get; set; } public IEnumerable<SelectionReadOnlyRequestModel> Collections { get; set; }
} }
public class OrganizationUserAcceptRequestModel public class OrganizationUserAcceptRequestModel
@ -33,7 +32,7 @@ namespace Bit.Core.Models.Api
[Required] [Required]
public Enums.OrganizationUserType? Type { get; set; } public Enums.OrganizationUserType? Type { get; set; }
public bool AccessAll { get; set; } public bool AccessAll { get; set; }
public IEnumerable<OrganizationUserCollectionRequestModel> Collections { get; set; } public IEnumerable<SelectionReadOnlyRequestModel> Collections { get; set; }
public OrganizationUser ToOrganizationUser(OrganizationUser existingUser) public OrganizationUser ToOrganizationUser(OrganizationUser existingUser)
{ {
@ -48,22 +47,4 @@ namespace Bit.Core.Models.Api
[Required] [Required]
public IEnumerable<string> GroupIds { get; set; } public IEnumerable<string> GroupIds { get; set; }
} }
public class OrganizationUserCollectionRequestModel
{
[Required]
public string CollectionId { get; set; }
public bool ReadOnly { get; set; }
public CollectionUser ToCollectionUser()
{
var collection = new CollectionUser
{
ReadOnly = ReadOnly,
CollectionId = new Guid(CollectionId)
};
return collection;
}
}
} }

View File

@ -0,0 +1,33 @@
using System;
using System.ComponentModel.DataAnnotations;
using Newtonsoft.Json;
using Bit.Core.Models.Table;
using Bit.Core.Models.Data;
namespace Bit.Core.Models.Api
{
public class SelectionReadOnlyRequestModel
{
[Required]
public string Id { get; set; }
public bool ReadOnly { get; set; }
public CollectionUser ToCollectionUser()
{
return new CollectionUser
{
ReadOnly = ReadOnly,
CollectionId = new Guid(Id)
};
}
public SelectionReadOnly ToSelectionReadOnly()
{
return new SelectionReadOnly
{
Id = new Guid(Id),
ReadOnly = ReadOnly
};
}
}
}

View File

@ -1,6 +1,8 @@
using System; using System;
using Bit.Core.Models.Table; using Bit.Core.Models.Table;
using System.Collections.Generic; using System.Collections.Generic;
using Bit.Core.Models.Data;
using System.Linq;
namespace Bit.Core.Models.Api namespace Bit.Core.Models.Api
{ {
@ -28,12 +30,12 @@ namespace Bit.Core.Models.Api
public class GroupDetailsResponseModel : GroupResponseModel public class GroupDetailsResponseModel : GroupResponseModel
{ {
public GroupDetailsResponseModel(Group group, IEnumerable<Guid> collectionIds) public GroupDetailsResponseModel(Group group, IEnumerable<SelectionReadOnly> collections)
: base(group, "groupDetails") : base(group, "groupDetails")
{ {
CollectionIds = collectionIds; Collections = collections.Select(c => new SelectionReadOnlyResponseModel(c));
} }
public IEnumerable<Guid> CollectionIds { get; set; } public IEnumerable<SelectionReadOnlyResponseModel> Collections { get; set; }
} }
} }

View File

@ -52,27 +52,10 @@ namespace Bit.Core.Models.Api
IEnumerable<SelectionReadOnly> collections) IEnumerable<SelectionReadOnly> collections)
: base(organizationUser, "organizationUserDetails") : base(organizationUser, "organizationUserDetails")
{ {
Collections = collections.Select(c => new CollectionSelection(c)); Collections = collections.Select(c => new SelectionReadOnlyResponseModel(c));
} }
public IEnumerable<CollectionSelection> Collections { get; set; } public IEnumerable<SelectionReadOnlyResponseModel> Collections { get; set; }
public class CollectionSelection
{
public CollectionSelection(Data.SelectionReadOnly selection)
{
if(selection == null)
{
throw new ArgumentNullException(nameof(selection));
}
Id = selection.Id.ToString();
ReadOnly = selection.ReadOnly;
}
public string Id { get; set; }
public bool ReadOnly { get; set; }
}
} }
public class OrganizationUserUserDetailsResponseModel : OrganizationUserResponseModel public class OrganizationUserUserDetailsResponseModel : OrganizationUserResponseModel
{ {

View File

@ -0,0 +1,22 @@
using System;
using Bit.Core.Models.Data;
namespace Bit.Core.Models.Api
{
public class SelectionReadOnlyResponseModel
{
public SelectionReadOnlyResponseModel(SelectionReadOnly selection)
{
if(selection == null)
{
throw new ArgumentNullException(nameof(selection));
}
Id = selection.Id.ToString();
ReadOnly = selection.ReadOnly;
}
public string Id { get; set; }
public bool ReadOnly { get; set; }
}
}

View File

@ -5,7 +5,7 @@ namespace Bit.Core.Models.Table
public class CollectionGroup public class CollectionGroup
{ {
public Guid CollectionId { get; set; } public Guid CollectionId { get; set; }
public Guid OrganizationUserId { get; set; } public Guid GroupId { get; set; }
public bool ReadOnly { get; set; } public bool ReadOnly { get; set; }
} }
} }

View File

@ -8,12 +8,12 @@ namespace Bit.Core.Repositories
{ {
public interface IGroupRepository : IRepository<Group, Guid> public interface IGroupRepository : IRepository<Group, Guid>
{ {
Task<Tuple<Group, ICollection<Guid>>> GetByIdWithCollectionsAsync(Guid id); Task<Tuple<Group, ICollection<SelectionReadOnly>>> GetByIdWithCollectionsAsync(Guid id);
Task<ICollection<Group>> GetManyByOrganizationIdAsync(Guid organizationId); Task<ICollection<Group>> GetManyByOrganizationIdAsync(Guid organizationId);
Task<ICollection<GroupUserUserDetails>> GetManyUserDetailsByIdAsync(Guid id); Task<ICollection<GroupUserUserDetails>> GetManyUserDetailsByIdAsync(Guid id);
Task<ICollection<Guid>> GetManyIdsByUserIdAsync(Guid organizationUserId); Task<ICollection<Guid>> GetManyIdsByUserIdAsync(Guid organizationUserId);
Task CreateAsync(Group obj, IEnumerable<Guid> collectionIds); Task CreateAsync(Group obj, IEnumerable<SelectionReadOnly> collections);
Task ReplaceAsync(Group obj, IEnumerable<Guid> collectionIds); Task ReplaceAsync(Group obj, IEnumerable<SelectionReadOnly> collections);
Task DeleteUserAsync(Guid groupId, Guid organizationUserId); Task DeleteUserAsync(Guid groupId, Guid organizationUserId);
} }
} }

View File

@ -22,7 +22,7 @@ namespace Bit.Core.Repositories.SqlServer
: base(connectionString) : base(connectionString)
{ } { }
public async Task<Tuple<Group, ICollection<Guid>>> GetByIdWithCollectionsAsync(Guid id) public async Task<Tuple<Group, ICollection<SelectionReadOnly>>> GetByIdWithCollectionsAsync(Guid id)
{ {
using(var connection = new SqlConnection(ConnectionString)) using(var connection = new SqlConnection(ConnectionString))
{ {
@ -32,9 +32,9 @@ namespace Bit.Core.Repositories.SqlServer
commandType: CommandType.StoredProcedure); commandType: CommandType.StoredProcedure);
var group = await results.ReadFirstOrDefaultAsync<Group>(); var group = await results.ReadFirstOrDefaultAsync<Group>();
var colletionIds = (await results.ReadAsync<Guid>()).ToList(); var colletions = (await results.ReadAsync<SelectionReadOnly>()).ToList();
return new Tuple<Group, ICollection<Guid>>(group, colletionIds); return new Tuple<Group, ICollection<SelectionReadOnly>>(group, colletions);
} }
} }
@ -77,11 +77,11 @@ namespace Bit.Core.Repositories.SqlServer
} }
} }
public async Task CreateAsync(Group obj, IEnumerable<Guid> collectionIds) public async Task CreateAsync(Group obj, IEnumerable<SelectionReadOnly> collections)
{ {
obj.SetNewId(); obj.SetNewId();
var objWithCollections = JsonConvert.DeserializeObject<GroupWithCollections>(JsonConvert.SerializeObject(obj)); var objWithCollections = JsonConvert.DeserializeObject<GroupWithCollections>(JsonConvert.SerializeObject(obj));
objWithCollections.CollectionIds = collectionIds.ToGuidIdArrayTVP(); objWithCollections.Collections = collections.ToArrayTVP();
using(var connection = new SqlConnection(ConnectionString)) using(var connection = new SqlConnection(ConnectionString))
{ {
@ -92,10 +92,10 @@ namespace Bit.Core.Repositories.SqlServer
} }
} }
public async Task ReplaceAsync(Group obj, IEnumerable<Guid> collectionIds) public async Task ReplaceAsync(Group obj, IEnumerable<SelectionReadOnly> collections)
{ {
var objWithCollections = JsonConvert.DeserializeObject<GroupWithCollections>(JsonConvert.SerializeObject(obj)); var objWithCollections = JsonConvert.DeserializeObject<GroupWithCollections>(JsonConvert.SerializeObject(obj));
objWithCollections.CollectionIds = collectionIds.ToGuidIdArrayTVP(); objWithCollections.Collections = collections.ToArrayTVP();
using(var connection = new SqlConnection(ConnectionString)) using(var connection = new SqlConnection(ConnectionString))
{ {
@ -119,7 +119,7 @@ namespace Bit.Core.Repositories.SqlServer
public class GroupWithCollections : Group public class GroupWithCollections : Group
{ {
public DataTable CollectionIds { get; set; } public DataTable Collections { get; set; }
} }
} }
} }

View File

@ -1,12 +1,12 @@
using System.Threading.Tasks; using System.Threading.Tasks;
using Bit.Core.Models.Table; using Bit.Core.Models.Table;
using System.Collections.Generic; using System.Collections.Generic;
using System; using Bit.Core.Models.Data;
namespace Bit.Core.Services namespace Bit.Core.Services
{ {
public interface IGroupService public interface IGroupService
{ {
Task SaveAsync(Group group, IEnumerable<Guid> collectionIds = null); Task SaveAsync(Group group, IEnumerable<SelectionReadOnly> collections = null);
} }
} }

View File

@ -4,6 +4,7 @@ using Bit.Core.Exceptions;
using Bit.Core.Models.Table; using Bit.Core.Models.Table;
using Bit.Core.Repositories; using Bit.Core.Repositories;
using System.Collections.Generic; using System.Collections.Generic;
using Bit.Core.Models.Data;
namespace Bit.Core.Services namespace Bit.Core.Services
{ {
@ -20,7 +21,7 @@ namespace Bit.Core.Services
_groupRepository = groupRepository; _groupRepository = groupRepository;
} }
public async Task SaveAsync(Group group, IEnumerable<Guid> collectionIds = null) public async Task SaveAsync(Group group, IEnumerable<SelectionReadOnly> collections = null)
{ {
var org = await _organizationRepository.GetByIdAsync(group.OrganizationId); var org = await _organizationRepository.GetByIdAsync(group.OrganizationId);
if(org == null) if(org == null)
@ -35,24 +36,24 @@ namespace Bit.Core.Services
if(group.Id == default(Guid)) if(group.Id == default(Guid))
{ {
if(collectionIds == null) if(collections == null)
{ {
await _groupRepository.CreateAsync(group); await _groupRepository.CreateAsync(group);
} }
else else
{ {
await _groupRepository.CreateAsync(group, collectionIds); await _groupRepository.CreateAsync(group, collections);
} }
} }
else else
{ {
if(collectionIds == null) if(collections == null)
{ {
await _groupRepository.ReplaceAsync(group); await _groupRepository.ReplaceAsync(group);
} }
else else
{ {
await _groupRepository.ReplaceAsync(group, collectionIds); await _groupRepository.ReplaceAsync(group, collections);
} }
} }
} }

View File

@ -1,4 +1,6 @@
using Dapper; using Bit.Core.Models.Data;
using Bit.Core.Models.Table;
using Dapper;
using System; using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Data; using System.Data;
@ -65,6 +67,30 @@ namespace Bit.Core.Utilities
return table; return table;
} }
public static DataTable ToArrayTVP(this IEnumerable<SelectionReadOnly> values)
{
var table = new DataTable();
table.SetTypeName("[dbo].[SelectionReadOnlyArray]");
var idColumn = new DataColumn("Id", typeof(Guid));
table.Columns.Add(idColumn);
var readOnlyColumn = new DataColumn("ReadOnly", typeof(bool));
table.Columns.Add(readOnlyColumn);
if(values != null)
{
foreach(var value in values)
{
var row = table.NewRow();
row[idColumn] = value.Id;
row[readOnlyColumn] = value.ReadOnly;
table.Rows.Add(row);
}
}
return table;
}
public static X509Certificate2 GetCertificate(string thumbprint) public static X509Certificate2 GetCertificate(string thumbprint)
{ {
// Clean possible garbage characters from thumbprint copy/paste // Clean possible garbage characters from thumbprint copy/paste

View File

@ -193,5 +193,6 @@
<Build Include="dbo\Stored Procedures\GroupUser_ReadGroupIdsByOrganizationUserId.sql" /> <Build Include="dbo\Stored Procedures\GroupUser_ReadGroupIdsByOrganizationUserId.sql" />
<Build Include="dbo\Stored Procedures\GroupUser_UpdateGroups.sql" /> <Build Include="dbo\Stored Procedures\GroupUser_UpdateGroups.sql" />
<Build Include="dbo\Stored Procedures\GroupUser_Delete.sql" /> <Build Include="dbo\Stored Procedures\GroupUser_Delete.sql" />
<Build Include="dbo\User Defined Types\SelectionReadOnlyArray.sql" />
</ItemGroup> </ItemGroup>
</Project> </Project>

View File

@ -5,7 +5,7 @@
@AccessAll BIT, @AccessAll BIT,
@CreationDate DATETIME2(7), @CreationDate DATETIME2(7),
@RevisionDate DATETIME2(7), @RevisionDate DATETIME2(7),
@CollectionIds AS [dbo].[GuidIdArray] READONLY @Collections AS [dbo].[SelectionReadOnlyArray] READONLY
AS AS
BEGIN BEGIN
SET NOCOUNT ON SET NOCOUNT ON
@ -29,9 +29,9 @@ BEGIN
SELECT SELECT
[Id], [Id],
@Id, @Id,
0 [ReadOnly]
FROM FROM
@CollectionIds @Collections
WHERE WHERE
[Id] IN (SELECT [Id] FROM [AvailableCollectionsCTE]) [Id] IN (SELECT [Id] FROM [AvailableCollectionsCTE])
END END

View File

@ -7,7 +7,8 @@ BEGIN
EXEC [dbo].[Group_ReadById] @Id EXEC [dbo].[Group_ReadById] @Id
SELECT SELECT
[CollectionId] [CollectionId] [Id],
[ReadOnly]
FROM FROM
[dbo].[CollectionGroup] [dbo].[CollectionGroup]
WHERE WHERE

View File

@ -5,7 +5,7 @@
@AccessAll BIT, @AccessAll BIT,
@CreationDate DATETIME2(7), @CreationDate DATETIME2(7),
@RevisionDate DATETIME2(7), @RevisionDate DATETIME2(7),
@CollectionIds AS [dbo].[GuidIdArray] READONLY @Collections AS [dbo].[SelectionReadOnlyArray] READONLY
AS AS
BEGIN BEGIN
SET NOCOUNT ON SET NOCOUNT ON
@ -23,7 +23,7 @@ BEGIN
MERGE MERGE
[dbo].[CollectionGroup] AS [Target] [dbo].[CollectionGroup] AS [Target]
USING USING
@CollectionIds AS [Source] @Collections AS [Source]
ON ON
[Target].[CollectionId] = [Source].[Id] [Target].[CollectionId] = [Source].[Id]
AND [Target].[GroupId] = @Id AND [Target].[GroupId] = @Id
@ -33,8 +33,10 @@ BEGIN
( (
[Source].[Id], [Source].[Id],
@Id, @Id,
0 [Source].[ReadOnly]
) )
WHEN MATCHED AND [Target].[ReadOnly] != [Source].[ReadOnly] THEN
UPDATE SET [Target].[ReadOnly] = [Source].[ReadOnly]
WHEN NOT MATCHED BY SOURCE WHEN NOT MATCHED BY SOURCE
AND [Target].[GroupId] = @Id THEN AND [Target].[GroupId] = @Id THEN
DELETE DELETE

View File

@ -0,0 +1,4 @@
CREATE TYPE [dbo].[SelectionReadOnlyArray] AS TABLE (
[Id] UNIQUEIDENTIFIER NOT NULL,
[ReadOnly] BIT NOT NULL);