diff --git a/src/db/bots.rs b/src/db/bots.rs index c711b77..cd412ae 100644 --- a/src/db/bots.rs +++ b/src/db/bots.rs @@ -75,4 +75,19 @@ impl BotInstance { .await?; Ok(()) } + + pub async fn update_script(db: &mut D, name: &str, script: &str) -> DbResult<()> { + let bi = db.get_collection::().await; + + bi.update_one( + doc! {"name": name}, + doc! { "$set": { + "script": script, + "restart_flag": true, + } + }, + ) + .await?; + Ok(()) + } }