project.rb 32.0 KB
Newer Older
1 2
require 'carrierwave/orm/activerecord'

G
gitlabhq 已提交
3
class Project < ActiveRecord::Base
4
  include Gitlab::ConfigHelper
5
  include Gitlab::ShellAdapter
6
  include Gitlab::VisibilityLevel
7
  include Gitlab::CurrentSettings
8 9
  include Referable
  include Sortable
10
  include AfterCommitQueue
11
  include CaseSensitivity
12
  include TokenAuthenticatable
R
Robert Speicher 已提交
13

14
  extend Gitlab::ConfigHelper
15

J
Jared Szechy 已提交
16 17
  UNKNOWN_IMPORT_URL = 'http://unknown.git'

18
  default_value_for :archived, false
19 20 21
  default_value_for :visibility_level, gitlab_config_features.visibility_level
  default_value_for :issues_enabled, gitlab_config_features.issues
  default_value_for :merge_requests_enabled, gitlab_config_features.merge_requests
22
  default_value_for :builds_enabled, gitlab_config_features.builds
23 24
  default_value_for :wiki_enabled, gitlab_config_features.wiki
  default_value_for :snippets_enabled, gitlab_config_features.snippets
25
  default_value_for :container_registry_enabled, gitlab_config_features.container_registry
K
Kamil Trzcinski 已提交
26
  default_value_for(:shared_runners_enabled) { current_application_settings.shared_runners_enabled }
27

28 29
  # set last_activity_at to the same as created_at
  after_create :set_last_activity_at
30
  def set_last_activity_at
31
    update_column(:last_activity_at, self.created_at)
32 33
  end

34
  # update visibility_level of forks
35 36 37 38 39 40 41 42 43 44 45 46
  after_update :update_forks_visibility_level
  def update_forks_visibility_level
    return unless visibility_level < visibility_level_was

    forks.each do |forked_project|
      if forked_project.visibility_level > visibility_level
        forked_project.visibility_level = visibility_level
        forked_project.save!
      end
    end
  end

47
  ActsAsTaggableOn.strict_case_match = true
48
  acts_as_taggable_on :tags
D
Dmitriy Zaporozhets 已提交
49

50
  attr_accessor :new_default_branch
51
  attr_accessor :old_path_with_namespace
52

53 54
  alias_attribute :title, :name

55
  # Relations
56
  belongs_to :creator, foreign_key: 'creator_id', class_name: 'User'
57
  belongs_to :group, -> { where(type: Group) }, foreign_key: 'namespace_id'
58
  belongs_to :namespace
59

60
  has_one :last_event, -> {order 'events.created_at DESC'}, class_name: 'Event', foreign_key: 'project_id'
61 62 63

  # Project services
  has_many :services
D
Dmitriy Zaporozhets 已提交
64
  has_one :campfire_service, dependent: :destroy
K
Kirilll Zaitsev 已提交
65
  has_one :drone_ci_service, dependent: :destroy
66
  has_one :emails_on_push_service, dependent: :destroy
67
  has_one :builds_email_service, dependent: :destroy
A
Aorimn 已提交
68
  has_one :irker_service, dependent: :destroy
69
  has_one :pivotaltracker_service, dependent: :destroy
R
Ronald van Eede 已提交
70
  has_one :hipchat_service, dependent: :destroy
71
  has_one :flowdock_service, dependent: :destroy
C
Carlos Paramio 已提交
72
  has_one :assembla_service, dependent: :destroy
J
Jeremy 已提交
73
  has_one :asana_service, dependent: :destroy
74
  has_one :gemnasium_service, dependent: :destroy
75
  has_one :slack_service, dependent: :destroy
76
  has_one :buildkite_service, dependent: :destroy
D
Drew Blessing 已提交
77
  has_one :bamboo_service, dependent: :destroy
78
  has_one :teamcity_service, dependent: :destroy
79
  has_one :pushover_service, dependent: :destroy
80 81
  has_one :jira_service, dependent: :destroy
  has_one :redmine_service, dependent: :destroy
M
Marin Jankovski 已提交
82
  has_one :custom_issue_tracker_service, dependent: :destroy
83
  has_one :gitlab_issue_tracker_service, dependent: :destroy
D
Dmitriy Zaporozhets 已提交
84
  has_one :external_wiki_service, dependent: :destroy
85

86 87 88 89 90
  has_one  :forked_project_link,  dependent: :destroy, foreign_key: "forked_to_project_id"
  has_one  :forked_from_project,  through:   :forked_project_link

  has_many :forked_project_links, foreign_key: "forked_from_project_id"
  has_many :forks,                through:     :forked_project_links, source: :forked_to_project
91

92
  # Merge Requests for target project should be removed with it
93
  has_many :merge_requests,     dependent: :destroy, foreign_key: 'target_project_id'
94
  # Merge requests from source project should be kept when source project was removed
95
  has_many :fork_merge_requests, foreign_key: 'source_project_id', class_name: MergeRequest
96
  has_many :issues,             dependent: :destroy
97
  has_many :labels,             dependent: :destroy
98 99
  has_many :services,           dependent: :destroy
  has_many :events,             dependent: :destroy
100 101
  has_many :milestones,         dependent: :destroy
  has_many :notes,              dependent: :destroy
102 103
  has_many :snippets,           dependent: :destroy, class_name: 'ProjectSnippet'
  has_many :hooks,              dependent: :destroy, class_name: 'ProjectHook'
104
  has_many :protected_branches, dependent: :destroy
