from config.pgDb import pg_pool


def insert_session(
    user_id: str, ai_id: str, session_id: str, session_title: str, available_module
):
    with pg_pool.getConn() as coon:
        with coon.cursor() as cur:
            cur.execute(
                "INSERT INTO ai_chat_sessions (id ,user_id, ai_id, title, available_module, created_at, updated_at) VALUES (%s, %s, %s, %s,%s, NOW(), NOW())",
                (session_id, user_id, ai_id, session_title, available_module),
            )
            coon.commit()


def update_session_updated_at(session_id: str):
    with pg_pool.getConn() as conn:
        with conn.cursor() as cur:
            cur.execute(
                "UPDATE ai_chat_sessions SET updated_at = NOW() WHERE id = %s",
                (session_id,),
            )
            conn.commit()


def get_sessions(user_id: str, available_module: str):
    with pg_pool.getConn() as conn:
        with conn.cursor() as cur:
            cur.execute(
                "SELECT id, title, updated_at "
                "FROM ai_chat_sessions "
                "WHERE user_id = %s AND available_module = %s "
                "ORDER BY updated_at DESC",
                (user_id, available_module),
            )

            sessions = cur.fetchall()
            return [
                {"id": row[0], "title": row[1], "updated_at": row[2]}
                for row in sessions
            ]