idgit 0.0.1

#![feature(with_options, assert_matches)]

use bstr::BString;
use idgit::{Error, FileContents, Meta, Repo, RepoFile, Result};
use rand::Rng;
use std::{
    fs::{self, File},
    io::Write,
    path::Path,
};

use cmd_lib::run_fun;
use insta::{assert_debug_snapshot, assert_snapshot};
use tempfile::TempDir;

#[allow(unused)]
use pretty_assertions::{assert_eq, assert_ne};
#[allow(unused)]
use tracing::{debug, error, info, instrument, span, warn};

fn log_cmd(cmd: std::result::Result<String, std::io::Error>) {
    let stdout = cmd.expect("Command failed");
    info!("Ran command, got: {}", stdout);
}

struct SampleRepoDir(TempDir);

impl SampleRepoDir {
    fn new() -> Self {
        let mut this = Self(tempfile::tempdir().unwrap());
        this.init();
        this
    }

    fn path(&self) -> &Path {
        self.0.path()
    }

    fn path_str(&self) -> &str {
        self.path().to_str().unwrap()
    }

    fn init(&mut self) {
        let path = self.path_str();

        log_cmd(run_fun! {
            cd $path;
            git init
        });
    }

    fn change_something(&mut self) {
        let name = format!("something_{}.txt", rand::thread_rng().gen::<u64>());
        self.set_file(&name, b"some change");
    }

    fn create_dir<N: AsRef<Path>>(&mut self, name: N) {
        fs::create_dir(self.path().join(name)).unwrap();
    }

    fn set_file<N: AsRef<Path>>(&mut self, name: N, contents: &'_ [u8]) {
        let path = self.path().join(name);

        File::with_options()
            .write(true)
            .create(true)
            .open(path)
            .unwrap()
            .write_all(contents)
            .unwrap();
    }

    fn remove_file<N: AsRef<Path>>(&mut self, name: N) {
        fs::remove_file(self.path().join(name)).unwrap();
    }

    fn commit_all(&mut self) {
        let path = self.path_str();
        log_cmd(run_fun! {
            cd $path;
            git add *;
            git commit -am "Make some change";
        });
    }

    fn add<N: AsRef<Path>>(&mut self, name: N) {
        let path = self.path_str();
        let name = name.as_ref().to_str().unwrap();

        log_cmd(run_fun! {
            cd $path;
            git add $name;
        });
    }

    fn status(&self) -> String {
        let path = self.path_str();
        (run_fun! {
            cd $path;
            git status;
        })
        .expect("Command failed")
    }

    fn assert_all_clean(&self) {
        assert_eq!(
            self.status(),
            "On branch master\nnothing to commit, working tree clean"
        );
    }
}

fn init_logs() {
    let _ = tracing_subscriber::fmt::fmt()
        .with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
        .pretty()
        .try_init();
}

#[test]
fn can_open_blank() {
    init_logs();

    let dir = SampleRepoDir::new();
    Repo::open(dir.path()).unwrap();
}

#[test]
fn can_open_used() -> Result<()> {
    init_logs();

    let mut dir = SampleRepoDir::new();
    dir.change_something();
    dir.commit_all();

    let path = dir.path_str();
    log_cmd(run_fun! {
        cd $path;
        git log;
    });

    let _repo = Repo::open(dir.path())?;

    Ok(())
}

#[test]
fn uncommitted_files() -> Result<()> {
    init_logs();

    let mut dir = SampleRepoDir::new();
    let repo = Repo::open(dir.path())?;

    dir.set_file("foo.txt", b"foo");
    dir.set_file("unchanged.txt", b"unchanged");
    dir.set_file("deleted.txt", b"deleted");
    dir.set_file("changed_to_bin", b"text");
    dir.set_file("renamed", b"renamed contents");
    dir.set_file(".gitignore", b"*.ignored\n");
    dir.commit_all();

    dir.set_file("example.ignored", b"ignored");

    dir.remove_file("deleted.txt");

    dir.set_file("staged.txt", b"already staged");
    dir.add("staged.txt");

    dir.set_file("foo.txt", b"foobar");
    dir.set_file("qux.txt", b"quz");

    dir.create_dir("example_dir");
    dir.set_file("example_dir/nested.txt", b"nested");

    dir.remove_file("renamed");
    dir.set_file("renamed_to", b"renamed contents");
    dir.add("renamed");
    dir.add("renamed_to");

    assert_debug_snapshot!(repo.uncommitted_files()?);

    Ok(())
}

