mirror of
https://github.com/pacnpal/Pac-cogs.git
synced 2025-12-20 02:41:06 -05:00
f8d383a55a
⚙️ Processing: Shows items currently being processed ✅ Completed: Shows successfully processed items ❌ Failed: Shows failed items
71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
"""Message management utilities"""
|
|
|
|
import asyncio
|
|
import logging
|
|
from typing import Dict, List
|
|
|
|
logger = logging.getLogger("VideoArchiver")
|
|
|
|
class MessageManager:
|
|
def __init__(self, message_duration: int, message_template: str):
|
|
self.message_duration = message_duration
|
|
self.message_template = message_template
|
|
self.scheduled_deletions: Dict[int, asyncio.Task] = {}
|
|
self._lock = asyncio.Lock()
|
|
|
|
def format_archive_message(
|
|
self, author: str, url: str, original_message: str
|
|
) -> str:
|
|
return self.message_template.format(
|
|
author=author, url=url, original_message=original_message
|
|
)
|
|
|
|
async def get_message_reactions(self, message) -> List[Dict]:
|
|
"""Get all reactions from a message"""
|
|
reactions = []
|
|
for reaction in message.reactions:
|
|
reactions.append({
|
|
'emoji': str(reaction.emoji),
|
|
'count': reaction.count,
|
|
'users': [user.id async for user in reaction.users()]
|
|
})
|
|
return reactions
|
|
|
|
async def restore_reactions(self, message, reactions: List[Dict]) -> None:
|
|
"""Restore reactions to a message"""
|
|
for reaction_data in reactions:
|
|
try:
|
|
await message.add_reaction(reaction_data['emoji'])
|
|
except Exception as e:
|
|
logger.error(f"Failed to restore reaction {reaction_data['emoji']}: {str(e)}")
|
|
|
|
async def schedule_message_deletion(self, message_id: int, delete_func) -> None:
|
|
if self.message_duration <= 0:
|
|
return
|
|
|
|
async with self._lock:
|
|
if message_id in self.scheduled_deletions:
|
|
self.scheduled_deletions[message_id].cancel()
|
|
|
|
async def delete_later():
|
|
try:
|
|
await asyncio.sleep(self.message_duration * 3600)
|
|
await delete_func()
|
|
except asyncio.CancelledError:
|
|
pass
|
|
except Exception as e:
|
|
logger.error(f"Failed to delete message {message_id}: {str(e)}")
|
|
finally:
|
|
async with self._lock:
|
|
self.scheduled_deletions.pop(message_id, None)
|
|
|
|
self.scheduled_deletions[message_id] = asyncio.create_task(delete_later())
|
|
|
|
async def cancel_all_deletions(self):
|
|
"""Cancel all scheduled message deletions"""
|
|
async with self._lock:
|
|
for task in self.scheduled_deletions.values():
|
|
task.cancel()
|
|
await asyncio.gather(*self.scheduled_deletions.values(), return_exceptions=True)
|
|
self.scheduled_deletions.clear()
|