eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-04 13:06:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Add group add/remove commands

Browse source
This commit is contained in:
Ske 2020-07-07 15:28:53 +02:00
parent 6c5cb8cea7
commit 0f4c40b344
3 changed files with 63 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

20
PluralKit.Core/Models/Patch/ModelPatchExt.cs
View file

@ -1,4 +1,6 @@
using System.Threading.Tasks;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Dapper;
@ -73,5 +75,21 @@ namespace PluralKit.Core
.Build("returning *");
return conn.QueryFirstAsync<PKGroup>(query, pms);
}
public static async Task AddMembersToGroup(this IPKConnection conn, GroupId group, IEnumerable<MemberId> members)
{
await using var w = conn.BeginBinaryImport("copy group_members (group_id, member_id) from stdin (format binary)");
foreach (var member in members)
{
await w.StartRowAsync();
await w.WriteAsync(group.Value);
await w.WriteAsync(member.Value);
}
await w.CompleteAsync();
}
public static Task RemoveMembersFromGroup(this IPKConnection conn, GroupId group, IEnumerable<MemberId> members) =>
conn.ExecuteAsync("delete from group_members where group_id = @Group and member_id = any(@Members)",
new {Group = group, Members = members.ToArray() });
}
}