repository

type BackendRepository interface {
	// Workspaces
	CreateWorkspace(ctx context.Context, name string) (*types.Workspace, error)
	GetWorkspace(ctx context.Context, id uint) (*types.Workspace, error)
	GetWorkspaceByExternalId(ctx context.Context, externalId string) (*types.Workspace, error)
	GetWorkspaceByName(ctx context.Context, name string) (*types.Workspace, error)
	ListWorkspaces(ctx context.Context) ([]*types.Workspace, error)
	DeleteWorkspace(ctx context.Context, id uint) error

	// Workspace Tool Settings
	GetWorkspaceToolSettings(ctx context.Context, workspaceId uint) (*types.WorkspaceToolSettings, error)
	GetWorkspaceToolSetting(ctx context.Context, workspaceId uint, toolName string) (*types.WorkspaceToolSetting, error)
	SetWorkspaceToolSetting(ctx context.Context, workspaceId uint, toolName string, enabled bool) error
	ListWorkspaceToolSettings(ctx context.Context, workspaceId uint) ([]types.WorkspaceToolSetting, error)

	// Members
	MemberRepository

	// Tokens
	TokenRepository

	// Integrations
	IntegrationRepository

	// Workspace Tools
	WorkspaceToolRepository

	// Tasks
	CreateTask(ctx context.Context, task *types.Task) error
	GetTask(ctx context.Context, externalId string) (*types.Task, error)
	GetTaskById(ctx context.Context, id uint) (*types.Task, error)
	ListTasks(ctx context.Context, workspaceId uint) ([]*types.Task, error)
	UpdateTaskStatus(ctx context.Context, externalId string, status types.TaskStatus) error
	SetTaskStarted(ctx context.Context, externalId string) error
	SetTaskResult(ctx context.Context, externalId string, exitCode int, errorMsg string) error
	CancelTask(ctx context.Context, externalId string) error
	DeleteTask(ctx context.Context, externalId string) error

	// Database access
	DB() *sql.DB

	// Utilities
	Ping(ctx context.Context) error
	Close() error
	RunMigrations() error
}

type ElasticsearchClient interface {
	Index(ctx context.Context, index, docID string, body []byte) error
	Search(ctx context.Context, index string, query map[string]interface{}, size int) ([]json.RawMessage, error)
	Get(ctx context.Context, index, docID string) ([]byte, error)
	Delete(ctx context.Context, index, docID string) error
	DeleteByQuery(ctx context.Context, index string, query map[string]interface{}) error
}

