refactor: refactor

src/main.rs

@@ -1,88 +1,7 @@
 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]))
-}
+use copyrat::{run, Opt};
 
 fn main() {
     let opt = Opt::parse();
@@ -93,67 +12,6 @@ fn main() {
 
     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();
-        }
-    }
+    run(buffer, opt);
 }