git-whence 0.2.0

interactive git reblame
use ansi_to_tui::IntoText;
use git2::{Oid, Repository};
use std::{
	error,
	path::{Path, PathBuf},
	process, time, vec,
};
use tui::{
	style::{Color, Style},
	text::{Span, Spans, Text},
};

#[derive(Debug)]
pub struct BlameLine<'a> {
	pub spans: Spans<'a>,
	pub commit: Oid,
	pub path: Option<PathBuf>,
}

pub fn blame<'a>(
	repo: &'a Repository,
	rel_path: &Path,
	start_commit: Oid,
) -> Result<Vec<BlameLine<'a>>, Box<dyn error::Error>> {
	let output = process::Command::new("git")
		.args([
			"blame",
			"--porcelain",
			rel_path.to_str().unwrap(),
			&start_commit.to_string(),
		])
		.current_dir(repo.path())
		.output()?;
	if !output.status.success() {
		return Err(std::str::from_utf8(&output.stderr)?.into());
	}
	let blame_output = std::str::from_utf8(&output.stdout)?;
	let blame = crate::git_blame_porcelain::parse_blame_porcelain(blame_output)?;

	let mut out = vec![];
	let now = time::SystemTime::now();
	let duration_formatter = timeago::Formatter::new();
	for b in blame {
		let commit_time = b.info.commit_time;
		let time_display = duration_formatter.convert(now.duration_since(commit_time).unwrap_or_default());
		let mut spans = vec![
			Span::styled(format!("{:.8}", b.commit), Style::default().fg(Color::Yellow)),
			Span::raw(format!(" {}", fmt_width(b.info.author, 12))),
			Span::styled(
				format!(" {}", fmt_width(&time_display, 13)),
				Style::default().fg(Color::LightRed),
			),
		];
		spans.append(&mut format_line_num_and_code(b.line_num, b.code[0]));
		let line_path = b.info.path;
		out.push(BlameLine {
			spans: Spans::from(spans),
			commit: Oid::from_str(b.commit)?,
			path: line_path.map(|p| p.to_owned()),
		});

		for i in 1..b.code.len() {
			let mut spans = vec![Span::raw(" ".repeat(35))];
			let line_num = b.line_num + i32::try_from(i).unwrap();
			spans.append(&mut format_line_num_and_code(line_num, b.code[i]));
			out.push(BlameLine {
				spans: Spans::from(spans),
				commit: Oid::from_str(b.commit)?,
				path: line_path.map(|p| p.to_owned()),
			});
		}
	}
	Ok(out)
}

fn fmt_width(s: &str, width: usize) -> String {
	let mut out = String::new();
	match s.char_indices().nth(width) {
		None => out.push_str(s),
		Some((i, _)) => out.push_str(&s[..i]),
	};
	if out.len() < width {
		out.push_str(&" ".repeat(width - out.len()));
	}
	out
}

fn format_line_num_and_code(line_num: i32, line: &str) -> Vec<Span<'static>> {
	vec![
		Span::styled(format!(" {:4} ", line_num), Style::default().fg(Color::DarkGray)),
		Span::raw(line.replace('\t', "    ")),
	]
}

pub fn log_follow(repo: &Repository, rel_path: &Path, line_num: usize, start_commit: Oid) -> Text<'static> {
	let repo_path = repo.workdir().unwrap();
	let output = process::Command::new("git")
		.args([
			"log",
			"--color=always",
			"--abbrev-commit",
			"--decorate",
			"--date=local",
			"--find-copies",
			"-L",
			&format!("{},{}:{}", line_num + 1, line_num + 1, rel_path.display()),
			&start_commit.to_string(),
		])
		.current_dir(repo_path)
		.output();

	let buf = match output {
		Ok(o) => {
			if o.status.success() {
				o.stdout
			} else {
				o.stderr
			}
		}
		Err(e) => {
			return Text::raw(e.to_string());
		}
	};
	match buf.into_text() {
		Ok(t) => t,
		Err(e) => Text::raw(format!("ansi_to_tui:\n{}", e)),
	}
}