3878e985b6
Signed-off-by: Andrew Thornton <art27@cantab.net> Co-authored-by: zeripath <art27@cantab.net>
108 lines
2.4 KiB
Go
108 lines
2.4 KiB
Go
// Copyright 2020 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package storage
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net/url"
|
|
"os"
|
|
|
|
"code.gitea.io/gitea/modules/setting"
|
|
)
|
|
|
|
var (
|
|
// ErrURLNotSupported represents url is not supported
|
|
ErrURLNotSupported = errors.New("url method not supported")
|
|
// ErrIterateObjectsNotSupported represents IterateObjects not supported
|
|
ErrIterateObjectsNotSupported = errors.New("iterateObjects method not supported")
|
|
)
|
|
|
|
// Object represents the object on the storage
|
|
type Object interface {
|
|
io.ReadCloser
|
|
io.Seeker
|
|
Stat() (os.FileInfo, error)
|
|
}
|
|
|
|
// ObjectStorage represents an object storage to handle a bucket and files
|
|
type ObjectStorage interface {
|
|
Open(path string) (Object, error)
|
|
Save(path string, r io.Reader) (int64, error)
|
|
Stat(path string) (os.FileInfo, error)
|
|
Delete(path string) error
|
|
URL(path, name string) (*url.URL, error)
|
|
IterateObjects(func(path string, obj Object) error) error
|
|
}
|
|
|
|
// Copy copys a file from source ObjectStorage to dest ObjectStorage
|
|
func Copy(dstStorage ObjectStorage, dstPath string, srcStorage ObjectStorage, srcPath string) (int64, error) {
|
|
f, err := srcStorage.Open(srcPath)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
defer f.Close()
|
|
|
|
return dstStorage.Save(dstPath, f)
|
|
}
|
|
|
|
var (
|
|
// Attachments represents attachments storage
|
|
Attachments ObjectStorage
|
|
|
|
// LFS represents lfs storage
|
|
LFS ObjectStorage
|
|
)
|
|
|
|
// Init init the stoarge
|
|
func Init() error {
|
|
if err := initAttachments(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return initLFS()
|
|
}
|
|
|
|
func initStorage(storageCfg setting.Storage) (ObjectStorage, error) {
|
|
var err error
|
|
var s ObjectStorage
|
|
switch storageCfg.Type {
|
|
case setting.LocalStorageType:
|
|
s, err = NewLocalStorage(storageCfg.Path)
|
|
case setting.MinioStorageType:
|
|
minio := storageCfg.Minio
|
|
s, err = NewMinioStorage(
|
|
context.Background(),
|
|
minio.Endpoint,
|
|
minio.AccessKeyID,
|
|
minio.SecretAccessKey,
|
|
minio.Bucket,
|
|
minio.Location,
|
|
minio.BasePath,
|
|
minio.UseSSL,
|
|
)
|
|
default:
|
|
return nil, fmt.Errorf("Unsupported attachment store type: %s", storageCfg.Type)
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return s, nil
|
|
}
|
|
|
|
func initAttachments() (err error) {
|
|
Attachments, err = initStorage(setting.Attachment.Storage)
|
|
return
|
|
}
|
|
|
|
func initLFS() (err error) {
|
|
LFS, err = initStorage(setting.LFS.Storage)
|
|
return
|
|
}
|