Backport upgrader script from GitLab

Signed-off-by: Dmitriy Zaporozhets <dmitriy.zaporozhets@gmail.com>

4 files changed, 173 insertions, 0 deletions

diff --git a/lib/upgrader.rb b/lib/upgrader.rb
new file mode 100644
index 0000000..3dff6a8
--- /dev/null
+++ b/lib/upgrader.rb
@@ -0,0 +1,94 @@
+require_relative "version_info"
+
+class Upgrader
+  def execute
+    puts "GitLab CI #{current_version.major} upgrade tool"
+    puts "Your version is #{current_version}"
+    puts "Latest available version for GitLab CI #{current_version.major} is #{latest_version}"
+
+    if latest_version?
+      puts "You are using the latest GitLab CI 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 ||= VersionInfo.parse(current_version_raw)
+  end
+
+  def latest_version
+    @latest_version ||= 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 = `git ls-remote --tags origin | grep tags\/v#{current_version.major}`
+    git_tags = git_tags.lines.to_a.select { |version| version =~ /v\d\.\d\.\d\Z/ }
+    last_tag = git_tags.last.match(/v\d\.\d\.\d/).to_s
+  end
+
+  def update_commands
+    {
+      "Stash changed files" => "git stash",
+      "Get latest code" => "git fetch",
+      "Switch to new version" => "git checkout v#{latest_version}",
+      "Install gems" => "bundle",
+      "Migrate DB" => "bundle exec rake db:migrate RAILS_ENV=production",
+      "Recompile assets" => "bundle exec rake assets:clean assets:precompile RAILS_ENV=production",
+      "Clear cache" => "bundle exec rake cache:clear RAILS_ENV=production"
+    }
+  end
+
+  def upgrade
+    update_commands.each do |title, cmd|
+      puts title
+      puts " -> #{cmd}"
+      if system(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
diff --git a/lib/version_info.rb b/lib/version_info.rb
new file mode 100644
index 0000000..2a87c91
--- /dev/null
+++ b/lib/version_info.rb
@@ -0,0 +1,52 @@
+class VersionInfo
+  include Comparable
+
+  attr_reader :major, :minor, :patch
+
+  def self.parse(str)
+    if str && m = str.match(/(\d+)\.(\d+)\.(\d+)/)
+      VersionInfo.new(m[1].to_i, m[2].to_i, m[3].to_i)
+    else
+      VersionInfo.new
+    end
+  end
+
+  def initialize(major = 0, minor = 0, patch = 0)
+    @major = major
+    @minor = minor
+    @patch = patch
+  end
+
+  def <=>(other)
+    return unless other.is_a? VersionInfo
+    return unless valid? && other.valid?
+
+    if other.major < @major
+      1
+    elsif @major < other.major
+      -1
+    elsif other.minor < @minor
+      1
+    elsif @minor < other.minor
+      -1
+    elsif other.patch < @patch
+      1
+    elsif @patch < other.patch
+      -1
+    else
+      0
+    end
+  end
+
+  def to_s
+    if valid?
+      "%d.%d.%d" % [@major, @minor, @patch]
+    else
+      "Unknown"
+    end
+  end
+
+  def valid?
+    @major >= 0 && @minor >= 0 && @patch >= 0 && @major + @minor + @patch > 0
+  end
+end
diff --git a/script/upgrade.rb b/script/upgrade.rb
new file mode 100644
index 0000000..86cbf6e
--- /dev/null
+++ b/script/upgrade.rb
@@ -0,0 +1,3 @@
+require_relative "../lib/upgrader"
+
+Upgrader.new.execute
diff --git a/spec/lib/upgrader_spec.rb b/spec/lib/upgrader_spec.rb
new file mode 100644
index 0000000..9674907
--- /dev/null
+++ b/spec/lib/upgrader_spec.rb
@@ -0,0 +1,24 @@
+require 'spec_helper'
+
+describe Upgrader do
+  let(:upgrader) { Upgrader.new }
+  let(:current_version) { GitlabCi::VERSION }
+
+  describe 'current_version_raw' do
+    it { upgrader.current_version_raw.should == current_version }
+  end
+
+  describe 'latest_version?' do
+    it 'should be true if newest version' do
+      upgrader.stub(latest_version_raw: current_version)
+      upgrader.latest_version?.should be_true
+    end
+  end
+
+  describe 'latest_version_raw' do
+    it 'should be latest version for GitLab 5' do
+      upgrader.stub(current_version_raw: "3.0.0")
+      upgrader.latest_version_raw.should == "v3.2.0"
+    end
+  end
+end