Commit c4299bb4 authored by Dmitriy Zaporozhets's avatar Dmitriy Zaporozhets

Move directory logic out of model. Use Gitlab:Shell class to interact with file system

parent 01033631
...@@ -13,6 +13,8 @@ ...@@ -13,6 +13,8 @@
# #
class Namespace < ActiveRecord::Base class Namespace < ActiveRecord::Base
include Gitlab::ShellAdapter
attr_accessible :name, :description, :path attr_accessible :name, :description, :path
has_many :projects, dependent: :destroy has_many :projects, dependent: :destroy
...@@ -31,7 +33,7 @@ class Namespace < ActiveRecord::Base ...@@ -31,7 +33,7 @@ class Namespace < ActiveRecord::Base
delegate :name, to: :owner, allow_nil: true, prefix: true delegate :name, to: :owner, allow_nil: true, prefix: true
after_create :ensure_dir_exist after_create :ensure_dir_exist
after_update :move_dir after_update :move_dir, if: :path_changed?
after_destroy :rm_dir after_destroy :rm_dir
scope :root, -> { where('type IS NULL') } scope :root, -> { where('type IS NULL') }
...@@ -53,48 +55,34 @@ class Namespace < ActiveRecord::Base ...@@ -53,48 +55,34 @@ class Namespace < ActiveRecord::Base
end end
def ensure_dir_exist def ensure_dir_exist
unless dir_exists? gitlab_shell.add_namespace(path)
FileUtils.mkdir( namespace_full_path, mode: 0770 )
end
end
def dir_exists?
File.exists?(namespace_full_path)
end end
def namespace_full_path def rm_dir
@namespace_full_path ||= File.join(Gitlab.config.gitlab_shell.repos_path, path) gitlab_shell.rm_namespace(path)
end end
def move_dir def move_dir
if path_changed? if gitlab_shell.mv_namespace(path_was, path)
old_path = File.join(Gitlab.config.gitlab_shell.repos_path, path_was) # If repositories moved successfully we need to remove old satellites
new_path = File.join(Gitlab.config.gitlab_shell.repos_path, path) # and send update instructions to users.
if File.exists?(new_path) # However we cannot allow rollback since we moved namespace dir
raise "Already exists" # So we basically we mute exceptions in next actions
end
begin begin
# Remove satellite when moving repo gitlab_shell.rm_satellites(path_was)
if path_was.present?
satellites_path = File.join(Gitlab.config.satellites.path, path_was)
FileUtils.rm_r( satellites_path, force: true )
end
FileUtils.mv( old_path, new_path )
send_update_instructions send_update_instructions
rescue Exception => e rescue
raise "Namespace move error #{old_path} #{new_path}" # Returning false does not rolback after_* transaction but gives
# us information about failing some of tasks
false
end 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('namespace directory cannot be moved')
end end
end end
def rm_dir
dir_path = File.join(Gitlab.config.gitlab_shell.repos_path, path)
FileUtils.rm_r( dir_path, force: true )
end
def send_update_instructions def send_update_instructions
projects.each(&:send_move_instructions) projects.each(&:send_move_instructions)
end end
......
...@@ -65,13 +65,72 @@ module Gitlab ...@@ -65,13 +65,72 @@ module Gitlab
system("#{gitlab_shell_user_home}/gitlab-shell/bin/gitlab-keys rm-key #{key_id} \"#{key_content}\"") system("#{gitlab_shell_user_home}/gitlab-shell/bin/gitlab-keys rm-key #{key_id} \"#{key_content}\"")
end end
# Add empty directory for storing repositories
#
# Ex.
# add_namespace("gitlab")
#
def add_namespace(name)
FileUtils.mkdir(full_path(name), mode: 0770) unless exists?(name)
end
# Remove directory from repositories storage
# Every repository inside this directory will be removed too
#
# Ex.
# rm_namespace("gitlab")
#
def rm_namespace(name)
FileUtils.rm_r(full_path(name), force: true)
end
# Move namespace directory inside repositories storage
#
# Ex.
# mv_namespace("gitlab", "gitlabhq")
#
def mv_namespace(old_name, new_name)
return false if exists?(new_name) || !exists?(old_name)
FileUtils.mv(full_path(old_name), full_path(new_name))
end
# Remove GitLab Satellites for provided path (namespace or repo dir)
#
# Ex.
# rm_satellites("gitlab")
#
# rm_satellites("gitlab/gitlab-ci.git")
#
def rm_satellites(path)
raise ArgumentError.new("Path can't be blank") if path.blank?
satellites_path = File.join(Gitlab.config.satellites.path, path)
FileUtils.rm_r(satellites_path, force: true)
end
def url_to_repo path def url_to_repo path
Gitlab.config.gitlab_shell.ssh_path_prefix + "#{path}.git" Gitlab.config.gitlab_shell.ssh_path_prefix + "#{path}.git"
end end
protected
def gitlab_shell_user_home def gitlab_shell_user_home
File.expand_path("~#{Gitlab.config.gitlab_shell.ssh_user}") File.expand_path("~#{Gitlab.config.gitlab_shell.ssh_user}")
end end
def repos_path
Gitlab.config.gitlab_shell.repos_path
end
def full_path(dir_name)
raise ArgumentError.new("Directory name can't be blank") if dir_name.blank?
File.join(repos_path, dir_name)
end
def exists?(dir_name)
File.exists?(full_path(dir_name))
end
end end
end end
...@@ -60,7 +60,7 @@ describe Namespace do ...@@ -60,7 +60,7 @@ describe Namespace do
end end
it "should raise error when dirtory exists" do it "should raise error when dirtory exists" do
expect { @namespace.move_dir }.to raise_error("Already exists") expect { @namespace.move_dir }.to raise_error("namespace directory cannot be moved")
end end
it "should move dir if path changed" do it "should move dir if path changed" do
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment