diff-tool 0.2.0

use std::{
    path::Path,
    process::{Command, Stdio},
};

#[derive(Default, Debug)]
pub struct Diff {
    old_diff: Vec<DiffLine>,
    current_diff: Vec<DiffLine>,
}

impl Diff {
    pub fn longest_diff_len(&self) -> usize {
        let old_diff = self.old_diff.len();
        let current_diff = self.current_diff.len();

        std::cmp::max(old_diff, current_diff) - 1
    }

    pub fn old_diff(&self) -> &[DiffLine] {
        &self.old_diff
    }

    pub fn current_diff(&self) -> &[DiffLine] {
        &self.current_diff
    }

    pub fn largest_line_number_char_len(&self) -> u16 {
        let (old_diff, current_diff) = self.largest_line_number();

        let largest_line_number = std::cmp::max(old_diff, current_diff);

        let length = std::cmp::min(largest_line_number.to_string().len(), u16::MAX.into());
        length.try_into().unwrap_or(4)
    }

    /// Gets the largest line number from each diff
    fn largest_line_number(&self) -> (usize, usize) {
        let old_diff = self
            .old_diff
            .iter()
            .map(|x| x.line_number().unwrap_or(0))
            .max()
            .unwrap_or(0);

        let current_diff = self
            .old_diff
            .iter()
            .map(|x| x.line_number().unwrap_or(0))
            .max()
            .unwrap_or(0);

        (old_diff, current_diff)
    }

    pub fn parse_diff(diff_string: &str) -> Self {
        let lines = diff_string.split("\n");

        let mut diff = Self::default();

        let mut start = false;
        let mut additions = 0;
        let mut removals = 0;
        let mut diff_one_line = 1;
        let mut diff_two_line = 1;

        for line in lines {
            if line.starts_with("@@") && line.ends_with("@@") {
                start = true;
                continue;
            }

            if !start {
                continue;
            }

            let (prefix, content) = remove_first_char(line);

            match prefix {
                '+' => {
                    diff.current_diff.push(DiffLine::new(
                        content,
                        DiffKind::Addition,
                        Some(diff_two_line),
                    ));
                    diff_two_line += 1;
                    if removals > 0 {
                        removals -= 1
                    } else {
                        additions += 1
                    }
                }

                '-' => {
                    diff.old_diff.push(DiffLine::new(
                        content,
                        DiffKind::Removal,
                        Some(diff_one_line),
                    ));
                    diff_one_line += 1;
                    removals += 1
                }
                _ => {
                    for _ in 0..removals {
                        diff.current_diff
                            .push(DiffLine::new("", DiffKind::Blank, None))
                    }

                    removals = 0;

                    for _ in 0..additions {
                        diff.old_diff.push(DiffLine::new("", DiffKind::Blank, None))
                    }

                    additions = 0;

                    diff.old_diff.push(DiffLine::new(
                        content,
                        DiffKind::Neutral,
                        Some(diff_one_line),
                    ));
                    diff_one_line += 1;
                    diff.current_diff.push(DiffLine::new(
                        content,
                        DiffKind::Neutral,
                        Some(diff_two_line),
                    ));
                    diff_two_line += 1
                }
            }
        }

        for _ in 0..removals {
            diff.current_diff
                .push(DiffLine::new("", DiffKind::Blank, None))
        }

        for _ in 0..additions {
            diff.old_diff.push(DiffLine::new("", DiffKind::Blank, None))
        }

        diff
    }
}

#[derive(Debug, Clone, Default)]
pub struct DiffLine {
    content: String,
    kind: DiffKind,
    line_number: Option<usize>,
}

impl DiffLine {
    fn new(content: &str, kind: DiffKind, line_number: Option<usize>) -> Self {
        let content = content.to_string();
        Self {
            content,
            kind,
            line_number,
        }
    }

    pub fn content(&self) -> &str {
        &self.content
    }

    pub fn kind(&self) -> &DiffKind {
        &self.kind
    }

    pub fn line_number(&self) -> &Option<usize> {
        &self.line_number
    }
}

#[derive(Debug, Clone, Copy, Default)]
pub enum DiffKind {
    Addition,
    Removal,
    Neutral,
    #[default]
    Blank,
}

impl DiffKind {
    pub fn value(&self) -> &str {
        match self {
            DiffKind::Addition => "+",
            DiffKind::Removal => "-",
            DiffKind::Neutral => " ",
            DiffKind::Blank => " ",
        }
    }
}

/// Performs 'git diff -U1000 <filename>' or 'git -C [path] diff -U1000 <filename>' and returns the result as a string
pub fn get_raw_diff(path: &Path, dir_flag: bool) -> String {
    let args = if !dir_flag {
        vec!["diff", "-U1000", path.to_str().unwrap()]
    } else {
        vec![
            "-C",
            path.parent().unwrap().to_str().unwrap(),
            "diff",
            "-U1000",
            path.file_name().unwrap().to_str().unwrap(),
        ]
    };

    // Process git diff <filename> command and save the stdout response
    let output = Command::new("git")
        .args(args)
        .stdout(Stdio::piped())
        .output()
        .expect("Failed to execute git diff");

    // Convert stdout response to a string and return
    String::from_utf8(output.stdout).expect("UTF8 data to convert to string")
}

fn remove_first_char(string: &str) -> (char, &str) {
    let mut chars = string.chars();
    (chars.next().unwrap_or(' '), chars.as_str())
}