photoprism/internal/server/routes.go

package server

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/gin-gonic/gin/binding"
	"github.com/photoprism/photoprism/internal/forms"
	"github.com/photoprism/photoprism/internal/photoprism"
	"net/http"
	"strconv"
)

func ConfigureRoutes(app *gin.Engine, conf *photoprism.Config) {
	app.LoadHTMLGlob(conf.GetTemplatesPath() + "/*")

	app.StaticFile("/favicon.ico", conf.GetFaviconsPath()+"/favicon.ico")
	app.StaticFile("/favicon.png", conf.GetFaviconsPath()+"/favicon.png")

	app.Static("/assets", conf.GetPublicPath())

	// JSON-REST API Version 1
	v1 := app.Group("/api/v1")
	{
		v1.GET("/photos", func(c *gin.Context) {
			var form forms.PhotoSearchForm

			search := photoprism.NewSearch(conf.OriginalsPath, conf.GetDb())

			c.MustBindWith(&form, binding.Form)

			result, err := search.Photos(form)

			if err != nil {
				c.AbortWithStatusJSON(400, gin.H{"error": err.Error()})
			}

			c.Header("x-result-count", strconv.Itoa(form.Count))
			c.Header("x-result-offset", strconv.Itoa(form.Offset))

			c.JSON(http.StatusOK, result)
		})

		// v1.OPTIONS()

		v1.GET("/files", func(c *gin.Context) {
			search := photoprism.NewSearch(conf.OriginalsPath, conf.GetDb())

			files := search.FindFiles(70, 0)

			c.JSON(http.StatusOK, files)
		})

		v1.GET("/files/:hash/thumbnail", func(c *gin.Context) {
			fileHash := c.Param("hash")
			size, _ := strconv.Atoi(c.Query("size"))

			search := photoprism.NewSearch(conf.OriginalsPath, conf.GetDb())

			file := search.FindFileByHash(fileHash)

			fileName := fmt.Sprintf("%s/%s", conf.OriginalsPath, file.FileName)

			if mediaFile, err := photoprism.NewMediaFile(fileName); err == nil {
				thumbnail, _ := mediaFile.GetThumbnail(conf.ThumbnailsPath, size)

				c.File(thumbnail.GetFilename())
			} else {
				c.Data(404, "image/svg+xml", notFoundSvg)
			}
		})

		v1.GET("/files/:hash/square_thumbnail", func(c *gin.Context) {
			fileHash := c.Param("hash")
			size, _ := strconv.Atoi(c.Query("size"))

			search := photoprism.NewSearch(conf.OriginalsPath, conf.GetDb())

			file := search.FindFileByHash(fileHash)

			fileName := fmt.Sprintf("%s/%s", conf.OriginalsPath, file.FileName)

			if mediaFile, err := photoprism.NewMediaFile(fileName); err == nil {
				thumbnail, _ := mediaFile.GetSquareThumbnail(conf.ThumbnailsPath, size)

				c.File(thumbnail.GetFilename())
			} else {
				c.Data(404, "image/svg+xml", notFoundSvg)
			}
		})

		v1.GET("/albums", func(c *gin.Context) {
			c.JSON(http.StatusOK, gin.H{"status": "ok"})
		})

		v1.GET("/tags", func(c *gin.Context) {
			c.JSON(http.StatusOK, gin.H{"status": "ok"})
		})
	}

	app.NoRoute(func(c *gin.Context) {
		c.HTML(http.StatusOK, "index.tmpl", conf.GetClientConfig())
	})
}