jigaido/apps/telegram-bot/commands.py

"""Telegram command handlers for JIGAIDO - Thin wrappers around core services."""

import time
from functools import wraps
from typing import Optional

import dateparser
from telegram import Update
from telegram.ext import ContextTypes

from adapters.storage.json_file import JsonFileRoomStorage, JsonFileTrackingStorage
from core.services import BountyService, TrackingService

ROOM_STORAGE = JsonFileRoomStorage()
TRACKING_STORAGE = JsonFileTrackingStorage()
BOUNTY_SERVICE = BountyService(ROOM_STORAGE)
TRACKING_SERVICE = TrackingService(TRACKING_STORAGE, ROOM_STORAGE)

TELEGRAM_BOT_USERNAME = "your_bot_username"


def extract_args(text: str) -> list[str]:
    if not text:
        return []
    tokens = text.strip().split()
    return tokens[1:] if len(tokens) > 1 else []


def parse_args(args: list[str]) -> tuple[Optional[str], Optional[str], Optional[int]]:
    text = None
    link = None
    due_date_ts = None

    remaining = []
    i = 0
    while i < len(args):
        arg = args[i]
        if not link and (arg.startswith("http://") or arg.startswith("https://")):
            link = arg
        elif due_date_ts is None:
            parsed = dateparser.parse(arg)
            if parsed:
                due_date_ts = int(parsed.timestamp())
                if i + 1 < len(args) and _is_time_format(args[i + 1]):
                    time_str = args[i + 1]
                    hour, minute = map(int, time_str.split(":"))
                    due_date_ts = _set_time_on_timestamp(due_date_ts, hour, minute)
                    i += 1
            else:
                remaining.append(arg)
        else:
            remaining.append(arg)
        i += 1

    text = " ".join(remaining) if remaining else None
    return text, link, due_date_ts


def _is_time_format(s: str) -> bool:
    """Check if string matches HH:MM format."""
    if not s or len(s) != 5:
        return False
    if s[2] != ":":
        return False
    try:
        h, m = map(int, s.split(":"))
        return 0 <= h <= 23 and 0 <= m <= 59
    except ValueError:
        return False


def _set_time_on_timestamp(ts: int, hour: int, minute: int) -> int:
    """Set time (hour:minute) on a Unix timestamp, keeping the date."""
    import datetime

    dt = datetime.datetime.fromtimestamp(ts)
    dt = dt.replace(hour=hour, minute=minute, second=0, microsecond=0)
    return int(dt.timestamp())


def format_bounty(b, show_id: bool = True) -> str:
    parts = []
    if show_id:
        parts.append(f"[#{b.id}]")
    if b.text:
        parts.append(b.text)
    if b.link:
        parts.append(f"🔗 {b.link}")
    if b.due_date_ts:
        due_str = time.strftime("%Y-%m-%d", time.localtime(b.due_date_ts))
        days_left = (b.due_date_ts - int(time.time())) // 86400
        if days_left < 0:
            parts.append(f"⏰ {due_str} (OVERDUE)")
        elif days_left == 0:
            parts.append(f"⏰ {due_str} (TODAY)")
        else:
            parts.append(f"⏰ {due_str} ({days_left}d)")
    if b.created_by_user_id:
        parts.append(f"by {b.created_by_user_id}")
    return " | ".join(parts)


def is_group(update: Update) -> bool:
    return update.effective_chat.type != "private"


def get_group_id(update: Update) -> int:
    return update.effective_chat.id


def get_user_id(update: Update) -> int:
    return update.effective_user.id


def get_room_id(update: Update) -> int:
    """Get room_id for the current context.

    For groups: negative group_id
    For DMs: positive user_id
    """
    if is_group(update):
        return get_group_id(update)
    return get_user_id(update)


async def cmd_bounty(update: Update, ctx: ContextTypes.DEFAULT_TYPE) -> None:
    room_id = get_room_id(update)
    bounties = BOUNTY_SERVICE.list_bounties(room_id)

    if not bounties:
        await update.message.reply_text("No bounties yet.")
        return

    lines = [format_bounty(b, show_id=True) for b in bounties]
    await update.message.reply_text("\n".join(lines), disable_web_page_preview=True)


async def cmd_my(update: Update, ctx: ContextTypes.DEFAULT_TYPE) -> None:
    user_id = get_user_id(update)

    if is_group(update):
        group_id = get_group_id(update)
        bounties = TRACKING_SERVICE.get_tracked_bounties(group_id, user_id)
    else:
        room_id = get_room_id(update)
        bounties = BOUNTY_SERVICE.list_bounties(room_id)

    if not bounties:
        msg = (
            "You are not tracking any bounties."
            if is_group(update)
            else "No personal bounties."
        )
        await update.message.reply_text(msg)
        return

    lines = [format_bounty(b, show_id=True) for b in bounties]
    await update.message.reply_text("\n".join(lines), disable_web_page_preview=True)


async def cmd_add(update: Update, ctx: ContextTypes.DEFAULT_TYPE) -> None:
    args = extract_args(update.message.text)
    if not args:
        await update.message.reply_text(
            "Usage: /add <text> [link] [date] [time]\n"
            "Example: /add Fix bug https://github.com/foo/bar april 15 14:30"
        )
        return

    text, link, due_date_ts = parse_args(args)
    if not text and not link:
        await update.message.reply_text("A bounty needs at least text or a link.")
        return

    user_id = get_user_id(update)
    room_id = get_room_id(update)

    try:
        bounty = BOUNTY_SERVICE.add_bounty(
            room_id=room_id,
            user_id=user_id,
            text=text,
            link=link,
            due_date_ts=due_date_ts,
        )
    except PermissionError as e:
        await update.message.reply_text(f"⛔ {e}")
        return
    except ValueError as e:
        await update.message.reply_text(f"⛔ {e}")
        return

    due_str = ""
    if due_date_ts:
        due_str = f" | Due: {time.strftime('%Y-%m-%d', time.localtime(due_date_ts))}"

    await update.message.reply_text(
        f"✅ Bounty added (#{bounty.id}){due_str}",
        disable_web_page_preview=True,
    )


