use crossterm::*;
use std::process::Command;
use rustyline::error::ReadlineError;
use rustyline::Editor;
use crate::select::{draw_select, Entry, SelectResult};
use crate::version_control_actions::VersionControlActions;
const RESET_COLOR: Attribute = Attribute::Reset;
const HEADER_COLOR: Colored = Colored::Fg(Color::Black);
const HEADER_BG_COLOR: Colored = Colored::Bg(Color::Magenta);
const ACTION_COLOR: Colored = Colored::Fg(Color::Rgb {
r: 255,
g: 100,
b: 180,
});
const ENTRY_COLOR: Colored = Colored::Fg(Color::Rgb {
r: 255,
g: 180,
b: 100,
});
const DONE_COLOR: Colored = Colored::Fg(Color::Green);
const CANCEL_COLOR: Colored = Colored::Fg(Color::Yellow);
const ERROR_COLOR: Colored = Colored::Fg(Color::Red);
const VERSION: &'static str = env!("CARGO_PKG_VERSION");
pub fn show_tui<'a, T: VersionControlActions>(repository_name: &str, version_control: &'a mut T) {
Tui::new(repository_name, version_control).show();
}
struct Tui<'a, T: VersionControlActions + 'a> {
repository_name: &'a str,
version_control: &'a mut T,
_crossterm: Crossterm,
terminal: Terminal,
input: TerminalInput,
cursor: TerminalCursor,
readline: Editor<()>,
}
impl<'a, T: VersionControlActions> Tui<'a, T> {
fn new(repository_name: &'a str, version_control: &'a mut T) -> Self {
let crossterm = Crossterm::new();
let terminal = crossterm.terminal();
let input = crossterm.input();
let cursor = crossterm.cursor();
Tui {
repository_name: repository_name,
version_control: version_control,
_crossterm: crossterm,
terminal,
input,
cursor,
readline: Editor::new(),
}
}
fn show(&mut self) {
self.show_header();
self.show_help();
loop {
match self.input.read_char() {
Ok(key) => {
if !self.handle_key(key) {
return;
}
}
Err(error) => {
println!("Error reading input: {}", error);
return;
}
}
}
}
fn handle_key(&mut self, key: char) -> bool {
self.terminal.clear(ClearType::CurrentLine).unwrap();
self.cursor.move_left(1);
const CTRL_C: char = 3u8 as char;
const CTRL_R: char = 18u8 as char;
const CTRL_B: char = 2u8 as char;
if key.is_control() {
match key {
CTRL_C => {
return false;
}
CTRL_R => {
self.show_action("merge taking local");
let result = self.version_control.take_local();
self.handle_result(result);
}
CTRL_B => {
self.show_action("close branch");
if let Some(input) = self.handle_input("branch to close (ctrl+c to cancel): ") {
let result = self.version_control.close_branch(&input[..]);
self.handle_result(result);
}
}
_ => (),
}
} else {
match key {
'q' => {
return false;
}
'h' => {
self.show_action("help");
self.show_help();
}
'e' => {
self.show_action("explorer");
self.open_explorer();
}
's' => {
self.show_action("status");
let result = self.version_control.status();
self.handle_result(result);
}
'l' => {
self.show_action("log");
let result = self.version_control.log();
self.handle_result(result);
}
'd' => {
self.show_action("revision changes");
if let Some(input) = self.handle_input("show changes from (ctrl+c to cancel): ")
{
let result = self.version_control.changes(&input[..]);
self.handle_result(result);
}
}
'D' => {
self.show_action("revision diff");
if let Some(input) = self.handle_input("show diff from (ctrl+c to cancel): ") {
let result = self.version_control.diff(&input[..]);
self.handle_result(result);
}
}
'c' => {
self.show_action("commit all");
if let Some(input) = self.handle_input("commit message (ctrl+c to cancel): ") {
let result = self.version_control.commit_all(&input[..]);
self.handle_result(result);
}
}
'C' => {
self.show_action("commit selected");
match self.version_control.get_files_to_commit() {
Ok(mut entries) => {
self.show_add_remove_ui(&mut entries);
print!("\n\n");
if entries.iter().any(|e| e.selected) {
if let Some(input) =
self.handle_input("commit message (ctrl+c to cancel): ")
{
let result =
self.version_control.commit_selected(&input[..], &entries);
self.handle_result(result);
}
} else {
print!("\n\n{}canceled{}\n\n", CANCEL_COLOR, RESET_COLOR);
}
}
Err(error) => self.handle_result(Err(error)),
}
}
'u' => {
self.show_action("update");
if let Some(input) = self.handle_input("update to (ctrl+c to cancel): ") {
let result = self.version_control.update(&input[..]);
self.handle_result(result);
}
}
'x' => {
self.show_action("revert");
let result = self.version_control.revert();
self.handle_result(result);
}
'X' => {
self.show_action("revert selected");
let result = self.version_control.revert();
self.handle_result(result);
}
'm' => {
self.show_action("merge");
if let Some(input) = self.handle_input("merge with (ctrl+c to cancel): ") {
let result = self.version_control.merge(&input[..]);
self.handle_result(result);
}
}
'r' => {
self.show_action("unresolved conflicts");
let result = self.version_control.conflicts();
self.handle_result(result);
}
'R' => {
self.show_action("merge taking other");
let result = self.version_control.take_other();
self.handle_result(result);
}
'f' => {
self.show_action("fetch");
let result = self.version_control.fetch();
self.handle_result(result);
}
'p' => {
self.show_action("pull");
let result = self.version_control.pull();
self.handle_result(result);
}
'P' => {
self.show_action("push");
let result = self.version_control.push();
self.handle_result(result);
}
'T' => {
self.show_action("create tag");
if let Some(input) = self.handle_input("tag name (ctrl+c to cancel): ") {
let result = self.version_control.create_tag(&input[..]);
self.handle_result(result);
}
}
'b' => {
self.show_action("list branches");
let result = self.version_control.list_branches();
self.handle_result(result);
}
'B' => {
self.show_action("create branch");
if let Some(input) = self.handle_input("branch name (ctrl+c to cancel): ") {
let result = self.version_control.create_branch(&input[..]);
self.handle_result(result);
}
}
_ => (),
}
}
true
}
fn handle_input(&mut self, prompt: &str) -> Option<String> {
print!("{}{}{}\n", ENTRY_COLOR, prompt, RESET_COLOR);
match self.readline.readline("") {
Ok(line) => Some(line),
Err(ReadlineError::Interrupted) | Err(ReadlineError::Eof) => {
print!("\n\n{}canceled{}\n\n", CANCEL_COLOR, RESET_COLOR);
None
}
Err(err) => {
println!("error {:?}\n\n", err);
None
}
}
}
fn handle_result(&mut self, result: std::result::Result<String, String>) {
match result {
Ok(output) => {
print!("{}\n\n", output);
print!("{}done{}\n\n", DONE_COLOR, RESET_COLOR);
}
Err(error) => {
print!("{}\n\n", error);
print!("{}error{}\n\n", ERROR_COLOR, RESET_COLOR);
}
}
}
fn show_header(&mut self) {
self.terminal.clear(ClearType::All).unwrap();
let (w, _) = self.terminal.terminal_size();
self.cursor.goto(0, 0).unwrap();
print!("{}{}", HEADER_COLOR, HEADER_BG_COLOR,);
print!("{}", " ".repeat(w as usize + 1));
self.cursor.goto(0, 0).unwrap();
print!(
"{}Verco @ {}{}{}\n\n",
HEADER_COLOR, self.repository_name, RESET_COLOR, RESET_COLOR,
);
}
fn show_action(&mut self, action_name: &str) {
self.show_header();
print!("{}{}{}\n\n", ACTION_COLOR, action_name, RESET_COLOR);
}
fn show_help(&mut self) {
print!("Verco {}\n\n", VERSION);
match self.version_control.version() {
Ok(version) => {
print!("{}", version);
print!("\n\n");
}
Err(error) => {
print!("{}{}", ERROR_COLOR, error);
panic!("Could not find version control in system");
}
}
print!("press a key and peform an action\n\n");
self.show_help_action("h", "help\n");
self.show_help_action("e", "explorer\n");
self.show_help_action("s", "status");
self.show_help_action("l", "log\n");
self.show_help_action("d", "revision changes");
self.show_help_action("shift+d", "revision diff\n");
self.show_help_action("c", "commit all");
self.show_help_action("shift+c", "commit selected");
self.show_help_action("x", "revert");
self.show_help_action("shift+x", "revert selected");
self.show_help_action("u", "update/checkout");
self.show_help_action("m", "merge\n");
self.show_help_action("r", "unresolved conflicts");
self.show_help_action("shift+r", "resolve taking other");
self.show_help_action("ctrl+r", "resolve taking local\n");
self.show_help_action("f", "fetch");
self.show_help_action("p", "pull");
self.show_help_action("shift+p", "push\n");
self.show_help_action("shift+t", "create tag\n");
self.show_help_action("b", "list branches");
self.show_help_action("shift+b", "create branch");
self.show_help_action("ctrl+b", "close branch\n");
}
fn show_help_action(&mut self, shortcut: &str, action: &str) {
print!(
"\t{}{}{}\t\t{}\n",
ENTRY_COLOR, shortcut, RESET_COLOR, action
);
}
fn open_explorer(&mut self) {
let mut command = Command::new("explorer");
command.arg(self.repository_name);
command.spawn().expect("failed to open explorer");
print!("{}done{}\n\n", DONE_COLOR, RESET_COLOR);
}
pub fn show_add_remove_ui(&mut self, entries: &mut Vec<Entry>) {
let mut index = 0;
loop {
self.terminal.clear(ClearType::All).unwrap();
self.show_action("commit selected");
match draw_select(
&mut self.terminal,
&mut self.cursor,
&mut self.input,
entries,
&mut index,
) {
SelectResult::Cancel => break,
SelectResult::Selected => break,
SelectResult::Repeat => (),
}
}
}
}