package teamUser

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

type TeamUserEntity struct {
	ID     uint `gorm:"primaryKey"`
	TeamID uint `gorm:"column:team_id;not null"`
	UserID uint `gorm:"column:user_id;not null"`

	// Add foreign key constraints
	Team team.TeamEntity `gorm:"foreignKey:TeamID"`
	User user.UserEntity `gorm:"foreignKey:UserID"`
}

func (TeamUserEntity) TableName() string {
	return "team_user"
}

func (entity TeamUserEntity) ToDTO() *TeamUserDTO {
	return &TeamUserDTO{
		ID:     entity.ID,
		TeamID: entity.TeamID,
		UserID: entity.UserID,
		Team:   *entity.Team.ToDTO(),
		User:   *entity.User.ToDTO(),
	}
}