vikunja-api/pkg/models/teams.go

97 lines
2.4 KiB
Go
Raw Normal View History

package models
// Team holds a team object
type Team struct {
2018-07-21 15:08:46 +02:00
ID int64 `xorm:"int(11) autoincr not null unique pk" json:"id" param:"team"`
2018-07-14 17:34:59 +02:00
Name string `xorm:"varchar(250) not null" json:"name"`
Description string `xorm:"varchar(250)" json:"description"`
2018-10-05 18:46:01 +02:00
CreatedByID int64 `xorm:"int(11) not null INDEX" json:"-"`
2018-07-14 17:34:59 +02:00
2018-07-16 08:43:47 +02:00
CreatedBy User `xorm:"-" json:"created_by"`
Members []*TeamUser `xorm:"-" json:"members"`
Created int64 `xorm:"created" json:"created"`
Updated int64 `xorm:"updated" json:"updated"`
2018-07-14 17:34:59 +02:00
CRUDable `xorm:"-" json:"-"`
Rights `xorm:"-" json:"-"`
}
// TableName makes beautiful table names
func (Team) TableName() string {
return "teams"
}
2018-07-14 17:37:46 +02:00
// AfterLoad gets the created by user object
2018-07-14 17:34:59 +02:00
func (t *Team) AfterLoad() {
// Get the owner
2018-08-30 19:14:02 +02:00
t.CreatedBy, _ = GetUserByID(t.CreatedByID)
2018-07-16 08:43:47 +02:00
// Get all members
x.Select("*").
Table("users").
Join("INNER", "team_members", "team_members.user_id = users.id").
Where("team_id = ?", t.ID).
Find(&t.Members)
2018-07-14 17:34:59 +02:00
}
// TeamMember defines the relationship between a user and a team
type TeamMember struct {
2018-07-26 10:08:01 +02:00
ID int64 `xorm:"int(11) autoincr not null unique pk" json:"id"`
2018-10-05 18:46:01 +02:00
TeamID int64 `xorm:"int(11) not null INDEX" json:"team_id" param:"team"`
UserID int64 `xorm:"int(11) not null INDEX" json:"user_id" param:"user"`
Admin bool `xorm:"tinyint(1) INDEX" json:"admin"`
2018-07-14 17:34:59 +02:00
Created int64 `xorm:"created" json:"created"`
Updated int64 `xorm:"updated" json:"updated"`
2018-07-14 18:29:24 +02:00
CRUDable `xorm:"-" json:"-"`
Rights `xorm:"-" json:"-"`
}
// TableName makes beautiful table names
func (TeamMember) TableName() string {
return "team_members"
}
2018-07-16 08:45:38 +02:00
// TeamUser is the team member type
2018-07-16 08:43:47 +02:00
type TeamUser struct {
2018-07-26 10:08:01 +02:00
User `xorm:"extends"`
2018-07-26 09:53:32 +02:00
Admin bool `json:"admin"`
2018-07-16 08:43:47 +02:00
}
2018-07-14 18:29:24 +02:00
// GetTeamByID gets a team by its ID
func GetTeamByID(id int64) (team Team, err error) {
if id < 1 {
return team, ErrTeamDoesNotExist{id}
}
2018-07-14 18:29:24 +02:00
exists, err := x.Where("id = ?", id).Get(&team)
if err != nil {
return
}
if !exists {
return team, ErrTeamDoesNotExist{id}
}
return
}
2018-07-16 08:43:47 +02:00
// ReadOne implements the CRUD method to get one team
2018-07-21 15:08:46 +02:00
func (t *Team) ReadOne() (err error) {
*t, err = GetTeamByID(t.ID)
2018-07-16 08:43:47 +02:00
return
}
// ReadAll gets all teams the user is part of
func (t *Team) ReadAll(user *User) (teams interface{}, err error) {
all := []*Team{}
err = x.Select("teams.*").
Table("teams").
Join("INNER", "team_members", "team_members.team_id = teams.id").
Where("team_members.user_id = ?", user.ID).
Find(&all)
return all, err
}