2021-12-05 17:17:28 +00:00
|
|
|
package server
|
|
|
|
|
|
|
|
import (
|
2022-07-27 15:25:08 +00:00
|
|
|
"net/http"
|
2021-12-05 17:17:28 +00:00
|
|
|
"strings"
|
|
|
|
|
2022-11-12 13:10:04 +00:00
|
|
|
"github.com/rs/zerolog"
|
|
|
|
|
2021-12-05 17:17:28 +00:00
|
|
|
"codeberg.org/codeberg/pages/html"
|
|
|
|
"codeberg.org/codeberg/pages/server/cache"
|
2022-07-27 15:25:08 +00:00
|
|
|
"codeberg.org/codeberg/pages/server/context"
|
2022-06-11 21:02:06 +00:00
|
|
|
"codeberg.org/codeberg/pages/server/gitea"
|
2021-12-05 17:17:28 +00:00
|
|
|
"codeberg.org/codeberg/pages/server/upstream"
|
|
|
|
)
|
|
|
|
|
|
|
|
// tryUpstream forwards the target request to the Gitea API, and shows an error page on failure.
|
2022-07-27 15:25:08 +00:00
|
|
|
func tryUpstream(ctx *context.Context, giteaClient *gitea.Client,
|
2022-08-28 14:21:37 +00:00
|
|
|
mainDomainSuffix, trimmedHost string,
|
2021-12-05 17:17:28 +00:00
|
|
|
|
|
|
|
targetOptions *upstream.Options,
|
2022-06-11 21:02:06 +00:00
|
|
|
targetOwner, targetRepo, targetBranch, targetPath string,
|
2021-12-05 17:17:28 +00:00
|
|
|
|
2022-07-27 13:39:46 +00:00
|
|
|
canonicalDomainCache cache.SetGetKey,
|
2022-03-27 19:54:06 +00:00
|
|
|
) {
|
2021-12-05 17:17:28 +00:00
|
|
|
// check if a canonical domain exists on a request on MainDomain
|
2022-08-28 14:21:37 +00:00
|
|
|
if strings.HasSuffix(trimmedHost, mainDomainSuffix) {
|
2022-06-11 21:02:06 +00:00
|
|
|
canonicalDomain, _ := upstream.CheckCanonicalDomain(giteaClient, targetOwner, targetRepo, targetBranch, "", string(mainDomainSuffix), canonicalDomainCache)
|
2021-12-05 17:17:28 +00:00
|
|
|
if !strings.HasSuffix(strings.SplitN(canonicalDomain, "/", 2)[0], string(mainDomainSuffix)) {
|
2022-07-27 15:25:08 +00:00
|
|
|
canonicalPath := ctx.Req.RequestURI
|
2021-12-05 17:17:28 +00:00
|
|
|
if targetRepo != "pages" {
|
2021-12-09 18:32:30 +00:00
|
|
|
path := strings.SplitN(canonicalPath, "/", 3)
|
|
|
|
if len(path) >= 3 {
|
2021-12-09 19:16:43 +00:00
|
|
|
canonicalPath = "/" + path[2]
|
2021-12-09 18:32:30 +00:00
|
|
|
}
|
2021-12-05 17:17:28 +00:00
|
|
|
}
|
2022-07-27 15:25:08 +00:00
|
|
|
ctx.Redirect("https://"+canonicalDomain+canonicalPath, http.StatusTemporaryRedirect)
|
2021-12-05 17:17:28 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-12-05 18:53:23 +00:00
|
|
|
targetOptions.TargetOwner = targetOwner
|
|
|
|
targetOptions.TargetRepo = targetRepo
|
|
|
|
targetOptions.TargetBranch = targetBranch
|
|
|
|
targetOptions.TargetPath = targetPath
|
2022-08-12 03:06:26 +00:00
|
|
|
targetOptions.Host = string(trimmedHost)
|
2021-12-05 18:53:23 +00:00
|
|
|
|
2021-12-05 17:17:28 +00:00
|
|
|
// Try to request the file from the Gitea API
|
2022-07-27 13:48:48 +00:00
|
|
|
if !targetOptions.Upstream(ctx, giteaClient) {
|
2022-11-11 05:38:09 +00:00
|
|
|
html.ReturnErrorPage(ctx, "", ctx.StatusCode)
|
2021-12-05 17:17:28 +00:00
|
|
|
}
|
|
|
|
}
|
2022-11-11 18:03:17 +00:00
|
|
|
|
2022-11-12 01:54:56 +00:00
|
|
|
// tryBranch checks if a branch exists and populates the target variables. If canonicalLink is non-empty,
|
|
|
|
// it will also disallow search indexing and add a Link header to the canonical URL.
|
|
|
|
func tryBranch(log zerolog.Logger, ctx *context.Context, giteaClient *gitea.Client,
|
|
|
|
repoOwner, repoName, branch, path string, canonicalLink bool,
|
|
|
|
) (*gitea.BranchTimestamp, bool) {
|
2022-11-11 18:03:17 +00:00
|
|
|
if repoName == "" {
|
2022-11-12 01:54:56 +00:00
|
|
|
log.Debug().Msg("tryBranch: repo is empty")
|
|
|
|
return nil, false
|
2022-11-11 18:03:17 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Replace "~" to "/" so we can access branch that contains slash character
|
|
|
|
// Branch name cannot contain "~" so doing this is okay
|
|
|
|
branch = strings.ReplaceAll(branch, "~", "/")
|
|
|
|
|
|
|
|
// Check if the branch exists, otherwise treat it as a file path
|
|
|
|
branchTimestampResult := upstream.GetBranchTimestamp(giteaClient, repoOwner, repoName, branch)
|
|
|
|
if branchTimestampResult == nil {
|
|
|
|
log.Debug().Msg("tryBranch: branch doesn't exist")
|
2022-11-12 01:54:56 +00:00
|
|
|
return nil, false
|
2022-11-11 18:03:17 +00:00
|
|
|
}
|
|
|
|
|
2022-11-12 01:54:56 +00:00
|
|
|
if canonicalLink {
|
2022-11-11 18:03:17 +00:00
|
|
|
// Hide from search machines & add canonical link
|
|
|
|
ctx.RespWriter.Header().Set("X-Robots-Tag", "noarchive, noindex")
|
|
|
|
ctx.RespWriter.Header().Set("Link",
|
2022-11-12 01:54:56 +00:00
|
|
|
giteaClient.ContentWebLink(repoOwner, repoName, branchTimestampResult.Branch, path)+
|
|
|
|
"; rel=\"canonical\"")
|
2022-11-11 18:03:17 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
log.Debug().Msg("tryBranch: true")
|
2022-11-12 01:54:56 +00:00
|
|
|
return branchTimestampResult, true
|
2022-11-11 18:03:17 +00:00
|
|
|
}
|