sqlc

type AddTaskDependencyParams struct {
	TaskID          string `json:"task_id"`
	DependsOnTaskID string `json:"depends_on_task_id"`
}

type AddTaskToWorkParams struct {
	WorkID   string `json:"work_id"`
	TaskID   string `json:"task_id"`
	Position int64  `json:"position"`
}

type AddWorkBeadParams struct {
	WorkID   string `json:"work_id"`
	BeadID   string `json:"bead_id"`
	Position int64  `json:"position"`
}

type AddWorkBeadsBatchParams struct {
	WorkID   string `json:"work_id"`
	BeadID   string `json:"bead_id"`
	Position int64  `json:"position"`
}

type Bead struct {
	ID            string       `json:"id"`
	Status        string       `json:"status"`
	Title         string       `json:"title"`
	PrUrl         string       `json:"pr_url"`
	ErrorMessage  string       `json:"error_message"`
	ZellijSession string       `json:"zellij_session"`
	ZellijPane    string       `json:"zellij_pane"`
	WorktreePath  string       `json:"worktree_path"`
	StartedAt     sql.NullTime `json:"started_at"`
	CompletedAt   sql.NullTime `json:"completed_at"`
	CreatedAt     time.Time    `json:"created_at"`
	UpdatedAt     time.Time    `json:"updated_at"`
}

type CacheComplexityParams struct {
	BeadID          string `json:"bead_id"`
	DescriptionHash string `json:"description_hash"`
	ComplexityScore int64  `json:"complexity_score"`
	EstimatedTokens int64  `json:"estimated_tokens"`
}

type CompleteBeadParams struct {
	PrUrl       string       `json:"pr_url"`
	CompletedAt sql.NullTime `json:"completed_at"`
	UpdatedAt   time.Time    `json:"updated_at"`
	ID          string       `json:"id"`
}

type CompleteTaskBeadParams struct {
	TaskID string `json:"task_id"`
	BeadID string `json:"bead_id"`
}

type CompleteTaskParams struct {
	PrUrl       string       `json:"pr_url"`
	CompletedAt sql.NullTime `json:"completed_at"`
	ID          string       `json:"id"`
}

type CompleteWorkParams struct {
	PrUrl       string       `json:"pr_url"`
	CompletedAt sql.NullTime `json:"completed_at"`
	ID          string       `json:"id"`
}

type ComplexityCache struct {
	BeadID          string    `json:"bead_id"`
	DescriptionHash string    `json:"description_hash"`
	ComplexityScore int64     `json:"complexity_score"`
	EstimatedTokens int64     `json:"estimated_tokens"`
	CreatedAt       time.Time `json:"created_at"`
}

type CountTaskBeadStatusesRow struct {
	Total     int64 `json:"total"`
	Completed int64 `json:"completed"`
}

type CreatePRFeedbackParams struct {
	ID           string         `json:"id"`
	WorkID       string         `json:"work_id"`
	PrUrl        string         `json:"pr_url"`
	FeedbackType string         `json:"feedback_type"`
	Title        string         `json:"title"`
	Description  string         `json:"description"`
	SourceUrl    sql.NullString `json:"source_url"`
	SourceID     sql.NullString `json:"source_id"`
	Priority     int64          `json:"priority"`
	SourceType   sql.NullString `json:"source_type"`
	SourceName   sql.NullString `json:"source_name"`
	Context      sql.NullString `json:"context"`
}

type CreateScheduledTaskParams struct {
	ID          string    `json:"id"`
	WorkID      string    `json:"work_id"`
	TaskType    string    `json:"task_type"`
	ScheduledAt time.Time `json:"scheduled_at"`
	Status      string    `json:"status"`
	Metadata    string    `json:"metadata"`
}

type CreateScheduledTaskWithRetryParams struct {
	ID             string         `json:"id"`
	WorkID         string         `json:"work_id"`
	TaskType       string         `json:"task_type"`
	ScheduledAt    time.Time      `json:"scheduled_at"`
	Status         string         `json:"status"`
	Metadata       string         `json:"metadata"`
	AttemptCount   int64          `json:"attempt_count"`
	MaxAttempts    int64          `json:"max_attempts"`
	IdempotencyKey sql.NullString `json:"idempotency_key"`
}

type CreateTaskBeadParams struct {
	TaskID string `json:"task_id"`
	BeadID string `json:"bead_id"`
}

type CreateTaskParams struct {
	ID               string `json:"id"`
	TaskType         string `json:"task_type"`
	ComplexityBudget int64  `json:"complexity_budget"`
	WorkID           string `json:"work_id"`
}

