108 lines
4.5 KiB
C#
Executable File
108 lines
4.5 KiB
C#
Executable File
using CatsOfMastodonBot.Models;
|
|
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)
|
|
{
|
|
var config = configData.fetchData();
|
|
|
|
// 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(config.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.");
|
|
}
|
|
|
|
}
|
|
}
|
|
} |