158 lines
3.8 KiB
Go
158 lines
3.8 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/OliveTin/OliveTin/internal/entityfiles"
|
|
"github.com/OliveTin/OliveTin/internal/executor"
|
|
grpcapi "github.com/OliveTin/OliveTin/internal/grpcapi"
|
|
"github.com/OliveTin/OliveTin/internal/httpservers"
|
|
"github.com/OliveTin/OliveTin/internal/installationinfo"
|
|
"github.com/OliveTin/OliveTin/internal/oncalendarfile"
|
|
"github.com/OliveTin/OliveTin/internal/oncron"
|
|
"github.com/OliveTin/OliveTin/internal/onfileindir"
|
|
"github.com/OliveTin/OliveTin/internal/onstartup"
|
|
updatecheck "github.com/OliveTin/OliveTin/internal/updatecheck"
|
|
"github.com/OliveTin/OliveTin/internal/websocket"
|
|
|
|
config "github.com/OliveTin/OliveTin/internal/config"
|
|
"github.com/fsnotify/fsnotify"
|
|
"github.com/spf13/viper"
|
|
"os"
|
|
)
|
|
|
|
var (
|
|
cfg *config.Config
|
|
version = "dev"
|
|
commit = "nocommit"
|
|
date = "nodate"
|
|
)
|
|
|
|
func init() {
|
|
initLog()
|
|
|
|
initViperConfig(initCliFlags())
|
|
|
|
initCheckEnvironment()
|
|
|
|
initInstallationInfo()
|
|
|
|
log.Info("OliveTin initialization complete")
|
|
}
|
|
|
|
func initLog() {
|
|
log.SetFormatter(&log.TextFormatter{
|
|
ForceQuote: true,
|
|
DisableTimestamp: true,
|
|
})
|
|
|
|
// Use debug this early on to catch details about startup errors. The
|
|
// default config will raise the log level later, if not set.
|
|
log.SetLevel(log.DebugLevel) // Default to debug, to catch cfg issue
|
|
}
|
|
|
|
func initCliFlags() string {
|
|
var configDir string
|
|
flag.StringVar(&configDir, "configdir", ".", "Config directory path")
|
|
|
|
var printVersion bool
|
|
flag.BoolVar(&printVersion, "version", false, "Prints the version number and exits")
|
|
flag.Parse()
|
|
|
|
// This log message should be the first log message OliveTin prints.
|
|
if printVersion {
|
|
logStartupMessage("OliveTin is just printing the startup message")
|
|
os.Exit(1)
|
|
} else {
|
|
logStartupMessage("OliveTin initializing")
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"value": configDir,
|
|
}).Debugf("Value of -configdir flag")
|
|
|
|
return configDir
|
|
}
|
|
|
|
func initViperConfig(configDir string) {
|
|
viper.AutomaticEnv()
|
|
viper.SetConfigName("config.yaml")
|
|
viper.SetConfigType("yaml")
|
|
viper.AddConfigPath(configDir)
|
|
viper.AddConfigPath("/config") // For containers.
|
|
viper.AddConfigPath("/etc/OliveTin/")
|
|
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
log.Errorf("Config file error at startup. %s", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
cfg = config.DefaultConfig()
|
|
|
|
viper.WatchConfig()
|
|
viper.OnConfigChange(func(e fsnotify.Event) {
|
|
if e.Op == fsnotify.Write {
|
|
log.Info("Config file changed:", e.String())
|
|
|
|
config.Reload(cfg)
|
|
}
|
|
})
|
|
|
|
config.Reload(cfg)
|
|
}
|
|
|
|
func initInstallationInfo() {
|
|
installationinfo.Config = cfg
|
|
installationinfo.Build.Version = version
|
|
installationinfo.Build.Commit = commit
|
|
installationinfo.Build.Date = date
|
|
}
|
|
|
|
func logStartupMessage(message string) {
|
|
log.WithFields(log.Fields{
|
|
"version": version,
|
|
"commit": commit,
|
|
"date": date,
|
|
}).Info(message)
|
|
}
|
|
|
|
func initCheckEnvironment() {
|
|
warnIfPuidGuid()
|
|
}
|
|
|
|
func warnIfPuidGuid() {
|
|
if os.Getenv("PUID") != "" || os.Getenv("PGID") != "" {
|
|
log.Warnf("PUID or PGID seem to be set to something, but they are ignored by OliveTin. Please check https://docs.olivetin.app/no-puid-pgid.html")
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
log.WithFields(log.Fields{
|
|
"configDir": cfg.GetDir(),
|
|
}).Infof("OliveTin started")
|
|
|
|
log.Debugf("Config: %+v", cfg)
|
|
|
|
executor := executor.DefaultExecutor(cfg)
|
|
executor.RebuildActionMap()
|
|
executor.AddListener(websocket.ExecutionListener)
|
|
config.AddListener(executor.RebuildActionMap)
|
|
|
|
go onstartup.Execute(cfg, executor)
|
|
go oncron.Schedule(cfg, executor)
|
|
go onfileindir.WatchFilesInDirectory(cfg, executor)
|
|
go oncalendarfile.Schedule(cfg, executor)
|
|
|
|
entityfiles.AddListener(websocket.OnEntityChanged)
|
|
entityfiles.AddListener(executor.RebuildActionMap)
|
|
go entityfiles.SetupEntityFileWatchers(cfg)
|
|
|
|
go updatecheck.StartUpdateChecker(version, commit, cfg, cfg.GetDir())
|
|
|
|
go grpcapi.Start(cfg, executor)
|
|
|
|
httpservers.StartServers(cfg)
|
|
}
|