db

type CreateFileParams struct {
	ID        string `json:"id"`
	SessionID string `json:"session_id"`
	Path      string `json:"path"`
	Content   string `json:"content"`
	Version   string `json:"version"`
}

type CreateFlowStateParams struct {
	SessionID      string         `json:"session_id"`
	RootSessionID  sql.NullString `json:"root_session_id"`
	FlowID         string         `json:"flow_id"`
	StepID         string         `json:"step_id"`
	Status         string         `json:"status"`
	Args           sql.NullString `json:"args"`
	Output         sql.NullString `json:"output"`
	IsStructOutput bool           `json:"is_struct_output"`
}

type CreateMessageParams struct {
	ID        string         `json:"id"`
	SessionID string         `json:"session_id"`
	Role      string         `json:"role"`
	Parts     string         `json:"parts"`
	Model     sql.NullString `json:"model"`
	Seq       sql.NullInt64  `json:"seq"`
}

type CreateSessionParams struct {
	ID               string         `json:"id"`
	ProjectID        sql.NullString `json:"project_id"`
	ParentSessionID  sql.NullString `json:"parent_session_id"`
	RootSessionID    sql.NullString `json:"root_session_id"`
	Title            string         `json:"title"`
	MessageCount     int64          `json:"message_count"`
	PromptTokens     int64          `json:"prompt_tokens"`
	CompletionTokens int64          `json:"completion_tokens"`
	Cost             float64        `json:"cost"`
}

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 DeleteFlowStatesByRootSessionParams struct {
	RootSessionID sql.NullString `json:"root_session_id"`
	Column2       interface{}    `json:"column_2"`
}

type File struct {
	ID        string `json:"id"`
	SessionID string `json:"session_id"`
	Path      string `json:"path"`
	Content   string `json:"content"`
	Version   string `json:"version"`
	CreatedAt int64  `json:"created_at"`
	UpdatedAt int64  `json:"updated_at"`
}

type FlowState struct {
	SessionID      string         `json:"session_id"`
	RootSessionID  sql.NullString `json:"root_session_id"`
	FlowID         string         `json:"flow_id"`
	StepID         string         `json:"step_id"`
	Status         string         `json:"status"`
	Args           sql.NullString `json:"args"`
	Output         sql.NullString `json:"output"`
	IsStructOutput bool           `json:"is_struct_output"`
	CreatedAt      int64          `json:"created_at"`
	UpdatedAt      int64          `json:"updated_at"`
}

type GetFileByPathAndSessionParams struct {
	Path      string `json:"path"`
	SessionID string `json:"session_id"`
}

type ListFlowStatesByRootSessionParams struct {
	RootSessionID sql.NullString `json:"root_session_id"`
	Column2       interface{}    `json:"column_2"`
}

type Message struct {
	ID         string         `json:"id"`
	SessionID  string         `json:"session_id"`
	Role       string         `json:"role"`
	Parts      string         `json:"parts"`
	Model      sql.NullString `json:"model"`
	CreatedAt  int64          `json:"created_at"`
	UpdatedAt  int64          `json:"updated_at"`
	FinishedAt sql.NullInt64  `json:"finished_at"`
	Seq        sql.NullInt64  `json:"seq"`
}

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

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

type Provider interface {
	// Connect establishes a connection to the database and returns a *sql.DB instance.
	// It should handle provider-specific connection setup, including:
	// - Connection string/DSN building
	// - Connection pooling configuration
	// - Provider-specific optimizations (pragmas, settings, etc.)
	// - Connection verification (ping)
	Connect() (*sql.DB, error)

	// Type returns the provider type (sqlite or mysql).
	Type() config.ProviderType

	// Dialect returns the SQL dialect name for migration purposes.
	Dialect() string
}

