BigW Consortium Gitlab

project.rb 30.2 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 9
  include Referable
  include Sortable
10
  include AfterCommitQueue
11
  include CaseSensitivity
12
  include TokenAuthenticatable
Robert Speicher committed
13

14
  extend Gitlab::ConfigHelper
15

Jared Szechy committed
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
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
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
64
  has_one :campfire_service, dependent: :destroy
Kirilll Zaitsev committed
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
Aorimn committed
68
  has_one :irker_service, dependent: :destroy
69
  has_one :pivotaltracker_service, dependent: :destroy
70
  has_one :hipchat_service, dependent: :destroy
71
  has_one :flowdock_service, dependent: :destroy
72
  has_one :assembla_service, dependent: :destroy
Jeremy committed
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
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
82
  has_one :custom_issue_tracker_service, dependent: :destroy
83
  has_one :gitlab_issue_tracker_service, dependent: :destroy
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
Ciro Santilli committed
109 110
  has_many :users_star_projects, dependent: :destroy
  has_many :starrers, through: :users_star_projects, source: :user
111
  has_many :releases, dependent: :destroy
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"
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
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

Andrey Kumanyaev committed
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,
141
              message: Gitlab::Regex.project_name_regex_message }
142 143 144
  validates :path,
    presence: true,
    length: { within: 0..255 },
145 146
    format: { with: Gitlab::Regex.project_path_regex,
              message: Gitlab::Regex.project_path_regex_message }
147
  validates :issues_enabled, :merge_requests_enabled,
Andrey Kumanyaev committed
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,
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 }
161
  validate :visibility_level_allowed_by_group
Douwe Maan committed
162
  validate :visibility_level_allowed_as_fork
163

164
  add_authentication_token_field :runners_token
165
  before_save :ensure_runners_token
166

167
  mount_uploader :avatar, AvatarUploader
Andrey Kumanyaev committed
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) }
176
  scope :personal, ->(user) { where(namespace_id: user.namespace_id) }
177
  scope :joined, ->(user) { where('namespace_id != ?', user.namespace_id) }
178
  scope :visible_to_user, ->(user) { where(id: user.authorized_projects.select(:id).reorder(nil)) }
179
  scope :non_archived, -> { where(archived: false) }
180
  scope :for_milestones, ->(ids) { joins(:milestones).where('milestones.id' => ids).distinct }
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

Andrey Kumanyaev committed
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})")
Andrey Kumanyaev committed
243
    end
244

245
    def search_by_visibility(level)
Josh Frye committed
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

256
    def find_with_namespace(id)
257
      namespace_path, project_path = id.split('/', 2)
258 259 260 261 262 263

      return nil if !namespace_path || !project_path

      # Use of unscoped ensures we're not secretly adding any ORDER BYs, which
      # have a negative impact on performance (and aren't needed for this
      # query).
264
      projects = unscoped.
265
        joins(:namespace).
266 267
        iwhere('namespaces.path' => namespace_path)

268
      projects.find_by('projects.path' => project_path) ||
269
        projects.iwhere('projects.path' => project_path).take
270
    end
271

272 273 274
    def visibility_levels
      Gitlab::VisibilityLevel.options
    end
275 276

    def sort(method)
277 278 279 280
      if method == 'repository_size_desc'
        reorder(repository_size: :desc, id: :desc)
      else
        order_by(method)
281 282
      end
    end
