project.rb 20.5 KB
Newer Older
1 2 3 4 5 6 7 8
# == Schema Information
#
# Table name: projects
#
#  id                     :integer          not null, primary key
#  name                   :string(255)
#  path                   :string(255)
#  description            :text
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
9 10
#  created_at             :datetime
#  updated_at             :datetime
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
11
#  creator_id             :integer
12 13 14 15
#  issues_enabled         :boolean          default(TRUE), not null
#  wall_enabled           :boolean          default(TRUE), not null
#  merge_requests_enabled :boolean          default(TRUE), not null
#  wiki_enabled           :boolean          default(TRUE), not null
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
16
#  namespace_id           :integer
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
17
#  issues_tracker         :string(255)      default("gitlab"), not null
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
18
#  issues_tracker_id      :string(255)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
19
#  snippets_enabled       :boolean          default(TRUE), not null
20
#  last_activity_at       :datetime
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
21
#  import_url             :string(255)
22
#  visibility_level       :integer          default(0), not null
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
23
#  archived               :boolean          default(FALSE), not null
Atsushi Ishida's avatar
Atsushi Ishida committed
24
#  avatar                 :string(255)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
25
#  import_status          :string(255)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
26 27
#  repository_size        :float            default(0.0)
#  star_count             :integer          default(0), not null
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
28 29
#  import_type            :string(255)
#  import_source          :string(255)
Atsushi Ishida's avatar
Atsushi Ishida committed
30
#  commit_count           :integer          default(0)
31 32
#

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

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

43
  extend Gitlab::ConfigHelper
44
  extend Enumerize
45

46
  default_value_for :archived, false
47 48 49 50
  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
51
  default_value_for :wall_enabled, false
52
  default_value_for :snippets_enabled, gitlab_config_features.snippets
53

54 55
  # set last_activity_at to the same as created_at
  after_create :set_last_activity_at
56
  def set_last_activity_at
57
    update_column(:last_activity_at, self.created_at)
58 59
  end

60
  ActsAsTaggableOn.strict_case_match = true
61
  acts_as_taggable_on :tags
62

63 64
  attr_accessor :new_default_branch

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

70
  has_one :last_event, -> {order 'events.created_at DESC'}, class_name: 'Event', foreign_key: 'project_id'
71 72 73

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

95
  has_one :forked_project_link, dependent: :destroy, foreign_key: "forked_to_project_id"
96

97
  has_one :forked_from_project, through: :forked_project_link
98
  # Merge Requests for target project should be removed with it
99
  has_many :merge_requests,     dependent: :destroy, foreign_key: 'target_project_id'
100
  # Merge requests from source project should be kept when source project was removed
101
  has_many :fork_merge_requests, foreign_key: 'source_project_id', class_name: MergeRequest
102
  has_many :issues,             dependent: :destroy
103
  has_many :labels,             dependent: :destroy
104 105
  has_many :services,           dependent: :destroy
  has_many :events,             dependent: :destroy
106 107
  has_many :milestones,         dependent: :destroy
  has_many :notes,              dependent: :destroy
108 109
  has_many :snippets,           dependent: :destroy, class_name: 'ProjectSnippet'
  has_many :hooks,              dependent: :destroy, class_name: 'ProjectHook'
110
  has_many :protected_branches, dependent: :destroy
111 112
  has_many :project_members, dependent: :destroy, as: :source, class_name: 'ProjectMember'
  has_many :users, through: :project_members
113 114
  has_many :deploy_keys_projects, dependent: :destroy
  has_many :deploy_keys, through: :deploy_keys_projects
Ciro Santilli's avatar
Ciro Santilli committed
115 116
  has_many :users_star_projects, dependent: :destroy
  has_many :starrers, through: :users_star_projects, source: :user
117

118 119
  has_one :import_data, dependent: :destroy, class_name: "ProjectImportData"

120
  delegate :name, to: :owner, allow_nil: true, prefix: true
121
  delegate :members, to: :team, prefix: true
122

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

Douwe Maan's avatar
Douwe Maan committed
151
  mount_uploader :avatar, AvatarUploader
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
152

153
  # Scopes
154
  scope :sorted_by_activity, -> { reorder(last_activity_at: :desc) }
155 156 157
  scope :sorted_by_stars, -> { reorder('projects.star_count DESC') }
  scope :sorted_by_names, -> { joins(:namespace).reorder('namespaces.name ASC, projects.name ASC') }

158 159 160
  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 ) }
161
  scope :in_namespace, ->(namespace_ids) { where(namespace_id: namespace_ids) }
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
162
  scope :in_group_namespace, -> { joins(:group) }
163
  scope :personal, ->(user) { where(namespace_id: user.namespace_id) }
164
  scope :joined, ->(user) { where('namespace_id != ?', user.namespace_id) }
