repository

type AssistantMessageRegenerationRepository interface {
	Create(ctx context.Context, regen *domain.AssistantMessageRegeneration) error

	ListByMessageID(ctx context.Context, messageID int64, limit int32) ([]*domain.AssistantMessageRegeneration, error)
}

type AuthRepos struct {
	User  UserRepository
	Token TokenRepository
}

type AuthTransactionRunner interface {
	WithinTx(ctx context.Context, fn func(ctx context.Context, r AuthRepos) error) error
}

type ChatPreferenceRepository interface {
	GetSelectedRunner(ctx context.Context, userID int) (string, error)

	SetSelectedRunner(ctx context.Context, userID int, runner string) error
}

type ChatRepos struct {
	Session         ChatSessionRepository
	Preference      ChatPreferenceRepository
	SessionSettings ChatSessionSettingsRepository
	Message         MessageRepository
	MessageEdit     MessageEditRepository
	AssistantRegen  AssistantMessageRegenerationRepository
	File            FileRepository
}

type ChatSessionRepository interface {
	Create(ctx context.Context, session *domain.ChatSession) error

	GetById(ctx context.Context, id int64) (*domain.ChatSession, error)

	GetByUserId(ctx context.Context, userID int, page, pageSize int32) ([]*domain.ChatSession, int32, error)

	Update(ctx context.Context, session *domain.ChatSession) error

	Delete(ctx context.Context, id int64) error
}

type ChatSessionSettingsRepository interface {
	GetBySessionID(ctx context.Context, sessionID int64) (*domain.ChatSessionSettings, error)

	Upsert(ctx context.Context, settings *domain.ChatSessionSettings) error
}

type ChatTransactionRunner interface {
	WithinTx(ctx context.Context, fn func(ctx context.Context, r ChatRepos) error) error
}

type DocumentRAGRepository interface {
	GetFileIndex(ctx context.Context, fileID int64) (*domain.FileRAGIndex, error)

	SaveFileRAGIndex(ctx context.Context, row *domain.FileRAGIndex) error

	MarkFileRAGIndexFailed(ctx context.Context, fileID int64, errMsg string) error

	ReplaceFileChunks(ctx context.Context, sessionID int64, userID int, fileID int64, pipelineVersion, embeddingModel, sourceSHA256 string, chunks []domain.DocumentRAGChunk) error

	DeleteIndexForFile(ctx context.Context, fileID int64) error

	SearchSessionTopK(ctx context.Context, sessionID int64, userID int, embeddingModel string, queryEmbedding []float32, topK int, fileID *int64) ([]domain.ScoredDocumentRAGChunk, error)

	GetChunksByFileChunkIndices(ctx context.Context, sessionID int64, userID int, fileID int64, embeddingModel string, chunkIndices []int) ([]domain.DocumentRAGChunk, error)
}

type EditorHistoryRepository interface {
	Save(ctx context.Context, userID int, runnerID *int64, text string) error
}

type FileRepository interface {
	Create(ctx context.Context, file *domain.File) error

	UpdateStoragePath(ctx context.Context, id int64, storagePath string) error

	GetById(ctx context.Context, id int64) (*domain.File, error)

	GetByIdWithExtractedCache(ctx context.Context, id int64) (*domain.File, error)

	ListByIds(ctx context.Context, ids []int64) ([]*domain.File, error)

	SaveExtractedTextCache(ctx context.Context, fileID int64, contentSHA256Hex string, extractedText string) error

	CountSessionTTLArtifacts(ctx context.Context, sessionID int64, userID int) (count int32, totalSize int64, err error)
}

