|
|
@@ -18,17 +18,30 @@ func init() {
|
|
|
bus.AddHandler("sql", GetTeamsByUser)
|
|
|
|
|
|
bus.AddHandler("sql", AddTeamMember)
|
|
|
+ bus.AddHandler("sql", UpdateTeamMember)
|
|
|
bus.AddHandler("sql", RemoveTeamMember)
|
|
|
bus.AddHandler("sql", GetTeamMembers)
|
|
|
}
|
|
|
|
|
|
+func getTeamSearchSqlBase() string {
|
|
|
+ return `SELECT
|
|
|
+ team.id as id,
|
|
|
+ team.org_id,
|
|
|
+ team.name as name,
|
|
|
+ team.email as email,
|
|
|
+ (SELECT COUNT(*) from team_member where team_member.team_id = team.id) as member_count,
|
|
|
+ team_member.permission
|
|
|
+ FROM team as team
|
|
|
+ INNER JOIN team_member on team.id = team_member.team_id AND team_member.user_id = ? `
|
|
|
+}
|
|
|
+
|
|
|
func getTeamSelectSqlBase() string {
|
|
|
return `SELECT
|
|
|
team.id as id,
|
|
|
team.org_id,
|
|
|
team.name as name,
|
|
|
team.email as email,
|
|
|
- (SELECT COUNT(*) from team_member where team_member.team_id = team.id) as member_count
|
|
|
+ (SELECT COUNT(*) from team_member where team_member.team_id = team.id) as member_count
|
|
|
FROM team as team `
|
|
|
}
|
|
|
|
|
|
@@ -91,10 +104,8 @@ func UpdateTeam(cmd *m.UpdateTeamCommand) error {
|
|
|
// DeleteTeam will delete a team, its member and any permissions connected to the team
|
|
|
func DeleteTeam(cmd *m.DeleteTeamCommand) error {
|
|
|
return inTransaction(func(sess *DBSession) error {
|
|
|
- if teamExists, err := teamExists(cmd.OrgId, cmd.Id, sess); err != nil {
|
|
|
+ if _, err := teamExists(cmd.OrgId, cmd.Id, sess); err != nil {
|
|
|
return err
|
|
|
- } else if !teamExists {
|
|
|
- return m.ErrTeamNotFound
|
|
|
}
|
|
|
|
|
|
deletes := []string{
|
|
|
@@ -117,7 +128,7 @@ func teamExists(orgId int64, teamId int64, sess *DBSession) (bool, error) {
|
|
|
if res, err := sess.Query("SELECT 1 from team WHERE org_id=? and id=?", orgId, teamId); err != nil {
|
|
|
return false, err
|
|
|
} else if len(res) != 1 {
|
|
|
- return false, nil
|
|
|
+ return false, m.ErrTeamNotFound
|
|
|
}
|
|
|
|
|
|
return true, nil
|
|
|
@@ -147,7 +158,12 @@ func SearchTeams(query *m.SearchTeamsQuery) error {
|
|
|
var sql bytes.Buffer
|
|
|
params := make([]interface{}, 0)
|
|
|
|
|
|
- sql.WriteString(getTeamSelectSqlBase())
|
|
|
+ if query.UserIdFilter > 0 {
|
|
|
+ sql.WriteString(getTeamSearchSqlBase())
|
|
|
+ params = append(params, query.UserIdFilter)
|
|
|
+ } else {
|
|
|
+ sql.WriteString(getTeamSelectSqlBase())
|
|
|
+ }
|
|
|
sql.WriteString(` WHERE team.org_id = ?`)
|
|
|
|
|
|
params = append(params, query.OrgId)
|
|
|
@@ -233,19 +249,18 @@ func AddTeamMember(cmd *m.AddTeamMemberCommand) error {
|
|
|
return m.ErrTeamMemberAlreadyAdded
|
|
|
}
|
|
|
|
|
|
- if teamExists, err := teamExists(cmd.OrgId, cmd.TeamId, sess); err != nil {
|
|
|
+ if _, err := teamExists(cmd.OrgId, cmd.TeamId, sess); err != nil {
|
|
|
return err
|
|
|
- } else if !teamExists {
|
|
|
- return m.ErrTeamNotFound
|
|
|
}
|
|
|
|
|
|
entity := m.TeamMember{
|
|
|
- OrgId: cmd.OrgId,
|
|
|
- TeamId: cmd.TeamId,
|
|
|
- UserId: cmd.UserId,
|
|
|
- External: cmd.External,
|
|
|
- Created: time.Now(),
|
|
|
- Updated: time.Now(),
|
|
|
+ OrgId: cmd.OrgId,
|
|
|
+ TeamId: cmd.TeamId,
|
|
|
+ UserId: cmd.UserId,
|
|
|
+ External: cmd.External,
|
|
|
+ Created: time.Now(),
|
|
|
+ Updated: time.Now(),
|
|
|
+ Permission: cmd.Permission,
|
|
|
}
|
|
|
|
|
|
_, err := sess.Insert(&entity)
|
|
|
@@ -253,13 +268,59 @@ func AddTeamMember(cmd *m.AddTeamMemberCommand) error {
|
|
|
})
|
|
|
}
|
|
|
|
|
|
+func getTeamMember(sess *DBSession, orgId int64, teamId int64, userId int64) (m.TeamMember, error) {
|
|
|
+ rawSql := `SELECT * FROM team_member WHERE org_id=? and team_id=? and user_id=?`
|
|
|
+ var member m.TeamMember
|
|
|
+ exists, err := sess.SQL(rawSql, orgId, teamId, userId).Get(&member)
|
|
|
+
|
|
|
+ if err != nil {
|
|
|
+ return member, err
|
|
|
+ }
|
|
|
+ if !exists {
|
|
|
+ return member, m.ErrTeamMemberNotFound
|
|
|
+ }
|
|
|
+
|
|
|
+ return member, nil
|
|
|
+}
|
|
|
+
|
|
|
+// UpdateTeamMember updates a team member
|
|
|
+func UpdateTeamMember(cmd *m.UpdateTeamMemberCommand) error {
|
|
|
+ return inTransaction(func(sess *DBSession) error {
|
|
|
+ member, err := getTeamMember(sess, cmd.OrgId, cmd.TeamId, cmd.UserId)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ if cmd.ProtectLastAdmin {
|
|
|
+ _, err := isLastAdmin(sess, cmd.OrgId, cmd.TeamId, cmd.UserId)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if cmd.Permission != m.PERMISSION_ADMIN { // make sure we don't get invalid permission levels in store
|
|
|
+ cmd.Permission = 0
|
|
|
+ }
|
|
|
+
|
|
|
+ member.Permission = cmd.Permission
|
|
|
+ _, err = sess.Cols("permission").Where("org_id=? and team_id=? and user_id=?", cmd.OrgId, cmd.TeamId, cmd.UserId).Update(member)
|
|
|
+
|
|
|
+ return err
|
|
|
+ })
|
|
|
+}
|
|
|
+
|
|
|
// RemoveTeamMember removes a member from a team
|
|
|
func RemoveTeamMember(cmd *m.RemoveTeamMemberCommand) error {
|
|
|
return inTransaction(func(sess *DBSession) error {
|
|
|
- if teamExists, err := teamExists(cmd.OrgId, cmd.TeamId, sess); err != nil {
|
|
|
+ if _, err := teamExists(cmd.OrgId, cmd.TeamId, sess); err != nil {
|
|
|
return err
|
|
|
- } else if !teamExists {
|
|
|
- return m.ErrTeamNotFound
|
|
|
+ }
|
|
|
+
|
|
|
+ if cmd.ProtectLastAdmin {
|
|
|
+ _, err := isLastAdmin(sess, cmd.OrgId, cmd.TeamId, cmd.UserId)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
var rawSql = "DELETE FROM team_member WHERE org_id=? and team_id=? and user_id=?"
|
|
|
@@ -276,6 +337,29 @@ func RemoveTeamMember(cmd *m.RemoveTeamMemberCommand) error {
|
|
|
})
|
|
|
}
|
|
|
|
|
|
+func isLastAdmin(sess *DBSession, orgId int64, teamId int64, userId int64) (bool, error) {
|
|
|
+ rawSql := "SELECT user_id FROM team_member WHERE org_id=? and team_id=? and permission=?"
|
|
|
+ userIds := []*int64{}
|
|
|
+ err := sess.SQL(rawSql, orgId, teamId, m.PERMISSION_ADMIN).Find(&userIds)
|
|
|
+ if err != nil {
|
|
|
+ return false, err
|
|
|
+ }
|
|
|
+
|
|
|
+ isAdmin := false
|
|
|
+ for _, adminId := range userIds {
|
|
|
+ if userId == *adminId {
|
|
|
+ isAdmin = true
|
|
|
+ break
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if isAdmin && len(userIds) == 1 {
|
|
|
+ return true, m.ErrLastTeamAdmin
|
|
|
+ }
|
|
|
+
|
|
|
+ return false, err
|
|
|
+}
|
|
|
+
|
|
|
// GetTeamMembers return a list of members for the specified team
|
|
|
func GetTeamMembers(query *m.GetTeamMembersQuery) error {
|
|
|
query.Result = make([]*m.TeamMemberDTO, 0)
|
|
|
@@ -293,7 +377,7 @@ func GetTeamMembers(query *m.GetTeamMembersQuery) error {
|
|
|
if query.External {
|
|
|
sess.Where("team_member.external=?", dialect.BooleanStr(true))
|
|
|
}
|
|
|
- sess.Cols("team_member.org_id", "team_member.team_id", "team_member.user_id", "user.email", "user.login", "team_member.external")
|
|
|
+ sess.Cols("team_member.org_id", "team_member.team_id", "team_member.user_id", "user.email", "user.login", "team_member.external", "team_member.permission")
|
|
|
sess.Asc("user.login", "user.email")
|
|
|
|
|
|
err := sess.Find(&query.Result)
|