BigW Consortium Gitlab

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

gitlabhq committed
3
class Project < ActiveRecord::Base
4
  include Gitlab::ConfigHelper
5
  include Gitlab::ShellAdapter
6
  include Gitlab::VisibilityLevel
7
  include Gitlab::CurrentSettings
8
  include AccessRequestable
9 10
  include Referable
  include Sortable
11
  include AfterCommitQueue
12
  include CaseSensitivity
13
  include TokenAuthenticatable
Robert Speicher committed
14

15
  extend Gitlab::ConfigHelper
16

Jared Szechy committed
17 18
  UNKNOWN_IMPORT_URL = 'http://unknown.git'

19
  default_value_for :archived, false
20 21 22
  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
23
  default_value_for :builds_enabled, gitlab_config_features.builds
24 25
  default_value_for :wiki_enabled, gitlab_config_features.wiki
  default_value_for :snippets_enabled, gitlab_config_features.snippets
26
  default_value_for :container_registry_enabled, gitlab_config_features.container_registry
27
  default_value_for(:repository_storage) { current_application_settings.repository_storage }
28
  default_value_for(:shared_runners_enabled) { current_application_settings.shared_runners_enabled }
29

30 31 32
  after_create :ensure_dir_exist
  after_save :ensure_dir_exist, if: :namespace_id_changed?

33 34
  # set last_activity_at to the same as created_at
  after_create :set_last_activity_at
35
  def set_last_activity_at
36
    update_column(:last_activity_at, self.created_at)
37 38
  end

39
  # update visibility_level of forks
40 41 42 43 44 45 46 47 48 49 50 51
  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

52
  ActsAsTaggableOn.strict_case_match = true
53
  acts_as_taggable_on :tags
54

55
  attr_accessor :new_default_branch
56
  attr_accessor :old_path_with_namespace
57

58 59
  alias_attribute :title, :name

60
  # Relations
61
  belongs_to :creator, foreign_key: 'creator_id', class_name: 'User'
62
  belongs_to :group, -> { where(type: Group) }, foreign_key: 'namespace_id'
63
  belongs_to :namespace
64

65
  has_one :last_event, -> {order 'events.created_at DESC'}, class_name: 'Event', foreign_key: 'project_id'
66 67 68

  # Project services
  has_many :services
69
  has_one :campfire_service, dependent: :destroy
Kirilll Zaitsev committed
70
  has_one :drone_ci_service, dependent: :destroy
71
  has_one :emails_on_push_service, dependent: :destroy
72
  has_one :builds_email_service, dependent: :destroy
Aorimn committed
73
  has_one :irker_service, dependent: :destroy
74
  has_one :pivotaltracker_service, dependent: :destroy
75
  has_one :hipchat_service, dependent: :destroy
76
  has_one :flowdock_service, dependent: :destroy
77
  has_one :assembla_service, dependent: :destroy
Jeremy committed
78
  has_one :asana_service, dependent: :destroy
79
  has_one :gemnasium_service, dependent: :destroy
80
  has_one :slack_service, dependent: :destroy
81
  has_one :buildkite_service, dependent: :destroy
82
  has_one :bamboo_service, dependent: :destroy
83
  has_one :teamcity_service, dependent: :destroy
84
  has_one :pushover_service, dependent: :destroy
85 86
  has_one :jira_service, dependent: :destroy
  has_one :redmine_service, dependent: :destroy
87
  has_one :custom_issue_tracker_service, dependent: :destroy
88
  has_one :bugzilla_service, dependent: :destroy
89
  has_one :gitlab_issue_tracker_service, dependent: :destroy, inverse_of: :project
90
  has_one :external_wiki_service, dependent: :destroy
91

92 93 94 95 96
  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
97

98
  # Merge Requests for target project should be removed with it
99
  has_many :merge_requests,     dependent: :destroy, foreign_key: 'target_project_id'
100
  # Merge requests from source project should be kept when source project was removed
101
  has_many :fork_merge_requests, foreign_key: 'source_project_id', class_name: MergeRequest
102
  has_many :issues,             dependent: :destroy