type CreateWorkParams struct {
	ID           string `json:"id"`
	Name         string `json:"name"`
	WorktreePath string `json:"worktree_path"`
	BranchName   string `json:"branch_name"`
	BaseBranch   string `json:"base_branch"`
	RootIssueID  string `json:"root_issue_id"`
	Auto         bool   `json:"auto"`
}

type DBTX interface {
	ExecContext(context.Context, string, ...interface{}) (sql.Result, error)
	PrepareContext(context.Context, string) (*sql.Stmt, error)
	QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error)
	QueryRowContext(context.Context, string, ...interface{}) *sql.Row
}

type DeleteTaskDependencyParams struct {
	TaskID          string `json:"task_id"`
	DependsOnTaskID string `json:"depends_on_task_id"`
}

type DeleteTaskMetadataParams struct {
	TaskID string `json:"task_id"`
	Key    string `json:"key"`
}

type FailBeadParams struct {
	ErrorMessage string       `json:"error_message"`
	CompletedAt  sql.NullTime `json:"completed_at"`
	UpdatedAt    time.Time    `json:"updated_at"`
	ID           string       `json:"id"`
}

type FailTaskBeadParams struct {
	TaskID string `json:"task_id"`
	BeadID string `json:"bead_id"`
}

type FailTaskParams struct {
	ErrorMessage string       `json:"error_message"`
	CompletedAt  sql.NullTime `json:"completed_at"`
	ID           string       `json:"id"`
}

type FailWorkParams struct {
	ErrorMessage string       `json:"error_message"`
	CompletedAt  sql.NullTime `json:"completed_at"`
	ID           string       `json:"id"`
}

type GetAllAssignedBeadsRow struct {
	BeadID string `json:"bead_id"`
	WorkID string `json:"work_id"`
}

type GetAllCachedComplexityRow struct {
	BeadID          string `json:"bead_id"`
	ComplexityScore int64  `json:"complexity_score"`
	EstimatedTokens int64  `json:"estimated_tokens"`
}

type GetAllTaskMetadataRow struct {
	Key   string `json:"key"`
	Value string `json:"value"`
}

type GetCachedComplexityParams struct {
	BeadID          string `json:"bead_id"`
	DescriptionHash string `json:"description_hash"`
}

type GetCachedComplexityRow struct {
	ComplexityScore int64 `json:"complexity_score"`
	EstimatedTokens int64 `json:"estimated_tokens"`
}

type GetMigrationDownSQLRow struct {
	Name    string `json:"name"`
	DownSql string `json:"down_sql"`
}

type GetPRFeedbackBySourceIDParams struct {
	WorkID   string         `json:"work_id"`
	SourceID sql.NullString `json:"source_id"`
}

type GetPRTaskForWorkRow struct {
	ID               string       `json:"id"`
	Status           string       `json:"status"`
	TaskType         string       `json:"task_type"`
	ComplexityBudget int64        `json:"complexity_budget"`
	ActualComplexity int64        `json:"actual_complexity"`
	WorkID           string       `json:"work_id"`
	WorktreePath     string       `json:"worktree_path"`
	PrUrl            string       `json:"pr_url"`
	ErrorMessage     string       `json:"error_message"`
	StartedAt        sql.NullTime `json:"started_at"`
	CompletedAt      sql.NullTime `json:"completed_at"`
	CreatedAt        time.Time    `json:"created_at"`
	SpawnedAt        sql.NullTime `json:"spawned_at"`
	SpawnStatus      string       `json:"spawn_status"`
}

type GetPendingTaskByTypeParams struct {
	WorkID   string `json:"work_id"`
	TaskType string `json:"task_type"`
}

type GetReadyTasksForWorkRow struct {
	ID               string       `json:"id"`
	Status           string       `json:"status"`
	TaskType         string       `json:"task_type"`
	ComplexityBudget int64        `json:"complexity_budget"`
	ActualComplexity int64        `json:"actual_complexity"`
	WorkID           string       `json:"work_id"`
	WorktreePath     string       `json:"worktree_path"`
	PrUrl            string       `json:"pr_url"`
	ErrorMessage     string       `json:"error_message"`
	StartedAt        sql.NullTime `json:"started_at"`
	CompletedAt      sql.NullTime `json:"completed_at"`
	CreatedAt        time.Time    `json:"created_at"`
	SpawnedAt        sql.NullTime `json:"spawned_at"`
	SpawnStatus      string       `json:"spawn_status"`
}

type GetTaskBeadStatusParams struct {
	TaskID string `json:"task_id"`
	BeadID string `json:"bead_id"`
}

