package teamSeverity

import (
	"houston/model/severity"
	"houston/model/team"
)

type TeamSeverityEntity struct {
	ID         uint `gorm:"primaryKey"`
	TeamID     uint `gorm:"column:team_id;not null"`
	SeverityID uint `gorm:"column:severity_id;not null"`
	Sla        int  `gorm:"column:sla"`

	// Add foreign key constraints
	Team     team.TeamEntity         `gorm:"foreignKey:TeamID"`
	Severity severity.SeverityEntity `gorm:"foreignKey:SeverityID"`
}

func (TeamSeverityEntity) TableName() string {
	return "team_severity"
}

func (entity TeamSeverityEntity) ToDTO() TeamSeverityDTO {
	return TeamSeverityDTO{
		ID:          entity.ID,
		TeamID:      entity.TeamID,
		SeverityID:  entity.SeverityID,
		Sla:         entity.Sla,
		TeamDTO:     *entity.Team.ToDTO(),
		SeverityDTO: entity.Severity.ToDTO(),
	}
}