mirror of https://git.jolheiser.com/ugit.git
104 lines
1.9 KiB
Go
104 lines
1.9 KiB
Go
package git
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/go-git/go-git/v5"
|
|
"github.com/go-git/go-git/v5/plumbing"
|
|
"github.com/go-git/go-git/v5/plumbing/object"
|
|
)
|
|
|
|
type Repo struct {
|
|
path string
|
|
Meta RepoMeta
|
|
}
|
|
|
|
func (r Repo) Name() string {
|
|
return strings.TrimSuffix(filepath.Base(r.path), ".git")
|
|
}
|
|
|
|
func NewRepo(dir, name string) (*Repo, error) {
|
|
if !strings.HasSuffix(name, ".git") {
|
|
name += ".git"
|
|
}
|
|
r := &Repo{
|
|
path: filepath.Join(dir, name),
|
|
}
|
|
|
|
_, err := os.Stat(r.path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := ensureJSONFile(r.metaPath()); err != nil {
|
|
return nil, err
|
|
}
|
|
fi, err := os.Open(r.metaPath())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer fi.Close()
|
|
|
|
if err := json.NewDecoder(fi).Decode(&r.Meta); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return r, nil
|
|
}
|
|
|
|
// DefaultBranch returns the branch referenced by HEAD, setting it if needed
|
|
func (r Repo) DefaultBranch() (string, error) {
|
|
repo, err := r.Git()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
ref, err := repo.Head()
|
|
if err != nil {
|
|
if !errors.Is(err, plumbing.ErrReferenceNotFound) {
|
|
return "", err
|
|
}
|
|
brs, err := repo.Branches()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer brs.Close()
|
|
fb, err := brs.Next()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
// Rename the default branch to the first branch available
|
|
ref = fb
|
|
sym := plumbing.NewSymbolicReference(plumbing.HEAD, fb.Name())
|
|
if err := repo.Storer.SetReference(sym); err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
|
|
return strings.TrimPrefix(ref.Name().String(), "refs/heads/"), nil
|
|
}
|
|
|
|
// Git allows access to the git repository
|
|
func (r Repo) Git() (*git.Repository, error) {
|
|
return git.PlainOpen(r.path)
|
|
}
|
|
|
|
// LastCommit returns the last commit of the repo
|
|
func (r Repo) LastCommit() (*object.Commit, error) {
|
|
repo, err := r.Git()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
head, err := repo.Head()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return repo.CommitObject(head.Hash())
|
|
}
|