diff options
Diffstat (limited to 'store/sql_team_store.go')
-rw-r--r-- | store/sql_team_store.go | 195 |
1 files changed, 195 insertions, 0 deletions
diff --git a/store/sql_team_store.go b/store/sql_team_store.go new file mode 100644 index 000000000..6e7fc1c1e --- /dev/null +++ b/store/sql_team_store.go @@ -0,0 +1,195 @@ +// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// See License.txt for license information. + +package store + +import ( + "github.com/mattermost/platform/model" + "strings" +) + +type SqlTeamStore struct { + *SqlStore +} + +func NewSqlTeamStore(sqlStore *SqlStore) TeamStore { + s := &SqlTeamStore{sqlStore} + + for _, db := range sqlStore.GetAllConns() { + table := db.AddTableWithName(model.Team{}, "Teams").SetKeys(false, "Id") + table.ColMap("Id").SetMaxSize(26) + table.ColMap("Name").SetMaxSize(64) + table.ColMap("Domain").SetMaxSize(64).SetUnique(true) + table.ColMap("Email").SetMaxSize(128) + table.ColMap("CompanyName").SetMaxSize(64) + table.ColMap("AllowedDomains").SetMaxSize(500) + } + + return s +} + +func (s SqlTeamStore) UpgradeSchemaIfNeeded() { +} + +func (s SqlTeamStore) CreateIndexesIfNotExists() { +} + +func (s SqlTeamStore) Save(team *model.Team) StoreChannel { + storeChannel := make(StoreChannel) + + go func() { + result := StoreResult{} + + if len(team.Id) > 0 { + result.Err = model.NewAppError("SqlTeamStore.Save", + "Must call update for exisiting team", "id="+team.Id) + storeChannel <- result + close(storeChannel) + return + } + + team.PreSave() + if result.Err = team.IsValid(); result.Err != nil { + storeChannel <- result + close(storeChannel) + return + } + + if err := s.GetMaster().Insert(team); err != nil { + if strings.Contains(err.Error(), "Duplicate entry") && strings.Contains(err.Error(), "for key 'Domain'") { + result.Err = model.NewAppError("SqlTeamStore.Save", "A team with that domain already exists", "id="+team.Id+", "+err.Error()) + } else { + result.Err = model.NewAppError("SqlTeamStore.Save", "We couldn't save the team", "id="+team.Id+", "+err.Error()) + } + } else { + result.Data = team + } + + storeChannel <- result + close(storeChannel) + }() + + return storeChannel +} + +func (s SqlTeamStore) Update(team *model.Team) StoreChannel { + + storeChannel := make(StoreChannel) + + go func() { + result := StoreResult{} + + team.PreUpdate() + + if result.Err = team.IsValid(); result.Err != nil { + storeChannel <- result + close(storeChannel) + return + } + + if oldResult, err := s.GetMaster().Get(model.Team{}, team.Id); err != nil { + result.Err = model.NewAppError("SqlTeamStore.Update", "We encounted an error finding the team", "id="+team.Id+", "+err.Error()) + } else if oldResult == nil { + result.Err = model.NewAppError("SqlTeamStore.Update", "We couldn't find the existing team to update", "id="+team.Id) + } else { + oldTeam := oldResult.(*model.Team) + team.CreateAt = oldTeam.CreateAt + team.Domain = oldTeam.Domain + + if count, err := s.GetMaster().Update(team); err != nil { + result.Err = model.NewAppError("SqlTeamStore.Update", "We encounted an error updating the team", "id="+team.Id+", "+err.Error()) + } else if count != 1 { + result.Err = model.NewAppError("SqlTeamStore.Update", "We couldn't update the team", "id="+team.Id) + } else { + result.Data = team + } + } + + storeChannel <- result + close(storeChannel) + }() + + return storeChannel +} + +func (s SqlTeamStore) UpdateName(name string, teamId string) StoreChannel { + + storeChannel := make(StoreChannel) + + go func() { + result := StoreResult{} + + if _, err := s.GetMaster().Exec("UPDATE Teams SET Name = ? WHERE Id = ?", name, teamId); err != nil { + result.Err = model.NewAppError("SqlTeamStore.UpdateName", "We couldn't update the team name", "team_id="+teamId) + } else { + result.Data = teamId + } + + storeChannel <- result + close(storeChannel) + }() + + return storeChannel +} + +func (s SqlTeamStore) Get(id string) StoreChannel { + storeChannel := make(StoreChannel) + + go func() { + result := StoreResult{} + + if obj, err := s.GetReplica().Get(model.Team{}, id); err != nil { + result.Err = model.NewAppError("SqlTeamStore.Get", "We encounted an error finding the team", "id="+id+", "+err.Error()) + } else if obj == nil { + result.Err = model.NewAppError("SqlTeamStore.Get", "We couldn't find the existing team", "id="+id) + } else { + result.Data = obj.(*model.Team) + } + + storeChannel <- result + close(storeChannel) + }() + + return storeChannel +} + +func (s SqlTeamStore) GetByDomain(domain string) StoreChannel { + storeChannel := make(StoreChannel) + + go func() { + result := StoreResult{} + + team := model.Team{} + + if err := s.GetReplica().SelectOne(&team, "SELECT * FROM Teams WHERE Domain=?", domain); err != nil { + result.Err = model.NewAppError("SqlTeamStore.GetByDomain", "We couldn't find the existing team", "domain="+domain+", "+err.Error()) + } + + result.Data = &team + + storeChannel <- result + close(storeChannel) + }() + + return storeChannel +} + +func (s SqlTeamStore) GetTeamsForEmail(email string) StoreChannel { + storeChannel := make(StoreChannel) + + go func() { + result := StoreResult{} + + var data []*model.Team + if _, err := s.GetReplica().Select(&data, "SELECT Teams.* FROM Teams, Users WHERE Teams.Id = Users.TeamId AND Users.Email = ?", email); err != nil { + result.Err = model.NewAppError("SqlTeamStore.GetTeamsForEmail", "We encounted a problem when looking up teams", "email="+email+", "+err.Error()) + } + + result.Data = data + + storeChannel <- result + close(storeChannel) + }() + + return storeChannel +} |