102 lines
2.9 KiB
Rust
102 lines
2.9 KiB
Rust
mod config;
|
|
#[cfg(test)]
|
|
mod test;
|
|
|
|
use std::error::Error;
|
|
use std::ops::{Deref, Not};
|
|
use std::path::PathBuf;
|
|
|
|
use anyhow::{anyhow, Result};
|
|
use clipboard::{ClipboardContext, ClipboardProvider};
|
|
use dirs::config_dir;
|
|
use tracing::{debug, error, Level};
|
|
use tracing_subscriber::FmtSubscriber;
|
|
|
|
use crate::config::{Act, Match, Replacements};
|
|
|
|
const VERSION_ARGS: [&str; 3] = ["version", "-v", "--version"];
|
|
|
|
fn main() -> Result<()> {
|
|
if check_for_version_arg() {
|
|
return Ok(());
|
|
}
|
|
configure_tracing();
|
|
let config_path = get_config_path()?;
|
|
let config_str = std::fs::read_to_string(config_path.as_path()).unwrap_or_default();
|
|
let config: Replacements<'_> = toml::from_str(&config_str)?;
|
|
loop_clipboard(config);
|
|
Ok(())
|
|
}
|
|
|
|
fn check_for_version_arg() -> bool {
|
|
let args: Vec<String> = std::env::args().collect();
|
|
for arg in args {
|
|
if VERSION_ARGS.contains(&arg.deref()) {
|
|
print_version();
|
|
return true;
|
|
}
|
|
}
|
|
false
|
|
}
|
|
|
|
fn print_version() {
|
|
println!(
|
|
"{}",
|
|
concat!(env!("CARGO_PKG_NAME"), " ", env!("CARGO_PKG_VERSION"))
|
|
);
|
|
}
|
|
|
|
fn configure_tracing() {
|
|
let subscriber = FmtSubscriber::builder()
|
|
.with_max_level(Level::TRACE)
|
|
.finish();
|
|
|
|
tracing::subscriber::set_global_default(subscriber).expect("setting default subscriber failed");
|
|
}
|
|
|
|
fn get_config_path() -> Result<PathBuf> {
|
|
let mut config_path = config_dir().ok_or_else(|| anyhow!("Failed to get config dir"))?;
|
|
config_path.push("substitutor");
|
|
config_path.push("config");
|
|
config_path.set_extension("toml");
|
|
Ok(config_path)
|
|
}
|
|
|
|
fn loop_clipboard(config: Replacements) {
|
|
let mut clipboard: ClipboardContext =
|
|
ClipboardProvider::new().expect("Failed to get clipboard");
|
|
let mut clipboard_contents = get_clipboard_contents(&mut clipboard);
|
|
while let Ok(contents) = clipboard_contents.as_deref() {
|
|
if let Some(subst) = config
|
|
.substitutors
|
|
.iter()
|
|
.find(|subst| subst.matcher.check_match(contents))
|
|
{
|
|
if subst.name.is_empty().not() {
|
|
debug!("{}: matched on {}...", &subst.name, truncate(contents, 40));
|
|
}
|
|
let result = subst.action.apply_action(contents);
|
|
if let Err(e) = clipboard.set_contents(result.to_owned()) {
|
|
error!("{e}");
|
|
}
|
|
};
|
|
while let Ok(new_contents) = get_clipboard_contents(&mut clipboard) {
|
|
if new_contents != contents {
|
|
clipboard_contents = Ok(new_contents);
|
|
break;
|
|
};
|
|
}
|
|
}
|
|
}
|
|
|
|
fn get_clipboard_contents(clipboard: &mut ClipboardContext) -> Result<String, Box<dyn Error>> {
|
|
clipboard.get_contents()
|
|
}
|
|
|
|
fn truncate(s: &str, max_chars: usize) -> &str {
|
|
match s.char_indices().nth(max_chars) {
|
|
None => s,
|
|
Some((idx, _)) => &s[..idx],
|
|
}
|
|
}
|