mirror of
https://codeberg.org/Codeberg/pages-server.git
synced 2025-04-24 22:06:57 +00:00
88 lines
2.6 KiB
Go
88 lines
2.6 KiB
Go
package upstream
|
|
|
|
import (
|
|
"fmt"
|
|
"path"
|
|
"time"
|
|
|
|
"github.com/valyala/fasthttp"
|
|
"github.com/valyala/fastjson"
|
|
|
|
"codeberg.org/codeberg/pages/server/cache"
|
|
)
|
|
|
|
type branchTimestamp struct {
|
|
Branch string
|
|
Timestamp time.Time
|
|
}
|
|
|
|
// GetBranchTimestamp finds the default branch (if branch is "") and returns the last modification time of the branch
|
|
// (or nil if the branch doesn't exist)
|
|
func GetBranchTimestamp(owner, repo, branch, giteaRoot, giteaAPIToken string, branchTimestampCache cache.SetGetKey) *branchTimestamp {
|
|
if result, ok := branchTimestampCache.Get(owner + "/" + repo + "/" + branch); ok {
|
|
if result == nil {
|
|
return nil
|
|
}
|
|
return result.(*branchTimestamp)
|
|
}
|
|
result := &branchTimestamp{
|
|
Branch: branch,
|
|
}
|
|
if len(branch) == 0 {
|
|
// Get default branch
|
|
defaultBranch, err := giteaGetRepoDefaultBranch(giteaRoot, owner, repo, giteaAPIToken)
|
|
if err != nil {
|
|
_ = branchTimestampCache.Set(owner+"/"+repo+"/", nil, defaultBranchCacheTimeout)
|
|
return nil
|
|
}
|
|
result.Branch = defaultBranch
|
|
}
|
|
|
|
timestamp, err := giteaGetRepoBranchTimestamp(giteaRoot, owner, repo, branch, giteaAPIToken)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
result.Timestamp = timestamp
|
|
_ = branchTimestampCache.Set(owner+"/"+repo+"/"+branch, result, branchExistenceCacheTimeout)
|
|
return result
|
|
}
|
|
|
|
func giteaGetRepoBranchTimestamp(giteaRoot, repoOwner, repoName, branchName, giteaAPIToken string) (time.Time, error) {
|
|
client := getFastHTTPClient(5 * time.Second)
|
|
|
|
req := fasthttp.AcquireRequest()
|
|
req.SetRequestURI(path.Join(giteaRoot, giteaAPIRepos, repoOwner, repoName, "branches", branchName))
|
|
req.Header.Set(fasthttp.HeaderAuthorization, giteaAPIToken)
|
|
res := fasthttp.AcquireResponse()
|
|
|
|
if err := client.Do(req, res); err != nil {
|
|
return time.Time{}, err
|
|
}
|
|
if res.StatusCode() != fasthttp.StatusOK {
|
|
return time.Time{}, fmt.Errorf("unexpected status code '%d'", res.StatusCode())
|
|
}
|
|
return time.Parse(time.RFC3339, fastjson.GetString(res.Body(), "commit", "timestamp"))
|
|
}
|
|
|
|
func giteaGetRepoDefaultBranch(giteaRoot, repoOwner, repoName, giteaAPIToken string) (string, error) {
|
|
client := getFastHTTPClient(5 * time.Second)
|
|
|
|
req := fasthttp.AcquireRequest()
|
|
req.SetRequestURI(path.Join(giteaRoot, giteaAPIRepos, repoOwner, repoName))
|
|
req.Header.Set(fasthttp.HeaderAuthorization, giteaAPIToken)
|
|
res := fasthttp.AcquireResponse()
|
|
|
|
if err := client.Do(req, res); err != nil {
|
|
return "", err
|
|
}
|
|
if res.StatusCode() != fasthttp.StatusOK {
|
|
return "", fmt.Errorf("unexpected status code '%d'", res.StatusCode())
|
|
}
|
|
return fastjson.GetString(res.Body(), "default_branch"), nil
|
|
}
|
|
|
|
type fileResponse struct {
|
|
exists bool
|
|
mimeType string
|
|
body []byte
|
|
}
|