You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
3.3 KiB
109 lines
3.3 KiB
package core |
|
|
|
import ( |
|
"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) { |
|
existingPlaylistContents, err := os.ReadFile(playlistFilePath) // nolint: gosec |
|
if err != nil { |
|
log.Debugln("unable to read existing playlist file", err) |
|
return |
|
} |
|
|
|
tmpFileName := fmt.Sprintf("tmp-stream-%d.m3u8", index) |
|
atomicWriteTmpPlaylistFile, err := os.CreateTemp(config.TempDir, tmpFileName) |
|
if err != nil { |
|
log.Errorln("error creating tmp playlist file to write to", playlistFilePath, err) |
|
return |
|
} |
|
|
|
// Write the existing playlist contents |
|
if _, err := atomicWriteTmpPlaylistFile.Write(existingPlaylistContents); err != nil { |
|
log.Debugln("error writing existing playlist contents to tmp playlist file", err) |
|
return |
|
} |
|
|
|
// Manually append 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) //nolint:gosec |
|
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(config.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 |
|
}
|
|
|