owncast/main.go

package main

import (
	"encoding/json"
	"net/http"
	"strconv"

	icore "github.com/ipfs/interface-go-ipfs-core"
	log "github.com/sirupsen/logrus"
)

var ipfs icore.CoreAPI
var configuration = getConfig()
var server *Server

var online = false

func main() {
	checkConfig(configuration)
	// resetDirectories()

	var hlsDirectoryPath = configuration.PublicHLSPath

	log.Println("Starting up.  Please wait...")

	if configuration.IPFS.Enabled {
		hlsDirectoryPath = configuration.PrivateHLSPath
		enableIPFS()
		go monitorVideoContent(hlsDirectoryPath, configuration, &ipfs)
	}

	go startChatServer()

	startRTMPService()
}

func enableIPFS() {
	log.Println("Enabling IPFS support...")

	ipfsInstance, node, _ := createIPFSInstance()
	ipfs = *ipfsInstance

	createIPFSDirectory(ipfsInstance, "./hls")
	go startIPFSNode(ipfs, node)
}

func startChatServer() {
	// log.SetFlags(log.Lshortfile)

	// websocket server
	server = NewServer("/entry")
	go server.Listen()

	// static files
	http.Handle("/", http.FileServer(http.Dir("webroot")))
	http.HandleFunc("/status", getStatus)

	log.Printf("Starting public web server on port %d", configuration.WebServerPort)

	log.Fatal(http.ListenAndServe(":"+strconv.Itoa(configuration.WebServerPort), nil))
}

func getStatus(w http.ResponseWriter, r *http.Request) {
	status := Status{
		Online:      online,
		ViewerCount: server.ClientCount(),
	}
	json.NewEncoder(w).Encode(status)
}

func streamConnected() {
	online = true
}

func streamDisconnected() {
	online = false
}

func viewerAdded() {
}

func viewerRemoved() {
}