db

type ImageDto struct {
	Id        int            `gorm:"column:Id;primaryKey;<-:false"`
	Name      string         `gorm:"column:Name;->;<-:create"`
	BaseImage string         `gorm:"column:BaseImage;->;<-:create"`
	Scans     []ImageScanDto `gorm:"foreignKey:ImageId"`
}

type ImageScanDto struct {
	Id              int                `gorm:"column:Id;primaryKey;<-:false"`
	ImageId         int                `gorm:"column:ImageId;<-:false"`
	ScanTime        time.Time          `gorm:"column:ScanTime;<-:false"`
	ScanSuccess     bool               `gorm:"column:ScanSuccess;<-:false"`
	Vulnerabilities []VulnerabilityDto `gorm:"many2many:ImageScanVulnerabilities;joinForeignKey:ImageScanId;joinReferences:VulnerabilityId"` // https://gorm.io/docs/many_to_many.html
}

type Repository interface {
	// GetLastImageScan returns the last vulnerability scan for an image
	GetLastImageScan(ctx context.Context, image string) (*ImageScanDto, error)
	// RegisterImageScan stores information about a vulnerability scan for an image
	RegisterImageScan(ctx context.Context, image string, baseImage *string, scanTime time.Time, success bool, vulnerabilities []VulnerabilityBulkDto, identifiers []VulnerabilityIdentifierBulkDto, references []VulnerabilityReferenceBulkDto) error
}

type VulnerabilityBulkDto struct {
	ExternalId       string
	PackageName      string
	Version          string
	Title            string
	Description      string
	Severity         string
	CVSS             *float32
	CVSSv3           string
	CreationTime     time.Time
	ModificationTime time.Time
	PublicationTime  time.Time
	DisclosureTime   *time.Time
}

type VulnerabilityDto struct {
	Id               int                          `gorm:"column:Id;primaryKey;<-:false"`
	ExternalId       string                       `gorm:"column:ExternalId;<-:false"`
	PackageName      string                       `gorm:"column:PackageName;<-:false"`
	Version          string                       `gorm:"column:Version;<-:false"`
	Description      string                       `gorm:"column:Description;<-:false"`
	Title            string                       `gorm:"column:Title;<-:false"`
	Severity         string                       `gorm:"column:Severity;<-:false"`
	CVSS             *float32                     `gorm:"column:CVSS;<-:false"`
	CVSSv3           string                       `gorm:"column:CVSSv3;<-:false"`
	CreationTime     time.Time                    `gorm:"column:CreationTime;<-:false"`
	ModificationTime time.Time                    `gorm:"column:ModificationTime;<-:false"`
	PublicationTime  time.Time                    `gorm:"column:PublicationTime;<-:false"`
	DisclosureTime   time.Time                    `gorm:"column:DisclosureTime;<-:false"`
	References       []VulnerabilityReferenceDto  `gorm:"foreignKey:VulnerabilityId"`
	Identifiers      []VulnerabilityIdentifierDto `gorm:"foreignKey:VulnerabilityId"`
}

type VulnerabilityIdentifierBulkDto struct {
	ExternalId     string
	IdentifierType string
	Identifier     string
}

type VulnerabilityIdentifierDto struct {
	VulnerabilityId int    `gorm:"column:VulnerabilityId;primaryKey;<-:false"`
	IdentifierType  string `gorm:"column:IdentifierType;primaryKey;<-:false"`
	Identifier      string `gorm:"column:Identifier;primaryKey;<-:false"`
}

type VulnerabilityReferenceBulkDto struct {
	ExternalId string
	Url        string
}

type VulnerabilityReferenceDto struct {
	VulnerabilityId int    `gorm:"column:VulnerabilityId;primaryKey;<-:false"`
	Url             string `gorm:"column:Url;primaryKey;<-:false"`
}