import os
import threading
import urllib.request
import json
import boto3


class TelegramAdapter:
    """Channel adapter for Telegram Bot API."""

    def __init__(self, chat_id: str, bot_token_secret_arn: str = ''):
        self.chat_id = str(chat_id)
        self._secret_arn = bot_token_secret_arn
        self._token: str | None = None
        self._lock = threading.Lock()

    def _get_token(self) -> str:
        if self._token is None:
            with self._lock:
                if self._token is None:
                    secret_arn = self._secret_arn or os.environ.get(
                        'TELEGRAM_BOT_TOKEN_SECRET_ARN',
                        'arn:aws:secretsmanager:us-east-1:495395224548:secret:agent-claw/telegram-bot-token-Oq3in3'
                    )
                    sm = boto3.client('secretsmanager')
                    self._token = sm.get_secret_value(
                        SecretId=secret_arn
                    )['SecretString']
        return self._token

    def _api(self, method: str, data: dict) -> dict:
        token = self._get_token()
        body = json.dumps(data).encode()
        req = urllib.request.Request(
            f'https://api.telegram.org/bot{token}/{method}',
            data=body,
            headers={'Content-Type': 'application/json'},
        )
        with urllib.request.urlopen(req, timeout=30) as resp:
            return json.loads(resp.read())

    def send(self, text: str) -> str:
        """Send message, return message_id."""
        resp = self._api('sendMessage', {
            'chat_id': self.chat_id,
            'text': text,
            'parse_mode': 'Markdown',
        })
        return str(resp.get('result', {}).get('message_id', ''))

    def send_typing(self) -> None:
        """Send typing action (best-effort)."""
        try:
            self._api('sendChatAction', {
                'chat_id': self.chat_id,
                'action': 'typing',
            })
        except Exception:
            pass

    def edit(self, message_id: str, text: str) -> None:
        """Edit an existing message in-place."""
        try:
            self._api('editMessageText', {
                'chat_id': self.chat_id,
                'message_id': int(message_id),
                'text': text,
                'parse_mode': 'Markdown',
            })
        except Exception:
            pass