add commands

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 @@ impl GitBranch {
     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/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 @@ pub fn get_commit_history(
         .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::{