BigW Consortium Gitlab

namespace.rb 5.25 KB
Newer Older
1
class Namespace < ActiveRecord::Base
2
  include Sortable
3 4
  include Gitlab::ShellAdapter

5
  has_many :projects, dependent: :destroy
6 7
  belongs_to :owner, class_name: "User"

8
  validates :owner, presence: true, unless: ->(n) { n.type == "Group" }
9 10
  validates :name,
    length: { within: 0..255 },
11 12 13
    namespace_name: true,
    presence: true,
    uniqueness: true
14

Andrew8xx8 committed
15
  validates :description, length: { within: 0..255 }
16 17
  validates :path,
    length: { within: 1..255 },
18 19 20
    namespace: true,
    presence: true,
    uniqueness: { case_sensitive: false }
21 22 23

  delegate :name, to: :owner, allow_nil: true, prefix: true

24
  after_update :move_dir, if: :path_changed?
25 26 27

  # Save the storage paths before the projects are destroyed to use them on after destroy
  before_destroy(prepend: true) { @old_repository_storage_paths = repository_storage_paths }
28
  after_destroy :rm_dir
29

30
  scope :root, -> { where('type IS NULL') }
31

32 33
  class << self
    def by_path(path)
34
      find_by('lower(path) = :value', value: path.downcase)
35 36 37 38 39 40 41
    end

    # Case insensetive search for namespace by path or name
    def find_by_path_or_name(path)
      find_by("lower(path) = :path OR lower(name) = :path", path: path.downcase)
    end

42 43 44 45 46 47 48
    # Searches for namespaces matching the given query.
    #
    # This method uses ILIKE on PostgreSQL and LIKE on MySQL.
    #
    # query - The search query as a String
    #
    # Returns an ActiveRecord::Relation
49
    def search(query)
50 51 52 53
      t = arel_table
      pattern = "%#{query}%"

      where(t[:name].matches(pattern).or(t[:path].matches(pattern)))
54 55 56
    end

    def clean_path(path)
57
      path = path.dup
58
      # Get the email username by removing everything after an `@` sign.
59
      path.gsub!(/@.*\z/,             "")
60
      # Usernames can't end in .git, so remove it.
61
      path.gsub!(/\.git\z/,           "")
62
      # Remove dashes at the start of the username.
63
      path.gsub!(/\A-+/,              "")
64
      # Remove periods at the end of the username.
65
      path.gsub!(/\.+\z/,             "")
66
      # Remove everything that's not in the list of allowed characters.
67 68
      path.gsub!(/[^a-zA-Z0-9_\-\.]/, "")

69
      # Users with the great usernames of "." or ".." would end up with a blank username.
70
      # Work around that by setting their username to "blank", followed by a counter.
71 72
      path = "blank" if path.blank?

73 74
      counter = 0
      base = path
75
      while Namespace.find_by_path_or_name(path)
76 77 78 79 80 81
        counter += 1
        path = "#{base}#{counter}"
      end

      path
    end
82 83
  end

84
  def to_param
85
    path
86
  end
87 88 89 90

  def human_name
    owner_name
  end
91

92
  def move_dir
93
    if any_project_has_container_registry_tags?
Kamil Trzcinski committed
94
      raise Exception.new('Namespace cannot be moved, because at least one project has tags in container registry')
95 96
    end

97 98 99 100 101 102 103 104 105
    # Move the namespace directory in all storages paths used by member projects
    repository_storage_paths.each do |repository_storage_path|
      # Ensure old directory exists before moving it
      gitlab_shell.add_namespace(repository_storage_path, path_was)

      unless gitlab_shell.mv_namespace(repository_storage_path, path_was, path)
        # 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('namespace directory cannot be moved')
106
      end
107 108 109 110 111 112 113 114 115 116 117 118 119 120
    end

    Gitlab::UploadsTransfer.new.rename_namespace(path_was, path)

    # If repositories moved successfully we need to
    # send update instructions to users.
    # However we cannot allow rollback since we moved namespace dir
    # So we basically we mute exceptions in next actions
    begin
      send_update_instructions
    rescue
      # Returning false does not rollback after_* transaction but gives
      # us information about failing some of tasks
      false
121
    end
122
  end
123

124
  def any_project_has_container_registry_tags?
Kamil Trzcinski committed
125
    projects.any?(&:has_container_registry_tags?)
126 127
  end

128
  def send_update_instructions
129 130 131
    projects.each do |project|
      project.send_move_instructions("#{path_was}/#{project.path}")
    end
132
  end
133 134 135 136

  def kind
    type == 'Group' ? 'group' : 'user'
  end
137 138

  def find_fork_of(project)
139
    projects.joins(:forked_project_link).find_by('forked_project_links.forked_from_project_id = ?', project.id)
140
  end
141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169

  private

  def repository_storage_paths
    # We need to get the storage paths for all the projects, even the ones that are
    # pending delete. Unscoping also get rids of the default order, which causes
    # problems with SELECT DISTINCT.
    Project.unscoped do
      projects.select('distinct(repository_storage)').to_a.map(&:repository_storage_path)
    end
  end

  def rm_dir
    # Remove the namespace directory in all storages paths used by member projects
    @old_repository_storage_paths.each do |repository_storage_path|
      # Move namespace directory into trash.
      # We will remove it later async
      new_path = "#{path}+#{id}+deleted"

      if gitlab_shell.mv_namespace(repository_storage_path, path, new_path)
        message = "Namespace directory \"#{path}\" moved to \"#{new_path}\""
        Gitlab::AppLogger.info message

        # Remove namespace directroy async with delay so
        # GitLab has time to remove all projects first
        GitlabShellWorker.perform_in(5.minutes, :rm_namespace, repository_storage_path, new_path)
      end
    end
  end
170
end