user.rb 8.4 KB
Newer Older
D
Dmitriy Zaporozhets 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
# == Schema Information
#
# Table name: users
#
#  id                     :integer          not null, primary key
#  email                  :string(255)      default(""), not null
#  encrypted_password     :string(255)      default(""), not null
#  reset_password_token   :string(255)
#  reset_password_sent_at :datetime
#  remember_created_at    :datetime
#  sign_in_count          :integer          default(0)
#  current_sign_in_at     :datetime
#  last_sign_in_at        :datetime
#  current_sign_in_ip     :string(255)
#  last_sign_in_ip        :string(255)
#  created_at             :datetime         not null
#  updated_at             :datetime         not null
#  name                   :string(255)
#  admin                  :boolean          default(FALSE), not null
#  projects_limit         :integer          default(10)
#  skype                  :string(255)      default(""), not null
#  linkedin               :string(255)      default(""), not null
#  twitter                :string(255)      default(""), not null
#  authentication_token   :string(255)
#  dark_scheme            :boolean          default(FALSE), not null
#  theme_id               :integer          default(1), not null
#  bio                    :string(255)
#  blocked                :boolean          default(FALSE), not null
#  failed_attempts        :integer          default(0)
#  locked_at              :datetime
#  extern_uid             :string(255)
#  provider               :string(255)
D
Dmitriy Zaporozhets 已提交
33
#  username               :string(255)
D
Dmitriy Zaporozhets 已提交
34 35
#

G
gitlabhq 已提交
36
class User < ActiveRecord::Base
37
  devise :database_authenticatable, :token_authenticatable, :lockable,
M
Marin Jankovski 已提交
38
         :recoverable, :rememberable, :trackable, :validatable, :omniauthable, :registerable
G
gitlabhq 已提交
39

40
  attr_accessible :email, :password, :password_confirmation, :remember_me, :bio, :name, :username,
N
Nihad Abbasov 已提交
41
                  :skype, :linkedin, :twitter, :dark_scheme, :theme_id, :force_random_password,
42 43
                  :extern_uid, :provider, as: [:default, :admin]
  attr_accessible :projects_limit, as: :admin
44 45

  attr_accessor :force_random_password
G
gitlabhq 已提交
46

47 48
  # Namespace for personal projects
  has_one :namespace, class_name: "Namespace", foreign_key: :owner_id, conditions: 'type IS NULL', dependent: :destroy
49
  has_many :groups, class_name: "Group", foreign_key: :owner_id
50

N
Nihad Abbasov 已提交
51 52 53 54 55 56 57 58 59
  has_many :keys, dependent: :destroy
  has_many :users_projects, dependent: :destroy
  has_many :issues, foreign_key: :author_id, dependent: :destroy
  has_many :notes, foreign_key: :author_id, dependent: :destroy
  has_many :merge_requests, foreign_key: :author_id, dependent: :destroy
  has_many :events, class_name: "Event", foreign_key: :author_id, dependent: :destroy
  has_many :recent_events, class_name: "Event", foreign_key: :author_id, order: "id DESC"
  has_many :assigned_issues, class_name: "Issue", foreign_key: :assignee_id, dependent: :destroy
  has_many :assigned_merge_requests, class_name: "MergeRequest", foreign_key: :assignee_id, dependent: :destroy
60

C
Cyril 已提交
61
  validates :name, presence: true
62
  validates :bio, length: { within: 0..255 }
63
  validates :extern_uid, allow_blank: true, uniqueness: {scope: :provider}
N
Nihad Abbasov 已提交
64
  validates :projects_limit, presence: true, numericality: {greater_than_or_equal_to: 0}
65 66 67 68
  validates :username, presence: true, uniqueness: true,
            format: { with: Gitlab::Regex.username_regex,
                      message: "only letters, digits & '_' '-' '.' allowed. Letter should be first" }

69

70 71
  validate :namespace_uniq, if: ->(user) { user.username_changed? }

72
  before_validation :generate_password, on: :create
N
Nihad Abbasov 已提交
73 74
  before_save :ensure_authentication_token
  alias_attribute :private_token, :authentication_token
75

76
  delegate :path, to: :namespace, allow_nil: true, prefix: true
77

A
Andrey Kumanyaev 已提交
78 79 80 81
  # Scopes
  scope :admins, where(admin:  true)
  scope :blocked, where(blocked:  true)
  scope :active, where(blocked:  false)
82
  scope :alphabetically, order('name ASC')
A
Andrey Kumanyaev 已提交
83

84 85 86
  #
  # Class methods
  #
A
Andrey Kumanyaev 已提交
87 88 89 90 91 92 93 94 95
  class << self
    def filter filter_name
      case filter_name
      when "admins"; self.admins
      when "blocked"; self.blocked
      when "wop"; self.without_projects
      else
        self.active
      end
96 97
    end

98 99 100 101 102 103 104 105
    def not_in_project(project)
      if project.users.present?
        where("id not in (:ids)", ids: project.users.map(&:id) )
      else
        scoped
      end
    end

