79 lines
2.8 KiB
C#
79 lines
2.8 KiB
C#
// Copyright (C) 2021-present sam (starshines.gay)
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published
|
|
// by the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
using Catalogger.Backend.Database.Models;
|
|
using Dapper;
|
|
using Remora.Rest.Core;
|
|
|
|
namespace Catalogger.Backend.Database.Dapper.Repositories;
|
|
|
|
public class InviteRepository(ILogger logger, DatabaseConnection conn)
|
|
: IDisposable,
|
|
IAsyncDisposable
|
|
{
|
|
private readonly ILogger _logger = logger.ForContext<InviteRepository>();
|
|
|
|
public async Task<List<Invite>> GetGuildInvitesAsync(Snowflake guildId) =>
|
|
(
|
|
await conn.QueryAsync<Invite>(
|
|
"select * from invites where guild_id = @GuildId",
|
|
new { GuildId = guildId.Value }
|
|
)
|
|
).ToList();
|
|
|
|
public async Task SetInviteNameAsync(Snowflake guildId, string code, string name) =>
|
|
await conn.ExecuteAsync(
|
|
"""
|
|
insert into invites (code, guild_id, name) values
|
|
(@Code, @GuildId, @Name) on conflict (code, guild_id) do update set name = @Name
|
|
""",
|
|
new
|
|
{
|
|
Code = code,
|
|
GuildId = guildId.Value,
|
|
Name = name,
|
|
}
|
|
);
|
|
|
|
public async Task<Invite?> GetInviteAsync(Snowflake guildId, string code) =>
|
|
await conn.QueryFirstOrDefaultAsync<Invite>(
|
|
"select * from invites where guild_id = @GuildId and code = @Code",
|
|
new { GuildId = guildId.Value, Code = code }
|
|
);
|
|
|
|
public async Task<string> GetInviteNameAsync(Snowflake guildId, string code) =>
|
|
await conn.ExecuteScalarAsync<string>(
|
|
"select name from invites where guild_id = @GuildId and code = @Code",
|
|
new { GuildId = guildId.Value, Code = code }
|
|
) ?? "(unnamed)";
|
|
|
|
public async Task<int> DeleteInviteAsync(Snowflake guildId, string code) =>
|
|
await conn.ExecuteAsync(
|
|
"delete from invites where guild_id = @GuildId and code = @Code",
|
|
new { GuildId = guildId.Value, Code = code }
|
|
);
|
|
|
|
public void Dispose()
|
|
{
|
|
conn.Dispose();
|
|
GC.SuppressFinalize(this);
|
|
}
|
|
|
|
public async ValueTask DisposeAsync()
|
|
{
|
|
await conn.DisposeAsync();
|
|
GC.SuppressFinalize(this);
|
|
}
|
|
}
|