willow/git/git.go

189 lines
4.0 KiB
Go
Raw Normal View History

2023-09-24 20:57:56 +00:00
// SPDX-FileCopyrightText: Amolith <amolith@secluded.site>
//
// SPDX-License-Identifier: Apache-2.0
package git
2023-09-21 18:03:21 +00:00
import (
"errors"
"fmt"
"net/url"
2023-09-21 18:03:21 +00:00
"os"
"strings"
"time"
"github.com/microcosm-cc/bluemonday"
2023-09-21 18:03:21 +00:00
"github.com/go-git/go-git/v5"
"github.com/go-git/go-git/v5/plumbing"
"github.com/go-git/go-git/v5/plumbing/transport"
)
type Release struct {
Tag string
Content string
URL string
Date time.Time
}
var (
bmUGC = bluemonday.UGCPolicy()
bmStrict = bluemonday.StrictPolicy()
)
2023-09-21 18:03:21 +00:00
// listRemoteTags lists all tags in a remote repository, whether HTTP(S) or SSH.
2023-09-24 21:02:00 +00:00
// func listRemoteTags(url string) (tags []string, err error) {
// // TODO: Implement listRemoteTags
// // https://pkg.go.dev/github.com/go-git/go-git/v5@v5.8.0#NewRemote
// return nil, nil
// }
2023-09-21 18:03:21 +00:00
// GetReleases fetches all releases in a remote repository, whether HTTP(S) or
// SSH.
func GetReleases(gitURI, forge string) ([]Release, error) {
r, err := minimalClone(gitURI)
2023-09-21 18:03:21 +00:00
if err != nil {
return nil, err
2023-09-21 18:03:21 +00:00
}
tagRefs, err := r.Tags()
if err != nil {
return nil, err
2023-09-21 18:03:21 +00:00
}
parsedURI, err := url.Parse(gitURI)
if err != nil {
fmt.Println("Error parsing URI: " + err.Error())
}
var httpURI string
if parsedURI.Scheme != "" {
httpURI = parsedURI.Host + parsedURI.Path
}
releases := make([]Release, 0)
2023-09-21 18:03:21 +00:00
err = tagRefs.ForEach(func(tagRef *plumbing.Reference) error {
tagObj, err := r.TagObject(tagRef.Hash())
var message string
var date time.Time
if errors.Is(err, plumbing.ErrObjectNotFound) {
commitTag, err := r.CommitObject(tagRef.Hash())
if err != nil {
return err
2023-09-21 18:03:21 +00:00
}
message = commitTag.Message
date = commitTag.Committer.When
} else {
message = tagObj.Message
date = tagObj.Tagger.When
}
tagURL := ""
tagName := bmStrict.Sanitize(tagRef.Name().Short())
switch forge {
case "sourcehut":
tagURL = "https://" + httpURI + "/refs/" + tagName
case "gitlab":
tagURL = "https://" + httpURI + "/-/releases/" + tagName
2023-09-21 18:03:21 +00:00
default:
tagURL = ""
2023-09-21 18:03:21 +00:00
}
releases = append(releases, Release{
Tag: tagName,
Content: bmUGC.Sanitize(message),
URL: tagURL,
Date: date,
})
2023-09-21 18:03:21 +00:00
return nil
})
if err != nil {
return nil, err
2023-09-21 18:03:21 +00:00
}
return releases, nil
2023-09-21 18:03:21 +00:00
}
// minimalClone clones a repository with a depth of 1 and no checkout.
func minimalClone(url string) (r *git.Repository, err error) {
path, err := stringifyRepo(url)
if err != nil {
return nil, err
}
if _, err := os.Stat(path); err == nil {
r, err := git.PlainOpen(path)
if err != nil {
return nil, err
}
err = r.Fetch(&git.FetchOptions{
RemoteName: "origin",
Depth: 1,
Tags: git.AllTags,
})
if errors.Is(err, git.NoErrAlreadyUpToDate) {
2023-09-21 18:03:21 +00:00
return r, nil
}
return r, err
}
r, err = git.PlainClone(path, false, &git.CloneOptions{
URL: url,
SingleBranch: true,
NoCheckout: true,
Depth: 1,
})
return r, err
}
// RemoveRepo removes a repository from the local filesystem.
func RemoveRepo(url string) (err error) {
2023-09-21 18:03:21 +00:00
path, err := stringifyRepo(url)
if err != nil {
return err
}
err = os.RemoveAll(path)
if err != nil {
return err
}
2024-02-29 20:51:42 +00:00
path = path[:strings.LastIndex(path, "/")]
dirs := strings.Split(path, "/")
for range dirs {
if path == "data" {
break
}
2024-02-29 20:51:42 +00:00
err = os.Remove(path)
if err != nil {
2024-02-29 20:51:42 +00:00
// This folder likely has data, so might as well save some time by
// not checking the parents we can't delete anyway.
break
}
path = path[:strings.LastIndex(path, "/")]
}
2024-02-29 20:51:42 +00:00
return nil
}
2023-09-21 18:03:21 +00:00
// stringifyRepo accepts a repository URI string and the corresponding local
// filesystem path, whether the URI is HTTP, HTTPS, or SSH.
func stringifyRepo(url string) (path string, err error) {
2024-02-23 20:51:24 +00:00
url = strings.TrimSuffix(url, ".git")
url = strings.TrimSuffix(url, "/")
2023-09-21 18:03:21 +00:00
ep, err := transport.NewEndpoint(url)
if err != nil {
return "", err
}
if ep.Protocol == "http" || ep.Protocol == "https" {
return "data/" + strings.Split(url, "://")[1], nil
} else if ep.Protocol == "ssh" {
2024-02-23 21:13:18 +00:00
return "data/" + ep.Host + "/" + ep.Path, nil
2023-09-21 18:03:21 +00:00
} else {
return "", errors.New("unsupported protocol")
}
}