project.rb 21.5 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's avatar
Dmitriy Zaporozhets committed
9
10
#  created_at             :datetime
#  updated_at             :datetime
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
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's avatar
Dmitriy Zaporozhets committed
16
#  namespace_id           :integer
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
17
#  issues_tracker         :string(255)      default("gitlab"), not null
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
18
#  issues_tracker_id      :string(255)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
19
#  snippets_enabled       :boolean          default(TRUE), not null
20
#  last_activity_at       :datetime
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
21
#  import_url             :string(255)
22
#  visibility_level       :integer          default(0), not null
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
23
#  archived               :boolean          default(FALSE), not null
Atsushi Ishida's avatar
Atsushi Ishida committed
24
#  avatar                 :string(255)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
25
#  import_status          :string(255)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
26
27
#  repository_size        :float            default(0.0)
#  star_count             :integer          default(0), not null
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
28
29
#  import_type            :string(255)
#  import_source          :string(255)
Atsushi Ishida's avatar
Atsushi Ishida committed
30
#  commit_count           :integer          default(0)
31
32
#

33
34
35
require 'carrierwave/orm/activerecord'
require 'file_size_validator'

gitlabhq's avatar
gitlabhq committed
36
class Project < ActiveRecord::Base
37
  include Gitlab::ConfigHelper
38
  include Gitlab::ShellAdapter
39
  include Gitlab::VisibilityLevel
40
41
  include Referable
  include Sortable
42
  include AfterCommitQueue
sue445's avatar
sue445 committed
43

44
  extend Gitlab::ConfigHelper
45
  extend Enumerize
46

Jared Szechy's avatar
Jared Szechy committed
47
48
  UNKNOWN_IMPORT_URL = 'http://unknown.git'

49
  default_value_for :archived, false
50
51
52
53
  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
  default_value_for :wiki_enabled, gitlab_config_features.wiki
54
  default_value_for :wall_enabled, false
55
  default_value_for :snippets_enabled, gitlab_config_features.snippets
56

57
58
  # set last_activity_at to the same as created_at
  after_create :set_last_activity_at
59
  def set_last_activity_at
60
    update_column(:last_activity_at, self.created_at)
61
62
  end

63
  ActsAsTaggableOn.strict_case_match = true
64
  acts_as_taggable_on :tags
65

66
67
  attr_accessor :new_default_branch

68
  # Relations
69
  belongs_to :creator, foreign_key: 'creator_id', class_name: 'User'
70
  belongs_to :group, -> { where(type: Group) }, foreign_key: 'namespace_id'
71
  belongs_to :namespace
72

73
  has_one :last_event, -> {order 'events.created_at DESC'}, class_name: 'Event', foreign_key: 'project_id'
74
75
76

  # Project services
  has_many :services
77
  has_one :gitlab_ci_service, dependent: :destroy
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
78
  has_one :campfire_service, dependent: :destroy
Kirilll Zaitsev's avatar
Kirilll Zaitsev committed
79
  has_one :drone_ci_service, dependent: :destroy
80
  has_one :emails_on_push_service, dependent: :destroy
Aorimn's avatar
Aorimn committed
81
  has_one :irker_service, dependent: :destroy
82
  has_one :pivotaltracker_service, dependent: :destroy
83
  has_one :hipchat_service, dependent: :destroy
84
  has_one :flowdock_service, dependent: :destroy
Carlos Paramio's avatar
Carlos Paramio committed
85
  has_one :assembla_service, dependent: :destroy
Jeremy's avatar
Jeremy committed
86
  has_one :asana_service, dependent: :destroy
87
  has_one :gemnasium_service, dependent: :destroy
88
  has_one :slack_service, dependent: :destroy
89
  has_one :buildkite_service, dependent: :destroy
Drew Blessing's avatar
Drew Blessing committed
90
  has_one :bamboo_service, dependent: :destroy
91
  has_one :teamcity_service, dependent: :destroy
92
  has_one :pushover_service, dependent: :destroy
93
94
  has_one :jira_service, dependent: :destroy
  has_one :redmine_service, dependent: :destroy
95
  has_one :custom_issue_tracker_service, dependent: :destroy
