29 lines
No EOL
1.1 KiB
Python
29 lines
No EOL
1.1 KiB
Python
"""Attachment queue handlers for MAX surface."""
|
|
from adapter.max.store import ChatStore
|
|
|
|
|
|
class AttachmentHandler:
|
|
def __init__(self, store: ChatStore):
|
|
self.store = store
|
|
|
|
def handle_list(self, max_chat_id: str) -> str:
|
|
attachments = self.store.get_attachments(max_chat_id)
|
|
if not attachments:
|
|
return "Attachment queue is empty."
|
|
lines = [f" {i+1}. {name}" for i, (_, name) in enumerate(attachments)]
|
|
return "\n".join(lines)
|
|
|
|
def handle_remove(self, max_chat_id: str, index: str) -> str:
|
|
attachments = self.store.staged_attachments.get(max_chat_id, [])
|
|
if index.lower() == "all":
|
|
self.store.staged_attachments[max_chat_id] = []
|
|
return "All attachments removed from queue."
|
|
|
|
try:
|
|
idx = int(index) - 1
|
|
if 0 <= idx < len(attachments):
|
|
removed = attachments.pop(idx)
|
|
return f"Removed: {removed[1]}"
|
|
return "Invalid index."
|
|
except ValueError:
|
|
return "Usage: !remove <number> or !remove all" |