mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
9a07b4d089
2978 files. The fork is 2026-07-15 (rebrand: gitea.dev -> github.com/hanzoai/git), so 2026 is the year, and "Copyright 2026 Hanzo AI, Inc." was already the form used by the handful of files that had ours. Shape is now: // Copyright 2026 Hanzo AI, Inc. All rights reserved. // Copyright 2015 The Gogs Authors. All rights reserved. // Copyright 2019 The Gitea Authors. All rights reserved. // SPDX-License-Identifier: MIT Ours on top, theirs retained verbatim. Retained because MIT requires it, and separately because they are true — those authors did write that code. Not renamed to "The Git Authors": Git is a different project under a different license, and putting its name here would attribute this code to people who have nothing to do with it. Adding our line is the fix; overwriting theirs would have been both a licence violation and a false statement. gofmt still reports import-order drift in some of these files. It predates this change (verified against a stash) and is left alone rather than burying a one-line-per-file legal change under 3000 files of unrelated reordering. Build clean. Co-authored-by: Hanzo Dev <[email protected]>
96 lines
3.2 KiB
Go
96 lines
3.2 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// Copyright 2020 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package web
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"image/png"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
|
|
"github.com/hanzoai/git/modules/assetfs"
|
|
"github.com/hanzoai/git/modules/avatar"
|
|
"github.com/hanzoai/git/modules/httpcache"
|
|
"github.com/hanzoai/git/modules/log"
|
|
"github.com/hanzoai/git/modules/setting"
|
|
"github.com/hanzoai/git/modules/storage"
|
|
"github.com/hanzoai/git/modules/util"
|
|
"github.com/hanzoai/git/modules/web/routing"
|
|
)
|
|
|
|
func avatarStorageHandler(storageSetting *setting.Storage, prefix string, objStore storage.ObjectStorage) http.HandlerFunc {
|
|
prefix = strings.Trim(prefix, "/")
|
|
funcInfo := routing.GetFuncInfo(avatarStorageHandler, prefix)
|
|
exeModTime := assetfs.GetExecutableModTime()
|
|
fallbackEtag := fmt.Sprintf(`"avatar-%s"`, exeModTime.Format("20060102150405"))
|
|
|
|
handleError := func(w http.ResponseWriter, req *http.Request, avatarPath string, err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
|
|
if errors.Is(err, os.ErrNotExist) || errors.Is(err, util.ErrNotExist) {
|
|
// if avatar doesn't exist, generate a random one and serve it with proper cache control headers
|
|
w.Header().Set("Content-Type", "image/png")
|
|
if !httpcache.HandleGenericETagPublicCache(req, w, fallbackEtag, &exeModTime) {
|
|
if req.Method == http.MethodGet {
|
|
img := avatar.RandomImageWithSize(96, []byte(avatarPath))
|
|
_ = png.Encode(w, img)
|
|
} // else: for HEAD request, just return the headers without body
|
|
}
|
|
} else {
|
|
// for internal errors, log the error and return 500
|
|
log.Error("Error when serving avatar %s: %s", req.URL.Path, err)
|
|
http.Error(w, "unable to serve avatar image", http.StatusInternalServerError)
|
|
}
|
|
return true
|
|
}
|
|
|
|
return func(w http.ResponseWriter, req *http.Request) {
|
|
defer routing.RecordFuncInfo(req.Context(), funcInfo)()
|
|
|
|
avatarPath, ok := strings.CutPrefix(req.URL.Path, "/"+prefix+"/")
|
|
if !ok {
|
|
http.Error(w, "invalid avatar path", http.StatusBadRequest)
|
|
return
|
|
}
|
|
avatarPath = util.PathJoinRelX(avatarPath)
|
|
if avatarPath == "" || avatarPath == "." {
|
|
http.Error(w, "not found", http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
if storageSetting.ServeDirect() {
|
|
// Old logic: no check for existence by Stat, so old code's "errors.Is(err, os.ErrNotExist)" didn't work.
|
|
// So in theory, it doesn't work with the non-existing avatar fallback, it just gets the URL and redirects to it.
|
|
// Checking "stat" requires one more request to the storage, which is inefficient.
|
|
// Workaround: disable "SERVE_DIRECT". Leave the problem to the future.
|
|
u, err := objStore.ServeDirectURL(avatarPath, path.Base(avatarPath), req.Method, nil)
|
|
if handleError(w, req, avatarPath, err) {
|
|
return
|
|
}
|
|
http.Redirect(w, req, u.String(), http.StatusTemporaryRedirect)
|
|
return
|
|
}
|
|
|
|
fr, err := objStore.Open(avatarPath)
|
|
if handleError(w, req, avatarPath, err) {
|
|
return
|
|
}
|
|
defer fr.Close()
|
|
|
|
fi, err := fr.Stat()
|
|
if handleError(w, req, avatarPath, err) {
|
|
return
|
|
}
|
|
|
|
httpcache.SetCacheControlInHeader(w.Header(), httpcache.CacheControlForPublicStatic())
|
|
http.ServeContent(w, req, path.Base(avatarPath), fi.ModTime(), fr)
|
|
}
|
|
}
|