import os
import asyncio
import logging
from dotenv import load_dotenv
from nio import AsyncClient, AsyncClientConfig, MatrixRoom, RoomMessageText
from nio.responses import LoginResponse

# --- Load environment variables ---
load_dotenv()
HOMESERVER_URL = os.getenv("HOMESERVER_URL")
USER_ID = os.getenv("USER_ID")
PASSWORD = os.getenv("PASSWORD")
LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO").upper()

# --- Logging Setup ---
# Convert string level to numeric
numeric_level = getattr(logging, LOG_LEVEL, logging.INFO)
logging.basicConfig(
    level=numeric_level,
    format="%(asctime)s %(levelname)s %(name)s: %(message)s"
)
logger = logging.getLogger(__name__)

async def message_callback(room: MatrixRoom, event: RoomMessageText):
    """
    Called when a new text message is received.
    """
    logger.debug("Message callback triggered")
    if event.sender == USER_ID:
        return
    
    body = event.body.strip().lower()
    logger.info("Message from %s in %s: %s", event.sender, room.display_name, event.body)

    if event.body.strip().lower() == "!ping":
        await client.room_send(
            room_id=room.room_id,
            message_type="m.room.message",
            content={"msgtype": "m.text", "body": "Pong!"}
        )
        logger.info("Replied with Pong! to %s", event.sender)
    elif body== "hello botbot":
        await client.room_send(
            room_id=room.room_id,
            message_type="m.room.message",
            content={
                "msgtype": "m.text", 
                "body": "Hello! How can I assist you today?",                
            },
            ignore_unverified_devices=True
        )
        logger.info("Replied with greeting to %s", event.sender)

async def main():
    global client
    logger.debug(HOMESERVER_URL)
    logger.debug(USER_ID)
    # Configure client with persistent store
    config = AsyncClientConfig(
        store_sync_tokens=True,
        encryption_enabled=True,
    )
    client = AsyncClient(
        HOMESERVER_URL,
        USER_ID,
        store_path="/app/data",
        config=config
    )

    # Login
    resp = await client.login(password=PASSWORD)
    logger.debug("Login response raw: %r", resp.__dict__)
    if not isinstance(resp, LoginResponse) or not resp.access_token:
        error_msg = getattr(resp, 'message', repr(resp))
        logger.error("Login failed: %s", error_msg)
        await client.close()
        return
    logger.info("Logged in successfully as %s", USER_ID)

    # Register callback and start syncing
    client.add_event_callback(message_callback, RoomMessageText)
    logger.info("Starting sync loop with timeout=30000ms")
    await client.sync_forever(timeout=30000)

if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        logger.info("Received interrupt, shutting down client")
        asyncio.run(client.close())