liz/PluralKit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

group import/export

Browse Source
This commit is contained in:
spiral
2021-09-21 21:42:41 -04:00
parent a09f819924
commit bc2c198a82
8 changed files with 219 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
PluralKit.Core/Utils/BulkImporter/BulkImporter.cs
View File

@@ -25,7 +25,12 @@ namespace PluralKit.Core
private readonly Dictionary<string, MemberId> _existingMemberHids = new();
private readonly Dictionary<string, MemberId> _existingMemberNames = new();
private readonly Dictionary<string, MemberId> _knownIdentifiers = new();
private readonly Dictionary<string, MemberId> _knownMemberIdentifiers = new();
private readonly Dictionary<string, GroupId> _existingGroupHids = new();
private readonly Dictionary<string, GroupId> _existingGroupNames = new();
private readonly Dictionary<string, GroupId> _knownGroupIdentifiers = new();
private ImportResultNew _result = new();
internal static async Task<ImportResultNew> PerformImport(IPKConnection conn, IPKTransaction tx, ModelRepository repo, ILogger logger,
@@ -58,6 +63,15 @@ namespace PluralKit.Core
importer._existingMemberNames[m.Name] = m.Id;
}
// same as above for groups
var groups = await conn.QueryAsync<PKGroup>("select id, hid, name from groups where system = @System",
new { System = system.Id });
foreach (var g in groups)
{
importer._existingGroupHids[g.Hid] = g.Id;
importer._existingGroupNames[g.Name] = g.Id;
}
try
{
if (importFile.ContainsKey("tuppers"))
@@ -89,7 +103,14 @@ namespace PluralKit.Core
return (null, false);
}
private async Task AssertLimitNotReached(int newMembers)
private (GroupId?, bool) TryGetExistingGroup(string hid, string name)
{
if (_existingGroupHids.TryGetValue(hid, out var byHid)) return (byHid, true);
if (_existingGroupNames.TryGetValue(name, out var byName)) return (byName, false);
return (null, false);
}
private async Task AssertMemberLimitNotReached(int newMembers)
{
var memberLimit = _system.MemberLimitOverride ?? Limits.MaxMemberCount;
var existingMembers = await _repo.GetSystemMemberCount(_conn, _system.Id);
@@ -97,6 +118,14 @@ namespace PluralKit.Core
throw new ImportException($"Import would exceed the maximum number of members ({memberLimit}).");
}
private async Task AssertGroupLimitNotReached(int newGroups)
{
var limit = _system.GroupLimitOverride ?? Limits.MaxGroupCount;
var existing = await _repo.GetSystemGroupCount(_conn, _system.Id);
if (existing + newGroups > limit)
throw new ImportException($"Import would exceed the maximum number of groups ({limit}).");
}
public async ValueTask DisposeAsync()
{
// try rolling back the transaction