type GetTaskMetadataParams struct {
	TaskID string `json:"task_id"`
	Key    string `json:"key"`
}

type GetTaskRow struct {
	ID               string       `json:"id"`
	Status           string       `json:"status"`
	TaskType         string       `json:"task_type"`
	ComplexityBudget int64        `json:"complexity_budget"`
	ActualComplexity int64        `json:"actual_complexity"`
	WorkID           string       `json:"work_id"`
	WorktreePath     string       `json:"worktree_path"`
	PrUrl            string       `json:"pr_url"`
	ErrorMessage     string       `json:"error_message"`
	StartedAt        sql.NullTime `json:"started_at"`
	CompletedAt      sql.NullTime `json:"completed_at"`
	CreatedAt        time.Time    `json:"created_at"`
	SpawnedAt        sql.NullTime `json:"spawned_at"`
	SpawnStatus      string       `json:"spawn_status"`
}

type GetWorkTasksRow struct {
	ID               string       `json:"id"`
	Status           string       `json:"status"`
	TaskType         string       `json:"task_type"`
	ComplexityBudget int64        `json:"complexity_budget"`
	ActualComplexity int64        `json:"actual_complexity"`
	WorkID           string       `json:"work_id"`
	WorktreePath     string       `json:"worktree_path"`
	PrUrl            string       `json:"pr_url"`
	ErrorMessage     string       `json:"error_message"`
	StartedAt        sql.NullTime `json:"started_at"`
	CompletedAt      sql.NullTime `json:"completed_at"`
	CreatedAt        time.Time    `json:"created_at"`
	SpawnedAt        sql.NullTime `json:"spawned_at"`
	SpawnStatus      string       `json:"spawn_status"`
}

type HasExistingFeedbackBySourceIDParams struct {
	WorkID   string         `json:"work_id"`
	SourceID sql.NullString `json:"source_id"`
}

type HasExistingFeedbackParams struct {
	WorkID     string         `json:"work_id"`
	Title      string         `json:"title"`
	SourceType sql.NullString `json:"source_type"`
	SourceName sql.NullString `json:"source_name"`
}

type IdleWorkWithPRParams struct {
	PrUrl string `json:"pr_url"`
	ID    string `json:"id"`
}

type IncrementAttemptAndRescheduleParams struct {
	ScheduledAt  time.Time      `json:"scheduled_at"`
	ErrorMessage sql.NullString `json:"error_message"`
	ID           string         `json:"id"`
}

type IsBeadInTaskParams struct {
	WorkID string `json:"work_id"`
	BeadID string `json:"bead_id"`
}

type IsOrchestratorAliveParams struct {
	WorkID  sql.NullString `json:"work_id"`
	Column2 sql.NullString `json:"column_2"`
}

type ListMigrationsWithDetailsRow struct {
	Version   string       `json:"version"`
	Name      string       `json:"name"`
	DownSql   string       `json:"down_sql"`
	AppliedAt sql.NullTime `json:"applied_at"`
}

type ListTasksByStatusRow struct {
	ID               string       `json:"id"`
	Status           string       `json:"status"`
	TaskType         string       `json:"task_type"`
	ComplexityBudget int64        `json:"complexity_budget"`
	ActualComplexity int64        `json:"actual_complexity"`
	WorkID           string       `json:"work_id"`
	WorktreePath     string       `json:"worktree_path"`
	PrUrl            string       `json:"pr_url"`
	ErrorMessage     string       `json:"error_message"`
	StartedAt        sql.NullTime `json:"started_at"`
	CompletedAt      sql.NullTime `json:"completed_at"`
	CreatedAt        time.Time    `json:"created_at"`
	SpawnedAt        sql.NullTime `json:"spawned_at"`
	SpawnStatus      string       `json:"spawn_status"`
}

type ListTasksRow struct {
	ID               string       `json:"id"`
	Status           string       `json:"status"`
	TaskType         string       `json:"task_type"`
	ComplexityBudget int64        `json:"complexity_budget"`
	ActualComplexity int64        `json:"actual_complexity"`
	WorkID           string       `json:"work_id"`
	WorktreePath     string       `json:"worktree_path"`
	PrUrl            string       `json:"pr_url"`
	ErrorMessage     string       `json:"error_message"`
	StartedAt        sql.NullTime `json:"started_at"`
	CompletedAt      sql.NullTime `json:"completed_at"`
	CreatedAt        time.Time    `json:"created_at"`
	SpawnedAt        sql.NullTime `json:"spawned_at"`
	SpawnStatus      string       `json:"spawn_status"`
}

