import asyncio
import logging
import zmq
import json
import time
import traceback
from telegram import Update
from telegram.ext import ApplicationBuilder, ContextTypes, CommandHandler

SUBCRIBERS = set()

async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
    await context.bot.send_message(chat_id=update.effective_chat.id, text="I'm a bot, please talk to me!")

async def subscribe(update: Update, context: ContextTypes.DEFAULT_TYPE):
    key = update.message.text
    if key == "/subscribe lam_bwatchers_bot":
        SUBCRIBERS.add(update.effective_chat.id)
        await context.bot.send_message(chat_id=update.effective_chat.id, text="ok, you're in")

def try_connect_zmq():
    try:
        context = zmq.asyncio.Context()
        socket = context.socket(zmq.SUB)
        socket.setsockopt(zmq.SUBSCRIBE, b"")
        socket.connect("tcp://127.0.0.1:2222")
        return socket
    except Exception:
        traceback.print_exc()
    return None

async def start_zmq(zmq_pipe, application):
    while True:
        message = await zmq_pipe.recv_json()
        for user in SUBCRIBERS:
            print("sending {message} to {user}")
            await application.bot.send_message(chat_id=user, text=json.dumps(message, indent=2))

async def run_app(api_key):
    
    application = ApplicationBuilder().token(api_key).build()
    start_handler = CommandHandler('start', start)
    subscribe_handler = CommandHandler('subscribe', subscribe)
    application.add_handler(start_handler)
    application.add_handler(subscribe_handler)
    
    ## sometimes this got async_cancel which make application not work
    ## we put it in try catch to prevent it
    zmq_pipe = try_connect_zmq()
    
    async with application:  # Calls `initialize` and `shutdown`
        await application.start()
        await application.updater.start_polling()
        await start_zmq(zmq_pipe, application)
        # Start other asyncio frameworks here
        # Add some logic that keeps the event loop running until you want to shutdown
        # Stop the other asyncio frameworks here
        await application.updater.stop()
        await application.stop()

async def start_app(options):
    await run_app(options.conf.telegram_api_key)

2024-03-25 21:27:14,965 [INFO] HTTP Request: POST https://api.telegram.org/bot6461521072:AAHYRcFVmxkdQ9W5DHmmkGO_PJXF85gW-tU/getUpdates "HTTP/1.1 200 OK"
2024-03-25 21:27:18,548 [WARNING] Fetching updates got a asyncio.CancelledError. Ignoring as this task may onlybe closed via `Application.stop`.

    async def __process_update_wrapper(self, update: object) -> None:
        await self._update_processor.process_update(update, self.process_update(update))
        self.update_queue.task_done()