botserver/src/basic/keywords/set_schedule.rs

use diesel::prelude::*;
use log::info;
use serde_json::{json, Value};
use uuid::Uuid;

use crate::shared::models::TriggerKind;


pub fn execute_set_schedule(
    conn: &mut diesel::PgConnection,
    cron: &str,
    script_name: &str,
    bot_uuid: Uuid,
) -> Result<Value, Box<dyn std::error::Error>> {
    info!(
        "Scheduling SET SCHEDULE cron: {}, script: {}, bot_id: {:?}",
        cron, script_name, bot_uuid
    );

    // First check if bot exists
    use crate::shared::models::bots::dsl::bots;
    let bot_exists: bool = diesel::select(diesel::dsl::exists(
        bots.filter(crate::shared::models::bots::dsl::id.eq(bot_uuid))
    ))
    .get_result(conn)?;
    
    if !bot_exists {
        return Err(format!("Bot with id {} does not exist", bot_uuid).into());
    }

    use crate::shared::models::system_automations::dsl::*;

    let new_automation = (
        bot_id.eq(bot_uuid),
        kind.eq(TriggerKind::Scheduled as i32),
        schedule.eq(cron),
        param.eq(script_name),
        is_active.eq(true),
    );

    let result = diesel::insert_into(system_automations)
        .values(&new_automation)
        .on_conflict((bot_id, kind, param))
        .do_update()
        .set((
            schedule.eq(cron),
            is_active.eq(true),
            last_triggered.eq(None::<chrono::DateTime<chrono::Utc>>),
        ))
        .execute(&mut *conn)?;

    Ok(json!({
        "command": "set_schedule",
        "schedule": cron,
        "script": script_name,
        "bot_id": bot_uuid.to_string(),
        "rows_affected": result
    }))
}