gitea源码

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. // Copyright 2021 The Gitea Authors. All rights reserved.
  2. // SPDX-License-Identifier: MIT
  3. package auth
  4. import (
  5. "context"
  6. "code.gitea.io/gitea/models/auth"
  7. "code.gitea.io/gitea/models/db"
  8. user_model "code.gitea.io/gitea/models/user"
  9. )
  10. // DeleteSource deletes a AuthSource record in DB.
  11. func DeleteSource(ctx context.Context, source *auth.Source) error {
  12. count, err := db.GetEngine(ctx).Count(&user_model.User{LoginSource: source.ID})
  13. if err != nil {
  14. return err
  15. } else if count > 0 {
  16. return auth.ErrSourceInUse{
  17. ID: source.ID,
  18. }
  19. }
  20. count, err = db.GetEngine(ctx).Count(&user_model.ExternalLoginUser{LoginSourceID: source.ID})
  21. if err != nil {
  22. return err
  23. } else if count > 0 {
  24. return auth.ErrSourceInUse{
  25. ID: source.ID,
  26. }
  27. }
  28. if registerableSource, ok := source.Cfg.(auth.RegisterableSource); ok {
  29. if err := registerableSource.UnregisterSource(); err != nil {
  30. return err
  31. }
  32. }
  33. _, err = db.GetEngine(ctx).ID(source.ID).Delete(new(auth.Source))
  34. return err
  35. }