db

type ActiveScan interface {
	RecordCredential(Credential)
	Finish() error
}

type Branch struct {
	Model

	RepositoryID uint

	Name            string
	CredentialCount uint
}

type BranchCredentialCount struct {
	Owner string
	Name  string

	Branch string

	CredentialCount int
}

type BranchRepository interface {
	GetBranches(repository Repository) ([]Branch, error)
	UpdateBranches(repository Repository, branches []Branch) error

	GetCredentialCountByOwner() ([]OwnerCredentialCount, error)
	GetCredentialCountForOwner(owner string) ([]RepositoryCredentialCount, error)
	GetCredentialCountForRepo(owner string, repo string) ([]BranchCredentialCount, error)
}

type Commit struct {
	Model
	Owner      string
	Repository string
	SHA        string
}

type CommitRepository interface {
	RegisterCommit(logger lager.Logger, commit *Commit) error
	IsCommitRegistered(logger lager.Logger, sha string) (bool, error)
	IsRepoRegistered(logger lager.Logger, owner, repo string) (bool, error)
}

type Credential struct {
	Model

	Scan   Scan
	ScanID uint

	Owner      string
	Repository string
	SHA        string
	Path       string
	LineNumber int
	MatchStart int
	MatchEnd   int
	Private    bool
}

type CredentialRepository interface {
	ForScanWithID(int) ([]Credential, error)
	UniqueSHAsForRepoAndRulesVersion(Repository, int) ([]string, error)
	CredentialReported(cred *Credential) (bool, error)
}

type Fetch struct {
	Model
	Repository   *Repository
	RepositoryID uint

	Path    string
	Changes []byte
}

type FetchRepository interface {
	RegisterFetch(lager.Logger, *Fetch) error
}

type Model struct {
	ID        uint `gorm:"primary_key"`
	CreatedAt time.Time
	UpdatedAt time.Time
}

type OwnerCredentialCount struct {
	Owner           string
	CredentialCount int
}

type PriorScan struct {
	ID          int
	Branch      string
	StartSHA    string
	StopSHA     string
	Repository  string
	Owner       string
	Credentials []Credential
}

type Repository struct {
	Model

	Cloned bool

	Name          string
	Owner         string
	Path          string
	SSHURL        string `gorm:"column:ssh_url"`
	Private       bool
	DefaultBranch string

	FailedFetches int `gorm:"column:failed_fetches"`
	Disabled      bool
}

type RepositoryCredentialCount struct {
	Owner   string
	Name    string
	Private bool

	CredentialCount int
}

type RepositoryRepository interface {
	Create(*Repository) error

	Update(*Repository) error

	Delete(*Repository) error

	Find(owner, name string) (Repository, bool, error)
	MustFind(owner, name string) (Repository, error)

	All() ([]Repository, error)
	Active() ([]Repository, error)
	AllForOrganization(string) ([]Repository, error)
	NotScannedWithVersion(int) ([]Repository, error)

	MarkAsCloned(owner, name, path string) error
	Reenable(owner, name string) error
	RegisterFailedFetch(lager.Logger, *Repository) error
}

type Scan struct {
	Model

	Type         string
	RulesVersion int

	ScanStart time.Time
	ScanEnd   time.Time

	Branch   string
	StartSHA string
	StopSHA  string

	Repository   *Repository
	RepositoryID *uint

	Credentials []Credential
}

type ScanRepository interface {
	Start(lager.Logger, string, string, string, string, *Repository) ActiveScan
	ScansNotYetRunWithVersion(lager.Logger, int) ([]PriorScan, error)
}

type StatsRepository interface {
	RepositoryCount() (int, error)
	DisabledRepositoryCount() (int, error)
	UnclonedRepositoryCount() (int, error)
	CredentialCount() (int, error)
	FetchCount() (int, error)
}