model

type Agent struct {
	gorm.Model
	ID         uuid.UUID `json:"id" gorm:"primaryKey"`
	Status     string
	StatusInfo string
	CredUrl    string
	Version    string
	SourceID   uuid.UUID
}

type AgentList []Agent

type Assessment struct {
	ID             uuid.UUID `gorm:"primaryKey;column:id;type:VARCHAR(255);"`
	CreatedAt      time.Time `gorm:"not null;default:now()"`
	UpdatedAt      *time.Time
	Name           string     `gorm:"not null;uniqueIndex:org_id_name"`
	OrgID          string     `gorm:"not null;uniqueIndex:org_id_name;index:assessments_org_id_idx"`
	Username       string     `gorm:"type:VARCHAR(255)"`
	OwnerFirstName *string    `gorm:"type:VARCHAR(100)"`
	OwnerLastName  *string    `gorm:"type:VARCHAR(100)"`
	SourceType     string     `gorm:"not null;type:VARCHAR(100)"`
	SourceID       *uuid.UUID `gorm:"type:TEXT"`
	Snapshots      []Snapshot `gorm:"foreignKey:AssessmentID;references:ID;constraint:OnDelete:CASCADE;"`
}

type AssessmentList []Assessment

type ImageInfra struct {
	gorm.Model
	SourceID         uuid.UUID `gorm:"primaryKey"`
	HttpProxyUrl     string
	HttpsProxyUrl    string
	NoProxyDomains   string
	CertificateChain string
	SshPublicKey     string
	ImageTokenKey    string
	IpAddress        string
	SubnetMask       string
	DefaultGateway   string
	Dns              string
}

type InventoryStats struct {
	Vms              VmStats
	Os               OsStats
	TotalCustomers   int
	TotalAssessments int
	TotalInventories int
	Storage          []StorageCustomerStats
}

type JSONField[T any] struct {
	Data T
}

type Key struct {
	gorm.Model
	OrgID      string          `gorm:"primaryKey"`
	ID         string          `gorm:"not null;unique"`
	PrivateKey *rsa.PrivateKey `gorm:"type:text;not null;serializer:key_serializer"`
}

type KeySerializer struct{}

type Label struct {
	Key      string `gorm:"primaryKey;column:key;type:VARCHAR;size:100;"`
	Value    string `gorm:"column:value;type:VARCHAR;size:100;not null"`
	SourceID string `gorm:"primaryKey;column:source_id;type:TEXT;"`
}

type OsStats struct {
	// TotalTypes is total of os types. (e.g. how many different os we found)
	Total int
}

type Snapshot struct {
	ID           uint                      `gorm:"primaryKey;autoIncrement"`
	CreatedAt    time.Time                 `gorm:"not null;default:now()"`
	Inventory    *JSONField[api.Inventory] `gorm:"type:jsonb;not null"`
	AssessmentID uuid.UUID                 `gorm:"not null;type:VARCHAR(255);"`
}

type Source struct {
	gorm.Model
	ID          uuid.UUID `gorm:"primaryKey;"`
	Name        string    `gorm:"uniqueIndex:name_org_id;not null"`
	VCenterID   string
	Username    string
	OrgID       string                    `gorm:"uniqueIndex:name_org_id;not null"`
	Inventory   *JSONField[api.Inventory] `gorm:"type:jsonb"`
	OnPremises  bool
	Agents      []Agent    `gorm:"constraint:OnDelete:CASCADE;"`
	ImageInfra  ImageInfra `gorm:"constraint:OnDelete:CASCADE;"`
	Labels      []Label    `gorm:"foreignKey:SourceID;references:ID;constraint:OnDelete:CASCADE;"`
	EmailDomain *string
}

type SourceList []Source

type StorageCustomerStats struct {
	TotalByProvider map[string]int
	Domain          string
}

type VmStats struct {
	// Total is the total number of vms
	Total int
	// Total number of vm by customer (key is organization id)
	TotalByCustomer map[string]int
	// Total number of vm by OS.
	TotalByOS map[string]int
}