project.rb 22.6 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
101
102
103
104
  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
105

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

128
  has_one :import_data, dependent: :destroy, class_name: "ProjectImportData"
129
  has_one :gitlab_ci_project, dependent: :destroy, class_name: "Ci::Project", foreign_key: :gitlab_id
130

131
  delegate :name, to: :owner, allow_nil: true, prefix: true
132
  delegate :members, to: :team, prefix: true
133

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

Douwe Maan's avatar
Douwe Maan committed
162
  mount_uploader :avatar, AvatarUploader
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
163

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

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

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

    event :import_finish do
186
      transition started: :finished
187
188
189
    end

    event :import_fail do
190
      transition started: :failed
191
192
193
    end

    event :import_retry do
194
      transition failed: :started
195
196
197
198
    end

    state :started
    state :finished
199
200
    state :failed

201
    after_transition any => :started, do: :schedule_add_import_job
202
    after_transition any => :finished, do: :clear_import_data
203
204
  end

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

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

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

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

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

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

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

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

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

257
258
259
    def visibility_levels
      Gitlab::VisibilityLevel.options
    end
260
261

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

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

    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
287
288
  end

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

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

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

301
  def saved?
302
    id && persisted?
303
304
  end

305
306
307
308
  def schedule_add_import_job
    run_after_commit(:add_import_job)
  end

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

319
  def clear_import_data
320
    self.import_data.destroy if self.import_data
321
322
  end

323
  def import?
324
325
326
327
    external_import? || forked?
  end

  def external_import?
328
329
330
    import_url.present?
  end

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

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

355
  def to_param
Vinnie Okada's avatar
Vinnie Okada committed
356
    path
357
358
  end

359
360
361
362
  def to_reference(_from_project = nil)
    path_with_namespace
  end

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

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

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

375
  def last_activity
376
    last_event
gitlabhq's avatar
gitlabhq committed
377
378
379
  end

  def last_activity_date
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
380
    last_activity_at || updated_at
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
381
  end
382

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
383
384
385
  def project_id
    self.id
  end
randx's avatar
randx committed
386

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

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

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

  def issues_tracker
    if external_issue_tracker
      external_issue_tracker
    else
      default_issue_tracker
    end
  end

411
  def default_issues_tracker?
412
    !external_issue_tracker
413
414
415
  end

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

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

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

427
  def build_missing_services
428
429
    services_templates = Service.where(template: true)

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

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

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

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

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

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

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

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

494
495
496
497
498
  # For compatibility with old code
  def code
    path
  end

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

508
509
  def send_move_instructions(old_path_with_namespace)
    NotificationService.new.project_was_moved(self, old_path_with_namespace)
510
  end
511
512

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

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

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

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

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

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

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

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

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

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

  def repo
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
578
    repository.raw
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
579
580
581
  end

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

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

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

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

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

  def ssh_url_to_repo
    url_to_repo
  end

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

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

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

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

636
637
638
639
  def personal?
    !group
  end

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

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

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

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

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

  def default_branch
    @default_branch ||= repository.root_ref if repository.exists?
  end
701
702
703
704
705

  def reload_default_branch
    @default_branch = nil
    default_branch
  end
706

707
708
709
  def visibility_level_field
    visibility_level
  end
710
711
712
713
714
715
716
717

  def archive!
    update_attribute(:archived, true)
  end

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

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

  def forked_from?(project)
    forked? && project == forked_from_project
  end
727
728
729
730

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

732
733
734
735
  def update_commit_count
    update_attribute(:commit_count, repository.commit_count)
  end

736
  def forks_count
737
738
739
    forks.count
  end

Robert Schilling's avatar
Robert Schilling committed
740
741
742
  def find_label(name)
    labels.find_by(name: name)
  end
743
744
745
746

  def origin_merge_requests
    merge_requests.where(source_project_id: self.id)
  end
747
748

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

  def repository_exists?
    !!repository.exists?
  end

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

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

  def ensure_ci_commit(sha)
    ci_commit(sha) || ci_commits.create(sha: sha)
778
  end
779

780
781
  def ensure_gitlab_ci_project
    gitlab_ci_project || create_gitlab_ci_project
Kamil Trzcinski's avatar
Kamil Trzcinski committed
782
783
  end

784
  def enable_ci
785
786
787
788
    service = gitlab_ci_service || create_gitlab_ci_service
    service.active = true
    service.save
  end
789
790
791

  def visibility_level_allowed?(level)
    return true unless forked?
Tomasz Maczukin's avatar
Tomasz Maczukin committed
792
    Gitlab::VisibilityLevel.allowed_fork_levels(forked_from_project.visibility_level).include?(level.to_i)
793
  end
gitlabhq's avatar
gitlabhq committed
794
end