123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169 |
- 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")
- enableReplayFeatures = flag.Bool("enableReplayFeatures", false, "Enable experimental replay features")
- )
- // nolint:cyclop
- func main() {
- flag.Parse()
- if *logDirectory != "" {
- config.LogDirectory = *logDirectory
- }
- if *backupDirectory != "" {
- config.BackupDirectory = *backupDirectory
- }
- config.EnableReplayFeatures = *enableReplayFeatures
- // 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,
- })
- }
|