db

type AuditLogDB interface {
	Add(context.Context, auditModels.Event) error
	Query(context.Context, QueryAuditLogOptions) ([]auditModels.Event, error)
}

type ConfigDB interface {
	Create(context.Context, models.Config) error
	Get(context.Context) (*models.Config, error)
}

type ContributorDB interface {
	Add(context.Context, models.Label, models.Email) (*models.Contributor, error)
	Get(context.Context, models.Label, models.Email) (*models.Contributor, error)
	List(context.Context, models.Label) ([]models.Contributor, error)
	Delete(context.Context, models.Label, models.Email) (*models.Contributor, error)
}

type DeleteStatus string

type GetSuggestionsOptions struct {
	Options *struct {
		Offset uint64
		Limit  uint64
	}

	FilterProjectLabel string
}

type Interface interface {
	Roles() RoleDB
	Users() UserDB
	Projects() ProjectsDB
	Suggestions() SuggestionsDB
	Contributors() ContributorDB
	Config() ConfigDB
	Secrets() SecretDB
	Tokens() TokensDB
	Session() SessionDB
	Recoveries() RecoveryDB
	OrgPolicyDB() OrgPolicyDB
	AuditLog() AuditLogDB
}

type ListPolicyOptions struct {
	Options *struct {
		Offset uint64
		Limit  uint64
	}

	FilterIDs []string
}

type ListRoleOptions struct {
	Offset uint64
	Limit  uint64
}

type ListUserOptions struct {
	Options *struct {
		Offset uint64
		Limit  uint64
	}

	FilterIDs []string
}

type Migrator interface {
	Up(context.Context) error
	Down(context.Context) error
}

type NewMigratorFunc func(dbURL *url.URL, migrations ...string) (Migrator, error)

type OrgPolicyDB interface {
	Get(context.Context) ([]models.Policy, error)
	AddPolicy(context.Context, string) error
	RemovePolicy(context.Context, string) error
}

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

type ProjectsDB interface {
	Get(context.Context, models.Label) (*models.Project, error)
	GetByID(context.Context, string) (*models.Project, error)
	Create(context.Context, models.Project) error
	Update(context.Context, models.Project) error

	List(context.Context) ([]models.Project, error)
	ListByStatus(context.Context, models.ProjectStatus) ([]models.Project, error)

	CreateProjectSpec(context.Context, models.Policy, SecretDB) error
	GetProjectSpec(context.Context, string, SecretDB) (*models.Policy, error)
}

type QueryAuditLogOptions struct {
	Offset     int
	Limit      int
	EventName  string
	GrepString string
}

type RecoveryDB interface {
	Get(context.Context, string) (*models.Recovery, error)
	Create(context.Context, models.Recovery) error
	Delete(context.Context, string) error
}

type RoleDB interface {
	Get(context.Context, models.Label) (*models.Role, error)
	GetByID(context.Context, string) (*models.Role, error)
	List(context.Context, *ListRoleOptions) ([]*models.Role, error)

	GetAll(context.Context, string) (*models.UserRoles, error)

	SetOrgRole(context.Context, models.Email, models.Label) (*models.Assignment, error)
	GetOrgRole(context.Context, models.Email) (*models.Role, error)
	SetProjectRole(context.Context, models.Email, models.Label, models.Label) (*models.Assignment, error)
	GetProjectRole(context.Context, models.Email, string) (*models.Role, error)

	CreateSystemRoles(context.Context) error
}

type SecretDB interface {
	Create(context.Context, models.SecretArg) error
	Delete(context.Context, string) (DeleteStatus, error)
	Get(context.Context, string) (*models.SecretArg, error)
}

type SessionDB interface {
	Get(context.Context, string) (*models.Session, error)
	Create(context.Context, models.Session) error
	Delete(context.Context, string) error
}

type SuggestionsDB interface {
	CreateSuggestion(context.Context, models.Suggestion) error
	GetSuggestions(context.Context, GetSuggestionsOptions) ([]models.Suggestion, error)
	GetSuggestion(context.Context, string) (*models.Suggestion, error)
	UpdateSuggestion(context.Context, models.Suggestion) error
}

type TokensDB interface {
	Get(context.Context, string) (*models.Token, error)
	Create(context.Context, models.Token) error
	Delete(context.Context, string) error
	ListByUserID(context.Context, string) ([]models.Token, error)
}

type UserDB interface {
	Create(context.Context, models.User) error
	Update(context.Context, string, models.User) error
	Delete(context.Context, models.Email) (DeleteStatus, error)
	Get(context.Context, models.Email) (*models.User, error)
	GetByID(context.Context, string) (*models.User, error)
	List(context.Context, *ListUserOptions) ([]models.User, error)
}