Rodrigo Rodriguez (Pragmatismo)
fd45f4e0dd
- Removed unused `id` and `app_state` fields from `ChatPanel`; updated constructor to accept but ignore the state, reducing memory footprint. - Switched database access in `ChatPanel` from a raw `Mutex` lock to a connection pool (`app_state.conn.get()`), improving concurrency and error handling. - Reordered and cleaned up imports in `status_panel.rs` and formatted struct fields for readability. - Updated VS Code launch configuration to pass `--noui` argument, enabling headless mode for debugging. - Bumped several crate versions in `Cargo.lock` (e.g., `bitflags` to 2.10.0, `syn` to 2.0.108, `cookie` to 0.16.2) and added the new `ashpd` dependency, aligning the project with latest library releases.
50 lines
2.4 KiB
Rust
50 lines
2.4 KiB
Rust
use rhai::Dynamic;
|
|
use rhai::Engine;
|
|
use std::error::Error;
|
|
use std::fs;
|
|
use std::io::Read;
|
|
use std::path::PathBuf;
|
|
use crate::shared::models::UserSession;
|
|
use crate::shared::state::AppState;
|
|
pub fn create_site_keyword(state: &AppState, _user: UserSession, engine: &mut Engine) {
|
|
let state_clone = state.clone();
|
|
engine
|
|
.register_custom_syntax(&["CREATE_SITE", "$expr$", ",", "$expr$", ",", "$expr$"], true, move |context, inputs| {
|
|
if inputs.len() < 3 {
|
|
return Err("Not enough arguments for CREATE SITE".into());
|
|
}
|
|
let alias = context.eval_expression_tree(&inputs[0])?;
|
|
let template_dir = context.eval_expression_tree(&inputs[1])?;
|
|
let prompt = context.eval_expression_tree(&inputs[2])?;
|
|
let config = state_clone.config.as_ref().expect("Config must be initialized").clone();
|
|
let fut = create_site(&config, alias, template_dir, prompt);
|
|
let result = tokio::task::block_in_place(|| tokio::runtime::Handle::current().block_on(fut))
|
|
.map_err(|e| format!("Site creation failed: {}", e))?;
|
|
Ok(Dynamic::from(result))
|
|
},
|
|
)
|
|
.unwrap();
|
|
}
|
|
async fn create_site(config: &crate::config::AppConfig, alias: Dynamic, template_dir: Dynamic, prompt: Dynamic) -> Result<String, Box<dyn Error + Send + Sync>> {
|
|
let base_path = PathBuf::from(&config.site_path);
|
|
let template_path = base_path.join(template_dir.to_string());
|
|
let alias_path = base_path.join(alias.to_string());
|
|
fs::create_dir_all(&alias_path).map_err(|e| e.to_string())?;
|
|
let mut combined_content = String::new();
|
|
for entry in fs::read_dir(&template_path).map_err(|e| e.to_string())? {
|
|
let entry = entry.map_err(|e| e.to_string())?;
|
|
let path = entry.path();
|
|
if path.extension().map_or(false, |ext| ext == "html") {
|
|
let mut file = fs::File::open(&path).map_err(|e| e.to_string())?;
|
|
let mut contents = String::new();
|
|
file.read_to_string(&mut contents).map_err(|e| e.to_string())?;
|
|
combined_content.push_str(&contents);
|
|
combined_content.push_str("\n\n--- TEMPLATE SEPARATOR ---\n\n");
|
|
}
|
|
}
|
|
let _full_prompt = format!("TEMPLATE FILES:\n{}\n\nPROMPT: {}\n\nGenerate a new HTML file cloning all previous TEMPLATE (keeping only the local _assets libraries use, no external resources), but turning this into this prompt:", combined_content, prompt.to_string());
|
|
let llm_result = "".to_string();
|
|
let index_path = alias_path.join("index.html");
|
|
fs::write(index_path, llm_result).map_err(|e| e.to_string())?;
|
|
Ok(alias_path.to_string_lossy().into_owned())
|
|
}
|