2020-08-18 07:23:45 +03:00
// Copyright 2020 The Gitea Authors. All rights reserved.
2022-11-27 21:20:29 +03:00
// SPDX-License-Identifier: MIT
2020-08-18 07:23:45 +03:00
package storage
import (
2020-10-13 06:58:34 +03:00
"context"
2023-03-21 23:02:49 +03:00
"fmt"
2020-08-18 07:23:45 +03:00
"io"
"net/url"
"os"
"path/filepath"
2020-10-16 06:51:06 +03:00
"code.gitea.io/gitea/modules/log"
2023-06-14 06:42:38 +03:00
"code.gitea.io/gitea/modules/setting"
2020-08-18 07:23:45 +03:00
"code.gitea.io/gitea/modules/util"
)
2022-01-20 20:46:10 +03:00
var _ ObjectStorage = & LocalStorage { }
2020-08-18 07:23:45 +03:00
// LocalStorage represents a local files storage
type LocalStorage struct {
2021-03-05 16:19:17 +03:00
ctx context . Context
dir string
tmpdir string
2020-08-18 07:23:45 +03:00
}
// NewLocalStorage returns a local files
2023-06-14 06:42:38 +03:00
func NewLocalStorage ( ctx context . Context , config * setting . Storage ) ( ObjectStorage , error ) {
2023-03-21 23:02:49 +03:00
if ! filepath . IsAbs ( config . Path ) {
return nil , fmt . Errorf ( "LocalStorageConfig.Path should have been prepared by setting/storage.go and should be an absolute path, but not: %q" , config . Path )
}
2020-10-16 06:51:06 +03:00
log . Info ( "Creating new Local Storage at %s" , config . Path )
2020-10-13 06:58:34 +03:00
if err := os . MkdirAll ( config . Path , os . ModePerm ) ; err != nil {
2020-08-18 07:23:45 +03:00
return nil , err
}
2021-03-05 16:19:17 +03:00
if config . TemporaryPath == "" {
2023-03-21 23:02:49 +03:00
config . TemporaryPath = filepath . Join ( config . Path , "tmp" )
}
if ! filepath . IsAbs ( config . TemporaryPath ) {
return nil , fmt . Errorf ( "LocalStorageConfig.TemporaryPath should be an absolute path, but not: %q" , config . TemporaryPath )
2021-03-05 16:19:17 +03:00
}
2020-08-18 07:23:45 +03:00
return & LocalStorage {
2021-03-05 16:19:17 +03:00
ctx : ctx ,
dir : config . Path ,
tmpdir : config . TemporaryPath ,
2020-08-18 07:23:45 +03:00
} , nil
}
2022-03-23 00:02:26 +03:00
func ( l * LocalStorage ) buildLocalPath ( p string ) string {
2023-03-21 23:02:49 +03:00
return util . FilePathJoinAbs ( l . dir , p )
2022-03-23 00:02:26 +03:00
}
2020-08-18 07:23:45 +03:00
// Open a file
2020-09-08 18:45:10 +03:00
func ( l * LocalStorage ) Open ( path string ) ( Object , error ) {
2022-03-23 00:02:26 +03:00
return os . Open ( l . buildLocalPath ( path ) )
2020-08-18 07:23:45 +03:00
}
// Save a file
2021-04-03 19:19:59 +03:00
func ( l * LocalStorage ) Save ( path string , r io . Reader , size int64 ) ( int64 , error ) {
2022-03-23 00:02:26 +03:00
p := l . buildLocalPath ( path )
2020-08-18 07:23:45 +03:00
if err := os . MkdirAll ( filepath . Dir ( p ) , os . ModePerm ) ; err != nil {
return 0 , err
}
2021-03-05 16:19:17 +03:00
// Create a temporary file to save to
if err := os . MkdirAll ( l . tmpdir , os . ModePerm ) ; err != nil {
2020-08-18 07:23:45 +03:00
return 0 , err
}
2021-09-22 08:38:34 +03:00
tmp , err := os . CreateTemp ( l . tmpdir , "upload-*" )
2021-03-05 16:19:17 +03:00
if err != nil {
return 0 , err
}
tmpRemoved := false
defer func ( ) {
if ! tmpRemoved {
_ = util . Remove ( tmp . Name ( ) )
}
} ( )
2020-08-18 07:23:45 +03:00
2021-03-05 16:19:17 +03:00
n , err := io . Copy ( tmp , r )
2020-08-18 07:23:45 +03:00
if err != nil {
return 0 , err
}
2021-03-05 16:19:17 +03:00
if err := tmp . Close ( ) ; err != nil {
return 0 , err
}
2021-07-15 18:46:07 +03:00
if err := util . Rename ( tmp . Name ( ) , p ) ; err != nil {
2021-03-05 16:19:17 +03:00
return 0 , err
}
2022-09-24 16:04:14 +03:00
// Golang's tmp file (os.CreateTemp) always have 0o600 mode, so we need to change the file to follow the umask (as what Create/MkDir does)
2022-12-19 03:50:36 +03:00
// but we don't want to make these files executable - so ensure that we mask out the executable bits
if err := util . ApplyUmask ( p , os . ModePerm & 0 o666 ) ; err != nil {
2022-09-24 16:04:14 +03:00
return 0 , err
}
2021-03-05 16:19:17 +03:00
tmpRemoved = true
return n , nil
2020-08-18 07:23:45 +03:00
}
2020-09-08 18:45:10 +03:00
// Stat returns the info of the file
2020-09-29 12:05:13 +03:00
func ( l * LocalStorage ) Stat ( path string ) ( os . FileInfo , error ) {
2022-03-23 00:02:26 +03:00
return os . Stat ( l . buildLocalPath ( path ) )
2022-03-14 18:18:27 +03:00
}
2020-08-18 07:23:45 +03:00
// Delete delete a file
func ( l * LocalStorage ) Delete ( path string ) error {
2022-03-23 00:02:26 +03:00
return util . Remove ( l . buildLocalPath ( path ) )
2020-08-18 07:23:45 +03:00
}
// URL gets the redirect URL to a file
2024-10-31 18:28:25 +03:00
func ( l * LocalStorage ) URL ( path , name string , reqParams url . Values ) ( * url . URL , error ) {
2020-08-18 07:23:45 +03:00
return nil , ErrURLNotSupported
}
2020-09-29 12:05:13 +03:00
// IterateObjects iterates across the objects in the local storage
2023-05-14 01:33:25 +03:00
func ( l * LocalStorage ) IterateObjects ( dirName string , fn func ( path string , obj Object ) error ) error {
dir := l . buildLocalPath ( dirName )
2023-03-13 13:23:51 +03:00
return filepath . WalkDir ( dir , func ( path string , d os . DirEntry , err error ) error {
2020-09-29 12:05:13 +03:00
if err != nil {
return err
}
2020-10-13 06:58:34 +03:00
select {
case <- l . ctx . Done ( ) :
return l . ctx . Err ( )
default :
}
2020-09-29 12:05:13 +03:00
if path == l . dir {
return nil
}
2023-01-16 19:21:44 +03:00
if d . IsDir ( ) {
2020-09-29 12:05:13 +03:00
return nil
}
relPath , err := filepath . Rel ( l . dir , path )
if err != nil {
return err
}
obj , err := os . Open ( path )
if err != nil {
return err
}
defer obj . Close ( )
return fn ( relPath , obj )
} )
}
2020-10-13 06:58:34 +03:00
func init ( ) {
2023-06-14 06:42:38 +03:00
RegisterStorageType ( setting . LocalStorageType , NewLocalStorage )
2020-10-13 06:58:34 +03:00
}