Add Checkout to Commit

src-tauri/src/git/git_branch.rs

@@ -1,5 +1,14 @@
+use std::{fs::OpenOptions, io::Write, path::PathBuf};
+
 use serde::{Deserialize, Serialize};
 
+use crate::errors::git_object_error::{CommitError, GitObjectError};
+
+use super::{
+    git_commit::GitCommit, git_files::GitFilesRequired, git_folders::GIT_FOLDER,
+    git_project::GitProject, object::GitObject,
+};
+
 #[derive(Debug, PartialEq, Serialize, Deserialize, Clone)]
 #[serde(rename_all = "camelCase")]
 pub struct GitBranch {
@@ -11,6 +20,22 @@
     pub fn new(name: String, commit: String) -> GitBranch {
         GitBranch { name, commit }
     }
+
+    pub fn checkout(&self, project: &GitProject) -> Result<(), GitObjectError> {
+        GitCommit::from_hash(project, &self.commit)?.checkout(project)?;
+
+        let head_path = PathBuf::from(project.get_directory())
+            .join(GIT_FOLDER)
+            .join(GitFilesRequired::HEAD.as_ref());
+        OpenOptions::new()
+            .write(true)
+            .open(head_path)
+            .map_err(|_| GitObjectError::InvalidCommitFile(CommitError::InvalidContent))?
+            .write_all(self.name.as_bytes())
+            .map_err(|_| GitObjectError::InvalidHash)?;
+
+        Ok(())
+    }
 }
 
 #[cfg(test)]

src-tauri/src/git/git_commit.rs

@@ -1,11 +1,15 @@
 use super::{
     git_commit_author::{GitCommitAuthor, GitCommitAuthorType},
+    git_files::GitFilesRequired,
+    git_folders::GIT_FOLDER,
     git_project::GitProject,
+    git_tree::GitTree,
     object::{GitObject, Header},
 };
 use crate::errors::git_object_error::{CommitError, GitObjectError};
 use core::fmt;
 use serde::{Deserialize, Serialize};
+use std::{fs::OpenOptions, io::Write, path::PathBuf};
 
 pub enum CommitPrefix {
     Tree,
@@ -149,6 +153,39 @@
 
         Ok(history)
     }
+
+    /**
+     * Checkout all the files in a commit
+     */
+    pub fn checkout(&self, project: &GitProject) -> Result<(), GitObjectError> {
+        let files =
+            GitTree::from_hash(project, self.get_tree_hash())?.get_object_blobs(project, None);
+
+        files.iter().for_each(|file| {
+            let path = PathBuf::from(project.get_directory()).join(&file.0);
+            let file_in_fs = OpenOptions::new()
+                .write(true)
+                .truncate(true)
+                .create(true)
+                .open(path);
+
+            if let Ok(mut file_in_fs) = file_in_fs {
+                let _ = file_in_fs.write_all(file.1.data());
+            }
+        });
+
+        let head_path = PathBuf::from(project.get_directory())
+            .join(GIT_FOLDER)
+            .join(GitFilesRequired::HEAD.as_ref());
+        OpenOptions::new()
+            .write(true)
+            .open(head_path)
+            .map_err(|_| GitObjectError::InvalidCommitFile(CommitError::InvalidContent))?
+            .write_all(self.get_hash().as_bytes())
+            .map_err(|_| GitObjectError::InvalidHash)?;
+
+        Ok(())
+    }
 }
 
 impl GitObject for GitCommit {

src-tauri/src/git/project_folder.rs

@@ -1,4 +1,5 @@
 use super::{
+    git_branch::GitBranch,
     git_commit::{GitCommit, GitCommitWithHash},
     git_folders::GitBranchType,
     git_project::GitProject,
@@ -83,6 +84,22 @@
         .map_err(|_| GitError::InvalidHistory)
 }
 
+#[tauri::command]
+pub fn checkout_branch(project: GitProject, branch: GitBranch) -> Result<(), GitError> {
+    branch
+        .checkout(&project)
+        .map_err(|_| GitError::NoLocalBranches)
+}
+
+#[tauri::command]
+pub fn checkout_commit(project: GitProject, hash: &str) -> Result<(), GitError> {
+    let commit = GitCommit::from_hash(&project, hash).map_err(|_| GitError::InvalidHistory)?;
+
+    commit
+        .checkout(&project)
+        .map_err(|_| GitError::InvalidHistory)
+}
+
 #[cfg(test)]
 mod tests {
     use std::{