owncast/core/storageproviders/local.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 {
			select {
			case <-_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) {
	s.Save(localFilePath, 0)
}

// VariantPlaylistWritten is called when a variant hls playlist is written
func (s *LocalStorage) VariantPlaylistWritten(localFilePath string) {
	_, error := s.Save(localFilePath, 0)
	if error != nil {
		log.Errorln(error)
		return
	}
}

// MasterPlaylistWritten is called when the master hls playlist is written
func (s *LocalStorage) MasterPlaylistWritten(localFilePath string) {
	s.Save(localFilePath, 0)
}

// 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)
	}

	utils.Copy(filePath, newPath)

	return newPath, nil
}