type MCPServerRepository interface {
	ListGlobal(ctx context.Context) ([]*mcpdomain.MCPServer, error)

	ListForUser(ctx context.Context, userID int) ([]*mcpdomain.MCPServer, error)

	ListActive(ctx context.Context) ([]*mcpdomain.MCPServer, error)

	GetByID(ctx context.Context, id int64) (*mcpdomain.MCPServer, error)

	GetByIDAccessible(ctx context.Context, id int64, userID int) (*mcpdomain.MCPServer, error)

	GetGlobalByID(ctx context.Context, id int64) (*mcpdomain.MCPServer, error)

	Create(ctx context.Context, s *mcpdomain.MCPServer) (*mcpdomain.MCPServer, error)

	UpdateGlobal(ctx context.Context, s *mcpdomain.MCPServer) error

	UpdateOwned(ctx context.Context, s *mcpdomain.MCPServer, ownerUserID int) error

	DeleteGlobal(ctx context.Context, id int64) error

	DeleteOwned(ctx context.Context, id int64, ownerUserID int) error

	CountOwnedByUser(ctx context.Context, userID int) (int64, error)
}

type MessageEditRepository interface {
	Create(ctx context.Context, edit *domain.MessageEdit) error

	ListByMessageID(ctx context.Context, messageID int64, limit int32) ([]*domain.MessageEdit, error)
}

type MessageRepository interface {
	Create(ctx context.Context, message *domain.Message) error

	UpdateContent(ctx context.Context, id int64, content string) error

	GetBySessionId(ctx context.Context, sessionID int64, page, pageSize int32) ([]*domain.Message, int32, error)

	ListLatestMessagesForSession(ctx context.Context, sessionID int64, limit int32) ([]*domain.Message, error)

	ListBySessionBeforeID(ctx context.Context, sessionID int64, beforeMessageID int64, limit int32) ([]*domain.Message, int32, error)

	SessionHasOlderMessages(ctx context.Context, sessionID int64, olderThanMessageID int64) (bool, error)

	GetByID(ctx context.Context, id int64) (*domain.Message, error)

	ListMessagesWithIDLessThan(ctx context.Context, sessionID int64, beforeMessageID int64) ([]*domain.Message, error)

	ListMessagesUpToID(ctx context.Context, sessionID int64, upToMessageID int64) ([]*domain.Message, error)

	SoftDeleteAfterID(ctx context.Context, sessionID int64, afterMessageID int64) error

	SoftDeleteRangeAfterID(ctx context.Context, sessionID int64, afterMessageID int64, upToMessageID int64) error

	ResetAssistantForRegenerate(ctx context.Context, sessionID int64, messageID int64) error

	MaxMessageIDInSession(ctx context.Context, sessionID int64) (int64, error)

	ListBySessionCreatedAtWindowIncludingDeleted(ctx context.Context, sessionID int64, fromInclusive, toExclusive time.Time) ([]*domain.Message, error)
}

type RunnerRepository interface {
	List(ctx context.Context) ([]domain.Runner, error)

	GetByID(ctx context.Context, id int64) (*domain.Runner, error)

	FirstEnabled(ctx context.Context) (*domain.Runner, error)

	Create(ctx context.Context, name, host string, port int32, enabled bool, selectedModel string) (*domain.Runner, error)

	Update(ctx context.Context, id int64, name, host string, port int32, enabled bool, selectedModel string) (*domain.Runner, error)

	SetEnabled(ctx context.Context, id int64, enabled bool) error

	Delete(ctx context.Context, id int64) error

	FindIDByListenAddress(ctx context.Context, listenAddr string) (id int64, ok bool, err error)
}

type TokenRepository interface {
	Create(ctx context.Context, token *domain.Token) error

	GetByToken(ctx context.Context, token string) (*domain.Token, error)

	DeleteByToken(ctx context.Context, token string) error

	DeleteByUserId(ctx context.Context, userId int, tokenType domain.TokenType) error
}

type UserRepository interface {
	Create(ctx context.Context, user *domain.User) error

	GetById(ctx context.Context, id int) (*domain.User, error)

	GetByUsername(ctx context.Context, username string) (*domain.User, error)

	Update(ctx context.Context, user *domain.User) error

	List(ctx context.Context, page, pageSize int32) ([]*domain.User, int32, error)

	UpdateLastVisitedAt(ctx context.Context, userID int) error
}

type WebSearchSettingsRepository interface {
	Get(ctx context.Context) (*domain.WebSearchSettings, error)

	Upsert(ctx context.Context, s *domain.WebSearchSettings) error
}