d1

type CastRule struct {
	SourceType string `json:"source_type"`
	TargetType string `json:"target_type"`
	Using      string `json:"using,omitempty"`
	Tables     string `json:"tables,omitempty"`
}

type ColumnSamples map[string]map[string][]string

type ColumnSchema struct {
	Name          string
	Type          string
	PrimaryKey    bool
	AutoIncrement bool
	NotNull       bool
	Unique        bool
	DefaultValue  string
	ForeignKey    string
	// CheckExprs holds raw (unconverted) column-level CHECK expressions, e.g.
	// ["age >= 0"] for `age INTEGER CHECK (age >= 0)`.
	CheckExprs []string
	// GeneratedExpr holds the raw (unconverted) expression of a
	// `GENERATED ALWAYS AS (...)` computed column, empty if the column is not generated.
	GeneratedExpr string
	// GeneratedMode is "STORED" or "VIRTUAL" as declared in the source DDL, empty if
	// GeneratedExpr is empty. SQLite defaults to VIRTUAL when omitted.
	GeneratedMode string
}

type CompleteResult struct {
	MigrationID      string   `json:"migration_id"`
	Status           string   `json:"status"`
	SkippedORMTables []string `json:"skipped_orm_tables,omitempty"`
}

type DefaultImportClient struct {
	Client *ps.Client
}

type DoctorCheck struct {
	Name        string `json:"name"`
	Status      string `json:"status"`
	Version     string `json:"version,omitempty"`
	Message     string `json:"message,omitempty"`
	Remediation string `json:"remediation,omitempty"`
}

type DoctorResult struct {
	Checks []DoctorCheck `json:"checks"`
	Ready  bool          `json:"ready"`
}

type ErrorInfo struct {
	Code        string `json:"code"`
	Message     string `json:"message"`
	Remediation string `json:"remediation,omitempty"`
}

type ImportClient interface {
	GetDatabase(ctx context.Context, org, database string) (*ps.Database, error)
}

type ImportOptions struct {
	Org            string
	Database       string
	Branch         string
	InputPath      string
	Method         string
	MigrationID    string
	DBName         string
	DBNameExplicit bool
	DryRun         bool
	DestURI        string // optional override for testing
	NotifyAPI      NotifyAPIConfig
	OnProgress     ImportProgressFunc
	// PgloaderVerbose emits full pgloader reports to stderr (defaults to false).
	PgloaderVerbose bool
	// contains filtered or unexported fields
}

type ImportPrepareResult struct {
	MigrationID   string      `json:"migration_id"`
	Method        string      `json:"method"`
	Lint          *LintResult `json:"lint"`
	Plan          *PlanResult `json:"plan"`
	CanProceed    bool        `json:"can_proceed"`
	BlockedReason string      `json:"blocked_reason,omitempty"`
}

type ImportProgress struct {
	Stage   string `json:"stage"`
	Current int    `json:"current,omitempty"`
	Total   int    `json:"total,omitempty"`
	Detail  string `json:"detail,omitempty"`
}

type ImportProgressFunc func(ImportProgress)

type ImportResult struct {
	MigrationID  string         `json:"migration_id"`
	Method       string         `json:"method"`
	DryRun       bool           `json:"dry_run"`
	TablesLoaded int            `json:"tables_loaded,omitempty"`
	Timings      *ImportTimings `json:"timings,omitempty"`
	Lint         *LintResult    `json:"lint,omitempty"`
	Plan         *PlanResult    `json:"plan,omitempty"`
	CanProceed   bool           `json:"can_proceed"`
}

type ImportTimings struct {
	TotalMs         int64             `json:"total_ms"`
	SQLiteStagingMs int64             `json:"sqlite_staging_ms,omitempty"`
	SchemaMs        int64             `json:"schema_ms,omitempty"`
	PgloaderMs      int64             `json:"pgloader_ms,omitempty"`
	IndexBuildMs    int64             `json:"index_build_ms,omitempty"`
	SequenceResetMs int64             `json:"sequence_reset_ms,omitempty"`
	TableLoads      []TableLoadTiming `json:"table_loads,omitempty"`
}

type IndexSchema struct {
	Name    string
	Table   string
	Unique  bool
	Columns string
	RawDDL  string
}

type Issue struct {
	Code        string `json:"code"`
	Severity    string `json:"severity"`
	Table       string `json:"table,omitempty"`
	Column      string `json:"column,omitempty"`
	Message     string `json:"message,omitempty"`
	Remediation string `json:"remediation"`
}

type LintResult struct {
	InputPath    string   `json:"input_path"`
	TableCount   int      `json:"table_count"`
	ErrorCount   int      `json:"error_count"`
	WarningCount int      `json:"warning_count"`
	Issues       []Issue  `json:"issues"`
	Tables       []string `json:"tables"`
}

type MigrationError struct {
	Info ErrorInfo
}

