project.rb 22.3 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
43
  include CaseSensitivity
sue445's avatar
sue445 committed
44

45
  extend Gitlab::ConfigHelper
46
  extend Enumerize
47

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

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

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

64
  ActsAsTaggableOn.strict_case_match = true
65
  acts_as_taggable_on :tags
66

67
68
  attr_accessor :new_default_branch

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

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

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

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

102
  has_one :forked_from_project, through: :forked_project_link
103
  # Merge Requests for target project should be removed with it
104
  has_many :merge_requests,     dependent: :destroy, foreign_key: 'target_project_id'
105
  # Merge requests from source project should be kept when source project was removed
106
  has_many :fork_merge_requests, foreign_key: 'source_project_id', class_name: MergeRequest
107
  has_many :issues,             dependent: :destroy
108
  has_many :labels,             dependent: :destroy
109
110
  has_many :services,           dependent: :destroy
  has_many :events,             dependent: :destroy
111
112
  has_many :milestones,         dependent: :destroy
  has_many :notes,              dependent: :destroy
113
114
  has_many :snippets,           dependent: :destroy, class_name: 'ProjectSnippet'
  has_many :hooks,              dependent: :destroy, class_name: 'ProjectHook'
115
  has_many :protected_branches, dependent: :destroy
116
117
  has_many :project_members, dependent: :destroy, as: :source, class_name: 'ProjectMember'
  has_many :users, through: :project_members
118
119
  has_many :deploy_keys_projects, dependent: :destroy
  has_many :deploy_keys, through: :deploy_keys_projects
Ciro Santilli's avatar
Ciro Santilli committed
120
121
  has_many :users_star_projects, dependent: :destroy
  has_many :starrers, through: :users_star_projects, source: :user
122
  has_many :ci_commits, dependent: :destroy, class_name: 'Ci::Commit', foreign_key: :gl_project_id
Kamil Trzcinski's avatar
Kamil Trzcinski committed
123
  has_many :ci_builds, through: :ci_commits, source: :builds, dependent: :destroy, class_name: 'Ci::Build'
124
  has_many :releases, dependent: :destroy
125

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

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

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

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

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

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

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

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

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

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

    state :started
    state :finished
197
198
    state :failed

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

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

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

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

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

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

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

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

239
    def find_with_namespace(id)
240
241
242
243
244
245
246
      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).
247
      projects = unscoped.
248
        joins(:namespace).
249
250
        iwhere('namespaces.path' => namespace_path)

251
      projects.where('projects.path' => project_path).take ||
252
        projects.iwhere('projects.path' => project_path).take
253
    end
254

255
256
257
    def visibility_levels
      Gitlab::VisibilityLevel.options
    end
258
259

    def sort(method)
260
261
262
263
      if method == 'repository_size_desc'
        reorder(repository_size: :desc, id: :desc)
      else
        order_by(method)
264
265
      end
    end
