namespace.rb 4,4 КБ
Newer Older
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
1
2
3
4
# == Schema Information
#
# Table name: namespaces
#
Andrew8xx8's avatar
Andrew8xx8 включено в состав коммита
5
6
7
#  id          :integer          not null, primary key
#  name        :string(255)      not null
#  path        :string(255)      not null
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
8
#  owner_id    :integer
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
9
10
#  created_at  :datetime
#  updated_at  :datetime
Andrew8xx8's avatar
Andrew8xx8 включено в состав коммита
11
#  type        :string(255)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
12
#  description :string(255)      default(""), not null
Steven Thonus's avatar
Steven Thonus включено в состав коммита
13
#  avatar      :string(255)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
14
#  public      :boolean          default(FALSE)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
15
16
#

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
17
class Namespace < ActiveRecord::Base
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
18
  include Sortable
19
20
  include Gitlab::ShellAdapter

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
21
  has_many :projects, dependent: :destroy
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
22
23
  belongs_to :owner, class_name: "User"

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
24
  validates :owner, presence: true, unless: ->(n) { n.type == "Group" }
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
25
26
27
  validates :name,
    presence: true, uniqueness: true,
    length: { within: 0..255 },
Douwe Maan's avatar
Douwe Maan включено в состав коммита
28
29
    format: { with: Gitlab::Regex.namespace_name_regex,
              message: Gitlab::Regex.namespace_name_regex_message }
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
30

Andrew8xx8's avatar
Andrew8xx8 включено в состав коммита
31
  validates :description, length: { within: 0..255 }
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
32
33
  validates :path,
    length: { within: 1..255 },
Robert Speicher's avatar
Robert Speicher включено в состав коммита
34
35
36
    namespace: true,
    presence: true,
    uniqueness: { case_sensitive: false }
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
37
38
39

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

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
40
  after_create :ensure_dir_exist
41
  after_update :move_dir, if: :path_changed?
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
42
  after_destroy :rm_dir
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
43

Andrew8xx8's avatar
Andrew8xx8 включено в состав коммита
44
  scope :root, -> { where('type IS NULL') }
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
45

Douwe Maan's avatar
Douwe Maan включено в состав коммита
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
  class << self
    def by_path(path)
      where('lower(path) = :value', value: path.downcase).first
    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

    def search(query)
      where("name LIKE :query OR path LIKE :query", query: "%#{query}%")
    end

    def clean_path(path)
Douwe Maan's avatar
Douwe Maan включено в состав коммита
61
      path = path.dup
Douwe Maan's avatar
Douwe Maan включено в состав коммита
62
      # Get the email username by removing everything after an `@` sign.
Douwe Maan's avatar
Douwe Maan включено в состав коммита
63
      path.gsub!(/@.*\z/,             "")
Douwe Maan's avatar
Douwe Maan включено в состав коммита
64
      # Usernames can't end in .git, so remove it.
Douwe Maan's avatar
Douwe Maan включено в состав коммита
65
      path.gsub!(/\.git\z/,           "")
Douwe Maan's avatar
Douwe Maan включено в состав коммита
66
      # Remove dashes at the start of the username.
J. Daniel Schmidt's avatar
J. Daniel Schmidt включено в состав коммита
67
      path.gsub!(/\A-+/,              "")
Douwe Maan's avatar
Douwe Maan включено в состав коммита
68
      # Remove periods at the end of the username.
J. Daniel Schmidt's avatar
J. Daniel Schmidt включено в состав коммита
69
      path.gsub!(/\.+\z/,             "")
Douwe Maan's avatar
Douwe Maan включено в состав коммита
70
      # Remove everything that's not in the list of allowed characters.
Douwe Maan's avatar
Douwe Maan включено в состав коммита
71
72
      path.gsub!(/[^a-zA-Z0-9_\-\.]/, "")

Douwe Maan's avatar
Douwe Maan включено в состав коммита
73
      # Users with the great usernames of "." or ".." would end up with a blank username.
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
74
      # Work around that by setting their username to "blank", followed by a counter.
Douwe Maan's avatar
Douwe Maan включено в состав коммита
75
76
      path = "blank" if path.blank?

Douwe Maan's avatar
Douwe Maan включено в состав коммита
77
78
      counter = 0
      base = path
Douwe Maan's avatar
Douwe Maan включено в состав коммита
79
      while Namespace.find_by_path_or_name(path)
Douwe Maan's avatar
Douwe Maan включено в состав коммита
80
81
82
83
84
85
        counter += 1
        path = "#{base}#{counter}"
      end

      path
    end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
86
87
  end

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
88
  def to_param
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
89
    path
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
90
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
91
92
93
94

  def human_name
    owner_name
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
95
96

  def ensure_dir_exist
97
    gitlab_shell.add_namespace(path)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
98
99
  end

100
  def rm_dir
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
101
102
103
104
    # Move namespace directory into trash.
    # We will remove it later async
    new_path = "#{path}+#{id}+deleted"

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
105
106
107
108
109
110
111
112
    if gitlab_shell.mv_namespace(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, new_path)
    end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
113
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
114
115

  def move_dir
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
116
117
118
    # Ensure old directory exists before moving it
    gitlab_shell.add_namespace(path_was)

119
    if gitlab_shell.mv_namespace(path_was, path)
Valery Sizov's avatar
Valery Sizov включено в состав коммита
120
121
      Gitlab::UploadsTransfer.new.rename_namespace(path_was, path)

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
122
123
      # If repositories moved successfully we need to
      # send update instructions to users.
124
125
      # However we cannot allow rollback since we moved namespace dir
      # So we basically we mute exceptions in next actions
GitLab's avatar
GitLab включено в состав коммита
126
      begin
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
127
        send_update_instructions
128
      rescue
Johannes Schleifenbaum's avatar
Johannes Schleifenbaum включено в состав коммита
129
        # Returning false does not rollback after_* transaction but gives
130
131
        # us information about failing some of tasks
        false
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
132
      end
133
134
135
136
    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('namespace directory cannot be moved')
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
137
    end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
138
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
139

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
140
  def send_update_instructions
Valery Sizov's avatar
Valery Sizov включено в состав коммита
141
142
143
    projects.each do |project|
      project.send_move_instructions("#{path_was}/#{project.path}")
    end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
144
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
145
146
147
148

  def kind
    type == 'Group' ? 'group' : 'user'
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
149
150
151
152

  def find_fork_of(project)
    projects.joins(:forked_project_link).where('forked_project_links.forked_from_project_id = ?', project.id).first
  end
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
153
end