#![allow(unused)]

use git_odb::FindExt;
use git_repository as git;
use git_repository::easy::Reference;

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let repo = git::discover(".")?;
    println!(
        "Repo: {}",
        repo.work_tree.as_deref().unwrap_or(repo.git_dir()).display()
    );
    let handle = repo.to_easy();
    let commit_ids = handle
        .head()?
        .into_fully_peeled_id()
        .ok_or_else(|| "There are no commits - nothing to do here.")??
        .ancestors()
        .all()
        .collect::<Result<Vec<_>, _>>()?;
    println!("Num Commits: {}", commit_ids.len());
    assert!(!commit_ids.is_empty(), "checked that before");

    let last_commit_id = &commit_ids[0];
    println!("Most recent commit message");

    let object = last_commit_id.object()?;
    let commit = object.into_commit();
    println!("{}", commit.message_raw()?);

    let tree = commit.tree()?;

    let mut delegate = visit::Tree::new(handle.clone());
    tree.traverse().breadthfirst(&mut delegate)?;
    let _files = tree.traverse().breadthfirst.files()?;

    println!("num trees: {}", delegate.num_trees);
    println!("num blobs: {}", delegate.num_blobs);
    println!("num blobs_executable: {}", delegate.num_blobs_exec);
    println!("num links: {}", delegate.num_links);
    println!("num submodules: {}", delegate.num_submodules);
    println!("total size in bytes: {}\n", delegate.num_bytes);

    // let num_branches = handle.branches()?;
    // let num_branches = handle.branches.remote("origin")?;
    let num_branches = handle.references()?.prefixed("refs/heads/")?.count();
    let num_remote_branches = handle.references()?.prefixed("refs/remotes/")?.count();
    let num_tags = handle.references()?.prefixed("refs/tags/")?.count();
    let broken_refs = handle
        .references()?
        .all()?
        .filter_map(Result::ok)
        .filter_map(|r: Reference| r.into_fully_peeled_id().err())
        .count();
    let inaccessible_refs = handle.references()?.all()?.filter(Result::is_err).count();

    println!("num local branches: {}", num_branches);
    println!("num remote branches: {}", num_remote_branches);
    println!("num tags: {}", num_tags);
    println!("refs with inaccessible objects: {}", broken_refs);
    println!("inaccessible refs: {}", inaccessible_refs);

    Ok(())
}

mod visit {
    use std::process::id;

    use git_hash::oid;
    use git_object::{bstr::BStr, tree::EntryRef};
    use git_repository as git;
    use git_traverse::tree::visit::Action;

    pub(crate) struct Tree {
        pub num_trees: usize,
        pub num_links: usize,
        pub num_blobs: usize,
        pub num_blobs_exec: usize,
        pub num_submodules: usize,
        pub num_bytes: u64,
        pub repo: git::easy::Handle,
    }

    impl Tree {
        pub fn new(repo: git::easy::Handle) -> Self {
            Tree {
                num_trees: 0,
                num_links: 0,
                num_blobs: 0,
                num_blobs_exec: 0,
                num_submodules: 0,
                num_bytes: 0,
                repo,
            }
        }

        pub(crate) fn count_bytes(&mut self, oid: &oid) {
            if let Ok(obj) = self.repo.find_object(oid) {
                self.num_bytes += obj.data.len() as u64;
            }
        }
    }
    impl git_traverse::tree::Visit for Tree {
        fn pop_front_tracked_path_and_set_current(&mut self) {}

        fn push_back_tracked_path_component(&mut self, component: &BStr) {}

        fn push_path_component(&mut self, component: &BStr) {}

        fn pop_path_component(&mut self) {}

        fn visit_tree(&mut self, entry: &EntryRef<'_>) -> Action {
            self.num_trees += 1;
            Action::Continue
        }

        fn visit_nontree(&mut self, entry: &EntryRef<'_>) -> Action {
            use git::objs::tree::EntryMode::*;
            match entry.mode {
                Commit => self.num_submodules += 1,
                Blob => {
                    self.count_bytes(entry.oid);
                    self.num_blobs += 1
                }
                BlobExecutable => {
                    self.count_bytes(entry.oid);
                    self.num_blobs_exec += 1
                }
                Link => self.num_links += 1,
                Tree => unreachable!("BUG"),
            }
            Action::Continue
        }
    }
}