certman/server/git.go

package server

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"
	"time"

	"git.nevets.tech/Steven/certman/common"
	"github.com/go-git/go-git/v5"
	gitconf "github.com/go-git/go-git/v5/config"
	"github.com/go-git/go-git/v5/plumbing/object"
	"github.com/go-git/go-git/v5/plumbing/transport/http"
)

type GitWorkspace common.GitWorkspace

func (ws *GitWorkspace) AddAndPushCerts(dataRoot, repoSuffix string, config *common.AppConfig) error {
	certFiles, err := os.ReadDir(dataRoot)
	if err != nil {
		fmt.Printf("Error reading from directory: %v\n", err)
		return err
	}
	for _, entry := range certFiles {
		if strings.HasSuffix(entry.Name(), ".crpt") {
			file, err := ws.FS.Create(entry.Name())
			if err != nil {
				fmt.Printf("Error copying file to memfs: %v\n", err)
				return err
			}
			certFile, err := os.ReadFile(filepath.Join(dataRoot, entry.Name()))
			if err != nil {
				fmt.Printf("Error reading file to memfs: %v\n", err)
				file.Close()
				return err
			}
			_, err = file.Write(certFile)
			if err != nil {
				fmt.Printf("Error writing to memfs: %v\n", err)
				file.Close()
				return err
			}
			_, err = ws.WorkTree.Add(file.Name())
			if err != nil {
				fmt.Printf("Error adding file %v: %v\n", file.Name(), err)
				file.Close()
				return err
			}
			err = file.Close()
			if err != nil {
				fmt.Printf("Error closing file: %v\n", err)
			}
		}
	}

	file, err := ws.FS.Create("/SERVER_ID")
	if err != nil {
		fmt.Printf("Error creating file in memfs: %v\n", err)
		return err
	}
	_, err = file.Write([]byte(config.App.UUID))
	if err != nil {
		fmt.Printf("Error writing to memfs: %v\n", err)
		file.Close()
		return err
	}
	_, err = ws.WorkTree.Add(file.Name())
	if err != nil {
		fmt.Printf("Error adding file %v: %v\n", file.Name(), err)
		file.Close()
		return err
	}
	err = file.Close()
	if err != nil {
		fmt.Printf("Error closing file: %v\n", err)
	}

	status, err := ws.WorkTree.Status()
	if err != nil {
		fmt.Printf("Error getting repo status: %v\n", err)
		return err
	}
	if status.IsClean() {
		fmt.Printf("Repository is clean, skipping commit...\n")
		return nil
	}

	fmt.Println("Work Tree Status:\n" + status.String())
	signature := &object.Signature{
		Name:  "Cert Manager",
		Email: config.Certificates.Email,
		When:  time.Now(),
	}
	_, err = ws.WorkTree.Commit("Update "+ws.Domain+" @ "+time.Now().Format("Mon Jan _2 2006 15:04:05 MST"), &git.CommitOptions{Author: signature, Committer: signature})
	if err != nil {
		fmt.Printf("Error committing certs: %v\n", err)
		return err
	}
	creds := &http.BasicAuth{
		Username: config.Git.Username,
		Password: config.Git.APIToken,
	}
	err = ws.Repo.Push(&git.PushOptions{
		Auth:       creds,
		Force:      true,
		RemoteName: "origin",
		RefSpecs: []gitconf.RefSpec{
			"refs/heads/master:refs/heads/master",
		},
	})
	if err != nil {
		fmt.Printf("Error pushing to origin: %v\n", err)
		return err
	}

	fmt.Println("Successfully uploaded to " + config.Git.Server + "/" + config.Git.OrgName + "/" + ws.Domain + repoSuffix + ".git")

	return nil
}