import asyncio, sys
sys.path.insert(0, ".")
from gateway.platforms.base import SendResult
from gateway.stream_consumer import GatewayStreamConsumer, StreamConsumerConfig

class FakeTelegram:
    """Simulates Telegram with persistent flood control once triggered."""
    MAX_MESSAGE_LENGTH = 4096
    def __init__(self):
        self.visible_messages = {}
        self.message_order = []
        self.edit_log = []
        self._next_id = 1
        self._flood_active = False
    async def send(self, *, chat_id, content, reply_to=None, metadata=None):
        mid = f"msg_{self._next_id}"; self._next_id += 1
        self.visible_messages[mid] = content
        self.message_order.append(mid)
        return SendResult(success=True, message_id=mid)
    async def edit_message(self, *, chat_id, message_id, content):
        self.edit_log.append(f"EDIT {message_id}: {content!r}")
        if self._flood_active:
            self.edit_log[-1] += "  <-- FAILED (flood still active)"
            return SendResult(success=False, error="flood_control:6.0")
        if "more" in content and content.endswith(" ▉"):
            self._flood_active = True
            self.edit_log[-1] += "  <-- FAILED (flood control triggered)"
            return SendResult(success=False, error="flood_control:6.0")
        self.visible_messages[message_id] = content
        return SendResult(success=True, message_id=message_id)
    def truncate_message(self, text, limit):
        return [text] if len(text) <= limit else [text[i:i+limit] for i in range(0, len(text), limit)]

async def main():
    tg = FakeTelegram()
    consumer = GatewayStreamConsumer(tg, "chat_123",
        StreamConsumerConfig(edit_interval=0.01, buffer_threshold=5, cursor=" ▉"))
    consumer.on_delta("Hello")
    task = asyncio.create_task(consumer.run())
    await asyncio.sleep(0.08)
    consumer.on_delta(" world"); await asyncio.sleep(0.08)
    consumer.on_delta(" more"); await asyncio.sleep(0.08)  # this edit fails
    consumer.on_delta(None)  # tool boundary
    consumer.on_delta("Here is the tool result.")
    consumer.finish()
    await task
    visible = " ".join(tg.visible_messages[m] for m in tg.message_order)
    for e in tg.edit_log: print(e)
    for mid in tg.message_order: print(f"{mid}: {tg.visible_messages[mid]!r}")
    print(f"Agent generated: \"Hello world more\" + [tool] + \"Here is the tool result.\"")
    print(f"User received:   {visible!r}")
    print("REPRODUCED" if "more" not in visible else "NOT REPRODUCED")

asyncio.run(main())