BigW Consortium Gitlab

project.rb 26.2 KB
Newer Older
1 2 3 4 5 6 7 8
# == Schema Information
#
# Table name: projects
#
#  id                     :integer          not null, primary key
#  name                   :string(255)
#  path                   :string(255)
#  description            :text
Dmitriy Zaporozhets committed
9 10
#  created_at             :datetime
#  updated_at             :datetime
11
#  creator_id             :integer
12 13 14 15
#  issues_enabled         :boolean          default(TRUE), not null
#  wall_enabled           :boolean          default(TRUE), not null
#  merge_requests_enabled :boolean          default(TRUE), not null
#  wiki_enabled           :boolean          default(TRUE), not null
Dmitriy Zaporozhets committed
16
#  namespace_id           :integer
Dmitriy Zaporozhets committed
17
#  issues_tracker         :string(255)      default("gitlab"), not null
Dmitriy Zaporozhets committed
18
#  issues_tracker_id      :string(255)
Dmitriy Zaporozhets committed
19
#  snippets_enabled       :boolean          default(TRUE), not null
20
#  last_activity_at       :datetime
Dmitriy Zaporozhets committed
21
#  import_url             :string(255)
22
#  visibility_level       :integer          default(0), not null
Dmitriy Zaporozhets committed
23
#  archived               :boolean          default(FALSE), not null
Atsushi Ishida committed
24
#  avatar                 :string(255)
Dmitriy Zaporozhets committed
25
#  import_status          :string(255)
Dmitriy Zaporozhets committed
26 27
#  repository_size        :float            default(0.0)
#  star_count             :integer          default(0), not null
Dmitriy Zaporozhets committed
28 29
#  import_type            :string(255)
#  import_source          :string(255)
Atsushi Ishida committed
30
#  commit_count           :integer          default(0)
Stan Hu committed
31
#  import_error           :text
Stan Hu committed
32 33 34 35 36 37 38
#  ci_id                  :integer
#  builds_enabled         :boolean          default(TRUE), not null
#  shared_runners_enabled :boolean          default(TRUE), not null
#  runners_token          :string
#  build_coverage_regex   :string
#  build_allow_git_fetch  :boolean          default(TRUE), not null
#  build_timeout          :integer          default(3600), not null
39
#  pending_delete         :boolean
40 41
#

42 43 44
require 'carrierwave/orm/activerecord'
require 'file_size_validator'

gitlabhq committed
45
class Project < ActiveRecord::Base
46
  include Gitlab::ConfigHelper
47
  include Gitlab::ShellAdapter
48
  include Gitlab::VisibilityLevel
49
  include Gitlab::CurrentSettings
50 51
  include Referable
  include Sortable
52
  include AfterCommitQueue
53
  include CaseSensitivity
54
  include TokenAuthenticatable
Robert Speicher committed
55

56
  extend Gitlab::ConfigHelper
57

Jared Szechy committed
58 59
  UNKNOWN_IMPORT_URL = 'http://unknown.git'

60
  default_value_for :archived, false
61 62 63
  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
64
  default_value_for :builds_enabled, gitlab_config_features.builds
65
  default_value_for :wiki_enabled, gitlab_config_features.wiki
66
  default_value_for :wall_enabled, false
67
  default_value_for :snippets_enabled, gitlab_config_features.snippets
68
  default_value_for(:shared_runners_enabled) { current_application_settings.shared_runners_enabled }
69

70 71
  # set last_activity_at to the same as created_at
  after_create :set_last_activity_at
72
  def set_last_activity_at
73
    update_column(:last_activity_at, self.created_at)
74 75
  end

76 77 78 79 80 81 82 83 84 85 86 87 88
  # update visibility_levet of forks
  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

89
  ActsAsTaggableOn.strict_case_match = true
90
  acts_as_taggable_on :tags
91

92
  attr_accessor :new_default_branch
93
  attr_accessor :old_path_with_namespace
94

95
  # Relations
96
  belongs_to :creator, foreign_key: 'creator_id', class_name: 'User'
97
  belongs_to :group, -> { where(type: Group) }, foreign_key: 'namespace_id'
98
  belongs_to :namespace
99

