Function build_rune_script

tools/attractor.rs:169–214

(args: &CmdlineOptions)

}

fn build_rune_script(args: &CmdlineOptions) -> eyre::Result<rune::Source> {
    let mut lines = Vec::new();
    lines.push("pub fn iterate(x, y) {".into());
    lines.push("// parameters".into());
    // Add the alphabetic parameters first
    if let Some(letters) = &args.letters {
        let mut params = params_from_letters(letters)?;
        lines.append(&mut params);
    }

    // Add the explicit parameters second (allows overrides if you so wanted)
    for parameter in &args.parameter {
        lines.push(format!("let {parameter};"));
    }

    // Add the script if given
    if let Some(script) = &args.script {
        lines.push("// script".into());
        if script == "-" {
            for maybe_line in std::io::stdin().lock().lines() {
                let line = maybe_line?;
                lines.push(line);
            }
        } else {
            let file = std::fs::File::open(script)?;
            let reader = std::io::BufReader::new(file);
            for maybe_line in reader.lines() {
                let line = maybe_line?;
                lines.push(line);
            }
        }
    }

    // Finally append any --math expressions
    if !args.math.is_empty() {
        lines.push("// math".into());
    }
    lines.append(&mut args.math.clone());

    lines.push("return (x_new, y_new);".into());
    lines.push("}".into());
    let script = lines.join("\n");
    tracing::debug!("Generated rune script:\n==========\n{script}\n==========");
    let source = rune::Source::memory(script)?;
    Ok(source)
}

fn main() -> eyre::Result<()> {
    color_eyre::install()?;