mirror of
https://github.com/Luzifer/gallery.git
synced 2024-12-22 19:01:23 +00:00
43 lines
917 B
Go
43 lines
917 B
Go
package main
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"path"
|
|
)
|
|
|
|
type fileStorage struct {
|
|
path string
|
|
}
|
|
|
|
// WriteFile takes the content of a file and writes it into the underlying
|
|
// storage system.
|
|
func (l fileStorage) WriteFile(filepath string, content io.Reader, contentType string) error {
|
|
fullPath := path.Join(l.path, filepath)
|
|
if err := os.MkdirAll(path.Dir(fullPath), 0755); err != nil {
|
|
return err
|
|
}
|
|
|
|
fp, err := os.Create(fullPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer fp.Close()
|
|
|
|
_, err = io.Copy(fp, content)
|
|
return err
|
|
}
|
|
|
|
// ReadFile retrieves a file from the underlying storage, needs to return
|
|
// errFileNotFound when file is not present.
|
|
func (l fileStorage) ReadFile(filepath string) (io.ReadCloser, error) {
|
|
fullPath := path.Join(l.path, filepath)
|
|
if _, err := os.Stat(fullPath); err != nil {
|
|
if os.IsNotExist(err) {
|
|
return nil, errFileNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
return os.Open(fullPath)
|
|
}
|