123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899 |
- // Copyright 2020 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package web
-
- import (
- "errors"
- "fmt"
- "net/http"
- "os"
- "path"
- "strings"
-
- "code.gitea.io/gitea/modules/httpcache"
- "code.gitea.io/gitea/modules/log"
- "code.gitea.io/gitea/modules/setting"
- "code.gitea.io/gitea/modules/storage"
- "code.gitea.io/gitea/modules/util"
- "code.gitea.io/gitea/modules/web/routing"
- )
-
- func storageHandler(storageSetting *setting.Storage, prefix string, objStore storage.ObjectStorage) func(next http.Handler) http.Handler {
- prefix = strings.Trim(prefix, "/")
- funcInfo := routing.GetFuncInfo(storageHandler, prefix)
- return func(next http.Handler) http.Handler {
- if storageSetting.MinioConfig.ServeDirect {
- return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
- if req.Method != "GET" && req.Method != "HEAD" {
- next.ServeHTTP(w, req)
- return
- }
-
- if !strings.HasPrefix(req.URL.Path, "/"+prefix+"/") {
- next.ServeHTTP(w, req)
- return
- }
- routing.UpdateFuncInfo(req.Context(), funcInfo)
-
- rPath := strings.TrimPrefix(req.URL.Path, "/"+prefix+"/")
- rPath = util.PathJoinRelX(rPath)
-
- u, err := objStore.URL(rPath, path.Base(rPath))
- if err != nil {
- if os.IsNotExist(err) || errors.Is(err, os.ErrNotExist) {
- log.Warn("Unable to find %s %s", prefix, rPath)
- http.Error(w, "file not found", http.StatusNotFound)
- return
- }
- log.Error("Error whilst getting URL for %s %s. Error: %v", prefix, rPath, err)
- http.Error(w, fmt.Sprintf("Error whilst getting URL for %s %s", prefix, rPath), http.StatusInternalServerError)
- return
- }
-
- http.Redirect(w, req, u.String(), http.StatusTemporaryRedirect)
- })
- }
-
- return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
- if req.Method != "GET" && req.Method != "HEAD" {
- next.ServeHTTP(w, req)
- return
- }
-
- if !strings.HasPrefix(req.URL.Path, "/"+prefix+"/") {
- next.ServeHTTP(w, req)
- return
- }
- routing.UpdateFuncInfo(req.Context(), funcInfo)
-
- rPath := strings.TrimPrefix(req.URL.Path, "/"+prefix+"/")
- rPath = util.PathJoinRelX(rPath)
- if rPath == "" || rPath == "." {
- http.Error(w, "file not found", http.StatusNotFound)
- return
- }
-
- fi, err := objStore.Stat(rPath)
- if err != nil {
- if os.IsNotExist(err) || errors.Is(err, os.ErrNotExist) {
- log.Warn("Unable to find %s %s", prefix, rPath)
- http.Error(w, "file not found", http.StatusNotFound)
- return
- }
- log.Error("Error whilst opening %s %s. Error: %v", prefix, rPath, err)
- http.Error(w, fmt.Sprintf("Error whilst opening %s %s", prefix, rPath), http.StatusInternalServerError)
- return
- }
-
- fr, err := objStore.Open(rPath)
- if err != nil {
- log.Error("Error whilst opening %s %s. Error: %v", prefix, rPath, err)
- http.Error(w, fmt.Sprintf("Error whilst opening %s %s", prefix, rPath), http.StatusInternalServerError)
- return
- }
- defer fr.Close()
- httpcache.ServeContentWithCacheControl(w, req, path.Base(rPath), fi.ModTime(), fr)
- })
- }
- }
|