type Querier interface {
	CreateFile(ctx context.Context, arg CreateFileParams) (File, error)
	CreateFlowState(ctx context.Context, arg CreateFlowStateParams) (FlowState, error)
	CreateMessage(ctx context.Context, arg CreateMessageParams) (Message, error)
	CreateSession(ctx context.Context, arg CreateSessionParams) (Session, error)
	DeleteFile(ctx context.Context, id string) error
	DeleteFlowStatesByRootSession(ctx context.Context, arg DeleteFlowStatesByRootSessionParams) error
	DeleteMessage(ctx context.Context, id string) error
	DeleteSession(ctx context.Context, id string) error
	DeleteSessionFiles(ctx context.Context, sessionID string) error
	DeleteSessionMessages(ctx context.Context, sessionID string) error
	GetFile(ctx context.Context, id string) (File, error)
	GetFileByPathAndSession(ctx context.Context, arg GetFileByPathAndSessionParams) (File, error)
	GetFlowState(ctx context.Context, sessionID string) (FlowState, error)
	GetMaxSeqBySession(ctx context.Context, sessionID string) (int64, error)
	GetMessage(ctx context.Context, id string) (Message, error)
	GetSessionByID(ctx context.Context, id string) (Session, error)
	ListChildSessions(ctx context.Context, rootSessionID sql.NullString) ([]Session, error)
	ListFilesByPath(ctx context.Context, path string) ([]File, error)
	ListFilesBySession(ctx context.Context, sessionID string) ([]File, error)
	ListFilesBySessionTree(ctx context.Context, rootSessionID sql.NullString) ([]File, error)
	ListFlowStatesByFlowID(ctx context.Context, flowID string) ([]FlowState, error)
	ListFlowStatesByRootSession(ctx context.Context, arg ListFlowStatesByRootSessionParams) ([]FlowState, error)
	ListLatestSessionFiles(ctx context.Context, sessionID string) ([]File, error)
	ListLatestSessionTreeFiles(ctx context.Context, rootSessionID sql.NullString) ([]File, error)
	ListMessagesBySession(ctx context.Context, sessionID string) ([]Message, error)
	ListSessions(ctx context.Context, projectID sql.NullString) ([]Session, error)
	UpdateFile(ctx context.Context, arg UpdateFileParams) (File, error)
	UpdateFlowState(ctx context.Context, arg UpdateFlowStateParams) (FlowState, error)
	UpdateMessage(ctx context.Context, arg UpdateMessageParams) error
	UpdateSession(ctx context.Context, arg UpdateSessionParams) (Session, error)
}

type QuerierWithTx interface {
	Querier
	WithTx(tx *sql.Tx) QuerierWithTx
}

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

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

type Session struct {
	ID               string         `json:"id"`
	ParentSessionID  sql.NullString `json:"parent_session_id"`
	Title            string         `json:"title"`
	MessageCount     int64          `json:"message_count"`
	PromptTokens     int64          `json:"prompt_tokens"`
	CompletionTokens int64          `json:"completion_tokens"`
	Cost             float64        `json:"cost"`
	UpdatedAt        int64          `json:"updated_at"`
	CreatedAt        int64          `json:"created_at"`
	SummaryMessageID sql.NullString `json:"summary_message_id"`
	ProjectID        sql.NullString `json:"project_id"`
	RootSessionID    sql.NullString `json:"root_session_id"`
}

type UpdateFileParams struct {
	Content string `json:"content"`
	Version string `json:"version"`
	ID      string `json:"id"`
}

type UpdateFlowStateParams struct {
	Status         string         `json:"status"`
	Args           sql.NullString `json:"args"`
	Output         sql.NullString `json:"output"`
	IsStructOutput bool           `json:"is_struct_output"`
	SessionID      string         `json:"session_id"`
}

type UpdateMessageParams struct {
	Parts      string        `json:"parts"`
	FinishedAt sql.NullInt64 `json:"finished_at"`
	ID         string        `json:"id"`
}

type UpdateSessionParams struct {
	Title            string         `json:"title"`
	PromptTokens     int64          `json:"prompt_tokens"`
	CompletionTokens int64          `json:"completion_tokens"`
	SummaryMessageID sql.NullString `json:"summary_message_id"`
	Cost             float64        `json:"cost"`
	ID               string         `json:"id"`
}