using JsonFlatFileDataStore;
using Microsoft.Extensions.Logging;
using mstdnCats.Models;
using Telegram.Bot;
using Telegram.Bot.Types;
using Telegram.Bot.Types.Enums;
using Telegram.Bot.Types.ReplyMarkups;

namespace mstdnCats.Services
{
    public class HandlePostAction
    {
        public static async Task HandleCallbackQuery(CallbackQuery callbackQuery, IDocumentCollection<Post> _db, TelegramBotClient _bot, ILogger<MastodonBot>? logger)
        {
            // Extract media ID from callback query data
            string[] parts = callbackQuery.Data.Split('-');
            if (parts.Length != 2)
            {
                logger?.LogError("Invalid callback query data format.");
                return;
            }

            string action = parts[0];
            string mediaId = parts[1];

            var post = _db.AsQueryable().FirstOrDefault(p => p.MediaAttachments.Any(m => m.MediaId == mediaId));

            if (post == null)
            {
                logger?.LogInformation("No matching post found.");
                return;
            }

            // Approve the media attachment
            if (action == "approve")
            {
                var mediaAttachment = post.MediaAttachments.FirstOrDefault(m => m.MediaId == mediaId);
                if (mediaAttachment != null)
                {
                    // Check if the media attachment is already approved
                    bool isAlreadyApproved = mediaAttachment.Approved;
                    if (isAlreadyApproved){
                        await _bot.AnswerCallbackQueryAsync(callbackQuery.Id, "Media attachment is already approved.",true);
                        return;
                    }
                    
                    mediaAttachment.Approved = true;

                    bool updated = await _db.UpdateOneAsync(p => p.mstdnPostId == post.mstdnPostId, post);
                    

                    if (updated)
                    {
                        // Send the media attachment to the channel
                        await _bot.SendPhotoAsync(DotNetEnv.Env.GetString("CHANNEL_NUMID"), post.MediaAttachments.First().Url, caption: $"Post from " + $"<a href=\"" + post.Account.Url + "\">" + post.Account.DisplayName + " </a>", parseMode: ParseMode.Html
                        , replyMarkup: new InlineKeyboardMarkup(InlineKeyboardButton.WithUrl("View on Mastodon", post.Url)));

                        await _bot.AnswerCallbackQueryAsync(callbackQuery.Id, "Media attachment approved and sent to channel.");
                        await _bot.DeleteMessageAsync(callbackQuery.Message.Chat.Id, callbackQuery.Message.MessageId);

                        logger?.LogTrace($"Media attachment {mediaId} approved.");

                    }
                    else
                    {
                        logger?.LogError($"Failed to update the media attachment {mediaId}. Record might not exist or was not found.");
                    }


                }
                else
                {
                    logger?.LogError($"No media attachment found with MediaId {mediaId}.");
                }
            }

            // Reject the media attachment
            else if (action == "reject")
            {
                // Check if the post has only one attachment, if so, do not delete it, else delete the associated attachment
                if (post.MediaAttachments.Count == 1 && post.MediaAttachments.First().MediaId == mediaId)
                {
                    await _bot.AnswerCallbackQueryAsync(callbackQuery.Id, "Post has only one attachment. No deletion performed.");
                    await _bot.DeleteMessageAsync(callbackQuery.Message.Chat.Id, callbackQuery.Message.MessageId);

                    logger?.LogTrace($"Post {post.mstdnPostId} has only one attachment. No deletion performed.");
                }
                else
                {
                    post.MediaAttachments.RemoveAll(m => m.MediaId == mediaId);
                    await _db.UpdateOneAsync(p => p.mstdnPostId == post.mstdnPostId, post);
                    await _bot.AnswerCallbackQueryAsync(callbackQuery.Id, "Media attachment rejected.");
                    await _bot.DeleteMessageAsync(callbackQuery.Message.Chat.Id, callbackQuery.Message.MessageId);

                    logger?.LogTrace($"Media attachment {mediaId} removed from post {post.mstdnPostId}.");
                }
            }
            else
            {
                logger?.LogError("Invalid action specified.");
            }

        }
    }
}