store.go

package fs

import (
	"io"
	"mime/multipart"
	"os"
	"path"

	"github.com/google/uuid"
	"github.com/kapmahc/h2o/web/uploader"
)

// NewFileSystemStore new file-system uploader store
func NewFileSystemStore(root, home string) (uploader.Store, error) {
	err := os.MkdirAll(root, 0755)
	return &FileSystemStore{home: home, root: root}, err
}

// FileSystemStore file-system storage
type FileSystemStore struct {
	home string
	root string
}

// Remove remove file
func (p *FileSystemStore) Remove(url string) error {
	return os.Remove(path.Join(p.root, url[len(p.home)+1:]))
}

// Save save file to file-system
func (p *FileSystemStore) Save(fh *multipart.FileHeader) (string, int64, error) {
	name := uuid.New().String() + path.Ext(fh.Filename)
	src, err := fh.Open()
	if err != nil {
		return "", 0, err
	}
	defer src.Close()
	dst, err := os.Create(path.Join(p.root, name))
	if err != nil {
		return "", 0, err
	}
	defer dst.Close()
	size, err := io.Copy(dst, src)
	return p.home + "/" + name, size, err
}