165 166
  scope :public_only, -> { where(visibility_level: Project::PUBLIC) }
  scope :public_and_internal_only, -> { where(visibility_level: Project.public_and_internal_levels) }
167 168
  scope :non_archived, -> { where(archived: false) }

169 170
  state_machine :import_status, initial: :none do
    event :import_start do
171
      transition [:none, :finished] => :started
172 173 174
    end

    event :import_finish do
175
      transition started: :finished
176 177 178
    end

    event :import_fail do
179
      transition started: :failed
180 181 182
    end

    event :import_retry do
183
      transition failed: :started
184 185 186 187
    end

    state :started
    state :finished
188 189
    state :failed

190
    after_transition any => :started, do: :add_import_job
191
    after_transition any => :finished, do: :clear_import_data
192 193
  end

Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
194
  class << self
195 196 197 198
    def public_and_internal_levels
      [Project::PUBLIC, Project::INTERNAL]
    end

199
    def abandoned
200
      where('projects.last_activity_at < ?', 6.months.ago)
201
    end
202

203 204
    def publicish(user)
      visibility_levels = [Project::PUBLIC]
205
      visibility_levels << Project::INTERNAL if user
206 207
      where(visibility_level: visibility_levels)
    end
208

209
    def with_push
210
      joins(:events).where('events.action = ?', Event::PUSHED)
211 212
    end

Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
213
    def active
214
      joins(:issues, :notes, :merge_requests).order('issues.created_at, notes.created_at, merge_requests.created_at DESC')
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
215
    end
216

217
    def search(query)
218 219
      joins(:namespace).where('projects.archived = ?', false).
        where('LOWER(projects.name) LIKE :query OR
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
220 221
              LOWER(projects.path) LIKE :query OR
              LOWER(namespaces.name) LIKE :query OR
222
              LOWER(projects.description) LIKE :query',
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
223
              query: "%#{query.try(:downcase)}%")
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
224
    end
225

226
    def search_by_title(query)
227
      where('projects.archived = ?', false).where('LOWER(projects.name) LIKE :query', query: "%#{query.downcase}%")
228 229
    end

230
    def find_with_namespace(id)
231
      return nil unless id.include?('/')
232

233
      id = id.split('/')
234 235 236 237
      namespace = Namespace.find_by(path: id.first)
      return nil unless namespace

      where(namespace_id: namespace.id).find_by(path: id.second)
238
    end
239

240 241 242
    def visibility_levels
      Gitlab::VisibilityLevel.options
    end
243 244

    def sort(method)
245 246 247 248
      if method == 'repository_size_desc'
        reorder(repository_size: :desc, id: :desc)
      else
        order_by(method)
249 250
      end
    end
251 252 253 254 255

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

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
258
  def team
259
    @team ||= ProjectTeam.new(self)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
260 261 262
  end

  def repository
263 264 265
    @repository ||= Repository.new(path_with_namespace, nil, self)
  end

266
  def commit(id = 'HEAD')
267
    repository.commit(id)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
268 269
  end

270
  def saved?
271
    id && persisted?
272 273
  end

274 275 276 277
  def add_import_job
    RepositoryImportWorker.perform_in(2.seconds, id)
  end

278
  def clear_import_data
279
    self.import_data.destroy if self.import_data
280 281
  end

282 283 284 285
  def import?
    import_url.present?
  end

286
  def imported?
287 288 289 290 291 292 293 294 295 296 297 298 299
    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'
300 301
  end

302
  def check_limit
303
    unless creator.can_create_project? or namespace.kind == 'group'
304
      errors[:limit_reached] << ("Your project limit is #{creator.projects_limit} projects! Please contact your administrator to increase it")
305 306
    end
  rescue
Robert Speicher's avatar
Robert Speicher committed
307
    errors[:base] << ("Can't check your ability to create project")
gitlabhq's avatar
gitlabhq committed
308 309
  end

310
  def to_param
Vinnie Okada's avatar
Vinnie Okada committed
311
    path
312 313
  end

314 315 316 317
  def to_reference(_from_project = nil)
    path_with_namespace
  end

318
  def web_url
Douwe Maan's avatar
Douwe Maan committed
319
    Rails.application.routes.url_helpers.namespace_project_url(self.namespace, self)
320 321
  end

322
  def web_url_without_protocol
323
    web_url.split('://')[1]
324 325
  end

326
  def build_commit_note(commit)
327
    notes.new(commit_id: commit.id, noteable_type: 'Commit')
gitlabhq's avatar
gitlabhq committed
328
  end
Nihad Abbasov's avatar
Nihad Abbasov committed
329

330
  def last_activity
331
    last_event
gitlabhq's avatar
gitlabhq committed
332 333 334
  end

  def last_activity_date