type MarkPRFeedbackProcessedParams struct {
	BeadID sql.NullString `json:"bead_id"`
	ID     string         `json:"id"`
}

type MarkTaskFailedParams struct {
	ErrorMessage sql.NullString `json:"error_message"`
	ID           string         `json:"id"`
}

type MergeWorkParams struct {
	CompletedAt sql.NullTime `json:"completed_at"`
	ID          string       `json:"id"`
}

type PlanSession struct {
	BeadID        string    `json:"bead_id"`
	ZellijSession string    `json:"zellij_session"`
	TabName       string    `json:"tab_name"`
	Pid           int64     `json:"pid"`
	StartedAt     time.Time `json:"started_at"`
}

type PrFeedback struct {
	ID           string         `json:"id"`
	WorkID       string         `json:"work_id"`
	PrUrl        string         `json:"pr_url"`
	FeedbackType string         `json:"feedback_type"`
	Title        string         `json:"title"`
	Description  string         `json:"description"`
	SourceUrl    sql.NullString `json:"source_url"`
	SourceID     sql.NullString `json:"source_id"`
	SourceType   sql.NullString `json:"source_type"`
	SourceName   sql.NullString `json:"source_name"`
	Context      sql.NullString `json:"context"`
	Priority     int64          `json:"priority"`
	BeadID       sql.NullString `json:"bead_id"`
	CreatedAt    time.Time      `json:"created_at"`
	ProcessedAt  sql.NullTime   `json:"processed_at"`
	ResolvedAt   sql.NullTime   `json:"resolved_at"`
}

type Process struct {
	ID          string         `json:"id"`
	ProcessType string         `json:"process_type"`
	WorkID      sql.NullString `json:"work_id"`
	Pid         int64          `json:"pid"`
	Hostname    string         `json:"hostname"`
	Heartbeat   time.Time      `json:"heartbeat"`
	StartedAt   time.Time      `json:"started_at"`
}

