photoprism/internal/api/upload.go

114 lines
2.3 KiB
Go
Raw Normal View History

2019-06-13 20:26:01 +02:00
package api
import (
"net/http"
"os"
"path"
2019-06-13 20:26:01 +02:00
"path/filepath"
"time"
"github.com/photoprism/photoprism/internal/acl"
"github.com/photoprism/photoprism/internal/event"
"github.com/photoprism/photoprism/internal/i18n"
"github.com/photoprism/photoprism/internal/service"
"github.com/photoprism/photoprism/pkg/txt"
2019-06-13 20:26:01 +02:00
"github.com/gin-gonic/gin"
)
// POST /api/v1/upload/:path
func Upload(router *gin.RouterGroup) {
router.POST("/upload/:path", func(c *gin.Context) {
conf := service.Config()
if conf.ReadOnly() || !conf.Settings().Features.Upload {
Abort(c, http.StatusForbidden, i18n.ErrReadOnly)
return
}
s := Auth(SessionID(c), acl.ResourcePhotos, acl.ActionUpload)
if s.Invalid() {
AbortUnauthorized(c)
return
}
2019-06-13 20:26:01 +02:00
start := time.Now()
subPath := c.Param("path")
2019-06-13 20:26:01 +02:00
f, err := c.MultipartForm()
2019-06-13 20:26:01 +02:00
if err != nil {
AbortBadRequest(c)
2019-06-13 20:26:01 +02:00
return
}
event.Publish("upload.start", event.Data{"time": start})
files := f.File["files"]
uploaded := len(files)
var uploads []string
2019-06-13 20:26:01 +02:00
p := path.Join(conf.ImportPath(), "upload", subPath)
2019-06-13 20:26:01 +02:00
if err := os.MkdirAll(p, os.ModePerm); err != nil {
AbortBadRequest(c)
2019-06-13 20:26:01 +02:00
return
}
for _, file := range files {
filename := path.Join(p, filepath.Base(file.Filename))
2019-06-13 20:26:01 +02:00
log.Debugf("upload: saving file %s", txt.Quote(file.Filename))
2019-06-13 20:26:01 +02:00
if err := c.SaveUploadedFile(file, filename); err != nil {
AbortBadRequest(c)
2019-06-13 20:26:01 +02:00
return
}
uploads = append(uploads, filename)
}
if !conf.UploadNSFW() {
nd := service.NsfwDetector()
containsNSFW := false
for _, filename := range uploads {
labels, err := nd.File(filename)
if err != nil {
log.Debug(err)
continue
}
if labels.IsSafe() {
continue
}
log.Infof("nsfw: %s might be offensive", txt.Quote(filename))
containsNSFW = true
}
if containsNSFW {
for _, filename := range uploads {
if err := os.Remove(filename); err != nil {
log.Errorf("nsfw: could not delete %s", txt.Quote(filename))
}
}
Abort(c, http.StatusForbidden, i18n.ErrOffensiveUpload)
return
}
2019-06-13 20:26:01 +02:00
}
elapsed := int(time.Since(start).Seconds())
2019-06-13 20:26:01 +02:00
msg := i18n.Msg(i18n.MsgFilesUploadedIn, uploaded, elapsed)
2019-06-13 20:26:01 +02:00
log.Info(msg)
c.JSON(http.StatusOK, i18n.Response{Code: http.StatusOK, Msg: msg})
2019-06-13 20:26:01 +02:00
})
}