diff --git a/apps/telegram-bot/bot.py b/apps/telegram-bot/bot.py index c72df9c..a34453b 100644 --- a/apps/telegram-bot/bot.py +++ b/apps/telegram-bot/bot.py @@ -14,6 +14,7 @@ from commands import ( cmd_help, cmd_my, cmd_start, + cmd_timezone, cmd_track, cmd_untrack, cmd_update, @@ -41,6 +42,7 @@ def build_app() -> Application: app.add_handler(CommandHandler("delete", cmd_delete)) app.add_handler(CommandHandler("track", cmd_track)) app.add_handler(CommandHandler("untrack", cmd_untrack)) + app.add_handler(CommandHandler("timezone", cmd_timezone)) app.add_handler(MessageHandler(filters.COMMAND, cmd_help)) @@ -56,6 +58,7 @@ async def post_init(app: Application) -> None: ("edit", "Edit a bounty"), ("track", "Track a bounty"), ("untrack", "Stop tracking"), + ("timezone", "Get/set room timezone"), ("help", "Show help"), ] ) diff --git a/apps/telegram-bot/commands.py b/apps/telegram-bot/commands.py index b26f59a..2f608ad 100644 --- a/apps/telegram-bot/commands.py +++ b/apps/telegram-bot/commands.py @@ -3,6 +3,7 @@ import time from functools import wraps from typing import Optional +from zoneinfo import ZoneInfo, ZoneInfoNotFoundError import dateparser from telegram import Update @@ -323,7 +324,37 @@ async def cmd_help(update: Update, ctx: ContextTypes.DEFAULT_TYPE) -> None: "/delete — delete bounty\n" "/track — track a bounty (groups only)\n" "/untrack — stop tracking (groups only)\n" + "/timezone [tz] — get/set room timezone (admin only)\n" "/start — re-initialize\n" "/help — this message", disable_web_page_preview=True, ) + + +async def cmd_timezone(update: Update, ctx: ContextTypes.DEFAULT_TYPE) -> None: + args = extract_args(update.message.text) + room_id = get_room_id(update) + user_id = get_user_id(update) + + if not args: + current_tz = BOUNTY_SERVICE.get_timezone(room_id) + await update.message.reply_text(f"Current timezone: {current_tz}") + return + + timezone_str = args[0] + + try: + ZoneInfo(timezone_str) + except (KeyError, ZoneInfoNotFoundError): + await update.message.reply_text( + "⛔ Invalid timezone. Use IANA format (e.g., Asia/Jakarta)" + ) + return + + try: + BOUNTY_SERVICE.set_timezone(room_id, timezone_str, user_id) + except PermissionError as e: + await update.message.reply_text(f"⛔ {e}") + return + + await update.message.reply_text(f"✅ Timezone set to {timezone_str}.")