45 lines
1.3 KiB
Rust
45 lines
1.3 KiB
Rust
|
use crate::{kak_response, open_command_fifo, KakMessage};
|
||
|
|
use evalexpr::{eval, Value};
|
||
|
|
use std::io::Write;
|
||
|
|
|
||
|
|
#[derive(clap::StructOpt, Debug)]
|
||
|
|
pub struct Options;
|
||
|
|
pub fn math_eval(_options: &Options) -> Result<KakMessage, KakMessage> {
|
||
|
|
let selections = kak_response("%val{selections}")?;
|
||
|
|
|
||
|
|
let mut f = open_command_fifo()?;
|
||
|
|
write!(f, "reg '\"'")?;
|
||
|
|
|
||
|
|
let mut err = None;
|
||
|
|
let mut err_count: usize = 0;
|
||
|
|
|
||
|
|
for i in selections.iter().map(|s| {
|
||
|
|
// TODO: Do all of these need to be strings?
|
||
|
|
match eval(s) {
|
||
|
|
Ok(Value::Float(f)) => Some(f.to_string()),
|
||
|
|
Ok(Value::Int(f)) => Some(f.to_string()),
|
||
|
|
// TODO: Should this be none?
|
||
|
|
Ok(_) => None,
|
||
|
|
Err(e) => {
|
||
|
|
eprintln!("Error: {:?}", e);
|
||
|
|
if err.is_none() {
|
||
|
|
err = Some(e);
|
||
|
|
err_count = err_count.saturating_add(1);
|
||
|
|
}
|
||
|
|
None
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}) {
|
||
|
|
// TODO: String allocation?
|
||
|
|
let new_selection = i.map(|s| s.replace('\'', "''"));
|
||
|
|
// .unwrap_or_else(|| "".to_string());
|
||
|
|
write!(f, " '{}'", new_selection.as_deref().unwrap_or(""))?;
|
||
|
|
}
|
||
|
|
write!(f, " ; exec R;")?;
|
||
|
|
|
||
|
|
Ok(KakMessage(
|
||
|
|
format!("MathEval {} selections", selections.len()),
|
||
|
|
None,
|
||
|
|
))
|
||
|
|
}
|