type Querier interface {
	AddTaskDependency(ctx context.Context, arg AddTaskDependencyParams) error
	AddTaskToWork(ctx context.Context, arg AddTaskToWorkParams) error
	AddWorkBead(ctx context.Context, arg AddWorkBeadParams) error
	AddWorkBeadsBatch(ctx context.Context, arg AddWorkBeadsBatchParams) error
	CacheComplexity(ctx context.Context, arg CacheComplexityParams) error
	CompleteBead(ctx context.Context, arg CompleteBeadParams) (int64, error)
	CompleteTask(ctx context.Context, arg CompleteTaskParams) (int64, error)
	CompleteTaskBead(ctx context.Context, arg CompleteTaskBeadParams) (int64, error)
	CompleteWork(ctx context.Context, arg CompleteWorkParams) (int64, error)
	CountEstimatedBeads(ctx context.Context, beadIds []string) (int64, error)
	CountPendingTasksForWork(ctx context.Context, workID string) (int64, error)
	CountTaskBeadStatuses(ctx context.Context, taskID string) (CountTaskBeadStatusesRow, error)
	// Count PR feedback items that have beads which are not yet assigned to any task and not resolved/closed.
	CountUnassignedFeedbackForWork(ctx context.Context, workID string) (int64, error)
	CreateMigrationsTable(ctx context.Context) error
	CreatePRFeedback(ctx context.Context, arg CreatePRFeedbackParams) error
	CreateScheduledTask(ctx context.Context, arg CreateScheduledTaskParams) error
	CreateScheduledTaskWithRetry(ctx context.Context, arg CreateScheduledTaskWithRetryParams) error
	CreateTask(ctx context.Context, arg CreateTaskParams) error
	CreateTaskBead(ctx context.Context, arg CreateTaskBeadParams) error
	CreateWork(ctx context.Context, arg CreateWorkParams) error
	DeleteCompletedTasksOlderThan(ctx context.Context, executedAt sql.NullTime) error
	DeleteControlPlaneProcess(ctx context.Context) error
	DeleteMigration(ctx context.Context, version string) error
	DeleteOrchestratorByWorkID(ctx context.Context, workID sql.NullString) error
	DeletePRFeedback(ctx context.Context, id string) error
	DeletePRFeedbackForWork(ctx context.Context, workID string) error
	DeleteProcess(ctx context.Context, id string) error
	DeleteScheduledTask(ctx context.Context, id string) error
	DeleteSchedulerForWork(ctx context.Context, workID string) (int64, error)
	DeleteStaleProcesses(ctx context.Context, dollar_1 sql.NullString) error
	DeleteTask(ctx context.Context, id string) (int64, error)
	DeleteTaskBeadsByTask(ctx context.Context, taskID string) (int64, error)
	DeleteTaskBeadsForWork(ctx context.Context, workID string) (int64, error)
	DeleteTaskDependencies(ctx context.Context, taskID string) (int64, error)
	DeleteTaskDependenciesForWork(ctx context.Context, workID string) (int64, error)
	DeleteTaskDependency(ctx context.Context, arg DeleteTaskDependencyParams) (int64, error)
	DeleteTaskMetadata(ctx context.Context, arg DeleteTaskMetadataParams) (int64, error)
	DeleteTasksForWork(ctx context.Context, workID string) (int64, error)
	DeleteWork(ctx context.Context, id string) (int64, error)
	DeleteWorkBeads(ctx context.Context, workID string) (int64, error)
	DeleteWorkTaskByTask(ctx context.Context, taskID string) (int64, error)
	DeleteWorkTasks(ctx context.Context, workID string) (int64, error)
	FailBead(ctx context.Context, arg FailBeadParams) (int64, error)
	FailTask(ctx context.Context, arg FailTaskParams) (int64, error)
	FailTaskBead(ctx context.Context, arg FailTaskBeadParams) (int64, error)
	FailWork(ctx context.Context, arg FailWorkParams) (int64, error)
	// Returns all beads assigned to any work, with their work ID.
	// This is used by plan mode to show which beads are already assigned.
	GetAllAssignedBeads(ctx context.Context) ([]GetAllAssignedBeadsRow, error)
	GetAllCachedComplexity(ctx context.Context) ([]GetAllCachedComplexityRow, error)
	GetAllProcesses(ctx context.Context) ([]Process, error)
	GetAllTaskMetadata(ctx context.Context, taskID string) ([]GetAllTaskMetadataRow, error)
	GetAndIncrementTaskCounter(ctx context.Context, workID string) (int64, error)
	GetAppliedMigrations(ctx context.Context) ([]string, error)
	GetBead(ctx context.Context, id string) (Bead, error)
	GetBeadStatus(ctx context.Context, id string) (string, error)
	GetCachedComplexity(ctx context.Context, arg GetCachedComplexityParams) (GetCachedComplexityRow, error)
	GetControlPlaneProcess(ctx context.Context) (Process, error)
	GetLastMigration(ctx context.Context) (string, error)
	GetLastWorkID(ctx context.Context) (string, error)
	GetMaxWorkBeadPosition(ctx context.Context, workID string) (int64, error)
	GetMigrationDownSQL(ctx context.Context, version string) (GetMigrationDownSQLRow, error)
	GetNextScheduledTask(ctx context.Context) (Scheduler, error)
	GetOrchestratorProcess(ctx context.Context, workID sql.NullString) (Process, error)
	GetOverdueTasks(ctx context.Context) ([]Scheduler, error)
	GetPRFeedback(ctx context.Context, id string) (PrFeedback, error)
	GetPRFeedbackByBead(ctx context.Context, beadID sql.NullString) (PrFeedback, error)
	GetPRFeedbackBySourceID(ctx context.Context, arg GetPRFeedbackBySourceIDParams) (PrFeedback, error)
	GetPRTaskForWork(ctx context.Context, workID string) (GetPRTaskForWorkRow, error)
	GetPendingTaskByType(ctx context.Context, arg GetPendingTaskByTypeParams) (Scheduler, error)
	GetProcess(ctx context.Context, id string) (Process, error)
	GetProcessByWorkID(ctx context.Context, workID sql.NullString) (Process, error)
	GetReadyTasksForWork(ctx context.Context, workID string) ([]GetReadyTasksForWorkRow, error)
	GetScheduledTaskByID(ctx context.Context, id string) (Scheduler, error)
	GetScheduledTasksForWork(ctx context.Context, workID string) ([]Scheduler, error)
	GetStaleProcesses(ctx context.Context, dollar_1 sql.NullString) ([]Process, error)
	GetTask(ctx context.Context, id string) (GetTaskRow, error)
	GetTaskBeadStatus(ctx context.Context, arg GetTaskBeadStatusParams) (string, error)
	GetTaskBeads(ctx context.Context, taskID string) ([]string, error)
	GetTaskBeadsForWork(ctx context.Context, workID string) ([]TaskBead, error)
	GetTaskBeadsWithStatus(ctx context.Context, taskID string) ([]TaskBead, error)
	GetTaskByIdempotencyKey(ctx context.Context, idempotencyKey sql.NullString) (Scheduler, error)
	GetTaskDependencies(ctx context.Context, taskID string) ([]string, error)
	GetTaskDependents(ctx context.Context, dependsOnTaskID string) ([]string, error)
	GetTaskForBead(ctx context.Context, beadID string) (string, error)
	GetTaskMetadata(ctx context.Context, arg GetTaskMetadataParams) (string, error)
	GetTasksWithActivity(ctx context.Context) ([]Task, error)
	// Get bead IDs from PR feedback items that are not yet assigned to any task and not resolved/closed.
	GetUnassignedFeedbackBeadIDs(ctx context.Context, workID string) ([]sql.NullString, error)
	GetUnassignedWorkBeads(ctx context.Context, workID string) ([]WorkBead, error)
	GetUnresolvedFeedbackForBeads(ctx context.Context, beadIds []sql.NullString) ([]PrFeedback, error)
	GetUnresolvedFeedbackForWork(ctx context.Context, workID string) ([]PrFeedback, error)
	GetWork(ctx context.Context, id string) (Work, error)
	GetWorkBeads(ctx context.Context, workID string) ([]WorkBead, error)
	GetWorkByDirectory(ctx context.Context, worktreePath string) (Work, error)
	GetWorkTasks(ctx context.Context, workID string) ([]GetWorkTasksRow, error)
	GetWorksWithPRs(ctx context.Context) ([]Work, error)
	GetWorksWithUnseenChanges(ctx context.Context) ([]Work, error)
	HasExistingFeedback(ctx context.Context, arg HasExistingFeedbackParams) (int64, error)
	HasExistingFeedbackBySourceID(ctx context.Context, arg HasExistingFeedbackBySourceIDParams) (int64, error)
	HasPendingDependencies(ctx context.Context, taskID string) (bool, error)
	IdleWork(ctx context.Context, id string) (int64, error)
	IdleWorkWithPR(ctx context.Context, arg IdleWorkWithPRParams) (int64, error)
	IncrementAttemptAndReschedule(ctx context.Context, arg IncrementAttemptAndRescheduleParams) error
	InitializeTaskCounter(ctx context.Context, workID string) error
	IsBeadInTask(ctx context.Context, arg IsBeadInTaskParams) (bool, error)
	IsControlPlaneAlive(ctx context.Context, dollar_1 sql.NullString) (int64, error)
	IsOrchestratorAlive(ctx context.Context, arg IsOrchestratorAliveParams) (int64, error)
	ListBeads(ctx context.Context) ([]Bead, error)
	ListBeadsByStatus(ctx context.Context, status string) ([]Bead, error)
	ListMigrationVersions(ctx context.Context) ([]string, error)
	ListMigrationsWithDetails(ctx context.Context) ([]ListMigrationsWithDetailsRow, error)
	ListPRFeedback(ctx context.Context, workID string) ([]PrFeedback, error)
	ListTasks(ctx context.Context) ([]ListTasksRow, error)
	ListTasksByStatus(ctx context.Context, status string) ([]ListTasksByStatusRow, error)
	ListUnprocessedPRFeedback(ctx context.Context, workID string) ([]PrFeedback, error)
	ListWorks(ctx context.Context) ([]Work, error)
	ListWorksByStatus(ctx context.Context, status string) ([]Work, error)
	MarkPRFeedbackProcessed(ctx context.Context, arg MarkPRFeedbackProcessedParams) error
	MarkPRFeedbackResolved(ctx context.Context, id string) error
	MarkTaskCompleted(ctx context.Context, id string) error
	MarkTaskCompletedByIdempotencyKey(ctx context.Context, idempotencyKey sql.NullString) error
	MarkTaskExecuting(ctx context.Context, id string) error
	MarkTaskFailed(ctx context.Context, arg MarkTaskFailedParams) error
	MarkWorkPRSeen(ctx context.Context, id string) (int64, error)
	MergeWork(ctx context.Context, arg MergeWorkParams) (int64, error)
	RecordMigration(ctx context.Context, version string) error
	RecordMigrationWithDown(ctx context.Context, arg RecordMigrationWithDownParams) error
	RegisterProcess(ctx context.Context, arg RegisterProcessParams) error
	RemoveWorkBead(ctx context.Context, arg RemoveWorkBeadParams) (int64, error)
	RescheduleTask(ctx context.Context, arg RescheduleTaskParams) error
	// Reset any tasks stuck in 'executing' status back to 'pending'.
	// Used when the control plane starts up to recover from a crash.
	ResetExecutingTasksToPending(ctx context.Context) (int64, error)
	ResetTaskBeadStatus(ctx context.Context, arg ResetTaskBeadStatusParams) (int64, error)
	ResetTaskBeadStatuses(ctx context.Context, taskID string) (int64, error)
	ResetTaskStatus(ctx context.Context, id string) (int64, error)
	RestartWork(ctx context.Context, id string) (int64, error)
	ResumeWork(ctx context.Context, id string) (int64, error)
	SetTaskMetadata(ctx context.Context, arg SetTaskMetadataParams) error
	SetWorkHasUnseenPRChanges(ctx context.Context, arg SetWorkHasUnseenPRChangesParams) (int64, error)
	SetWorkPRURL(ctx context.Context, arg SetWorkPRURLParams) (int64, error)
	SpawnTask(ctx context.Context, arg SpawnTaskParams) (int64, error)
	StartBead(ctx context.Context, arg StartBeadParams) error
	StartTask(ctx context.Context, arg StartTaskParams) (int64, error)
	StartWork(ctx context.Context, arg StartWorkParams) (int64, error)
	UpdateHeartbeat(ctx context.Context, id string) error
	UpdateHeartbeatWithTime(ctx context.Context, arg UpdateHeartbeatWithTimeParams) error
	UpdateMigrationDownSQL(ctx context.Context, arg UpdateMigrationDownSQLParams) error
	UpdateScheduledTaskTime(ctx context.Context, arg UpdateScheduledTaskTimeParams) error
	UpdateTaskActivity(ctx context.Context, arg UpdateTaskActivityParams) (int64, error)
	UpdateWorkPRStatus(ctx context.Context, arg UpdateWorkPRStatusParams) (int64, error)
	UpdateWorkWorktreePath(ctx context.Context, arg UpdateWorkWorktreePathParams) (int64, error)
	WatchSchedulerChanges(ctx context.Context, updatedAt time.Time) ([]Scheduler, error)
}

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

