package api import ( "net/http" "strconv" "github.com/gin-gonic/gin" "github.com/photoprism/photoprism/internal/ai/classify" "github.com/photoprism/photoprism/internal/auth/acl" "github.com/photoprism/photoprism/internal/entity" "github.com/photoprism/photoprism/internal/entity/query" "github.com/photoprism/photoprism/internal/event" "github.com/photoprism/photoprism/internal/form" "github.com/photoprism/photoprism/pkg/clean" "github.com/photoprism/photoprism/pkg/txt" ) // AddPhotoLabel adds a label to a photo. // // @Summary adds a label to a photo // @Id AddPhotoLabel // @Tags Labels, Photos // @Accept json // @Produce json // @Success 200 {object} entity.Photo // @Failure 400,401,403,404,429,500 {object} i18n.Response // @Param label body form.Label true "label properties" // @Param uid path string true "photo uid" // @Router /api/v1/photos/{uid}/label [post] func AddPhotoLabel(router *gin.RouterGroup) { router.POST("/photos/:uid/label", func(c *gin.Context) { s := Auth(c, acl.ResourcePhotos, acl.ActionUpdate) if s.Abort(c) { return } m, err := query.PhotoByUID(clean.UID(c.Param("uid"))) if err != nil { AbortEntityNotFound(c) return } frm := &form.Label{} // Assign and validate request form values. if err = c.BindJSON(frm); err != nil { AbortBadRequest(c) return } else if err = frm.Validate(); err != nil { AbortInvalidName(c) return } labelEntity := entity.FirstOrCreateLabel(entity.NewLabel(frm.LabelName, frm.LabelPriority)) if labelEntity == nil { AbortInvalidName(c) return } if err = labelEntity.Restore(); err != nil { c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": "could not restore label"}) return } photoLabel := entity.FirstOrCreatePhotoLabel(entity.NewPhotoLabel(m.ID, labelEntity.ID, frm.Uncertainty, "manual")) if photoLabel == nil { c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": "failed to update photo label"}) return } if photoLabel.Uncertainty > frm.Uncertainty { if updateErr := photoLabel.Updates(map[string]interface{}{ "Uncertainty": frm.Uncertainty, "LabelSrc": entity.SrcManual, }); updateErr != nil { log.Errorf("label: %s", updateErr) } } p, err := query.PhotoPreloadByUID(clean.UID(c.Param("uid"))) if err != nil { AbortEntityNotFound(c) return } if err = p.SaveLabels(); err != nil { c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": txt.UpperFirst(err.Error())}) return } PublishPhotoEvent(StatusUpdated, c.Param("uid"), c) event.Success("label updated") c.JSON(http.StatusOK, p) }) } // RemovePhotoLabel removes a label from a photo. // // @Summary removes a label from a photo // @Id RemovePhotoLabel // @Tags Labels, Photos // @Accept json // @Produce json // @Success 200 {object} entity.Photo // @Failure 400,401,403,404,429,500 {object} i18n.Response // @Param uid path string true "photo uid" // @Param id path string true "label id" // @Router /api/v1/photos/{uid}/label/{id} [delete] func RemovePhotoLabel(router *gin.RouterGroup) { router.DELETE("/photos/:uid/label/:id", func(c *gin.Context) { s := Auth(c, acl.ResourcePhotos, acl.ActionUpdate) if s.Abort(c) { return } m, err := query.PhotoByUID(clean.UID(c.Param("uid"))) if err != nil { AbortEntityNotFound(c) return } labelId, err := strconv.Atoi(clean.Token(c.Param("id"))) if err != nil { c.AbortWithStatusJSON(http.StatusNotFound, gin.H{"error": txt.UpperFirst(err.Error())}) return } label, err := query.PhotoLabel(m.ID, uint(labelId)) if err != nil { c.AbortWithStatusJSON(http.StatusNotFound, gin.H{"error": txt.UpperFirst(err.Error())}) return } if label.LabelSrc == classify.SrcManual || label.LabelSrc == classify.SrcTitle || label.LabelSrc == classify.SrcCaption || label.LabelSrc == classify.SrcSubject || label.LabelSrc == classify.SrcKeyword { logErr("label", entity.Db().Delete(&label).Error) } else { label.Uncertainty = 100 logErr("label", entity.Db().Save(&label).Error) } p, err := query.PhotoPreloadByUID(clean.UID(c.Param("uid"))) if err != nil { AbortEntityNotFound(c) return } logErr("label", p.RemoveKeyword(label.Label.LabelName)) if err := p.SaveLabels(); err != nil { c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": txt.UpperFirst(err.Error())}) return } PublishPhotoEvent(StatusUpdated, clean.UID(c.Param("uid")), c) event.Success("label removed") c.JSON(http.StatusOK, p) }) } // UpdatePhotoLabel changes a photo label. // // @Summary changes a photo label // @Id UpdatePhotoLabel // @Tags Labels, Photos // @Accept json // @Produce json // @Success 200 {object} entity.Photo // @Failure 400,401,403,404,429,500 {object} i18n.Response // @Param uid path string true "photo uid" // @Param id path string true "label id" // @Param label body form.Label true "properties to be updated (currently supports: uncertainty)" // @Router /api/v1/photos/{uid}/label/{id} [put] func UpdatePhotoLabel(router *gin.RouterGroup) { router.PUT("/photos/:uid/label/:id", func(c *gin.Context) { s := Auth(c, acl.ResourcePhotos, acl.ActionUpdate) if s.Abort(c) { return } // TODO: Clean up and simplify this. m, err := query.PhotoByUID(clean.UID(c.Param("uid"))) if err != nil { AbortEntityNotFound(c) return } labelId, err := strconv.Atoi(clean.Token(c.Param("id"))) if err != nil { c.AbortWithStatusJSON(http.StatusNotFound, gin.H{"error": txt.UpperFirst(err.Error())}) return } label, err := query.PhotoLabel(m.ID, uint(labelId)) if err != nil { c.AbortWithStatusJSON(http.StatusNotFound, gin.H{"error": txt.UpperFirst(err.Error())}) return } if err = c.BindJSON(label); err != nil { AbortBadRequest(c) return } if err = label.Save(); err != nil { c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": txt.UpperFirst(err.Error())}) return } p, err := query.PhotoPreloadByUID(clean.UID(c.Param("uid"))) if err != nil { AbortEntityNotFound(c) return } if err = p.SaveLabels(); err != nil { c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": txt.UpperFirst(err.Error())}) return } PublishPhotoEvent(StatusUpdated, clean.UID(c.Param("uid")), c) event.Success("label saved") c.JSON(http.StatusOK, p) }) }