owncast/playlistMonitor.go

package main

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

	log "github.com/sirupsen/logrus"

	icore "github.com/ipfs/interface-go-ipfs-core"
	"github.com/radovskyb/watcher"
)

var filesToUpload = make(map[string]string)

func monitorVideoContent(pathToMonitor string, configuration Config, ipfs *icore.CoreAPI) {
	log.Printf("Using %s for IPFS 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" {

					if configuration.IPFS.Enabled {
						for filePath, objectID := range filesToUpload {
							if objectID != "" {
								continue
							}

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

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

					if configuration.IPFS.Enabled {
						playlistString = generateRemotePlaylist(playlistString, configuration.IPFS.Gateway, filesToUpload)
					}
					writePlaylist(playlistString, path.Join(configuration.PublicHLSPath, "/stream.m3u8"))

				} else if filepath.Ext(event.Path) == ".ts" {
					if configuration.IPFS.Enabled {
						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)
	}
}