100
  has_one :last_event, -> {order 'events.created_at DESC'}, class_name: 'Event', foreign_key: 'project_id'
101 102 103

  # Project services
  has_many :services
104
  has_one :campfire_service, dependent: :destroy
Kirilll Zaitsev committed
105
  has_one :drone_ci_service, dependent: :destroy
106
  has_one :emails_on_push_service, dependent: :destroy
107
  has_one :builds_email_service, dependent: :destroy
Aorimn committed
108
  has_one :irker_service, dependent: :destroy
109
  has_one :pivotaltracker_service, dependent: :destroy
110
  has_one :hipchat_service, dependent: :destroy
111
  has_one :flowdock_service, dependent: :destroy
112
  has_one :assembla_service, dependent: :destroy
Jeremy committed
113
  has_one :asana_service, dependent: :destroy
114
  has_one :gemnasium_service, dependent: :destroy
115
  has_one :slack_service, dependent: :destroy
116
  has_one :buildkite_service, dependent: :destroy
117
  has_one :bamboo_service, dependent: :destroy
118
  has_one :teamcity_service, dependent: :destroy
119
  has_one :pushover_service, dependent: :destroy
120 121
  has_one :jira_service, dependent: :destroy
  has_one :redmine_service, dependent: :destroy
122
  has_one :custom_issue_tracker_service, dependent: :destroy
123
  has_one :gitlab_issue_tracker_service, dependent: :destroy
124
  has_one :external_wiki_service, dependent: :destroy
125

126 127 128 129 130
  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
131

132
  # Merge Requests for target project should be removed with it
133
  has_many :merge_requests,     dependent: :destroy, foreign_key: 'target_project_id'
134
  # Merge requests from source project should be kept when source project was removed
135
  has_many :fork_merge_requests, foreign_key: 'source_project_id', class_name: MergeRequest
136
  has_many :issues,             dependent: :destroy
137
  has_many :labels,             dependent: :destroy
138 139
  has_many :services,           dependent: :destroy
  has_many :events,             dependent: :destroy
140 141
  has_many :milestones,         dependent: :destroy
  has_many :notes,              dependent: :destroy
142 143
  has_many :snippets,           dependent: :destroy, class_name: 'ProjectSnippet'
  has_many :hooks,              dependent: :destroy, class_name: 'ProjectHook'
144
  has_many :protected_branches, dependent: :destroy
145 146
  has_many :project_members, dependent: :destroy, as: :source, class_name: 'ProjectMember'
  has_many :users, through: :project_members
147 148
  has_many :deploy_keys_projects, dependent: :destroy
  has_many :deploy_keys, through: :deploy_keys_projects
Ciro Santilli committed
149 150
  has_many :users_star_projects, dependent: :destroy
  has_many :starrers, through: :users_star_projects, source: :user
151
  has_many :releases, dependent: :destroy
152 153
  has_many :lfs_objects_projects, dependent: :destroy
  has_many :lfs_objects, through: :lfs_objects_projects
154

155
  has_one :import_data, dependent: :destroy, class_name: "ProjectImportData"
156

157
  has_many :commit_statuses, dependent: :destroy, class_name: 'CommitStatus', foreign_key: :gl_project_id
158
  has_many :ci_commits, dependent: :destroy, class_name: 'Ci::Commit', foreign_key: :gl_project_id
159 160 161 162 163
  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
164

165
  accepts_nested_attributes_for :variables, allow_destroy: true
166

167
  delegate :name, to: :owner, allow_nil: true, prefix: true
168
  delegate :members, to: :team, prefix: true
169

Andrey Kumanyaev committed
170
  # Validations
171
  validates :creator, presence: true, on: :create
172
  validates :description, length: { maximum: 2000 }, allow_blank: true
173 174 175 176
  validates :name,
    presence: true,
    length: { within: 0..255 },
    format: { with: Gitlab::Regex.project_name_regex,
177
              message: Gitlab::Regex.project_name_regex_message }
178 179 180
  validates :path,
    presence: true,
    length: { within: 0..255 },
181 182
    format: { with: Gitlab::Regex.project_path_regex,
              message: Gitlab::Regex.project_path_regex_message }
183
  validates :issues_enabled, :merge_requests_enabled,
