ugit/internal/git/git.go

package git

import (
	"errors"
	"io/fs"
	"os"
	"path/filepath"
	"sort"

	"github.com/dustin/go-humanize"
	"github.com/go-git/go-git/v5"
	"github.com/go-git/go-git/v5/plumbing"
	"github.com/go-git/go-git/v5/plumbing/object"
)

// EnsureRepo ensures that the repo exists in the given directory
func EnsureRepo(dir string, repo string) error {
	exists, err := PathExists(dir)
	if err != nil {
		return err
	}
	if !exists {
		err = os.MkdirAll(dir, os.ModeDir|os.FileMode(0o700))
		if err != nil {
			return err
		}
	}
	rp := filepath.Join(dir, repo)
	exists, err = PathExists(rp)
	if err != nil {
		return err
	}
	if !exists {
		_, err := git.PlainInit(rp, true)
		if err != nil {
			return err
		}
	}
	return nil
}

// PathExists checks if a path exists and returns true if it does
func PathExists(path string) (bool, error) {
	_, err := os.Stat(path)
	if err == nil {
		return true, nil
	}
	if errors.Is(err, fs.ErrNotExist) {
		return false, nil
	}
	return true, err
}

// Tree returns the git tree at a given ref/rev
func (r Repo) Tree(ref string) (*object.Tree, error) {
	g, err := r.Git()
	if err != nil {
		return nil, err
	}

	hash, err := g.ResolveRevision(plumbing.Revision(ref))
	if err != nil {
		return nil, err
	}

	c, err := g.CommitObject(*hash)
	if err != nil {
		return nil, err
	}

	return c.Tree()
}

// FileInfo is the information for a file in a tree
type FileInfo struct {
	Path  string
	IsDir bool
	Mode  string
	Size  string
}

// Name returns the last part of the FileInfo.Path
func (f FileInfo) Name() string {
	return filepath.Base(f.Path)
}

// Dir returns the given dirpath in the given ref as a slice of FileInfo
// Sorted alphabetically, dirs first
func (r Repo) Dir(ref, path string) ([]FileInfo, error) {
	t, err := r.Tree(ref)
	if err != nil {
		return nil, err
	}
	if path != "" {
		t, err = t.Tree(path)
		if err != nil {
			return nil, err
		}
	}

	fis := make([]FileInfo, 0)
	for _, entry := range t.Entries {
		fm, err := entry.Mode.ToOSFileMode()
		if err != nil {
			return nil, err
		}
		size, err := t.Size(entry.Name)
		if err != nil {
			return nil, err
		}
		fis = append(fis, FileInfo{
			Path:  filepath.Join(path, entry.Name),
			IsDir: fm.IsDir(),
			Mode:  fm.String(),
			Size:  humanize.Bytes(uint64(size)),
		})
	}
	sort.Slice(fis, func(i, j int) bool {
		fi1 := fis[i]
		fi2 := fis[j]
		return (fi1.IsDir && !fi2.IsDir) || fi1.Name() < fi2.Name()
	})

	return fis, nil
}

// FileContent returns the content of a file in the git tree at a given ref/rev
func (r Repo) FileContent(ref, file string) (string, error) {
	t, err := r.Tree(ref)
	if err != nil {
		return "", err
	}

	f, err := t.File(file)
	if err != nil {
		return "", err
	}

	content, err := f.Contents()
	if err != nil {
		return "", err
	}

	return content, nil
}