283 284 285 286 287

    def reference_pattern
      name_pattern = Gitlab::Regex::NAMESPACE_REGEX_STR
      %r{(?<project>#{name_pattern}/#{name_pattern})}
    end
288 289 290 291 292 293 294 295 296 297 298 299 300 301

    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
302 303
  end

304
  def team
305
    @team ||= ProjectTeam.new(self)
306 307 308
  end

  def repository
309
    @repository ||= Repository.new(path_with_namespace, self)
310 311
  end

312 313 314 315
  def container_registry_path_with_namespace
    path_with_namespace.downcase
  end

316
  def container_registry_repository
Kamil Trzcinski committed
317 318
    return unless Gitlab.config.registry.enabled

319
    @container_registry_repository ||= begin
320
      token = Auth::ContainerRegistryAuthenticationService.full_access_token(container_registry_path_with_namespace)
321
      url = Gitlab.config.registry.api_url
322 323
      host_port = Gitlab.config.registry.host_port
      registry = ContainerRegistry::Registry.new(url, token: token, path: host_port)
324
      registry.repository(container_registry_path_with_namespace)
325
    end
326 327
  end

328
  def container_registry_repository_url
Kamil Trzcinski committed
329
    if Gitlab.config.registry.enabled
330
      "#{Gitlab.config.registry.host_port}/#{container_registry_path_with_namespace}"
331
    end
332 333
  end

334
  def has_container_registry_tags?
Kamil Trzcinski committed
335 336 337
    return unless container_registry_repository

    container_registry_repository.tags.any?
338 339
  end

340
  def commit(id = 'HEAD')
341
    repository.commit(id)
342 343
  end

344
  def merge_base_commit(first_commit_id, second_commit_id)
Douwe Maan committed
345 346
    sha = repository.merge_base(first_commit_id, second_commit_id)
    repository.commit(sha) if sha
347 348
  end

349
  def saved?
350
    id && persisted?
351 352
  end

353
  def add_import_job
354
    if forked?
355
      job_id = RepositoryForkWorker.perform_async(self.id, forked_from_project.path_with_namespace, self.namespace.path)
356
    else
357 358 359 360 361 362 363
      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}"
364
    end
365 366
  end

367
  def reset_cache_and_import_attrs
368 369 370 371
    update(import_error: nil)

    ProjectCacheWorker.perform_async(self.id)

372
    self.import_data.destroy if self.import_data
373 374
  end

375
  def import_url=(value)
376
    import_url = Gitlab::UrlSanitizer.new(value)
377
    create_or_update_import_data(credentials: import_url.credentials)
James Lopez committed
378
    super(import_url.sanitized_url)
379 380 381
  end

  def import_url
382
    if import_data && super
383
      import_url = Gitlab::UrlSanitizer.new(super, credentials: import_data.credentials)
James Lopez committed
384 385 386
      import_url.full_url
    else
      super
387 388
    end
  end
389

390
  def create_or_update_import_data(data: nil, credentials: nil)
James Lopez committed
391
    project_import_data = import_data || build_import_data
392 393 394 395
    if data
      project_import_data.data ||= {}
      project_import_data.data = project_import_data.data.merge(data)
    end
396 397 398 399
    if credentials
      project_import_data.credentials ||= {}
      project_import_data.credentials = project_import_data.credentials.merge(credentials)
    end
400 401

    project_import_data.save
402
  end
403

404
  def import?
405 406 407
    external_import? || forked?
  end

408 409 410 411
  def no_import?
    import_status == 'none'
  end

412
  def external_import?
413 414 415
    import_url.present?
  end

416
  def imported?
417 418 419 420 421 422 423 424 425 426 427 428 429
    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'
430 431
  end

432
  def safe_import_url
433
    Gitlab::UrlSanitizer.new(import_url).masked_url
434 435
  end

436
  def check_limit
437
    unless creator.can_create_project? or namespace.kind == 'group'
438 439 440
      projects_limit = creator.projects_limit

      if projects_limit == 0
441
        self.errors.add(:limit_reached, "Personal project creation is not allowed. Please contact your administrator with questions")
442
      else
443
        self.errors.add(:limit_reached, "Your project limit is #{projects_limit} projects! Please contact your administrator to increase it")
444
      end
445 446
    end
  rescue
447
    self.errors.add(:base, "Can't check your ability to create project")
gitlabhq committed
448 449
  end

450 451 452 453 454 455 456 457 458 459 460 461 462
  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.")
463 464
  end

465
  def to_param
Vinnie Okada committed
466
    path
467 468
  end

469 470 471 472
  def to_reference(_from_project = nil)
    path_with_namespace
  end

473
  def web_url
474
    Gitlab::Routing.url_helpers.namespace_project_url(self.namespace, self)
475 476
  end

477
  def web_url_without_protocol
478
    web_url.split('://')[1]
479 480
  end

481
  def build_commit_note(commit)
482
    notes.new(commit_id: commit.id, noteable_type: 'Commit')
gitlabhq committed
483
  end
Nihad Abbasov committed
484

485
  def last_activity
486
    last_event
gitlabhq committed
487 488 489
  end

  def last_activity_date
Andrey Kumanyaev committed
490
    last_activity_at || updated_at
Dmitriy Zaporozhets committed
491
  end
492

493 494 495
  def project_id
    self.id
  end
randx committed
496

497
  def get_issue(issue_id)
498
    if default_issues_tracker?
499
      issues.find_by(iid: issue_id)
500
    else
501
      ExternalIssue.new(issue_id, self)
502 503 504
    end
  end

505
  def issue_exists?(issue_id)
506
    get_issue(issue_id)
507 508
  end

509
  def default_issue_tracker
510
    gitlab_issue_tracker_service || create_gitlab_issue_tracker_service
511 512 513 514 515 516 517 518 519 520
  end

  def issues_tracker
    if external_issue_tracker
      external_issue_tracker
    else
      default_issue_tracker
    end
  end

521
  def default_issues_tracker?
522
    !external_issue_tracker
523 524 525
  end

  def external_issue_tracker
526 527 528 529 530 531 532 533 534 535 536 537 538 539 540
    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?)
