apitech
/
grafana
mirror of https://github.com/grafana/grafana
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Chore: Remove x from team (#47905)

Browse Source 
* Chore: Remove x from team

* Update pkg/services/sqlstore/team.go

Co-authored-by: ying-jeanne <74549700+ying-jeanne@users.noreply.github.com>

* Update pkg/services/sqlstore/team.go

Co-authored-by: ying-jeanne <74549700+ying-jeanne@users.noreply.github.com>

* Refactor dialects and add ISAdminOfTeams to Store

* Add IsAdminOfTeams to mockstore

Co-authored-by: ying-jeanne <74549700+ying-jeanne@users.noreply.github.com>
pull/48016/head
Kat Yang 4 years ago committed by GitHub
parent 72b5af8d9b
commit 68478e908a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 158 additions and 145 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 2
      pkg/middleware/auth.go
  2. 4
      pkg/services/sqlstore/mockstore/mockstore.go
  3. 1
      pkg/services/sqlstore/store.go
  4. 34
      pkg/services/sqlstore/team.go
  5. 4
      pkg/services/sqlstore/team_test.go

2
pkg/middleware/auth.go
Unescape View File

@ -207,7 +207,7 @@ func OrgAdminFolderAdminOrTeamAdmin(ss sqlstore.Store) func(c *models.ReqContext
} }
isAdminOfTeamsQuery := models.IsAdminOfTeamsQuery{SignedInUser: c.SignedInUser} isAdminOfTeamsQuery := models.IsAdminOfTeamsQuery{SignedInUser: c.SignedInUser}
if err := sqlstore.IsAdminOfTeams(c.Req.Context(), &isAdminOfTeamsQuery); err != nil { if err := ss.IsAdminOfTeams(c.Req.Context(), &isAdminOfTeamsQuery); err != nil {
c.JsonApiErr(500, "Failed to check if user is a team admin", err) c.JsonApiErr(500, "Failed to check if user is a team admin", err)
} }

4
pkg/services/sqlstore/mockstore/mockstore.go
Unescape View File

@ -672,3 +672,7 @@ func (m *SQLStoreMock) GetDashboardPermissionsForUser(ctx context.Context, query
func (m *SQLStoreMock) GetDashboardSlugById(ctx context.Context, query *models.GetDashboardSlugByIdQuery) error { func (m *SQLStoreMock) GetDashboardSlugById(ctx context.Context, query *models.GetDashboardSlugByIdQuery) error {
return m.ExpectedError return m.ExpectedError
} }
func (m *SQLStoreMock) IsAdminOfTeams(ctx context.Context, query *models.IsAdminOfTeamsQuery) error {
return m.ExpectedError
}

1
pkg/services/sqlstore/store.go
Unescape View File

@ -149,4 +149,5 @@ type Store interface {
HasAdminPermissionInFolders(ctx context.Context, query *models.HasAdminPermissionInFoldersQuery) error HasAdminPermissionInFolders(ctx context.Context, query *models.HasAdminPermissionInFoldersQuery) error
GetDashboardPermissionsForUser(ctx context.Context, query *models.GetDashboardPermissionsForUserQuery) error GetDashboardPermissionsForUser(ctx context.Context, query *models.GetDashboardPermissionsForUserQuery) error
GetDashboardSlugById(ctx context.Context, query *models.GetDashboardSlugByIdQuery) error GetDashboardSlugById(ctx context.Context, query *models.GetDashboardSlugByIdQuery) error
IsAdminOfTeams(ctx context.Context, query *models.IsAdminOfTeamsQuery) error
} }

34
pkg/services/sqlstore/team.go
Unescape View File

