Refactor git shit to its own module

2025-03-27 22:20:22 +01:00
parent f008efd5e1
commit 89eed3f847
3 changed files with 132 additions and 118 deletions
--- a/utils/git.go
+++ b/utils/git.go
@@ -0,0 +1,97 @@
+package utils
+
+import (
+	"fmt"
+	"modify/logger"
+	"os"
+	"path/filepath"
+	"time"
+
+	"github.com/go-git/go-git/v5/plumbing/object"
+	"github.com/go-git/go-git/v5"
+)
+
+var (
+	Repo     *git.Repository
+	Worktree *git.Worktree
+)
+
+func SetupGit() error {
+	cwd, err := os.Getwd()
+	if err != nil {
+		return fmt.Errorf("failed to get current working directory: %w", err)
+	}
+	logger.Debug("Current working directory obtained: %s", cwd)
+
+	logger.Debug("Attempting to open git repository at %s", cwd)
+	Repo, err = git.PlainOpen(cwd)
+	if err != nil {
+		logger.Debug("No existing git repository found at %s, attempting to initialize a new git repository.", cwd)
+		Repo, err = git.PlainInit(cwd, false)
+		if err != nil {
+			return fmt.Errorf("failed to initialize a new git repository at %s: %w", cwd, err)
+		}
+		logger.Info("Successfully initialized a new git repository at %s", cwd)
+	} else {
+		logger.Info("Successfully opened existing git repository at %s", cwd)
+	}
+
+	logger.Debug("Attempting to obtain worktree for repository at %s", cwd)
+	Worktree, err = Repo.Worktree()
+	if err != nil {
+		return fmt.Errorf("failed to obtain worktree for repository at %s: %w", cwd, err)
+	}
+	logger.Debug("Successfully obtained worktree for repository at %s", cwd)
+	return nil
+}
+
+func CleanupGitFiles(files []string) error {
+	for _, file := range files {
+		logger.Debug("Checking git status for file: %s", file)
+		status, err := Worktree.Status()
+		if err != nil {
+			logger.Error("Error getting worktree status: %v", err)
+			fmt.Fprintf(os.Stderr, "Error getting worktree status: %v\n", err)
+			return fmt.Errorf("error getting worktree status: %w", err)
+		}
+		if status.IsUntracked(file) {
+			logger.Info("Detected untracked file: %s. Adding to git index.", file)
+			_, err = Worktree.Add(file)
+			if err != nil {
+				logger.Error("Error adding file to git: %v", err)
+				fmt.Fprintf(os.Stderr, "Error adding file to git: %v\n", err)
+				return fmt.Errorf("error adding file to git: %w", err)
+			}
+
+			filename := filepath.Base(file)
+			logger.Info("File %s added successfully. Committing with message: 'Track %s'", filename, filename)
+			_, err = Worktree.Commit("Track "+filename, &git.CommitOptions{
+				Author: &object.Signature{
+					Name:  "Big Chef",
+					Email: "bigchef@bigchef.com",
+					When:  time.Now(),
+				},
+			})
+			if err != nil {
+				logger.Error("Error committing file: %v", err)
+				fmt.Fprintf(os.Stderr, "Error committing file: %v\n", err)
+				return fmt.Errorf("error committing file: %w", err)
+			}
+			logger.Info("Successfully committed file: %s", filename)
+		} else {
+			logger.Info("File %s is already tracked. Restoring it to the working tree.", file)
+			err := Worktree.Restore(&git.RestoreOptions{
+				Files:    []string{file},
+				Staged:   true,
+				Worktree: true,
+			})
+			if err != nil {
+				logger.Error("Error restoring file: %v", err)
+				fmt.Fprintf(os.Stderr, "Error restoring file: %v\n", err)
+				return fmt.Errorf("error restoring file: %w", err)
+			}
+			logger.Info("File %s restored successfully", file)
+		}
+	}
+	return nil
+}