Add support for deleting reminders

This commit is contained in:
Starbeamrainbowlabs 2018-11-17 17:58:25 +00:00
parent bcc6c82c59
commit 6fafd3b480
Signed by: sbrl
GPG Key ID: 1BE5172E637709C2
3 changed files with 46 additions and 3 deletions

View File

@ -4,6 +4,7 @@ using System.Linq;
using System.Text.RegularExpressions; using System.Text.RegularExpressions;
using Microsoft.Recognizers.Text; using Microsoft.Recognizers.Text;
using Microsoft.Recognizers.Text.DateTime; using Microsoft.Recognizers.Text.DateTime;
using Microsoft.Recognizers.Text.Number;
using SBRL.Geometry; using SBRL.Geometry;
namespace RhinoReminds namespace RhinoReminds
@ -11,6 +12,14 @@ namespace RhinoReminds
public static class AIRecogniser public static class AIRecogniser
{ {
public static IEnumerable<int> RecogniseNumbers(string source)
{
List<ModelResult> aiResults = NumberRecognizer.RecognizeNumber(source, Culture.English);
foreach (ModelResult nextResult in aiResults) {
yield return int.Parse(nextResult.Text);
}
}
public static (DateTime, TimeSpan) RecogniseDateTimeRange(string source, out string rawString) public static (DateTime, TimeSpan) RecogniseDateTimeRange(string source, out string rawString)
{ {
List<ModelResult> aiResults = DateTimeRecognizer.RecognizeDateTime(source, Culture.English); List<ModelResult> aiResults = DateTimeRecognizer.RecognizeDateTime(source, Culture.English);

View File

@ -110,6 +110,33 @@ namespace RhinoReminds
switch (parts[0].ToLower()) switch (parts[0].ToLower())
{ {
case "delete":
case "remove":
List<int> failed = new List<int>(),
succeeded = new List<int>();
foreach (int nextId in AIRecogniser.RecogniseNumbers(message.Body)) {
Reminder nextReminder = reminderList.GetById(nextId);
if (nextReminder.JidObj != message.From.GetBareJid()) {
failed.Add(nextId);
continue;
}
reminderList.DeleteReminder(nextReminder);
succeeded.Add(nextId);
}
if (failed.Count > 0) {
string response = string.Join(", ", failed.Select((int nextId) => $"#{nextId}"));
response = $"Sorry! I can't delete reminder{(failed.Count != 1 ? "s" : "")} {response}, as you didn't create {(failed.Count != 1 ? "them":"it")}.";
sendChatReply(message, response);
}
if (succeeded.Count > 0) {
string response = string.Join(", ", succeeded.Select((int nextId) => $"#{nextId}"));
response = $"Deleted reminder{(succeeded.Count != 1 ? "s" : "")} {response} successfully.";
sendChatReply(message, response);
}
break;
case "list": case "list":
case "show": case "show":
if (parts.Select((n) => n.ToLower()).Contains("all")) { if (parts.Select((n) => n.ToLower()).Contains("all")) {
@ -165,6 +192,7 @@ namespace RhinoReminds
Reminder newReminder = reminderList.CreateReminder(message.From, dateTime, reminder); Reminder newReminder = reminderList.CreateReminder(message.From, dateTime, reminder);
reminderList.Save(ReminderFilePath); reminderList.Save(ReminderFilePath);
break; break;
default: default:
sendChatReply(message, "I don't understand that. Try rephrasing it or asking for help."); sendChatReply(message, "I don't understand that. Try rephrasing it or asking for help.");
break; break;

View File

@ -19,12 +19,14 @@ namespace RhinoReminds
public event OnReminderListUpdateHandler OnReminderListUpdate; public event OnReminderListUpdateHandler OnReminderListUpdate;
public ReminderList() { public ReminderList()
{
} }
public Reminder CreateReminder(Jid inJid, DateTime time, string message) { public Reminder CreateReminder(Jid inJid, DateTime time, string message)
{
Reminder result = new Reminder(nextId++, $"{inJid.Node}@{inJid.Domain}", time, message); Reminder result = new Reminder(nextId++, $"{inJid.Node}@{inJid.Domain}", time, message);
Console.WriteLine($"[Rhino/ReminderList] Created reminder {result}"); Console.WriteLine($"[Rhino/ReminderList] Created reminder {result}");
while (Reminders.ContainsKey(time)) while (Reminders.ContainsKey(time))
@ -34,7 +36,6 @@ namespace RhinoReminds
return result; return result;
} }
public Reminder GetNextReminder() public Reminder GetNextReminder()
{ {
if (Reminders.Count == 0) if (Reminders.Count == 0)
@ -43,6 +44,11 @@ namespace RhinoReminds
return Reminders.Values[0]; return Reminders.Values[0];
} }
public Reminder GetById(int targetId)
{
return Reminders.First((KeyValuePair<DateTime, Reminder> nextPair) => nextPair.Value.Id == targetId).Value;
}
public void DeleteReminder(Reminder nextReminder) public void DeleteReminder(Reminder nextReminder)
{ {
Reminders.Remove(nextReminder.Time); Reminders.Remove(nextReminder.Time);