photoprism/internal/api/auth_share.go

package api

import (
	"fmt"
	"net/http"
	"path"

	"github.com/gin-gonic/gin"

	"github.com/photoprism/photoprism/internal/entity"
	"github.com/photoprism/photoprism/internal/query"
	"github.com/photoprism/photoprism/internal/service"
	"github.com/photoprism/photoprism/pkg/clean"
)

// Shares handles link share
//
// GET /s/:token/...
func Shares(router *gin.RouterGroup) {
	router.GET("/:token", func(c *gin.Context) {
		conf := service.Config()

		token := clean.Token(c.Param("token"))
		links := entity.FindValidLinks(token, "")

		if len(links) == 0 {
			log.Debugf("share: invalid token")
			c.Redirect(http.StatusTemporaryRedirect, conf.BaseUri(""))
			return
		}

		clientConfig := conf.ClientShare()
		clientConfig.SiteUrl = fmt.Sprintf("%ss/%s", clientConfig.SiteUrl, token)

		uri := conf.BaseUri("/albums")
		c.HTML(http.StatusOK, "share.tmpl", gin.H{"shared": gin.H{"token": token, "uri": uri}, "config": clientConfig})
	})

	router.GET("/:token/:shared", func(c *gin.Context) {
		conf := service.Config()

		token := clean.Token(c.Param("token"))
		shared := clean.Token(c.Param("shared"))

		links := entity.FindValidLinks(token, shared)

		if len(links) < 1 {
			log.Debugf("share: invalid token or slug")
			c.Redirect(http.StatusTemporaryRedirect, conf.BaseUri(""))
			return
		}

		uid := links[0].ShareUID
		clientConfig := conf.ClientShare()
		clientConfig.SiteUrl = fmt.Sprintf("%s/%s", clientConfig.SiteUrl, path.Join("s", token, uid))
		clientConfig.SitePreview = fmt.Sprintf("%s/preview", clientConfig.SiteUrl)

		if a, err := query.AlbumByUID(uid); err == nil {
			clientConfig.SiteCaption = a.AlbumTitle

			if a.AlbumDescription != "" {
				clientConfig.SiteDescription = a.AlbumDescription
			}
		}

		uri := conf.BaseUri(path.Join("/albums", uid, shared))

		c.HTML(http.StatusOK, "share.tmpl", gin.H{"shared": gin.H{"token": token, "uri": uri}, "config": clientConfig})
	})
}