187 lines
3.9 KiB
Go
187 lines
3.9 KiB
Go
package config
|
|
|
|
import (
|
|
"context"
|
|
"runtime"
|
|
"time"
|
|
|
|
"github.com/jinzhu/gorm"
|
|
_ "github.com/jinzhu/gorm/dialects/mysql"
|
|
_ "github.com/jinzhu/gorm/dialects/sqlite"
|
|
gc "github.com/patrickmn/go-cache"
|
|
"github.com/photoprism/photoprism/internal/event"
|
|
"github.com/sirupsen/logrus"
|
|
tensorflow "github.com/tensorflow/tensorflow/tensorflow/go"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
var log = event.Log
|
|
|
|
type Config struct {
|
|
db *gorm.DB
|
|
cache *gc.Cache
|
|
config *Params
|
|
}
|
|
|
|
func initLogger(debug bool) {
|
|
log.SetFormatter(&logrus.TextFormatter{
|
|
DisableColors: false,
|
|
FullTimestamp: true,
|
|
})
|
|
|
|
if debug {
|
|
log.SetLevel(logrus.DebugLevel)
|
|
} else {
|
|
log.SetLevel(logrus.InfoLevel)
|
|
}
|
|
}
|
|
|
|
func NewConfig(ctx *cli.Context) *Config {
|
|
initLogger(ctx.GlobalBool("debug"))
|
|
|
|
c := &Config{
|
|
config: NewParams(ctx),
|
|
}
|
|
|
|
log.SetLevel(c.LogLevel())
|
|
|
|
return c
|
|
}
|
|
|
|
// Name returns the application name.
|
|
func (c *Config) Name() string {
|
|
return c.config.Name
|
|
}
|
|
|
|
// Url returns the public server URL (default is "http://localhost:2342/").
|
|
func (c *Config) Url() string {
|
|
if c.config.Url == "" {
|
|
return "http://localhost:2342/"
|
|
}
|
|
|
|
return c.config.Url
|
|
}
|
|
|
|
// Title returns the site title (default is application name).
|
|
func (c *Config) Title() string {
|
|
if c.config.Title == "" {
|
|
return c.Name()
|
|
}
|
|
|
|
return c.config.Title
|
|
}
|
|
|
|
// Subtitle returns the site title.
|
|
func (c *Config) Subtitle() string {
|
|
return c.config.Subtitle
|
|
}
|
|
|
|
// Description returns the site title.
|
|
func (c *Config) Description() string {
|
|
return c.config.Description
|
|
}
|
|
|
|
// Author returns the site author / copyright.
|
|
func (c *Config) Author() string {
|
|
return c.config.Author
|
|
}
|
|
|
|
// Description returns the twitter handle for sharing.
|
|
func (c *Config) Twitter() string {
|
|
return c.config.Twitter
|
|
}
|
|
|
|
// Version returns the application version.
|
|
func (c *Config) Version() string {
|
|
return c.config.Version
|
|
}
|
|
|
|
// TensorFlowVersion returns the TenorFlow framework version.
|
|
func (c *Config) TensorFlowVersion() string {
|
|
return tensorflow.Version()
|
|
}
|
|
|
|
// Copyright returns the application copyright.
|
|
func (c *Config) Copyright() string {
|
|
return c.config.Copyright
|
|
}
|
|
|
|
// Debug returns true if Debug mode is on.
|
|
func (c *Config) Debug() bool {
|
|
return c.config.Debug
|
|
}
|
|
|
|
// Public returns true if app requires no authentication.
|
|
func (c *Config) Public() bool {
|
|
return c.config.Public
|
|
}
|
|
|
|
// Experimental returns true if experimental features should be enabled.
|
|
func (c *Config) Experimental() bool {
|
|
return c.config.Experimental
|
|
}
|
|
|
|
// ReadOnly returns true if photo directories are write protected.
|
|
func (c *Config) ReadOnly() bool {
|
|
return c.config.ReadOnly
|
|
}
|
|
|
|
// HideNSFW returns true if NSFW photos are hidden by default.
|
|
func (c *Config) HideNSFW() bool {
|
|
return c.config.HideNSFW
|
|
}
|
|
|
|
// UploadNSFW returns true if NSFW photos can be uploaded.
|
|
func (c *Config) UploadNSFW() bool {
|
|
return c.config.UploadNSFW
|
|
}
|
|
|
|
// AdminPassword returns the admin password.
|
|
func (c *Config) AdminPassword() string {
|
|
if c.config.AdminPassword == "" {
|
|
return "photoprism"
|
|
}
|
|
|
|
return c.config.AdminPassword
|
|
}
|
|
|
|
// LogLevel returns the logrus log level.
|
|
func (c *Config) LogLevel() logrus.Level {
|
|
if c.Debug() {
|
|
c.config.LogLevel = "debug"
|
|
}
|
|
|
|
if logLevel, err := logrus.ParseLevel(c.config.LogLevel); err == nil {
|
|
return logLevel
|
|
} else {
|
|
return logrus.InfoLevel
|
|
}
|
|
}
|
|
|
|
// Cache returns the in-memory cache.
|
|
func (c *Config) Cache() *gc.Cache {
|
|
if c.cache == nil {
|
|
c.cache = gc.New(336*time.Hour, 30*time.Minute)
|
|
}
|
|
|
|
return c.cache
|
|
}
|
|
|
|
// Init initialises the Database.
|
|
func (c *Config) Init(ctx context.Context) error {
|
|
return c.connectToDatabase(ctx)
|
|
}
|
|
|
|
// Shutdown closes open database connections.
|
|
func (c *Config) Shutdown() {
|
|
if err := c.CloseDb(); err != nil {
|
|
log.Errorf("could not close database connection: %s", err)
|
|
} else {
|
|
log.Info("closed database connection")
|
|
}
|
|
}
|
|
|
|
// Workers returns the number of workers e.g. for indexing files.
|
|
func(c *Config) Workers() int {
|
|
return runtime.NumCPU()
|
|
}
|