#[test]
fn uncommitted_no_commits_no_untracked() -> Result<()> {
    init_logs();

    let dir = SampleRepoDir::new();
    let repo = Repo::open(dir.path())?;
    assert_eq!(repo.uncommitted_files()?.len(), 0);

    Ok(())
}

#[test]
fn uncommitted_no_commits_with_untracked() -> Result<()> {
    init_logs();

    let mut dir = SampleRepoDir::new();
    dir.set_file("name", b"contents");
    dir.create_dir("example_dir");
    dir.set_file("example_dir/child_recursed_into", b"");

    let repo = Repo::open(dir.path())?;

    assert_matches!(repo.uncommitted_files()?.as_slice(), [
        Meta::Untracked(a), Meta::Untracked(b)] if
            a.rel_path().to_str().unwrap() == "example_dir/child_recursed_into" &&
            b.rel_path().to_str().unwrap() == "name"
    );

    Ok(())
}

#[test]
fn uncommitted_details() -> Result<()> {
    init_logs();
    let mut dir = SampleRepoDir::new();
    let repo = Repo::open(&dir.path())?;

    let old = b"
         fn undo(&mut self, target: &mut Self::Target) -> undo::Result<Self> {
            match self {
                Change::StageFile(file) => target.do_stage_file(file),
                Change::UnstageFile(file) => target.do_unstage_file(file),
            }
         }
    ";

    let new = b"
         fn undo(&mut self, target: &mut Self::Target) -> undo::Result<Self> {
            match self {
                Change::StageFile(file) => target.do_unstage_file(file),
                Change::UnstageFile(file) => target.do_stage_file(file),
            }
         }
    ";

    dir.set_file("file", old);
    dir.commit_all();
    dir.set_file("file", new);

    let uncommitted = repo.uncommitted_files()?;
    let diff = &uncommitted[0];

    let changes = repo.diff_details(diff)?;
    debug!(?changes);

    Ok(())
}

#[test]
fn cant_get_details_of_file_in_untracked_dir() -> Result<()> {
    init_logs();

    let mut dir = SampleRepoDir::new();
    let repo = Repo::open(dir.path())?;

    dir.create_dir("example_dir");
    dir.set_file("example_dir/nested.txt", b"nested");

    let uncommitted = repo.uncommitted_files()?;
    let diff = &uncommitted[0];

    assert_matches!(repo.diff_details(diff), Err(Error::PathNotFound(_)));

    Ok(())
}

#[test]
fn stage_file() -> Result<()> {
    init_logs();

    let mut dir = SampleRepoDir::new();

    dir.set_file("a.txt", b"a");
    dir.set_file("b.txt", b"b");

    let mut repo = Repo::open(dir.path())?;

    repo.stage_file(RepoFile::new("a.txt"))?;

    assert_matches!(
        repo.uncommitted_files()?.as_slice(),
        [Meta::Added(_), Meta::Untracked(_)]
    );

    Ok(())
}

#[test]
fn unstage_file() -> Result<()> {
    init_logs();
    let mut dir = SampleRepoDir::new();
    let mut repo = Repo::open(dir.path())?;

    dir.set_file("f", b"contents");
    let file = RepoFile::new("f");

    let uncommitted = repo.uncommitted_files()?;
    assert_eq!(uncommitted.len(), 1);

    repo.stage_file(file.clone())?;
    assert_matches!(repo.uncommitted_files()?.as_slice(), [Meta::Added(_)]);

    repo.unstage_file(file)?;
    assert_matches!(repo.uncommitted_files()?.as_slice(), [Meta::Untracked(_)]);

    Ok(())
}

