package main import ( "flag" "os" "strconv" "github.com/owncast/owncast/logging" log "github.com/sirupsen/logrus" "github.com/owncast/owncast/config" "github.com/owncast/owncast/core" "github.com/owncast/owncast/core/data" "github.com/owncast/owncast/metrics" "github.com/owncast/owncast/router" "github.com/owncast/owncast/utils" ) var ( dbFile = flag.String("database", "", "Path to the database file.") logDirectory = flag.String("logdir", "", "Directory where logs will be written to") backupDirectory = flag.String("backupdir", "", "Directory where backups will be written to") enableDebugOptions = flag.Bool("enableDebugFeatures", false, "Enable additional debugging options.") enableVerboseLogging = flag.Bool("enableVerboseLogging", false, "Enable additional logging.") restoreDatabaseFile = flag.String("restoreDatabase", "", "Restore an Owncast database backup") newAdminPassword = flag.String("adminpassword", "", "Set your admin password") newStreamKey = flag.String("streamkey", "", "Set a temporary stream key for this session") webServerPortOverride = flag.String("webserverport", "", "Force the web server to listen on a specific port") webServerIPOverride = flag.String("webserverip", "", "Force web server to listen on this IP address") rtmpPortOverride = flag.Int("rtmpport", 0, "Set listen port for the RTMP server") ) // nolint:cyclop func main() { flag.Parse() if *logDirectory != "" { config.LogDirectory = *logDirectory } if *backupDirectory != "" { config.BackupDirectory = *backupDirectory } // Create the data directory if needed if !utils.DoesFileExists("data") { if err := os.Mkdir("./data", 0o700); err != nil { log.Fatalln("Cannot create data directory", err) } } // Migrate old (pre 0.1.0) emoji to new location if they exist. utils.MigrateCustomEmojiLocations() // Otherwise save the default emoji to the data directory. if err := data.SetupEmojiDirectory(); err != nil { log.Fatalln("Cannot set up emoji directory", err) } // Recreate the temp dir if utils.DoesFileExists(config.TempDir) { err := os.RemoveAll(config.TempDir) if err != nil { log.Fatalln("Unable to remove temp dir! Check permissions.", config.TempDir, err) } } if err := os.Mkdir(config.TempDir, 0o700); err != nil { log.Fatalln("Unable to create temp dir!", err) } configureLogging(*enableDebugOptions, *enableVerboseLogging) log.Infoln(config.GetReleaseString()) // Allows a user to restore a specific database backup if *restoreDatabaseFile != "" { databaseFile := config.DatabaseFilePath if *dbFile != "" { databaseFile = *dbFile } if err := utils.Restore(*restoreDatabaseFile, databaseFile); err != nil { log.Fatalln(err) } log.Println("Database has been restored. Restart Owncast.") log.Exit(0) } config.EnableDebugFeatures = *enableDebugOptions if *dbFile != "" { config.DatabaseFilePath = *dbFile } if err := data.SetupPersistence(config.DatabaseFilePath); err != nil { log.Fatalln("failed to open database", err) } handleCommandLineFlags() // starts the core if err := core.Start(); err != nil { log.Fatalln("failed to start the core package", err) } go metrics.Start(core.GetStatus) if err := router.Start(); err != nil { log.Fatalln("failed to start/run the router", err) } } func handleCommandLineFlags() { if *newAdminPassword != "" { if err := data.SetAdminPassword(*newAdminPassword); err != nil { log.Errorln("Error setting your admin password.", err) log.Exit(1) } else { log.Infoln("Admin password changed") } } if *newStreamKey != "" { log.Println("Temporary stream key is set for this session.") config.TemporaryStreamKey = *newStreamKey } // Set the web server port if *webServerPortOverride != "" { portNumber, err := strconv.Atoi(*webServerPortOverride) if err != nil { log.Warnln(err) return } log.Println("Saving new web server port number to", portNumber) if err := data.SetHTTPPortNumber(float64(portNumber)); err != nil { log.Errorln(err) } } config.WebServerPort = data.GetHTTPPortNumber() // Set the web server ip if *webServerIPOverride != "" { log.Println("Saving new web server listen IP address to", *webServerIPOverride) if err := data.SetHTTPListenAddress(*webServerIPOverride); err != nil { log.Errorln(err) } } config.WebServerIP = data.GetHTTPListenAddress() // Set the rtmp server port if *rtmpPortOverride > 0 { log.Println("Saving new RTMP server port number to", *rtmpPortOverride) if err := data.SetRTMPPortNumber(float64(*rtmpPortOverride)); err != nil { log.Errorln(err) } } } func configureLogging(enableDebugFeatures bool, enableVerboseLogging bool) { logging.Setup(enableDebugFeatures, enableVerboseLogging) log.SetFormatter(&log.TextFormatter{ FullTimestamp: true, }) }