PluralKit/PluralKit.Core/Database/Repository/ModelRepository.CommandMessage.cs

using System.Threading.Tasks;

using Dapper;

namespace PluralKit.Core
{
    public partial class ModelRepository
	{
		public Task SaveCommandMessage(IPKConnection conn, ulong messageId, ulong authorId) =>
			conn.QueryAsync("insert into command_messages (message_id, author_id) values (@Message, @Author)",
				new {Message = messageId, Author = authorId });

		public Task<CommandMessage> GetCommandMessage(IPKConnection conn, ulong messageId) =>
			conn.QuerySingleOrDefaultAsync<CommandMessage>("select * from command_messages where message_id = @Message", 
				new {Message = messageId});

        public Task<int> DeleteCommandMessagesBefore(IPKConnection conn, ulong messageIdThreshold) =>
            conn.ExecuteAsync("delete from command_messages where message_id < @Threshold",
                new {Threshold = messageIdThreshold});
    }

	public class CommandMessage
	{
        public ulong AuthorId { get; set; }
        public ulong MessageId { get; set; }
	}
}