tmux-copyrat/src/main.rs

use clap::Clap;
use std::fs::OpenOptions;
use std::io::prelude::*;
use std::io::{self, Read};
use std::path;

mod alphabets;
mod colors;
mod error;
mod state;
mod view;

/// Main configuration, parsed from command line.
#[derive(Clap, Debug)]
#[clap(author, about, version)]
struct Opt {
    /// Alphabet to draw hints from.
    ///
    /// Possible values are "{A}", "{A}-homerow", "{A}-left-hand",
    /// "{A}-right-hand", where "{A}" is one of "qwerty", "azerty", "qwertz",
    /// "dvorak", "colemak". Examples: "qwerty", "dvorak-homerow".
    #[clap(short = "k", long, default_value = "qwerty",
                parse(try_from_str = alphabets::parse_alphabet))]
    alphabet: alphabets::Alphabet,

    /// Enable multi-selection.
    #[clap(short, long)]
    multi_selection: bool,

    #[clap(flatten)]
    colors: view::ViewColors,

    /// Reverse the order for assigned hints.
    #[clap(short, long)]
    reverse: bool,

    /// Keep the same hint for identical matches.
    #[clap(short, long)]
    unique: bool,

    /// Align hint with its match.
    #[clap(short = "a", long, arg_enum, default_value = "Leading")]
    hint_alignment: view::HintAlignment,

    /// Additional regex patterns.
    #[clap(short = "c", long)]
    custom_regex: Vec<String>,

    /// Optional hint styling.
    ///
    /// Underline or surround the hint for increased visibility.
    /// If not provided, only the hint colors will be used.
    #[clap(short = "s", long, arg_enum)]
    hint_style: Option<HintStyleCli>,

    /// Chars surrounding each hint, used with `Surrounded` style.
    #[clap(long, default_value = "{}",
                parse(try_from_str = parse_chars))]
    hint_surroundings: (char, char),

    /// Target path where to store the selected matches.
    #[clap(short = "o", long = "output", parse(from_os_str))]
    target_path: Option<path::PathBuf>,

    /// Only output if key was uppercased.
    #[clap(long)]
    uppercased: bool,
}

/// Type introduced due to parsing limitation,
/// as we cannot directly parse into view::HintStyle.
#[derive(Debug, Clap)]
enum HintStyleCli {
    Underlined,
    Surrounded,
}

fn parse_chars(src: &str) -> Result<(char, char), error::ParseError> {
    if src.len() != 2 {
        return Err(error::ParseError::ExpectedSurroundingPair);
    }

    let chars: Vec<char> = src.chars().collect();
    Ok((chars[0], chars[1]))
}

fn main() {
    let opt = Opt::parse();

    // Copy the pane contents (piped in via stdin) into a buffer, and split lines.
    let stdin = io::stdin();
    let mut handle = stdin.lock();

    let mut buffer = String::new();
    handle.read_to_string(&mut buffer).unwrap();
    let lines: Vec<&str> = buffer.split('\n').collect();

    let mut state = state::State::new(&lines, &opt.alphabet, &opt.custom_regex);

    let hint_style = match opt.hint_style {
        None => None,
        Some(style) => match style {
            HintStyleCli::Underlined => Some(view::HintStyle::Underlined),
            HintStyleCli::Surrounded => {
                let (open, close) = opt.hint_surroundings;
                Some(view::HintStyle::Surrounded(open, close))
            }
        },
    };
    let uppercase_flag = opt.uppercased;

    let selections = {
        let mut viewbox = view::View::new(
            &mut state,
            opt.multi_selection,
            opt.reverse,
            opt.unique,
            opt.hint_alignment,
            &opt.colors,
            hint_style,
        );

        viewbox.present()
    };

    // Early exit, signaling tmux we had no selections.
    if selections.is_empty() {
        ::std::process::exit(1);
    }

    let output = selections
        .iter()
        .map(|(text, uppercased)| {
            let upcase_value = if *uppercased { "true" } else { "false" };

            let output = if uppercase_flag { upcase_value } else { text };
            // let mut output = &opt.format;

            // output = str::replace(&output, "%U", upcase_value);
            // output = str::replace(&output, "%H", text.as_str());
            output
        })
        .collect::<Vec<&str>>()
        .join("\n");

    match opt.target_path {
        None => println!("{}", output),
        Some(target) => {
            let mut file = OpenOptions::new()
                .create(true)
                .truncate(true)
                .write(true)
                .open(target)
                .expect("Unable to open the target file");

            file.write(output.as_bytes()).unwrap();
        }
    }
}