type RecordMigrationWithDownParams struct {
	Version string `json:"version"`
	Name    string `json:"name"`
	DownSql string `json:"down_sql"`
}

type RegisterProcessParams struct {
	ID          string         `json:"id"`
	ProcessType string         `json:"process_type"`
	WorkID      sql.NullString `json:"work_id"`
	Pid         int64          `json:"pid"`
	Hostname    string         `json:"hostname"`
}

type RemoveWorkBeadParams struct {
	WorkID string `json:"work_id"`
	BeadID string `json:"bead_id"`
}

type RescheduleTaskParams struct {
	ScheduledAt time.Time `json:"scheduled_at"`
	ID          string    `json:"id"`
}

type ResetTaskBeadStatusParams struct {
	TaskID string `json:"task_id"`
	BeadID string `json:"bead_id"`
}

type Scheduler struct {
	ID             string         `json:"id"`
	WorkID         string         `json:"work_id"`
	TaskType       string         `json:"task_type"`
	ScheduledAt    time.Time      `json:"scheduled_at"`
	ExecutedAt     sql.NullTime   `json:"executed_at"`
	Status         string         `json:"status"`
	ErrorMessage   sql.NullString `json:"error_message"`
	Metadata       string         `json:"metadata"`
	AttemptCount   int64          `json:"attempt_count"`
	MaxAttempts    int64          `json:"max_attempts"`
	IdempotencyKey sql.NullString `json:"idempotency_key"`
	CreatedAt      time.Time      `json:"created_at"`
	UpdatedAt      time.Time      `json:"updated_at"`
}

