Updated a few things
Signed-off-by: Emotion <emotion@emotionchild.com>
This commit is contained in:
parent
218672665c
commit
12557ea609
5 changed files with 45 additions and 16 deletions
|
@ -1,4 +1,6 @@
|
|||
#nullable disable
|
||||
using Ellie.Services.Database.Models;
|
||||
|
||||
namespace Ellie.Modules.Utility.Services;
|
||||
|
||||
public interface IRemindService
|
||||
|
@ -8,5 +10,6 @@ public interface IRemindService
|
|||
ulong? guildId,
|
||||
bool isPrivate,
|
||||
DateTime time,
|
||||
string message);
|
||||
string message,
|
||||
ReminderType reminderType);
|
||||
}
|
|
@ -9,4 +9,11 @@ public class Reminder : DbEntity
|
|||
public ulong UserId { get; set; }
|
||||
public string Message { get; set; }
|
||||
public bool IsPrivate { get; set; }
|
||||
public ReminderType Type { get; set; }
|
||||
}
|
||||
|
||||
public enum ReminderType
|
||||
{
|
||||
User,
|
||||
Timely
|
||||
}
|
|
@ -149,11 +149,25 @@ public partial class Gambling : GamblingModule<GamblingService>
|
|||
ctx.Guild?.Id,
|
||||
true,
|
||||
when,
|
||||
GetText(strs.timely_time));
|
||||
GetText(strs.timely_time),
|
||||
ReminderType.Timely);
|
||||
|
||||
await smc.RespondConfirmAsync(_eb, GetText(strs.remind_timely(tt)), ephemeral: true);
|
||||
}
|
||||
|
||||
private EllieInteraction CreateRemindMeInteraction(int period)
|
||||
{
|
||||
return _inter
|
||||
.Create(ctx.User.Id,
|
||||
new SimpleInteraction<DateTime>(
|
||||
new ButtonBuilder(
|
||||
label: "Remind me",
|
||||
emote: Emoji.Parse("⏰"),
|
||||
customId: "timely:remind_me"),
|
||||
RemindTimelyAction,
|
||||
DateTime.UtcNow.Add(TimeSpan.FromHours(period))));
|
||||
}
|
||||
|
||||
[Cmd]
|
||||
public async Task Timely()
|
||||
{
|
||||
|
@ -165,11 +179,19 @@ public partial class Gambling : GamblingModule<GamblingService>
|
|||
return;
|
||||
}
|
||||
|
||||
var inter = CreateRemindMeInteraction(period);
|
||||
|
||||
if (await _service.ClaimTimelyAsync(ctx.User.Id, period) is { } rem)
|
||||
{
|
||||
// Removes timely button if there is a timely reminder in DB
|
||||
if (_service.UserHasTimelyReminder(ctx.User.Id))
|
||||
{
|
||||
inter = null;
|
||||
}
|
||||
|
||||
var now = DateTime.UtcNow;
|
||||
var relativeTag = TimestampTag.FromDateTime(now.Add(rem), TimestampTagStyles.Relative);
|
||||
await ReplyPendingLocalizedAsync(strs.timely_already_claimed(relativeTag));
|
||||
await ReplyPendingLocalizedAsync(strs.timely_already_claimed(relativeTag), inter);
|
||||
return;
|
||||
}
|
||||
|
||||
|
@ -179,16 +201,6 @@ public partial class Gambling : GamblingModule<GamblingService>
|
|||
|
||||
await _cs.AddAsync(ctx.User.Id, val, new("timely", "claim"));
|
||||
|
||||
var inter = _inter
|
||||
.Create(ctx.User.Id,
|
||||
new SimpleInteraction<DateTime>(
|
||||
new ButtonBuilder(
|
||||
label: "Remind me",
|
||||
emote: Emoji.Parse("⏰"),
|
||||
customId: "timely:remind_me"),
|
||||
RemindTimelyAction,
|
||||
DateTime.UtcNow.Add(TimeSpan.FromHours(period))));
|
||||
|
||||
await ReplyConfirmLocalizedAsync(strs.timely(N(val), period), inter);
|
||||
}
|
||||
|
||||
|
|
|
@ -209,6 +209,13 @@ public class GamblingService : IEService, IReadyExecutor
|
|||
}
|
||||
}
|
||||
|
||||
public bool UserHasTimelyReminder(ulong userId)
|
||||
{
|
||||
var db = _db.GetDbContext();
|
||||
return db.GetTable<Reminder>().Any(x => x.UserId == userId
|
||||
&& x.Type == ReminderType.Timely);
|
||||
}
|
||||
|
||||
public async Task RemoveAllTimelyClaimsAsync()
|
||||
=> await _cache.RemoveAsync(_timelyKey);
|
||||
}
|
||||
|
|
|
@ -63,7 +63,7 @@ helpText: |-
|
|||
"inline": false
|
||||
},
|
||||
{
|
||||
"name": "Nadeko Support Server",
|
||||
"name": "Ellie Support Server",
|
||||
"value": "https://discord.elliebot.net/ ",
|
||||
"inline": true
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue