database

type DAL interface {
	NewSession(ctx context.Context) (DBSession, error)
	StoreSession(dbs DBSession, secret string, uid int64, durationSeconds int64, scope string, client string, ipAddr string) error
	DeleteSession(dbs DBSession, secret string) error
	GetSessions(dbs DBSession, uid int64) ([]*types.SessionInfo, error)
	GetSessionAuthInfo(dbs DBSession, secret string) (*types.SessionInfo, bool, error)
	RevokeSession(dbs DBSession, uid int64, sessionID int64) error

	SetClientSecret(dbs DBSession, clientID string, clientSecret string) error
	GetClientSecret(dbs DBSession, clientID string) (string, error)

	StoreDiscordUser(dbs DBSession, discordUser *types.DiscordUser) error
	GetDiscordUser(dbs DBSession, uid int64) (*types.DiscordUser, error)
	StoreDiscordServerRoles(dbs DBSession, roles []types.DiscordRole) error
	StoreDiscordUserRoles(dbs DBSession, uid int64, roles []int64) error
	GetDiscordUserRoles(dbs DBSession, uid int64) ([]string, error)

	StoreSubmission(dbs DBSession, submissionLevel string) (int64, error)
	// History/cache mutations require LockSubmissions before the transaction
	// performs any consistent read, unless the parent was created in that transaction.
	StoreSubmissionFile(dbs DBSession, s *types.SubmissionFile) (int64, error)
	GetSubmissionFiles(dbs DBSession, sfids []int64) ([]*types.SubmissionFile, error)
	GetExtendedSubmissionFilesBySubmissionID(dbs DBSession, sid int64) ([]*types.ExtendedSubmissionFile, error)
	GetSubmissionUploaderIDs(dbs DBSession, sid int64) ([]int64, error)

	SearchSubmissions(dbs DBSession, filter *types.SubmissionsFilter) ([]*types.ExtendedSubmission, int64, error)

	StoreCurationMeta(dbs DBSession, cm *types.CurationMeta) error
	GetCurationMetaBySubmissionFileID(dbs DBSession, sfid int64) (*types.CurationMeta, error)

	StoreComment(dbs DBSession, c *types.Comment) (int64, error)
	GetExtendedCommentsBySubmissionID(dbs DBSession, sid int64) ([]*types.ExtendedComment, error)
	GetCommentByID(dbs DBSession, cid int64) (*types.Comment, error)

	SoftDeleteSubmissionFile(dbs DBSession, sfid int64, deleteReason string) error
	SoftDeleteSubmission(dbs DBSession, sid int64, deleteReason string) error
	SoftDeleteComment(dbs DBSession, cid int64, deleteReason string) error

	StoreNotificationSettings(dbs DBSession, uid int64, actions []string) error
	GetNotificationSettingsByUserID(dbs DBSession, uid int64) ([]string, error)

	SubscribeUserToSubmission(dbs DBSession, uid, sid int64) error
	UnsubscribeUserFromSubmission(dbs DBSession, uid, sid int64) error
	IsUserSubscribedToSubmission(dbs DBSession, uid, sid int64) (bool, error)

	StoreNotification(dbs DBSession, msg, notificationType string) error
	GetUsersForNotification(dbs DBSession, authorID, sid int64, action string) ([]int64, error)
	GetUsersForUniversalNotification(dbs DBSession, authorID int64, action string) ([]int64, error)
	GetOldestUnsentNotification(dbs DBSession) (*types.Notification, error)
	MarkNotificationAsSent(dbs DBSession, nid int64) error

	StoreCurationImage(dbs DBSession, c *types.CurationImage) (int64, error)
	GetCurationImagesBySubmissionFileID(dbs DBSession, sfid int64) ([]*types.CurationImage, error)
	GetCurationImage(dbs DBSession, ciid int64) (*types.CurationImage, error)

	GetNextSubmission(dbs DBSession, sid int64) (int64, error)
	GetPreviousSubmission(dbs DBSession, sid int64) (int64, error)

	UpdateSubmissionCacheTable(dbs DBSession, sid int64) error
	ListSubmissionIDsForCacheRebuild(dbs DBSession, afterID int64, limit int) ([]int64, error)
	RebuildSubmissionCacheTable(dbs DBSession, sid int64) error

	StoreMasterDBGames(dbs DBSession, games []*types.MasterDatabaseGame) error

	GetAllSimilarityAttributes(dbs DBSession) ([]*types.SimilarityAttributes, error)

	DeleteUserSessions(dbs DBSession, uid int64) (int64, error)

	GetTotalCommentsCount(dbs DBSession) (int64, error)
	GetTotalUserCount(dbs DBSession) (int64, error)
	GetTotalSubmissionFilesize(dbs DBSession) (int64, error)

	GetUsers(dbs DBSession) ([]*types.User, error)
	GetSiteStatistics(dbs DBSession) (*types.StatisticsPageData, error)
	GetUserStatisticsAggregate(dbs DBSession, uid int64) ([]*types.UserStatistics, error)
	GetAllUserStatistics(dbs DBSession) ([]*types.UserStatistics, error)
	GetCommentsByUserIDAndAction(dbs DBSession, uid int64, action string) ([]*types.Comment, error)

	PopulateRevisionInfo(dbs DBSession, revisions []*types.RevisionInfo) error

	FreezeSubmission(dbs DBSession, sid int64) error
	UnfreezeSubmission(dbs DBSession, sid int64) error

	NukeSessionTable(dbs DBSession) error

	UpdateSubmissionAutofreeze(dbs DBSession, sid int64, shouldAutofreeze bool) error
}

