owncast/playlistMonitor.go

package main

import (
	"io/ioutil"
	"path"
	"path/filepath"
	"time"

	log "github.com/sirupsen/logrus"

	"github.com/radovskyb/watcher"
)

var filesToUpload = make(map[string]string)

func monitorVideoContent(pathToMonitor string, configuration Config, storage ChunkStorage) {
	log.Printf("Using %s files...\n", pathToMonitor)

	w := watcher.New()

	go func() {
		for {
			select {
			case event := <-w.Event:
				if event.Op != watcher.Write {
					continue
				}
				if filepath.Base(event.Path) == "temp.m3u8" {

					for filePath, objectID := range filesToUpload {
						if objectID != "" {
							continue
						}

						newObjectPath := storage.Save(path.Join(configuration.PrivateHLSPath, filePath))
						filesToUpload[filePath] = newObjectPath
					}

					playlistBytes, err := ioutil.ReadFile(event.Path)
					verifyError(err)
					playlistString := string(playlistBytes)

					playlistString = storage.GenerateRemotePlaylist(playlistString, filesToUpload)
					writePlaylist(playlistString, path.Join(configuration.PublicHLSPath, "/stream.m3u8"))

				} else if filepath.Ext(event.Path) == ".ts" {
					filesToUpload[filepath.Base(event.Path)] = ""
				}
			case err := <-w.Error:
				log.Fatalln(err)
			case <-w.Closed:
				return
			}
		}
	}()

	// Watch this folder for changes.
	if err := w.Add(pathToMonitor); err != nil {
		log.Fatalln(err)
	}

	if err := w.Start(time.Millisecond * 100); err != nil {
		log.Fatalln(err)
	}
}