105 106
  has_many :project_members, dependent: :destroy, as: :source, class_name: 'ProjectMember'
  has_many :users, through: :project_members
107 108
  has_many :deploy_keys_projects, dependent: :destroy
  has_many :deploy_keys, through: :deploy_keys_projects
C
Ciro Santilli 已提交
109 110
  has_many :users_star_projects, dependent: :destroy
  has_many :starrers, through: :users_star_projects, source: :user
111
  has_many :releases, dependent: :destroy
M
Marin Jankovski 已提交
112 113
  has_many :lfs_objects_projects, dependent: :destroy
  has_many :lfs_objects, through: :lfs_objects_projects
114 115
  has_many :project_group_links, dependent: :destroy
  has_many :invited_groups, through: :project_group_links, source: :group
116
  has_many :todos, dependent: :destroy
117
  has_many :notification_settings, dependent: :destroy, as: :source
118

119
  has_one :import_data, dependent: :destroy, class_name: "ProjectImportData"
K
Kamil Trzcinski 已提交
120

121
  has_many :commit_statuses, dependent: :destroy, class_name: 'CommitStatus', foreign_key: :gl_project_id
122
  has_many :pipelines, dependent: :destroy, class_name: 'Ci::Pipeline', foreign_key: :gl_project_id
123 124 125 126 127
  has_many :builds, class_name: 'Ci::Build', foreign_key: :gl_project_id # the builds are created from the commit_statuses
  has_many :runner_projects, dependent: :destroy, class_name: 'Ci::RunnerProject', foreign_key: :gl_project_id
  has_many :runners, through: :runner_projects, source: :runner, class_name: 'Ci::Runner'
  has_many :variables, dependent: :destroy, class_name: 'Ci::Variable', foreign_key: :gl_project_id
  has_many :triggers, dependent: :destroy, class_name: 'Ci::Trigger', foreign_key: :gl_project_id
K
Kamil Trzcinski 已提交
128

129
  accepts_nested_attributes_for :variables, allow_destroy: true
130

131
  delegate :name, to: :owner, allow_nil: true, prefix: true
132
  delegate :members, to: :team, prefix: true
133

A
Andrey Kumanyaev 已提交
134
  # Validations
135
  validates :creator, presence: true, on: :create
136
  validates :description, length: { maximum: 2000 }, allow_blank: true
137 138 139 140
  validates :name,
    presence: true,
    length: { within: 0..255 },
    format: { with: Gitlab::Regex.project_name_regex,
D
Douwe Maan 已提交
141
              message: Gitlab::Regex.project_name_regex_message }
142 143 144
  validates :path,
    presence: true,
    length: { within: 0..255 },
D
Douwe Maan 已提交
145 146
    format: { with: Gitlab::Regex.project_path_regex,
              message: Gitlab::Regex.project_path_regex_message }
D
Dmitriy Zaporozhets 已提交
147
  validates :issues_enabled, :merge_requests_enabled,
A
Andrey Kumanyaev 已提交
148
            :wiki_enabled, inclusion: { in: [true, false] }
149
  validates :issues_tracker_id, length: { maximum: 255 }, allow_blank: true
150
  validates :namespace, presence: true
151 152
  validates_uniqueness_of :name, scope: :namespace_id
  validates_uniqueness_of :path, scope: :namespace_id
153
  validates :import_url,
R
Robert Speicher 已提交
154
    url: { protocols: %w(ssh git http https) },
155
    if: :external_import?
156
  validates :star_count, numericality: { greater_than_or_equal_to: 0 }
157
  validate :check_limit, on: :create
158
  validate :avatar_type,
159
    if: ->(project) { project.avatar.present? && project.avatar_changed? }
160
  validates :avatar, file_size: { maximum: 200.kilobytes.to_i }
D
Douwe Maan 已提交
161
  validate :visibility_level_allowed_by_group
D
Douwe Maan 已提交
162
  validate :visibility_level_allowed_as_fork
163

164
  add_authentication_token_field :runners_token
165
  before_save :ensure_runners_token
K
Kamil Trzcinski 已提交
166

D
Douwe Maan 已提交
167
  mount_uploader :avatar, AvatarUploader
A
Andrey Kumanyaev 已提交
168

169
  # Scopes
170 171
  default_scope { where(pending_delete: false) }

172
  scope :sorted_by_activity, -> { reorder(last_activity_at: :desc) }
173 174
  scope :sorted_by_stars, -> { reorder('projects.star_count DESC') }

175
  scope :in_namespace, ->(namespace_ids) { where(namespace_id: namespace_ids) }
D
Dmitriy Zaporozhets 已提交
176
  scope :personal, ->(user) { where(namespace_id: user.namespace_id) }
177
  scope :joined, ->(user) { where('namespace_id != ?', user.namespace_id) }
R
Rémy Coutable 已提交
178
  scope :visible_to_user, ->(user) { where(id: user.authorized_projects.select(:id).reorder(nil)) }
179
  scope :non_archived, -> { where(archived: false) }
R
Rubén Dávila 已提交
180
  scope :for_milestones, ->(ids) { joins(:milestones).where('milestones.id' => ids).distinct }
R
Rémy Coutable 已提交
181 182 183 184
  scope :with_push, -> { joins(:events).where('events.action = ?', Event::PUSHED) }

  scope :active, -> { joins(:issues, :notes, :merge_requests).order('issues.created_at, notes.created_at, merge_requests.created_at DESC') }
  scope :abandoned, -> { where('projects.last_activity_at < ?', 6.months.ago) }
