47 lines
855 B
Go
47 lines
855 B
Go
package fstorage
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type fileStorage struct {
|
|
basePath string
|
|
}
|
|
|
|
func New(cfg *Config) FileStorage {
|
|
if cfg == nil {
|
|
cfg = getDefaultConfig()
|
|
}
|
|
|
|
return &fileStorage{
|
|
cfg.BasePath,
|
|
}
|
|
}
|
|
|
|
func (storage *fileStorage) Put(file io.Reader, filename string) error {
|
|
fullPath := path.Join(storage.basePath, filename)
|
|
f, err := os.Create(fullPath)
|
|
if err != nil {
|
|
return errors.Wrap(err, "couldn't create a file")
|
|
}
|
|
defer f.Close()
|
|
_, err = io.Copy(f, file)
|
|
if err != nil {
|
|
return errors.Wrap(err, "couldn't write a file")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (storage *fileStorage) Remove(filename string) error {
|
|
fullPath := path.Join(storage.basePath, filename)
|
|
err := os.Remove(fullPath)
|
|
if err != nil && !os.IsNotExist(err) {
|
|
return errors.Wrap(err, "couldn't remove a file")
|
|
}
|
|
return nil
|
|
}
|