botserver/src/basic/keywords/find.rs

71 lines
2.4 KiB
Rust
Raw Normal View History

use crate::shared::models::UserSession;
use crate::shared::state::AppState;
use crate::shared::utils;
use crate::shared::utils::to_array;
2025-10-11 20:02:14 -03:00
use diesel::pg::PgConnection;
2025-10-11 12:29:03 -03:00
use diesel::prelude::*;
use log::error;
use log::trace;
2025-10-06 10:30:17 -03:00
use rhai::Dynamic;
use rhai::Engine;
use serde_json::{json, Value};
2025-10-11 20:02:14 -03:00
pub fn find_keyword(state: &AppState, _user: UserSession, engine: &mut Engine) {
let connection = state.conn.clone();
2025-10-11 20:02:14 -03:00
engine
.register_custom_syntax(&["FIND", "$expr$", ",", "$expr$"], false, {
move |context, inputs| {
let table_name = context.eval_expression_tree(&inputs[0])?;
let filter = context.eval_expression_tree(&inputs[1])?;
let mut binding = connection.get().map_err(|e| format!("DB error: {}", e))?;
2025-10-11 20:02:14 -03:00
let binding2 = table_name.to_string();
let binding3 = filter.to_string();
let result = tokio::task::block_in_place(|| {
tokio::runtime::Handle::current()
.block_on(async { execute_find(&mut binding, &binding2, &binding3).await })
})
2025-10-11 13:29:38 -03:00
.map_err(|e| format!("DB error: {}", e))?;
2025-10-11 20:02:14 -03:00
if let Some(results) = result.get("results") {
let array = to_array(utils::json_value_to_dynamic(results));
Ok(Dynamic::from(array))
} else {
Err("No results".into())
}
2025-10-06 10:30:17 -03:00
}
2025-10-11 20:02:14 -03:00
})
.unwrap();
2025-10-06 10:30:17 -03:00
}
2025-10-11 20:02:14 -03:00
pub async fn execute_find(
conn: &mut PgConnection,
2025-10-06 10:30:17 -03:00
table_str: &str,
filter_str: &str,
) -> Result<Value, String> {
trace!(
2025-10-06 10:30:17 -03:00
"Starting execute_find with table: {}, filter: {}",
table_str,
filter_str
2025-10-06 10:30:17 -03:00
);
2025-10-11 20:02:14 -03:00
let (where_clause, params) = utils::parse_filter(filter_str).map_err(|e| e.to_string())?;
2025-10-06 10:30:17 -03:00
let query = format!(
"SELECT * FROM {} WHERE {} LIMIT 10",
table_str, where_clause
);
let _raw_result = diesel::sql_query(&query)
2025-10-11 20:02:14 -03:00
.bind::<diesel::sql_types::Text, _>(&params[0])
.execute(conn)
2025-10-06 10:30:17 -03:00
.map_err(|e| {
error!("SQL execution error: {}", e);
e.to_string()
})?;
let mut results = Vec::new();
2025-10-11 20:02:14 -03:00
let json_row = serde_json::json!({
"note": "Dynamic row deserialization not implemented - need table schema"
2025-10-11 20:02:14 -03:00
});
results.push(json_row);
2025-10-06 10:30:17 -03:00
Ok(json!({
"command": "find",
"table": table_str,
"filter": filter_str,
"results": results
2025-10-06 10:30:17 -03:00
}))
}