A
Andrey Kumanyaev 已提交
106 107
    def without_projects
      where('id NOT IN (SELECT DISTINCT(user_id) FROM users_projects)')
108 109
    end

A
Andrey Kumanyaev 已提交
110 111 112
    def create_from_omniauth(auth, ldap = false)
      gitlab_auth.create_from_omniauth(auth, ldap)
    end
G
gitlabhq 已提交
113

A
Andrey Kumanyaev 已提交
114 115 116
    def find_or_new_for_omniauth(auth)
      gitlab_auth.find_or_new_for_omniauth(auth)
    end
F
Florian Unglaub 已提交
117

A
Andrey Kumanyaev 已提交
118 119 120
    def find_for_ldap_auth(auth, signed_in_resource = nil)
      gitlab_auth.find_for_ldap_auth(auth, signed_in_resource)
    end
121

A
Andrey Kumanyaev 已提交
122 123 124
    def gitlab_auth
      Gitlab::Auth.new
    end
125

A
Andrey Kumanyaev 已提交
126 127 128
    def search query
      where("name LIKE :query or email LIKE :query", query: "%#{query}%")
    end
V
vsizov 已提交
129
  end
R
randx 已提交
130

131 132 133
  #
  # Instance methods
  #
A
Andrey Kumanyaev 已提交
134 135 136 137
  def generate_password
    if self.force_random_password
      self.password = self.password_confirmation = Devise.friendly_token.first(8)
    end
R
randx 已提交
138
  end
139

140 141 142 143 144 145
  def namespace_uniq
    namespace_name = self.username
    if Namespace.find_by_path(namespace_name)
      self.errors.add :username, "already exist"
    end
  end
146 147 148 149 150 151 152 153 154

  # Namespaces user has access to
  def namespaces
    namespaces = []

    # Add user account namespace
    namespaces << self.namespace if self.namespace

    # Add groups you can manage
155 156
    namespaces += groups.all

157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192
    namespaces
  end

  # Groups where user is an owner
  def owned_groups
    groups
  end

  # Groups user has access to
  def authorized_groups
    @authorized_groups ||= begin
                           groups = Group.where(id: self.authorized_projects.pluck(:namespace_id)).all
                           groups = groups + self.groups
                           groups.uniq
                         end
  end


  # Projects user has access to
  def authorized_projects
    project_ids = users_projects.pluck(:project_id)
    project_ids = project_ids | owned_projects.pluck(:id)
    Project.where(id: project_ids)
  end

  # Projects in user namespace
  def personal_projects
    Project.personal(self)
  end

  # Projects where user is an owner
  def owned_projects
    Project.where("(projects.namespace_id IN (:namespaces)) OR
                  (projects.namespace_id IS NULL AND projects.creator_id = :user_id)",
                  namespaces: namespaces.map(&:id), user_id: self.id)
  end
193

194 195 196
  # Team membership in authorized projects
  def tm_in_authorized_projects
    UsersProject.where(project_id:  authorized_projects.map(&:id), user_id: self.id)
197
  end
D
Dmitriy Zaporozhets 已提交
198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233

  # Returns a string for use as a Gitolite user identifier
  #
  # Note that Gitolite 2.x requires the following pattern for users:
  #
  #   ^@?[0-9a-zA-Z][0-9a-zA-Z._\@+-]*$
  def identifier
    # Replace non-word chars with underscores, then make sure it starts with
    # valid chars
    email.gsub(/\W/, '_').gsub(/\A([\W\_])+/, '')
  end

  def is_admin?
    admin
  end

  def require_ssh_key?
    keys.count == 0
  end

  def can_create_project?
    projects_limit > personal_projects.count
  end

  def can_create_group?
    is_admin?
  end

  def abilities
    @abilities ||= begin
                     abilities = Six.new
                     abilities << Ability
                     abilities
                   end
  end

234 235 236 237
  def can_select_namespace?
    several_namespaces? || admin
  end

D
Dmitriy Zaporozhets 已提交
238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285
  def can? action, subject
    abilities.allowed?(self, action, subject)
  end

  def first_name
    name.split.first unless name.blank?
  end

  def cared_merge_requests
    MergeRequest.where("author_id = :id or assignee_id = :id", id: self.id)
  end

  # Remove user from all projects and
  # set blocked attribute to true
  def block
    users_projects.find_each do |membership|
      return false unless membership.destroy
    end

    self.blocked = true
    save
  end

  def projects_limit_percent
    return 100 if projects_limit.zero?
    (personal_projects.count.to_f / projects_limit) * 100
  end

  def recent_push project_id = nil
    # Get push events not earlier than 2 hours ago
    events = recent_events.code_push.where("created_at > ?", Time.now - 2.hours)
    events = events.where(project_id: project_id) if project_id

    # Take only latest one
    events = events.recent.limit(1).first
  end

  def projects_sorted_by_activity
    authorized_projects.sorted_by_activity
  end

  def several_namespaces?
    namespaces.size > 1
  end

  def namespace_id
    namespace.try :id
  end
G
gitlabhq 已提交
286
end