36 lines
1.2 KiB
Go
36 lines
1.2 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/usememos/memos/internal/storage/s3"
|
|
storepb "github.com/usememos/memos/proto/gen/store"
|
|
)
|
|
|
|
// Driver provides object operations for a configured attachment storage.
|
|
type Driver interface {
|
|
UploadObject(ctx context.Context, key string, fileType string, content io.Reader) (string, error)
|
|
PresignGetObject(ctx context.Context, key string) (string, error)
|
|
GetObject(ctx context.Context, key string) ([]byte, error)
|
|
GetObjectStream(ctx context.Context, key string) (io.ReadCloser, error)
|
|
DeleteObject(ctx context.Context, key string) error
|
|
}
|
|
|
|
// NewDriver creates the driver for a configured storage.
|
|
func NewDriver(ctx context.Context, configuredStorage *storepb.Storage) (Driver, error) {
|
|
if configuredStorage == nil {
|
|
return nil, errors.New("storage is required")
|
|
}
|
|
switch configuredStorage.Type {
|
|
case storepb.StorageType_STORAGE_TYPE_S3:
|
|
if configuredStorage.GetS3Config() == nil {
|
|
return nil, errors.Errorf("S3 config is missing for storage %q", configuredStorage.Id)
|
|
}
|
|
return s3.NewDriver(ctx, configuredStorage.GetS3Config())
|
|
default:
|
|
return nil, errors.Errorf("storage %q has unsupported driver type %s", configuredStorage.Id, configuredStorage.Type.String())
|
|
}
|
|
}
|