type SchemaMigration struct {
	Version   string       `json:"version"`
	AppliedAt sql.NullTime `json:"applied_at"`
	Name      string       `json:"name"`
	DownSql   string       `json:"down_sql"`
}

type SetTaskMetadataParams struct {
	TaskID string `json:"task_id"`
	Key    string `json:"key"`
	Value  string `json:"value"`
}

type SetWorkHasUnseenPRChangesParams struct {
	HasUnseenPrChanges bool   `json:"has_unseen_pr_changes"`
	ID                 string `json:"id"`
}

type SetWorkPRURLParams struct {
	PrUrl string `json:"pr_url"`
	ID    string `json:"id"`
}

type SpawnTaskParams struct {
	SpawnedAt   sql.NullTime `json:"spawned_at"`
	SpawnStatus string       `json:"spawn_status"`
	ID          string       `json:"id"`
}

type StartBeadParams struct {
	ID            string       `json:"id"`
	Title         string       `json:"title"`
	ZellijSession string       `json:"zellij_session"`
	ZellijPane    string       `json:"zellij_pane"`
	WorktreePath  string       `json:"worktree_path"`
	StartedAt     sql.NullTime `json:"started_at"`
	UpdatedAt     time.Time    `json:"updated_at"`
}

type StartTaskParams struct {
	WorktreePath string       `json:"worktree_path"`
	StartedAt    sql.NullTime `json:"started_at"`
	ID           string       `json:"id"`
}

