gitlab-org--gitlab-foss/lib/gitlab/upgrader.rb
2015-11-03 17:11:09 -05:00

107 lines
3 KiB
Ruby

require_relative "popen"
require_relative "version_info"
module Gitlab
class Upgrader
def execute
puts "GitLab #{current_version.major} upgrade tool"
puts "Your version is #{current_version}"
puts "Latest available version for GitLab #{current_version.major} is #{latest_version}"
if latest_version?
puts "You are using the latest GitLab version"
else
puts "Newer GitLab version is available"
answer = if ARGV.first == "-y"
"yes"
else
prompt("Do you want to upgrade (yes/no)? ", %w{yes no})
end
if answer == "yes"
upgrade
else
exit 0
end
end
end
def latest_version?
current_version >= latest_version
end
def current_version
@current_version ||= Gitlab::VersionInfo.parse(current_version_raw)
end
def latest_version
@latest_version ||= Gitlab::VersionInfo.parse(latest_version_raw)
end
def current_version_raw
File.read(File.join(gitlab_path, "VERSION")).strip
end
def latest_version_raw
git_tags = fetch_git_tags
git_tags = git_tags.select { |version| version =~ /v\d+\.\d+\.\d+\Z/ }
git_versions = git_tags.map { |tag| Gitlab::VersionInfo.parse(tag.match(/v\d+\.\d+\.\d+/).to_s) }
"v#{git_versions.sort.last.to_s}"
end
def fetch_git_tags
remote_tags, _ = Gitlab::Popen.popen(%W(#{Gitlab.config.git.bin_path} ls-remote --tags https://gitlab.com/gitlab-org/gitlab-ce.git))
remote_tags.split("\n").grep(/tags\/v#{current_version.major}/)
end
def update_commands
{
"Stash changed files" => %W(#{Gitlab.config.git.bin_path} stash),
"Get latest code" => %W(#{Gitlab.config.git.bin_path} fetch),
"Switch to new version" => %W(#{Gitlab.config.git.bin_path} checkout v#{latest_version}),
"Install gems" => %W(bundle),
"Migrate DB" => %W(bundle exec rake db:migrate),
"Recompile assets" => %W(bundle exec rake assets:clean assets:precompile),
"Clear cache" => %W(bundle exec rake cache:clear)
}
end
def env
{ 'RAILS_ENV' => 'production' }
end
def upgrade
update_commands.each do |title, cmd|
puts title
puts " -> #{cmd.join(' ')}"
if system(env, *cmd)
puts " -> OK"
else
puts " -> FAILED"
puts "Failed to upgrade. Try to repeat task or proceed with upgrade manually "
exit 1
end
end
puts "Done"
end
def gitlab_path
File.expand_path(File.join(File.dirname(__FILE__), '../..'))
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.include?(answer)
answer
end
end
end