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() }