90 lines
No EOL
3.3 KiB
Rust
90 lines
No EOL
3.3 KiB
Rust
use rhai::Dynamic;
|
|
use rhai::Engine;
|
|
use std::error::Error;
|
|
use std::fs;
|
|
use std::path::{ PathBuf};
|
|
use std::io::Read;
|
|
|
|
use crate::services::state::AppState;
|
|
use crate::services::utils;
|
|
|
|
pub fn create_site_keyword(state: &AppState, 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::services::config::AppConfig,
|
|
alias: Dynamic,
|
|
template_dir: Dynamic,
|
|
prompt: Dynamic,
|
|
) -> Result<String, Box<dyn Error + Send + Sync>> {
|
|
// Convert paths to platform-specific format
|
|
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());
|
|
|
|
// Create destination directory
|
|
fs::create_dir_all(&alias_path).map_err(|e| e.to_string())?;
|
|
|
|
// Process all HTML files in template directory
|
|
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");
|
|
}
|
|
}
|
|
|
|
// Combine template content with prompt
|
|
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()
|
|
);
|
|
|
|
// Call LLM with the combined prompt
|
|
println!("Asking LLM to create site.");
|
|
let llm_result = utils::call_llm(&full_prompt, &config.ai).await?;
|
|
|
|
// Write the generated HTML file
|
|
let index_path = alias_path.join("index.html");
|
|
fs::write(index_path, llm_result).map_err(|e| e.to_string())?;
|
|
|
|
println!("Site created at: {}", alias_path.display());
|
|
Ok(alias_path.to_string_lossy().into_owned())
|
|
} |