123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138 |
- 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.")
- var logDirectory = flag.String("logdir", "", "Directory where logs will be written to")
- var backupDirectory = flag.String("backupdir", "", "Directory where backups will be written to")
- var enableDebugOptions = flag.Bool("enableDebugFeatures", false, "Enable additional debugging options.")
- var enableVerboseLogging = flag.Bool("enableVerboseLogging", false, "Enable additional logging.")
- var restoreDatabaseFile = flag.String("restoreDatabase", "", "Restore an Owncast database backup")
- var newStreamKey = flag.String("streamkey", "", "Set your stream key/admin password")
- var webServerPortOverride = flag.String("webserverport", "", "Force the web server to listen on a specific port")
- var webServerIPOverride = flag.String("webserverip", "", "Force web server to listen on this IP address")
- var rtmpPortOverride = flag.Int("rtmpport", 0, "Set listen port for the RTMP server")
- 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", 0700); err != nil {
- log.Fatalln("Cannot create data directory", 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
- }
- go metrics.Start()
- 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)
- }
- if err := router.Start(); err != nil {
- log.Fatalln("failed to start/run the router", err)
- }
- }
- func handleCommandLineFlags() {
- if *newStreamKey != "" {
- if err := data.SetStreamKey(*newStreamKey); err != nil {
- log.Errorln("Error setting your stream key.", err)
- log.Exit(1)
- } else {
- log.Infoln("Stream key changed to", *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,
- })
- }
|