feat(States): add a new state

TelegramBot/States/DeleteContact.cs

@@ -0,0 +1,127 @@
+using Telegram.Bot;
+using MediaTelegramBot.Utils;
+using Telegram.Bot.Types;
+using Telegram.Bot.Types.ReplyMarkups;
+using Telegram.Bot.Types.Enums;
+using TelegramMediaRelayBot;
+using DataBase;
+
+
+namespace MediaTelegramBot;
+
+public class ProcessRemoveUser : IUserState
+{
+    public UsersStandartState currentState;
+    private List<int> preparedTargetUserIds = new List<int>();
+    private Message statusMessage;
+    bool isDeleteSuccessful = false;
+
+    public ProcessRemoveUser(Message statusMessage)
+    {
+        this.statusMessage = statusMessage;
+        currentState = UsersStandartState.ProcessAction;
+    }
+
+    public string GetCurrentState()
+    {
+        return currentState.ToString();
+    }
+
+    public async Task ProcessState(ITelegramBotClient botClient, Update update, CancellationToken cancellationToken)
+    {
+        long chatId = Utils.Utils.GetIDfromUpdate(update);
+
+        switch (currentState)
+        {
+            case UsersStandartState.ProcessAction:
+                if (await Utils.Utils.HandleStateBreakCommand(botClient, update, chatId, removeReplyMarkup: false)) return;
+                if (update.Message != null)
+                {
+                    string input = update.Message.Text;
+                    if (string.IsNullOrWhiteSpace(input))
+                    {
+                        await botClient.SendMessage(chatId, Config.GetResourceString("PleaseEnterContactIDs"), cancellationToken: cancellationToken);
+                        return;
+                    }
+
+                    string[] ids = input.Split(' ');
+                    if (ids.All(id => int.TryParse(id, out _)))
+                    {
+                        preparedTargetUserIds = ids.Select(int.Parse).ToList();
+                        bool isSuccessful = await RetrieveAndDisplayUserInfo(botClient, update, chatId, cancellationToken);
+                        if (isSuccessful) currentState = UsersStandartState.ProcessData;
+                    }
+                    else
+                    {
+                        await botClient.SendMessage(chatId, Config.GetResourceString("InvalidInputNumbers"), cancellationToken: cancellationToken);
+                    }
+                }
+                break;
+
+            case UsersStandartState.ProcessData:
+                if (await Utils.Utils.HandleStateBreakCommand(botClient, update, chatId, removeReplyMarkup: false)) return;
+
+                if (update.CallbackQuery != null)
+                {
+                    string callbackData = update.CallbackQuery.Data!;
+                    if (callbackData == "confirm_removal")
+                    {
+                        RemoveUsersFromContacts(botClient, chatId, cancellationToken);
+                        await botClient.EditMessageText(chatId, statusMessage.MessageId, Config.GetResourceString("RemovalProcessCompleted"), cancellationToken: cancellationToken);
+
+                        string text = isDeleteSuccessful ? Config.GetResourceString("SuccessActionResult") : Config.GetResourceString("ErrorActionResult");
+                        await KeyboardUtils.SendInlineKeyboardMenu(botClient, update, cancellationToken, text);
+
+                        TelegramBot.userStates.Remove(chatId);
+                    }
+                    else if (callbackData == "cancel_removal")
+                    {
+                        await botClient.EditMessageText(chatId, statusMessage.MessageId, Config.GetResourceString("PleaseEnterContactIDs"), replyMarkup: KeyboardUtils.GetReturnButtonMarkup(), cancellationToken: cancellationToken);
+                        currentState = UsersStandartState.ProcessAction;
+                    }
+                }
+                break;
+        }
+    }
+
+    private async Task<bool> RetrieveAndDisplayUserInfo(ITelegramBotClient botClient, Update update, long chatId, CancellationToken cancellationToken)
+    {
+        List<long> contactUserTGIds = await CoreDB.GetAllContactUserTGIds(DBforGetters.GetUserIDbyTelegramID(chatId));
+        List<long> preparedTargetUserTGIds = preparedTargetUserIds.Select(id => DBforGetters.GetTelegramIDbyUserID(id)).ToList();
+
+        contactUserTGIds = contactUserTGIds.Intersect(preparedTargetUserTGIds).ToList();
+        List<string> contactUsersInfo = new List<string>();
+
+        foreach (var contactUserId in contactUserTGIds)
+        {
+            int id = DBforGetters.GetUserIDbyTelegramID(contactUserId);
+            string username = DBforGetters.GetUserNameByTelegramID(contactUserId);
+            string link = DBforGetters.GetSelfLink(contactUserId);
+
+            contactUsersInfo.Add(string.Format(Config.GetResourceString("ContactInfo"), id, username, link));
+        }
+
+        if (contactUsersInfo.Any())
+        {
+            string messageText = $"{Config.GetResourceString("ConfirmRemovalMessage")}\n\n{string.Join("\n", contactUsersInfo)}";
+            InlineKeyboardMarkup keyboard = Utils.KeyboardUtils.GetConfirmForActionKeyboardMarkup("confirm_removal", "cancel_removal");
+
+            await botClient.EditMessageText(chatId, statusMessage.MessageId, messageText, replyMarkup: keyboard, cancellationToken: cancellationToken, parseMode: ParseMode.Html);
+            return true;
+        }
+        else
+        {
+            await botClient.EditMessageText(chatId, statusMessage.MessageId, Config.GetResourceString("NoUsersFound"), cancellationToken: cancellationToken);
+        }
+        return false;
+    }
+
+    private void RemoveUsersFromContacts(ITelegramBotClient botClient, long chatId, CancellationToken cancellationToken)
+    {
+        foreach (int contactId in preparedTargetUserIds)
+        {
+            int userId = DBforGetters.GetUserIDbyTelegramID(chatId);
+            isDeleteSuccessful = CoreDB.RemoveUserFromContacts(userId, contactId);
+        }
+    }
+}