4d2066a76d
* Start cleaning up linter errors. For #357 * Fix unmarshalling NullTime values * More linter fixes * Remove commented code * Move defer up * Consolidate error check lines * Move error check to make sure row iteration was successful * Cleaner error check + do not recreate pipe if it exists * Consolidate hashing to generate client id
72 lines
1.8 KiB
Go
72 lines
1.8 KiB
Go
package storageproviders
|
|
|
|
import (
|
|
"path/filepath"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/owncast/owncast/config"
|
|
"github.com/owncast/owncast/core/ffmpeg"
|
|
"github.com/owncast/owncast/utils"
|
|
)
|
|
|
|
type LocalStorage struct {
|
|
}
|
|
|
|
// Cleanup old public HLS content every N min from the webroot.
|
|
var _onlineCleanupTicker *time.Ticker
|
|
|
|
// Setup configures this storage provider.
|
|
func (s *LocalStorage) Setup() error {
|
|
// NOTE: This cleanup timer will have to be disabled to support recordings in the future
|
|
// as all HLS segments have to be publicly available on disk to keep a recording of them.
|
|
_onlineCleanupTicker = time.NewTicker(1 * time.Minute)
|
|
go func() {
|
|
for range _onlineCleanupTicker.C {
|
|
ffmpeg.CleanupOldContent(config.PublicHLSStoragePath)
|
|
}
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
// SegmentWritten is called when a single segment of video is written.
|
|
func (s *LocalStorage) SegmentWritten(localFilePath string) {
|
|
_, err := s.Save(localFilePath, 0)
|
|
if err != nil {
|
|
log.Warnln(err)
|
|
}
|
|
}
|
|
|
|
// VariantPlaylistWritten is called when a variant hls playlist is written.
|
|
func (s *LocalStorage) VariantPlaylistWritten(localFilePath string) {
|
|
_, err := s.Save(localFilePath, 0)
|
|
if err != nil {
|
|
log.Errorln(err)
|
|
return
|
|
}
|
|
}
|
|
|
|
// MasterPlaylistWritten is called when the master hls playlist is written.
|
|
func (s *LocalStorage) MasterPlaylistWritten(localFilePath string) {
|
|
_, err := s.Save(localFilePath, 0)
|
|
if err != nil {
|
|
log.Warnln(err)
|
|
}
|
|
}
|
|
|
|
// Save will save a local filepath using the storage provider.
|
|
func (s *LocalStorage) Save(filePath string, retryCount int) (string, error) {
|
|
newPath := ""
|
|
|
|
// This is a hack
|
|
if filePath == "hls/stream.m3u8" {
|
|
newPath = filepath.Join(config.PublicHLSStoragePath, filepath.Base(filePath))
|
|
} else {
|
|
newPath = filepath.Join(config.WebRoot, filePath)
|
|
}
|
|
|
|
err := utils.Copy(filePath, newPath)
|
|
return newPath, err
|
|
}
|