photoprism/internal/config/client.go

package config

import (
	"strings"
	"time"

	"github.com/photoprism/photoprism/internal/entity"
	"github.com/photoprism/photoprism/pkg/colors"
	"github.com/photoprism/photoprism/pkg/fs"
)

// ClientConfig contains HTTP client / Web UI config values
type ClientConfig map[string]interface{}

// PublicClientConfig returns reduced config values for non-public sites.
func (c *Config) PublicClientConfig() ClientConfig {
	if c.Public() {
		return c.ClientConfig()
	}

	jsHash := fs.Checksum(c.HttpStaticBuildPath() + "/app.js")
	cssHash := fs.Checksum(c.HttpStaticBuildPath() + "/app.css")

	// Feature Flags
	var flags []string

	if c.Public() {
		flags = append(flags, "public")
	}
	if c.Debug() {
		flags = append(flags, "debug")
	}
	if c.Experimental() {
		flags = append(flags, "experimental")
	}
	if c.ReadOnly() {
		flags = append(flags, "readonly")
	}

	var noPos = struct {
		PhotoUUID  string    `json:"photo"`
		LocationID string    `json:"location"`
		TakenAt    time.Time `json:"utc"`
		PhotoLat   float64   `json:"lat"`
		PhotoLng   float64   `json:"lng"`
	}{}

	var count = struct {
		Photos    uint `json:"photos"`
		Favorites uint `json:"favorites"`
		Private   uint `json:"private"`
		Stories   uint `json:"stories"`
		Labels    uint `json:"labels"`
		Albums    uint `json:"albums"`
		Countries uint `json:"countries"`
		Places    uint `json:"places"`
	}{}

	result := ClientConfig{
		"settings":     c.Settings(),
		"flags":        strings.Join(flags, " "),
		"name":         c.Name(),
		"url":          c.Url(),
		"title":        c.Title(),
		"subtitle":     c.Subtitle(),
		"description":  c.Description(),
		"author":       c.Author(),
		"twitter":      c.Twitter(),
		"version":      c.Version(),
		"copyright":    c.Copyright(),
		"debug":        c.Debug(),
		"readonly":     c.ReadOnly(),
		"uploadNSFW":   c.UploadNSFW(),
		"public":       c.Public(),
		"experimental": c.Experimental(),
		"albums":       []string{},
		"cameras":      []string{},
		"lenses":       []string{},
		"countries":    []string{},
		"thumbnails":   Thumbnails,
		"jsHash":       jsHash,
		"cssHash":      cssHash,
		"count":        count,
		"pos":          noPos,
		"years":        []int{},
		"colors":       colors.All.List(),
		"categories":   []string{},
	}

	return result
}

// ClientConfig returns a loaded and set configuration entity.
func (c *Config) ClientConfig() ClientConfig {
	db := c.Db()

	var cameras []*entity.Camera
	var lenses []*entity.Lens
	var albums []*entity.Album

	var position struct {
		PhotoUUID  string    `json:"photo"`
		LocationID string    `json:"location"`
		TakenAt    time.Time `json:"utc"`
		PhotoLat   float64   `json:"lat"`
		PhotoLng   float64   `json:"lng"`
	}

	db.Table("photos").
		Select("photo_uuid, location_id, photo_lat, photo_lng, taken_at").
		Where("deleted_at IS NULL AND photo_lat != 0 AND photo_lng != 0").
		Order("taken_at DESC").
		Limit(1).Offset(0).
		Take(&position)

	var count = struct {
		Photos    uint `json:"photos"`
		Favorites uint `json:"favorites"`
		Private   uint `json:"private"`
		Stories   uint `json:"stories"`
		Labels    uint `json:"labels"`
		Albums    uint `json:"albums"`
		Countries uint `json:"countries"`
		Places    uint `json:"places"`
	}{}

	db.Table("photos").
		Select("COUNT(*) AS photos, SUM(photo_favorite) AS favorites, SUM(photo_private) AS private, SUM(photo_story) AS stories").
		Where("deleted_at IS NULL").
		Take(&count)

	db.Table("labels").
		Select("COUNT(*) AS labels").
		Where("(label_priority >= 0 || label_favorite = 1) && deleted_at IS NULL").
		Take(&count)

	db.Table("albums").
		Select("COUNT(*) AS albums").
		Where("deleted_at IS NULL").
		Take(&count)

	db.Table("countries").
		Select("(COUNT(*) - 1) AS countries").
		Take(&count)

	db.Table("places").
		Select("(COUNT(*) - 1) AS places").
		Take(&count)

	type country struct {
		ID          string `json:"code"`
		CountryName string `json:"name"`
	}

	var countries []country

	db.Model(&entity.Country{}).
		Select("id, country_name").
		Order("country_name").
		Scan(&countries)

	db.Where("deleted_at IS NULL").
		Limit(1000).Order("camera_model").
		Find(&cameras)

	db.Where("deleted_at IS NULL").
		Limit(1000).Order("lens_model").
		Find(&lenses)

	db.Where("deleted_at IS NULL AND album_favorite = 1").
		Limit(20).Order("album_name").
		Find(&albums)

	var years []int

	db.Table("photos").
		Order("photo_year DESC").
		Pluck("DISTINCT photo_year", &years)

	type CategoryLabel struct {
		LabelName string
		Title     string
	}

	var categories []CategoryLabel

	db.Table("categories").
		Select("l.label_name").
		Joins("JOIN labels l ON categories.category_id = l.id").
		Group("l.label_name").
		Order("l.label_name").
		Limit(1000).Offset(0).
		Scan(&categories)

	for i, l := range categories {
		categories[i].Title = strings.Title(l.LabelName)
	}

	jsHash := fs.Checksum(c.HttpStaticBuildPath() + "/app.js")
	cssHash := fs.Checksum(c.HttpStaticBuildPath() + "/app.css")

	// Feature Flags
	var flags []string

	if c.Public() {
		flags = append(flags, "public")
	}
	if c.Debug() {
		flags = append(flags, "debug")
	}
	if c.Experimental() {
		flags = append(flags, "experimental")
	}
	if c.ReadOnly() {
		flags = append(flags, "readonly")
	}

	result := ClientConfig{
		"flags":        strings.Join(flags, " "),
		"name":         c.Name(),
		"url":          c.Url(),
		"title":        c.Title(),
		"subtitle":     c.Subtitle(),
		"description":  c.Description(),
		"author":       c.Author(),
		"twitter":      c.Twitter(),
		"version":      c.Version(),
		"copyright":    c.Copyright(),
		"debug":        c.Debug(),
		"readonly":     c.ReadOnly(),
		"uploadNSFW":   c.UploadNSFW(),
		"public":       c.Public(),
		"experimental": c.Experimental(),
		"albums":       albums,
		"cameras":      cameras,
		"lenses":       lenses,
		"countries":    countries,
		"thumbnails":   Thumbnails,
		"jsHash":       jsHash,
		"cssHash":      cssHash,
		"settings":     c.Settings(),
		"count":        count,
		"pos":          position,
		"years":        years,
		"colors":       colors.All.List(),
		"categories":   categories,
	}

	return result
}