Сущность MyGit

Cargo.toml

@@ -6,3 +6,5 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+git2 = "0.18.3"
+walkdir = "2.5.0"

src/arch.rs

@@ -1,3 +1,15 @@
+use walkdir::WalkDir;
+
 pub fn read() -> String {
-    todo!()
-}
+    todo!("")
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::arch::read;
+
+    #[test]
+    fn it_works() {
+        read();
+    }
+}

src/main.rs

@@ -1,19 +1,17 @@
 use std::env::args;
 
-use clap::Parser;
-
-mod git;
+mod mygit;
 mod arch;
 
 fn main() {
     let from_branch = args().nth(1).expect("expected source branch name");
-    let into_branch = git::current_branch();
+    let into_branch = mygit::current_branch();
 
-    git::co(&from_branch);
+    mygit::co(&from_branch);
     let from_branch = arch::read();
 
-    git::co(&into_branch);
+    mygit::co(&into_branch);
     let into_branch = arch::read();
 
-    println!("{}", git::diff(&from_branch, &into_branch));
+    println!("{}", mygit::diff(&from_branch, &into_branch));
 }

src/mygit.rs

@@ -0,0 +1,68 @@
+use std::process::Command;
+use std::str;
+
+use git2::{BranchType, Repository};
+
+struct MyGit {
+    repo: Repository,
+}
+
+impl MyGit {
+    pub fn new() -> Self {
+        match Repository::init(".") {
+            Ok(repo) => Self { repo },
+            Err(e) => panic!("failed to init: {}", e)
+        }
+    }
+
+    pub fn current_branch_name(&self) -> String {
+        let repo = &self.repo;
+        let head = repo.head().unwrap();
+        let name = head.name().unwrap();
+        let name = name.trim_start_matches("refs/heads/");
+        name.to_string()
+    }
+
+    pub fn checkout(&self, branch_name: &str) {
+        let repo = &self.repo;
+    }
+}
+
+pub fn co(branch: &str) {
+    let status =
+        Command::new("git")
+            .arg("checkout")
+            .arg(branch)
+            .status()
+            .expect("failed to find 'git' command");
+    assert!(status.success())
+}
+
+pub fn current_branch() -> String {
+    let output =
+        Command::new("git")
+            .arg("branch")
+            .arg("--show-current")
+            .output()
+            .expect("failed to find 'git' command")
+            .stdout;
+    str::from_utf8(&output)
+        .expect("failed to find current branch")
+        .trim()
+        .to_string()
+}
+
+pub fn diff(from: &str, into: &str) -> String {
+    todo!()
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::mygit::{current_branch, MyGit};
+
+    #[test]
+    fn it_works() {
+        let branch = MyGit::new().current_branch_name();
+        assert_eq!(branch, "main")
+    }
+}