Andrey Kumanyaev committed
184
            :wiki_enabled, inclusion: { in: [true, false] }
185
  validates :issues_tracker_id, length: { maximum: 255 }, allow_blank: true
186
  validates :namespace, presence: true
187 188
  validates_uniqueness_of :name, scope: :namespace_id
  validates_uniqueness_of :path, scope: :namespace_id
189
  validates :import_url,
190
    url: { protocols: %w(ssh git http https) },
191
    if: :external_import?
192
  validates :star_count, numericality: { greater_than_or_equal_to: 0 }
193
  validate :check_limit, on: :create
194
  validate :avatar_type,
195
    if: ->(project) { project.avatar.present? && project.avatar_changed? }
196
  validates :avatar, file_size: { maximum: 200.kilobytes.to_i }
197

198
  add_authentication_token_field :runners_token
199
  before_save :ensure_runners_token
200

201
  mount_uploader :avatar, AvatarUploader
Andrey Kumanyaev committed
202

203
  # Scopes
204
  scope :sorted_by_activity, -> { reorder(last_activity_at: :desc) }
205 206 207
  scope :sorted_by_stars, -> { reorder('projects.star_count DESC') }
  scope :sorted_by_names, -> { joins(:namespace).reorder('namespaces.name ASC, projects.name ASC') }

208 209 210
  scope :without_user, ->(user)  { where('projects.id NOT IN (:ids)', ids: user.authorized_projects.map(&:id) ) }
  scope :without_team, ->(team) { team.projects.present? ? where('projects.id NOT IN (:ids)', ids: team.projects.map(&:id)) : scoped  }
  scope :not_in_group, ->(group) { where('projects.id NOT IN (:ids)', ids: group.project_ids ) }
211
  scope :in_namespace, ->(namespace_ids) { where(namespace_id: namespace_ids) }
Andrey Kumanyaev committed
212
  scope :in_group_namespace, -> { joins(:group) }
213
  scope :personal, ->(user) { where(namespace_id: user.namespace_id) }
214
  scope :joined, ->(user) { where('namespace_id != ?', user.namespace_id) }
215 216
  scope :public_only, -> { where(visibility_level: Project::PUBLIC) }
  scope :public_and_internal_only, -> { where(visibility_level: Project.public_and_internal_levels) }
217 218
  scope :non_archived, -> { where(archived: false) }

219 220
  state_machine :import_status, initial: :none do
    event :import_start do
221
      transition [:none, :finished] => :started
222 223 224
    end

    event :import_finish do
225
      transition started: :finished
226 227 228
    end

    event :import_fail do
229
      transition started: :failed
230 231 232
    end

    event :import_retry do
233
      transition failed: :started
234 235 236 237
    end

    state :started
    state :finished
238 239
    state :failed

240
    after_transition any => :started, do: :schedule_add_import_job
241
    after_transition any => :finished, do: :clear_import_data
242 243
  end

Andrey Kumanyaev committed
244
  class << self
245 246 247 248
    def public_and_internal_levels
      [Project::PUBLIC, Project::INTERNAL]
    end

249
    def abandoned
250
      where('projects.last_activity_at < ?', 6.months.ago)
251
    end
252

253 254
    def publicish(user)
      visibility_levels = [Project::PUBLIC]
255
      visibility_levels << Project::INTERNAL if user
256 257
      where(visibility_level: visibility_levels)
    end
258

259
    def with_push
260
      joins(:events).where('events.action = ?', Event::PUSHED)
261 262
    end

Andrey Kumanyaev committed
263
    def active
264
      joins(:issues, :notes, :merge_requests).order('issues.created_at, notes.created_at, merge_requests.created_at DESC')
Andrey Kumanyaev committed
265
    end
266

267
    def search(query)
268
      joins(:namespace).
269
        where('LOWER(projects.name) LIKE :query OR
Dmitriy Zaporozhets committed
270 271
              LOWER(projects.path) LIKE :query OR
              LOWER(namespaces.name) LIKE :query OR
272
              LOWER(projects.description) LIKE :query',
Dmitriy Zaporozhets committed
273
              query: "%#{query.try(:downcase)}%")