96
  has_one :gitlab_issue_tracker_service, dependent: :destroy
97
  has_one :external_wiki_service, dependent: :destroy
98

99
  has_one :forked_project_link, dependent: :destroy, foreign_key: "forked_to_project_id"
100

101
  has_one :forked_from_project, through: :forked_project_link
102
  # Merge Requests for target project should be removed with it
103
  has_many :merge_requests,     dependent: :destroy, foreign_key: 'target_project_id'
104
  # Merge requests from source project should be kept when source project was removed
105
  has_many :fork_merge_requests, foreign_key: 'source_project_id', class_name: MergeRequest
106
  has_many :issues,             dependent: :destroy
107
  has_many :labels,             dependent: :destroy
108
109
  has_many :services,           dependent: :destroy
  has_many :events,             dependent: :destroy
110
111
  has_many :milestones,         dependent: :destroy
  has_many :notes,              dependent: :destroy
112
113
  has_many :snippets,           dependent: :destroy, class_name: 'ProjectSnippet'
  has_many :hooks,              dependent: :destroy, class_name: 'ProjectHook'
114
  has_many :protected_branches, dependent: :destroy
115
116
  has_many :project_members, dependent: :destroy, as: :source, class_name: 'ProjectMember'
  has_many :users, through: :project_members
117
118
  has_many :deploy_keys_projects, dependent: :destroy
  has_many :deploy_keys, through: :deploy_keys_projects
Ciro Santilli's avatar
Ciro Santilli committed
119
120
  has_many :users_star_projects, dependent: :destroy
  has_many :starrers, through: :users_star_projects, source: :user
Kamil Trzcinski's avatar
Kamil Trzcinski committed
121
122
  has_many :ci_commits, ->() { order('CASE WHEN ci_commits.committed_at IS NULL THEN 0 ELSE 1 END', :committed_at, :id) }, dependent: :destroy, class_name: 'Ci::Commit', foreign_key: :gl_project_id
  has_many :ci_builds, through: :ci_commits, source: :builds, dependent: :destroy, class_name: 'Ci::Build'
123

124
  has_one :import_data, dependent: :destroy, class_name: "ProjectImportData"
125
  has_one :gitlab_ci_project, dependent: :destroy, class_name: "Ci::Project", foreign_key: :gitlab_id
126

127
  delegate :name, to: :owner, allow_nil: true, prefix: true
128
  delegate :members, to: :team, prefix: true
129

Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
130
  # Validations
131
  validates :creator, presence: true, on: :create
132
  validates :description, length: { maximum: 2000 }, allow_blank: true
133
134
135
136
  validates :name,
    presence: true,
    length: { within: 0..255 },
    format: { with: Gitlab::Regex.project_name_regex,
Douwe Maan's avatar
Douwe Maan committed
137
              message: Gitlab::Regex.project_name_regex_message }
138
139
140
  validates :path,
    presence: true,
    length: { within: 0..255 },
Douwe Maan's avatar
Douwe Maan committed
141
142
    format: { with: Gitlab::Regex.project_path_regex,
              message: Gitlab::Regex.project_path_regex_message }
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
143
  validates :issues_enabled, :merge_requests_enabled,
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
144
            :wiki_enabled, inclusion: { in: [true, false] }
145
  validates :issues_tracker_id, length: { maximum: 255 }, allow_blank: true
146
  validates :namespace, presence: true
147
148
  validates_uniqueness_of :name, scope: :namespace_id
  validates_uniqueness_of :path, scope: :namespace_id
149
  validates :import_url,
150
    format: { with: /\A#{URI.regexp(%w(ssh git http https))}\z/, message: 'should be a valid url' },
151
    if: :external_import?
152
  validates :star_count, numericality: { greater_than_or_equal_to: 0 }
153
  validate :check_limit, on: :create
154
  validate :avatar_type,
155
    if: ->(project) { project.avatar.present? && project.avatar_changed? }
156
  validates :avatar, file_size: { maximum: 200.kilobytes.to_i }
157

Douwe Maan's avatar
Douwe Maan committed
158
  mount_uploader :avatar, AvatarUploader
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
159

160
  # Scopes
161
  scope :sorted_by_activity, -> { reorder(last_activity_at: :desc) }
162
163
164
  scope :sorted_by_stars, -> { reorder('projects.star_count DESC') }
  scope :sorted_by_names, -> { joins(:namespace).reorder('namespaces.name ASC, projects.name ASC') }

165
166
167
  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 ) }
