65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"log"
|
||
|
|
"path/filepath"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/radovskyb/watcher"
|
||
|
|
)
|
||
|
|
|
||
|
|
var filesToUpload = make(map[string]string)
|
||
|
|
|
||
|
|
func monitorVideoContent(path string) {
|
||
|
|
w := watcher.New()
|
||
|
|
|
||
|
|
go func() {
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case event := <-w.Event:
|
||
|
|
if event.Op != watcher.Write {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
if filepath.Base(event.Path) == "temp.m3u8" {
|
||
|
|
// fmt.Printf("Upload playlist + %d files\n", len(filesToUpload))
|
||
|
|
|
||
|
|
for filePath, objectID := range filesToUpload {
|
||
|
|
|
||
|
|
if objectID != "" {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
newObjectPath := save("hls/" + filePath)
|
||
|
|
fmt.Println(filePath, newObjectPath)
|
||
|
|
|
||
|
|
filesToUpload[filePath] = newObjectPath
|
||
|
|
}
|
||
|
|
|
||
|
|
playlistBytes, err := ioutil.ReadFile(event.Path)
|
||
|
|
verifyError(err)
|
||
|
|
playlistString := string(playlistBytes)
|
||
|
|
remotePlaylistString := generateRemotePlaylist(playlistString, filesToUpload)
|
||
|
|
writePlaylist(remotePlaylistString, "webroot/stream.m3u8")
|
||
|
|
} else if filepath.Ext(event.Path) == ".ts" {
|
||
|
|
filesToUpload[filepath.Base(event.Path)] = ""
|
||
|
|
// copy(event.Path, "webroot/"+filepath.Base(event.Path))
|
||
|
|
}
|
||
|
|
case err := <-w.Error:
|
||
|
|
log.Fatalln(err)
|
||
|
|
case <-w.Closed:
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
// Watch this folder for changes.
|
||
|
|
if err := w.Add(path); err != nil {
|
||
|
|
log.Fatalln(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := w.Start(time.Millisecond * 500); err != nil {
|
||
|
|
log.Fatalln(err)
|
||
|
|
}
|
||
|
|
}
|
