owncast/storage/configrepository/configrepository.go

package configrepository

import (
	"time"

	"github.com/owncast/owncast/models"
	"github.com/owncast/owncast/services/config"
	"github.com/owncast/owncast/storage/data"
	"github.com/owncast/owncast/utils"
	log "github.com/sirupsen/logrus"
)

type ConfigRepository interface {
	GetSuggestedUsernamesList() []string
	GetExtraPageBodyContent() string
	SetExtraPageBodyContent(content string) error
	GetStreamTitle() string
	SetStreamTitle(title string) error
	GetAdminPassword() string
	SetAdminPassword(key string) error
	GetLogoPath() string
	SetLogoPath(logo string) error
	SetLogoUniquenessString(uniqueness string) error
	GetLogoUniquenessString() string
	GetServerSummary() string
	SetServerSummary(summary string) error
	GetServerWelcomeMessage() string
	SetServerWelcomeMessage(welcomeMessage string) error
	GetServerName() string
	SetServerName(name string) error
	GetServerURL() string
	SetServerURL(url string) error
	GetHTTPPortNumber() int
	SetWebsocketOverrideHost(host string) error
	GetWebsocketOverrideHost() string
	SetHTTPPortNumber(port float64) error
	GetHTTPListenAddress() string
	SetHTTPListenAddress(address string) error
	GetRTMPPortNumber() int
	SetRTMPPortNumber(port float64) error
	GetServerMetadataTags() []string
	SetServerMetadataTags(tags []string) error
	GetDirectoryEnabled() bool
	SetDirectoryEnabled(enabled bool) error
	SetDirectoryRegistrationKey(key string) error
	GetDirectoryRegistrationKey() string
	GetSocialHandles() []models.SocialHandle
	SetSocialHandles(socialHandles []models.SocialHandle) error
	GetPeakSessionViewerCount() int
	SetPeakSessionViewerCount(count int) error
	GetPeakOverallViewerCount() int
	SetPeakOverallViewerCount(count int) error
	GetLastDisconnectTime() (*utils.NullTime, error)
	SetLastDisconnectTime(disconnectTime time.Time) error
	SetNSFW(isNSFW bool) error
	GetNSFW() bool
	SetFfmpegPath(path string) error
	GetFfMpegPath() string
	GetS3Config() models.S3
	SetS3Config(config models.S3) error
	GetStreamLatencyLevel() models.LatencyLevel
	SetStreamLatencyLevel(level float64) error
	GetStreamOutputVariants() []models.StreamOutputVariant
	SetStreamOutputVariants(variants []models.StreamOutputVariant) error
	SetChatDisabled(disabled bool) error
	GetChatDisabled() bool
	SetChatEstablishedUsersOnlyMode(enabled bool) error
	GetChatEstbalishedUsersOnlyMode() bool
	GetExternalActions() []models.ExternalAction
	SetExternalActions(actions []models.ExternalAction) error
	SetCustomStyles(styles string) error
	GetCustomStyles() string
	SetCustomJavascript(styles string) error
	GetCustomJavascript() string
	SetVideoCodec(codec string) error
	GetVideoCodec() string
	VerifySettings() error
	FindHighestVideoQualityIndex(qualities []models.StreamOutputVariant) int
	GetForbiddenUsernameList() []string
	SetForbiddenUsernameList(usernames []string) error
	SetSuggestedUsernamesList(usernames []string) error
	GetServerInitTime() (*utils.NullTime, error)
	SetServerInitTime(t time.Time) error
	SetFederationEnabled(enabled bool) error
	GetFederationEnabled() bool
	SetFederationUsername(username string) error
	GetFederationUsername() string
	SetFederationGoLiveMessage(message string) error
	GetFederationGoLiveMessage() string
	SetFederationIsPrivate(isPrivate bool) error
	GetFederationIsPrivate() bool
	SetFederationShowEngagement(showEngagement bool) error
	GetFederationShowEngagement() bool
	SetBlockedFederatedDomains(domains []string) error
	GetBlockedFederatedDomains() []string
	SetChatJoinPartMessagesEnabled(enabled bool) error
	GetChatJoinPartMessagesEnabled() bool
	SetNotificationsEnabled(enabled bool) error
	GetNotificationsEnabled() bool
	GetDiscordConfig() models.DiscordConfiguration
	SetDiscordConfig(config models.DiscordConfiguration) error
	GetBrowserPushConfig() models.BrowserNotificationConfiguration
	SetBrowserPushConfig(config models.BrowserNotificationConfiguration) error
	SetBrowserPushPublicKey(key string) error
	GetBrowserPushPublicKey() (string, error)
	SetBrowserPushPrivateKey(key string) error
	GetBrowserPushPrivateKey() (string, error)
	SetHasPerformedInitialNotificationsConfig(hasConfigured bool) error
	GetHasPerformedInitialNotificationsConfig() bool
	GetHideViewerCount() bool
	SetHideViewerCount(hide bool) error
	GetCustomOfflineMessage() string
	SetCustomOfflineMessage(message string) error
	SetCustomColorVariableValues(variables map[string]string) error
	GetCustomColorVariableValues() map[string]string
	GetStreamKeys() []models.StreamKey
	SetStreamKeys(actions []models.StreamKey) error
	SetDisableSearchIndexing(disableSearchIndexing bool) error
	GetDisableSearchIndexing() bool
	GetVideoServingEndpoint() string
	SetVideoServingEndpoint(message string) error
	GetDefaultFederationUsername() string
}

type SqlConfigRepository struct {
	datastore *data.Store
}

// New will create a new config repository.
func New(datastore *data.Store) *SqlConfigRepository {
	r := SqlConfigRepository{
		datastore: datastore,
	}
	temporaryGlobalInstance = &r
	return &r
}

// NOTE: This is temporary during the transition period.
var temporaryGlobalInstance *SqlConfigRepository

// GetUserRepository will return the temporary repository singleton.
func Get() *SqlConfigRepository {
	if temporaryGlobalInstance == nil {
		i := New(data.GetDatastore())
		temporaryGlobalInstance = i
	}
	return temporaryGlobalInstance
}

// Setup will create the datastore table and perform initial initialization.
func (cr *SqlConfigRepository) Setup() {
	if !cr.HasPopulatedDefaults() {
		cr.PopulateDefaults()
	}

	if !cr.hasPopulatedFederationDefaults() {
		if err := cr.SetFederationGoLiveMessage(config.GetDefaults().FederationGoLiveMessage); err != nil {
			log.Errorln(err)
		}
		if err := cr.datastore.SetBool("HAS_POPULATED_FEDERATION_DEFAULTS", true); err != nil {
			log.Errorln(err)
		}
	}

	// Set the server initialization date if needed.
	if hasSetInitDate, _ := cr.GetServerInitTime(); hasSetInitDate == nil || !hasSetInitDate.Valid {
		_ = cr.SetServerInitTime(time.Now())
	}
}