type StartWorkParams struct {
	ZellijSession string       `json:"zellij_session"`
	ZellijTab     string       `json:"zellij_tab"`
	StartedAt     sql.NullTime `json:"started_at"`
	ID            string       `json:"id"`
}

type Task struct {
	ID               string       `json:"id"`
	Status           string       `json:"status"`
	TaskType         string       `json:"task_type"`
	ComplexityBudget int64        `json:"complexity_budget"`
	ActualComplexity int64        `json:"actual_complexity"`
	WorkID           string       `json:"work_id"`
	WorktreePath     string       `json:"worktree_path"`
	PrUrl            string       `json:"pr_url"`
	ErrorMessage     string       `json:"error_message"`
	StartedAt        sql.NullTime `json:"started_at"`
	CompletedAt      sql.NullTime `json:"completed_at"`
	CreatedAt        time.Time    `json:"created_at"`
	SpawnedAt        sql.NullTime `json:"spawned_at"`
	SpawnStatus      string       `json:"spawn_status"`
	LastActivity     sql.NullTime `json:"last_activity"`
}

type TaskBead struct {
	TaskID string `json:"task_id"`
	BeadID string `json:"bead_id"`
	Status string `json:"status"`
}

type TaskDependency struct {
	TaskID          string    `json:"task_id"`
	DependsOnTaskID string    `json:"depends_on_task_id"`
	CreatedAt       time.Time `json:"created_at"`
}

type TaskMetadatum struct {
	TaskID    string    `json:"task_id"`
	Key       string    `json:"key"`
	Value     string    `json:"value"`
	CreatedAt time.Time `json:"created_at"`
}

type UpdateHeartbeatWithTimeParams struct {
	Heartbeat time.Time `json:"heartbeat"`
	ID        string    `json:"id"`
}

type UpdateMigrationDownSQLParams struct {
	Name    string `json:"name"`
	DownSql string `json:"down_sql"`
	Version string `json:"version"`
}

type UpdateScheduledTaskTimeParams struct {
	ScheduledAt time.Time `json:"scheduled_at"`
	ID          string    `json:"id"`
}

type UpdateTaskActivityParams struct {
	LastActivity sql.NullTime `json:"last_activity"`
	ID           string       `json:"id"`
}

type UpdateWorkPRStatusParams struct {
	CiStatus       string       `json:"ci_status"`
	ApprovalStatus string       `json:"approval_status"`
	Approvers      string       `json:"approvers"`
	PrState        string       `json:"pr_state"`
	MergeableState string       `json:"mergeable_state"`
	LastPrPollAt   sql.NullTime `json:"last_pr_poll_at"`
	ID             string       `json:"id"`
}

type UpdateWorkWorktreePathParams struct {
	WorktreePath string `json:"worktree_path"`
	ID           string `json:"id"`
}

type Work struct {
	ID                 string       `json:"id"`
	Status             string       `json:"status"`
	Name               string       `json:"name"`
	ZellijSession      string       `json:"zellij_session"`
	ZellijTab          string       `json:"zellij_tab"`
	WorktreePath       string       `json:"worktree_path"`
	BranchName         string       `json:"branch_name"`
	BaseBranch         string       `json:"base_branch"`
	RootIssueID        string       `json:"root_issue_id"`
	PrUrl              string       `json:"pr_url"`
	ErrorMessage       string       `json:"error_message"`
	StartedAt          sql.NullTime `json:"started_at"`
	CompletedAt        sql.NullTime `json:"completed_at"`
	CreatedAt          time.Time    `json:"created_at"`
	Auto               bool         `json:"auto"`
	CiStatus           string       `json:"ci_status"`
	ApprovalStatus     string       `json:"approval_status"`
	Approvers          string       `json:"approvers"`
	LastPrPollAt       sql.NullTime `json:"last_pr_poll_at"`
	HasUnseenPrChanges bool         `json:"has_unseen_pr_changes"`
	PrState            string       `json:"pr_state"`
	MergeableState     string       `json:"mergeable_state"`
}

type WorkBead struct {
	WorkID    string    `json:"work_id"`
	BeadID    string    `json:"bead_id"`
	Position  int64     `json:"position"`
	CreatedAt time.Time `json:"created_at"`
}

type WorkTask struct {
	WorkID   string `json:"work_id"`
	TaskID   string `json:"task_id"`
	Position int64  `json:"position"`
}

type WorkTaskCounter struct {
	WorkID      string `json:"work_id"`
	NextTaskNum int64  `json:"next_task_num"`
}