type MigrationState struct {
	MigrationID   string    `json:"migration_id"`
	Org           string    `json:"org"`
	Database      string    `json:"database"`
	Branch        string    `json:"branch"`
	InputPath     string    `json:"input_path"`
	SQLitePath    string    `json:"sqlite_path,omitempty"`
	DBName        string    `json:"db_name,omitempty"`
	Method        string    `json:"method,omitempty"`
	Phase         string    `json:"phase"`
	SchemaApplied bool      `json:"schema_applied,omitempty"`
	LoadedTables  []string  `json:"loaded_tables,omitempty"`
	CreatedAt     time.Time `json:"created_at"`
	UpdatedAt     time.Time `json:"updated_at"`
}

type NextStep struct {
	Tool    string `json:"tool,omitempty"`
	Command string `json:"command,omitempty"`
	Reason  string `json:"reason"`
}

type NotifyAPIConfig struct {
	Client *ps.Client
	// Disabled skips notifications (internal use and tests).
	Disabled bool
}

type PgloaderOptions struct {
	SQLitePath string
	DestURI    string
	InputPath  string // dump path for column-level CAST rules
	WorkDir    string
	DryRun     bool
	DataOnly   bool
	// Tables loads one table per pgloader invocation when set (recommended for
	// large databases — avoids SBCL heap exhaustion from whole-catalog planning).
	Tables []string
	// SkipTables skips tables already loaded during a resumed import.
	SkipTables []string
	// OnTableLoaded is called after each table load succeeds (for resume checkpoints).
	OnTableLoaded func(table string) error
	// OnProgress reports per-table load progress.
	OnProgress ImportProgressFunc
	// PgloaderVerbose writes full pgloader output to stderr after each table.
	PgloaderVerbose bool
}

type PlanOptions struct {
	InputPath   string
	Org         string
	Database    string
	Branch      string
	Method      string
	MigrationID string      // optional: reuse an existing migration ID from plan/start
	Lint        *LintResult // optional: skip re-lint when already computed
}

type PlanResult struct {
	MigrationID        string      `json:"migration_id"`
	InputPath          string      `json:"input_path"`
	Org                string      `json:"org"`
	Database           string      `json:"database"`
	Branch             string      `json:"branch"`
	RecommendedMethod  string      `json:"recommended_method"`
	EstimatedSizeBytes int64       `json:"estimated_size_bytes,omitempty"`
	Tables             []TablePlan `json:"tables"`
	CastRules          []CastRule  `json:"cast_rules"`
	LoadOrder          []string    `json:"load_order"`
	Issues             []Issue     `json:"issues"`
}

type Response struct {
	Status      string     `json:"status"`
	Command     string     `json:"command,omitempty"`
	Phase       string     `json:"phase,omitempty"`
	MigrationID string     `json:"migration_id,omitempty"`
	Issues      []Issue    `json:"issues,omitempty"`
	NextSteps   []NextStep `json:"next_steps,omitempty"`
	Reminder    string     `json:"reminder,omitempty"`
	Data        any        `json:"data,omitempty"`
	Error       *ErrorInfo `json:"error,omitempty"`
}

type SchemaParts struct {
	Tables  string
	Indexes string
}

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

type TableLoadTiming struct {
	Table string `json:"table"`
	Ms    int64  `json:"ms"`
}

type TablePlan struct {
	Name        string `json:"name"`
	RowEstimate int    `json:"row_estimate,omitempty"`
	HasFK       bool   `json:"has_foreign_keys"`
}

type TableSchema struct {
	Name        string
	Columns     []ColumnSchema
	Constraints []string
	RawDDL      string
}

type TableVerifyResult struct {
	Table      string `json:"table"`
	SourceRows int64  `json:"source_rows"`
	DestRows   int64  `json:"dest_rows"`
	Match      bool   `json:"match"`
}

type TypeCoercionContext struct {
	Samples ColumnSamples
}

type VerifyCheckResult struct {
	Name    string `json:"name"`
	Table   string `json:"table,omitempty"`
	Column  string `json:"column,omitempty"`
	Matched bool   `json:"matched"`
	Message string `json:"message,omitempty"`
	Source  string `json:"source,omitempty"`
	Dest    string `json:"dest,omitempty"`
}

type VerifyOptions struct {
	Org         string
	Database    string
	Branch      string
	MigrationID string
	InputPath   string
	SQLitePath  string
	DestURI     string
	DBName      string // destination PostgreSQL database name (default postgres)
	NotifyAPI   NotifyAPIConfig
	OnProgress  ImportProgressFunc
	// contains filtered or unexported fields
}

type VerifyResult struct {
	MigrationID string              `json:"migration_id"`
	Matched     bool                `json:"matched"`
	Tables      []TableVerifyResult `json:"tables"`
	Checks      []VerifyCheckResult `json:"checks,omitempty"`
}