Andrey Kumanyaev's avatar
Andrey Kumanyaev committed
335
    last_activity_at || updated_at
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
336
  end
337

338 339 340
  def project_id
    self.id
  end
randx's avatar
randx committed
341

Robert Speicher's avatar
Robert Speicher committed
342
  def get_issue(issue_id)
343
    if default_issues_tracker?
Robert Speicher's avatar
Robert Speicher committed
344
      issues.find_by(iid: issue_id)
345
    else
Robert Speicher's avatar
Robert Speicher committed
346
      ExternalIssue.new(issue_id, self)
347 348 349
    end
  end

Robert Speicher's avatar
Robert Speicher committed
350
  def issue_exists?(issue_id)
351
    get_issue(issue_id)
Robert Speicher's avatar
Robert Speicher committed
352 353
  end

354
  def default_issue_tracker
355
    gitlab_issue_tracker_service || create_gitlab_issue_tracker_service
356 357 358 359 360 361 362 363 364 365
  end

  def issues_tracker
    if external_issue_tracker
      external_issue_tracker
    else
      default_issue_tracker
    end
  end

366
  def default_issues_tracker?
367
    !external_issue_tracker
368 369 370
  end

  def external_issues_trackers
371
    services.select(&:issue_tracker?).reject(&:default?)
372 373 374 375 376 377
  end

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

Andrew8xx8's avatar
Andrew8xx8 committed
378
  def can_have_issues_tracker_id?
379
    self.issues_enabled && !self.default_issues_tracker?
Andrew8xx8's avatar
Andrew8xx8 committed
380 381
  end

382
  def build_missing_services
383 384
    services_templates = Service.where(template: true)

385
    Service.available_services_names.each do |service_name|
386
      service = find_service(services, service_name)
387 388

      # If service is available but missing in db
389 390 391 392 393 394 395 396 397 398 399
      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
400 401 402
    end
  end

403 404 405
  def find_service(list, name)
    list.find { |service| service.to_param == name }
  end
406 407 408 409

  def gitlab_ci?
    gitlab_ci_service && gitlab_ci_service.active
  end
410

411 412 413 414 415
  def ci_services
    services.select { |service| service.category == :ci }
  end

  def ci_service
416
    @ci_service ||= ci_services.select(&:activated?).first
417 418
  end

419
  def avatar_type
420 421
    unless self.avatar.image?
      self.errors.add :avatar, 'only images allowed'
422 423 424 425 426 427 428 429 430 431
    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
432 433 434 435
  def avatar_url
    if avatar.present?
      [gitlab_config.url, avatar.url].join
    elsif avatar_in_git
Douwe Maan's avatar
Douwe Maan committed
436
      Rails.application.routes.url_helpers.namespace_project_avatar_url(namespace, self)
sue445's avatar
sue445 committed
437 438 439
    end
  end

440 441 442 443 444
  # For compatibility with old code
  def code
    path
  end

445
  def items_for(entity)
446 447 448 449 450 451 452
    case entity
    when 'issue' then
      issues
    when 'merge_request' then
      merge_requests
    end
  end
453 454

  def send_move_instructions
455
    NotificationService.new.project_was_moved(self)
456
  end
457 458

  def owner
459 460
    if group
      group
461
    else
462
      namespace.try(:owner)
463 464
    end
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
465

466
  def project_member_by_name_or_email(name = nil, email = nil)
467
    user = users.where('name like ? or email like ?', name, email).first
468
    project_members.where(user: user) if user
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
469 470 471
  end

  # Get Team Member record by user id
472
  def project_member_by_id(user_id)
473
    project_members.find_by(user_id: user_id)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
474 475 476 477 478
  end

  def name_with_namespace
    @name_with_namespace ||= begin
                               if namespace
479
                                 namespace.human_name + ' / ' + name
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
480 481 482 483 484 485 486 487 488 489 490 491 492 493
                               else
                                 name
                               end
                             end
  end

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

494 495
  def execute_hooks(data, hooks_scope = :push_hooks)
    hooks.send(hooks_scope).each do |hook|
496
      hook.async_execute(data, hooks_scope.to_s)
497
    end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
498 499
  end

500 501 502
  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|
503
      service.async_execute(data)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
504 505 506 507
    end
  end

  def update_merge_requests(oldrev, newrev, ref, user)
508 509
    MergeRequests::RefreshService.new(self, user).
      execute(oldrev, newrev, ref)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
510 511 512
  end

  def valid_repo?
513
    repository.exists?
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
514
  rescue
515
    errors.add(:path, 'Invalid repository path')
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
516 517 518 519
    false
  end

  def empty_repo?
520
    !repository.exists? || repository.empty?
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
521 522
  end