185

186 187
  state_machine :import_status, initial: :none do
    event :import_start do
188
      transition [:none, :finished] => :started
189 190 191
    end

    event :import_finish do
192
      transition started: :finished
193 194 195
    end

    event :import_fail do
196
      transition started: :failed
197 198 199
    end

    event :import_retry do
200
      transition failed: :started
201 202 203 204
    end

    state :started
    state :finished
205 206
    state :failed

207
    after_transition any => :finished, do: :reset_cache_and_import_attrs
208 209
  end

A
Andrey Kumanyaev 已提交
210
  class << self
211 212 213 214 215 216 217
    # Searches for a list of projects based on the query given in `query`.
    #
    # On PostgreSQL this method uses "ILIKE" to perform a case-insensitive
    # search. On MySQL a regular "LIKE" is used as it's already
    # case-insensitive.
    #
    # query - The search query as a String.
218
    def search(query)
219
      ptable  = arel_table
220 221 222 223 224 225 226 227 228
      ntable  = Namespace.arel_table
      pattern = "%#{query}%"

      projects = select(:id).where(
        ptable[:path].matches(pattern).
          or(ptable[:name].matches(pattern)).
          or(ptable[:description].matches(pattern))
      )

229 230 231 232 233 234
      # We explicitly remove any eager loading clauses as they're:
      #
      # 1. Not needed by this query
      # 2. Combined with .joins(:namespace) lead to all columns from the
      #    projects & namespaces tables being selected, leading to a SQL error
      #    due to the columns of all UNION'd queries no longer being the same.
235
      namespaces = select(:id).
236
        except(:includes).
237 238 239 240 241 242
        joins(:namespace).
        where(ntable[:name].matches(pattern))

      union = Gitlab::SQL::Union.new([projects, namespaces])

      where("projects.id IN (#{union.to_sql})")
A
Andrey Kumanyaev 已提交
243
    end
244

245
    def search_by_visibility(level)
J
Josh Frye 已提交
246
      where(visibility_level: Gitlab::VisibilityLevel.const_get(level.upcase))
247 248
    end

249
    def search_by_title(query)
250 251 252 253
      pattern = "%#{query}%"
      table   = Project.arel_table

      non_archived.where(table[:name].matches(pattern))
254 255
    end

Y
Yorick Peterse 已提交
256 257 258 259 260 261 262 263
    # Finds a single project for the given path.
    #
    # path - The full project path (including namespace path).
    #
    # Returns a Project, or nil if no project could be found.
    def find_with_namespace(path)
      where_paths_in([path]).reorder(nil).take
    end
264

Y
Yorick Peterse 已提交
265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288
    # Builds a relation to find multiple projects by their full paths.
    #
    # Each path must be in the following format:
    #
    #     namespace_path/project_path
    #
    # For example:
    #
    #     gitlab-org/gitlab-ce
    #
    # Usage:
    #
    #     Project.where_paths_in(%w{gitlab-org/gitlab-ce gitlab-org/gitlab-ee})
    #
    # This would return the projects with the full paths matching the values
    # given.
    #
    # paths - An Array of full paths (namespace path + project path) for which
    #         to find the projects.
    #
    # Returns an ActiveRecord::Relation.
    def where_paths_in(paths)
      wheres = []
      cast_lower = Gitlab::Database.postgresql?
289

Y
Yorick Peterse 已提交
290 291
      paths.each do |path|
        namespace_path, project_path = path.split('/', 2)
292

Y
Yorick Peterse 已提交
293 294 295 296
        next unless namespace_path && project_path

        namespace_path = connection.quote(namespace_path)
        project_path = connection.quote(project_path)
297

Y
Yorick Peterse 已提交
298 299
        where = "(namespaces.path = #{namespace_path}
          AND projects.path = #{project_path})"
300

