live-streamer/main.go

75 lines
1.8 KiB
Go
Raw Normal View History

2024-10-22 16:39:10 -04:00
package main
import (
"live-streamer/config"
2024-10-23 10:47:37 -04:00
"live-streamer/server"
2024-10-22 16:39:10 -04:00
"live-streamer/streamer"
"live-streamer/utils"
2024-10-23 14:35:37 -04:00
"live-streamer/websocket"
2024-10-22 16:39:10 -04:00
"log"
"github.com/fsnotify/fsnotify"
)
var GlobalStreamer *streamer.Streamer
2024-10-23 14:35:37 -04:00
var outputer websocket.Outputer
2024-10-22 16:39:10 -04:00
func main() {
2024-10-23 14:35:37 -04:00
server.NewServer(":8080", websocketRequestHandler)
2024-10-23 10:47:37 -04:00
server.GlobalServer.Run()
2024-10-23 14:35:37 -04:00
outputer = server.GlobalServer
2024-10-22 16:39:10 -04:00
if !utils.HasFFMPEG() {
log.Fatal("ffmpeg not found")
}
2024-10-23 14:35:37 -04:00
GlobalStreamer = streamer.NewStreamer(config.GlobalConfig.VideoList, outputer)
2024-10-22 16:39:10 -04:00
go startWatcher()
GlobalStreamer.Stream()
GlobalStreamer.Close()
}
func startWatcher() {
watcher, err := fsnotify.NewWatcher()
if err != nil {
log.Fatalf("failed to create watcher: %v", err)
}
defer watcher.Close()
2024-10-23 05:11:52 -04:00
for _, item := range config.GlobalConfig.InputItems {
2024-10-22 16:39:10 -04:00
if item.ItemType == "dir" {
err = watcher.Add(item.Path)
if err != nil {
log.Fatalf("failed to add dir to watcher: %v", err)
}
2024-10-23 14:35:37 -04:00
log.Println("watching dir:", item.Path)
2024-10-22 16:39:10 -04:00
}
}
if err != nil {
log.Fatalf("failed to start watcher: %v", err)
}
for {
select {
case event, ok := <-watcher.Events:
if !ok {
return
}
if event.Op&fsnotify.Create == fsnotify.Create {
if utils.IsSupportedVideo(event.Name) {
2024-10-23 14:35:37 -04:00
log.Println("new video added:", event.Name)
2024-10-22 16:39:10 -04:00
GlobalStreamer.Add(event.Name)
2024-10-23 14:35:37 -04:00
server.GlobalServer.Broadcast(websocket.MakeResponse(websocket.TypeAddVideo, true, event.Name, ""))
2024-10-22 16:39:10 -04:00
}
}
if event.Op&fsnotify.Remove == fsnotify.Remove {
2024-10-23 14:35:37 -04:00
server.GlobalServer.Broadcast(websocket.MakeResponse(websocket.TypeRemoveVideo, true, event.Name, ""))
log.Println("video removed:", event.Name)
2024-10-22 16:39:10 -04:00
GlobalStreamer.Remove(event.Name)
}
case err, ok := <-watcher.Errors:
if !ok {
return
}
2024-10-23 14:35:37 -04:00
log.Println("watcher error:", err)
2024-10-22 16:39:10 -04:00
}
}
}