mirror of https://github.com/lus/pasty.git
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package storage
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/lus/pasty/internal/env"
|
|
"github.com/lus/pasty/internal/pastes"
|
|
)
|
|
|
|
// Current holds the current storage driver
|
|
var Current Driver
|
|
|
|
// Driver represents a storage driver
|
|
type Driver interface {
|
|
Initialize() error
|
|
Terminate() error
|
|
ListIDs() ([]string, error)
|
|
Get(id string) (*pastes.Paste, error)
|
|
Save(paste *pastes.Paste) error
|
|
Delete(id string) error
|
|
Cleanup() (int, error)
|
|
}
|
|
|
|
// Load loads the current storage driver
|
|
func Load() error {
|
|
// Define the driver to use
|
|
storageType := env.Get("STORAGE_TYPE", "file")
|
|
driver, err := GetDriver(storageType)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Initialize the driver
|
|
err = driver.Initialize()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
Current = driver
|
|
return nil
|
|
}
|
|
|
|
// GetDriver returns the driver with the given type string if it exists
|
|
func GetDriver(storageType string) (Driver, error) {
|
|
switch strings.ToLower(storageType) {
|
|
case "file":
|
|
return new(FileDriver), nil
|
|
case "s3":
|
|
return new(S3Driver), nil
|
|
case "mongodb":
|
|
return new(MongoDBDriver), nil
|
|
case "sql":
|
|
return new(SQLDriver), nil
|
|
default:
|
|
return nil, fmt.Errorf("invalid storage type '%s'", storageType)
|
|
}
|
|
}
|