datasources

type CreateParams struct {
	Slug           slug.Slug
	Name           string
	Type           model.DataSourceType
	ConfigJSON     json.RawMessage
	CredentialSlug slug.Slug
	ReadOnly       bool
}

type CredentialMetadata struct {
	Slug                 slug.Slug       `json:"slug"`
	Kind                 string          `json:"kind"`
	Status               string          `json:"status"`
	Provider             string          `json:"provider"`
	UpdatedAt            time.Time       `json:"updated_at"`
	AccessTokenExpiresAt *time.Time      `json:"access_token_expires_at"`
	Scopes               []string        `json:"scopes"`
	Account              ProviderAccount `json:"account"`
	LastRefreshedAt      *time.Time      `json:"last_refreshed_at"`
	RevokedAt            *time.Time      `json:"revoked_at"`
	FailureCode          string          `json:"failure_code"`
	FailureMessage       string          `json:"failure_message"`
}

type DataSource struct {
	OrganizationGitLabConnection bool
	Slug                         slug.Slug
	Name                         string
	Type                         model.DataSourceType
	Config                       json.RawMessage
	EncryptedCredentials         []byte
	IsActive                     bool
	ReadOnly                     bool
	SchemaImportStatus           model.DataSourceSchemaImportStatus
	SchemaImportDiagnostic       *SchemaImportDiagnostic
	SchemaImportStartedAt        *time.Time
	SchemaImportFinishedAt       *time.Time
	LastTestedAt                 *time.Time
	TestError                    *string
	ConnectionTestRevision       int64
	Schema                       json.RawMessage
	CreatedAt                    time.Time
	UpdatedAt                    time.Time
	Credential                   *CredentialMetadata
}

type MarkSchemaImportFailedParams struct {
	AttemptID    string
	ErrorCode    string
	ErrorMessage string
	FinishedAt   time.Time
}

type MarkTestSuccessParams struct {
	ExpectedRevision int64
	TestedAt         time.Time
}

type ProviderAccount struct {
	ID    string `json:"ID"`
	Name  string `json:"Name"`
	Email string `json:"Email"`
}

type Repository interface {
	// Create inserts a new data source and returns it.
	Create(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug slug.Slug,
		params *CreateParams,
	) (*DataSource, error)

	// FindBySlug fetches a data source by its slug within an organization.
	FindBySlug(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
	) (*DataSource, error)

	// IsSlugReserved reports whether an active or soft-deleted data source reserves the slug.
	IsSlugReserved(ctx context.Context, tx postgres.DBTX, orgSlug, dsSlug slug.Slug) (bool, error)

	// ListByOrg lists all data sources in an organization.
	ListByOrg(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug slug.Slug,
		req model.ListRequest,
	) ([]*DataSource, error)

	// Update updates an existing data source by slug.
	Update(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
		params *UpdateParams,
	) (*DataSource, error)

	// UpdateCredentials updates credentials for an existing data source by slug.
	UpdateCredentials(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
		params *UpdateCredentialsParams,
	) (*DataSource, error)

	// MarkTestSuccess marks a data source as active and stores the introspected schema.
	MarkTestSuccess(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
		params *MarkTestSuccessParams,
	) (*DataSource, error)

	// MarkTestFailure marks a data source as inactive and records the test error.
	MarkTestFailure(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
		expectedRevision int64,
		errMsg string,
		schema json.RawMessage,
	) (*DataSource, error)

	// Delete soft-deletes a data source by slug.
	Delete(ctx context.Context, tx postgres.DBTX, orgSlug, dsSlug slug.Slug) error

	// DataSourceHasProjectBackendReferences reports whether any project uses this data source as its pipeline backend.
	DataSourceHasProjectBackendReferences(ctx context.Context, tx postgres.DBTX, orgSlug, dsSlug slug.Slug) (bool, error)

	// LockDataSourceForUpdate locks an active data source row for update within the current transaction.
	LockDataSourceForUpdate(ctx context.Context, tx postgres.DBTX, orgSlug, dsSlug slug.Slug) error

	// Restore clears the deletion tombstone for a data source by slug.
	Restore(ctx context.Context, tx postgres.DBTX, orgSlug, dsSlug slug.Slug) error

	// SetReadOnly sets the read_only flag for a data source.
	SetReadOnly(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
		readOnly bool,
	) error
}

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

type SchemaImportDiagnostic struct {
	Code    string
	Message string
}

type SchemaImportState struct {
	DataSourceSlug slug.Slug
	Status         model.DataSourceSchemaImportStatus
	Diagnostic     *SchemaImportDiagnostic
	StartedAt      *time.Time
	FinishedAt     *time.Time
	// Started is true when this call transitioned the import into importing instead of reusing an in-flight attempt.
	Started bool
}

type SchemaMetadata struct {
	Slug       slug.Slug
	SchemaData json.RawMessage
	CreatedAt  time.Time
	UpdatedAt  time.Time
}

type SchemaMetadataRepository interface {
	FindByDataSourceSlug(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
	) (*SchemaMetadata, error)
	DeleteByDataSourceSlug(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
	) error
	UpsertByDataSourceSlug(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
		params *UpsertSchemaMetadataParams,
	) (*SchemaMetadata, error)
	FindImportStateByDataSourceSlug(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
	) (*SchemaImportState, error)
	StartImportByDataSourceSlug(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
		attemptID string,
		startedAt time.Time,
	) (*SchemaImportState, error)
	FindCurrentImportAttemptByDataSourceSlug(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
		attemptID string,
	) (*SchemaImportState, error)
	MarkImportReadyByDataSourceSlug(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
		attemptID string,
		finishedAt time.Time,
	) (*SchemaImportState, error)
	MarkImportFailedByDataSourceSlug(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
		params *MarkSchemaImportFailedParams,
	) (*SchemaImportState, error)
	ResetImportByDataSourceSlug(
		ctx context.Context,
		tx postgres.DBTX,
		orgSlug, dsSlug slug.Slug,
	) (*SchemaImportState, error)
}

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

type UpdateCredentialsParams struct {
	CredentialSlug slug.Slug
}

type UpdateParams struct {
	Name        string
	ConfigJSON  json.RawMessage
	IsActive    *bool
	ReadOnly    bool
	ReadOnlySet bool
}

type UpsertSchemaMetadataParams struct {
	SchemaData json.RawMessage
}