type FilesystemStore interface {

	// CreateQuery stores a new filesystem query definition.
	CreateQuery(ctx context.Context, query *types.FilesystemQuery) (*types.FilesystemQuery, error)

	// GetQuery retrieves a query by workspace and path.
	GetQuery(ctx context.Context, workspaceId uint, path string) (*types.FilesystemQuery, error)

	// GetQueryByExternalId retrieves a query by its external ID.
	GetQueryByExternalId(ctx context.Context, externalId string) (*types.FilesystemQuery, error)

	// ListQueries returns all queries under a parent path.
	ListQueries(ctx context.Context, workspaceId uint, parentPath string) ([]*types.FilesystemQuery, error)

	// UpdateQuery updates an existing query definition.
	UpdateQuery(ctx context.Context, query *types.FilesystemQuery) error

	// DeleteQuery removes a query by external ID.
	DeleteQuery(ctx context.Context, externalId string) error

	// GetQueryResults retrieves cached results for a query path.
	GetQueryResults(ctx context.Context, workspaceId uint, queryPath string) ([]QueryResult, error)

	// StoreQueryResults caches query results with the specified TTL.
	StoreQueryResults(ctx context.Context, workspaceId uint, queryPath string, results []QueryResult, ttl time.Duration) error

	// GetResultContent retrieves cached content for a specific result.
	GetResultContent(ctx context.Context, workspaceId uint, queryPath, resultID string) ([]byte, error)

	// StoreResultContent caches content for a specific result.
	StoreResultContent(ctx context.Context, workspaceId uint, queryPath, resultID string, content []byte) error

	// SearchContent performs full-text search across all materialized content.
	SearchContent(ctx context.Context, workspaceId uint, query string, limit int) ([]SearchHit, error)

	// IndexContent indexes content for full-text search.
	IndexContent(ctx context.Context, workspaceId uint, queryPath, resultID, filename string, content []byte) error

	// GetFileMeta retrieves file metadata.
	GetFileMeta(ctx context.Context, path string) (*types.FileMeta, error)

	// GetDirMeta retrieves directory metadata.
	GetDirMeta(ctx context.Context, path string) (*types.DirMeta, error)

	// SaveFileMeta stores file metadata.
	SaveFileMeta(ctx context.Context, meta *types.FileMeta) error

	// SaveDirMeta stores directory metadata.
	SaveDirMeta(ctx context.Context, meta *types.DirMeta) error

	// DeleteFileMeta removes file metadata.
	DeleteFileMeta(ctx context.Context, path string) error

	// DeleteDirMeta removes directory metadata.
	DeleteDirMeta(ctx context.Context, path string) error

	// ListDir returns directory entries.
	ListDir(ctx context.Context, path string) ([]types.DirEntry, error)

	// SaveDirListing stores a directory listing.
	SaveDirListing(ctx context.Context, path string, entries []types.DirEntry) error

	// GetSymlink retrieves a symlink target.
	GetSymlink(ctx context.Context, path string) (string, error)

	// SaveSymlink stores a symlink.
	SaveSymlink(ctx context.Context, path, target string) error

	// DeleteSymlink removes a symlink.
	DeleteSymlink(ctx context.Context, path string) error

	// InvalidatePath removes a specific path from all caches.
	InvalidatePath(ctx context.Context, path string) error

	// InvalidatePrefix removes all paths with a given prefix.
	InvalidatePrefix(ctx context.Context, prefix string) error

	// InvalidateQuery removes all cached data for a query (results + content).
	InvalidateQuery(ctx context.Context, workspaceId uint, queryPath string) error
}

type IntegrationRepository interface {
	SaveConnection(ctx context.Context, workspaceId uint, memberId *uint, integrationType string, creds *types.IntegrationCredentials, scope string) (*types.IntegrationConnection, error)
	GetConnection(ctx context.Context, workspaceId uint, memberId uint, integrationType string) (*types.IntegrationConnection, error)
	GetConnectionByExternalId(ctx context.Context, externalId string) (*types.IntegrationConnection, error)
	ListConnections(ctx context.Context, workspaceId uint) ([]types.IntegrationConnection, error)
	DeleteConnection(ctx context.Context, externalId string) error
}

type MemberRepository interface {
	CreateMember(ctx context.Context, workspaceId uint, email, name string, role types.MemberRole) (*types.WorkspaceMember, error)
	GetMember(ctx context.Context, externalId string) (*types.WorkspaceMember, error)
	GetMemberByEmail(ctx context.Context, workspaceId uint, email string) (*types.WorkspaceMember, error)
	ListMembers(ctx context.Context, workspaceId uint) ([]types.WorkspaceMember, error)
	UpdateMember(ctx context.Context, externalId string, name string, role types.MemberRole) (*types.WorkspaceMember, error)
	DeleteMember(ctx context.Context, externalId string) error
}

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

type QueryResult struct {
	ID       string            `json:"id"`       // Provider-specific ID (e.g., Gmail message ID)
	Filename string            `json:"filename"` // Generated filename for the result
	Metadata map[string]string `json:"metadata"` // Key-value metadata (date, subject, from, etc.)
	Size     int64             `json:"size"`     // Content size in bytes
	Mtime    int64             `json:"mtime"`    // Last modified timestamp (Unix)
}

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

type SearchHit struct {
	WorkspaceID uint    `json:"workspace_id"`
	QueryPath   string  `json:"query_path"` // Path of the query that produced this result
	ResultID    string  `json:"result_id"`  // Provider-specific ID
	Filename    string  `json:"filename"`
	Snippet     string  `json:"snippet"` // Text snippet with match highlighted
	Score       float64 `json:"score"`   // Relevance score
}

type TaskLogEntry struct {
	TaskID    string `json:"task_id"`
	Timestamp int64  `json:"timestamp"`
	Stream    string `json:"stream"` // "stdout" or "stderr"
	Data      string `json:"data"`
}