type DBSession interface {
	Commit() error
	Rollback() error
	Tx() *sql.Tx
	Ctx() context.Context
}

type MysqlSession struct {
	// contains filtered or unexported fields
}

type PGDAL interface {
	Stat() *PostgresStats
	NewSession(ctx context.Context) (PGDBSession, error)

	CountSinceDate(dbs PGDBSession, modifiedAfter *string) (int, error)

	SearchTags(dbs PGDBSession, modifiedAfter *string) ([]*types.Tag, error)
	SearchPlatforms(dbs PGDBSession, modifiedAfter *string) ([]*types.Platform, error)
	SearchGames(dbs PGDBSession, modifiedAfter *string, modifiedBefore *string, broad bool, afterId *string) ([]*types.Game, []*types.AdditionalApp, []*types.GameData, [][]string, [][]string, error)
	SearchDeletedGames(dbs PGDBSession, modifiedAfter *string) ([]*types.DeletedGame, error)

	GetTagCategories(dbs PGDBSession) ([]*types.TagCategory, error)
	GetGamesUsingTagTotal(dbs PGDBSession, tagId int64) (int64, error)
	SaveGame(dbs PGDBSession, game *types.Game, uid int64) error
	SaveGameData(dbs PGDBSession, gameId string, date int64, gameData *types.GameData) error
	SaveTag(dbs PGDBSession, tag *types.Tag, uid int64) error
	DeveloperImportDatabaseJson(dbs PGDBSession, data *types.LauncherDump) error

	GetTagCategory(dbs PGDBSession, categoryId int64) (*types.TagCategory, error)
	GetTag(dbs PGDBSession, tagId int64) (*types.Tag, error)
	GetTagByName(dbs PGDBSession, tagName string) (*types.Tag, error)
	GetPlatform(dbs PGDBSession, platformId int64) (*types.Platform, error)
	GetPlatformByName(dbs PGDBSession, platformName string) (*types.Platform, error)
	GetGamesSlimInfo(dbs PGDBSession, gameIds []string) ([]*types.GameSlimInfo, error)
	GetGames(dbs PGDBSession, gameIds []string) ([]*types.Game, error)
	GetChangelogGame(dbs PGDBSession, gameId string, createdAt time.Time) (*types.Game, error)
	GetGame(dbs PGDBSession, gameId string) (*types.Game, error)
	GetGameData(dbs PGDBSession, gameId string, date int64) (*types.GameData, error)
	GetGameDataIndex(dbs PGDBSession, gameId string, date int64) (*types.GameDataIndex, error)
	GetGameRevisionInfo(dbs PGDBSession, gameId string) ([]*types.RevisionInfo, error)
	GetTagRevisionInfo(dbs PGDBSession, tagId int64) ([]*types.RevisionInfo, error)

	GetMetadataStats(dbs PGDBSession) (*types.MetadataStatsPageDataBare, error)

	DeleteGame(dbs PGDBSession, gameId string, uid int64, reason string, imagesPath string, gamesPath string,
		deletedImagesPath string, deletedGamesPath string, frozenGamesPath string) error

	RestoreGame(dbs PGDBSession, gameId string, uid int64, reason string, imagesPath string, gamesPath string,
		deletedImagesPath string, deletedGamesPath string, frozenGamesPath string) error

	GetOrCreateTagCategory(dbs PGDBSession, categoryName string) (*types.TagCategory, error)
	GetOrCreateTag(dbs PGDBSession, tagName string, tagCategory string, reason string, uid int64) (*types.Tag, error)
	GetOrCreatePlatform(dbs PGDBSession, platformName string, reason string, uid int64) (*types.Platform, error)

	AddSubmissionFromValidator(dbs PGDBSession, uid int64, vr *types.ValidatorRepackResponse, frozen bool) (*types.Game, error)
	AddGameData(dbs PGDBSession, uid int64, gameId string, vr *types.ValidatorRepackResponse) (*types.GameData, error)

	IndexerGetNext(ctx context.Context) (*types.GameData, error)
	IndexerInsert(ctx context.Context, crc32sum []byte, md5sum []byte, sha256sum []byte, sha1sum []byte,
		size uint64, path string, gameId string, zipDate time.Time) error
	IndexerMarkFailure(ctx context.Context, gameId string, zipDate time.Time) error
	IndexerMarkEmpty(ctx context.Context, gameId string, zipDate time.Time) error

	GetIndexMatchesHash(dbs PGDBSession, hashType string, hashStr string) ([]*types.IndexMatchData, error)
	GetIndexMatchesPath(dbs PGDBSession, paths []string) ([]*types.IndexMatchData, error)

	UpdateTagsFromTagsList(dbs PGDBSession, tagsList []types.Tag) error
	ApplyGamePatch(dbs PGDBSession, uid int64, game *types.Game, patch *types.GameContentPatch, addApps []*types.CurationAdditionalApp) error

	GetGameRedirectTo(dbs PGDBSession, gameId string) (string, error)
	GetGameRedirects(dbs PGDBSession) ([]*types.GameRedirect, error)
	AddGameRedirect(dbs PGDBSession, srcId string, destId string) error
	RemoveGameRedirectsTo(dbs PGDBSession, srcId string) error
	RemoveGameRedirectsFrom(dbs PGDBSession, srcId string) error
	UpdateGameRedirects(dbs PGDBSession, srcId string, destId string) error

	CreateActivityEvent(dbs PGDBSession, event *activityevents.ActivityEvent) error
	GetActivityEvents(dbs PGDBSession, filter *types.ActivityEventsFilter) ([]*activityevents.ActivityEvent, error)
	GetLatestSubmissionActivity(dbs PGDBSession, userIDs []int64) (map[int64]time.Time, error)

	GetFrozenGames(dbs PGDBSession) ([]*types.AutounfreezerGame, error)
}

type PGDBSession interface {
	Commit() error
	Rollback() error
	Tx() pgx.Tx
	Ctx() context.Context
}

type PostgresConfig struct {
	MaxConns          int32         `json:"MaxConns"`
	MinConns          int32         `json:"MinConns"`
	MaxConnLifetime   time.Duration `json:"MaxConnLifetime"`
	MaxConnIdleTime   time.Duration `json:"MaxConnIdleTime"`
	HealthCheckPeriod time.Duration `json:"HealthCheckPeriod"`
}

type PostgresSession struct {
	// contains filtered or unexported fields
}

type PostgresStats struct {
	AcquireCount            int64           `json:"AcquireCount"`
	CancelledAcquireCount   int64           `json:"CancelledAcquireCount"`
	EmptyAcquireCount       int64           `json:"EmptyAcquireCount"`
	MaxLifetimeDestroyCount int64           `json:"MaxLifetimeDestroyCount"`
	MaxIdleDestroyCount     int64           `json:"MaxIdleDestroyCount"`
	AcquiredConns           int32           `json:"AcquiredConns"`
	ConstructingConns       int32           `json:"ConstructingConns"`
	IdleConns               int32           `json:"IdleConns"`
	TotalConns              int32           `json:"TotalConns"`
	Config                  *PostgresConfig `json:"config"`
}