70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
|
package file
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
)
|
||
|
|
|
||
|
|
// A Service to handle file-uploads in golang
|
||
|
|
type Service struct {
|
||
|
|
StorageType string `toml:"storage_type"`
|
||
|
|
Path string `toml:"path"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check if Service is configurated and useable
|
||
|
|
func (s *Service) Check() error {
|
||
|
|
mgmt, ok := managers[s.StorageType]
|
||
|
|
if !ok {
|
||
|
|
return ErrUnsupportedStorageType
|
||
|
|
}
|
||
|
|
return mgmt.Check(s)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Upload a file to storage
|
||
|
|
func (s *Service) Upload(request *http.Request) (*File, error) {
|
||
|
|
mgmt, ok := managers[s.StorageType]
|
||
|
|
if !ok {
|
||
|
|
return nil, ErrUnsupportedStorageType
|
||
|
|
}
|
||
|
|
file, fileRequest, err := request.FormFile("file")
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
fileObj := File{
|
||
|
|
Filename: filepath.Base(fileRequest.Filename),
|
||
|
|
}
|
||
|
|
|
||
|
|
// detect contenttype
|
||
|
|
buffer := make([]byte, 512)
|
||
|
|
n, err := file.Read(buffer)
|
||
|
|
if err != nil && err != io.EOF {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
fileObj.ContentType = http.DetectContentType(buffer[:n])
|
||
|
|
|
||
|
|
// Reset the read pointer
|
||
|
|
file.Seek(0, io.SeekStart)
|
||
|
|
if err := mgmt.Save(s, &fileObj, file); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return &fileObj, nil
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// GINUpload a file to storage using gin-gonic
|
||
|
|
func (s *Service) GINUpload(c *gin.Context) (*File, error) {
|
||
|
|
return s.Upload(c.Request)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Read a file to storage
|
||
|
|
func (s *Service) Read(file *File) (io.Reader, error) {
|
||
|
|
mgmt, ok := managers[s.StorageType]
|
||
|
|
if !ok {
|
||
|
|
return nil, ErrUnsupportedStorageType
|
||
|
|
}
|
||
|
|
return mgmt.Read(s, file)
|
||
|
|
}
|