2014-04-11 02:20:58 +08:00
// Copyright 2014 The Gogs Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package models
import (
2014-11-21 23:58:08 +08:00
"bytes"
2014-09-24 03:30:04 +08:00
"container/list"
2017-02-25 22:58:57 +08:00
"crypto/md5"
2014-04-11 02:20:58 +08:00
"crypto/sha256"
2016-12-03 13:49:17 +08:00
"crypto/subtle"
2014-04-11 02:20:58 +08:00
"encoding/hex"
"errors"
"fmt"
2014-11-21 23:58:08 +08:00
"image"
2016-11-29 00:47:46 +08:00
// Needed for jpeg support
2015-08-09 11:46:10 +08:00
_ "image/jpeg"
2015-11-14 05:43:43 +08:00
"image/png"
2014-04-11 02:20:58 +08:00
"os"
"path/filepath"
"strings"
"time"
2016-07-23 18:58:18 +08:00
"unicode/utf8"
2014-04-11 02:20:58 +08:00
2014-07-26 12:24:27 +08:00
"github.com/Unknwon/com"
2017-02-25 21:42:20 +08:00
"github.com/go-xorm/builder"
2015-09-02 00:19:52 +08:00
"github.com/go-xorm/xorm"
2014-11-21 23:58:08 +08:00
"github.com/nfnt/resize"
2016-12-15 09:24:27 +08:00
"golang.org/x/crypto/pbkdf2"
2014-04-11 02:20:58 +08:00
2016-11-11 00:24:48 +08:00
"code.gitea.io/git"
2016-11-11 17:39:44 +08:00
api "code.gitea.io/sdk/gitea"
2016-11-11 00:24:48 +08:00
"code.gitea.io/gitea/modules/avatar"
"code.gitea.io/gitea/modules/base"
2018-02-19 02:14:37 +08:00
"code.gitea.io/gitea/modules/generate"
2016-11-11 00:24:48 +08:00
"code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/setting"
2017-10-25 01:36:19 +08:00
"code.gitea.io/gitea/modules/util"
2014-04-11 02:20:58 +08:00
)
2016-11-29 00:47:46 +08:00
// UserType defines the user type
2014-06-25 12:44:48 +08:00
type UserType int
2014-04-11 02:20:58 +08:00
const (
2016-11-29 00:47:46 +08:00
// UserTypeIndividual defines an individual user
2016-11-08 00:53:22 +08:00
UserTypeIndividual UserType = iota // Historic reason to make it starts at 0.
2016-11-29 00:47:46 +08:00
// UserTypeOrganization defines an organization
2016-11-08 00:53:22 +08:00
UserTypeOrganization
2014-04-11 02:20:58 +08:00
)
2017-05-10 21:10:18 +08:00
const syncExternalUsers = "sync_external_users"
2014-04-11 02:20:58 +08:00
var (
2016-11-29 00:47:46 +08:00
// ErrUserNotKeyOwner user does not own this key error
ErrUserNotKeyOwner = errors . New ( "User does not own this public key" )
// ErrEmailNotExist e-mail does not exist error
ErrEmailNotExist = errors . New ( "E-mail does not exist" )
// ErrEmailNotActivated e-mail address has not been activated error
ErrEmailNotActivated = errors . New ( "E-mail address has not been activated" )
// ErrUserNameIllegal user name contains illegal characters error
ErrUserNameIllegal = errors . New ( "User name contains illegal characters" )
// ErrLoginSourceNotActived login source is not actived error
2014-05-11 14:12:45 +08:00
ErrLoginSourceNotActived = errors . New ( "Login source is not actived" )
2016-11-29 00:47:46 +08:00
// ErrUnsupportedLoginType login source is unknown error
ErrUnsupportedLoginType = errors . New ( "Login source is unknown" )
2014-04-11 02:20:58 +08:00
)
// User represents the object of individual and member of organization.
type User struct {
2016-07-24 01:08:22 +08:00
ID int64 ` xorm:"pk autoincr" `
2014-12-17 16:26:19 +08:00
LowerName string ` xorm:"UNIQUE NOT NULL" `
Name string ` xorm:"UNIQUE NOT NULL" `
FullName string
2015-12-21 20:24:11 +08:00
// Email is the primary email address (to be used for communication)
2017-01-08 11:12:03 +08:00
Email string ` xorm:"NOT NULL" `
KeepEmailPrivate bool
Passwd string ` xorm:"NOT NULL" `
LoginType LoginType
LoginSource int64 ` xorm:"NOT NULL DEFAULT 0" `
LoginName string
Type UserType
OwnedOrgs [ ] * User ` xorm:"-" `
Orgs [ ] * User ` xorm:"-" `
Repos [ ] * Repository ` xorm:"-" `
Location string
Website string
Rands string ` xorm:"VARCHAR(10)" `
Salt string ` xorm:"VARCHAR(10)" `
2018-05-05 08:28:30 +08:00
Language string ` xorm:"VARCHAR(5)" `
2016-03-10 08:53:30 +08:00
2017-12-11 12:37:04 +08:00
CreatedUnix util . TimeStamp ` xorm:"INDEX created" `
UpdatedUnix util . TimeStamp ` xorm:"INDEX updated" `
LastLoginUnix util . TimeStamp ` xorm:"INDEX" `
2014-11-21 23:58:08 +08:00
2015-10-25 16:26:26 +08:00
// Remember visibility choice for convenience, true for private
2015-08-28 16:44:04 +08:00
LastRepoVisibility bool
2016-11-27 19:59:12 +08:00
// Maximum repository creation limit, -1 means use global default
2015-12-11 01:48:45 +08:00
MaxRepoCreation int ` xorm:"NOT NULL DEFAULT -1" `
2015-08-28 16:44:04 +08:00
2015-12-21 20:24:11 +08:00
// Permissions
2017-01-06 23:14:33 +08:00
IsActive bool ` xorm:"INDEX" ` // Activate primary email
2016-12-31 10:33:30 +08:00
IsAdmin bool
AllowGitHook bool
AllowImportLocal bool // Allow migrate repository by local path
AllowCreateOrganization bool ` xorm:"DEFAULT true" `
2017-09-16 08:18:25 +08:00
ProhibitLogin bool ` xorm:"NOT NULL DEFAULT false" `
2014-11-21 23:58:08 +08:00
2015-12-21 20:24:11 +08:00
// Avatar
2014-11-21 23:58:08 +08:00
Avatar string ` xorm:"VARCHAR(2048) NOT NULL" `
AvatarEmail string ` xorm:"NOT NULL" `
UseCustomAvatar bool
2015-12-21 20:24:11 +08:00
// Counters
NumFollowers int
2015-12-22 08:09:28 +08:00
NumFollowing int ` xorm:"NOT NULL DEFAULT 0" `
2015-12-21 20:24:11 +08:00
NumStars int
NumRepos int
2014-06-25 12:44:48 +08:00
2015-12-21 20:24:11 +08:00
// For organization
2014-06-27 15:37:01 +08:00
Description string
NumTeams int
NumMembers int
2014-06-29 03:43:25 +08:00
Teams [ ] * Team ` xorm:"-" `
Members [ ] * User ` xorm:"-" `
2016-11-13 10:54:04 +08:00
// Preferences
DiffViewStyle string ` xorm:"NOT NULL DEFAULT ''" `
2014-04-11 02:20:58 +08:00
}
2016-11-29 00:47:46 +08:00
// BeforeUpdate is invoked from XORM before updating this object.
2015-12-11 01:37:53 +08:00
func ( u * User ) BeforeUpdate ( ) {
2015-12-11 01:46:05 +08:00
if u . MaxRepoCreation < - 1 {
u . MaxRepoCreation = - 1
2015-12-11 01:37:53 +08:00
}
2018-01-13 17:45:16 +08:00
// Organization does not need email
u . Email = strings . ToLower ( u . Email )
if ! u . IsOrganization ( ) {
if len ( u . AvatarEmail ) == 0 {
u . AvatarEmail = u . Email
}
2018-02-04 06:37:05 +08:00
if len ( u . AvatarEmail ) > 0 && u . Avatar == "" {
2018-01-13 17:45:16 +08:00
u . Avatar = base . HashEmail ( u . AvatarEmail )
}
}
u . LowerName = strings . ToLower ( u . Name )
u . Location = base . TruncateString ( u . Location , 255 )
u . Website = base . TruncateString ( u . Website , 255 )
u . Description = base . TruncateString ( u . Description , 255 )
2015-12-11 01:37:53 +08:00
}
2016-11-29 00:47:46 +08:00
// SetLastLogin set time to last login
2016-11-09 18:53:45 +08:00
func ( u * User ) SetLastLogin ( ) {
2017-12-11 12:37:04 +08:00
u . LastLoginUnix = util . TimeStampNow ( )
2016-11-09 18:53:45 +08:00
}
2016-11-29 00:47:46 +08:00
// UpdateDiffViewStyle updates the users diff view style
2016-11-13 10:54:04 +08:00
func ( u * User ) UpdateDiffViewStyle ( style string ) error {
u . DiffViewStyle = style
2017-08-12 22:18:44 +08:00
return UpdateUserCols ( u , "diff_view_style" )
2016-11-13 10:54:04 +08:00
}
2017-01-08 11:12:03 +08:00
// getEmail returns an noreply email, if the user has set to keep his
// email address private, otherwise the primary email address.
func ( u * User ) getEmail ( ) string {
if u . KeepEmailPrivate {
return fmt . Sprintf ( "%s@%s" , u . LowerName , setting . Service . NoReplyAddress )
}
return u . Email
}
2016-11-29 00:47:46 +08:00
// APIFormat converts a User to api.User
2016-08-14 18:32:24 +08:00
func ( u * User ) APIFormat ( ) * api . User {
return & api . User {
ID : u . ID ,
UserName : u . Name ,
FullName : u . FullName ,
2017-01-08 11:12:03 +08:00
Email : u . getEmail ( ) ,
2016-11-29 16:25:47 +08:00
AvatarURL : u . AvatarLink ( ) ,
2018-05-05 08:28:30 +08:00
Language : u . Language ,
2016-08-14 18:32:24 +08:00
}
}
2016-11-29 00:47:46 +08:00
// IsLocal returns true if user login type is LoginPlain.
2015-12-11 08:02:57 +08:00
func ( u * User ) IsLocal ( ) bool {
2016-11-08 00:30:04 +08:00
return u . LoginType <= LoginPlain
2015-12-11 08:02:57 +08:00
}
2017-02-22 15:14:37 +08:00
// IsOAuth2 returns true if user login type is LoginOAuth2.
func ( u * User ) IsOAuth2 ( ) bool {
return u . LoginType == LoginOAuth2
}
2015-11-16 23:14:12 +08:00
// HasForkedRepo checks if user has already forked a repository with given ID.
func ( u * User ) HasForkedRepo ( repoID int64 ) bool {
2016-07-24 01:08:22 +08:00
_ , has := HasForkedRepo ( u . ID , repoID )
2015-11-16 23:14:12 +08:00
return has
}
2017-05-24 08:27:08 +08:00
// MaxCreationLimit returns the number of repositories a user is allowed to create
func ( u * User ) MaxCreationLimit ( ) int {
2015-12-11 01:46:05 +08:00
if u . MaxRepoCreation <= - 1 {
2015-12-11 01:37:53 +08:00
return setting . Repository . MaxCreationLimit
}
return u . MaxRepoCreation
}
2016-11-29 00:47:46 +08:00
// CanCreateRepo returns if user login can create a repository
2015-12-11 01:37:53 +08:00
func ( u * User ) CanCreateRepo ( ) bool {
2017-05-20 11:51:19 +08:00
if u . IsAdmin {
return true
}
2015-12-11 01:46:05 +08:00
if u . MaxRepoCreation <= - 1 {
2015-12-12 04:11:13 +08:00
if setting . Repository . MaxCreationLimit <= - 1 {
2015-12-11 05:27:47 +08:00
return true
}
2015-12-11 01:37:53 +08:00
return u . NumRepos < setting . Repository . MaxCreationLimit
}
return u . NumRepos < u . MaxRepoCreation
}
2016-12-31 10:33:30 +08:00
// CanCreateOrganization returns true if user can create organisation.
func ( u * User ) CanCreateOrganization ( ) bool {
2017-02-14 20:16:00 +08:00
return u . IsAdmin || ( u . AllowCreateOrganization && ! setting . Admin . DisableRegularOrgCreation )
2016-12-31 10:33:30 +08:00
}
2015-11-04 07:40:52 +08:00
// CanEditGitHook returns true if user can edit Git hooks.
func ( u * User ) CanEditGitHook ( ) bool {
2017-09-12 17:25:42 +08:00
return ! setting . DisableGitHooks && ( u . IsAdmin || u . AllowGitHook )
2015-11-04 07:40:52 +08:00
}
// CanImportLocal returns true if user can migrate repository by local path.
func ( u * User ) CanImportLocal ( ) bool {
2017-01-23 09:19:50 +08:00
if ! setting . ImportLocalPaths {
return false
}
2015-11-04 07:40:52 +08:00
return u . IsAdmin || u . AllowImportLocal
}
2014-07-26 12:24:27 +08:00
// DashboardLink returns the user dashboard page link.
func ( u * User ) DashboardLink ( ) string {
if u . IsOrganization ( ) {
2016-11-27 18:14:25 +08:00
return setting . AppSubURL + "/org/" + u . Name + "/dashboard/"
2014-07-26 12:24:27 +08:00
}
2016-11-27 18:14:25 +08:00
return setting . AppSubURL + "/"
2014-07-26 12:24:27 +08:00
}
2015-08-26 21:45:51 +08:00
// HomeLink returns the user or organization home page link.
2014-06-25 12:44:48 +08:00
func ( u * User ) HomeLink ( ) string {
2016-11-27 18:14:25 +08:00
return setting . AppSubURL + "/" + u . Name
2014-04-11 02:20:58 +08:00
}
2015-09-17 13:54:12 +08:00
2017-02-11 20:57:33 +08:00
// HTMLURL returns the user or organization's full link.
func ( u * User ) HTMLURL ( ) string {
return setting . AppURL + u . Name
}
2015-09-17 13:54:12 +08:00
// GenerateEmailActivateCode generates an activate code based on user information and given e-mail.
func ( u * User ) GenerateEmailActivateCode ( email string ) string {
code := base . CreateTimeLimitCode (
2016-07-24 01:08:22 +08:00
com . ToStr ( u . ID ) + email + u . LowerName + u . Passwd + u . Rands ,
2015-09-17 13:54:12 +08:00
setting . Service . ActiveCodeLives , nil )
// Add tail hex username
code += hex . EncodeToString ( [ ] byte ( u . LowerName ) )
return code
}
// GenerateActivateCode generates an activate code based on user information.
func ( u * User ) GenerateActivateCode ( ) string {
return u . GenerateEmailActivateCode ( u . Email )
}
2014-04-11 02:20:58 +08:00
2015-09-06 22:08:14 +08:00
// CustomAvatarPath returns user custom avatar file path.
func ( u * User ) CustomAvatarPath ( ) string {
2017-02-25 22:58:57 +08:00
return filepath . Join ( setting . AvatarUploadPath , u . Avatar )
2015-09-06 22:08:14 +08:00
}
// GenerateRandomAvatar generates a random avatar for user.
func ( u * User ) GenerateRandomAvatar ( ) error {
2017-03-08 23:05:15 +08:00
return u . generateRandomAvatar ( x )
}
func ( u * User ) generateRandomAvatar ( e Engine ) error {
2015-09-06 22:08:14 +08:00
seed := u . Email
if len ( seed ) == 0 {
seed = u . Name
}
img , err := avatar . RandomImage ( [ ] byte ( seed ) )
if err != nil {
return fmt . Errorf ( "RandomImage: %v" , err )
}
2017-03-06 16:15:40 +08:00
// NOTICE for random avatar, it still uses id as avatar name, but custom avatar use md5
// since random image is not a user's photo, there is no security for enumable
2018-02-08 23:43:08 +08:00
if u . Avatar == "" {
u . Avatar = fmt . Sprintf ( "%d" , u . ID )
}
2016-02-02 23:22:27 +08:00
if err = os . MkdirAll ( filepath . Dir ( u . CustomAvatarPath ( ) ) , os . ModePerm ) ; err != nil {
2015-09-06 22:08:14 +08:00
return fmt . Errorf ( "MkdirAll: %v" , err )
}
fw , err := os . Create ( u . CustomAvatarPath ( ) )
if err != nil {
return fmt . Errorf ( "Create: %v" , err )
}
defer fw . Close ( )
2017-10-05 12:43:04 +08:00
if _ , err := e . ID ( u . ID ) . Cols ( "avatar" ) . Update ( u ) ; err != nil {
2017-03-08 23:05:15 +08:00
return err
}
2016-07-21 15:31:14 +08:00
if err = png . Encode ( fw , img ) ; err != nil {
2015-09-06 22:08:14 +08:00
return fmt . Errorf ( "Encode: %v" , err )
}
2016-07-24 01:08:22 +08:00
log . Info ( "New random avatar created: %d" , u . ID )
2015-09-06 22:08:14 +08:00
return nil
}
2017-12-03 19:55:13 +08:00
// SizedRelAvatarLink returns a relative link to the user's avatar. When
// applicable, the link is for an avatar of the indicated size (in pixels).
func ( u * User ) SizedRelAvatarLink ( size int ) string {
2016-07-24 01:08:22 +08:00
if u . ID == - 1 {
2017-06-30 00:10:33 +08:00
return base . DefaultAvatarLink ( )
2015-08-15 02:48:05 +08:00
}
2014-11-21 23:58:08 +08:00
switch {
case u . UseCustomAvatar :
2017-02-27 08:42:22 +08:00
if ! com . IsFile ( u . CustomAvatarPath ( ) ) {
2017-06-30 00:10:33 +08:00
return base . DefaultAvatarLink ( )
2015-08-09 11:46:10 +08:00
}
2017-02-25 22:58:57 +08:00
return setting . AppSubURL + "/avatars/" + u . Avatar
2015-03-13 08:32:38 +08:00
case setting . DisableGravatar , setting . OfflineMode :
2017-02-27 08:42:22 +08:00
if ! com . IsFile ( u . CustomAvatarPath ( ) ) {
2015-09-06 22:08:14 +08:00
if err := u . GenerateRandomAvatar ( ) ; err != nil {
log . Error ( 3 , "GenerateRandomAvatar: %v" , err )
2015-08-09 11:46:10 +08:00
}
}
2017-02-25 22:58:57 +08:00
return setting . AppSubURL + "/avatars/" + u . Avatar
2014-04-11 02:20:58 +08:00
}
2017-12-03 19:55:13 +08:00
return base . SizedAvatarLink ( u . AvatarEmail , size )
}
// RelAvatarLink returns a relative link to the user's avatar. The link
// may either be a sub-URL to this site, or a full URL to an external avatar
// service.
func ( u * User ) RelAvatarLink ( ) string {
return u . SizedRelAvatarLink ( base . DefaultAvatarSize )
2014-04-11 02:20:58 +08:00
}
2016-08-06 03:12:54 +08:00
// AvatarLink returns user avatar absolute link.
2015-08-28 23:36:13 +08:00
func ( u * User ) AvatarLink ( ) string {
link := u . RelAvatarLink ( )
2015-09-02 17:16:30 +08:00
if link [ 0 ] == '/' && link [ 1 ] != '/' {
2016-11-27 18:14:25 +08:00
return setting . AppURL + strings . TrimPrefix ( link , setting . AppSubURL ) [ 1 : ]
2015-08-28 23:36:13 +08:00
}
return link
}
2016-11-29 00:47:46 +08:00
// GetFollowers returns range of user's followers.
2015-12-21 20:24:11 +08:00
func ( u * User ) GetFollowers ( page int ) ( [ ] * User , error ) {
users := make ( [ ] * User , 0 , ItemsPerPage )
2016-11-10 23:16:32 +08:00
sess := x .
Limit ( ItemsPerPage , ( page - 1 ) * ItemsPerPage ) .
Where ( "follow.follow_id=?" , u . ID )
2015-12-21 20:24:11 +08:00
if setting . UsePostgreSQL {
2018-07-20 10:10:17 +08:00
sess = sess . Join ( "LEFT" , "follow" , "`user`.id=follow.user_id" )
2015-12-21 20:24:11 +08:00
} else {
2018-07-20 10:10:17 +08:00
sess = sess . Join ( "LEFT" , "follow" , "`user`.id=follow.user_id" )
2015-12-21 20:24:11 +08:00
}
return users , sess . Find ( & users )
}
2016-11-29 00:47:46 +08:00
// IsFollowing returns true if user is following followID.
2015-12-21 20:24:11 +08:00
func ( u * User ) IsFollowing ( followID int64 ) bool {
2016-07-24 01:08:22 +08:00
return IsFollowing ( u . ID , followID )
2015-12-21 20:24:11 +08:00
}
// GetFollowing returns range of user's following.
func ( u * User ) GetFollowing ( page int ) ( [ ] * User , error ) {
users := make ( [ ] * User , 0 , ItemsPerPage )
2016-11-10 23:16:32 +08:00
sess := x .
Limit ( ItemsPerPage , ( page - 1 ) * ItemsPerPage ) .
Where ( "follow.user_id=?" , u . ID )
2015-12-21 20:24:11 +08:00
if setting . UsePostgreSQL {
2018-07-20 10:10:17 +08:00
sess = sess . Join ( "LEFT" , "follow" , "`user`.id=follow.follow_id" )
2015-12-21 20:24:11 +08:00
} else {
2018-07-20 10:10:17 +08:00
sess = sess . Join ( "LEFT" , "follow" , "`user`.id=follow.follow_id" )
2015-12-21 20:24:11 +08:00
}
return users , sess . Find ( & users )
}
2014-04-11 02:20:58 +08:00
// NewGitSig generates and returns the signature of given user.
2014-06-25 12:44:48 +08:00
func ( u * User ) NewGitSig ( ) * git . Signature {
2014-04-11 02:20:58 +08:00
return & git . Signature {
2016-08-28 04:37:55 +08:00
Name : u . DisplayName ( ) ,
2017-01-08 11:12:03 +08:00
Email : u . getEmail ( ) ,
2014-04-11 02:20:58 +08:00
When : time . Now ( ) ,
}
}
2018-01-12 06:19:38 +08:00
func hashPassword ( passwd , salt string ) string {
tempPasswd := pbkdf2 . Key ( [ ] byte ( passwd ) , [ ] byte ( salt ) , 10000 , 50 , sha256 . New )
return fmt . Sprintf ( "%x" , tempPasswd )
}
2018-01-09 06:28:18 +08:00
// HashPassword hashes a password using PBKDF.
2018-01-12 06:19:38 +08:00
func ( u * User ) HashPassword ( passwd string ) {
u . Passwd = hashPassword ( passwd , u . Salt )
2014-06-25 12:44:48 +08:00
}
2015-04-17 02:40:39 +08:00
// ValidatePassword checks if given password matches the one belongs to the user.
2015-04-17 02:36:32 +08:00
func ( u * User ) ValidatePassword ( passwd string ) bool {
2018-01-12 06:19:38 +08:00
tempHash := hashPassword ( passwd , u . Salt )
return subtle . ConstantTimeCompare ( [ ] byte ( u . Passwd ) , [ ] byte ( tempHash ) ) == 1
2014-08-03 01:47:33 +08:00
}
2017-02-22 15:14:37 +08:00
// IsPasswordSet checks if the password is set or left empty
func ( u * User ) IsPasswordSet ( ) bool {
return ! u . ValidatePassword ( "" )
}
2014-11-21 23:58:08 +08:00
// UploadAvatar saves custom avatar for user.
2014-12-07 09:22:48 +08:00
// FIXME: split uploads to different subdirs in case we have massive users.
2014-11-21 23:58:08 +08:00
func ( u * User ) UploadAvatar ( data [ ] byte ) error {
2018-07-03 11:56:32 +08:00
imgCfg , _ , err := image . DecodeConfig ( bytes . NewReader ( data ) )
if err != nil {
return fmt . Errorf ( "DecodeConfig: %v" , err )
}
if imgCfg . Width > setting . AvatarMaxWidth {
return fmt . Errorf ( "Image width is to large: %d > %d" , imgCfg . Width , setting . AvatarMaxWidth )
}
if imgCfg . Height > setting . AvatarMaxHeight {
return fmt . Errorf ( "Image height is to large: %d > %d" , imgCfg . Height , setting . AvatarMaxHeight )
}
2014-11-21 23:58:08 +08:00
img , _ , err := image . Decode ( bytes . NewReader ( data ) )
if err != nil {
2015-11-14 05:43:43 +08:00
return fmt . Errorf ( "Decode: %v" , err )
2014-11-21 23:58:08 +08:00
}
2015-09-07 06:19:30 +08:00
2016-11-25 16:37:04 +08:00
m := resize . Resize ( avatar . AvatarSize , avatar . AvatarSize , img , resize . NearestNeighbor )
2014-11-21 23:58:08 +08:00
sess := x . NewSession ( )
2017-06-21 08:57:05 +08:00
defer sess . Close ( )
2014-11-21 23:58:08 +08:00
if err = sess . Begin ( ) ; err != nil {
return err
}
2015-11-14 05:43:43 +08:00
u . UseCustomAvatar = true
2017-02-25 22:58:57 +08:00
u . Avatar = fmt . Sprintf ( "%x" , md5 . Sum ( data ) )
2015-11-14 05:43:43 +08:00
if err = updateUser ( sess , u ) ; err != nil {
return fmt . Errorf ( "updateUser: %v" , err )
2014-11-21 23:58:08 +08:00
}
2016-12-01 07:56:15 +08:00
if err := os . MkdirAll ( setting . AvatarUploadPath , os . ModePerm ) ; err != nil {
2017-01-30 04:13:57 +08:00
return fmt . Errorf ( "Failed to create dir %s: %v" , setting . AvatarUploadPath , err )
2016-12-01 07:56:15 +08:00
}
2014-11-22 23:22:53 +08:00
fw , err := os . Create ( u . CustomAvatarPath ( ) )
2014-11-21 23:58:08 +08:00
if err != nil {
2015-11-14 05:43:43 +08:00
return fmt . Errorf ( "Create: %v" , err )
2014-11-21 23:58:08 +08:00
}
defer fw . Close ( )
2015-09-07 06:19:30 +08:00
2015-11-14 05:43:43 +08:00
if err = png . Encode ( fw , m ) ; err != nil {
return fmt . Errorf ( "Encode: %v" , err )
2014-11-21 23:58:08 +08:00
}
return sess . Commit ( )
}
2016-03-07 00:36:30 +08:00
// DeleteAvatar deletes the user's custom avatar.
func ( u * User ) DeleteAvatar ( ) error {
2016-07-24 01:08:22 +08:00
log . Trace ( "DeleteAvatar[%d]: %s" , u . ID , u . CustomAvatarPath ( ) )
2017-03-06 16:15:40 +08:00
if len ( u . Avatar ) > 0 {
if err := os . Remove ( u . CustomAvatarPath ( ) ) ; err != nil {
return fmt . Errorf ( "Failed to remove %s: %v" , u . CustomAvatarPath ( ) , err )
}
2016-12-01 07:56:15 +08:00
}
2016-03-07 00:36:30 +08:00
u . UseCustomAvatar = false
2017-03-06 16:15:40 +08:00
u . Avatar = ""
2017-10-05 12:43:04 +08:00
if _ , err := x . ID ( u . ID ) . Cols ( "avatar, use_custom_avatar" ) . Update ( u ) ; err != nil {
2016-03-07 02:24:42 +08:00
return fmt . Errorf ( "UpdateUser: %v" , err )
2016-03-07 00:36:30 +08:00
}
return nil
}
2015-08-14 02:43:40 +08:00
// IsAdminOfRepo returns true if user has admin or higher access of repository.
func ( u * User ) IsAdminOfRepo ( repo * Repository ) bool {
2017-03-15 08:51:46 +08:00
has , err := HasAccess ( u . ID , repo , AccessModeAdmin )
2016-03-06 09:45:23 +08:00
if err != nil {
log . Error ( 3 , "HasAccess: %v" , err )
2015-08-14 02:43:40 +08:00
}
2016-03-06 09:45:23 +08:00
return has
2015-08-14 02:43:40 +08:00
}
2016-03-06 09:45:23 +08:00
// IsWriterOfRepo returns true if user has write access to given repository.
func ( u * User ) IsWriterOfRepo ( repo * Repository ) bool {
2017-03-15 08:51:46 +08:00
has , err := HasAccess ( u . ID , repo , AccessModeWrite )
2016-03-05 04:43:01 +08:00
if err != nil {
log . Error ( 3 , "HasAccess: %v" , err )
}
return has
}
2014-06-28 12:40:07 +08:00
// IsOrganization returns true if user is actually a organization.
2014-06-25 12:44:48 +08:00
func ( u * User ) IsOrganization ( ) bool {
2016-11-08 00:53:22 +08:00
return u . Type == UserTypeOrganization
2014-06-25 12:44:48 +08:00
}
2014-08-15 18:29:41 +08:00
// IsUserOrgOwner returns true if user is in the owner team of given organization.
2016-11-29 00:47:46 +08:00
func ( u * User ) IsUserOrgOwner ( orgID int64 ) bool {
2017-12-21 15:43:26 +08:00
isOwner , err := IsOrganizationOwner ( orgID , u . ID )
if err != nil {
log . Error ( 4 , "IsOrganizationOwner: %v" , err )
return false
}
return isOwner
2014-08-15 18:29:41 +08:00
}
2017-01-20 13:16:10 +08:00
// IsPublicMember returns true if user public his/her membership in given organization.
2016-11-29 00:47:46 +08:00
func ( u * User ) IsPublicMember ( orgID int64 ) bool {
2017-12-21 15:43:26 +08:00
isMember , err := IsPublicMembership ( orgID , u . ID )
if err != nil {
log . Error ( 4 , "IsPublicMembership: %v" , err )
return false
}
return isMember
2014-08-15 18:29:41 +08:00
}
2015-09-06 20:54:08 +08:00
func ( u * User ) getOrganizationCount ( e Engine ) ( int64 , error ) {
2016-11-10 23:16:32 +08:00
return e .
Where ( "uid=?" , u . ID ) .
Count ( new ( OrgUser ) )
2015-09-06 20:54:08 +08:00
}
2014-06-29 03:43:25 +08:00
// GetOrganizationCount returns count of membership of organization of user.
func ( u * User ) GetOrganizationCount ( ) ( int64 , error ) {
2015-09-06 20:54:08 +08:00
return u . getOrganizationCount ( x )
2014-06-29 03:43:25 +08:00
}
2016-07-24 14:32:46 +08:00
// GetRepositories returns repositories that user owns, including private repositories.
func ( u * User ) GetRepositories ( page , pageSize int ) ( err error ) {
2017-02-04 20:20:20 +08:00
u . Repos , err = GetUserRepositories ( u . ID , true , page , pageSize , "" )
2014-08-24 21:09:05 +08:00
return err
}
2018-06-22 00:00:13 +08:00
// GetRepositoryIDs returns repositories IDs where user owned and has unittypes
func ( u * User ) GetRepositoryIDs ( units ... UnitType ) ( [ ] int64 , error ) {
2017-02-17 08:58:19 +08:00
var ids [ ] int64
2018-06-22 00:00:13 +08:00
sess := x . Table ( "repository" ) . Cols ( "repository.id" )
if len ( units ) > 0 {
sess = sess . Join ( "INNER" , "repo_unit" , "repository.id = repo_unit.repo_id" )
sess = sess . In ( "repo_unit.type" , units )
}
return ids , sess . Where ( "owner_id = ?" , u . ID ) . Find ( & ids )
2017-02-17 08:58:19 +08:00
}
2018-06-22 00:00:13 +08:00
// GetOrgRepositoryIDs returns repositories IDs where user's team owned and has unittypes
func ( u * User ) GetOrgRepositoryIDs ( units ... UnitType ) ( [ ] int64 , error ) {
2017-02-17 08:58:19 +08:00
var ids [ ] int64
2018-06-22 00:00:13 +08:00
sess := x . Table ( "repository" ) .
2017-02-17 08:58:19 +08:00
Cols ( "repository.id" ) .
2018-06-22 00:00:13 +08:00
Join ( "INNER" , "team_user" , "repository.owner_id = team_user.org_id" ) .
Join ( "INNER" , "team_repo" , "repository.is_private != ? OR (team_user.team_id = team_repo.team_id AND repository.id = team_repo.repo_id)" , true )
if len ( units ) > 0 {
sess = sess . Join ( "INNER" , "team_unit" , "team_unit.team_id = team_user.team_id" )
sess = sess . In ( "team_unit.type" , units )
}
return ids , sess .
Where ( "team_user.uid = ?" , u . ID ) .
2017-02-17 08:58:19 +08:00
GroupBy ( "repository.id" ) . Find ( & ids )
}
2017-03-15 08:52:01 +08:00
// GetAccessRepoIDs returns all repositories IDs where user's or user is a team member organizations
2018-06-22 00:00:13 +08:00
func ( u * User ) GetAccessRepoIDs ( units ... UnitType ) ( [ ] int64 , error ) {
ids , err := u . GetRepositoryIDs ( units ... )
2017-02-17 08:58:19 +08:00
if err != nil {
return nil , err
}
2018-06-22 00:00:13 +08:00
ids2 , err := u . GetOrgRepositoryIDs ( units ... )
2017-02-17 08:58:19 +08:00
if err != nil {
return nil , err
}
return append ( ids , ids2 ... ) , nil
}
2016-11-29 00:47:46 +08:00
// GetMirrorRepositories returns mirror repositories that user owns, including private repositories.
2016-07-24 14:32:46 +08:00
func ( u * User ) GetMirrorRepositories ( ) ( [ ] * Repository , error ) {
return GetUserMirrorRepositories ( u . ID )
}
2015-09-08 01:58:23 +08:00
// GetOwnedOrganizations returns all organizations that user owns.
func ( u * User ) GetOwnedOrganizations ( ) ( err error ) {
2016-07-24 01:08:22 +08:00
u . OwnedOrgs , err = GetOwnedOrgsByUserID ( u . ID )
2015-09-08 01:58:23 +08:00
return err
}
2014-06-28 12:40:07 +08:00
// GetOrganizations returns all organizations that user belongs to.
2015-12-17 15:28:47 +08:00
func ( u * User ) GetOrganizations ( all bool ) error {
2016-07-24 01:08:22 +08:00
ous , err := GetOrgUsersByUserID ( u . ID , all )
2014-06-25 12:44:48 +08:00
if err != nil {
return err
}
u . Orgs = make ( [ ] * User , len ( ous ) )
for i , ou := range ous {
2015-08-08 22:43:14 +08:00
u . Orgs [ i ] , err = GetUserByID ( ou . OrgID )
2014-06-25 12:44:48 +08:00
if err != nil {
return err
}
}
return nil
2014-04-11 02:20:58 +08:00
}
2015-08-27 13:26:38 +08:00
// DisplayName returns full name if it's not empty,
// returns username otherwise.
func ( u * User ) DisplayName ( ) string {
if len ( u . FullName ) > 0 {
return u . FullName
2014-09-17 21:11:51 +08:00
}
2015-08-27 13:26:38 +08:00
return u . Name
2014-09-17 21:11:51 +08:00
}
2016-11-29 00:47:46 +08:00
// ShortName ellipses username to length
2015-11-19 06:42:20 +08:00
func ( u * User ) ShortName ( length int ) string {
2016-01-11 20:41:43 +08:00
return base . EllipsisString ( u . Name , length )
2015-11-19 06:42:20 +08:00
}
2017-03-15 08:52:01 +08:00
// IsMailable checks if a user is eligible
2017-02-02 20:33:36 +08:00
// to receive emails.
func ( u * User ) IsMailable ( ) bool {
return u . IsActive
}
2017-10-03 14:29:26 +08:00
func isUserExist ( e Engine , uid int64 , name string ) ( bool , error ) {
2014-04-11 02:20:58 +08:00
if len ( name ) == 0 {
return false , nil
}
2017-10-03 14:29:26 +08:00
return e .
2016-11-10 23:16:32 +08:00
Where ( "id!=?" , uid ) .
Get ( & User { LowerName : strings . ToLower ( name ) } )
2014-04-11 02:20:58 +08:00
}
2017-10-03 14:29:26 +08:00
// IsUserExist checks if given user name exist,
// the user name should be noncased unique.
// If uid is presented, then check will rule out that one,
// it is used when update a user name in settings page.
func IsUserExist ( uid int64 , name string ) ( bool , error ) {
return isUserExist ( x , uid , name )
}
2017-03-15 08:52:01 +08:00
// GetUserSalt returns a random user salt token.
2016-12-20 20:32:02 +08:00
func GetUserSalt ( ) ( string , error ) {
2018-02-19 02:14:37 +08:00
return generate . GetRandomString ( 10 )
2014-04-11 02:20:58 +08:00
}
2016-08-14 18:32:24 +08:00
// NewGhostUser creates and returns a fake user for someone has deleted his/her account.
func NewGhostUser ( ) * User {
2015-08-15 02:48:05 +08:00
return & User {
2016-07-24 01:08:22 +08:00
ID : - 1 ,
2016-08-14 18:32:24 +08:00
Name : "Ghost" ,
LowerName : "ghost" ,
2015-08-15 02:48:05 +08:00
}
}
2016-07-23 18:58:18 +08:00
var (
2018-08-13 13:02:18 +08:00
reservedUsernames = [ ] string {
"admin" ,
"api" ,
"assets" ,
"avatars" ,
"commits" ,
"css" ,
"debug" ,
"error" ,
"explore" ,
"help" ,
"img" ,
"install" ,
"issues" ,
"js" ,
"less" ,
"new" ,
"org" ,
"plugins" ,
"pulls" ,
"raw" ,
"repo" ,
"stars" ,
"template" ,
"user" ,
"vendor" ,
"." ,
".." ,
}
2016-11-12 20:26:45 +08:00
reservedUserPatterns = [ ] string { "*.keys" }
2016-07-23 18:58:18 +08:00
)
// isUsableName checks if name is reserved or pattern of name is not allowed
2016-11-12 20:26:45 +08:00
// based on given reserved names and patterns.
2016-07-23 18:58:18 +08:00
// Names are exact match, patterns can be prefix or suffix match with placeholder '*'.
func isUsableName ( names , patterns [ ] string , name string ) error {
name = strings . TrimSpace ( strings . ToLower ( name ) )
if utf8 . RuneCountInString ( name ) == 0 {
return ErrNameEmpty
}
for i := range names {
if name == names [ i ] {
return ErrNameReserved { name }
}
}
for _ , pat := range patterns {
if pat [ 0 ] == '*' && strings . HasSuffix ( name , pat [ 1 : ] ) ||
( pat [ len ( pat ) - 1 ] == '*' && strings . HasPrefix ( name , pat [ : len ( pat ) - 1 ] ) ) {
return ErrNamePatternNotAllowed { pat }
}
}
return nil
}
2016-11-29 00:47:46 +08:00
// IsUsableUsername returns an error when a username is reserved
2016-07-23 18:58:18 +08:00
func IsUsableUsername ( name string ) error {
2016-11-12 20:26:45 +08:00
return isUsableName ( reservedUsernames , reservedUserPatterns , name )
2016-07-23 18:58:18 +08:00
}
2014-06-25 12:44:48 +08:00
// CreateUser creates record of a new user.
2015-03-27 05:11:47 +08:00
func CreateUser ( u * User ) ( err error ) {
2016-07-23 18:58:18 +08:00
if err = IsUsableUsername ( u . Name ) ; err != nil {
2015-03-27 05:11:47 +08:00
return err
2014-06-25 12:44:48 +08:00
}
2017-10-03 14:29:26 +08:00
sess := x . NewSession ( )
defer sess . Close ( )
if err = sess . Begin ( ) ; err != nil {
return err
}
isExist , err := isUserExist ( sess , 0 , u . Name )
2014-06-25 12:44:48 +08:00
if err != nil {
2014-07-26 12:24:27 +08:00
return err
2014-06-25 12:44:48 +08:00
} else if isExist {
2015-03-27 05:11:47 +08:00
return ErrUserAlreadyExist { u . Name }
2014-06-25 12:44:48 +08:00
}
2015-11-24 09:43:04 +08:00
u . Email = strings . ToLower ( u . Email )
2017-10-03 14:29:26 +08:00
isExist , err = sess .
2017-01-05 08:52:20 +08:00
Where ( "email=?" , u . Email ) .
Get ( new ( User ) )
if err != nil {
return err
2017-10-03 14:29:26 +08:00
} else if isExist {
2017-01-05 08:52:20 +08:00
return ErrEmailAlreadyUsed { u . Email }
}
2017-10-03 14:29:26 +08:00
isExist , err = isEmailUsed ( sess , u . Email )
2014-06-25 12:44:48 +08:00
if err != nil {
2014-07-26 12:24:27 +08:00
return err
2014-06-25 12:44:48 +08:00
} else if isExist {
2015-03-27 05:11:47 +08:00
return ErrEmailAlreadyUsed { u . Email }
2014-06-25 12:44:48 +08:00
}
2017-01-08 11:12:03 +08:00
u . KeepEmailPrivate = setting . Service . DefaultKeepEmailPrivate
2014-06-25 12:44:48 +08:00
u . LowerName = strings . ToLower ( u . Name )
u . AvatarEmail = u . Email
2016-02-15 12:14:55 +08:00
u . Avatar = base . HashEmail ( u . AvatarEmail )
2016-12-20 20:32:02 +08:00
if u . Rands , err = GetUserSalt ( ) ; err != nil {
return err
}
if u . Salt , err = GetUserSalt ( ) ; err != nil {
return err
}
2018-01-12 06:19:38 +08:00
u . HashPassword ( u . Passwd )
2017-05-09 03:51:53 +08:00
u . AllowCreateOrganization = setting . Service . DefaultAllowCreateOrganization
2015-12-12 04:11:13 +08:00
u . MaxRepoCreation = - 1
2014-06-25 12:44:48 +08:00
if _ , err = sess . Insert ( u ) ; err != nil {
2014-07-26 12:24:27 +08:00
return err
} else if err = os . MkdirAll ( UserPath ( u . Name ) , os . ModePerm ) ; err != nil {
return err
2014-06-25 12:44:48 +08:00
}
2014-04-23 00:55:27 +08:00
2015-08-19 04:58:45 +08:00
return sess . Commit ( )
2014-06-25 12:44:48 +08:00
}
2015-08-07 00:00:11 +08:00
func countUsers ( e Engine ) int64 {
2016-11-10 23:16:32 +08:00
count , _ := e .
Where ( "type=0" ) .
Count ( new ( User ) )
2015-08-07 00:00:11 +08:00
return count
}
2014-07-07 16:15:08 +08:00
// CountUsers returns number of users.
func CountUsers ( ) int64 {
2015-08-07 00:00:11 +08:00
return countUsers ( x )
2014-07-07 16:15:08 +08:00
}
2017-01-05 08:50:34 +08:00
// get user by verify code
2014-04-11 02:20:58 +08:00
func getVerifyUser ( code string ) ( user * User ) {
if len ( code ) <= base . TimeLimitCodeLength {
return nil
}
// use tail hex username query user
hexStr := code [ base . TimeLimitCodeLength : ]
if b , err := hex . DecodeString ( hexStr ) ; err == nil {
if user , err = GetUserByName ( string ( b ) ) ; user != nil {
return user
}
2014-07-26 12:24:27 +08:00
log . Error ( 4 , "user.getVerifyUser: %v" , err )
2014-04-11 02:20:58 +08:00
}
return nil
}
2016-11-29 00:47:46 +08:00
// VerifyUserActiveCode verifies active code when active account
2014-04-11 02:20:58 +08:00
func VerifyUserActiveCode ( code string ) ( user * User ) {
2014-05-26 08:11:25 +08:00
minutes := setting . Service . ActiveCodeLives
2014-04-11 02:20:58 +08:00
if user = getVerifyUser ( code ) ; user != nil {
// time limit code
prefix := code [ : base . TimeLimitCodeLength ]
2016-07-24 01:08:22 +08:00
data := com . ToStr ( user . ID ) + user . Email + user . LowerName + user . Passwd + user . Rands
2014-04-11 02:20:58 +08:00
if base . VerifyTimeLimitCode ( data , minutes , prefix ) {
return user
}
}
return nil
}
2016-11-29 00:47:46 +08:00
// VerifyActiveEmailCode verifies active email code when active account
2014-12-17 23:40:10 +08:00
func VerifyActiveEmailCode ( code , email string ) * EmailAddress {
minutes := setting . Service . ActiveCodeLives
if user := getVerifyUser ( code ) ; user != nil {
// time limit code
prefix := code [ : base . TimeLimitCodeLength ]
2016-07-24 01:08:22 +08:00
data := com . ToStr ( user . ID ) + email + user . LowerName + user . Passwd + user . Rands
2014-12-17 23:40:10 +08:00
if base . VerifyTimeLimitCode ( data , minutes , prefix ) {
emailAddress := & EmailAddress { Email : email }
if has , _ := x . Get ( emailAddress ) ; has {
return emailAddress
}
}
}
return nil
}
2014-04-11 02:20:58 +08:00
// ChangeUserName changes all corresponding setting from old user name to new one.
2014-07-26 12:24:27 +08:00
func ChangeUserName ( u * User , newUserName string ) ( err error ) {
2016-07-23 18:58:18 +08:00
if err = IsUsableUsername ( newUserName ) ; err != nil {
2015-03-27 05:11:47 +08:00
return err
}
isExist , err := IsUserExist ( 0 , newUserName )
if err != nil {
return err
} else if isExist {
return ErrUserAlreadyExist { newUserName }
2014-07-26 12:24:27 +08:00
}
2016-02-06 03:11:53 +08:00
if err = ChangeUsernameInPullRequests ( u . Name , newUserName ) ; err != nil {
2016-01-28 19:07:16 +08:00
return fmt . Errorf ( "ChangeUsernameInPullRequests: %v" , err )
2016-01-28 05:45:03 +08:00
}
2016-02-06 03:11:53 +08:00
// Delete all local copies of repository wiki that user owns.
2017-10-21 14:26:23 +08:00
if err = x . BufferSize ( setting . IterateBufferSize ) .
2016-11-10 23:16:32 +08:00
Where ( "owner_id=?" , u . ID ) .
Iterate ( new ( Repository ) , func ( idx int , bean interface { } ) error {
repo := bean . ( * Repository )
RemoveAllWithNotice ( "Delete repository wiki local copy" , repo . LocalWikiPath ( ) )
return nil
} ) ; err != nil {
2016-02-06 03:11:53 +08:00
return fmt . Errorf ( "Delete repository wiki local copy: %v" , err )
}
2016-01-28 19:07:16 +08:00
return os . Rename ( UserPath ( u . Name ) , UserPath ( newUserName ) )
2014-04-11 02:20:58 +08:00
}
2017-02-25 22:53:57 +08:00
// checkDupEmail checks whether there are the same email with the user
func checkDupEmail ( e Engine , u * User ) error {
u . Email = strings . ToLower ( u . Email )
has , err := e .
Where ( "id!=?" , u . ID ) .
And ( "type=?" , u . Type ) .
And ( "email=?" , u . Email ) .
Get ( new ( User ) )
if err != nil {
return err
} else if has {
return ErrEmailAlreadyUsed { u . Email }
}
return nil
}
2015-08-30 01:13:24 +08:00
func updateUser ( e Engine , u * User ) error {
2017-10-05 12:43:04 +08:00
_ , err := e . ID ( u . ID ) . AllCols ( ) . Update ( u )
2014-04-11 02:20:58 +08:00
return err
}
2015-08-30 01:13:24 +08:00
// UpdateUser updates user's information.
func UpdateUser ( u * User ) error {
return updateUser ( x , u )
2017-02-25 22:53:57 +08:00
}
2017-08-12 22:18:44 +08:00
// UpdateUserCols update user according special columns
func UpdateUserCols ( u * User , cols ... string ) error {
2017-09-25 12:59:27 +08:00
return updateUserCols ( x , u , cols ... )
}
func updateUserCols ( e Engine , u * User , cols ... string ) error {
2017-10-05 12:43:04 +08:00
_ , err := e . ID ( u . ID ) . Cols ( cols ... ) . Update ( u )
2017-08-12 22:18:44 +08:00
return err
}
2017-02-25 22:53:57 +08:00
// UpdateUserSetting updates user's settings.
func UpdateUserSetting ( u * User ) error {
if ! u . IsOrganization ( ) {
if err := checkDupEmail ( x , u ) ; err != nil {
return err
}
}
return updateUser ( x , u )
2015-08-30 01:13:24 +08:00
}
2015-09-06 20:54:08 +08:00
// deleteBeans deletes all given beans, beans should contain delete conditions.
func deleteBeans ( e Engine , beans ... interface { } ) ( err error ) {
2015-03-18 09:51:39 +08:00
for i := range beans {
if _ , err = e . Delete ( beans [ i ] ) ; err != nil {
return err
}
}
return nil
}
2014-11-13 18:27:01 +08:00
// FIXME: need some kind of mechanism to record failure. HINT: system notice
2015-09-06 20:54:08 +08:00
func deleteUser ( e * xorm . Session , u * User ) error {
2015-08-17 17:05:37 +08:00
// Note: A user owns any repository or belongs to any organization
// cannot perform delete operation.
2014-04-11 02:20:58 +08:00
// Check ownership of repository.
2015-09-06 20:54:08 +08:00
count , err := getRepositoryCount ( e , u )
2014-04-11 02:20:58 +08:00
if err != nil {
2015-03-18 09:51:39 +08:00
return fmt . Errorf ( "GetRepositoryCount: %v" , err )
2014-04-11 02:20:58 +08:00
} else if count > 0 {
2016-07-24 01:08:22 +08:00
return ErrUserOwnRepos { UID : u . ID }
2014-04-11 02:20:58 +08:00
}
2014-06-27 15:37:01 +08:00
// Check membership of organization.
2015-09-06 20:54:08 +08:00
count , err = u . getOrganizationCount ( e )
2014-06-27 15:37:01 +08:00
if err != nil {
2015-03-18 09:51:39 +08:00
return fmt . Errorf ( "GetOrganizationCount: %v" , err )
2014-06-27 15:37:01 +08:00
} else if count > 0 {
2016-07-24 01:08:22 +08:00
return ErrUserHasOrgs { UID : u . ID }
2014-06-27 15:37:01 +08:00
}
2015-08-17 17:05:37 +08:00
// ***** START: Watch *****
2017-05-20 16:48:22 +08:00
watchedRepoIDs := make ( [ ] int64 , 0 , 10 )
if err = e . Table ( "watch" ) . Cols ( "watch.repo_id" ) .
Where ( "watch.user_id = ?" , u . ID ) . Find ( & watchedRepoIDs ) ; err != nil {
2015-03-18 09:51:39 +08:00
return fmt . Errorf ( "get all watches: %v" , err )
2014-04-11 02:20:58 +08:00
}
2018-07-05 05:47:05 +08:00
if _ , err = e . Decr ( "num_watches" ) . In ( "id" , watchedRepoIDs ) . NoAutoTime ( ) . Update ( new ( Repository ) ) ; err != nil {
2017-05-20 16:48:22 +08:00
return fmt . Errorf ( "decrease repository num_watches: %v" , err )
2014-04-12 09:47:39 +08:00
}
2015-08-17 17:05:37 +08:00
// ***** END: Watch *****
2015-03-18 09:51:39 +08:00
2015-08-17 17:05:37 +08:00
// ***** START: Star *****
2017-05-20 16:48:22 +08:00
starredRepoIDs := make ( [ ] int64 , 0 , 10 )
if err = e . Table ( "star" ) . Cols ( "star.repo_id" ) .
Where ( "star.uid = ?" , u . ID ) . Find ( & starredRepoIDs ) ; err != nil {
2015-08-17 17:05:37 +08:00
return fmt . Errorf ( "get all stars: %v" , err )
2018-07-05 05:47:05 +08:00
} else if _ , err = e . Decr ( "num_stars" ) . In ( "id" , starredRepoIDs ) . NoAutoTime ( ) . Update ( new ( Repository ) ) ; err != nil {
2017-05-20 16:48:22 +08:00
return fmt . Errorf ( "decrease repository num_stars: %v" , err )
2015-08-17 17:05:37 +08:00
}
// ***** END: Star *****
2015-03-18 09:51:39 +08:00
2015-08-17 17:05:37 +08:00
// ***** START: Follow *****
2017-05-20 16:48:22 +08:00
followeeIDs := make ( [ ] int64 , 0 , 10 )
if err = e . Table ( "follow" ) . Cols ( "follow.follow_id" ) .
Where ( "follow.user_id = ?" , u . ID ) . Find ( & followeeIDs ) ; err != nil {
return fmt . Errorf ( "get all followees: %v" , err )
} else if _ , err = e . Decr ( "num_followers" ) . In ( "id" , followeeIDs ) . Update ( new ( User ) ) ; err != nil {
return fmt . Errorf ( "decrease user num_followers: %v" , err )
2015-08-17 17:05:37 +08:00
}
2017-05-20 16:48:22 +08:00
followerIDs := make ( [ ] int64 , 0 , 10 )
if err = e . Table ( "follow" ) . Cols ( "follow.user_id" ) .
Where ( "follow.follow_id = ?" , u . ID ) . Find ( & followerIDs ) ; err != nil {
return fmt . Errorf ( "get all followers: %v" , err )
} else if _ , err = e . Decr ( "num_following" ) . In ( "id" , followerIDs ) . Update ( new ( User ) ) ; err != nil {
return fmt . Errorf ( "decrease user num_following: %v" , err )
2014-04-11 02:20:58 +08:00
}
2015-08-17 17:05:37 +08:00
// ***** END: Follow *****
2015-03-18 09:51:39 +08:00
2015-09-06 20:54:08 +08:00
if err = deleteBeans ( e ,
2016-07-24 01:08:22 +08:00
& AccessToken { UID : u . ID } ,
& Collaboration { UserID : u . ID } ,
& Access { UserID : u . ID } ,
& Watch { UserID : u . ID } ,
& Star { UID : u . ID } ,
2017-05-20 16:48:22 +08:00
& Follow { UserID : u . ID } ,
2016-07-24 01:08:22 +08:00
& Follow { FollowID : u . ID } ,
& Action { UserID : u . ID } ,
& IssueUser { UID : u . ID } ,
& EmailAddress { UID : u . ID } ,
2017-03-17 22:16:08 +08:00
& UserOpenID { UID : u . ID } ,
2017-12-04 07:14:26 +08:00
& Reaction { UserID : u . ID } ,
2015-03-18 09:51:39 +08:00
) ; err != nil {
2015-12-01 09:45:55 +08:00
return fmt . Errorf ( "deleteBeans: %v" , err )
2014-04-11 02:20:58 +08:00
}
2015-03-18 09:51:39 +08:00
2015-08-17 17:05:37 +08:00
// ***** START: PublicKey *****
2014-05-07 04:28:52 +08:00
keys := make ( [ ] * PublicKey , 0 , 10 )
2016-07-24 01:08:22 +08:00
if err = e . Find ( & keys , & PublicKey { OwnerID : u . ID } ) ; err != nil {
2015-08-17 17:05:37 +08:00
return fmt . Errorf ( "get all public keys: %v" , err )
2014-04-11 02:20:58 +08:00
}
2016-07-26 17:26:48 +08:00
keyIDs := make ( [ ] int64 , len ( keys ) )
for i := range keys {
keyIDs [ i ] = keys [ i ] . ID
}
if err = deletePublicKeys ( e , keyIDs ... ) ; err != nil {
return fmt . Errorf ( "deletePublicKeys: %v" , err )
2014-04-11 02:20:58 +08:00
}
2015-08-17 17:05:37 +08:00
// ***** END: PublicKey *****
2014-04-11 02:20:58 +08:00
2015-08-15 02:48:05 +08:00
// Clear assignee.
2018-05-10 00:29:04 +08:00
if err = clearAssigneeByUserID ( e , u . ID ) ; err != nil {
2015-08-17 17:05:37 +08:00
return fmt . Errorf ( "clear assignee: %v" , err )
2015-08-15 02:48:05 +08:00
}
2017-02-22 15:14:37 +08:00
// ***** START: ExternalLoginUser *****
2017-03-20 22:13:52 +08:00
if err = removeAllAccountLinks ( e , u ) ; err != nil {
2017-02-22 15:14:37 +08:00
return fmt . Errorf ( "ExternalLoginUser: %v" , err )
}
// ***** END: ExternalLoginUser *****
2017-10-05 12:43:04 +08:00
if _ , err = e . ID ( u . ID ) . Delete ( new ( User ) ) ; err != nil {
2015-08-17 17:05:37 +08:00
return fmt . Errorf ( "Delete: %v" , err )
2015-03-18 09:51:39 +08:00
}
2015-08-17 17:05:37 +08:00
// FIXME: system notice
// Note: There are something just cannot be roll back,
// so just keep error logs of those operations.
2016-12-01 07:56:15 +08:00
path := UserPath ( u . Name )
2015-08-17 17:05:37 +08:00
2016-12-01 07:56:15 +08:00
if err := os . RemoveAll ( path ) ; err != nil {
2017-01-30 04:13:57 +08:00
return fmt . Errorf ( "Failed to RemoveAll %s: %v" , path , err )
2016-12-01 07:56:15 +08:00
}
2017-03-06 16:15:40 +08:00
if len ( u . Avatar ) > 0 {
avatarPath := u . CustomAvatarPath ( )
if com . IsExist ( avatarPath ) {
if err := os . Remove ( avatarPath ) ; err != nil {
return fmt . Errorf ( "Failed to remove %s: %v" , avatarPath , err )
}
2016-12-27 10:02:14 +08:00
}
2016-12-01 07:56:15 +08:00
}
2014-04-11 02:20:58 +08:00
2015-08-17 17:05:37 +08:00
return nil
2014-06-21 12:51:41 +08:00
}
2015-09-06 20:54:08 +08:00
// DeleteUser completely and permanently deletes everything of a user,
// but issues/comments/pulls will be kept and shown as someone has been deleted.
func DeleteUser ( u * User ) ( err error ) {
sess := x . NewSession ( )
2017-06-21 08:57:05 +08:00
defer sess . Close ( )
2015-09-06 20:54:08 +08:00
if err = sess . Begin ( ) ; err != nil {
return err
}
if err = deleteUser ( sess , u ) ; err != nil {
2015-09-14 01:26:20 +08:00
// Note: don't wrapper error here.
return err
2015-09-06 20:54:08 +08:00
}
2016-07-26 17:26:48 +08:00
if err = sess . Commit ( ) ; err != nil {
return err
}
return RewriteAllPublicKeys ( )
2015-09-06 20:54:08 +08:00
}
2014-12-17 16:26:19 +08:00
// DeleteInactivateUsers deletes all inactivate users and email addresses.
2015-08-17 17:05:37 +08:00
func DeleteInactivateUsers ( ) ( err error ) {
users := make ( [ ] * User , 0 , 10 )
2016-11-10 23:16:32 +08:00
if err = x .
Where ( "is_active = ?" , false ) .
Find ( & users ) ; err != nil {
2015-08-17 17:05:37 +08:00
return fmt . Errorf ( "get all inactive users: %v" , err )
}
2016-07-26 17:26:48 +08:00
// FIXME: should only update authorized_keys file once after all deletions.
2015-08-17 17:05:37 +08:00
for _ , u := range users {
if err = DeleteUser ( u ) ; err != nil {
// Ignore users that were set inactive by admin.
if IsErrUserOwnRepos ( err ) || IsErrUserHasOrgs ( err ) {
continue
}
return err
}
2014-12-17 16:26:19 +08:00
}
2015-08-17 17:05:37 +08:00
2016-11-10 23:16:32 +08:00
_ , err = x .
Where ( "is_activated = ?" , false ) .
Delete ( new ( EmailAddress ) )
2014-04-11 02:20:58 +08:00
return err
}
// UserPath returns the path absolute path of user repositories.
func UserPath ( userName string ) string {
2014-05-26 08:11:25 +08:00
return filepath . Join ( setting . RepoRootPath , strings . ToLower ( userName ) )
2014-04-11 02:20:58 +08:00
}
2016-11-12 07:01:09 +08:00
// GetUserByKeyID get user information by user's public key id
2015-11-05 10:57:10 +08:00
func GetUserByKeyID ( keyID int64 ) ( * User , error ) {
2016-11-12 07:01:09 +08:00
var user User
has , err := x . Join ( "INNER" , "public_key" , "`public_key`.owner_id = `user`.id" ) .
2016-11-10 23:16:32 +08:00
Where ( "`public_key`.id=?" , keyID ) .
2016-11-12 16:30:46 +08:00
Get ( & user )
2016-11-12 07:01:09 +08:00
if err != nil {
return nil , err
}
if ! has {
return nil , ErrUserNotExist { 0 , "" , keyID }
}
return & user , nil
2014-04-11 02:20:58 +08:00
}
2015-08-08 22:43:14 +08:00
func getUserByID ( e Engine , id int64 ) ( * User , error ) {
2014-06-06 10:07:35 +08:00
u := new ( User )
2017-10-05 12:43:04 +08:00
has , err := e . ID ( id ) . Get ( u )
2014-04-11 02:20:58 +08:00
if err != nil {
return nil , err
2014-06-06 10:07:35 +08:00
} else if ! has {
2016-11-12 07:01:09 +08:00
return nil , ErrUserNotExist { id , "" , 0 }
2014-04-11 02:20:58 +08:00
}
2014-06-06 10:07:35 +08:00
return u , nil
2014-04-11 02:20:58 +08:00
}
2015-08-08 22:43:14 +08:00
// GetUserByID returns the user object by given ID if exists.
func GetUserByID ( id int64 ) ( * User , error ) {
return getUserByID ( x , id )
2015-02-13 13:58:46 +08:00
}
2018-05-10 00:29:04 +08:00
// GetUserIfHasWriteAccess returns the user with write access of repository by given ID.
func GetUserIfHasWriteAccess ( repo * Repository , userID int64 ) ( * User , error ) {
2017-03-15 08:51:46 +08:00
has , err := HasAccess ( userID , repo , AccessModeWrite )
2015-08-10 21:47:23 +08:00
if err != nil {
return nil , err
} else if ! has {
2016-11-12 07:01:09 +08:00
return nil , ErrUserNotExist { userID , "" , 0 }
2015-08-10 21:47:23 +08:00
}
return GetUserByID ( userID )
}
2014-10-25 06:43:17 +08:00
// GetUserByName returns user by given name.
2014-04-11 02:20:58 +08:00
func GetUserByName ( name string ) ( * User , error ) {
2017-08-30 12:31:33 +08:00
return getUserByName ( x , name )
}
func getUserByName ( e Engine , name string ) ( * User , error ) {
2014-04-11 02:20:58 +08:00
if len ( name ) == 0 {
2016-11-12 07:01:09 +08:00
return nil , ErrUserNotExist { 0 , name , 0 }
2014-04-11 02:20:58 +08:00
}
2014-10-25 06:43:17 +08:00
u := & User { LowerName : strings . ToLower ( name ) }
2017-08-30 12:31:33 +08:00
has , err := e . Get ( u )
2014-04-11 02:20:58 +08:00
if err != nil {
return nil , err
} else if ! has {
2016-11-12 07:01:09 +08:00
return nil , ErrUserNotExist { 0 , name , 0 }
2014-04-11 02:20:58 +08:00
}
2014-10-25 06:43:17 +08:00
return u , nil
2014-04-11 02:20:58 +08:00
}
2014-10-12 06:02:48 +08:00
// GetUserEmailsByNames returns a list of e-mails corresponds to names.
2014-04-11 02:20:58 +08:00
func GetUserEmailsByNames ( names [ ] string ) [ ] string {
2017-08-30 12:31:33 +08:00
return getUserEmailsByNames ( x , names )
}
func getUserEmailsByNames ( e Engine , names [ ] string ) [ ] string {
2014-04-11 02:20:58 +08:00
mails := make ( [ ] string , 0 , len ( names ) )
for _ , name := range names {
2017-08-30 12:31:33 +08:00
u , err := getUserByName ( e , name )
2014-04-11 02:20:58 +08:00
if err != nil {
continue
}
2017-02-02 20:33:36 +08:00
if u . IsMailable ( ) {
mails = append ( mails , u . Email )
}
2014-04-11 02:20:58 +08:00
}
return mails
}
2016-11-10 16:10:35 +08:00
// GetUsersByIDs returns all resolved users from a list of Ids.
func GetUsersByIDs ( ids [ ] int64 ) ( [ ] * User , error ) {
ous := make ( [ ] * User , 0 , len ( ids ) )
2017-01-20 10:31:46 +08:00
if len ( ids ) == 0 {
return ous , nil
}
err := x . In ( "id" , ids ) .
2016-11-10 23:16:32 +08:00
Asc ( "name" ) .
Find ( & ous )
2016-11-10 16:10:35 +08:00
return ous , err
}
2016-07-16 10:08:04 +08:00
// GetUserIDsByNames returns a slice of ids corresponds to names.
func GetUserIDsByNames ( names [ ] string ) [ ] int64 {
2014-05-08 04:51:14 +08:00
ids := make ( [ ] int64 , 0 , len ( names ) )
for _ , name := range names {
u , err := GetUserByName ( name )
if err != nil {
continue
}
2016-07-24 01:08:22 +08:00
ids = append ( ids , u . ID )
2014-05-08 04:51:14 +08:00
}
return ids
}
2014-12-07 09:22:48 +08:00
// UserCommit represents a commit with validation of user.
2014-09-24 03:30:04 +08:00
type UserCommit struct {
2014-11-21 23:58:08 +08:00
User * User
2015-12-10 09:46:05 +08:00
* git . Commit
2014-09-24 03:30:04 +08:00
}
2017-01-05 08:50:34 +08:00
// ValidateCommitWithEmail check if author's e-mail of commit is corresponding to a user.
2015-12-10 09:46:05 +08:00
func ValidateCommitWithEmail ( c * git . Commit ) * User {
2017-10-26 15:45:14 +08:00
if c . Author == nil {
return nil
}
2014-09-26 20:55:13 +08:00
u , err := GetUserByEmail ( c . Author . Email )
2014-11-21 23:58:08 +08:00
if err != nil {
return nil
2014-09-26 20:55:13 +08:00
}
2014-11-21 23:58:08 +08:00
return u
2014-09-26 20:55:13 +08:00
}
// ValidateCommitsWithEmails checks if authors' e-mails of commits are corresponding to users.
func ValidateCommitsWithEmails ( oldCommits * list . List ) * list . List {
2015-08-05 17:36:22 +08:00
var (
u * User
emails = map [ string ] * User { }
newCommits = list . New ( )
e = oldCommits . Front ( )
)
2014-09-24 03:30:04 +08:00
for e != nil {
2015-12-10 09:46:05 +08:00
c := e . Value . ( * git . Commit )
2014-09-24 03:30:04 +08:00
2017-10-26 15:45:14 +08:00
if c . Author != nil {
if v , ok := emails [ c . Author . Email ] ; ! ok {
u , _ = GetUserByEmail ( c . Author . Email )
emails [ c . Author . Email ] = u
} else {
u = v
}
2014-09-24 11:18:14 +08:00
} else {
2017-10-26 15:45:14 +08:00
u = nil
2014-09-24 03:30:04 +08:00
}
newCommits . PushBack ( UserCommit {
2014-11-21 23:58:08 +08:00
User : u ,
Commit : c ,
2014-09-24 03:30:04 +08:00
} )
e = e . Next ( )
}
return newCommits
}
2014-04-11 02:20:58 +08:00
// GetUserByEmail returns the user object by given e-mail if exists.
func GetUserByEmail ( email string ) ( * User , error ) {
if len ( email ) == 0 {
2016-11-12 07:01:09 +08:00
return nil , ErrUserNotExist { 0 , email , 0 }
2014-04-11 02:20:58 +08:00
}
2015-08-05 17:36:22 +08:00
email = strings . ToLower ( email )
2014-12-17 16:26:19 +08:00
// First try to find the user by primary email
2015-08-05 17:36:22 +08:00
user := & User { Email : email }
2014-06-21 12:51:41 +08:00
has , err := x . Get ( user )
2014-04-11 02:20:58 +08:00
if err != nil {
return nil , err
}
2014-12-17 16:26:19 +08:00
if has {
return user , nil
}
// Otherwise, check in alternative list for activated email addresses
2015-08-05 17:36:22 +08:00
emailAddress := & EmailAddress { Email : email , IsActivated : true }
2014-12-17 16:26:19 +08:00
has , err = x . Get ( emailAddress )
if err != nil {
return nil , err
}
if has {
2015-09-10 23:40:34 +08:00
return GetUserByID ( emailAddress . UID )
2014-12-17 16:26:19 +08:00
}
2016-11-12 07:01:09 +08:00
return nil , ErrUserNotExist { 0 , email , 0 }
2014-04-11 02:20:58 +08:00
}
2017-02-22 15:14:37 +08:00
// GetUser checks if a user already exists
func GetUser ( user * User ) ( bool , error ) {
return x . Get ( user )
}
2016-11-29 00:47:46 +08:00
// SearchUserOptions contains the options for searching
2016-03-12 04:33:12 +08:00
type SearchUserOptions struct {
2017-11-26 16:40:38 +08:00
Keyword string
Type UserType
2018-05-24 09:03:42 +08:00
OrderBy SearchOrderBy
2017-11-26 16:40:38 +08:00
Page int
PageSize int // Can be smaller than or equal to setting.UI.ExplorePagingNum
IsActive util . OptionalBool
SearchByEmail bool // Search by email as well as username/full name
2016-03-12 04:33:12 +08:00
}
2017-10-25 01:36:19 +08:00
func ( opts * SearchUserOptions ) toConds ( ) builder . Cond {
var cond builder . Cond = builder . Eq { "type" : opts . Type }
if len ( opts . Keyword ) > 0 {
lowerKeyword := strings . ToLower ( opts . Keyword )
2017-11-26 16:40:38 +08:00
keywordCond := builder . Or (
2017-10-25 01:36:19 +08:00
builder . Like { "lower_name" , lowerKeyword } ,
builder . Like { "LOWER(full_name)" , lowerKeyword } ,
2017-11-26 16:40:38 +08:00
)
if opts . SearchByEmail {
keywordCond = keywordCond . Or ( builder . Like { "LOWER(email)" , lowerKeyword } )
}
cond = cond . And ( keywordCond )
2016-03-12 04:33:12 +08:00
}
2017-10-25 01:36:19 +08:00
if ! opts . IsActive . IsNone ( ) {
cond = cond . And ( builder . Eq { "is_active" : opts . IsActive . IsTrue ( ) } )
2016-03-12 04:33:12 +08:00
}
2017-10-25 01:36:19 +08:00
return cond
}
2016-03-12 04:33:12 +08:00
2017-10-25 01:36:19 +08:00
// SearchUsers takes options i.e. keyword and part of user name to search,
// it returns results in given range and number of total results.
func SearchUsers ( opts * SearchUserOptions ) ( users [ ] * User , _ int64 , _ error ) {
cond := opts . toConds ( )
2017-02-25 21:42:20 +08:00
count , err := x . Where ( cond ) . Count ( new ( User ) )
2016-03-12 04:33:12 +08:00
if err != nil {
return nil , 0 , fmt . Errorf ( "Count: %v" , err )
2014-05-01 11:48:01 +08:00
}
2017-10-25 01:36:19 +08:00
if opts . PageSize <= 0 || opts . PageSize > setting . UI . ExplorePagingNum {
opts . PageSize = setting . UI . ExplorePagingNum
}
if opts . Page <= 0 {
opts . Page = 1
2016-03-12 05:11:33 +08:00
}
2017-10-25 01:36:19 +08:00
if len ( opts . OrderBy ) == 0 {
opts . OrderBy = "name ASC"
}
users = make ( [ ] * User , 0 , opts . PageSize )
return users , count , x . Where ( cond ) .
Limit ( opts . PageSize , ( opts . Page - 1 ) * opts . PageSize ) .
2018-05-24 09:03:42 +08:00
OrderBy ( opts . OrderBy . String ( ) ) .
2017-10-25 01:36:19 +08:00
Find ( & users )
2014-05-01 11:48:01 +08:00
}
2016-11-15 06:33:58 +08:00
// GetStarredRepos returns the repos starred by a particular user
func GetStarredRepos ( userID int64 , private bool ) ( [ ] * Repository , error ) {
sess := x . Where ( "star.uid=?" , userID ) .
Join ( "LEFT" , "star" , "`repository`.id=`star`.repo_id" )
if ! private {
sess = sess . And ( "is_private=?" , false )
}
repos := make ( [ ] * Repository , 0 , 10 )
err := sess . Find ( & repos )
if err != nil {
return nil , err
}
return repos , nil
}
2016-12-24 09:53:11 +08:00
// GetWatchedRepos returns the repos watched by a particular user
func GetWatchedRepos ( userID int64 , private bool ) ( [ ] * Repository , error ) {
sess := x . Where ( "watch.user_id=?" , userID ) .
Join ( "LEFT" , "watch" , "`repository`.id=`watch`.repo_id" )
if ! private {
sess = sess . And ( "is_private=?" , false )
}
repos := make ( [ ] * Repository , 0 , 10 )
err := sess . Find ( & repos )
if err != nil {
return nil , err
}
return repos , nil
}
2017-05-10 21:10:18 +08:00
2018-05-24 12:59:02 +08:00
// deleteKeysMarkedForDeletion returns true if ssh keys needs update
func deleteKeysMarkedForDeletion ( keys [ ] string ) ( bool , error ) {
// Start session
sess := x . NewSession ( )
defer sess . Close ( )
if err := sess . Begin ( ) ; err != nil {
return false , err
}
// Delete keys marked for deletion
var sshKeysNeedUpdate bool
for _ , KeyToDelete := range keys {
key , err := SearchPublicKeyByContent ( KeyToDelete )
if err != nil {
log . Error ( 4 , "SearchPublicKeyByContent: %v" , err )
continue
}
if err = deletePublicKeys ( sess , key . ID ) ; err != nil {
log . Error ( 4 , "deletePublicKeys: %v" , err )
continue
}
sshKeysNeedUpdate = true
}
if err := sess . Commit ( ) ; err != nil {
return false , err
}
return sshKeysNeedUpdate , nil
}
func addLdapSSHPublicKeys ( s * LoginSource , usr * User , SSHPublicKeys [ ] string ) bool {
var sshKeysNeedUpdate bool
for _ , sshKey := range SSHPublicKeys {
if strings . HasPrefix ( strings . ToLower ( sshKey ) , "ssh" ) {
sshKeyName := fmt . Sprintf ( "%s-%s" , s . Name , sshKey [ 0 : 40 ] )
if _ , err := AddPublicKey ( usr . ID , sshKeyName , sshKey , s . ID ) ; err != nil {
log . Error ( 4 , "addLdapSSHPublicKeys[%s]: Error adding LDAP Public SSH Key for user %s: %v" , s . Name , usr . Name , err )
} else {
log . Trace ( "addLdapSSHPublicKeys[%s]: Added LDAP Public SSH Key for user %s" , s . Name , usr . Name )
sshKeysNeedUpdate = true
}
} else {
log . Warn ( "addLdapSSHPublicKeys[%s]: Skipping invalid LDAP Public SSH Key for user %s: %v" , s . Name , usr . Name , sshKey )
}
}
return sshKeysNeedUpdate
}
func synchronizeLdapSSHPublicKeys ( s * LoginSource , SSHPublicKeys [ ] string , usr * User ) bool {
var sshKeysNeedUpdate bool
log . Trace ( "synchronizeLdapSSHPublicKeys[%s]: Handling LDAP Public SSH Key synchronization for user %s" , s . Name , usr . Name )
// Get Public Keys from DB with current LDAP source
var giteaKeys [ ] string
keys , err := ListPublicLdapSSHKeys ( usr . ID , s . ID )
if err != nil {
log . Error ( 4 , "synchronizeLdapSSHPublicKeys[%s]: Error listing LDAP Public SSH Keys for user %s: %v" , s . Name , usr . Name , err )
}
for _ , v := range keys {
giteaKeys = append ( giteaKeys , v . OmitEmail ( ) )
}
// Get Public Keys from LDAP and skip duplicate keys
var ldapKeys [ ] string
for _ , v := range SSHPublicKeys {
ldapKey := strings . Join ( strings . Split ( v , " " ) [ : 2 ] , " " )
if ! util . ExistsInSlice ( ldapKey , ldapKeys ) {
ldapKeys = append ( ldapKeys , ldapKey )
}
}
// Check if Public Key sync is needed
if util . IsEqualSlice ( giteaKeys , ldapKeys ) {
log . Trace ( "synchronizeLdapSSHPublicKeys[%s]: LDAP Public Keys are already in sync for %s (LDAP:%v/DB:%v)" , s . Name , usr . Name , len ( ldapKeys ) , len ( giteaKeys ) )
return false
}
log . Trace ( "synchronizeLdapSSHPublicKeys[%s]: LDAP Public Key needs update for user %s (LDAP:%v/DB:%v)" , s . Name , usr . Name , len ( ldapKeys ) , len ( giteaKeys ) )
// Add LDAP Public SSH Keys that doesn't already exist in DB
var newLdapSSHKeys [ ] string
for _ , LDAPPublicSSHKey := range ldapKeys {
if ! util . ExistsInSlice ( LDAPPublicSSHKey , giteaKeys ) {
newLdapSSHKeys = append ( newLdapSSHKeys , LDAPPublicSSHKey )
}
}
if addLdapSSHPublicKeys ( s , usr , newLdapSSHKeys ) {
sshKeysNeedUpdate = true
}
// Mark LDAP keys from DB that doesn't exist in LDAP for deletion
var giteaKeysToDelete [ ] string
for _ , giteaKey := range giteaKeys {
if ! util . ExistsInSlice ( giteaKey , ldapKeys ) {
log . Trace ( "synchronizeLdapSSHPublicKeys[%s]: Marking LDAP Public SSH Key for deletion for user %s: %v" , s . Name , usr . Name , giteaKey )
giteaKeysToDelete = append ( giteaKeysToDelete , giteaKey )
}
}
// Delete LDAP keys from DB that doesn't exist in LDAP
needUpd , err := deleteKeysMarkedForDeletion ( giteaKeysToDelete )
if err != nil {
log . Error ( 4 , "synchronizeLdapSSHPublicKeys[%s]: Error deleting LDAP Public SSH Keys marked for deletion for user %s: %v" , s . Name , usr . Name , err )
}
if needUpd {
sshKeysNeedUpdate = true
}
return sshKeysNeedUpdate
}
2017-05-10 21:10:18 +08:00
// SyncExternalUsers is used to synchronize users with external authorization source
func SyncExternalUsers ( ) {
2017-05-31 16:57:17 +08:00
if ! taskStatusTable . StartIfNotRunning ( syncExternalUsers ) {
2017-05-10 21:10:18 +08:00
return
}
defer taskStatusTable . Stop ( syncExternalUsers )
log . Trace ( "Doing: SyncExternalUsers" )
ls , err := LoginSources ( )
if err != nil {
log . Error ( 4 , "SyncExternalUsers: %v" , err )
return
}
updateExisting := setting . Cron . SyncExternalUsers . UpdateExisting
for _ , s := range ls {
if ! s . IsActived || ! s . IsSyncEnabled {
continue
}
2018-05-24 12:59:02 +08:00
2017-05-10 21:10:18 +08:00
if s . IsLDAP ( ) {
log . Trace ( "Doing: SyncExternalUsers[%s]" , s . Name )
var existingUsers [ ] int64
2018-05-24 12:59:02 +08:00
var isAttributeSSHPublicKeySet = len ( strings . TrimSpace ( s . LDAP ( ) . AttributeSSHPublicKey ) ) > 0
var sshKeysNeedUpdate bool
2017-05-10 21:10:18 +08:00
// Find all users with this login type
var users [ ] User
x . Where ( "login_type = ?" , LoginLDAP ) .
And ( "login_source = ?" , s . ID ) .
Find ( & users )
sr := s . LDAP ( ) . SearchEntries ( )
for _ , su := range sr {
if len ( su . Username ) == 0 {
continue
}
if len ( su . Mail ) == 0 {
su . Mail = fmt . Sprintf ( "%s@localhost" , su . Username )
}
var usr * User
// Search for existing user
for _ , du := range users {
if du . LowerName == strings . ToLower ( su . Username ) {
usr = & du
break
}
}
fullName := composeFullName ( su . Name , su . Surname , su . Username )
// If no existing user found, create one
if usr == nil {
log . Trace ( "SyncExternalUsers[%s]: Creating user %s" , s . Name , su . Username )
usr = & User {
LowerName : strings . ToLower ( su . Username ) ,
Name : su . Username ,
FullName : fullName ,
LoginType : s . Type ,
LoginSource : s . ID ,
LoginName : su . Username ,
Email : su . Mail ,
IsAdmin : su . IsAdmin ,
IsActive : true ,
}
err = CreateUser ( usr )
2018-05-24 12:59:02 +08:00
2017-05-10 21:10:18 +08:00
if err != nil {
log . Error ( 4 , "SyncExternalUsers[%s]: Error creating user %s: %v" , s . Name , su . Username , err )
2018-05-24 12:59:02 +08:00
} else if isAttributeSSHPublicKeySet {
log . Trace ( "SyncExternalUsers[%s]: Adding LDAP Public SSH Keys for user %s" , s . Name , usr . Name )
if addLdapSSHPublicKeys ( s , usr , su . SSHPublicKey ) {
sshKeysNeedUpdate = true
}
2017-05-10 21:10:18 +08:00
}
} else if updateExisting {
existingUsers = append ( existingUsers , usr . ID )
2018-05-24 12:59:02 +08:00
// Synchronize SSH Public Key if that attribute is set
if isAttributeSSHPublicKeySet && synchronizeLdapSSHPublicKeys ( s , su . SSHPublicKey , usr ) {
sshKeysNeedUpdate = true
}
2017-05-10 21:10:18 +08:00
// Check if user data has changed
if ( len ( s . LDAP ( ) . AdminFilter ) > 0 && usr . IsAdmin != su . IsAdmin ) ||
strings . ToLower ( usr . Email ) != strings . ToLower ( su . Mail ) ||
usr . FullName != fullName ||
! usr . IsActive {
log . Trace ( "SyncExternalUsers[%s]: Updating user %s" , s . Name , usr . Name )
usr . FullName = fullName
usr . Email = su . Mail
// Change existing admin flag only if AdminFilter option is set
if len ( s . LDAP ( ) . AdminFilter ) > 0 {
usr . IsAdmin = su . IsAdmin
}
usr . IsActive = true
2017-08-12 22:18:44 +08:00
err = UpdateUserCols ( usr , "full_name" , "email" , "is_admin" , "is_active" )
2017-05-10 21:10:18 +08:00
if err != nil {
log . Error ( 4 , "SyncExternalUsers[%s]: Error updating user %s: %v" , s . Name , usr . Name , err )
}
}
}
}
2018-05-24 12:59:02 +08:00
// Rewrite authorized_keys file if LDAP Public SSH Key attribute is set and any key was added or removed
if sshKeysNeedUpdate {
RewriteAllPublicKeys ( )
}
2017-05-10 21:10:18 +08:00
// Deactivate users not present in LDAP
if updateExisting {
for _ , usr := range users {
found := false
for _ , uid := range existingUsers {
if usr . ID == uid {
found = true
break
}
}
if ! found {
log . Trace ( "SyncExternalUsers[%s]: Deactivating user %s" , s . Name , usr . Name )
usr . IsActive = false
2017-08-12 22:18:44 +08:00
err = UpdateUserCols ( & usr , "is_active" )
2017-05-10 21:10:18 +08:00
if err != nil {
log . Error ( 4 , "SyncExternalUsers[%s]: Error deactivating user %s: %v" , s . Name , usr . Name , err )
}
}
}
}
}
}
}