async def cmd_update(update: Update, ctx: ContextTypes.DEFAULT_TYPE) -> None:
    args = extract_args(update.message.text)
    if len(args) < 1:
        await update.message.reply_text(
            "Usage: /update <bounty_id> [text] [link] [due_date]"
        )
        return

    try:
        bounty_id = int(args[0])
    except ValueError:
        await update.message.reply_text("Invalid bounty ID.")
        return

    text, link, due_date_ts = parse_args(args[1:])
    if not text and not link and due_date_ts is None:
        await update.message.reply_text("Nothing to update.")
        return

    user_id = get_user_id(update)
    room_id = get_room_id(update)

    try:
        success = BOUNTY_SERVICE.update_bounty(
            room_id=room_id,
            bounty_id=bounty_id,
            user_id=user_id,
            text=text,
            link=link,
            due_date_ts=due_date_ts,
        )
    except PermissionError as e:
        await update.message.reply_text(f"⛔ {e}")
        return

    if success:
        await update.message.reply_text(f"✅ Bounty #{bounty_id} updated.")
    else:
        await update.message.reply_text("Bounty not found.")


cmd_edit = cmd_update


async def cmd_delete(update: Update, ctx: ContextTypes.DEFAULT_TYPE) -> None:
    args = extract_args(update.message.text)
    if not args:
        await update.message.reply_text("Usage: /delete <bounty_id>")
        return

    try:
        bounty_id = int(args[0])
    except ValueError:
        await update.message.reply_text("Invalid bounty ID.")
        return

    user_id = get_user_id(update)
    room_id = get_room_id(update)

    try:
        success = BOUNTY_SERVICE.delete_bounty(
            room_id=room_id,
            bounty_id=bounty_id,
            user_id=user_id,
        )
    except PermissionError as e:
        await update.message.reply_text(f"⛔ {e}")
        return

    if success:
        await update.message.reply_text(f"✅ Bounty #{bounty_id} deleted.")
    else:
        await update.message.reply_text("Bounty not found.")


async def cmd_track(update: Update, ctx: ContextTypes.DEFAULT_TYPE) -> None:
    if not is_group(update):
        await update.message.reply_text("⛔ /track is only available in groups.")
        return

    args = extract_args(update.message.text)
    if not args:
        await update.message.reply_text("Usage: /track <bounty_id>")
        return

    try:
        bounty_id = int(args[0])
    except ValueError:
        await update.message.reply_text("Invalid bounty ID.")
        return

    user_id = get_user_id(update)
    room_id = get_room_id(update)

    try:
        if TRACKING_SERVICE.track_bounty(room_id, user_id, bounty_id):
            await update.message.reply_text(f"✅ Now tracking bounty #{bounty_id}.")
        else:
            await update.message.reply_text(f"Already tracking bounty #{bounty_id}.")
    except ValueError as e:
        await update.message.reply_text(str(e))


async def cmd_untrack(update: Update, ctx: ContextTypes.DEFAULT_TYPE) -> None:
    if not is_group(update):
        await update.message.reply_text("⛔ /untrack is only available in groups.")
        return

    args = extract_args(update.message.text)
    if not args:
        await update.message.reply_text("Usage: /untrack <bounty_id>")
        return

    try:
        bounty_id = int(args[0])
    except ValueError:
        await update.message.reply_text("Invalid bounty ID.")
        return

    user_id = get_user_id(update)
    room_id = get_room_id(update)

    if TRACKING_SERVICE.untrack_bounty(room_id, user_id, bounty_id):
        await update.message.reply_text(f"✅ Untracked bounty #{bounty_id}.")
    else:
        await update.message.reply_text("Not tracking bounty #{bounty_id}.")


async def cmd_start(update: Update, ctx: ContextTypes.DEFAULT_TYPE) -> None:
    if is_group(update):
        await update.message.reply_text(
            "👻 JIGAIDO is watching.\n\n"
            "This group's bounty list is now active.\n"
            "/bounty — list bounties\n"
            "/add — create a bounty\n"
            "/track — track a bounty\n"
            "/my — your tracked bounties"
        )
    else:
        await update.message.reply_text(
            "👻 JIGAIDO activated.\n\n"
            "Personal bounty list ready.\n"
            "/bounty — list your bounties\n"
            "/add — create a bounty\n"
            "/my — your tracked bounties"
        )


async def cmd_help(update: Update, ctx: ContextTypes.DEFAULT_TYPE) -> None:
    await update.message.reply_text(
        "👻 JIGAIDO Commands:\n\n"
        "/bounty — list all bounties\n"
        "/my — bounties you're tracking\n"
        "/add <text> [link] [date] [time] — add bounty (admin only)\n"
        "/update <id> [text] [link] [due] — update bounty (admin only)\n"
        "/delete <id> — delete bounty (admin only)\n"
        "/track <id> — track a bounty (groups only)\n"
        "/untrack <id> — stop tracking (groups only)\n"
        "/start — re-initialize\n"
        "/help — this message",
        disable_web_page_preview=True,
    )