headless_lms_models/
chatbot_page_sync_statuses.rs

use std::collections::HashMap;

use crate::prelude::*;

#[derive(Debug, Serialize, Deserialize, PartialEq, Clone)]
#[cfg_attr(feature = "ts_rs", derive(TS))]
pub struct ChatbotPageSyncStatus {
    pub id: Uuid,
    pub created_at: DateTime<Utc>,
    pub updated_at: DateTime<Utc>,
    pub deleted_at: Option<DateTime<Utc>>,
    pub course_id: Uuid,
    pub page_id: Uuid,
    pub error_message: Option<String>,
    pub synced_page_revision_id: Option<Uuid>,
}

pub async fn ensure_sync_statuses_exist(
    conn: &mut PgConnection,
    course_ids: &[Uuid],
) -> ModelResult<HashMap<Uuid, Vec<ChatbotPageSyncStatus>>> {
    sqlx::query!(
        r#"
INSERT INTO chatbot_page_sync_statuses (course_id, page_id)
SELECT course_id,
  id
FROM pages
WHERE course_id = ANY($1)
  AND deleted_at IS NULL
  AND hidden IS FALSE ON CONFLICT (page_id, deleted_at) DO NOTHING
        "#,
        course_ids
    )
    .execute(&mut *conn)
    .await?;

    let all_statuses = sqlx::query_as!(
        ChatbotPageSyncStatus,
        r#"
SELECT *
FROM chatbot_page_sync_statuses
WHERE course_id = ANY($1)
        "#,
        course_ids
    )
    .fetch_all(&mut *conn)
    .await?
    .into_iter()
    .fold(
        HashMap::<Uuid, Vec<ChatbotPageSyncStatus>>::new(),
        |mut map, status| {
            map.entry(status.course_id).or_default().push(status);
            map
        },
    );

    Ok(all_statuses)
}

// Given a mapping from page id to the new revision id, update the sync statuses
pub async fn update_page_revision_ids(
    conn: &mut PgConnection,
    page_id_to_new_revision_id: HashMap<Uuid, Uuid>,
) -> ModelResult<()> {
    // If there are no updates to perform, return early
    if page_id_to_new_revision_id.is_empty() {
        return Ok(());
    }
    let (page_ids, revision_ids): (Vec<Uuid>, Vec<Uuid>) =
        page_id_to_new_revision_id.into_iter().unzip();

    sqlx::query!(
        r#"
UPDATE chatbot_page_sync_statuses AS cps
SET synced_page_revision_id = data.synced_page_revision_id
FROM (
    SELECT unnest($1::uuid []) AS page_id,
      unnest($2::uuid []) AS synced_page_revision_id
  ) AS data
WHERE cps.page_id = data.page_id
AND cps.deleted_at IS NULL
    "#,
        &page_ids,
        &revision_ids
    )
    .execute(conn)
    .await?;

    Ok(())
}
headless_lms_models/chatbot_page_sync_statuses.rs

headless_lms_models/
chatbot_page_sync_statuses.rs
