module Gitlab class SatelliteNotExistError < StandardError; end module Satellite class Satellite PARKING_BRANCH = "__parking_branch" attr_accessor :project def initialize(project) @project = project end def raise_no_satellite raise SatelliteNotExistError.new("Satellite doesn't exist") end def clear_and_update! raise_no_satellite unless exists? delete_heads! clear_working_dir! update_from_source! end def create create_cmd = "git clone #{project.url_to_repo} #{path}" if system(create_cmd) true else Gitlab::GitLogger.error("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 Rails.root.join("tmp", "repo_satellites", project.path_with_namespace) end def repo raise_no_satellite unless exists? @repo ||= Grit::Repo.new(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