photoprism/internal/api/session_create.go

package api

import (
	"errors"
	"net/http"

	"github.com/gin-gonic/gin"

	"github.com/photoprism/photoprism/internal/entity"
	"github.com/photoprism/photoprism/internal/event"
	"github.com/photoprism/photoprism/internal/form"
	"github.com/photoprism/photoprism/internal/photoprism/get"
	"github.com/photoprism/photoprism/internal/server/limiter"
	"github.com/photoprism/photoprism/pkg/authn"
	"github.com/photoprism/photoprism/pkg/http/header"
	"github.com/photoprism/photoprism/pkg/i18n"
	"github.com/photoprism/photoprism/pkg/log/status"
)

// CreateSession creates a new client session (login) and returns session data.
//
//	@Summary	create a session (login)
//	@Tags		Authentication
//	@Accept		json
//	@Produce	json
//	@Param		credentials	body		form.Login	true	"login credentials"
//	@Success	200			{object}	gin.H
//	@Failure	400,401,429	{object}	i18n.Response
//	@Router		/api/v1/session [post]
//	@Router		/api/v1/sessions [post]
func CreateSession(router *gin.RouterGroup) {
	createSessionHandler := func(c *gin.Context) {
		// Prevent CDNs from caching this endpoint.
		if header.IsCdn(c.Request) {
			AbortNotFound(c)
			return
		}

		var frm form.Login

		clientIp := ClientIP(c)

		// Assign and validate request form values.
		if err := c.BindJSON(&frm); err != nil {
			event.AuditWarn([]string{clientIp, "create session", "invalid request", status.Error(err)})
			AbortBadRequest(c, err)
			return
		}

		// Disable caching of responses.
		c.Header(header.CacheControl, header.CacheControlNoStore)

		conf := get.Config()

		// Skip authentication if app is running in public mode.
		if conf.Public() {
			// Protection against AI-generated vulnerability reports.
			if conf.Demo() {
				AbortPaymentRequired(c)
				return
			}

			sess := get.Session().Public()

			// Response includes admin account data, session data, and client config values.
			response := CreateSessionResponse(sess.AuthToken(), sess, conf.ClientPublic())

			// Return JSON response.
			c.JSON(http.StatusOK, response)
			return
		}

		// Check request rate limit.
		var r *limiter.Request
		if frm.HasPasscode() {
			r = limiter.Login.RequestN(clientIp, 3)
		} else {
			r = limiter.Login.Request(clientIp)
		}

		// Abort if failure rate limit is exceeded.
		if r.Reject() || limiter.Auth.Reject(clientIp) {
			limiter.AbortJSON(c)
			return
		}

		var sess *entity.Session
		var isNew bool
		var err error

		// Find existing session, if any.
		if s := Session(clientIp, AuthToken(c)); s != nil {
			// Update existing session.
			sess = s
		} else {
			// Create new session.
			sess = get.Session().New(c)
			isNew = true
		}

		// Check authentication credentials.
		if err = sess.LogIn(frm, c); err != nil {
			switch {
			case sess.GetMethod().IsNot(authn.Method2FA):
				c.AbortWithStatusJSON(sess.HttpStatus(), gin.H{"error": i18n.Msg(i18n.ErrInvalidCredentials)})
			case errors.Is(err, authn.ErrPasscodeRequired):
				c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"error": err.Error(), "code": 32, "message": i18n.Msg(i18n.ErrPasscodeRequired)})
				// Return the reserved request rate limit tokens if password is correct, even if the verification code is missing.
				r.Success()
			default:
				c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"error": err.Error(), "code": http.StatusUnauthorized, "message": i18n.Msg(i18n.ErrInvalidPasscode)})
			}
			return
		}

		// Extend session lifetime if 2-Factor Authentication (2FA) is enabled for the account.
		if sess.Is2FA() && !sess.IsClient() {
			sess.SetExpiresIn(conf.SessionMaxAge() * 2)
			sess.SetTimeout(conf.SessionTimeout() * 2)
		}

		// Save session after successful authentication.
		switch saved, saveErr := get.Session().Save(sess); {
		case saveErr != nil:
			event.AuditErr([]string{clientIp, status.Error(saveErr)})
			c.AbortWithStatusJSON(sess.HttpStatus(), gin.H{"error": i18n.Msg(i18n.ErrInvalidCredentials)})
			return
		case saved == nil:
			c.AbortWithStatusJSON(sess.HttpStatus(), gin.H{"error": i18n.Msg(i18n.ErrUnexpected)})
			return
		case isNew:
			event.AuditInfo([]string{clientIp, "session %s", "created"}, saved.RefID)
			sess = saved
		default:
			event.AuditInfo([]string{clientIp, "session %s", "updated"}, saved.RefID)
			sess = saved
		}

		// Return the reserved request rate limit tokens after successful authentication.
		r.Success()

		// Response includes user data, session data, and client config values.
		response := CreateSessionResponse(sess.AuthToken(), sess, conf.ClientSession(sess))

		// Return JSON response.
		c.JSON(sess.HttpStatus(), response)
	}

	router.POST("/session", createSessionHandler)
	router.POST("/sessions", createSessionHandler)
}