#[test]
fn undo_redo_stage_file() -> Result<()> {
    init_logs();
    let mut dir = SampleRepoDir::new();
    let mut repo = Repo::open(dir.path())?;

    dir.set_file("f", b"contents");
    let uncommitted = repo.uncommitted_files()?;
    assert_eq!(uncommitted.len(), 1);
    let file = RepoFile::new("f");

    repo.stage_file(file)?;
    assert_matches!(repo.uncommitted_files()?.as_slice(), [Meta::Added(_)]);

    repo.undo()?;
    assert_matches!(repo.uncommitted_files()?.as_slice(), [Meta::Untracked(_)]);

    repo.redo()?;
    assert_matches!(repo.uncommitted_files()?.as_slice(), [Meta::Added(_)]);

    Ok(())
}

#[test]
fn undo_redo_unstage_file() -> Result<()> {
    init_logs();
    let mut dir = SampleRepoDir::new();
    let mut repo = Repo::open(dir.path())?;

    dir.set_file("f", b"contents");
    let file = RepoFile::new("f");

    repo.stage_file(file.clone())?;

    let uncommitted = repo.uncommitted_files()?;
    assert_matches!(uncommitted.as_slice(), [Meta::Added(_)]);

    repo.unstage_file(file)?;
    assert_matches!(repo.uncommitted_files()?.as_slice(), [Meta::Untracked(_)]);

    repo.undo()?;
    assert_matches!(repo.uncommitted_files()?.as_slice(), [Meta::Added(_)]);

    repo.redo()?;
    assert_matches!(repo.uncommitted_files()?.as_slice(), [Meta::Untracked(_)]);

    Ok(())
}

#[test]
fn uncommitted_change_for_nonexistent_errors() -> Result<()> {
    init_logs();
    let mut dir = SampleRepoDir::new();
    let repo = Repo::open(&dir.path())?;

    dir.set_file("file", b"contents");

    let uncommitted = repo.uncommitted_files()?;

    dir.remove_file("file");

    let delta = &uncommitted[0];
    assert_matches!(repo.diff_details(delta), Err(idgit::Error::PathNotFound(_)));

    Ok(())
}

#[test]
fn stage_with_empty_contents() -> Result<()> {
    init_logs();
    let mut dir = SampleRepoDir::new();
    let mut repo = Repo::open(&dir.path())?;

    dir.set_file("file.txt", b"initial content");
    dir.commit_all();

    let file = RepoFile::new("file.txt");
    let contents = FileContents::new(BString::from(""));

    dir.assert_all_clean();

    repo.stage_contents(file, contents)?;

    assert_snapshot!(dir.status());

    Ok(())
}

#[test]
fn stage_to_match_workdir() -> Result<()> {
    init_logs();
    let mut dir = SampleRepoDir::new();
    let mut repo = Repo::open(&dir.path())?;

    dir.set_file("file.txt", b"initial content");
    dir.commit_all();

    let file = RepoFile::new("file.txt");
    let contents = FileContents::new(BString::from("changed content"));

    dir.assert_all_clean();
    dir.set_file("file.txt", b"changed content");

    repo.stage_contents(file, contents)?;

    assert_snapshot!(dir.status());

    Ok(())
}

#[test]
fn stage_not_matching_workdir() -> Result<()> {
    init_logs();
    let mut dir = SampleRepoDir::new();
    let mut repo = Repo::open(&dir.path())?;

    dir.set_file("file.txt", b"initial content");
    dir.commit_all();

    let file = RepoFile::new("file.txt");
    let contents = FileContents::new(BString::from("changed content"));
    dir.assert_all_clean();

    repo.stage_contents(file, contents)?;

    assert_snapshot!(dir.status());

    Ok(())
}