2016-11-29 13:21:25 -05:00
|
|
|
require 'rainbow/ext/string'
|
2017-11-17 07:27:16 -05:00
|
|
|
require 'gitlab/utils/strong_memoize'
|
2016-11-29 13:21:25 -05:00
|
|
|
|
|
|
|
module Gitlab
|
|
|
|
TaskFailedError = Class.new(StandardError)
|
|
|
|
TaskAbortedByUserError = Class.new(StandardError)
|
|
|
|
|
|
|
|
module TaskHelpers
|
2017-11-17 07:27:16 -05:00
|
|
|
include Gitlab::Utils::StrongMemoize
|
|
|
|
|
2017-08-08 11:14:29 -04:00
|
|
|
extend self
|
|
|
|
|
2016-11-29 13:21:25 -05:00
|
|
|
# Ask if the user wants to continue
|
|
|
|
#
|
|
|
|
# Returns "yes" the user chose to continue
|
|
|
|
# Raises Gitlab::TaskAbortedByUserError if the user chose *not* to continue
|
|
|
|
def ask_to_continue
|
|
|
|
answer = prompt("Do you want to continue (yes/no)? ".color(:blue), %w{yes no})
|
|
|
|
raise Gitlab::TaskAbortedByUserError unless answer == "yes"
|
|
|
|
end
|
|
|
|
|
|
|
|
# Check which OS is running
|
|
|
|
#
|
|
|
|
# It will primarily use lsb_relase to determine the OS.
|
|
|
|
# It has fallbacks to Debian, SuSE, OS X and systems running systemd.
|
|
|
|
def os_name
|
2017-02-22 13:18:40 -05:00
|
|
|
os_name = run_command(%w(lsb_release -irs))
|
|
|
|
os_name ||=
|
2017-02-23 10:27:36 -05:00
|
|
|
if File.readable?('/etc/system-release')
|
|
|
|
File.read('/etc/system-release')
|
|
|
|
elsif File.readable?('/etc/debian_version')
|
|
|
|
"Debian #{File.read('/etc/debian_version')}"
|
|
|
|
elsif File.readable?('/etc/SuSE-release')
|
|
|
|
File.read('/etc/SuSE-release')
|
|
|
|
elsif os_x_version = run_command(%w(sw_vers -productVersion))
|
2017-02-22 13:18:40 -05:00
|
|
|
"Mac OS X #{os_x_version}"
|
2017-02-23 10:27:36 -05:00
|
|
|
elsif File.readable?('/etc/os-release')
|
|
|
|
File.read('/etc/os-release').match(/PRETTY_NAME=\"(.+)\"/)[1]
|
2017-02-22 13:18:40 -05:00
|
|
|
end
|
2017-02-23 10:27:36 -05:00
|
|
|
|
2016-11-29 13:21:25 -05:00
|
|
|
os_name.try(:squish!)
|
|
|
|
end
|
|
|
|
|
|
|
|
# Prompt the user to input something
|
|
|
|
#
|
|
|
|
# message - the message to display before input
|
|
|
|
# choices - array of strings of acceptable answers or nil for any answer
|
|
|
|
#
|
|
|
|
# Returns the user's answer
|
|
|
|
def prompt(message, choices = nil)
|
|
|
|
begin
|
|
|
|
print(message)
|
|
|
|
answer = STDIN.gets.chomp
|
|
|
|
end while choices.present? && !choices.include?(answer)
|
|
|
|
answer
|
|
|
|
end
|
|
|
|
|
|
|
|
# Runs the given command and matches the output against the given pattern
|
|
|
|
#
|
|
|
|
# Returns nil if nothing matched
|
|
|
|
# Returns the MatchData if the pattern matched
|
|
|
|
#
|
|
|
|
# see also #run_command
|
|
|
|
# see also String#match
|
|
|
|
def run_and_match(command, regexp)
|
|
|
|
run_command(command).try(:match, regexp)
|
|
|
|
end
|
|
|
|
|
|
|
|
# Runs the given command
|
|
|
|
#
|
|
|
|
# Returns '' if the command was not found
|
|
|
|
# Returns the output of the command otherwise
|
|
|
|
#
|
|
|
|
# see also #run_and_match
|
|
|
|
def run_command(command)
|
|
|
|
output, _ = Gitlab::Popen.popen(command)
|
|
|
|
output
|
|
|
|
rescue Errno::ENOENT
|
|
|
|
'' # if the command does not exist, return an empty string
|
|
|
|
end
|
|
|
|
|
|
|
|
# Runs the given command and raises a Gitlab::TaskFailedError exception if
|
|
|
|
# the command does not exit with 0
|
|
|
|
#
|
|
|
|
# Returns the output of the command otherwise
|
|
|
|
def run_command!(command)
|
|
|
|
output, status = Gitlab::Popen.popen(command)
|
|
|
|
|
2017-03-14 12:48:37 -04:00
|
|
|
raise Gitlab::TaskFailedError.new(output) unless status.zero?
|
2016-11-29 13:21:25 -05:00
|
|
|
|
|
|
|
output
|
|
|
|
end
|
|
|
|
|
|
|
|
def uid_for(user_name)
|
|
|
|
run_command(%W(id -u #{user_name})).chomp.to_i
|
|
|
|
end
|
|
|
|
|
|
|
|
def gid_for(group_name)
|
|
|
|
begin
|
|
|
|
Etc.getgrnam(group_name).gid
|
|
|
|
rescue ArgumentError # no group
|
|
|
|
"group #{group_name} doesn't exist"
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2017-05-31 05:32:11 -04:00
|
|
|
def gitlab_user
|
|
|
|
Gitlab.config.gitlab.user
|
|
|
|
end
|
|
|
|
|
2017-08-24 12:58:09 -04:00
|
|
|
def gitlab_user?
|
2017-11-17 07:27:16 -05:00
|
|
|
strong_memoize(:is_gitlab_user) do
|
|
|
|
current_user = run_command(%w(whoami)).chomp
|
|
|
|
current_user == gitlab_user
|
|
|
|
end
|
2017-05-31 05:32:11 -04:00
|
|
|
end
|
|
|
|
|
2016-11-29 13:21:25 -05:00
|
|
|
def warn_user_is_not_gitlab
|
2017-11-17 07:27:16 -05:00
|
|
|
return if gitlab_user?
|
2017-05-30 13:06:58 -04:00
|
|
|
|
2017-11-17 07:27:16 -05:00
|
|
|
strong_memoize(:warned_user_not_gitlab) do
|
2017-02-22 13:18:40 -05:00
|
|
|
current_user = run_command(%w(whoami)).chomp
|
2017-05-30 13:06:58 -04:00
|
|
|
|
|
|
|
puts " Warning ".color(:black).background(:yellow)
|
|
|
|
puts " You are running as user #{current_user.color(:magenta)}, we hope you know what you are doing."
|
|
|
|
puts " Things may work\/fail for the wrong reasons."
|
|
|
|
puts " For correct results you should run this as user #{gitlab_user.color(:magenta)}."
|
|
|
|
puts ""
|
2016-11-29 13:21:25 -05:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def all_repos
|
2017-02-28 16:08:40 -05:00
|
|
|
Gitlab.config.repositories.storages.each_value do |repository_storage|
|
2017-12-21 09:42:25 -05:00
|
|
|
IO.popen(%W(find #{repository_storage['path']} -mindepth 2 -type d -name *.git)) do |find|
|
2016-11-29 13:21:25 -05:00
|
|
|
find.each_line do |path|
|
|
|
|
yield path.chomp
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def repository_storage_paths_args
|
2017-02-28 16:08:40 -05:00
|
|
|
Gitlab.config.repositories.storages.values.map { |rs| rs['path'] }
|
2016-11-29 13:21:25 -05:00
|
|
|
end
|
|
|
|
|
|
|
|
def user_home
|
|
|
|
Rails.env.test? ? Rails.root.join('tmp/tests') : Gitlab.config.gitlab.user_home
|
|
|
|
end
|
|
|
|
|
2017-04-10 19:15:48 -04:00
|
|
|
def checkout_or_clone_version(version:, repo:, target_dir:)
|
|
|
|
version =
|
|
|
|
if version.starts_with?("=")
|
|
|
|
version.sub(/\A=/, '') # tag or branch
|
|
|
|
else
|
|
|
|
"v#{version}" # tag
|
|
|
|
end
|
2016-11-29 13:21:25 -05:00
|
|
|
|
2017-04-10 19:15:48 -04:00
|
|
|
clone_repo(repo, target_dir) unless Dir.exist?(target_dir)
|
|
|
|
checkout_version(version, target_dir)
|
2016-11-29 13:21:25 -05:00
|
|
|
end
|
|
|
|
|
|
|
|
def clone_repo(repo, target_dir)
|
|
|
|
run_command!(%W[#{Gitlab.config.git.bin_path} clone -- #{repo} #{target_dir}])
|
|
|
|
end
|
|
|
|
|
2017-04-10 19:15:48 -04:00
|
|
|
def checkout_version(version, target_dir)
|
2017-07-20 17:30:29 -04:00
|
|
|
run_command!(%W[#{Gitlab.config.git.bin_path} -C #{target_dir} fetch --quiet origin #{version}])
|
|
|
|
run_command!(%W[#{Gitlab.config.git.bin_path} -C #{target_dir} checkout -f --quiet FETCH_HEAD --])
|
2016-11-29 13:21:25 -05:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|