541 542
  end

Andrew8xx8 committed
543
  def can_have_issues_tracker_id?
544
    self.issues_enabled && !self.default_issues_tracker?
Andrew8xx8 committed
545 546
  end

547
  def build_missing_services
548 549
    services_templates = Service.where(template: true)

550
    Service.available_services_names.each do |service_name|
551
      service = find_service(services, service_name)
552 553

      # If service is available but missing in db
554 555 556 557 558 559
      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`
560
          self.send :"create_#{service_name}_service"
561 562 563 564
        else
          Service.create_from_template(self.id, template)
        end
      end
565 566 567
    end
  end

568 569 570 571 572 573 574 575 576
  def create_labels
    Label.templates.each do |label|
      label = label.dup
      label.template = nil
      label.project_id = self.id
      label.save
    end
  end

577 578 579
  def find_service(list, name)
    list.find { |service| service.to_param == name }
  end
580

581
  def ci_services
582
    services.where(category: :ci)
583 584 585
  end

  def ci_service
586
    @ci_service ||= ci_services.reorder(nil).find_by(active: true)
587 588
  end

Drew Blessing committed
589 590 591 592
  def jira_tracker?
    issues_tracker.to_param == 'jira'
  end

593
  def avatar_type
594 595
    unless self.avatar.image?
      self.errors.add :avatar, 'only images allowed'
596 597 598 599
    end
  end

  def avatar_in_git
600
    repository.avatar
601 602
  end

603 604 605 606
  def avatar_url
    if avatar.present?
      [gitlab_config.url, avatar.url].join
    elsif avatar_in_git
607
      Gitlab::Routing.url_helpers.namespace_project_avatar_url(namespace, self)
608 609 610
    end
  end

611 612 613 614 615
  # For compatibility with old code
  def code
    path
  end

616
  def items_for(entity)
617 618 619 620 621 622 623
    case entity
    when 'issue' then
      issues
    when 'merge_request' then
      merge_requests
    end
  end
624

625
  def send_move_instructions(old_path_with_namespace)
626 627 628
    # 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) }
629
  end
630 631

  def owner
632 633
    if group
      group
634
    else
635
      namespace.try(:owner)
636 637
    end
  end
Dmitriy Zaporozhets committed
638

639
  def project_member_by_name_or_email(name = nil, email = nil)
640
    user = users.find_by('name like ? or email like ?', name, email)
641
    project_members.where(user: user) if user
Dmitriy Zaporozhets committed
642 643 644
  end

  # Get Team Member record by user id
645
  def project_member_by_id(user_id)
646
    project_members.find_by(user_id: user_id)
Dmitriy Zaporozhets committed
647 648 649 650 651
  end

  def name_with_namespace
    @name_with_namespace ||= begin
                               if namespace
652
                                 namespace.human_name + ' / ' + name
Dmitriy Zaporozhets committed
653 654 655 656 657 658 659 660 661 662 663 664 665 666
                               else
                                 name
                               end
                             end
  end

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

667 668
  def execute_hooks(data, hooks_scope = :push_hooks)
    hooks.send(hooks_scope).each do |hook|
669
      hook.async_execute(data, hooks_scope.to_s)
670
    end
Dmitriy Zaporozhets committed
671 672
  end

673 674 675
  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|
676
      service.async_execute(data)
Dmitriy Zaporozhets committed
677 678 679 680
    end
  end

  def update_merge_requests(oldrev, newrev, ref, user)
681 682
    MergeRequests::RefreshService.new(self, user).
      execute(oldrev, newrev, ref)
Dmitriy Zaporozhets committed
683 684 685
  end

  def valid_repo?
686
    repository.exists?
Dmitriy Zaporozhets committed
687
  rescue
688
    errors.add(:path, 'Invalid repository path')
Dmitriy Zaporozhets committed
689 690 691 692
    false
  end

  def empty_repo?
693
    !repository.exists? || !repository.has_visible_content?
Dmitriy Zaporozhets committed
694 695 696
  end

  def repo
697
    repository.raw
Dmitriy Zaporozhets committed
698 699 700
  end

  def url_to_repo
701
    gitlab_shell.url_to_repo(path_with_namespace)
Dmitriy Zaporozhets committed
702 703 704 705 706 707 708
  end

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

  def repo_exists?
709
    @repo_exists ||= repository.exists?
Dmitriy Zaporozhets committed
710 711 712 713 714
  rescue
    @repo_exists = false
  end

  def open_branches
715 716 717
    # 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)
718

719 720
    repository.branches.reject do |branch|
      protected_set.include?(branch.name)
721 722 723
    end
  end

724 725
  def protected_branch_names
    @protected_branch_names ||= protected_branches.pluck(:name)
Dmitriy Zaporozhets committed
726 727 728
  end

  def root_ref?(branch)
729
    repository.root_ref == branch
Dmitriy Zaporozhets committed
730 731 732 733 734 735 736
  end

  def ssh_url_to_repo
    url_to_repo
  end

  def http_url_to_repo
Douwe Maan committed
737
    "#{web_url}.git"
Dmitriy Zaporozhets committed
738 739 740
  end

  # Check if current branch name is marked as protected in the system
741
  def protected_branch?(branch_name)
742
    protected_branch_names.include?(branch_name)
Dmitriy Zaporozhets committed
743
  end
744

745
  def developers_can_push_to_protected_branch?(branch_name)
746
    protected_branches.any? { |pb| pb.name == branch_name && pb.developers_can_push }
747 748
  end

749 750 751
  def forked?
    !(forked_project_link.nil? || forked_project_link.forked_from_project.nil?)
  end
752

753 754 755 756
  def personal?
    !group
  end

757
  def rename_repo
758
    path_was = previous_changes['path'].first
759 760 761
    old_path_with_namespace = File.join(namespace_dir, path_was)
    new_path_with_namespace = File.join(namespace_dir, path)

762 763
    expire_caches_before_rename(old_path_with_namespace)

764 765
    if has_container_registry_tags?
      # we currently doesn't support renaming repository if it contains tags in container registry
Kamil Trzcinski committed
766
      raise Exception.new('Project cannot be renamed, because tags are present in its container registry')
767 768
    end

769
    if gitlab_shell.mv_repository(old_path_with_namespace, new_path_with_namespace)
770
      # If repository moved successfully we need to send update instructions to users.
771 772 773
      # However we cannot allow rollback since we moved repository
      # So we basically we mute exceptions in next actions
      begin
774
        gitlab_shell.mv_repository("#{old_path_with_namespace}.wiki", "#{new_path_with_namespace}.wiki")
775
        send_move_instructions(old_path_with_namespace)
776
        reset_events_cache
777 778 779 780 781

        @old_path_with_namespace = old_path_with_namespace

        SystemHooksService.new.execute_hooks_for(self, :rename)

782
        @repository = nil
783
      rescue
Johannes Schleifenbaum committed
784
        # Returning false does not rollback after_* transaction but gives
785 786 787 788 789 790 791 792
        # 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
793 794

    Gitlab::UploadsTransfer.new.rename_project(path_was, path, namespace.path)
795
  end
796

797 798 799 800 801 802
  # 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?
803
      repo.before_delete
804 805 806
    end

    if wiki.exists?
807
      wiki.before_delete
808 809 810
    end
  end

811 812
  def hook_attrs(backward: true)
    attrs = {
Kirill Zaitsev committed
813
      name: name,
814
      description: description,
Kirilll Zaitsev committed
815
      web_url: web_url,
816 817 818
      avatar_url: avatar_url,
      git_ssh_url: ssh_url_to_repo,
      git_http_url: http_url_to_repo,
Kirill Zaitsev committed
819
      namespace: namespace.name,
820 821 822
      visibility_level: visibility_level,
      path_with_namespace: path_with_namespace,
      default_branch: default_branch,
Kirill Zaitsev committed
823
    }
824 825 826 827 828 829 830 831 832 833 834 835

    # 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
836 837
  end

838 839 840 841 842
  # 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
843
  # * when the project avatar changes
844 845 846 847 848 849 850 851 852
  # 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
853 854

  def project_member(user)
855
    project_members.find_by(user_id: user)
856
  end
857 858 859 860

  def default_branch
    @default_branch ||= repository.root_ref if repository.exists?
  end
861 862 863 864 865

  def reload_default_branch
    @default_branch = nil
    default_branch
  end
866

867 868 869
  def visibility_level_field
    visibility_level
  end
870 871 872 873 874 875 876 877

  def archive!
    update_attribute(:archived, true)
  end

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

879
  def change_head(branch)
880
    repository.before_change_head
P.S.V.R committed
881 882 883
    repository.rugged.references.create('HEAD',
                                        "refs/heads/#{branch}",
                                        force: true)
884
    repository.copy_gitattributes(branch)
885 886
    reload_default_branch
  end
887 888 889 890

  def forked_from?(project)
    forked? && project == forked_from_project
  end
891 892 893 894

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

896 897 898 899
  def update_commit_count
    update_attribute(:commit_count, repository.commit_count)
  end

900
  def forks_count
901
    forks.count
902
  end
903 904 905 906

  def find_label(name)
    labels.find_by(name: name)
  end
907 908 909 910

  def origin_merge_requests
    merge_requests.where(source_project_id: self.id)
  end
911 912

  def create_repository
913 914
    # Forked import is handled asynchronously
    unless forked?
915
      if gitlab_shell.add_repository(path_with_namespace)
916
        repository.after_create
917 918
        true
      else
919
        errors.add(:base, 'Failed to create repository via gitlab-shell')
920 921
        false
      end
922 923 924 925 926 927 928 929 930 931
    end
  end

  def repository_exists?
    !!repository.exists?
  end

  def create_wiki
    ProjectWiki.new(self, self.owner).wiki
    true
932
  rescue ProjectWiki::CouldNotCreateWikiError
933
    errors.add(:base, 'Failed create wiki')
934 935
    false
  end
936

Drew Blessing committed
937 938 939 940
  def jira_tracker_active?
    jira_tracker? && jira_service.active
  end

941
  def allowed_to_share_with_group?
942
    !namespace.share_with_group_lock
943 944
  end

945 946
  def pipeline(sha, ref)
    pipelines.order(id: :desc).find_by(sha: sha, ref: ref)
947 948
  end

949 950
  def ensure_pipeline(sha, ref)
    pipeline(sha, ref) || pipelines.create(sha: sha, ref: ref)
951
  end
952

953 954 955
  def enable_ci
    self.builds_enabled = true
  end
956

957
  def any_runners?(&block)
958
    if runners.active.any?(&block)
959 960 961 962 963 964
      return true
    end

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

965
  def valid_runners_token?(token)
966
    self.runners_token && ActiveSupport::SecurityUtils.variable_size_secure_compare(token, self.runners_token)
Kamil Trzcinski committed
967 968
  end

Kamil Trzcinski committed
969 970
  # 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
971
  def valid_build_token?(token)
972
    self.builds_enabled? && self.runners_token && ActiveSupport::SecurityUtils.variable_size_secure_compare(token, self.runners_token)
973 974 975 976 977 978 979 980 981 982 983 984 985
  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
986

987 988 989
  def open_issues_count
    issues.opened.count
  end
990

991
  def visibility_level_allowed_as_fork?(level = self.visibility_level)
Douwe Maan committed
992
    return true unless forked?
993

Douwe Maan committed
994 995 996 997 998 999
    # 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
1000
  end
1001

1002 1003
  def visibility_level_allowed_by_group?(level = self.visibility_level)
    return true unless group
1004

1005
    level <= group.visibility_level
1006
  end
1007

1008 1009
  def visibility_level_allowed?(level = self.visibility_level)
    visibility_level_allowed_as_fork?(level) && visibility_level_allowed_by_group?(level)
Felipe Artur committed
1010 1011
  end

1012 1013 1014
  def runners_token
    ensure_runners_token!
  end
1015 1016 1017 1018

  def wiki
    @wiki ||= ProjectWiki.new(self, self.owner)
  end
1019 1020 1021 1022 1023 1024 1025

  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
1026

Josh Frye committed
1027 1028
  def running_or_pending_build_count(force: false)
    Rails.cache.fetch(['projects', id, 'running_or_pending_build_count'], force: force) do
1029 1030 1031
      builds.running_or_pending.count(:all)
    end
  end
1032 1033

  def mark_import_as_failed(error_message)
1034 1035 1036
    original_errors = errors.dup
    sanitized_message = Gitlab::UrlSanitizer.sanitize(error_message)

1037
    import_fail
1038 1039 1040 1041 1042
    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
1043
  end
gitlabhq committed
1044
end