168
  scope :in_namespace, ->(namespace_ids) { where(namespace_id: namespace_ids) }
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
169
  scope :in_group_namespace, -> { joins(:group) }
170
  scope :personal, ->(user) { where(namespace_id: user.namespace_id) }
171
  scope :joined, ->(user) { where('namespace_id != ?', user.namespace_id) }
172
173
  scope :public_only, -> { where(visibility_level: Project::PUBLIC) }
  scope :public_and_internal_only, -> { where(visibility_level: Project.public_and_internal_levels) }
174
175
  scope :non_archived, -> { where(archived: false) }

176
177
  state_machine :import_status, initial: :none do
    event :import_start do
178
      transition [:none, :finished] => :started
179
180
181
    end

    event :import_finish do
182
      transition started: :finished
183
184
185
    end

    event :import_fail do
186
      transition started: :failed
187
188
189
    end

    event :import_retry do
190
      transition failed: :started
191
192
193
194
    end

    state :started
    state :finished
195
196
    state :failed

197
    after_transition any => :started, do: :schedule_add_import_job
198
    after_transition any => :finished, do: :clear_import_data
199
200
  end

Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
201
  class << self
202
203
204
205
    def public_and_internal_levels
      [Project::PUBLIC, Project::INTERNAL]
    end

206
    def abandoned
207
      where('projects.last_activity_at < ?', 6.months.ago)
208
    end
209

210
211
    def publicish(user)
      visibility_levels = [Project::PUBLIC]
212
      visibility_levels << Project::INTERNAL if user
213
214
      where(visibility_level: visibility_levels)
    end
215

216
    def with_push
217
      joins(:events).where('events.action = ?', Event::PUSHED)
218
219
    end

Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
220
    def active
221
      joins(:issues, :notes, :merge_requests).order('issues.created_at, notes.created_at, merge_requests.created_at DESC')
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
222
    end
223

224
    def search(query)
225
      joins(:namespace).
226
        where('LOWER(projects.name) LIKE :query OR
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
227
228
              LOWER(projects.path) LIKE :query OR
              LOWER(namespaces.name) LIKE :query OR
229
              LOWER(projects.description) LIKE :query',
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
230
              query: "%#{query.try(:downcase)}%")
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
231
    end
232

233
    def search_by_title(query)
234
      where('projects.archived = ?', false).where('LOWER(projects.name) LIKE :query', query: "%#{query.downcase}%")
235
236
    end

237
    def find_with_namespace(id)
238
      return nil unless id.include?('/')
239

240
      id = id.split('/')
241
      namespace = Namespace.by_path(id.first)
242
243
      return nil unless namespace

244
      where(namespace_id: namespace.id).where("LOWER(projects.path) = :path", path: id.second.downcase).first
245
    end
246

247
248
249
    def visibility_levels
      Gitlab::VisibilityLevel.options
    end
250
251

    def sort(method)
252
253
254
255
      if method == 'repository_size_desc'
        reorder(repository_size: :desc, id: :desc)
      else
        order_by(method)
256
257
      end
    end
