import time

import pika
import requests
from loguru import logger


def _send_message_to_user(bot_token, target_id, message):
    base_url = f"https://api.telegram.org/bot{bot_token}/sendMessage"
    params = {
        "chat_id": target_id,
        "text": message,
        "parse_mode": "MarkdownV2"
    }
    max_retry = 3
    retry_count = 0
    while retry_count < max_retry:
        response = requests.post(base_url, params=params)
        if response.status_code == 200:
            logger.debug(f'消息发送成功:{message}')
            return  # 如果发送成功，立即返回
        else:
            logger.debug('消息发送失败，重试中...')
            logger.error(response.text)
            time.sleep(10)
            retry_count += 1

    logger.debug('消息发送失败')


# 你的消息发送逻辑

class MessageServer:
    def __init__(self):
        self.credentials = pika.PlainCredentials('bot', 'xiaom@i123')
        self.connection = pika.BlockingConnection(
            pika.ConnectionParameters('45.89.233.249', 5672, '/', credentials=self.credentials))
        self.channel = self.connection.channel()
        self.channel.queue_declare(queue='message_queue')

    def start(self):
        try:
            def callback(ch, method, properties, body):
                logger.info(body.decode())
                bot_token, target_id, message = body.decode().split('|')
                _send_message_to_user(bot_token, target_id, message)

            self.channel.basic_consume(queue='message_queue', on_message_callback=callback, auto_ack=True)
            self.channel.start_consuming()
        except Exception as e:
            logger.error(e)

    def stop(self):
        self.connection.close()


server = MessageServer()
server.start()