import asyncio
import sys
from nio import AsyncClient, LoginResponse, RoomSendResponse
from room import Config

async def main():
    if len(sys.argv) < 2 or len(sys.argv) > 4:
        print("Usage: python3 simulate_suggestions.py simulation_suggestions.txt [delay_seconds] [config_path]")
        return

    file_path = sys.argv[1]
    delay = float(sys.argv[2]) if len(sys.argv) >= 3 else 5.0
    config_path = sys.argv[3] if len(sys.argv) == 4 else "configs/config.yaml"

    with open(file_path, "r") as f:
        suggestions = [line.strip() for line in f if line.strip()]

    if not suggestions:
        print("No suggestions found in file.")
        return

    config = Config(config_path)
    client = AsyncClient(config.homeserver, config.username)
    resp = await client.login(config.password)
    if not isinstance(resp, LoginResponse):
        print(f"Failed to log in: {resp}")
        return

    print(f"Posting {len(suggestions)} suggestions to {config.room_id} "
          f"(every {delay:g}s to avoid rate limits)...")
    for i, suggestion in enumerate(suggestions, start=1):
        content = {
            "msgtype": "m.text",
            "body": f"!suggest {suggestion}",
        }
        resp = await client.room_send(
            config.room_id,
            message_type="m.room.message",
            content=content,
        )
        if isinstance(resp, RoomSendResponse):
            print(f"Posted {i}/{len(suggestions)}: {suggestion}")
        else:
            print(f"Failed to post suggestion {i}: {resp}")
        await asyncio.sleep(delay)

    await client.close()

if __name__ == "__main__":
    asyncio.run(main())