2020-11-13 20:51:07 +08:00
// Copyright 2020 The Gitea Authors. All rights reserved.
2022-11-27 13:20:29 -05:00
// SPDX-License-Identifier: MIT
2020-11-13 20:51:07 +08:00
2021-06-09 07:33:54 +08:00
package web
2020-11-13 20:51:07 +08:00
import (
"errors"
"fmt"
"net/http"
"os"
"path"
"strings"
2020-11-17 23:44:52 +01:00
"code.gitea.io/gitea/modules/httpcache"
2020-11-13 20:51:07 +08:00
"code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/setting"
"code.gitea.io/gitea/modules/storage"
2023-03-08 20:17:39 +08:00
"code.gitea.io/gitea/modules/util"
2022-01-20 19:41:25 +08:00
"code.gitea.io/gitea/modules/web/routing"
2020-11-13 20:51:07 +08:00
)
2023-10-06 13:23:14 +00:00
func storageHandler ( storageSetting * setting . Storage , prefix string , objStore storage . ObjectStorage ) http . HandlerFunc {
2022-03-22 21:02:26 +00:00
prefix = strings . Trim ( prefix , "/" )
2022-01-20 19:41:25 +08:00
funcInfo := routing . GetFuncInfo ( storageHandler , prefix )
2020-11-13 20:51:07 +08:00
2024-05-30 15:33:50 +08:00
if storageSetting . ServeDirect ( ) {
2020-11-13 20:51:07 +08:00
return http . HandlerFunc ( func ( w http . ResponseWriter , req * http . Request ) {
if req . Method != "GET" && req . Method != "HEAD" {
2023-10-06 13:23:14 +00:00
http . Error ( w , http . StatusText ( http . StatusMethodNotAllowed ) , http . StatusMethodNotAllowed )
2020-11-13 20:51:07 +08:00
return
}
2022-03-22 21:02:26 +00:00
if ! strings . HasPrefix ( req . URL . Path , "/" + prefix + "/" ) {
2023-10-06 13:23:14 +00:00
http . Error ( w , http . StatusText ( http . StatusNotFound ) , http . StatusNotFound )
2020-11-13 20:51:07 +08:00
return
}
2022-01-20 19:41:25 +08:00
routing . UpdateFuncInfo ( req . Context ( ) , funcInfo )
2020-11-13 20:51:07 +08:00
2022-03-22 21:02:26 +00:00
rPath := strings . TrimPrefix ( req . URL . Path , "/" + prefix + "/" )
2023-03-22 04:02:49 +08:00
rPath = util . PathJoinRelX ( rPath )
2020-11-17 23:44:52 +01:00
2024-10-31 23:28:25 +08:00
u , err := objStore . URL ( rPath , path . Base ( rPath ) , nil )
2020-11-13 20:51:07 +08:00
if err != nil {
if os . IsNotExist ( err ) || errors . Is ( err , os . ErrNotExist ) {
log . Warn ( "Unable to find %s %s" , prefix , rPath )
2023-10-06 13:23:14 +00:00
http . Error ( w , http . StatusText ( http . StatusNotFound ) , http . StatusNotFound )
2020-11-13 20:51:07 +08:00
return
}
2023-10-06 13:23:14 +00:00
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 )
2020-11-13 20:51:07 +08:00
return
}
2023-10-06 13:23:14 +00:00
http . Redirect ( w , req , u . String ( ) , http . StatusTemporaryRedirect )
2020-11-13 20:51:07 +08:00
} )
}
2023-10-06 13:23:14 +00:00
return http . HandlerFunc ( func ( w http . ResponseWriter , req * http . Request ) {
if req . Method != "GET" && req . Method != "HEAD" {
http . Error ( w , http . StatusText ( http . StatusMethodNotAllowed ) , http . StatusMethodNotAllowed )
return
}
if ! strings . HasPrefix ( req . URL . Path , "/" + prefix + "/" ) {
http . Error ( w , http . StatusText ( http . StatusNotFound ) , http . StatusNotFound )
return
}
routing . UpdateFuncInfo ( req . Context ( ) , funcInfo )
rPath := strings . TrimPrefix ( req . URL . Path , "/" + prefix + "/" )
rPath = util . PathJoinRelX ( rPath )
if rPath == "" || rPath == "." {
http . Error ( w , http . StatusText ( http . StatusNotFound ) , 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 , http . StatusText ( http . StatusNotFound ) , 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 )
} )
2020-11-13 20:51:07 +08:00
}