103
  has_many :labels,             dependent: :destroy
104 105
  has_many :services,           dependent: :destroy
  has_many :events,             dependent: :destroy
106 107
  has_many :milestones,         dependent: :destroy
  has_many :notes,              dependent: :destroy
108 109
  has_many :snippets,           dependent: :destroy, class_name: 'ProjectSnippet'
  has_many :hooks,              dependent: :destroy, class_name: 'ProjectHook'
110
  has_many :protected_branches, dependent: :destroy
111 112

  has_many :project_members, -> { where(requested_at: nil) }, dependent: :destroy, as: :source, class_name: 'ProjectMember'
113
  alias_method :members, :project_members
114 115 116 117
  has_many :users, through: :project_members

  has_many :requesters, -> { where.not(requested_at: nil) }, dependent: :destroy, as: :source, class_name: 'ProjectMember'

118 119
  has_many :deploy_keys_projects, dependent: :destroy
  has_many :deploy_keys, through: :deploy_keys_projects
Ciro Santilli committed
120 121
  has_many :users_star_projects, dependent: :destroy
  has_many :starrers, through: :users_star_projects, source: :user
122
  has_many :releases, dependent: :destroy
123 124
  has_many :lfs_objects_projects, dependent: :destroy
  has_many :lfs_objects, through: :lfs_objects_projects
125 126
  has_many :project_group_links, dependent: :destroy
  has_many :invited_groups, through: :project_group_links, source: :group
127
  has_many :todos, dependent: :destroy
128
  has_many :notification_settings, dependent: :destroy, as: :source
129

130
  has_one :import_data, dependent: :destroy, class_name: "ProjectImportData"
131

132
  has_many :commit_statuses, dependent: :destroy, class_name: 'CommitStatus', foreign_key: :gl_project_id
133
  has_many :pipelines, dependent: :destroy, class_name: 'Ci::Pipeline', foreign_key: :gl_project_id
134 135 136 137 138
  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
139 140
  has_many :environments, dependent: :destroy
  has_many :deployments, dependent: :destroy
141

142
  accepts_nested_attributes_for :variables, allow_destroy: true
143

144
  delegate :name, to: :owner, allow_nil: true, prefix: true
145
  delegate :members, to: :team, prefix: true
146

Andrey Kumanyaev committed
147
  # Validations
148
  validates :creator, presence: true, on: :create
149
  validates :description, length: { maximum: 2000 }, allow_blank: true
150 151 152 153
  validates :name,
    presence: true,
    length: { within: 0..255 },
    format: { with: Gitlab::Regex.project_name_regex,
154
              message: Gitlab::Regex.project_name_regex_message }
155 156 157
  validates :path,
    presence: true,
    length: { within: 0..255 },
158 159
    format: { with: Gitlab::Regex.project_path_regex,
              message: Gitlab::Regex.project_path_regex_message }
160
  validates :issues_enabled, :merge_requests_enabled,
Andrey Kumanyaev committed
161
            :wiki_enabled, inclusion: { in: [true, false] }
162
  validates :namespace, presence: true
163 164
  validates_uniqueness_of :name, scope: :namespace_id
  validates_uniqueness_of :path, scope: :namespace_id
165
  validates :import_url,
166
    url: { protocols: %w(ssh git http https) },
167
    if: :external_import?
168
  validates :star_count, numericality: { greater_than_or_equal_to: 0 }
169
  validate :check_limit, on: :create
170
  validate :avatar_type,
171
    if: ->(project) { project.avatar.present? && project.avatar_changed? }
172
  validates :avatar, file_size: { maximum: 200.kilobytes.to_i }
173
  validate :visibility_level_allowed_by_group
Douwe Maan committed
174
  validate :visibility_level_allowed_as_fork
175
  validate :check_wiki_path_conflict
176 177 178
  validates :repository_storage,
    presence: true,
    inclusion: { in: ->(_object) { Gitlab.config.repositories.storages.keys } }
179

180
  add_authentication_token_field :runners_token
181
  before_save :ensure_runners_token
182

183
  mount_uploader :avatar, AvatarUploader
