photoprism/internal/api/connect.go
Michael Mayer b7bbc59fca Upgrade: Improve error handling and messages #3813
Signed-off-by: Michael Mayer <michael@photoprism.app>
2023-10-13 12:43:20 +02:00

85 lines
1.9 KiB
Go

package api
import (
"net/http"
"github.com/gin-gonic/gin"
"github.com/photoprism/photoprism/internal/acl"
"github.com/photoprism/photoprism/internal/form"
"github.com/photoprism/photoprism/internal/get"
"github.com/photoprism/photoprism/internal/i18n"
"github.com/photoprism/photoprism/internal/mutex"
"github.com/photoprism/photoprism/pkg/clean"
)
// Connect confirms external service accounts using a token.
//
// PUT /api/v1/connect/:name
func Connect(router *gin.RouterGroup) {
router.PUT("/connect/:name", func(c *gin.Context) {
name := clean.ID(c.Param("name"))
if name == "" {
log.Errorf("connect: empty service name")
AbortBadRequest(c)
return
}
var f form.Connect
if err := c.BindJSON(&f); err != nil {
log.Warnf("connect: invalid form values (%s)", clean.Log(name))
Abort(c, http.StatusBadRequest, i18n.ErrAccountConnect)
return
}
if f.Invalid() {
log.Warnf("connect: invalid token %s", clean.Log(f.Token))
Abort(c, http.StatusBadRequest, i18n.ErrAccountConnect)
return
}
conf := get.Config()
if conf.Public() {
Abort(c, http.StatusForbidden, i18n.ErrPublic)
return
}
s := Auth(c, acl.ResourceConfig, acl.ActionUpdate)
if !s.IsSuperAdmin() {
log.Errorf("connect: %s not authorized", clean.Log(s.User().UserName))
AbortForbidden(c)
return
}
var err error
var restart bool
switch name {
case "hub":
old := conf.Hub().Session
err = conf.RenewApiKeysWithToken(f.Token)
restart = old != conf.Hub().Session
default:
log.Errorf("connect: invalid service %s", clean.Log(name))
Abort(c, http.StatusBadRequest, i18n.ErrAccountConnect)
return
}
// Set restart flag and update client config if necessary.
if restart {
mutex.Restart.Store(true)
conf.Propagate()
UpdateClientConfig()
}
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
} else {
c.JSON(http.StatusOK, gin.H{"code": http.StatusOK})
}
})
}