@ -176,6 +176,7 @@ func isTeamNameTaken(orgId int64, name string, existingId int64, sess *DBSession
} }
func (ss *SQLStore) SearchTeams(ctx context.Context, query *models.SearchTeamsQuery) error { func (ss *SQLStore) SearchTeams(ctx context.Context, query *models.SearchTeamsQuery) error {
return ss.WithDbSession(ctx, func(sess *DBSession) error {
query.Result = models.SearchTeamQueryResult{ query.Result = models.SearchTeamQueryResult{
Teams: make([]*models.TeamDTO, 0), Teams: make([]*models.TeamDTO, 0),
} }
@ -200,7 +201,7 @@ func (ss *SQLStore) SearchTeams(ctx context.Context, query *models.SearchTeamsQu
params = append(params, query.OrgId) params = append(params, query.OrgId)
if query.Query != "" { if query.Query != "" {
sql.WriteString(` and team.name ` + dialect.LikeStr() + ` ?`) sql.WriteString(` and team.name ` + ss.Dialect.LikeStr() + ` ?`)
params = append(params, queryWithWildcards) params = append(params, queryWithWildcards)
} }
@ -226,15 +227,15 @@ func (ss *SQLStore) SearchTeams(ctx context.Context, query *models.SearchTeamsQu
if query.Limit != 0 { if query.Limit != 0 {
offset := query.Limit * (query.Page - 1) offset := query.Limit * (query.Page - 1)
sql.WriteString(dialect.LimitOffset(int64(query.Limit), int64(offset))) sql.WriteString(ss.Dialect.LimitOffset(int64(query.Limit), int64(offset)))
} }
if err := x.SQL(sql.String(), params...).Find(&query.Result.Teams); err != nil { if err := sess.SQL(sql.String(), params...).Find(&query.Result.Teams); err != nil {
return err return err
} }
team := models.Team{} team := models.Team{}
countSess := x.Table("team") countSess := sess.Table("team")
countSess.Where("team.org_id=?", query.OrgId) countSess.Where("team.org_id=?", query.OrgId)
if query.Query != "" { if query.Query != "" {
@ -266,9 +267,11 @@ func (ss *SQLStore) SearchTeams(ctx context.Context, query *models.SearchTeamsQu
query.Result.TotalCount = count query.Result.TotalCount = count
return err return err
})
} }
func (ss *SQLStore) GetTeamById(ctx context.Context, query *models.GetTeamByIdQuery) error { func (ss *SQLStore) GetTeamById(ctx context.Context, query *models.GetTeamByIdQuery) error {
return ss.WithDbSession(ctx, func(sess *DBSession) error {
var sql bytes.Buffer var sql bytes.Buffer
params := make([]interface{}, 0) params := make([]interface{}, 0)
@ -287,7 +290,7 @@ func (ss *SQLStore) GetTeamById(ctx context.Context, query *models.GetTeamByIdQu
params = append(params, query.OrgId, query.Id) params = append(params, query.OrgId, query.Id)
var team models.TeamDTO var team models.TeamDTO
exists, err := x.SQL(sql.String(), params...).Get(&team) exists, err := sess.SQL(sql.String(), params...).Get(&team)
if err != nil { if err != nil {
return err return err
@ -299,6 +302,7 @@ func (ss *SQLStore) GetTeamById(ctx context.Context, query *models.GetTeamByIdQu
query.Result = &team query.Result = &team
return nil return nil
})
} }
// GetTeamsByUser is used by the Guardian when checking a users' permissions // GetTeamsByUser is used by the Guardian when checking a users' permissions
@ -513,7 +517,7 @@ func (ss *SQLStore) GetTeamMembers(ctx context.Context, query *models.GetTeamMem
// Note we assume that checking SignedInUser is allowed to see team members for this team has already been performed // Note we assume that checking SignedInUser is allowed to see team members for this team has already been performed
// If the signed in user is not set no member will be returned // If the signed in user is not set no member will be returned
if ss.Cfg.IsFeatureToggleEnabled(featuremgmt.FlagAccesscontrol) { if ss.Cfg.IsFeatureToggleEnabled(featuremgmt.FlagAccesscontrol) {
sqlID := fmt.Sprintf("%s.%s", x.Dialect().Quote("user"), x.Dialect().Quote("id")) sqlID := fmt.Sprintf("%s.%s", ss.engine.Dialect().Quote("user"), ss.engine.Dialect().Quote("id"))
*acFilter, err = ac.Filter(query.SignedInUser, sqlID, "users:id:", ac.ActionOrgUsersRead) *acFilter, err = ac.Filter(query.SignedInUser, sqlID, "users:id:", ac.ActionOrgUsersRead)
if err != nil { if err != nil {
return err return err
@ -525,10 +529,11 @@ func (ss *SQLStore) GetTeamMembers(ctx context.Context, query *models.GetTeamMem
// getTeamMembers return a list of members for the specified team // getTeamMembers return a list of members for the specified team
func (ss *SQLStore) getTeamMembers(ctx context.Context, query *models.GetTeamMembersQuery, acUserFilter *ac.SQLFilter) error { func (ss *SQLStore) getTeamMembers(ctx context.Context, query *models.GetTeamMembersQuery, acUserFilter *ac.SQLFilter) error {
return ss.WithDbSession(ctx, func(dbSess *DBSession) error {
query.Result = make([]*models.TeamMemberDTO, 0) query.Result = make([]*models.TeamMemberDTO, 0)
sess := x.Table("team_member") sess := dbSess.Table("team_member")
sess.Join("INNER", x.Dialect().Quote("user"), sess.Join("INNER", ss.Dialect.Quote("user"),
fmt.Sprintf("team_member.user_id=%s.%s", x.Dialect().Quote("user"), x.Dialect().Quote("id")), fmt.Sprintf("team_member.user_id=%s.%s", ss.Dialect.Quote("user"), ss.Dialect.Quote("id")),
) )
if acUserFilter != nil { if acUserFilter != nil {
@ -540,7 +545,7 @@ func (ss *SQLStore) getTeamMembers(ctx context.Context, query *models.GetTeamMem
SELECT id from user_auth SELECT id from user_auth
WHERE user_auth.user_id = team_member.user_id WHERE user_auth.user_id = team_member.user_id
ORDER BY user_auth.created DESC ` ORDER BY user_auth.created DESC `
authJoinCondition = "user_auth.id=" + authJoinCondition + dialect.Limit(1) + ")" authJoinCondition = "user_auth.id=" + authJoinCondition + ss.Dialect.Limit(1) + ")"
sess.Join("LEFT", "user_auth", authJoinCondition) sess.Join("LEFT", "user_auth", authJoinCondition)
if query.OrgId != 0 { if query.OrgId != 0 {
@ -553,7 +558,7 @@ func (ss *SQLStore) getTeamMembers(ctx context.Context, query *models.GetTeamMem
sess.Where("team_member.user_id=?", query.UserId) sess.Where("team_member.user_id=?", query.UserId)
} }
if query.External { if query.External {
sess.Where("team_member.external=?", dialect.BooleanStr(true)) sess.Where("team_member.external=?", ss.Dialect.BooleanStr(true))
} }
sess.Cols( sess.Cols(
"team_member.org_id", "team_member.org_id",
@ -570,9 +575,11 @@ func (ss *SQLStore) getTeamMembers(ctx context.Context, query *models.GetTeamMem
err := sess.Find(&query.Result) err := sess.Find(&query.Result)
return err return err
})
} }
func IsAdminOfTeams(ctx context.Context, query *models.IsAdminOfTeamsQuery) error { func (ss *SQLStore) IsAdminOfTeams(ctx context.Context, query *models.IsAdminOfTeamsQuery) error {
return ss.WithDbSession(ctx, func(sess *DBSession) error {
builder := &SQLBuilder{} builder := &SQLBuilder{}
builder.Write("SELECT COUNT(team.id) AS count FROM team INNER JOIN team_member ON team_member.team_id = team.id WHERE team.org_id = ? AND team_member.user_id = ? AND team_member.permission = ?", query.SignedInUser.OrgId, query.SignedInUser.UserId, models.PERMISSION_ADMIN) builder.Write("SELECT COUNT(team.id) AS count FROM team INNER JOIN team_member ON team_member.team_id = team.id WHERE team.org_id = ? AND team_member.user_id = ? AND team_member.permission = ?", query.SignedInUser.OrgId, query.SignedInUser.UserId, models.PERMISSION_ADMIN)
@ -581,11 +588,12 @@ func IsAdminOfTeams(ctx context.Context, query *models.IsAdminOfTeamsQuery) erro
} }
resp := make([]*teamCount, 0) resp := make([]*teamCount, 0)
if err := x.SQL(builder.GetSQLString(), builder.params...).Find(&resp); err != nil { if err := sess.SQL(builder.GetSQLString(), builder.params...).Find(&resp); err != nil {
return err return err
} }
query.Result = len(resp) > 0 && resp[0].Count > 0 query.Result = len(resp) > 0 && resp[0].Count > 0
return nil return nil
})
} }

4
pkg/services/sqlstore/team_test.go
Unescape View File

@ -302,12 +302,12 @@ func TestTeamCommandsAndQueries(t *testing.T) {
require.NoError(t, err) require.NoError(t, err)
query := &models.IsAdminOfTeamsQuery{SignedInUser: &models.SignedInUser{OrgId: testOrgID, UserId: userIds[0]}} query := &models.IsAdminOfTeamsQuery{SignedInUser: &models.SignedInUser{OrgId: testOrgID, UserId: userIds[0]}}
err = IsAdminOfTeams(context.Background(), query) err = sqlStore.IsAdminOfTeams(context.Background(), query)
require.NoError(t, err) require.NoError(t, err)
require.False(t, query.Result) require.False(t, query.Result)
query = &models.IsAdminOfTeamsQuery{SignedInUser: &models.SignedInUser{OrgId: testOrgID, UserId: userIds[1]}} query = &models.IsAdminOfTeamsQuery{SignedInUser: &models.SignedInUser{OrgId: testOrgID, UserId: userIds[1]}}
err = IsAdminOfTeams(context.Background(), query) err = sqlStore.IsAdminOfTeams(context.Background(), query)
require.NoError(t, err) require.NoError(t, err)
require.True(t, query.Result) require.True(t, query.Result)
}) })

Write Preview
Loading…
Cancel
Save