diff --git a/internal/storage/driver.go b/internal/storage/driver.go index 98c9831..3290a88 100644 --- a/internal/storage/driver.go +++ b/internal/storage/driver.go @@ -6,6 +6,9 @@ import ( "github.com/lus/pasty/internal/config" "github.com/lus/pasty/internal/shared" + "github.com/lus/pasty/internal/storage/file" + "github.com/lus/pasty/internal/storage/mongodb" + "github.com/lus/pasty/internal/storage/s3" ) // Current holds the current storage driver @@ -43,14 +46,14 @@ func Load() error { func GetDriver(storageType shared.StorageType) (Driver, error) { switch storageType { case shared.StorageTypeFile: - return new(FileDriver), nil + return new(file.FileDriver), nil case shared.StorageTypePostgres: // TODO: Implement Postgres driver return nil, errors.New("TODO") case shared.StorageTypeMongoDB: - return new(MongoDBDriver), nil + return new(mongodb.MongoDBDriver), nil case shared.StorageTypeS3: - return new(S3Driver), nil + return new(s3.S3Driver), nil default: return nil, fmt.Errorf("invalid storage type '%s'", storageType) } diff --git a/internal/storage/file_driver.go b/internal/storage/file/file_driver.go similarity index 99% rename from internal/storage/file_driver.go rename to internal/storage/file/file_driver.go index 055f1fd..f394f2b 100644 --- a/internal/storage/file_driver.go +++ b/internal/storage/file/file_driver.go @@ -1,4 +1,4 @@ -package storage +package file import ( "encoding/base64" diff --git a/internal/storage/mongodb_driver.go b/internal/storage/mongodb/mongodb_driver.go similarity index 99% rename from internal/storage/mongodb_driver.go rename to internal/storage/mongodb/mongodb_driver.go index 66937ab..2831422 100644 --- a/internal/storage/mongodb_driver.go +++ b/internal/storage/mongodb/mongodb_driver.go @@ -1,4 +1,4 @@ -package storage +package mongodb import ( "context" diff --git a/internal/storage/s3_driver.go b/internal/storage/s3/s3_driver.go similarity index 99% rename from internal/storage/s3_driver.go rename to internal/storage/s3/s3_driver.go index e79b40c..e3f0e06 100644 --- a/internal/storage/s3_driver.go +++ b/internal/storage/s3/s3_driver.go @@ -1,4 +1,4 @@ -package storage +package s3 import ( "bytes"