diff options
Diffstat (limited to 'store')
-rw-r--r-- | store/sql_audit_store.go | 2 | ||||
-rw-r--r-- | store/sql_audit_store_test.go | 2 | ||||
-rw-r--r-- | store/sql_channel_store.go | 5 | ||||
-rw-r--r-- | store/sql_channel_store_test.go | 2 | ||||
-rw-r--r-- | store/sql_oauth_store.go | 2 | ||||
-rw-r--r-- | store/sql_oauth_store_test.go | 2 | ||||
-rw-r--r-- | store/sql_post_store.go | 2 | ||||
-rw-r--r-- | store/sql_post_store_test.go | 2 | ||||
-rw-r--r-- | store/sql_session_store.go | 2 | ||||
-rw-r--r-- | store/sql_session_store_test.go | 2 | ||||
-rw-r--r-- | store/sql_store.go | 9 | ||||
-rw-r--r-- | store/sql_store_test.go | 2 | ||||
-rw-r--r-- | store/sql_system_store.go | 2 | ||||
-rw-r--r-- | store/sql_system_store_test.go | 2 | ||||
-rw-r--r-- | store/sql_team_store.go | 3 | ||||
-rw-r--r-- | store/sql_team_store_test.go | 2 | ||||
-rw-r--r-- | store/sql_user_store.go | 34 | ||||
-rw-r--r-- | store/sql_user_store_test.go | 25 | ||||
-rw-r--r-- | store/sql_webhook_store.go | 2 | ||||
-rw-r--r-- | store/sql_webhook_store_test.go | 2 | ||||
-rw-r--r-- | store/store.go | 3 |
21 files changed, 79 insertions, 30 deletions
diff --git a/store/sql_audit_store.go b/store/sql_audit_store.go index a595d50e2..898cf8f78 100644 --- a/store/sql_audit_store.go +++ b/store/sql_audit_store.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_audit_store_test.go b/store/sql_audit_store_test.go index 4d804d150..e265db837 100644 --- a/store/sql_audit_store_test.go +++ b/store/sql_audit_store_test.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_channel_store.go b/store/sql_channel_store.go index 3bbe7e716..56e190fee 100644 --- a/store/sql_channel_store.go +++ b/store/sql_channel_store.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store @@ -38,6 +38,8 @@ func NewSqlChannelStore(sqlStore *SqlStore) ChannelStore { } func (s SqlChannelStore) UpgradeSchemaIfNeeded() { + + // BEGIN REMOVE AFTER 1.1.0 if s.CreateColumnIfNotExists("ChannelMembers", "NotifyProps", "varchar(2000)", "varchar(2000)", "{}") { // populate NotifyProps from existing NotifyLevel field @@ -80,6 +82,7 @@ func (s SqlChannelStore) UpgradeSchemaIfNeeded() { s.RemoveColumnIfExists("ChannelMembers", "NotifyLevel") } + // END REMOVE AFTER 1.1.0 } func (s SqlChannelStore) CreateIndexesIfNotExists() { diff --git a/store/sql_channel_store_test.go b/store/sql_channel_store_test.go index b6d05684b..b4e0f7593 100644 --- a/store/sql_channel_store_test.go +++ b/store/sql_channel_store_test.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_oauth_store.go b/store/sql_oauth_store.go index 2a6fa3118..db52e379b 100644 --- a/store/sql_oauth_store.go +++ b/store/sql_oauth_store.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_oauth_store_test.go b/store/sql_oauth_store_test.go index 08e1388e0..3f05f1c92 100644 --- a/store/sql_oauth_store_test.go +++ b/store/sql_oauth_store_test.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_post_store.go b/store/sql_post_store.go index 8d62eaad0..07077bd64 100644 --- a/store/sql_post_store.go +++ b/store/sql_post_store.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_post_store_test.go b/store/sql_post_store_test.go index 62d7b0100..9a7679454 100644 --- a/store/sql_post_store_test.go +++ b/store/sql_post_store_test.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_session_store.go b/store/sql_session_store.go index 22411389d..15ec6924b 100644 --- a/store/sql_session_store.go +++ b/store/sql_session_store.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_session_store_test.go b/store/sql_session_store_test.go index 3d8aafe25..068e5fc19 100644 --- a/store/sql_session_store_test.go +++ b/store/sql_session_store_test.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_store.go b/store/sql_store.go index 3ef9cfbc4..900543460 100644 --- a/store/sql_store.go +++ b/store/sql_store.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store @@ -84,13 +84,6 @@ func NewSqlStore() Store { } } - // Temporary upgrade code, remove after 0.8.0 release - if sqlStore.DoesTableExist("Sessions") { - if sqlStore.DoesColumnExist("Sessions", "AltId") { - sqlStore.GetMaster().Exec("DROP TABLE IF EXISTS Sessions") - } - } - sqlStore.team = NewSqlTeamStore(sqlStore) sqlStore.channel = NewSqlChannelStore(sqlStore) sqlStore.post = NewSqlPostStore(sqlStore) diff --git a/store/sql_store_test.go b/store/sql_store_test.go index 84dbf5705..a9e25cb33 100644 --- a/store/sql_store_test.go +++ b/store/sql_store_test.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_system_store.go b/store/sql_system_store.go index ca22de2a6..a4cb52d4d 100644 --- a/store/sql_system_store.go +++ b/store/sql_system_store.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_system_store_test.go b/store/sql_system_store_test.go index 0f03b8f0e..8ff5445cc 100644 --- a/store/sql_system_store_test.go +++ b/store/sql_system_store_test.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_team_store.go b/store/sql_team_store.go index 109fe5401..de44782cf 100644 --- a/store/sql_team_store.go +++ b/store/sql_team_store.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store @@ -28,7 +28,6 @@ func NewSqlTeamStore(sqlStore *SqlStore) TeamStore { } func (s SqlTeamStore) UpgradeSchemaIfNeeded() { - s.RemoveColumnIfExists("Teams", "AllowValet") } func (s SqlTeamStore) CreateIndexesIfNotExists() { diff --git a/store/sql_team_store_test.go b/store/sql_team_store_test.go index 1f13e466c..3d9b4d435 100644 --- a/store/sql_team_store_test.go +++ b/store/sql_team_store_test.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_user_store.go b/store/sql_user_store.go index 0a723d965..011acd7e4 100644 --- a/store/sql_user_store.go +++ b/store/sql_user_store.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store @@ -41,7 +41,6 @@ func NewSqlUserStore(sqlStore *SqlStore) UserStore { } func (us SqlUserStore) UpgradeSchemaIfNeeded() { - us.CreateColumnIfNotExists("Users", "ThemeProps", "varchar(2000)", "character varying(2000)", "{}") } func (us SqlUserStore) CreateIndexesIfNotExists() { @@ -370,6 +369,37 @@ func (us SqlUserStore) GetProfiles(teamId string) StoreChannel { return storeChannel } +func (us SqlUserStore) GetSystemAdminProfiles() StoreChannel { + + storeChannel := make(StoreChannel) + + go func() { + result := StoreResult{} + + var users []*model.User + + if _, err := us.GetReplica().Select(&users, "SELECT * FROM Users WHERE Roles = :Roles", map[string]interface{}{"Roles": "system_admin"}); err != nil { + result.Err = model.NewAppError("SqlUserStore.GetSystemAdminProfiles", "We encounted an error while finding user profiles", err.Error()) + } else { + + userMap := make(map[string]*model.User) + + for _, u := range users { + u.Password = "" + u.AuthData = "" + userMap[u.Id] = u + } + + result.Data = userMap + } + + storeChannel <- result + close(storeChannel) + }() + + return storeChannel +} + func (us SqlUserStore) GetByEmail(teamId string, email string) StoreChannel { storeChannel := make(StoreChannel) diff --git a/store/sql_user_store_test.go b/store/sql_user_store_test.go index e2a454023..be21c8bd2 100644 --- a/store/sql_user_store_test.go +++ b/store/sql_user_store_test.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store @@ -259,6 +259,29 @@ func TestUserStoreGetProfiles(t *testing.T) { } } +func TestUserStoreGetSystemAdminProfiles(t *testing.T) { + Setup() + + u1 := model.User{} + u1.TeamId = model.NewId() + u1.Email = model.NewId() + Must(store.User().Save(&u1)) + + u2 := model.User{} + u2.TeamId = u1.TeamId + u2.Email = model.NewId() + Must(store.User().Save(&u2)) + + if r1 := <-store.User().GetSystemAdminProfiles(); r1.Err != nil { + t.Fatal(r1.Err) + } else { + users := r1.Data.(map[string]*model.User) + if len(users) <= 0 { + t.Fatal("invalid returned system admin users") + } + } +} + func TestUserStoreGetByEmail(t *testing.T) { Setup() diff --git a/store/sql_webhook_store.go b/store/sql_webhook_store.go index e309f79e4..42a91a80e 100644 --- a/store/sql_webhook_store.go +++ b/store/sql_webhook_store.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/sql_webhook_store_test.go b/store/sql_webhook_store_test.go index 0a015eaf9..6f4ef4354 100644 --- a/store/sql_webhook_store_test.go +++ b/store/sql_webhook_store_test.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store diff --git a/store/store.go b/store/store.go index 887913bc6..1c4d08e36 100644 --- a/store/store.go +++ b/store/store.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store @@ -104,6 +104,7 @@ type UserStore interface { UpdateFailedPasswordAttempts(userId string, attempts int) StoreChannel GetForExport(teamId string) StoreChannel GetTotalUsersCount() StoreChannel + GetSystemAdminProfiles() StoreChannel } type SessionStore interface { |