258
259
260
261
262

    def reference_pattern
      name_pattern = Gitlab::Regex::NAMESPACE_REGEX_STR
      %r{(?<project>#{name_pattern}/#{name_pattern})}
    end
263
264
  end

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
265
  def team
266
    @team ||= ProjectTeam.new(self)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
267
268
269
  end

  def repository
270
271
272
    @repository ||= Repository.new(path_with_namespace, nil, self)
  end

273
  def commit(id = 'HEAD')
274
    repository.commit(id)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
275
276
  end

277
  def saved?
278
    id && persisted?
279
280
  end

281
282
283
284
  def schedule_add_import_job
    run_after_commit(:add_import_job)
  end

285
  def add_import_job
286
287
288
289
290
    if forked?
      unless RepositoryForkWorker.perform_async(id, forked_from_project.path_with_namespace, self.namespace.path)
        import_fail
      end
    else
291
      RepositoryImportWorker.perform_async(id)
292
    end
293
294
  end

295
  def clear_import_data
296
    self.import_data.destroy if self.import_data
297
298
  end

299
  def import?
300
301
302
303
    external_import? || forked?
  end

  def external_import?
304
305
306
    import_url.present?
  end

307
  def imported?
308
309
310
311
312
313
314
315
316
317
318
319
320
    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'
321
322
  end

323
  def check_limit
324
    unless creator.can_create_project? or namespace.kind == 'group'
Josh Frye's avatar
Josh Frye committed
325
      errors[:limit_reached] << ("Your project limit is #{creator.projects_limit} projects! Please contact your administrator to increase it")
326
327
    end
  rescue
Robert Speicher's avatar
Robert Speicher committed
328
    errors[:base] << ("Can't check your ability to create project")
gitlabhq's avatar
gitlabhq committed
329
330
  end

331
  def to_param
Vinnie Okada's avatar
Vinnie Okada committed
332
    path
333
334
  end

335
336
337
338
  def to_reference(_from_project = nil)
    path_with_namespace
  end

339
  def web_url
Douwe Maan's avatar
Douwe Maan committed
340
    Gitlab::Application.routes.url_helpers.namespace_project_url(self.namespace, self)
341
342
  end

343
  def web_url_without_protocol
344
    web_url.split('://')[1]
345
346
  end

347
  def build_commit_note(commit)
348
    notes.new(commit_id: commit.id, noteable_type: 'Commit')
gitlabhq's avatar
gitlabhq committed
349
  end
Nihad Abbasov's avatar
Nihad Abbasov committed
350

351
  def last_activity
352
    last_event
gitlabhq's avatar
gitlabhq committed
353
354
355
  end

  def last_activity_date
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
356
    last_activity_at || updated_at
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
357
  end
358

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
359
360
361
  def project_id
    self.id
  end
randx's avatar
randx committed
362

Robert Speicher's avatar
Robert Speicher committed
363
  def get_issue(issue_id)
364
    if default_issues_tracker?
Robert Speicher's avatar
Robert Speicher committed
365
      issues.find_by(iid: issue_id)
366
    else
Robert Speicher's avatar
Robert Speicher committed
367
      ExternalIssue.new(issue_id, self)
368
369
370
    end
  end

Robert Speicher's avatar
Robert Speicher committed
371
  def issue_exists?(issue_id)
372
    get_issue(issue_id)
Robert Speicher's avatar
Robert Speicher committed
373
374
  end

375
  def default_issue_tracker
376
    gitlab_issue_tracker_service || create_gitlab_issue_tracker_service
377
378
379
380
381
382
383
384
385
386
  end

  def issues_tracker
    if external_issue_tracker
      external_issue_tracker
    else
      default_issue_tracker
    end
  end

387
  def default_issues_tracker?
388
    !external_issue_tracker
389
390
391
  end

  def external_issues_trackers
392
    services.select(&:issue_tracker?).reject(&:default?)
393
394
395
396
397
398
  end

  def external_issue_tracker
    @external_issues_tracker ||= external_issues_trackers.select(&:activated?).first
  end

Andrew8xx8's avatar
Andrew8xx8 committed
399
  def can_have_issues_tracker_id?
400
    self.issues_enabled && !self.default_issues_tracker?
Andrew8xx8's avatar
Andrew8xx8 committed
401
402
  end

403
  def build_missing_services
404
405
    services_templates = Service.where(template: true)

406
    Service.available_services_names.each do |service_name|
407
      service = find_service(services, service_name)
408
409

      # If service is available but missing in db
410
411
412
413
414
415
      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`
416
          self.send :"create_#{service_name}_service"
417
418
419
420
        else
          Service.create_from_template(self.id, template)
        end
      end
421
422
423
    end
  end

Valery Sizov's avatar
Valery Sizov committed
424
425
426
427
428
429
430
431
432
  def create_labels
    Label.templates.each do |label|
      label = label.dup
      label.template = nil
      label.project_id = self.id
      label.save
    end
  end

433
434
435
  def find_service(list, name)
    list.find { |service| service.to_param == name }
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
436
437

  def gitlab_ci?
438
    gitlab_ci_service && gitlab_ci_service.active && gitlab_ci_project.present?
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
439
  end
440

441
442
443
444
445
  def ci_services
    services.select { |service| service.category == :ci }
  end

  def ci_service
446
    @ci_service ||= ci_services.select(&:activated?).first
447
448
  end

449
  def avatar_type
450
451
    unless self.avatar.image?
      self.errors.add :avatar, 'only images allowed'
452
453
454
455
456
457
458
459
460
461
    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

sue445's avatar
sue445 committed
462
463
464
465
  def avatar_url
    if avatar.present?
      [gitlab_config.url, avatar.url].join
    elsif avatar_in_git
Douwe Maan's avatar
Douwe Maan committed
466
      Gitlab::Application.routes.url_helpers.namespace_project_avatar_url(namespace, self)
sue445's avatar
sue445 committed
467
468
469
    end
  end

470
471
472
473
474
  # For compatibility with old code
  def code
    path
  end

475
  def items_for(entity)
476
477
478
479
480
481
482
    case entity
    when 'issue' then
      issues
    when 'merge_request' then
      merge_requests
    end
  end
483

484
485
  def send_move_instructions(old_path_with_namespace)
    NotificationService.new.project_was_moved(self, old_path_with_namespace)
486
  end
487
488

  def owner
489
490
    if group
      group
491
    else
492
      namespace.try(:owner)
493
494
    end
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
495

496
  def project_member_by_name_or_email(name = nil, email = nil)
497
    user = users.where('name like ? or email like ?', name, email).first
498
    project_members.where(user: user) if user
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
499
500
501
  end

  # Get Team Member record by user id
502
  def project_member_by_id(user_id)
503
    project_members.find_by(user_id: user_id)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
504
505
506
507
508
  end

  def name_with_namespace
    @name_with_namespace ||= begin
                               if namespace
509
                                 namespace.human_name + ' / ' + name
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
510
511
512
513
514
515
516
517
518
519
520
521
522
523
                               else
                                 name
                               end
                             end
  end

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

524
525
  def execute_hooks(data, hooks_scope = :push_hooks)
    hooks.send(hooks_scope).each do |hook|
526
      hook.async_execute(data, hooks_scope.to_s)
527
    end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
528
529
  end

530
531
532
  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|
533
      service.async_execute(data)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
534
535
536
537
    end
  end

  def update_merge_requests(oldrev, newrev, ref, user)
538
539
    MergeRequests::RefreshService.new(self, user).
      execute(oldrev, newrev, ref)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
540
541
542
  end

  def valid_repo?
543
    repository.exists?
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
544
  rescue
545
    errors.add(:path, 'Invalid repository path')
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
546
547
548
549
    false
  end

  def empty_repo?
550
    !repository.exists? || repository.empty?
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
551
552
553
  end

  def repo
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
554
    repository.raw
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
555
556
557
  end

  def url_to_repo
558
    gitlab_shell.url_to_repo(path_with_namespace)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
559
560
561
562
563
564
565
  end

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

  def repo_exists?
566
    @repo_exists ||= repository.exists?
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
567
568
569
570
571
  rescue
    @repo_exists = false
  end

  def open_branches
572
573
574
575
576
577
578
579
580
581
582
583
584
    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's avatar
Dmitriy Zaporozhets committed
585
586
587
  end

  def root_ref?(branch)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
588
    repository.root_ref == branch
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
589
590
591
592
593
594
595
  end

  def ssh_url_to_repo
    url_to_repo
  end

  def http_url_to_repo
Douwe Maan's avatar
Douwe Maan committed
596
    "#{web_url}.git"
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
597
598
599
  end

  # Check if current branch name is marked as protected in the system
600
  def protected_branch?(branch_name)
601
    protected_branches_names.include?(branch_name)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
602
  end
603

604
  def developers_can_push_to_protected_branch?(branch_name)
605
    protected_branches.any? { |pb| pb.name == branch_name && pb.developers_can_push }
606
607
  end

608
609
610
  def forked?
    !(forked_project_link.nil? || forked_project_link.forked_from_project.nil?)
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
611

612
613
614
615
  def personal?
    !group
  end

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
616
  def rename_repo
617
    path_was = previous_changes['path'].first
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
618
619
620
621
    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)
622
      # If repository moved successfully we need to send update instructions to users.
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
623
624
625
      # However we cannot allow rollback since we moved repository
      # So we basically we mute exceptions in next actions
      begin
626
        gitlab_shell.mv_repository("#{old_path_with_namespace}.wiki", "#{new_path_with_namespace}.wiki")
627
        send_move_instructions(old_path_with_namespace)
628
        reset_events_cache
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
629
      rescue
Johannes Schleifenbaum's avatar
Johannes Schleifenbaum committed
630
        # Returning false does not rollback after_* transaction but gives
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
631
632
633
634
635
636
637
638
639
        # 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
  end
640

Kirill Zaitsev's avatar
Kirill Zaitsev committed
641
642
643
644
645
  def hook_attrs
    {
      name: name,
      ssh_url: ssh_url_to_repo,
      http_url: http_url_to_repo,
Kirilll Zaitsev's avatar
Kirilll Zaitsev committed
646
      web_url: web_url,
Kirill Zaitsev's avatar
Kirill Zaitsev committed
647
648
649
650
651
      namespace: namespace.name,
      visibility_level: visibility_level
    }
  end

652
653
654
655
656
  # 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
657
  # * when the project avatar changes
658
659
660
661
662
663
664
665
666
  # 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
667
668

  def project_member(user)
669
    project_members.where(user_id: user).first
670
  end
671
672
673
674

  def default_branch
    @default_branch ||= repository.root_ref if repository.exists?
  end
675
676
677
678
679

  def reload_default_branch
    @default_branch = nil
    default_branch
  end
680

681
682
683
  def visibility_level_field
    visibility_level
  end
684
685
686
687
688
689
690
691

  def archive!
    update_attribute(:archived, true)
  end

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

693
694
695
696
  def change_head(branch)
    gitlab_shell.update_repository_head(self.path_with_namespace, branch)
    reload_default_branch
  end
697
698
699
700

  def forked_from?(project)
    forked? && project == forked_from_project
  end
701
702
703
704

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

706
707
708
709
  def update_commit_count
    update_attribute(:commit_count, repository.commit_count)
  end

710
711
712
  def forks_count
    ForkedProjectLink.where(forked_from_project_id: self.id).count
  end
Robert Schilling's avatar
Robert Schilling committed
713
714
715
716

  def find_label(name)
    labels.find_by(name: name)
  end
717
718
719
720

  def origin_merge_requests
    merge_requests.where(source_project_id: self.id)
  end
721
722

  def create_repository
723
724
    # Forked import is handled asynchronously
    unless forked?
725
726
727
      if gitlab_shell.add_repository(path_with_namespace)
        true
      else
728
        errors.add(:base, 'Failed to create repository via gitlab-shell')
729
730
        false
      end
731
732
733
734
735
736
737
738
739
740
    end
  end

  def repository_exists?
    !!repository.exists?
  end

  def create_wiki
    ProjectWiki.new(self, self.owner).wiki
    true
741
  rescue ProjectWiki::CouldNotCreateWikiError
742
    errors.add(:base, 'Failed create wiki')
743
744
    false
  end
745
746

  def ci_commit(sha)
Kamil Trzcinski's avatar
Kamil Trzcinski committed
747
748
749
750
751
    ci_commits.find_by(sha: sha)
  end

  def ensure_ci_commit(sha)
    ci_commit(sha) || ci_commits.create(sha: sha)
752
  end
753

754
755
  def ensure_gitlab_ci_project
    gitlab_ci_project || create_gitlab_ci_project
Kamil Trzcinski's avatar
Kamil Trzcinski committed
756
757
  end

758
  def enable_ci
759
760
761
762
    service = gitlab_ci_service || create_gitlab_ci_service
    service.active = true
    service.save
  end
gitlabhq's avatar
gitlabhq committed
763
end