Andrey Kumanyaev committed
184

185
  # Scopes
186 187
  default_scope { where(pending_delete: false) }

188
  scope :sorted_by_activity, -> { reorder(last_activity_at: :desc) }
189 190
  scope :sorted_by_stars, -> { reorder('projects.star_count DESC') }

191
  scope :in_namespace, ->(namespace_ids) { where(namespace_id: namespace_ids) }
192
  scope :personal, ->(user) { where(namespace_id: user.namespace_id) }
193
  scope :joined, ->(user) { where('namespace_id != ?', user.namespace_id) }
194
  scope :visible_to_user, ->(user) { where(id: user.authorized_projects.select(:id).reorder(nil)) }
195
  scope :non_archived, -> { where(archived: false) }
196
  scope :for_milestones, ->(ids) { joins(:milestones).where('milestones.id' => ids).distinct }
197 198 199 200
  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) }
201

202 203
  state_machine :import_status, initial: :none do
    event :import_start do
204
      transition [:none, :finished] => :started
205 206 207
    end

    event :import_finish do
208
      transition started: :finished
209 210 211
    end

    event :import_fail do
212
      transition started: :failed
213 214 215
    end

    event :import_retry do
216
      transition failed: :started
217 218 219 220
    end

    state :started
    state :finished
221 222
    state :failed

223
    after_transition any => :finished, do: :reset_cache_and_import_attrs
224 225
  end

Andrey Kumanyaev committed
226
  class << self
227 228 229 230 231 232 233
    # 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.
234
    def search(query)
235
      ptable  = arel_table
236 237 238 239 240 241 242 243 244
      ntable  = Namespace.arel_table
      pattern = "%#{query}%"

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

245 246 247 248 249 250
      # 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.
251
      namespaces = select(:id).
252
        except(:includes).
253 254 255 256 257 258
        joins(:namespace).
        where(ntable[:name].matches(pattern))

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

      where("projects.id IN (#{union.to_sql})")
Andrey Kumanyaev committed
259
    end
260

261
    def search_by_visibility(level)
Josh Frye committed
262
      where(visibility_level: Gitlab::VisibilityLevel.const_get(level.upcase))
263 264
    end

265
    def search_by_title(query)
266 267 268 269
      pattern = "%#{query}%"
      table   = Project.arel_table

      non_archived.where(table[:name].matches(pattern))
270 271
    end

272 273 274 275 276 277
    # 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)
278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294
      namespace_path, project_path = path.split('/', 2)

      return unless namespace_path && project_path

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

      # On MySQL we want to ensure the ORDER BY uses a case-sensitive match so
      # any literal matches come first, for this we have to use "BINARY".
      # Without this there's still no guarantee in what order MySQL will return
      # rows.
      binary = Gitlab::Database.mysql? ? 'BINARY' : ''

      order_sql = "(CASE WHEN #{binary} namespaces.path = #{namespace_path} " \
        "AND #{binary} projects.path = #{project_path} THEN 0 ELSE 1 END)"

      where_paths_in([path]).reorder(order_sql).take
295
    end
296

297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320
    # 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?
321

322 323
      paths.each do |path|
        namespace_path, project_path = path.split('/', 2)
324

325 326 327 328
        next unless namespace_path && project_path

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

330 331
        where = "(namespaces.path = #{namespace_path}
          AND projects.path = #{project_path})"
332

