headless_lms_server/controllers/main_frontend/courses/
chatbots.rs

1//! Controllers for requests starting with `/api/v0/main-frontend/{course_id}/chatbots`.
2use crate::prelude::*;
3
4use headless_lms_models::chatbot_configurations::NewChatbotConf;
5use models::chatbot_configurations::ChatbotConfiguration;
6
7/// GET `/api/v0/main-frontend/courses/{course_id}/chatbots`
8#[instrument(skip(pool))]
9async fn get_chatbots(
10    course_id: web::Path<Uuid>,
11    pool: web::Data<PgPool>,
12    user: AuthUser,
13) -> ControllerResult<web::Json<Vec<ChatbotConfiguration>>> {
14    let mut conn = pool.acquire().await?;
15    let token = authorize(&mut conn, Act::Edit, Some(user.id), Res::Course(*course_id)).await?;
16
17    let configurations =
18        models::chatbot_configurations::get_for_course(&mut conn, *course_id).await?;
19    token.authorized_ok(web::Json(configurations))
20}
21
22/// POST `/api/v0/main-frontend/courses/{course_id}/chatbots`
23#[instrument(skip(pool, payload))]
24async fn create_chatbot(
25    course_id: web::Path<Uuid>,
26    payload: web::Json<String>,
27    pool: web::Data<PgPool>,
28    user: AuthUser,
29) -> ControllerResult<web::Json<ChatbotConfiguration>> {
30    let mut conn = pool.acquire().await?;
31    let token = authorize(&mut conn, Act::Edit, Some(user.id), Res::Course(*course_id)).await?;
32    let mut tx = conn.begin().await?;
33
34    let course = models::courses::get_course(&mut tx, *course_id).await?;
35
36    if !course.can_add_chatbot {
37        return Err(ControllerError::new(
38            ControllerErrorType::BadRequest,
39            "Course doesn't allow creating chatbots.".to_string(),
40            None,
41        ));
42    }
43
44    let configuration = models::chatbot_configurations::insert(
45        &mut tx,
46        NewChatbotConf {
47            chatbot_name: payload.into_inner(),
48            course_id: *course_id,
49            ..Default::default()
50        },
51    )
52    .await?;
53    tx.commit().await?;
54
55    token.authorized_ok(web::Json(configuration))
56}
57
58/// POST `/api/v0/main-frontend/courses/{course_id}/chatbots/{chatbot_configuration_id}/set-as-default`
59#[instrument(skip(pool))]
60async fn set_default_chatbot(
61    ids: web::Path<(Uuid, Uuid)>,
62    pool: web::Data<PgPool>,
63    user: AuthUser,
64) -> ControllerResult<web::Json<ChatbotConfiguration>> {
65    let mut conn = pool.acquire().await?;
66    let (course_id, chatbot_configuration_id) = *ids;
67
68    let token = authorize(&mut conn, Act::Edit, Some(user.id), Res::Course(course_id)).await?;
69    let mut tx = conn.begin().await?;
70
71    models::chatbot_configurations::remove_default_chatbot_from_course(&mut tx, course_id).await?;
72
73    let chatbot =
74        models::chatbot_configurations::get_by_id(&mut tx, chatbot_configuration_id).await?;
75
76    if course_id != chatbot.course_id {
77        return Err(ControllerError::new(
78            ControllerErrorType::BadRequest,
79            "Chatbot course id doesn't match the course id provided.".to_string(),
80            None,
81        ));
82    }
83
84    let configuration = models::chatbot_configurations::set_default_chatbot_for_course(
85        &mut tx,
86        chatbot_configuration_id,
87    )
88    .await?;
89    tx.commit().await?;
90
91    token.authorized_ok(web::Json(configuration))
92}
93
94pub fn _add_routes(cfg: &mut web::ServiceConfig) {
95    cfg.route("", web::get().to(get_chatbots))
96        .route("", web::post().to(create_chatbot))
97        .route(
98            "/{chatbot_configuration_id}/set-as-default",
99            web::post().to(set_default_chatbot),
100        );
101}
headless_lms_server/controllers/main_frontend/courses/chatbots.rs

headless_lms_server/controllers/main_frontend/courses/
chatbots.rs