Y
Yorick Peterse 已提交
301 302 303 304 305 306 307 308 309
        if cast_lower
          where = "(
            #{where}
            OR (
              LOWER(namespaces.path) = LOWER(#{namespace_path})
              AND LOWER(projects.path) = LOWER(#{project_path})
            )
          )"
        end
310

Y
Yorick Peterse 已提交
311 312
        wheres << where
      end
313

Y
Yorick Peterse 已提交
314 315 316 317 318
      if wheres.empty?
        none
      else
        joins(:namespace).where(wheres.join(' OR '))
      end
K
Kamil Trzcinski 已提交
319 320
    end

321 322 323
    def visibility_levels
      Gitlab::VisibilityLevel.options
    end
324 325

    def sort(method)
326 327 328 329
      if method == 'repository_size_desc'
        reorder(repository_size: :desc, id: :desc)
      else
        order_by(method)
330 331
      end
    end
332 333 334 335 336

    def reference_pattern
      name_pattern = Gitlab::Regex::NAMESPACE_REGEX_STR
      %r{(?<project>#{name_pattern}/#{name_pattern})}
    end
Y
Yorick Peterse 已提交
337 338 339 340 341 342 343 344 345 346 347 348 349 350

    def trending(since = 1.month.ago)
      # By counting in the JOIN we don't expose the GROUP BY to the outer query.
      # This means that calls such as "any?" and "count" just return a number of
      # the total count, instead of the counts grouped per project as a Hash.
      join_body = "INNER JOIN (
        SELECT project_id, COUNT(*) AS amount
        FROM notes
        WHERE created_at >= #{sanitize(since)}
        GROUP BY project_id
      ) join_note_counts ON projects.id = join_note_counts.project_id"

      joins(join_body).reorder('join_note_counts.amount DESC')
    end
J
James Lopez 已提交
351 352 353 354 355

    # Deletes gitlab project export files older than 24 hours
    def archive_gitlab_exports!
      Gitlab::Popen.popen(%W(find #{export_path} -not -path #{export_path} -mmin +1440 -delete))
    end
356 357
  end

D
Dmitriy Zaporozhets 已提交
358
  def team
359
    @team ||= ProjectTeam.new(self)
D
Dmitriy Zaporozhets 已提交
360 361 362
  end

  def repository
363
    @repository ||= Repository.new(path_with_namespace, self)
364 365
  end

366 367 368 369
  def container_registry_path_with_namespace
    path_with_namespace.downcase
  end

370
  def container_registry_repository
K
Kamil Trzcinski 已提交
371 372
    return unless Gitlab.config.registry.enabled

373
    @container_registry_repository ||= begin
374
      token = Auth::ContainerRegistryAuthenticationService.full_access_token(container_registry_path_with_namespace)
375
      url = Gitlab.config.registry.api_url
376 377
      host_port = Gitlab.config.registry.host_port
      registry = ContainerRegistry::Registry.new(url, token: token, path: host_port)
378
      registry.repository(container_registry_path_with_namespace)
379
    end
380 381
  end

382
  def container_registry_repository_url
K
Kamil Trzcinski 已提交
383
    if Gitlab.config.registry.enabled
384
      "#{Gitlab.config.registry.host_port}/#{container_registry_path_with_namespace}"
385
    end
386 387
  end

388
  def has_container_registry_tags?
K
Kamil Trzcinski 已提交
389 390 391
    return unless container_registry_repository

    container_registry_repository.tags.any?
392 393
  end

394
  def commit(id = 'HEAD')
395
    repository.commit(id)
D
Dmitriy Zaporozhets 已提交
396 397
  end

398
  def merge_base_commit(first_commit_id, second_commit_id)
D
Douwe Maan 已提交
399 400
    sha = repository.merge_base(first_commit_id, second_commit_id)
    repository.commit(sha) if sha
401 402
  end

403
  def saved?
404
    id && persisted?
405 406
  end

407
  def add_import_job
408
    if forked?
409
      job_id = RepositoryForkWorker.perform_async(self.id, forked_from_project.path_with_namespace, self.namespace.path)
410
    else
411 412 413 414 415 416 417
      job_id = RepositoryImportWorker.perform_async(self.id)
    end

    if job_id
      Rails.logger.info "Import job started for #{path_with_namespace} with job ID #{job_id}"
    else
      Rails.logger.error "Import job failed to start for #{path_with_namespace}"
418
    end
419 420
  end

421
  def reset_cache_and_import_attrs
422 423 424 425
    update(import_error: nil)

    ProjectCacheWorker.perform_async(self.id)

426
    self.import_data.destroy if self.import_data
427 428
  end

J
James Lopez 已提交
429
  def import_url=(value)
430
    import_url = Gitlab::UrlSanitizer.new(value)
431
    create_or_update_import_data(credentials: import_url.credentials)
J
James Lopez 已提交
432
    super(import_url.sanitized_url)
J
James Lopez 已提交
433 434 435
  end

  def import_url
436
    if import_data && super
437
      import_url = Gitlab::UrlSanitizer.new(super, credentials: import_data.credentials)
J
James Lopez 已提交
438 439 440
      import_url.full_url
    else
      super
J
James Lopez 已提交
441 442
    end
  end
J
James Lopez 已提交
443

444
  def create_or_update_import_data(data: nil, credentials: nil)
J
James Lopez 已提交
445
    project_import_data = import_data || build_import_data
J
James Lopez 已提交
446 447 448 449
    if data
      project_import_data.data ||= {}
      project_import_data.data = project_import_data.data.merge(data)
    end
450 451 452 453
    if credentials
      project_import_data.credentials ||= {}
      project_import_data.credentials = project_import_data.credentials.merge(credentials)
    end
454 455

    project_import_data.save
J
James Lopez 已提交
456
  end
J
James Lopez 已提交
457

458
  def import?
459
    external_import? || forked? || gitlab_project_import?
460 461
  end

462 463 464 465
  def no_import?
    import_status == 'none'
  end

466
  def external_import?
467 468 469
    import_url.present?
  end

470
  def imported?
471 472 473 474 475 476 477 478 479 480 481 482 483
    import_finished?
  end

  def import_in_progress?
    import? && import_status == 'started'
  end

  def import_failed?
    import_status == 'failed'
  end

  def import_finished?
    import_status == 'finished'
484 485
  end

D
Douwe Maan 已提交
486
  def safe_import_url
487
    Gitlab::UrlSanitizer.new(import_url).masked_url
D
Douwe Maan 已提交
488 489
  end

490 491 492 493
  def gitlab_project_import?
    import_type == 'gitlab_project'
  end

494
  def check_limit
495
    unless creator.can_create_project? or namespace.kind == 'group'
496 497 498
      projects_limit = creator.projects_limit

      if projects_limit == 0
P
Phil Hughes 已提交
499
        self.errors.add(:limit_reached, "Personal project creation is not allowed. Please contact your administrator with questions")
500
      else
P
Phil Hughes 已提交
501
        self.errors.add(:limit_reached, "Your project limit is #{projects_limit} projects! Please contact your administrator to increase it")
502
      end
503 504
    end
  rescue
D
Douwe Maan 已提交
505
    self.errors.add(:base, "Can't check your ability to create project")
G
gitlabhq 已提交
506 507
  end

D
Douwe Maan 已提交
508 509 510 511 512 513 514 515 516 517 518 519 520
  def visibility_level_allowed_by_group
    return if visibility_level_allowed_by_group?

    level_name = Gitlab::VisibilityLevel.level_name(self.visibility_level).downcase
    group_level_name = Gitlab::VisibilityLevel.level_name(self.group.visibility_level).downcase
    self.errors.add(:visibility_level, "#{level_name} is not allowed in a #{group_level_name} group.")
  end

  def visibility_level_allowed_as_fork
    return if visibility_level_allowed_as_fork?

    level_name = Gitlab::VisibilityLevel.level_name(self.visibility_level).downcase
    self.errors.add(:visibility_level, "#{level_name} is not allowed since the fork source project has lower visibility.")
521 522
  end

523
  def to_param
V
Vinnie Okada 已提交
524
    path
525 526
  end

527 528 529 530
  def to_reference(_from_project = nil)
    path_with_namespace
  end

531
  def web_url
532
    Gitlab::Routing.url_helpers.namespace_project_url(self.namespace, self)
533 534
  end

535
  def web_url_without_protocol
536
    web_url.split('://')[1]
537 538
  end

539
  def build_commit_note(commit)
540
    notes.new(commit_id: commit.id, noteable_type: 'Commit')
G
gitlabhq 已提交
541
  end
N
Nihad Abbasov 已提交
542

N
Nihad Abbasov 已提交
543
  def last_activity
544
    last_event
G
gitlabhq 已提交
545 546 547
  end

  def last_activity_date
A
Andrey Kumanyaev 已提交
548
    last_activity_at || updated_at
D
Dmitriy Zaporozhets 已提交
549
  end
550

D
Dmitriy Zaporozhets 已提交
551 552 553
  def project_id
    self.id
  end
R
randx 已提交
554

R
Robert Speicher 已提交
555
  def get_issue(issue_id)
556
    if default_issues_tracker?
R
Robert Speicher 已提交
557
      issues.find_by(iid: issue_id)
558
    else
R
Robert Speicher 已提交
559
      ExternalIssue.new(issue_id, self)
560 561 562
    end
  end

R
Robert Speicher 已提交
563
  def issue_exists?(issue_id)
564
    get_issue(issue_id)
R
Robert Speicher 已提交
565 566
  end

567
  def default_issue_tracker
568
    gitlab_issue_tracker_service || create_gitlab_issue_tracker_service
569 570 571 572 573 574 575 576 577 578
  end

  def issues_tracker
    if external_issue_tracker
      external_issue_tracker
    else
      default_issue_tracker
    end
  end

579
  def default_issues_tracker?
580
    !external_issue_tracker
581 582 583
  end

  def external_issue_tracker
584 585 586 587 588 589 590 591 592 593 594 595 596 597 598
    if has_external_issue_tracker.nil? # To populate existing projects
      cache_has_external_issue_tracker
    end

    if has_external_issue_tracker?
      return @external_issue_tracker if defined?(@external_issue_tracker)

      @external_issue_tracker = services.external_issue_trackers.first
    else
      nil
    end
  end

  def cache_has_external_issue_tracker
    update_column(:has_external_issue_tracker, services.external_issue_trackers.any?)
599 600
  end

A
Andrew8xx8 已提交
601
  def can_have_issues_tracker_id?
602
    self.issues_enabled && !self.default_issues_tracker?
A
Andrew8xx8 已提交
603 604
  end

605
  def build_missing_services
M
Marin Jankovski 已提交
606 607
    services_templates = Service.where(template: true)

608
    Service.available_services_names.each do |service_name|
M
Marin Jankovski 已提交
609
      service = find_service(services, service_name)
610 611

      # If service is available but missing in db
M
Marin Jankovski 已提交
612 613 614 615 616 617
      if service.nil?
        # We should check if template for the service exists
        template = find_service(services_templates, service_name)

        if template.nil?
          # If no template, we should create an instance. Ex `create_gitlab_ci_service`
G
Guilherme Garnier 已提交
618
          self.send :"create_#{service_name}_service"
M
Marin Jankovski 已提交
619 620 621 622
        else
          Service.create_from_template(self.id, template)
        end
      end
623 624 625
    end
  end

V
Valery Sizov 已提交
626 627 628 629 630 631 632 633 634
  def create_labels
    Label.templates.each do |label|
      label = label.dup
      label.template = nil
      label.project_id = self.id
      label.save
    end
  end

M
Marin Jankovski 已提交
635 636 637
  def find_service(list, name)
    list.find { |service| service.to_param == name }
  end
D
Dmitriy Zaporozhets 已提交
638

639
  def ci_services
Y
Yorick Peterse 已提交
640
    services.where(category: :ci)
641 642 643
  end

  def ci_service
Y
Yorick Peterse 已提交
644
    @ci_service ||= ci_services.reorder(nil).find_by(active: true)
645 646
  end

D
Drew Blessing 已提交
647 648 649 650
  def jira_tracker?
    issues_tracker.to_param == 'jira'
  end

651
  def avatar_type
652 653
    unless self.avatar.image?
      self.errors.add :avatar, 'only images allowed'
654 655 656 657
    end
  end

  def avatar_in_git
658
    repository.avatar
659 660
  end

S
sue445 已提交
661 662 663 664
  def avatar_url
    if avatar.present?
      [gitlab_config.url, avatar.url].join
    elsif avatar_in_git
665
      Gitlab::Routing.url_helpers.namespace_project_avatar_url(namespace, self)
S
sue445 已提交
666 667 668
    end
  end

669 670 671 672 673
  # For compatibility with old code
  def code
    path
  end

674
  def items_for(entity)
D
Dmitriy Zaporozhets 已提交
675 676 677 678 679 680 681
    case entity
    when 'issue' then
      issues
    when 'merge_request' then
      merge_requests
    end
  end
682

683
  def send_move_instructions(old_path_with_namespace)
684 685 686
    # New project path needs to be committed to the DB or notification will
    # retrieve stale information
    run_after_commit { NotificationService.new.project_was_moved(self, old_path_with_namespace) }
687
  end
688 689

  def owner
690 691
    if group
      group
692
    else
693
      namespace.try(:owner)
694 695
    end
  end
D
Dmitriy Zaporozhets 已提交
696

697
  def project_member_by_name_or_email(name = nil, email = nil)
G
Gabriel Mazetto 已提交
698
    user = users.find_by('name like ? or email like ?', name, email)
699
    project_members.where(user: user) if user
D
Dmitriy Zaporozhets 已提交
700 701 702
  end

  # Get Team Member record by user id
703
  def project_member_by_id(user_id)
704
    project_members.find_by(user_id: user_id)
D
Dmitriy Zaporozhets 已提交
705 706 707 708 709
  end

  def name_with_namespace
    @name_with_namespace ||= begin
                               if namespace
710
                                 namespace.human_name + ' / ' + name
D
Dmitriy Zaporozhets 已提交
711 712 713 714 715 716 717 718 719 720 721 722 723 724
                               else
                                 name
                               end
                             end
  end

  def path_with_namespace
    if namespace
      namespace.path + '/' + path
    else
      path
    end
  end

725 726
  def execute_hooks(data, hooks_scope = :push_hooks)
    hooks.send(hooks_scope).each do |hook|
727
      hook.async_execute(data, hooks_scope.to_s)
728
    end
D
Dmitriy Zaporozhets 已提交
729 730
  end

731 732 733
  def execute_services(data, hooks_scope = :push_hooks)
    # Call only service hooks that are active for this scope
    services.send(hooks_scope).each do |service|
734
      service.async_execute(data)
D
Dmitriy Zaporozhets 已提交
735 736 737 738
    end
  end

  def update_merge_requests(oldrev, newrev, ref, user)
739 740
    MergeRequests::RefreshService.new(self, user).
      execute(oldrev, newrev, ref)
D
Dmitriy Zaporozhets 已提交
741 742 743
  end

  def valid_repo?
744
    repository.exists?
D
Dmitriy Zaporozhets 已提交
745
  rescue
746
    errors.add(:path, 'Invalid repository path')
D
Dmitriy Zaporozhets 已提交
747 748 749 750
    false
  end

  def empty_repo?
751
    !repository.exists? || !repository.has_visible_content?
D
Dmitriy Zaporozhets 已提交
752 753 754
  end

  def repo
D
Dmitriy Zaporozhets 已提交
755
    repository.raw
D
Dmitriy Zaporozhets 已提交
756 757 758
  end

  def url_to_repo
759
    gitlab_shell.url_to_repo(path_with_namespace)
D
Dmitriy Zaporozhets 已提交
760 761 762 763 764 765 766
  end

  def namespace_dir
    namespace.try(:path) || ''
  end

  def repo_exists?
767
    @repo_exists ||= repository.exists?
D
Dmitriy Zaporozhets 已提交
768 769 770 771 772
  rescue
    @repo_exists = false
  end

  def open_branches
773 774 775
    # We're using a Set here as checking values in a large Set is faster than
    # checking values in a large Array.
    protected_set = Set.new(protected_branch_names)
776

777 778
    repository.branches.reject do |branch|
      protected_set.include?(branch.name)
779 780 781
    end
  end

782 783
  def protected_branch_names
    @protected_branch_names ||= protected_branches.pluck(:name)
D
Dmitriy Zaporozhets 已提交
784 785 786
  end

  def root_ref?(branch)
D
Dmitriy Zaporozhets 已提交
787
    repository.root_ref == branch
D
Dmitriy Zaporozhets 已提交
788 789 790 791 792 793 794
  end

  def ssh_url_to_repo
    url_to_repo
  end

  def http_url_to_repo
D
Douwe Maan 已提交
795
    "#{web_url}.git"
D
Dmitriy Zaporozhets 已提交
796 797 798
  end

  # Check if current branch name is marked as protected in the system
799
  def protected_branch?(branch_name)
800
    protected_branch_names.include?(branch_name)
D
Dmitriy Zaporozhets 已提交
801
  end
802

803
  def developers_can_push_to_protected_branch?(branch_name)
804
    protected_branches.any? { |pb| pb.name == branch_name && pb.developers_can_push }
805 806
  end

807 808 809
  def forked?
    !(forked_project_link.nil? || forked_project_link.forked_from_project.nil?)
  end
D
Dmitriy Zaporozhets 已提交
810

811 812 813 814
  def personal?
    !group
  end

D
Dmitriy Zaporozhets 已提交
815
  def rename_repo
816
    path_was = previous_changes['path'].first
D
Dmitriy Zaporozhets 已提交
817 818 819
    old_path_with_namespace = File.join(namespace_dir, path_was)
    new_path_with_namespace = File.join(namespace_dir, path)

820 821
    expire_caches_before_rename(old_path_with_namespace)

822 823
    if has_container_registry_tags?
      # we currently doesn't support renaming repository if it contains tags in container registry
K
Kamil Trzcinski 已提交
824
      raise Exception.new('Project cannot be renamed, because tags are present in its container registry')
825 826
    end

D
Dmitriy Zaporozhets 已提交
827
    if gitlab_shell.mv_repository(old_path_with_namespace, new_path_with_namespace)
828
      # If repository moved successfully we need to send update instructions to users.
D
Dmitriy Zaporozhets 已提交
829 830 831
      # However we cannot allow rollback since we moved repository
      # So we basically we mute exceptions in next actions
      begin
832
        gitlab_shell.mv_repository("#{old_path_with_namespace}.wiki", "#{new_path_with_namespace}.wiki")
833
        send_move_instructions(old_path_with_namespace)
834
        reset_events_cache
835 836 837 838 839

        @old_path_with_namespace = old_path_with_namespace

        SystemHooksService.new.execute_hooks_for(self, :rename)

840
        @repository = nil
D
Dmitriy Zaporozhets 已提交
841
      rescue
J
Johannes Schleifenbaum 已提交
842
        # Returning false does not rollback after_* transaction but gives
D
Dmitriy Zaporozhets 已提交
843 844 845 846 847 848 849 850
        # us information about failing some of tasks
        false
      end
    else
      # if we cannot move namespace directory we should rollback
      # db changes in order to prevent out of sync between db and fs
      raise Exception.new('repository cannot be renamed')
    end
851 852

    Gitlab::UploadsTransfer.new.rename_project(path_was, path, namespace.path)
D
Dmitriy Zaporozhets 已提交
853
  end
854

855 856 857 858 859 860
  # Expires various caches before a project is renamed.
  def expire_caches_before_rename(old_path)
    repo = Repository.new(old_path, self)
    wiki = Repository.new("#{old_path}.wiki", self)

    if repo.exists?
861
      repo.before_delete
862 863 864
    end

    if wiki.exists?
865
      wiki.before_delete
866 867 868
    end
  end

869 870
  def hook_attrs(backward: true)
    attrs = {
K
Kirill Zaitsev 已提交
871
      name: name,
872
      description: description,
K
Kirilll Zaitsev 已提交
873
      web_url: web_url,
874 875 876
      avatar_url: avatar_url,
      git_ssh_url: ssh_url_to_repo,
      git_http_url: http_url_to_repo,
K
Kirill Zaitsev 已提交
877
      namespace: namespace.name,
878 879 880
      visibility_level: visibility_level,
      path_with_namespace: path_with_namespace,
      default_branch: default_branch,
K
Kirill Zaitsev 已提交
881
    }
882 883 884 885 886 887 888 889 890 891 892 893

    # Backward compatibility
    if backward
      attrs.merge!({
                    homepage: web_url,
                    url: url_to_repo,
                    ssh_url: ssh_url_to_repo,
                    http_url: http_url_to_repo
                  })
    end

    attrs
K
Kirill Zaitsev 已提交
894 895
  end

896 897 898 899 900
  # Reset events cache related to this project
  #
  # Since we do cache @event we need to reset cache in special cases:
  # * when project was moved
  # * when project was renamed
901
  # * when the project avatar changes
902 903 904 905 906 907 908 909 910
  # Events cache stored like  events/23-20130109142513.
  # The cache key includes updated_at timestamp.
  # Thus it will automatically generate a new fragment
  # when the event is updated because the key changes.
  def reset_events_cache
    Event.where(project_id: self.id).
      order('id DESC').limit(100).
      update_all(updated_at: Time.now)
  end
911 912

  def project_member(user)
G
Gabriel Mazetto 已提交
913
    project_members.find_by(user_id: user)
914
  end
915 916 917 918

  def default_branch
    @default_branch ||= repository.root_ref if repository.exists?
  end
919 920 921 922 923

  def reload_default_branch
    @default_branch = nil
    default_branch
  end
924

925 926 927
  def visibility_level_field
    visibility_level
  end
928 929 930 931 932 933 934 935

  def archive!
    update_attribute(:archived, true)
  end

  def unarchive!
    update_attribute(:archived, false)
  end
936

937
  def change_head(branch)
938
    repository.before_change_head
P
P.S.V.R 已提交
939 940 941
    repository.rugged.references.create('HEAD',
                                        "refs/heads/#{branch}",
                                        force: true)
942
    repository.copy_gitattributes(branch)
943 944
    reload_default_branch
  end
945 946 947 948

  def forked_from?(project)
    forked? && project == forked_from_project
  end
949 950 951 952

  def update_repository_size
    update_attribute(:repository_size, repository.size)
  end
953

954 955 956 957
  def update_commit_count
    update_attribute(:commit_count, repository.commit_count)
  end

958
  def forks_count
959
    forks.count
960
  end
R
Robert Schilling 已提交
961 962 963 964

  def find_label(name)
    labels.find_by(name: name)
  end
965 966 967 968

  def origin_merge_requests
    merge_requests.where(source_project_id: self.id)
  end
969 970

  def create_repository
971 972
    # Forked import is handled asynchronously
    unless forked?
973
      if gitlab_shell.add_repository(path_with_namespace)
974
        repository.after_create
975 976
        true
      else
977
        errors.add(:base, 'Failed to create repository via gitlab-shell')
978 979
        false
      end
980 981 982 983 984 985 986 987 988 989
    end
  end

  def repository_exists?
    !!repository.exists?
  end

  def create_wiki
    ProjectWiki.new(self, self.owner).wiki
    true
G
Guilherme Garnier 已提交
990
  rescue ProjectWiki::CouldNotCreateWikiError
991
    errors.add(:base, 'Failed create wiki')
992 993
    false
  end
994

D
Drew Blessing 已提交
995 996 997 998
  def jira_tracker_active?
    jira_tracker? && jira_service.active
  end

999
  def allowed_to_share_with_group?
1000
    !namespace.share_with_group_lock
1001 1002
  end

1003 1004
  def pipeline(sha, ref)
    pipelines.order(id: :desc).find_by(sha: sha, ref: ref)
K
Kamil Trzcinski 已提交
1005 1006
  end

1007 1008
  def ensure_pipeline(sha, ref)
    pipeline(sha, ref) || pipelines.create(sha: sha, ref: ref)
1009
  end
1010

1011 1012 1013
  def enable_ci
    self.builds_enabled = true
  end
M
Marin Jankovski 已提交
1014

K
Kamil Trzcinski 已提交
1015
  def any_runners?(&block)
1016
    if runners.active.any?(&block)
K
Kamil Trzcinski 已提交
1017 1018 1019 1020 1021 1022
      return true
    end

    shared_runners_enabled? && Ci::Runner.shared.active.any?(&block)
  end

1023
  def valid_runners_token?(token)
J
James Lopez 已提交
1024
    self.runners_token && ActiveSupport::SecurityUtils.variable_size_secure_compare(token, self.runners_token)
K
Kamil Trzcinski 已提交
1025 1026
  end

K
Kamil Trzcinski 已提交
1027 1028
  # TODO (ayufan): For now we use runners_token (backward compatibility)
  # In 8.4 every build will have its own individual token valid for time of build
1029
  def valid_build_token?(token)
J
James Lopez 已提交
1030
    self.builds_enabled? && self.runners_token && ActiveSupport::SecurityUtils.variable_size_secure_compare(token, self.runners_token)
K
Kamil Trzcinski 已提交
1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043
  end

  def build_coverage_enabled?
    build_coverage_regex.present?
  end

  def build_timeout_in_minutes
    build_timeout / 60
  end

  def build_timeout_in_minutes=(value)
    self.build_timeout = value.to_i * 60
  end
1044

S
Stan Hu 已提交
1045 1046 1047
  def open_issues_count
    issues.opened.count
  end
1048

D
Douwe Maan 已提交
1049
  def visibility_level_allowed_as_fork?(level = self.visibility_level)
D
Douwe Maan 已提交
1050
    return true unless forked?
D
Douwe Maan 已提交
1051

D
Douwe Maan 已提交
1052 1053 1054 1055 1056 1057
    # self.forked_from_project will be nil before the project is saved, so
    # we need to go through the relation
    original_project = forked_project_link.forked_from_project
    return true unless original_project

    level <= original_project.visibility_level
D
Douwe Maan 已提交
1058
  end
1059

D
Douwe Maan 已提交
1060 1061
  def visibility_level_allowed_by_group?(level = self.visibility_level)
    return true unless group
1062

D
Douwe Maan 已提交
1063
    level <= group.visibility_level
M
Marin Jankovski 已提交
1064
  end
1065

D
Douwe Maan 已提交
1066 1067
  def visibility_level_allowed?(level = self.visibility_level)
    visibility_level_allowed_as_fork?(level) && visibility_level_allowed_by_group?(level)
F
Felipe Artur 已提交
1068 1069
  end

1070 1071 1072
  def runners_token
    ensure_runners_token!
  end
V
Valery Sizov 已提交
1073 1074 1075 1076

  def wiki
    @wiki ||= ProjectWiki.new(self, self.owner)
  end
1077

1078 1079 1080 1081 1082 1083
  def schedule_delete!(user_id, params)
    # Queue this task for after the commit, so once we mark pending_delete it will run
    run_after_commit { ProjectDestroyWorker.perform_async(id, user_id, params) }

    update_attribute(:pending_delete, true)
  end
1084

J
Josh Frye 已提交
1085 1086
  def running_or_pending_build_count(force: false)
    Rails.cache.fetch(['projects', id, 'running_or_pending_build_count'], force: force) do
1087 1088 1089
      builds.running_or_pending.count(:all)
    end
  end
J
James Lopez 已提交
1090 1091

  def mark_import_as_failed(error_message)
1092 1093 1094
    original_errors = errors.dup
    sanitized_message = Gitlab::UrlSanitizer.sanitize(error_message)

J
James Lopez 已提交
1095
    import_fail
1096 1097 1098 1099 1100
    update_column(:import_error, sanitized_message)
  rescue ActiveRecord::ActiveRecordError => e
    Rails.logger.error("Error setting import status to failed: #{e.message}. Original error: #{sanitized_message}")
  ensure
    @errors = original_errors
J
James Lopez 已提交
1101
  end
J
James Lopez 已提交
1102

1103 1104 1105 1106 1107 1108 1109 1110 1111
  def add_export_job(current_user_id:)
    job_id = ProjectExportWorker.perform_async(current_user_id, self.id)

    if job_id
      Rails.logger.info "Export job started for project ID #{self.id} with job ID #{job_id}"
    else
      Rails.logger.error "Export job failed to start for project ID #{self.id}"
    end
  end
J
James Lopez 已提交
1112 1113 1114 1115

  def export_path
    File.join(ImportExport.storage_path, path_with_namespace)
  end
G
gitlabhq 已提交
1116
end