project.rb 21.4 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
WIP    
Kamil Trzcinski committed
121
  has_many :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
Kamil Trzcinski's avatar
WIP    
Kamil Trzcinski committed
122
  has_many :builds, through: :commits, dependent: :destroy, class_name: 'Ci::Build'
Kamil Trzcinski's avatar
WIP    
Kamil Trzcinski committed
123
  has_one :last_commit, -> { order 'ci_commits.created_at DESC' }, class_name: 'Ci::Commit', foreign_key: :gl_project_id
124

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

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

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

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

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

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

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

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

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

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

    state :started
    state :finished
196
197
    state :failed

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

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

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

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

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

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

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

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

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

241
      id = id.split('/')
242
243
244
245
      namespace = Namespace.find_by(path: id.first)
      return nil unless namespace

      where(namespace_id: namespace.id).find_by(path: id.second)
246
    end
247

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

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

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

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

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

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

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

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

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

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

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

  def external_import?
305
306
307
    import_url.present?
  end

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

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

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

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

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

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

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

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

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

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

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

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

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

  def issues_tracker
    if external_issue_tracker
      external_issue_tracker
    else
      default_issue_tracker
    end
  end

388
  def default_issues_tracker?
389
    !external_issue_tracker
390
391
392
  end

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

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

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

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

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

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

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

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

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

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

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

450
  def avatar_type
451
452
    unless self.avatar.image?
      self.errors.add :avatar, 'only images allowed'
453
454
455
456
457
458
459
460
461
462
    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
463
464
465
466
  def avatar_url
    if avatar.present?
      [gitlab_config.url, avatar.url].join
    elsif avatar_in_git
Douwe Maan's avatar
Douwe Maan committed
467
      Gitlab::Application.routes.url_helpers.namespace_project_avatar_url(namespace, self)
sue445's avatar
sue445 committed
468
469
470
    end
  end

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

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

  def send_move_instructions
486
    NotificationService.new.project_was_moved(self)
487
  end
488
489

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  def open_branches
573
574
575
576
577
578
579
580
581
582
583
584
585
    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
586
587
588
  end

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

  def ssh_url_to_repo
    url_to_repo
  end

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

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

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

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

613
614
615
616
  def personal?
    !group
  end

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

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

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

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

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

  def reload_default_branch
    @default_branch = nil
    default_branch
  end
681

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

  def archive!
    update_attribute(:archived, true)
  end

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

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

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

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

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

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

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

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

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

  def repository_exists?
    !!repository.exists?
  end

  def create_wiki
    ProjectWiki.new(self, self.owner).wiki
    true
  rescue ProjectWiki::CouldNotCreateWikiError => ex
743
    errors.add(:base, 'Failed create wiki')
744
745
    false
  end
746
747
748
749

  def ci_commit(sha)
    gitlab_ci_project.commits.find_by(sha: sha) if gitlab_ci?
  end
750
751
752
753
754
755
756
757

  def enable_ci(user)
    # Enable service
    service = gitlab_ci_service || create_gitlab_ci_service
    service.active = true
    service.save

    # Create Ci::Project
758
    Ci::CreateProjectService.new.execute(user, self)
759
  end
gitlabhq's avatar
gitlabhq committed
760
end