333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350
        if cast_lower
          where = "(
            #{where}
            OR (
              LOWER(namespaces.path) = LOWER(#{namespace_path})
              AND LOWER(projects.path) = LOWER(#{project_path})
            )
          )"
        end

        wheres << where
      end

      if wheres.empty?
        none
      else
        joins(:namespace).where(wheres.join(' OR '))
      end
351
    end
352

353 354 355
    def visibility_levels
      Gitlab::VisibilityLevel.options
    end
356 357

    def sort(method)
358 359 360 361
      if method == 'repository_size_desc'
        reorder(repository_size: :desc, id: :desc)
      else
        order_by(method)
362 363
      end
    end
364 365 366 367 368

    def reference_pattern
      name_pattern = Gitlab::Regex::NAMESPACE_REGEX_STR
      %r{(?<project>#{name_pattern}/#{name_pattern})}
    end
369 370 371 372 373 374 375 376 377 378 379 380 381 382

    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
James Lopez committed
383 384

    # Deletes gitlab project export files older than 24 hours
385 386
    def remove_gitlab_exports!
      Gitlab::Popen.popen(%W(find #{Gitlab::ImportExport.storage_path} -not -path #{Gitlab::ImportExport.storage_path} -mmin +1440 -delete))
James Lopez committed
387
    end
388 389
  end

390 391 392 393
  def repository_storage_path
    Gitlab.config.repositories.storages[repository_storage]
  end

394
  def team
395
    @team ||= ProjectTeam.new(self)
396 397 398
  end

  def repository
399
    @repository ||= Repository.new(path_with_namespace, self)
400 401
  end

402 403 404 405
  def container_registry_path_with_namespace
    path_with_namespace.downcase
  end

406
  def container_registry_repository
Kamil Trzcinski committed
407 408
    return unless Gitlab.config.registry.enabled

409
    @container_registry_repository ||= begin
410
      token = Auth::ContainerRegistryAuthenticationService.full_access_token(container_registry_path_with_namespace)
411
      url = Gitlab.config.registry.api_url
412 413
      host_port = Gitlab.config.registry.host_port
      registry = ContainerRegistry::Registry.new(url, token: token, path: host_port)
414
      registry.repository(container_registry_path_with_namespace)
415
    end
416 417
  end

418
  def container_registry_repository_url
Kamil Trzcinski committed
419
    if Gitlab.config.registry.enabled
420
      "#{Gitlab.config.registry.host_port}/#{container_registry_path_with_namespace}"
421
    end
422 423
  end

424
  def has_container_registry_tags?
Kamil Trzcinski committed
425 426 427
    return unless container_registry_repository

    container_registry_repository.tags.any?
428 429
  end

430
  def commit(id = 'HEAD')
431
    repository.commit(id)
432 433
  end

434
  def merge_base_commit(first_commit_id, second_commit_id)
Douwe Maan committed
435 436
    sha = repository.merge_base(first_commit_id, second_commit_id)
    repository.commit(sha) if sha
437 438
  end

439
  def saved?
440
    id && persisted?
441 442
  end

443
  def add_import_job
444
    if forked?
445
      job_id = RepositoryForkWorker.perform_async(self.id, forked_from_project.path_with_namespace, self.namespace.path)
446
    else
447 448 449 450 451 452 453
      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}"
454
    end
455 456
  end

457
  def reset_cache_and_import_attrs
458 459 460 461
    update(import_error: nil)

    ProjectCacheWorker.perform_async(self.id)

462
    self.import_data.destroy if self.import_data
463 464
  end

465
  def import_url=(value)
466
    import_url = Gitlab::UrlSanitizer.new(value)
467
    create_or_update_import_data(credentials: import_url.credentials)
James Lopez committed
468
    super(import_url.sanitized_url)
469 470 471
  end

  def import_url
472
    if import_data && super
473
      import_url = Gitlab::UrlSanitizer.new(super, credentials: import_data.credentials)
James Lopez committed
474 475 476
      import_url.full_url
    else
      super
477 478
    end
  end
479

480
  def create_or_update_import_data(data: nil, credentials: nil)
James Lopez committed
481
    project_import_data = import_data || build_import_data
482 483 484 485
    if data
      project_import_data.data ||= {}
      project_import_data.data = project_import_data.data.merge(data)
    end
486 487 488 489
    if credentials
      project_import_data.credentials ||= {}
      project_import_data.credentials = project_import_data.credentials.merge(credentials)
    end
490 491

    project_import_data.save
492
  end
493

494
  def import?
495
    external_import? || forked? || gitlab_project_import?
496 497
  end

498 499 500 501
  def no_import?
    import_status == 'none'
  end

502
  def external_import?
503 504 505
    import_url.present?
  end

506
  def imported?
507 508 509 510 511 512 513 514 515 516 517 518 519
    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'
520 521
  end

522
  def safe_import_url
523
    Gitlab::UrlSanitizer.new(import_url).masked_url
524 525
  end

526 527 528 529
  def gitlab_project_import?
    import_type == 'gitlab_project'
  end

530
  def check_limit
531
    unless creator.can_create_project? or namespace.kind == 'group'
532 533 534
      projects_limit = creator.projects_limit

      if projects_limit == 0
535
        self.errors.add(:limit_reached, "Personal project creation is not allowed. Please contact your administrator with questions")
536
      else
537
        self.errors.add(:limit_reached, "Your project limit is #{projects_limit} projects! Please contact your administrator to increase it")
538
      end
539 540
    end
  rescue
541
    self.errors.add(:base, "Can't check your ability to create project")
gitlabhq committed
542 543
  end

544 545 546 547 548 549 550 551 552 553 554 555 556
  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.")
557 558
  end

559 560 561 562 563 564 565 566 567 568
  def check_wiki_path_conflict
    return if path.blank?

    path_to_check = path.ends_with?('.wiki') ? path.chomp('.wiki') : "#{path}.wiki"

    if Project.where(namespace_id: namespace_id, path: path_to_check).exists?
      errors.add(:name, 'has already been taken')
    end
  end

569
  def to_param
Vinnie Okada committed
570
    path
571 572
  end

573 574 575 576
  def to_reference(_from_project = nil)
    path_with_namespace
  end

577
  def web_url
578
    Gitlab::Routing.url_helpers.namespace_project_url(self.namespace, self)
579 580
  end

581
  def web_url_without_protocol
582
    web_url.split('://')[1]
583 584
  end

585
  def build_commit_note(commit)
586
    notes.new(commit_id: commit.id, noteable_type: 'Commit')
gitlabhq committed
587
  end
Nihad Abbasov committed
588

589
  def last_activity
590
    last_event
gitlabhq committed
591 592 593
  end

  def last_activity_date
Andrey Kumanyaev committed
594
    last_activity_at || updated_at
Dmitriy Zaporozhets committed
595
  end
596

597 598 599
  def project_id
    self.id
  end
randx committed
600

601
  def get_issue(issue_id)
602
    if default_issues_tracker?
603
      issues.find_by(iid: issue_id)
604
    else
605
      ExternalIssue.new(issue_id, self)
606 607 608
    end
  end

609
  def issue_exists?(issue_id)
610
    get_issue(issue_id)
611 612
  end

613
  def default_issue_tracker
614
    gitlab_issue_tracker_service || create_gitlab_issue_tracker_service
615 616 617 618 619 620 621 622 623 624
  end

  def issues_tracker
    if external_issue_tracker
      external_issue_tracker
    else
      default_issue_tracker
    end
  end

625
  def default_issues_tracker?
626
    !external_issue_tracker
627 628 629
  end

  def external_issue_tracker
630 631 632 633 634 635 636 637 638 639 640 641 642 643 644
    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?)
645 646
  end

647
  def build_missing_services
648 649
    services_templates = Service.where(template: true)

650
    Service.available_services_names.each do |service_name|
651
      service = find_service(services, service_name)
652 653

      # If service is available but missing in db
654 655 656 657 658 659
      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`
660
          self.send :"create_#{service_name}_service"
661 662 663 664
        else
          Service.create_from_template(self.id, template)
        end
      end
665 666 667
    end
  end

668 669 670 671 672 673 674 675 676
  def create_labels
    Label.templates.each do |label|
      label = label.dup
      label.template = nil
      label.project_id = self.id
      label.save
    end
  end

677 678 679
  def find_service(list, name)
    list.find { |service| service.to_param == name }
  end
680

681
  def ci_services
682
    services.where(category: :ci)
683 684 685
  end

  def ci_service
686
    @ci_service ||= ci_services.reorder(nil).find_by(active: true)
687 688
  end

Drew Blessing committed
689 690 691 692
  def jira_tracker?
    issues_tracker.to_param == 'jira'
  end

693
  def avatar_type
694 695
    unless self.avatar.image?
      self.errors.add :avatar, 'only images allowed'
696 697 698 699
    end
  end

  def avatar_in_git
700
    repository.avatar
701 702
  end

703 704 705 706
  def avatar_url
    if avatar.present?
      [gitlab_config.url, avatar.url].join
    elsif avatar_in_git
707
      Gitlab::Routing.url_helpers.namespace_project_avatar_url(namespace, self)
708 709 710
    end
  end

711 712 713 714 715
  # For compatibility with old code
  def code
    path
  end

716
  def items_for(entity)
717 718 719 720 721 722 723
    case entity
    when 'issue' then
      issues
    when 'merge_request' then
      merge_requests
    end
  end
724

725
  def send_move_instructions(old_path_with_namespace)
726 727 728
    # 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) }
729
  end
730 731

  def owner
732 733
    if group
      group
734
    else
735
      namespace.try(:owner)
736 737
    end
  end
Dmitriy Zaporozhets committed
738 739 740 741

  def name_with_namespace
    @name_with_namespace ||= begin
                               if namespace
742
                                 namespace.human_name + ' / ' + name
Dmitriy Zaporozhets committed
743 744 745 746 747
                               else
                                 name
                               end
                             end
  end
748
  alias_method :human_name, :name_with_namespace
Dmitriy Zaporozhets committed
749 750 751 752 753 754 755 756 757

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

758 759
  def execute_hooks(data, hooks_scope = :push_hooks)
    hooks.send(hooks_scope).each do |hook|
760
      hook.async_execute(data, hooks_scope.to_s)
761
    end
Dmitriy Zaporozhets committed
762 763
  end

764 765 766
  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|
767
      service.async_execute(data)
Dmitriy Zaporozhets committed
768 769 770 771
    end
  end

  def update_merge_requests(oldrev, newrev, ref, user)
772 773
    MergeRequests::RefreshService.new(self, user).
      execute(oldrev, newrev, ref)
Dmitriy Zaporozhets committed
774 775 776
  end

  def valid_repo?
777
    repository.exists?
Dmitriy Zaporozhets committed
778
  rescue
779
    errors.add(:path, 'Invalid repository path')
Dmitriy Zaporozhets committed
780 781 782 783
    false
  end

  def empty_repo?
784
    !repository.exists? || !repository.has_visible_content?
Dmitriy Zaporozhets committed
785 786 787
  end

  def repo
788
    repository.raw
Dmitriy Zaporozhets committed
789 790 791
  end

  def url_to_repo
792
    gitlab_shell.url_to_repo(path_with_namespace)
Dmitriy Zaporozhets committed
793 794 795 796 797 798 799
  end

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

  def repo_exists?
800
    @repo_exists ||= repository.exists?
Dmitriy Zaporozhets committed
801 802 803 804 805
  rescue
    @repo_exists = false
  end

  def open_branches
806
    repository.branches.reject { |branch| self.protected_branch?(branch.name) }
Dmitriy Zaporozhets committed
807 808 809
  end

  def root_ref?(branch)
810
    repository.root_ref == branch
Dmitriy Zaporozhets committed
811 812 813 814 815 816 817
  end

  def ssh_url_to_repo
    url_to_repo
  end

  def http_url_to_repo
Douwe Maan committed
818
    "#{web_url}.git"
Dmitriy Zaporozhets committed
819 820 821
  end

  # Check if current branch name is marked as protected in the system
822
  def protected_branch?(branch_name)
823 824
    @protected_branches ||= self.protected_branches.to_a
    ProtectedBranch.matching(branch_name, protected_branches: @protected_branches).present?
Dmitriy Zaporozhets committed
825
  end
826

827
  def developers_can_push_to_protected_branch?(branch_name)
828
    protected_branches.matching(branch_name).any?(&:developers_can_push)
829 830
  end

831 832 833
  def forked?
    !(forked_project_link.nil? || forked_project_link.forked_from_project.nil?)
  end
834

835 836 837 838
  def personal?
    !group
  end

839
  def rename_repo
840
    path_was = previous_changes['path'].first
841 842 843
    old_path_with_namespace = File.join(namespace_dir, path_was)
    new_path_with_namespace = File.join(namespace_dir, path)

844 845
    expire_caches_before_rename(old_path_with_namespace)

846 847
    if has_container_registry_tags?
      # we currently doesn't support renaming repository if it contains tags in container registry
Kamil Trzcinski committed
848
      raise Exception.new('Project cannot be renamed, because tags are present in its container registry')
849 850
    end

851
    if gitlab_shell.mv_repository(repository_storage_path, old_path_with_namespace, new_path_with_namespace)
852
      # If repository moved successfully we need to send update instructions to users.
853 854 855
      # However we cannot allow rollback since we moved repository
      # So we basically we mute exceptions in next actions
      begin
856
        gitlab_shell.mv_repository(repository_storage_path, "#{old_path_with_namespace}.wiki", "#{new_path_with_namespace}.wiki")
857
        send_move_instructions(old_path_with_namespace)
858
        reset_events_cache
859 860 861 862 863

        @old_path_with_namespace = old_path_with_namespace

        SystemHooksService.new.execute_hooks_for(self, :rename)

864
        @repository = nil
865
      rescue
Johannes Schleifenbaum committed
866
        # Returning false does not rollback after_* transaction but gives
867 868 869 870 871 872 873 874
        # 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
875 876

    Gitlab::UploadsTransfer.new.rename_project(path_was, path, namespace.path)
877
  end
878

879 880 881 882 883 884
  # 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?
885
      repo.before_delete
886 887 888
    end

    if wiki.exists?
889
      wiki.before_delete
890 891 892
    end
  end

893 894
  def hook_attrs(backward: true)
    attrs = {
Kirill Zaitsev committed
895
      name: name,
896
      description: description,
Kirilll Zaitsev committed
897
      web_url: web_url,
898 899 900
      avatar_url: avatar_url,
      git_ssh_url: ssh_url_to_repo,
      git_http_url: http_url_to_repo,
Kirill Zaitsev committed
901
      namespace: namespace.name,
902 903 904
      visibility_level: visibility_level,
      path_with_namespace: path_with_namespace,
      default_branch: default_branch,
Kirill Zaitsev committed
905
    }
906 907 908 909 910 911 912 913 914 915 916 917

    # 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
Kirill Zaitsev committed
918 919
  end

920 921 922 923 924
  # 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
925
  # * when the project avatar changes
926 927 928 929 930 931 932 933 934
  # 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
935 936

  def project_member(user)
937
    project_members.find_by(user_id: user)
938
  end
939 940 941 942

  def default_branch
    @default_branch ||= repository.root_ref if repository.exists?
  end
943 944 945 946 947

  def reload_default_branch
    @default_branch = nil
    default_branch
  end
948

949 950 951
  def visibility_level_field
    visibility_level
  end
952 953 954 955 956 957 958 959

  def archive!
    update_attribute(:archived, true)
  end

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

961
  def change_head(branch)
962
    repository.before_change_head
P.S.V.R committed
963 964 965
    repository.rugged.references.create('HEAD',
                                        "refs/heads/#{branch}",
                                        force: true)
966
    repository.copy_gitattributes(branch)
967 968
    reload_default_branch
  end
969 970 971 972

  def forked_from?(project)
    forked? && project == forked_from_project
  end
973 974 975 976

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

978 979 980 981
  def update_commit_count
    update_attribute(:commit_count, repository.commit_count)
  end

982
  def forks_count
983
    forks.count
984
  end
985 986 987 988

  def find_label(name)
    labels.find_by(name: name)
  end
989 990 991 992

  def origin_merge_requests
    merge_requests.where(source_project_id: self.id)
  end
993 994

  def create_repository
995 996
    # Forked import is handled asynchronously
    unless forked?
997
      if gitlab_shell.add_repository(repository_storage_path, path_with_namespace)
998
        repository.after_create
999 1000
        true
      else
1001
        errors.add(:base, 'Failed to create repository via gitlab-shell')
1002 1003
        false
      end
1004 1005 1006 1007 1008 1009 1010 1011 1012 1013
    end
  end

  def repository_exists?
    !!repository.exists?
  end

  def create_wiki
    ProjectWiki.new(self, self.owner).wiki
    true
1014
  rescue ProjectWiki::CouldNotCreateWikiError
1015
    errors.add(:base, 'Failed create wiki')
1016 1017
    false
  end
1018

Drew Blessing committed
1019 1020 1021 1022
  def jira_tracker_active?
    jira_tracker? && jira_service.active
  end

1023
  def allowed_to_share_with_group?
1024
    !namespace.share_with_group_lock
1025 1026
  end

1027 1028
  def pipeline(sha, ref)
    pipelines.order(id: :desc).find_by(sha: sha, ref: ref)
1029 1030
  end

1031 1032
  def ensure_pipeline(sha, ref)
    pipeline(sha, ref) || pipelines.create(sha: sha, ref: ref)
1033
  end
1034

1035 1036 1037
  def enable_ci
    self.builds_enabled = true
  end
1038

1039
  def any_runners?(&block)
1040
    if runners.active.any?(&block)
1041 1042 1043 1044 1045 1046
      return true
    end

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

1047
  def valid_runners_token?(token)
1048
    self.runners_token && ActiveSupport::SecurityUtils.variable_size_secure_compare(token, self.runners_token)
Kamil Trzcinski committed
1049 1050
  end

Kamil Trzcinski committed
1051 1052
  # 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
1053
  def valid_build_token?(token)
1054
    self.builds_enabled? && self.runners_token && ActiveSupport::SecurityUtils.variable_size_secure_compare(token, self.runners_token)
1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067
  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
1068

1069 1070 1071
  def open_issues_count
    issues.opened.count
  end
1072

1073
  def visibility_level_allowed_as_fork?(level = self.visibility_level)
Douwe Maan committed
1074
    return true unless forked?
1075

Douwe Maan committed
1076 1077 1078 1079 1080 1081
    # 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
1082
  end
1083

1084 1085
  def visibility_level_allowed_by_group?(level = self.visibility_level)
    return true unless group
1086

1087
    level <= group.visibility_level
1088
  end
1089

1090 1091
  def visibility_level_allowed?(level = self.visibility_level)
    visibility_level_allowed_as_fork?(level) && visibility_level_allowed_by_group?(level)
Felipe Artur committed
1092 1093
  end

1094 1095 1096
  def runners_token
    ensure_runners_token!
  end
1097 1098 1099 1100

  def wiki
    @wiki ||= ProjectWiki.new(self, self.owner)
  end
1101 1102 1103 1104 1105 1106 1107

  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
1108

Josh Frye committed
1109 1110
  def running_or_pending_build_count(force: false)
    Rails.cache.fetch(['projects', id, 'running_or_pending_build_count'], force: force) do
1111 1112 1113
      builds.running_or_pending.count(:all)
    end
  end
1114 1115

  def mark_import_as_failed(error_message)
1116 1117 1118
    original_errors = errors.dup
    sanitized_message = Gitlab::UrlSanitizer.sanitize(error_message)

1119
    import_fail
1120 1121 1122 1123 1124
    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
1125
  end
James Lopez committed
1126

1127 1128
  def add_export_job(current_user:)
    job_id = ProjectExportWorker.perform_async(current_user.id, self.id)
1129 1130 1131 1132 1133 1134 1135

    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
James Lopez committed
1136 1137

  def export_path
1138
    File.join(Gitlab::ImportExport.storage_path, path_with_namespace)
James Lopez committed
1139
  end
1140 1141 1142 1143 1144 1145 1146 1147 1148

  def export_project_path
    Dir.glob("#{export_path}/*export.tar.gz").max_by { |f| File.ctime(f) }
  end

  def remove_exports
    _, status = Gitlab::Popen.popen(%W(find #{export_path} -not -path #{export_path} -delete))
    status.zero?
  end
1149 1150 1151 1152

  def ensure_dir_exist
    gitlab_shell.add_namespace(repository_storage_path, namespace.path)
  end
gitlabhq committed
1153
end