botserver/src/basic/keywords/create_draft.rs

69 lines
2.4 KiB
Rust

use crate::email::save_email_draft;
use crate::email::{fetch_latest_sent_to, SaveDraftRequest};
use crate::shared::state::AppState;
use rhai::Dynamic;
use rhai::Engine;
pub fn create_draft_keyword(state: &AppState, engine: &mut Engine) {
let state_clone = state.clone();
engine
.register_custom_syntax(
&["CREATE_DRAFT", "$expr$", ",", "$expr$", ",", "$expr$"],
true, // Statement
move |context, inputs| {
// Extract arguments
let to = context.eval_expression_tree(&inputs[0])?.to_string();
let subject = context.eval_expression_tree(&inputs[1])?.to_string();
let reply_text = context.eval_expression_tree(&inputs[2])?.to_string();
// Execute async operations using the same pattern as FIND
let fut = execute_create_draft(&state_clone, &to, &subject, &reply_text);
let result =
tokio::task::block_in_place(|| tokio::runtime::Handle::current().block_on(fut))
.map_err(|e| format!("Draft creation error: {}", e))?;
Ok(Dynamic::from(result))
},
)
.unwrap();
}
async fn execute_create_draft(
state: &AppState,
to: &str,
subject: &str,
reply_text: &str,
) -> Result<String, String> {
let get_result = fetch_latest_sent_to(&state.config.clone().unwrap().email, to).await;
let email_body = if let Ok(get_result_str) = get_result {
if !get_result_str.is_empty() {
let email_separator = "<br><hr><br>"; // Horizontal rule in HTML
let formatted_reply_text = reply_text.to_string();
let formatted_old_text = get_result_str.replace("\n", "<br>");
let fixed_reply_text = formatted_reply_text.replace("FIX", "Fixed");
format!(
"{}{}{}",
fixed_reply_text, email_separator, formatted_old_text
)
} else {
reply_text.to_string()
}
} else {
reply_text.to_string()
};
// Create and save draft
let draft_request = SaveDraftRequest {
to: to.to_string(),
subject: subject.to_string(),
cc: None,
text: email_body,
};
let save_result = save_email_draft(&state.config.clone().unwrap().email, &draft_request).await;
match save_result {
Ok(_) => Ok("Draft saved successfully".to_string()),
Err(e) => Err(e.to_string()),
}
}