Fork to maintain patches against the official gitea for https://code.ceondo.com https://github.com/go-gitea/gitea

external_login_user.go 2.2KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. // Copyright 2017 The Gitea Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package models
  5. import "github.com/markbates/goth"
  6. // ExternalLoginUser makes the connecting between some existing user and additional external login sources
  7. type ExternalLoginUser struct {
  8. ExternalID string `xorm:"pk NOT NULL"`
  9. UserID int64 `xorm:"INDEX NOT NULL"`
  10. LoginSourceID int64 `xorm:"pk NOT NULL"`
  11. }
  12. // GetExternalLogin checks if a externalID in loginSourceID scope already exists
  13. func GetExternalLogin(externalLoginUser *ExternalLoginUser) (bool, error) {
  14. return x.Get(externalLoginUser)
  15. }
  16. // ListAccountLinks returns a map with the ExternalLoginUser and its LoginSource
  17. func ListAccountLinks(user *User) ([]*ExternalLoginUser, error) {
  18. externalAccounts := make([]*ExternalLoginUser, 0, 5)
  19. err := x.Where("user_id=?", user.ID).
  20. Desc("login_source_id").
  21. Find(&externalAccounts)
  22. if err != nil {
  23. return nil, err
  24. }
  25. return externalAccounts, nil
  26. }
  27. // LinkAccountToUser link the gothUser to the user
  28. func LinkAccountToUser(user *User, gothUser goth.User) error {
  29. loginSource, err := GetActiveOAuth2LoginSourceByName(gothUser.Provider)
  30. if err != nil {
  31. return err
  32. }
  33. externalLoginUser := &ExternalLoginUser{
  34. ExternalID: gothUser.UserID,
  35. UserID: user.ID,
  36. LoginSourceID: loginSource.ID,
  37. }
  38. has, err := x.Get(externalLoginUser)
  39. if err != nil {
  40. return err
  41. } else if has {
  42. return ErrExternalLoginUserAlreadyExist{gothUser.UserID, user.ID, loginSource.ID}
  43. }
  44. _, err = x.Insert(externalLoginUser)
  45. return err
  46. }
  47. // RemoveAccountLink will remove all external login sources for the given user
  48. func RemoveAccountLink(user *User, loginSourceID int64) (int64, error) {
  49. deleted, err := x.Delete(&ExternalLoginUser{UserID: user.ID, LoginSourceID: loginSourceID})
  50. if err != nil {
  51. return deleted, err
  52. }
  53. if deleted < 1 {
  54. return deleted, ErrExternalLoginUserNotExist{user.ID, loginSourceID}
  55. }
  56. return deleted, err
  57. }
  58. // removeAllAccountLinks will remove all external login sources for the given user
  59. func removeAllAccountLinks(e Engine, user *User) error {
  60. _, err := e.Delete(&ExternalLoginUser{UserID: user.ID})
  61. return err
  62. }