523 524 525 526 527 528 529 530
  def ensure_satellite_exists
    self.satellite.create unless self.satellite.exists?
  end

  def satellite
    @satellite ||= Gitlab::Satellite::Satellite.new(self)
  end

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
531
  def repo
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
532
    repository.raw
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
533 534 535
  end

  def url_to_repo
536
    gitlab_shell.url_to_repo(path_with_namespace)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
537 538 539 540 541 542 543
  end

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

  def repo_exists?
544
    @repo_exists ||= repository.exists?
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
545 546 547 548 549
  rescue
    @repo_exists = false
  end

  def open_branches
550 551 552 553 554 555 556 557 558 559 560 561 562
    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
563 564 565
  end

  def root_ref?(branch)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
566
    repository.root_ref == branch
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
567 568 569 570 571 572 573
  end

  def ssh_url_to_repo
    url_to_repo
  end

  def http_url_to_repo
Douwe Maan's avatar
Douwe Maan committed
574
    "#{web_url}.git"
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
575 576 577
  end

  # Check if current branch name is marked as protected in the system
578
  def protected_branch?(branch_name)
579
    protected_branches_names.include?(branch_name)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
580
  end
581

582
  def developers_can_push_to_protected_branch?(branch_name)
583
    protected_branches.any? { |pb| pb.name == branch_name && pb.developers_can_push }
584 585
  end

586 587 588
  def forked?
    !(forked_project_link.nil? || forked_project_link.forked_from_project.nil?)
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
589

590 591 592 593
  def personal?
    !group
  end

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
594
  def rename_repo
595
    path_was = previous_changes['path'].first
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
596 597 598 599
    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)
600 601
      # If repository moved successfully we need to remove old satellite
      # and send update instructions to users.
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
602 603 604
      # However we cannot allow rollback since we moved repository
      # So we basically we mute exceptions in next actions
      begin
605
        gitlab_shell.mv_repository("#{old_path_with_namespace}.wiki", "#{new_path_with_namespace}.wiki")
606 607
        gitlab_shell.rm_satellites(old_path_with_namespace)
        ensure_satellite_exists
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
608
        send_move_instructions
609
        reset_events_cache
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
610
      rescue
Johannes Schleifenbaum's avatar
Johannes Schleifenbaum committed
611
        # Returning false does not rollback after_* transaction but gives
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
612 613 614 615 616 617 618 619 620
        # 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
621

Kirill Zaitsev's avatar
Kirill Zaitsev committed
622 623 624 625 626 627 628 629 630 631
  def hook_attrs
    {
      name: name,
      ssh_url: ssh_url_to_repo,
      http_url: http_url_to_repo,
      namespace: namespace.name,
      visibility_level: visibility_level
    }
  end

632 633 634 635 636
  # 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
637
  # * when the project avatar changes
638 639 640 641 642 643 644 645 646
  # 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
647 648

  def project_member(user)
649
    project_members.where(user_id: user).first
650
  end
651 652 653 654

  def default_branch
    @default_branch ||= repository.root_ref if repository.exists?
  end
655 656 657 658 659

  def reload_default_branch
    @default_branch = nil
    default_branch
  end
660

661 662 663
  def visibility_level_field
    visibility_level
  end
664 665 666 667 668 669 670 671

  def archive!
    update_attribute(:archived, true)
  end

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

673 674 675 676
  def change_head(branch)
    gitlab_shell.update_repository_head(self.path_with_namespace, branch)
    reload_default_branch
  end
677 678 679 680

  def forked_from?(project)
    forked? && project == forked_from_project
  end
681 682 683 684

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

686 687 688 689
  def update_commit_count
    update_attribute(:commit_count, repository.commit_count)
  end

690 691 692
  def forks_count
    ForkedProjectLink.where(forked_from_project_id: self.id).count
  end
693 694 695 696

  def find_label(name)
    labels.find_by(name: name)
  end
697 698 699 700

  def origin_merge_requests
    merge_requests.where(source_project_id: self.id)
  end
701 702

  def create_repository
703 704
    if forked?
      if gitlab_shell.fork_repository(forked_from_project.path_with_namespace, self.namespace.path)
705
        ensure_satellite_exists
706 707
        true
      else
708
        errors.add(:base, 'Failed to fork repository via gitlab-shell')
709 710
        false
      end
711
    else
712 713 714
      if gitlab_shell.add_repository(path_with_namespace)
        true
      else
715
        errors.add(:base, 'Failed to create repository via gitlab-shell')
716 717
        false
      end
718 719 720 721 722 723 724 725 726 727 728
    end
  end

  def repository_exists?
    !!repository.exists?
  end

  def create_wiki
    ProjectWiki.new(self, self.owner).wiki
    true
  rescue ProjectWiki::CouldNotCreateWikiError => ex
729
    errors.add(:base, 'Failed create wiki')
730 731
    false
  end
gitlabhq's avatar
gitlabhq committed
732
end