104 lines
2.2 KiB
Go
104 lines
2.2 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/photoprism/photoprism/internal/config"
|
|
"github.com/photoprism/photoprism/internal/util"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// POST /api/v1/upload/:path
|
|
func Upload(router *gin.RouterGroup, conf *config.Config) {
|
|
router.POST("/upload/:path", func(c *gin.Context) {
|
|
if conf.ReadOnly() {
|
|
c.AbortWithStatusJSON(http.StatusForbidden, ErrReadOnly)
|
|
return
|
|
}
|
|
|
|
if Unauthorized(c, conf) {
|
|
c.AbortWithStatusJSON(http.StatusUnauthorized, ErrUnauthorized)
|
|
return
|
|
}
|
|
|
|
start := time.Now()
|
|
subPath := c.Param("path")
|
|
|
|
f, err := c.MultipartForm()
|
|
|
|
if err != nil {
|
|
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": util.UcFirst(err.Error())})
|
|
return
|
|
}
|
|
|
|
files := f.File["files"]
|
|
uploaded := len(files)
|
|
var uploads []string
|
|
|
|
p := path.Join(conf.ImportPath(), "upload", subPath)
|
|
|
|
if err := os.MkdirAll(p, os.ModePerm); err != nil {
|
|
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": util.UcFirst(err.Error())})
|
|
return
|
|
}
|
|
|
|
for _, file := range files {
|
|
filename := path.Join(p, filepath.Base(file.Filename))
|
|
|
|
log.Debugf("upload: saving file \"%s\"", file.Filename)
|
|
|
|
if err := c.SaveUploadedFile(file, filename); err != nil {
|
|
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": util.UcFirst(err.Error())})
|
|
return
|
|
}
|
|
|
|
uploads = append(uploads, filename)
|
|
}
|
|
|
|
if !conf.UploadNSFW() {
|
|
initNsfwDetector(conf)
|
|
|
|
containsNSFW := false
|
|
|
|
for _, filename := range uploads {
|
|
labels, err := nsfwDetector.LabelsFromFile(filename)
|
|
|
|
if err != nil {
|
|
log.Debug(err)
|
|
continue
|
|
}
|
|
|
|
if labels.IsSafe() {
|
|
continue
|
|
}
|
|
|
|
log.Infof("nsfw: \"%s\" might be offensive", filename)
|
|
|
|
containsNSFW = true
|
|
}
|
|
|
|
if containsNSFW {
|
|
for _, filename := range uploads {
|
|
if err := os.Remove(filename); err != nil {
|
|
log.Errorf("nsfw: could not delete \"%s\"", filename)
|
|
}
|
|
}
|
|
|
|
c.AbortWithStatusJSON(http.StatusForbidden, ErrUploadNSFW)
|
|
return
|
|
}
|
|
}
|
|
|
|
elapsed := time.Since(start)
|
|
|
|
log.Infof("%d files uploaded in %s", uploaded, elapsed)
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": fmt.Sprintf("%d files uploaded in %s", uploaded, elapsed)})
|
|
})
|
|
}
|