266
267
268
269
270

    def reference_pattern
      name_pattern = Gitlab::Regex::NAMESPACE_REGEX_STR
      %r{(?<project>#{name_pattern}/#{name_pattern})}
    end
271
272
273
274
275
276
277
278
279
280
281
282
283
284

    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
285
286
  end

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
287
  def team
288
    @team ||= ProjectTeam.new(self)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
289
290
291
  end

  def repository
292
293
294
    @repository ||= Repository.new(path_with_namespace, nil, self)
  end

295
  def commit(id = 'HEAD')
296
    repository.commit(id)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
297
298
  end

299
  def saved?
300
    id && persisted?
301
302
  end

303
304
305
306
  def schedule_add_import_job
    run_after_commit(:add_import_job)
  end

307
  def add_import_job
308
309
310
311
312
    if forked?
      unless RepositoryForkWorker.perform_async(id, forked_from_project.path_with_namespace, self.namespace.path)
        import_fail
      end
    else
313
      RepositoryImportWorker.perform_async(id)
314
    end
315
316
  end

317
  def clear_import_data
318
    self.import_data.destroy if self.import_data
319
320
  end

321
  def import?
322
323
324
325
    external_import? || forked?
  end

  def external_import?
326
327
328
    import_url.present?
  end

329
  def imported?
330
331
332
333
334
335
336
337
338
339
340
341
342
    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'
343
344
  end

345
  def check_limit
346
    unless creator.can_create_project? or namespace.kind == 'group'
Josh Frye's avatar
Josh Frye committed
347
      errors[:limit_reached] << ("Your project limit is #{creator.projects_limit} projects! Please contact your administrator to increase it")
348
349
    end
  rescue
Robert Speicher's avatar
Robert Speicher committed
350
    errors[:base] << ("Can't check your ability to create project")
gitlabhq's avatar
gitlabhq committed
351
352
  end

353
  def to_param
Vinnie Okada's avatar
Vinnie Okada committed
354
    path
355
356
  end

357
358
359
360
  def to_reference(_from_project = nil)
    path_with_namespace
  end

361
  def web_url
Douwe Maan's avatar
Douwe Maan committed
362
    Gitlab::Application.routes.url_helpers.namespace_project_url(self.namespace, self)
363
364
  end

365
  def web_url_without_protocol
366
    web_url.split('://')[1]
367
368
  end

369
  def build_commit_note(commit)
370
    notes.new(commit_id: commit.id, noteable_type: 'Commit')
gitlabhq's avatar
gitlabhq committed
371
  end
Nihad Abbasov's avatar
Nihad Abbasov committed
372

373
  def last_activity
374
    last_event
gitlabhq's avatar
gitlabhq committed
375
376
377
  end

  def last_activity_date
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
378
    last_activity_at || updated_at
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
379
  end
380

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
381
382
383
  def project_id
    self.id
  end
randx's avatar
randx committed
384

Robert Speicher's avatar
Robert Speicher committed
385
  def get_issue(issue_id)
386
    if default_issues_tracker?
Robert Speicher's avatar
Robert Speicher committed
387
      issues.find_by(iid: issue_id)
388
    else
Robert Speicher's avatar
Robert Speicher committed
389
      ExternalIssue.new(issue_id, self)
390
391
392
    end
  end

Robert Speicher's avatar
Robert Speicher committed
393
  def issue_exists?(issue_id)
394
    get_issue(issue_id)
Robert Speicher's avatar
Robert Speicher committed
395
396
  end

397
  def default_issue_tracker
398
    gitlab_issue_tracker_service || create_gitlab_issue_tracker_service
399
400
401
402
403
404
405
406
407
408
  end

  def issues_tracker
    if external_issue_tracker
      external_issue_tracker
    else
      default_issue_tracker
    end
  end

409
  def default_issues_tracker?
410
    !external_issue_tracker
411
412
413
  end

  def external_issues_trackers
414
    services.select(&:issue_tracker?).reject(&:default?)
415
416
417
418
419
420
  end

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

Andrew8xx8's avatar
Andrew8xx8 committed
421
  def can_have_issues_tracker_id?
422
    self.issues_enabled && !self.default_issues_tracker?
Andrew8xx8's avatar
Andrew8xx8 committed
423
424
  end

425
  def build_missing_services
426
427
    services_templates = Service.where(template: true)

428
    Service.available_services_names.each do |service_name|
429
      service = find_service(services, service_name)
430
431

      # If service is available but missing in db
432
433
434
435
436
437
      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`
438
          self.send :"create_#{service_name}_service"
439
440
441
442
        else
          Service.create_from_template(self.id, template)
        end
      end
443
444
445
    end
  end

Valery Sizov's avatar
Valery Sizov committed
446
447
448
449
450
451
452
453
454
  def create_labels
    Label.templates.each do |label|
      label = label.dup
      label.template = nil
      label.project_id = self.id
      label.save
    end
  end

455
456
457
  def find_service(list, name)
    list.find { |service| service.to_param == name }
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
458
459

  def gitlab_ci?
460
    gitlab_ci_service && gitlab_ci_service.active && gitlab_ci_project.present?
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
461
  end
462

463
464
465
466
467
  def ci_services
    services.select { |service| service.category == :ci }
  end

  def ci_service
468
    @ci_service ||= ci_services.select(&:activated?).first
469
470
  end

471
  def avatar_type
472
473
    unless self.avatar.image?
      self.errors.add :avatar, 'only images allowed'
474
475
476
477
478
479
480
481
482
483
    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
484
485
486
487
  def avatar_url
    if avatar.present?
      [gitlab_config.url, avatar.url].join
    elsif avatar_in_git
Douwe Maan's avatar
Douwe Maan committed
488
      Gitlab::Application.routes.url_helpers.namespace_project_avatar_url(namespace, self)
sue445's avatar
sue445 committed
489
490
491
    end
  end

492
493
494
495
496
  # For compatibility with old code
  def code
    path
  end

497
  def items_for(entity)
498
499
500
501
502
503
504
    case entity
    when 'issue' then
      issues
    when 'merge_request' then
      merge_requests
    end
  end
505

506
507
  def send_move_instructions(old_path_with_namespace)
    NotificationService.new.project_was_moved(self, old_path_with_namespace)
508
  end
509
510

  def owner
511
512
    if group
      group
513
    else
514
      namespace.try(:owner)
515
516
    end
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
517

518
  def project_member_by_name_or_email(name = nil, email = nil)
519
    user = users.where('name like ? or email like ?', name, email).first
520
    project_members.where(user: user) if user
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
521
522
523
  end

  # Get Team Member record by user id
524
  def project_member_by_id(user_id)
525
    project_members.find_by(user_id: user_id)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
526
527
528
529
530
  end

  def name_with_namespace
    @name_with_namespace ||= begin
                               if namespace
531
                                 namespace.human_name + ' / ' + name
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
532
533
534
535
536
537
538
539
540
541
542
543
544
545
                               else
                                 name
                               end
                             end
  end

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

546
547
  def execute_hooks(data, hooks_scope = :push_hooks)
    hooks.send(hooks_scope).each do |hook|
548
      hook.async_execute(data, hooks_scope.to_s)
549
    end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
550
551
  end

552
553
554
  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|
555
      service.async_execute(data)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
556
557
558
559
    end
  end

  def update_merge_requests(oldrev, newrev, ref, user)
560
561
    MergeRequests::RefreshService.new(self, user).
      execute(oldrev, newrev, ref)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
562
563
564
  end

  def valid_repo?
565
    repository.exists?
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
566
  rescue
567
    errors.add(:path, 'Invalid repository path')
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
568
569
570
571
    false
  end

  def empty_repo?
572
    !repository.exists? || !repository.has_visible_content?
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
573
574
575
  end

  def repo
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
576
    repository.raw
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
577
578
579
  end

  def url_to_repo
580
    gitlab_shell.url_to_repo(path_with_namespace)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
581
582
583
584
585
586
587
  end

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

  def repo_exists?
588
    @repo_exists ||= repository.exists?
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
589
590
591
592
593
  rescue
    @repo_exists = false
  end

  def open_branches
594
595
596
597
598
599
600
601
602
603
604
605
606
    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
607
608
609
  end

  def root_ref?(branch)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
610
    repository.root_ref == branch
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
611
612
613
614
615
616
617
  end

  def ssh_url_to_repo
    url_to_repo
  end

  def http_url_to_repo
Douwe Maan's avatar
Douwe Maan committed
618
    "#{web_url}.git"
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
619
620
621
  end

  # Check if current branch name is marked as protected in the system
622
  def protected_branch?(branch_name)
623
    protected_branches_names.include?(branch_name)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
624
  end
625

626
  def developers_can_push_to_protected_branch?(branch_name)
627
    protected_branches.any? { |pb| pb.name == branch_name && pb.developers_can_push }
628
629
  end

630
631
632
  def forked?
    !(forked_project_link.nil? || forked_project_link.forked_from_project.nil?)
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
633

634
635
636
637
  def personal?
    !group
  end

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
638
  def rename_repo
639
    path_was = previous_changes['path'].first
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
640
641
642
643
    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)
644
      # If repository moved successfully we need to send update instructions to users.
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
645
646
647
      # However we cannot allow rollback since we moved repository
      # So we basically we mute exceptions in next actions
      begin
648
        gitlab_shell.mv_repository("#{old_path_with_namespace}.wiki", "#{new_path_with_namespace}.wiki")
649
        send_move_instructions(old_path_with_namespace)
650
        reset_events_cache
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
651
      rescue
Johannes Schleifenbaum's avatar
Johannes Schleifenbaum committed
652
        # Returning false does not rollback after_* transaction but gives
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
653
654
655
656
657
658
659
660
        # 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
661
662

    Gitlab::UploadsTransfer.new.rename_project(path_was, path, namespace.path)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
663
  end
664

Kirill Zaitsev's avatar
Kirill Zaitsev committed
665
666
667
668
669
  def hook_attrs
    {
      name: name,
      ssh_url: ssh_url_to_repo,
      http_url: http_url_to_repo,
Kirilll Zaitsev's avatar
Kirilll Zaitsev committed
670
      web_url: web_url,
Kirill Zaitsev's avatar
Kirill Zaitsev committed
671
672
673
674
675
      namespace: namespace.name,
      visibility_level: visibility_level
    }
  end

676
677
678
679
680
  # 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
681
  # * when the project avatar changes
682
683
684
685
686
687
688
689
690
  # 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
691
692

  def project_member(user)
693
    project_members.where(user_id: user).first
694
  end
695
696
697
698

  def default_branch
    @default_branch ||= repository.root_ref if repository.exists?
  end
699
700
701
702
703

  def reload_default_branch
    @default_branch = nil
    default_branch
  end
704

705
706
707
  def visibility_level_field
    visibility_level
  end
708
709
710
711
712
713
714
715

  def archive!
    update_attribute(:archived, true)
  end

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

717
718
719
720
  def change_head(branch)
    gitlab_shell.update_repository_head(self.path_with_namespace, branch)
    reload_default_branch
  end
721
722
723
724

  def forked_from?(project)
    forked? && project == forked_from_project
  end
725
726
727
728

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

730
731
732
733
  def update_commit_count
    update_attribute(:commit_count, repository.commit_count)
  end

734
735
736
  def forks_count
    ForkedProjectLink.where(forked_from_project_id: self.id).count
  end
Robert Schilling's avatar
Robert Schilling committed
737
738
739
740

  def find_label(name)
    labels.find_by(name: name)
  end
741
742
743
744

  def origin_merge_requests
    merge_requests.where(source_project_id: self.id)
  end
745
746

  def create_repository
747
748
    # Forked import is handled asynchronously
    unless forked?
749
750
751
      if gitlab_shell.add_repository(path_with_namespace)
        true
      else
752
        errors.add(:base, 'Failed to create repository via gitlab-shell')
753
754
        false
      end
755
756
757
758
759
760
761
762
763
764
    end
  end

  def repository_exists?
    !!repository.exists?
  end

  def create_wiki
    ProjectWiki.new(self, self.owner).wiki
    true
765
  rescue ProjectWiki::CouldNotCreateWikiError
766
    errors.add(:base, 'Failed create wiki')
767
768
    false
  end
769
770

  def ci_commit(sha)
Kamil Trzcinski's avatar
Kamil Trzcinski committed
771
772
773
774
775
    ci_commits.find_by(sha: sha)
  end

  def ensure_ci_commit(sha)
    ci_commit(sha) || ci_commits.create(sha: sha)
776
  end
777

778
779
  def ensure_gitlab_ci_project
    gitlab_ci_project || create_gitlab_ci_project
Kamil Trzcinski's avatar
Kamil Trzcinski committed
780
781
  end

782
  def enable_ci
783
784
785
786
    service = gitlab_ci_service || create_gitlab_ci_service
    service.active = true
    service.save
  end
gitlabhq's avatar
gitlabhq committed
787
end