Andrey Kumanyaev committed
274
    end
275

276
    def search_by_visibility(level)
Josh Frye committed
277
      where(visibility_level: Gitlab::VisibilityLevel.const_get(level.upcase))
278 279
    end

280
    def search_by_title(query)
281
      where('projects.archived = ?', false).where('LOWER(projects.name) LIKE :query', query: "%#{query.downcase}%")
282 283
    end

284
    def find_with_namespace(id)
285 286 287 288 289 290 291
      namespace_path, project_path = id.split('/')

      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).
292
      projects = unscoped.
293
        joins(:namespace).
294 295
        iwhere('namespaces.path' => namespace_path)

296
      projects.find_by('projects.path' => project_path) ||
297
        projects.iwhere('projects.path' => project_path).take
298
    end
299

300
    def find_by_ci_id(id)
Kamil Trzcinski committed
301
      find_by(ci_id: id.to_i)
302 303
    end

304 305 306
    def visibility_levels
      Gitlab::VisibilityLevel.options
    end
307 308

    def sort(method)
309 310 311 312
      if method == 'repository_size_desc'
        reorder(repository_size: :desc, id: :desc)
      else
        order_by(method)
313 314
      end
    end
315 316 317 318 319

    def reference_pattern
      name_pattern = Gitlab::Regex::NAMESPACE_REGEX_STR
      %r{(?<project>#{name_pattern}/#{name_pattern})}
    end
320 321 322 323 324 325 326 327 328 329 330 331 332 333

    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
334 335 336 337

    def visible_to_user(user)
      where(id: user.authorized_projects.select(:id).reorder(nil))
    end
338 339
  end

340
  def team
341
    @team ||= ProjectTeam.new(self)
342 343 344
  end

  def repository
345 346 347
    @repository ||= Repository.new(path_with_namespace, nil, self)
  end

348
  def commit(id = 'HEAD')
349
    repository.commit(id)
350 351
  end

352
  def merge_base_commit(first_commit_id, second_commit_id)
Douwe Maan committed
353 354
    sha = repository.merge_base(first_commit_id, second_commit_id)
    repository.commit(sha) if sha
355 356
  end

357
  def saved?
358
    id && persisted?
359 360
  end

361 362 363 364
  def schedule_add_import_job
    run_after_commit(:add_import_job)
  end

365
  def add_import_job
366
    if forked?
367
      RepositoryForkWorker.perform_async(self.id, forked_from_project.path_with_namespace, self.namespace.path)
368
    else
369
      RepositoryImportWorker.perform_async(self.id)
370
    end
371 372
  end

373
  def clear_import_data
374 375 376 377
    update(import_error: nil)

    ProjectCacheWorker.perform_async(self.id)

378
    self.import_data.destroy if self.import_data
379 380
  end

381
  def import?
382 383 384 385
    external_import? || forked?
  end

  def external_import?
386 387 388
    import_url.present?
  end

389
  def imported?
390 391 392 393 394 395 396 397 398 399 400 401 402
    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'
403 404
  end

405 406 407 408 409
  def safe_import_url
    result = URI.parse(self.import_url)
    result.password = '*****' unless result.password.nil?
    result.to_s
  rescue
410
    self.import_url
411 412
  end

413
  def check_limit
414
    unless creator.can_create_project? or namespace.kind == 'group'
415
      errors[:limit_reached] << ("Your project limit is #{creator.projects_limit} projects! Please contact your administrator to increase it")
416 417
    end
  rescue
Robert Speicher committed
418
    errors[:base] << ("Can't check your ability to create project")
gitlabhq committed
419 420
  end

421
  def to_param
Vinnie Okada committed
422
    path
423 424
  end

425 426 427 428
  def to_reference(_from_project = nil)
    path_with_namespace
  end

429
  def web_url
430
    Gitlab::Application.routes.url_helpers.namespace_project_url(self.namespace, self)
431 432
  end

433
  def web_url_without_protocol
434
    web_url.split('://')[1]
435 436
  end

437
  def build_commit_note(commit)
438
    notes.new(commit_id: commit.id, noteable_type: 'Commit')
gitlabhq committed
439
  end
Nihad Abbasov committed
440

441
  def last_activity
442
    last_event
gitlabhq committed
443 444 445
  end

  def last_activity_date
Andrey Kumanyaev committed
446
    last_activity_at || updated_at
Dmitriy Zaporozhets committed
447
  end
448

449 450 451
  def project_id
    self.id
  end
randx committed
452

453
  def get_issue(issue_id)
454
    if default_issues_tracker?
455
      issues.find_by(iid: issue_id)
456
    else
457
      ExternalIssue.new(issue_id, self)
458 459 460
    end
  end

461
  def issue_exists?(issue_id)
462
    get_issue(issue_id)
463 464
  end

465
  def default_issue_tracker
466
    gitlab_issue_tracker_service || create_gitlab_issue_tracker_service
467 468 469 470 471 472 473 474 475 476
  end

  def issues_tracker
    if external_issue_tracker
      external_issue_tracker
    else
      default_issue_tracker
    end
  end

477
  def default_issues_tracker?
478
    !external_issue_tracker
479 480 481
  end

  def external_issue_tracker
482 483
    @external_issue_tracker ||=
      services.issue_trackers.active.without_defaults.first
484 485
  end

Andrew8xx8 committed
486
  def can_have_issues_tracker_id?
487
    self.issues_enabled && !self.default_issues_tracker?
Andrew8xx8 committed
488 489
  end

490
  def build_missing_services
491 492
    services_templates = Service.where(template: true)

493
    Service.available_services_names.each do |service_name|
494
      service = find_service(services, service_name)
495 496

      # If service is available but missing in db
497 498 499 500 501 502
      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`
503
          self.send :"create_#{service_name}_service"
504 505 506 507
        else
          Service.create_from_template(self.id, template)
        end
      end
508 509 510
    end
  end

511 512 513 514 515 516 517 518 519
  def create_labels
    Label.templates.each do |label|
      label = label.dup
      label.template = nil
      label.project_id = self.id
      label.save
    end
  end

520 521 522
  def find_service(list, name)
    list.find { |service| service.to_param == name }
  end
523

524 525 526 527 528
  def ci_services
    services.select { |service| service.category == :ci }
  end

  def ci_service
529
    @ci_service ||= ci_services.find(&:activated?)
530 531
  end

Drew Blessing committed
532 533 534 535
  def jira_tracker?
    issues_tracker.to_param == 'jira'
  end

536
  def avatar_type
537 538
    unless self.avatar.image?
      self.errors.add :avatar, 'only images allowed'
539 540 541 542 543 544 545 546 547 548
    end
  end

  def avatar_in_git
    @avatar_file ||= 'logo.png' if repository.blob_at_branch('master', 'logo.png')
    @avatar_file ||= 'logo.jpg' if repository.blob_at_branch('master', 'logo.jpg')
    @avatar_file ||= 'logo.gif' if repository.blob_at_branch('master', 'logo.gif')
    @avatar_file
  end

549 550 551 552
  def avatar_url
    if avatar.present?
      [gitlab_config.url, avatar.url].join
    elsif avatar_in_git
553
      Gitlab::Application.routes.url_helpers.namespace_project_avatar_url(namespace, self)
554 555 556
    end
  end

557 558 559 560 561
  # For compatibility with old code
  def code
    path
  end

562
  def items_for(entity)
563 564 565 566 567 568 569
    case entity
    when 'issue' then
      issues
    when 'merge_request' then
      merge_requests
    end
  end
570

571
  def send_move_instructions(old_path_with_namespace)
572 573 574
    # 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) }
575
  end
576 577

  def owner
578 579
    if group
      group
580
    else
581
      namespace.try(:owner)
582 583
    end
  end
Dmitriy Zaporozhets committed
584

585
  def project_member_by_name_or_email(name = nil, email = nil)
586
    user = users.find_by('name like ? or email like ?', name, email)
587
    project_members.where(user: user) if user
Dmitriy Zaporozhets committed
588 589 590
  end

  # Get Team Member record by user id
591
  def project_member_by_id(user_id)
592
    project_members.find_by(user_id: user_id)
Dmitriy Zaporozhets committed
593 594 595 596 597
  end

  def name_with_namespace
    @name_with_namespace ||= begin
                               if namespace
598
                                 namespace.human_name + ' / ' + name
Dmitriy Zaporozhets committed
599 600 601 602 603 604 605 606 607 608 609 610 611 612
                               else
                                 name
                               end
                             end
  end

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

613 614
  def execute_hooks(data, hooks_scope = :push_hooks)
    hooks.send(hooks_scope).each do |hook|
615
      hook.async_execute(data, hooks_scope.to_s)
616
    end
Dmitriy Zaporozhets committed
617 618
  end

619 620 621
  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|
622
      service.async_execute(data)
Dmitriy Zaporozhets committed
623 624 625 626
    end
  end

  def update_merge_requests(oldrev, newrev, ref, user)
627 628
    MergeRequests::RefreshService.new(self, user).
      execute(oldrev, newrev, ref)
Dmitriy Zaporozhets committed
629 630 631
  end

  def valid_repo?
632
    repository.exists?
Dmitriy Zaporozhets committed
633
  rescue
634
    errors.add(:path, 'Invalid repository path')
Dmitriy Zaporozhets committed
635 636 637 638
    false
  end

  def empty_repo?
639
    !repository.exists? || !repository.has_visible_content?
Dmitriy Zaporozhets committed
640 641 642
  end

  def repo
643
    repository.raw
Dmitriy Zaporozhets committed
644 645 646
  end

  def url_to_repo
647
    gitlab_shell.url_to_repo(path_with_namespace)
Dmitriy Zaporozhets committed
648 649 650 651 652 653 654
  end

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

  def repo_exists?
655
    @repo_exists ||= repository.exists?
Dmitriy Zaporozhets committed
656 657 658 659 660
  rescue
    @repo_exists = false
  end

  def open_branches
661 662 663 664 665 666 667 668 669 670 671 672 673
    all_branches = repository.branches

    if protected_branches.present?
      all_branches.reject! do |branch|
        protected_branches_names.include?(branch.name)
      end
    end

    all_branches
  end

  def protected_branches_names
    @protected_branches_names ||= protected_branches.map(&:name)
Dmitriy Zaporozhets committed
674 675 676
  end

  def root_ref?(branch)
677
    repository.root_ref == branch
Dmitriy Zaporozhets committed
678 679 680 681 682 683 684
  end

  def ssh_url_to_repo
    url_to_repo
  end

  def http_url_to_repo
Douwe Maan committed
685
    "#{web_url}.git"
Dmitriy Zaporozhets committed
686 687 688
  end

  # Check if current branch name is marked as protected in the system
689
  def protected_branch?(branch_name)
690
    protected_branches_names.include?(branch_name)
Dmitriy Zaporozhets committed
691
  end
692

693
  def developers_can_push_to_protected_branch?(branch_name)
694
    protected_branches.any? { |pb| pb.name == branch_name && pb.developers_can_push }
695 696
  end

697 698 699
  def forked?
    !(forked_project_link.nil? || forked_project_link.forked_from_project.nil?)
  end
700

701 702 703 704
  def personal?
    !group
  end

705
  def rename_repo
706
    path_was = previous_changes['path'].first
707 708 709 710
    old_path_with_namespace = File.join(namespace_dir, path_was)
    new_path_with_namespace = File.join(namespace_dir, path)

    if gitlab_shell.mv_repository(old_path_with_namespace, new_path_with_namespace)
711
      # If repository moved successfully we need to send update instructions to users.
712 713 714
      # However we cannot allow rollback since we moved repository
      # So we basically we mute exceptions in next actions
      begin
715
        gitlab_shell.mv_repository("#{old_path_with_namespace}.wiki", "#{new_path_with_namespace}.wiki")
716
        send_move_instructions(old_path_with_namespace)
717
        reset_events_cache
718 719 720 721 722

        @old_path_with_namespace = old_path_with_namespace

        SystemHooksService.new.execute_hooks_for(self, :rename)

723
        @repository = nil
724
      rescue
Johannes Schleifenbaum committed
725
        # Returning false does not rollback after_* transaction but gives
726 727 728 729 730 731 732 733
        # 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
734 735

    Gitlab::UploadsTransfer.new.rename_project(path_was, path, namespace.path)
736
  end
737

Kirill Zaitsev committed
738 739 740 741 742
  def hook_attrs
    {
      name: name,
      ssh_url: ssh_url_to_repo,
      http_url: http_url_to_repo,
Kirilll Zaitsev committed
743
      web_url: web_url,
Kirill Zaitsev committed
744 745 746 747 748
      namespace: namespace.name,
      visibility_level: visibility_level
    }
  end

749 750 751 752 753
  # 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
754
  # * when the project avatar changes
755 756 757 758 759 760 761 762 763
  # 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
764 765

  def project_member(user)
766
    project_members.find_by(user_id: user)
767
  end
768 769 770 771

  def default_branch
    @default_branch ||= repository.root_ref if repository.exists?
  end
772 773 774 775 776

  def reload_default_branch
    @default_branch = nil
    default_branch
  end
777

778 779 780
  def visibility_level_field
    visibility_level
  end
781 782 783 784 785 786 787 788

  def archive!
    update_attribute(:archived, true)
  end

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

790
  def change_head(branch)
791 792
    # Cached divergent commit counts are based on repository head
    repository.expire_branch_cache
793 794 795
    gitlab_shell.update_repository_head(self.path_with_namespace, branch)
    reload_default_branch
  end
796 797 798 799

  def forked_from?(project)
    forked? && project == forked_from_project
  end
800 801 802 803

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

805 806 807 808
  def update_commit_count
    update_attribute(:commit_count, repository.commit_count)
  end

809
  def forks_count
810
    forks.count
811
  end
812 813 814 815

  def find_label(name)
    labels.find_by(name: name)
  end
816 817 818 819

  def origin_merge_requests
    merge_requests.where(source_project_id: self.id)
  end
820 821

  def create_repository
822 823
    # Forked import is handled asynchronously
    unless forked?
824 825 826
      if gitlab_shell.add_repository(path_with_namespace)
        true
      else
827
        errors.add(:base, 'Failed to create repository via gitlab-shell')
828 829
        false
      end
830 831 832 833 834 835 836 837 838 839
    end
  end

  def repository_exists?
    !!repository.exists?
  end

  def create_wiki
    ProjectWiki.new(self, self.owner).wiki
    true
840
  rescue ProjectWiki::CouldNotCreateWikiError
841
    errors.add(:base, 'Failed create wiki')
842 843
    false
  end
844

Drew Blessing committed
845 846 847 848
  def jira_tracker_active?
    jira_tracker? && jira_service.active
  end

849
  def ci_commit(sha)
850 851 852 853 854
    ci_commits.find_by(sha: sha)
  end

  def ensure_ci_commit(sha)
    ci_commit(sha) || ci_commits.create(sha: sha)
855
  end
856

857 858 859
  def enable_ci
    self.builds_enabled = true
  end
860 861 862 863 864 865 866 867 868 869

  def unlink_fork
    if forked?
      forked_from_project.lfs_objects.find_each do |lfs_object|
        lfs_object.projects << self
      end

      forked_project_link.destroy
    end
  end
870 871

  def any_runners?(&block)
872
    if runners.active.any?(&block)
873 874 875 876 877 878
      return true
    end

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

Kamil Trzcinski committed
879
  def valid_runners_token? token
Kamil Trzcinski committed
880
    self.runners_token && self.runners_token == token
Kamil Trzcinski committed
881 882
  end

Kamil Trzcinski committed
883 884
  # 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
Kamil Trzcinski committed
885
  def valid_build_token? token
Kamil Trzcinski committed
886
    self.builds_enabled? && self.runners_token && self.runners_token == token
887 888 889 890 891 892 893 894 895 896 897 898 899
  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
900

901 902 903
  def open_issues_count
    issues.opened.count
  end
904 905 906 907

  def visibility_level_allowed?(level)
    return true unless forked?
    Gitlab::VisibilityLevel.allowed_fork_levels(forked_from_project.visibility_level).include?(level.to_i)
908
  end
909 910 911 912

  def runners_token
    ensure_runners_token!
  end
913 914 915 916

  def wiki
    @wiki ||= ProjectWiki.new(self, self.owner)
  end
gitlabhq committed
917
end