diff options
author | Christopher Speller <crspeller@gmail.com> | 2015-12-08 13:38:43 -0500 |
---|---|---|
committer | Christopher Speller <crspeller@gmail.com> | 2015-12-16 17:30:15 -0500 |
commit | 58358ddd7cd0152bf16a7326e1d595524fb51246 (patch) | |
tree | 350cd462f9b530529e0f098fa1d458c3a36abd4a /model/gitlab | |
parent | 4f881046bf2a4c74fb44d71e2e78826c70719a8c (diff) | |
download | chat-58358ddd7cd0152bf16a7326e1d595524fb51246.tar.gz chat-58358ddd7cd0152bf16a7326e1d595524fb51246.tar.bz2 chat-58358ddd7cd0152bf16a7326e1d595524fb51246.zip |
Some refactoring
Diffstat (limited to 'model/gitlab')
-rw-r--r-- | model/gitlab/gitlab.go | 84 |
1 files changed, 84 insertions, 0 deletions
diff --git a/model/gitlab/gitlab.go b/model/gitlab/gitlab.go new file mode 100644 index 000000000..8b96c64f6 --- /dev/null +++ b/model/gitlab/gitlab.go @@ -0,0 +1,84 @@ +// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. +// See License.txt for license information. + +package oauthgitlab + +import ( + "encoding/json" + "github.com/mattermost/platform/einterfaces" + "github.com/mattermost/platform/model" + "io" + "strconv" + "strings" +) + +const ( + USER_AUTH_SERVICE_GITLAB = "gitlab" +) + +type GitLabProvider struct { +} + +type GitLabUser struct { + Id int64 `json:"id"` + Username string `json:"username"` + Login string `json:"login"` + Email string `json:"email"` + Name string `json:"name"` +} + +func init() { + provider := &GitLabProvider{} + einterfaces.RegisterOauthProvider(USER_AUTH_SERVICE_GITLAB, provider) +} + +func userFromGitLabUser(glu *GitLabUser) *model.User { + user := &model.User{} + username := glu.Username + if username == "" { + username = glu.Login + } + user.Username = model.CleanUsername(username) + splitName := strings.Split(glu.Name, " ") + if len(splitName) == 2 { + user.FirstName = splitName[0] + user.LastName = splitName[1] + } else if len(splitName) >= 2 { + user.FirstName = splitName[0] + user.LastName = strings.Join(splitName[1:], " ") + } else { + user.FirstName = glu.Name + } + user.Email = glu.Email + user.AuthData = strconv.FormatInt(glu.Id, 10) + user.AuthService = USER_AUTH_SERVICE_GITLAB + + return user +} + +func gitLabUserFromJson(data io.Reader) *GitLabUser { + decoder := json.NewDecoder(data) + var glu GitLabUser + err := decoder.Decode(&glu) + if err == nil { + return &glu + } else { + return nil + } +} + +func (glu *GitLabUser) getAuthData() string { + return strconv.FormatInt(glu.Id, 10) +} + +func (m *GitLabProvider) GetIdentifier() string { + return USER_AUTH_SERVICE_GITLAB +} + +func (m *GitLabProvider) GetUserFromJson(data io.Reader) *model.User { + return userFromGitLabUser(gitLabUserFromJson(data)) +} + +func (m *GitLabProvider) GetAuthDataFromJson(data io.Reader) string { + return gitLabUserFromJson(data).getAuthData() +} |