using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.Threading.Tasks; using Telegram.Bot.Types; using TelegramBotBase.Args; using TelegramBotBase.Base; using TelegramBotBase.Enums; namespace TelegramBotBase.Form { /// /// A form which cleans up old messages sent within /// public class AutoCleanForm : FormBase { List OldMessages { get; set; } public eDeleteMode DeleteMode { get; set; } public eDeleteSide DeleteSide { get; set; } public AutoCleanForm() { this.OldMessages = new List(); this.DeleteMode = eDeleteMode.OnEveryCall; this.DeleteSide = eDeleteSide.BotOnly; this.Init += AutoCleanForm_Init; this.Closed += AutoCleanForm_Closed; } private async Task AutoCleanForm_Init(object sender, InitEventArgs e) { if (this.Device == null) return; this.Device.MessageSent += Device_MessageSent; this.Device.MessageReceived += Device_MessageReceived; } private void Device_MessageReceived(object sender, MessageReceivedEventArgs e) { if (this.DeleteSide == eDeleteSide.BotOnly) return; this.OldMessages.Add(e.Message); } private void Device_MessageSent(object sender, MessageSentEventArgs e) { if (this.DeleteSide == eDeleteSide.UserOnly) return; this.OldMessages.Add(e.Message); } public override async Task PreLoad(MessageResult message) { if (this.DeleteMode != eDeleteMode.OnEveryCall) return; await MessageCleanup(); } /// /// Adds a message to this of removable ones /// /// public void AddMessage(Message m) { this.OldMessages.Add(m); } /// /// Keeps the message by removing it from the list /// /// public void LeaveMessage(int Id) { var m = this.OldMessages.FirstOrDefault(a => a.MessageId == Id); if (m == null) return; this.OldMessages.Remove(m); } /// /// Keeps the last sent message /// public void LeaveLastMessage() { if (this.OldMessages.Count == 0) return; this.OldMessages.RemoveAt(this.OldMessages.Count - 1); } private async Task AutoCleanForm_Closed(object sender, EventArgs e) { if (this.DeleteMode != eDeleteMode.OnLeavingForm) return; MessageCleanup().Wait(); } /// /// Cleans up all remembered messages. /// /// public async Task MessageCleanup() { while (this.OldMessages.Count > 0) { if (!await this.Device.DeleteMessage(this.OldMessages[0].MessageId)) { //Message can't be deleted cause it seems not to exist anymore if (this.OldMessages.Count > 0) this.OldMessages.RemoveAt(0); } } } } }