type TaskQueue interface {
	Push(ctx context.Context, task *types.Task) error
	Pop(ctx context.Context, workerID string) (*types.Task, error)
	Complete(ctx context.Context, taskID string, result *types.TaskResult) error
	Fail(ctx context.Context, taskID string, err error) error
	GetState(ctx context.Context, taskID string) (*types.TaskState, error)
	GetResult(ctx context.Context, taskID string) (*types.TaskResult, error)
	Len(ctx context.Context) (int64, error)
	InFlightCount(ctx context.Context) (int64, error)

	// Log streaming
	PublishLog(ctx context.Context, taskID string, stream string, data string) error
	PublishStatus(ctx context.Context, taskID string, status types.TaskStatus, exitCode *int, errorMsg string) error
	SubscribeLogs(ctx context.Context, taskID string) (<-chan []byte, func(), error)
	GetLogBuffer(ctx context.Context, taskID string) ([][]byte, error)
}

type TaskStatusEvent struct {
	TaskID    string           `json:"task_id"`
	Timestamp int64            `json:"timestamp"`
	Status    types.TaskStatus `json:"status"`
	ExitCode  *int             `json:"exit_code,omitempty"`
	Error     string           `json:"error,omitempty"`
}

type TokenRepository interface {
	// Workspace member tokens
	CreateToken(ctx context.Context, workspaceId, memberId uint, name string, expiresAt *time.Time, tokenType types.TokenType) (*types.Token, string, error)
	GetToken(ctx context.Context, externalId string) (*types.Token, error)
	ListTokens(ctx context.Context, workspaceId uint) ([]types.Token, error)
	RevokeToken(ctx context.Context, externalId string) error

	// Worker tokens (cluster-level)
	CreateWorkerToken(ctx context.Context, name string, poolName *string, expiresAt *time.Time) (*types.Token, string, error)
	ListWorkerTokens(ctx context.Context) ([]types.Token, error)

	// Validation
	ValidateToken(ctx context.Context, rawToken string) (*types.TokenValidationResult, error)
	AuthorizeToken(ctx context.Context, rawToken string) (*types.AuthInfo, error)
}

type WorkerPoolRepository interface {
	SetPoolState(ctx context.Context, poolName string, state *types.WorkerPoolState) error
	GetPoolState(ctx context.Context, poolName string) (*types.WorkerPoolState, error)
}

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

type WorkerRepository interface {
	AddWorker(ctx context.Context, worker *types.Worker) error
	GetWorker(ctx context.Context, workerId string) (*types.Worker, error)
	GetAllWorkers(ctx context.Context) ([]*types.Worker, error)
	GetAvailableWorkers(ctx context.Context) ([]*types.Worker, error)
	RemoveWorker(ctx context.Context, workerId string) error
	SetWorkerKeepAlive(ctx context.Context, workerId string) error
	UpdateWorkerStatus(ctx context.Context, workerId string, status types.WorkerStatus) error
}

type WorkspaceToolRepository interface {
	CreateWorkspaceTool(ctx context.Context, workspaceId uint, createdByMemberId *uint, name string, providerType types.WorkspaceToolProviderType, config json.RawMessage, manifest json.RawMessage) (*types.WorkspaceTool, error)
	GetWorkspaceTool(ctx context.Context, id uint) (*types.WorkspaceTool, error)
	GetWorkspaceToolByExternalId(ctx context.Context, externalId string) (*types.WorkspaceTool, error)
	GetWorkspaceToolByName(ctx context.Context, workspaceId uint, name string) (*types.WorkspaceTool, error)
	ListWorkspaceTools(ctx context.Context, workspaceId uint) ([]*types.WorkspaceTool, error)
	UpdateWorkspaceToolManifest(ctx context.Context, id uint, manifest json.RawMessage) error
	UpdateWorkspaceToolConfig(ctx context.Context, id uint, config json.RawMessage) error
	DeleteWorkspaceTool(ctx context.Context, id uint) error
	DeleteWorkspaceToolByName(ctx context.Context, workspaceId uint, name string) error
}