mirror of
https://github.com/owncast/owncast.git
synced 2024-10-24 05:25:58 +03:00
119 lines
3.4 KiB
Go
119 lines
3.4 KiB
Go
package core
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/grafov/m3u8"
|
|
"github.com/owncast/owncast/config"
|
|
"github.com/owncast/owncast/static"
|
|
"github.com/owncast/owncast/utils"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func appendOfflineToVariantPlaylist(index int, playlistFilePath string) {
|
|
f, err := os.OpenFile(playlistFilePath, os.O_CREATE|os.O_RDWR, os.ModePerm) //nolint
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
|
|
playlist, _, err := m3u8.DecodeFrom(bufio.NewReader(f), true)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
|
|
if err := f.Close(); err != nil {
|
|
log.Errorln("error closing playlist file", err)
|
|
}
|
|
|
|
variantPlaylist := playlist.(*m3u8.MediaPlaylist)
|
|
variantPlaylist.MediaType = m3u8.EVENT
|
|
|
|
tmpFileName := fmt.Sprintf("tmp-stream-%d.m3u8", index)
|
|
atomicWriteTmpPlaylistFile, err := os.CreateTemp(os.TempDir(), tmpFileName)
|
|
if err != nil {
|
|
log.Errorln("error creating tmp playlist file to write to", playlistFilePath, err)
|
|
return
|
|
}
|
|
|
|
if _, err := atomicWriteTmpPlaylistFile.Write(variantPlaylist.Encode().Bytes()); err != nil {
|
|
log.Errorln(err)
|
|
}
|
|
|
|
// Manually add the offline clip to the end of the media playlist.
|
|
_, _ = atomicWriteTmpPlaylistFile.WriteString("#EXT-X-DISCONTINUITY\n")
|
|
// If "offline" content gets changed then change the duration below
|
|
_, _ = atomicWriteTmpPlaylistFile.WriteString("#EXTINF:8.000000,\n")
|
|
_, _ = atomicWriteTmpPlaylistFile.WriteString("offline.ts\n")
|
|
_, _ = atomicWriteTmpPlaylistFile.WriteString("#EXT-X-ENDLIST\n")
|
|
|
|
if err := atomicWriteTmpPlaylistFile.Close(); err != nil {
|
|
log.Errorln(err)
|
|
}
|
|
|
|
if err := utils.Move(atomicWriteTmpPlaylistFile.Name(), playlistFilePath); err != nil {
|
|
log.Errorln("error moving temp playlist to overwrite existing one", err)
|
|
}
|
|
}
|
|
|
|
func makeVariantIndexOffline(index int, offlineFilePath string, offlineFilename string) {
|
|
playlistFilePath := fmt.Sprintf(filepath.Join(config.HLSStoragePath, "%d/stream.m3u8"), index)
|
|
segmentFilePath := fmt.Sprintf(filepath.Join(config.HLSStoragePath, "%d/%s"), index, offlineFilename)
|
|
|
|
if err := utils.Copy(offlineFilePath, segmentFilePath); err != nil {
|
|
log.Warnln(err)
|
|
}
|
|
|
|
if _, err := _storage.Save(segmentFilePath, 0); err != nil {
|
|
log.Warnln(err)
|
|
}
|
|
|
|
if utils.DoesFileExists(playlistFilePath) {
|
|
appendOfflineToVariantPlaylist(index, playlistFilePath)
|
|
} else {
|
|
createEmptyOfflinePlaylist(playlistFilePath, offlineFilename)
|
|
}
|
|
if _, err := _storage.Save(playlistFilePath, 0); err != nil {
|
|
log.Warnln(err)
|
|
}
|
|
}
|
|
|
|
func createEmptyOfflinePlaylist(playlistFilePath string, offlineFilename string) {
|
|
p, err := m3u8.NewMediaPlaylist(1, 1)
|
|
if err != nil {
|
|
log.Errorln(err)
|
|
}
|
|
|
|
// If "offline" content gets changed then change the duration below
|
|
if err := p.Append(offlineFilename, 8.0, ""); err != nil {
|
|
log.Errorln(err)
|
|
}
|
|
|
|
p.Close()
|
|
f, err := os.Create(playlistFilePath)
|
|
if err != nil {
|
|
log.Errorln(err)
|
|
}
|
|
defer f.Close()
|
|
if _, err := f.Write(p.Encode().Bytes()); err != nil {
|
|
log.Errorln(err)
|
|
}
|
|
}
|
|
|
|
func saveOfflineClipToDisk(offlineFilename string) (string, error) {
|
|
offlineFileData := static.GetOfflineSegment()
|
|
offlineTmpFile, err := os.CreateTemp(os.TempDir(), offlineFilename)
|
|
if err != nil {
|
|
log.Errorln("unable to create temp file for offline video segment", err)
|
|
}
|
|
|
|
if _, err = offlineTmpFile.Write(offlineFileData); err != nil {
|
|
return "", fmt.Errorf("unable to write offline segment to disk: %s", err)
|
|
}
|
|
|
|
offlineFilePath := offlineTmpFile.Name()
|
|
|
|
return offlineFilePath, nil
|
|
}
|