BigW Consortium Gitlab

group.rb 4.25 KB
Newer Older
1 2
# == Schema Information
#
Dmitriy Zaporozhets committed
3
# Table name: namespaces
4
#
5 6 7 8 9
#  id                     :integer          not null, primary key
#  name                   :string(255)      not null
#  path                   :string(255)      not null
#  owner_id               :integer
#  visibility_level       :integer          default(20), not null
Douwe Maan committed
10
#  created_at             :datetime
11 12 13 14
#  updated_at             :datetime
#  type                   :string(255)
#  description            :string(255)      default(""), not null
#  avatar                 :string(255)
15 16
#

Steven Thonus committed
17 18 19
require 'carrierwave/orm/activerecord'
require 'file_size_validator'

20
class Group < Namespace
21
  include Gitlab::ConfigHelper
22
  include Gitlab::VisibilityLevel
23
  include Referable
24

25
  has_many :group_members, dependent: :destroy, as: :source, class_name: 'GroupMember'
26
  alias_method :members, :group_members
27
  has_many :users, through: :group_members
28 29
  has_many :project_group_links, dependent: :destroy
  has_many :shared_projects, through: :project_group_links, source: :project
Andrey Kumanyaev committed
30

31
  validate :avatar_type, if: ->(user) { user.avatar.present? && user.avatar_changed? }
32 33
  validate :visibility_level_allowed_by_projects

34
  validates :avatar, file_size: { maximum: 200.kilobytes.to_i }
Steven Thonus committed
35

36
  mount_uploader :avatar, AvatarUploader
37

38 39 40
  after_create :post_create_hook
  after_destroy :post_destroy_hook

41
  class << self
42 43 44 45 46 47 48
    # Searches for groups matching the given query.
    #
    # This method uses ILIKE on PostgreSQL and LIKE on MySQL.
    #
    # query - The search query as a String
    #
    # Returns an ActiveRecord::Relation.
49
    def search(query)
50
      table   = Namespace.arel_table
51 52 53
      pattern = "%#{query}%"

      where(table[:name].matches(pattern).or(table[:path].matches(pattern)))
54 55 56 57 58
    end

    def sort(method)
      order_by(method)
    end
59 60

    def reference_prefix
61 62 63 64 65
      User.reference_prefix
    end

    def reference_pattern
      User.reference_pattern
66
    end
67 68 69 70

    def visible_to_user(user)
      where(id: user.authorized_groups.select(:id).reorder(nil))
    end
71 72 73 74
  end

  def to_reference(_from_project = nil)
    "#{self.class.reference_prefix}#{name}"
75 76
  end

77 78 79
  def human_name
    name
  end
80

Felipe Artur committed
81 82 83 84
  def visibility_level_field
    visibility_level
  end

85
  def visibility_level_allowed_by_projects
Douwe Maan committed
86
    allowed_by_projects = self.projects.where('visibility_level > ?', self.visibility_level).none?
87 88 89 90 91 92 93 94 95

    unless allowed_by_projects
      level_name = Gitlab::VisibilityLevel.level_name(visibility_level).downcase
      self.errors.add(:visibility_level, "#{level_name} is not allowed since there are projects with higher visibility.")
    end

    allowed_by_projects
  end

96 97 98 99 100 101
  def avatar_url(size = nil)
    if avatar.present?
      [gitlab_config.url, avatar.url].join
    end
  end

Douwe Maan committed
102 103 104 105
  def owners
    @owners ||= group_members.owners.includes(:user).map(&:user)
  end

106
  def add_users(user_ids, access_level, current_user = nil)
107 108
    user_ids.each do |user_id|
      Member.add_user(self.group_members, user_id, access_level, current_user)
109 110 111
    end
  end

112 113
  def add_user(user, access_level, current_user = nil)
    add_users([user], access_level, current_user)
114 115
  end

116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131
  def add_guest(user, current_user = nil)
    add_user(user, Gitlab::Access::GUEST, current_user)
  end

  def add_reporter(user, current_user = nil)
    add_user(user, Gitlab::Access::REPORTER, current_user)
  end

  def add_developer(user, current_user = nil)
    add_user(user, Gitlab::Access::DEVELOPER, current_user)
  end

  def add_master(user, current_user = nil)
    add_user(user, Gitlab::Access::MASTER, current_user)
  end

Douwe Maan committed
132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147
  def add_owner(user, current_user = nil)
    add_user(user, Gitlab::Access::OWNER, current_user)
  end

  def has_owner?(user)
    owners.include?(user)
  end

  def has_master?(user)
    members.masters.where(user_id: user).any?
  end

  def last_owner?(user)
    has_owner?(user) && owners.size == 1
  end

Steven Thonus committed
148 149 150 151 152
  def avatar_type
    unless self.avatar.image?
      self.errors.add :avatar, "only images allowed"
    end
  end
153

154
  def post_create_hook
155 156
    Gitlab::AppLogger.info("Group \"#{name}\" was created")

157 158 159 160
    system_hook_service.execute_hooks_for(self, :create)
  end

  def post_destroy_hook
161 162
    Gitlab::AppLogger.info("Group \"#{name}\" was removed")

163 164 165 166 167 168
    system_hook_service.execute_hooks_for(self, :destroy)
  end

  def system_hook_service
    SystemHooksService.new
  end
169
end