db

type ObservationReader interface {
	GetObservationByID(ctx context.Context, id int64) (*models.Observation, error)
	GetObservationsByIDs(ctx context.Context, ids []int64, orderBy string, limit int) ([]*models.Observation, error)
	GetRecentObservations(ctx context.Context, project string, limit int) ([]*models.Observation, error)
	GetActiveObservations(ctx context.Context, project string, limit int) ([]*models.Observation, error)
	GetAllRecentObservations(ctx context.Context, limit int) ([]*models.Observation, error)
	GetAllObservations(ctx context.Context) ([]*models.Observation, error)
	SearchObservationsFTS(ctx context.Context, query, project string, limit int) ([]*models.Observation, error)
	GetObservationCount(ctx context.Context, project string) (int, error)
}

type ObservationStore interface {
	ObservationReader
	ObservationWriter
}

type ObservationWriter interface {
	StoreObservation(ctx context.Context, sdkSessionID, project string, obs *models.ParsedObservation, promptNumber int, discoveryTokens int64) (int64, int64, error)
	DeleteObservations(ctx context.Context, ids []int64) (int64, error)
}

type PromptReader interface {
	GetPromptsByIDs(ctx context.Context, ids []int64, orderBy string, limit int) ([]*models.UserPromptWithSession, error)
	GetAllRecentUserPrompts(ctx context.Context, limit int) ([]*models.UserPromptWithSession, error)
	GetAllPrompts(ctx context.Context) ([]*models.UserPromptWithSession, error)
	GetRecentUserPromptsByProject(ctx context.Context, project string, limit int) ([]*models.UserPromptWithSession, error)
	FindRecentPromptByText(ctx context.Context, claudeSessionID, promptText string, withinSeconds int) (int64, int, bool)
}

type PromptStore interface {
	PromptReader
	PromptWriter
}

type PromptWriter interface {
	SaveUserPromptWithMatches(ctx context.Context, claudeSessionID string, promptNumber int, promptText string, matchedObservations int) (int64, error)
}

type SummaryReader interface {
	GetSummariesByIDs(ctx context.Context, ids []int64, orderBy string, limit int) ([]*models.SessionSummary, error)
	GetRecentSummaries(ctx context.Context, project string, limit int) ([]*models.SessionSummary, error)
	GetAllRecentSummaries(ctx context.Context, limit int) ([]*models.SessionSummary, error)
	GetAllSummaries(ctx context.Context) ([]*models.SessionSummary, error)
}

type SummaryStore interface {
	SummaryReader
	SummaryWriter
}

type SummaryWriter interface {
	StoreSummary(ctx context.Context, sdkSessionID, project string, summary *models.ParsedSummary, promptNumber int, discoveryTokens int64) (int64, int64, error)
}