from config.pgDb import pg_pool


def get_available_knowledge_bases(available_module: str):
    with pg_pool.getConn() as conn:
        with conn.cursor() as cursor:
            cursor.execute(
                """
                SELECT id, name, description, is_active
                FROM ai_knowledge
                WHERE available_module::jsonb @> %s::jsonb
                """,
                (f'["{available_module}"]',),
            )

            return [
                {
                    "id": row[0],
                    "name": row[1],
                    "description": row[2],
                    "is_active": row[3],
                }
                for row in cursor.fetchall()
            ]


def add_knowledge_base(name, description, user_id):
    with pg_pool.getConn() as conn:
        with conn.cursor() as cursor:
            cursor.execute(
                """
                INSERT INTO ai_knowledge (name, description, created_by, created_at)
                VALUES (%s, %s, %s, now())
                RETURNING id
                """,
                (name, description, user_id),
            )
            new_id = cursor.fetchone()[0]  # 取返回的 id
            return new_id