152 lines
3.2 KiB
Go
152 lines
3.2 KiB
Go
package operator
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"path"
|
|
"strconv"
|
|
|
|
"mstore/app/descr"
|
|
"mstore/pkg/auxtool"
|
|
"mstore/pkg/auxuuid"
|
|
)
|
|
|
|
// File exists
|
|
type FileExistsParams struct {
|
|
Filepath string
|
|
Source string
|
|
Dest string
|
|
}
|
|
type FileExistsResult struct {
|
|
Descr *descr.File
|
|
}
|
|
|
|
func (oper *Operator) FileExists(param *FileExistsParams) (int, *FileExistsResult, error) {
|
|
var err error
|
|
code := http.StatusNotFound
|
|
res := &FileExistsResult{}
|
|
|
|
filename := path.Base(param.Filepath)
|
|
collection := path.Dir(param.Filepath)
|
|
|
|
exist, file, err := oper.mdb.GetFileByCollection(collection, filename)
|
|
if exist {
|
|
code = http.StatusOK
|
|
res.Descr = file
|
|
}
|
|
return code, res, err
|
|
}
|
|
|
|
// Put file
|
|
type PutFileParams struct {
|
|
ContentType string
|
|
ContentLength string
|
|
Filepath string
|
|
Source io.ReadCloser
|
|
}
|
|
type PutFileResult struct {
|
|
Descr *descr.File
|
|
}
|
|
|
|
const defaultContentType = "application/octet-stream"
|
|
|
|
func (oper *Operator) PutFile(param *PutFileParams) (int, *PutFileResult, error) {
|
|
var err error
|
|
res := &PutFileResult{}
|
|
|
|
size, err := strconv.ParseInt(param.ContentLength, 10, 64)
|
|
if err != nil {
|
|
code := http.StatusLengthRequired
|
|
return code, res, err
|
|
}
|
|
contentType := param.ContentType
|
|
if contentType == "" {
|
|
contentType = defaultContentType
|
|
}
|
|
|
|
filename := path.Base(param.Filepath)
|
|
collection := path.Dir(param.Filepath)
|
|
oper.logg.Debugf("Put file %s %s", collection, filename)
|
|
|
|
tmpname, size, checksum, err := oper.store.WriteTempFile(param.Source)
|
|
if err != nil {
|
|
code := http.StatusInternalServerError
|
|
return code, res, err
|
|
}
|
|
|
|
exists, fileDescr, err := oper.mdb.GetFileByCollection(collection, filename)
|
|
if err != nil {
|
|
code := http.StatusInternalServerError
|
|
return code, res, err
|
|
}
|
|
|
|
now := auxtool.TimeNow()
|
|
if exists {
|
|
fileDescr.Size = size
|
|
fileDescr.Checksum = checksum
|
|
fileDescr.UpdatedAt = now
|
|
fileDescr.Type = contentType
|
|
err = oper.mdb.UpdateFileByID(fileDescr.ID, fileDescr)
|
|
if err != nil {
|
|
code := http.StatusInternalServerError
|
|
return code, res, err
|
|
}
|
|
} else {
|
|
fileDescr = &descr.File{
|
|
ID: auxuuid.NewUUID(),
|
|
Name: filename,
|
|
Collection: collection,
|
|
Size: size,
|
|
Type: contentType,
|
|
Checksum: checksum,
|
|
CreatedAt: now,
|
|
UpdatedAt: now,
|
|
}
|
|
err = oper.mdb.InsertFile(fileDescr)
|
|
if err != nil {
|
|
code := http.StatusInternalServerError
|
|
return code, res, err
|
|
}
|
|
}
|
|
|
|
err = oper.store.LinkFile(tmpname, collection, filename)
|
|
if err != nil {
|
|
code := http.StatusInternalServerError
|
|
return code, res, err
|
|
}
|
|
|
|
res.Descr = fileDescr
|
|
code := http.StatusOK
|
|
return code, res, err
|
|
}
|
|
|
|
// Get file
|
|
type GetFileParams struct {
|
|
Filepath string
|
|
}
|
|
type GetFileResult struct {
|
|
ContentType string
|
|
ContentLength string
|
|
Source io.ReadCloser
|
|
}
|
|
|
|
func (oper *Operator) GetFile(param *GetFileParams) (int, *GetFileResult, error) {
|
|
var err error
|
|
res := &GetFileResult{}
|
|
code := http.StatusOK
|
|
return code, res, err
|
|
}
|
|
|
|
// Delete file
|
|
type DeleteFileParams struct {
|
|
Filepath string
|
|
}
|
|
type DeleteFileResult struct{}
|
|
|
|
func (oper *Operator) DeleteFile(param *DeleteFileParams) (int, *DeleteFileResult, error) {
|
|
var err error
|
|
res := &DeleteFileResult{}
|
|
code := http.StatusOK
|
|
return code, res, err
|
|
}
|