openproject/lib/tasks/install.rake

namespace :redmine do
  namespace :backlogs do 

    desc "Install and configure Redmine Backlogs"
    task :install => :environment do |t|
      ENV["RAILS_ENV"] ||= "development"
      puts "\n"
      puts "====================================================="
      puts "             Redmine Backlogs Installer"
      puts "====================================================="
      puts "Installing to the #{ENV['RAILS_ENV']} environment."
      print "Fetching card labels from http://git.gnome.org..."
      STDOUT.flush
      Cards::TaskboardCards.fetch_labels
      print "done!\n"
      
      trackers = Tracker.find(:all)

      puts "Configuring story and task trackers..."
      invalid = true
      while invalid
        puts "-----------------------------------------------------"
        puts "Which trackers do you want to use for your stories?"
        trackers.each_with_index { |t, i| puts "  #{ i + 1 }. #{ t.name }" }
        print "Separate values with a space (e.g. 1 3): "
        STDOUT.flush
        selection = (STDIN.gets.chomp!).split(/\D+/)
        
        # Check that all values correspond to an items in the list
        invalid = false
        invalid_value = nil
        tracker_names = []
        selection.each do |s|
          if s.to_i > trackers.length
            invalid = true
            invalid_value = s
            break
          else
            tracker_names << trackers[s.to_i-1].name
          end
        end
        
        if invalid
          puts "Oooops! You entered an invalid value (#{invalid_value}). Please try again."
        else
          print "You selected the following trackers: #{tracker_names.join(', ')}. Is this correct? (y/n) "
          STDOUT.flush
          invalid = !(STDIN.gets.chomp!).match("y")
        end
      end

      settings = { :story_trackers => selection.map{ |s| trackers[s.to_i-1].id } }
      
      # Check if there is at least one tracker available
      puts "-----------------------------------------------------"
      if settings[:story_trackers].length < trackers.length
        invalid = true
        while invalid
          # If there's at least one, ask the user to pick one
          puts "Which tracker do you want to use for your tasks?"
          available_trackers = trackers.select{|t| !settings[:story_trackers].include? t.id}
          j = 0
          available_trackers.each_with_index { |t, i| puts "  #{ j = i + 1 }. #{ t.name }" }
          puts "  #{ j + 1 }. <<new>>"
          print "Choose one from above (or choose none to create a new tracker): "
          STDOUT.flush
          selection = (STDIN.gets.chomp!).split(/\D+/)
                
          if selection.length > 0 and selection.first.to_i <= available_trackers.length
            # If the user picked one, use that
            print "You selected #{available_trackers[selection.first.to_i-1].name}. Is this correct? (y/n) "
            STDOUT.flush
            if (STDIN.gets.chomp!).match("y")
              settings[:task_tracker] = available_trackers[selection.first.to_i-1].id
              invalid = false
            end
          elsif selection.length == 0 or selection.first.to_i == j + 1
            # If the user chose to create a new one, then ask for the name
            settings[:task_tracker] = create_new_tracker
            invalid = false
          else
            puts "Oooops! That's not a valid selection. Please try again."
          end
        end
      else
        # If there's none, ask to create one
        settings[:task_tracker] = create_new_tracker
      end

      Setting.plugin_redmine_backlogs = settings
      puts "Story and task trackers are now set."
      
      print "Migrating the database..."
      STDOUT.flush
      system('rake db:migrate_plugins --trace > redmine_backlogs_install.log')
      if $?==0
        puts "done!"
        puts "Installation complete. Please restart Redmine."
        puts "Thank you for trying out Redmine Backlogs!"
      else
        puts "ERROR!"
        puts "*******************************************************"
        puts " Whoa! An error occurred during database migration."
        puts " Please see redmine_backlogs_install.log for more info."
        puts "*******************************************************"
      end
    end
    
    def create_new_tracker
      repeat = true
      puts "Creating a new task tracker."
      while repeat
        print "Please type the tracker's name: "
        STDOUT.flush
        name = STDIN.gets.chomp!
        if Tracker.find(:first, :conditions => "name='#{name}'")
          puts "Ooops! That name is already taken."
          next
        end
        print "You typed '#{name}'. Is this correct? (y/n) "
        STDOUT.flush
        
        if (STDIN.gets.chomp!).match("y")
          tracker = Tracker.new(:name => name)
          tracker.save!
          repeat = false
        end
      end
      tracker.id
    end
  end
end