120 lines
2.8 KiB
Ruby
120 lines
2.8 KiB
Ruby
module Gitlab
|
|
class SatelliteNotExistError < StandardError; end
|
|
|
|
module Satellite
|
|
class Satellite
|
|
include Gitlab::Popen
|
|
|
|
PARKING_BRANCH = "__parking_branch"
|
|
|
|
attr_accessor :project
|
|
|
|
def initialize(project)
|
|
@project = project
|
|
end
|
|
|
|
def log message
|
|
Gitlab::Satellite::Logger.error(message)
|
|
end
|
|
|
|
def raise_no_satellite
|
|
raise SatelliteNotExistError.new("Satellite doesn't exist")
|
|
end
|
|
|
|
def clear_and_update!
|
|
raise_no_satellite unless exists?
|
|
|
|
clear_working_dir!
|
|
delete_heads!
|
|
update_from_source!
|
|
end
|
|
|
|
def create
|
|
output, status = popen("git clone #{project.repository.path_to_repo} #{path}",
|
|
Gitlab.config.satellites.path)
|
|
|
|
log("PID: #{project.id}: git clone #{project.repository.path_to_repo} #{path}")
|
|
log("PID: #{project.id}: -> #{output}")
|
|
|
|
if status.zero?
|
|
true
|
|
else
|
|
log("Failed to create satellite for #{project.name_with_namespace}")
|
|
false
|
|
end
|
|
end
|
|
|
|
def exists?
|
|
File.exists? path
|
|
end
|
|
|
|
# * Locks the satellite
|
|
# * Changes the current directory to the satellite's working dir
|
|
# * Yields
|
|
def lock
|
|
raise_no_satellite unless exists?
|
|
|
|
File.open(lock_file, "w+") do |f|
|
|
f.flock(File::LOCK_EX)
|
|
|
|
Dir.chdir(path) do
|
|
return yield
|
|
end
|
|
end
|
|
end
|
|
|
|
def lock_file
|
|
Rails.root.join("tmp", "satellite_#{project.id}.lock")
|
|
end
|
|
|
|
def path
|
|
File.join(Gitlab.config.satellites.path, project.path_with_namespace)
|
|
end
|
|
|
|
def repo
|
|
raise_no_satellite unless exists?
|
|
|
|
@repo ||= Grit::Repo.new(path)
|
|
end
|
|
|
|
def destroy
|
|
FileUtils.rm_rf(path)
|
|
end
|
|
|
|
private
|
|
|
|
# Clear the working directory
|
|
def clear_working_dir!
|
|
repo.git.reset(hard: true)
|
|
end
|
|
|
|
# Deletes all branches except the parking branch
|
|
#
|
|
# This ensures we have no name clashes or issues updating branches when
|
|
# working with the satellite.
|
|
def delete_heads!
|
|
heads = repo.heads.map(&:name)
|
|
|
|
# update or create the parking branch
|
|
if heads.include? PARKING_BRANCH
|
|
repo.git.checkout({}, PARKING_BRANCH)
|
|
else
|
|
repo.git.checkout({b: true}, PARKING_BRANCH)
|
|
end
|
|
|
|
# remove the parking branch from the list of heads ...
|
|
heads.delete(PARKING_BRANCH)
|
|
# ... and delete all others
|
|
heads.each { |head| repo.git.branch({D: true}, head) }
|
|
end
|
|
|
|
# Updates the satellite from Gitolite
|
|
#
|
|
# Note: this will only update remote branches (i.e. origin/*)
|
|
def update_from_source!
|
|
repo.git.fetch({timeout: true}, :origin)
|
|
end
|
|
end
|
|
end
|
|
end
|