rime-auto-deploy/lib/core.rb

module FontStylePatch
  RESET_COLOR = "\e[0m" #重置所有颜色和样式
  COLORS = {
    black: "\e[30m", #黑色文本
    red: "\e[31m", #红色文本
    green: "\e[32m", #绿色文本
    yellow: "\e[33m", #黄色文本
    blue: "\e[34m", #蓝色文本
    carmine: "\e[35m", #洋红色文本
    cyan: "\e[36m", #青色文本
    white: "\e[37m" #白色文本
  }
  COLORS.keys.each do |color_name|
    define_method(color_name) do
      return "#{COLORS[color_name]}#{self}#{RESET_COLOR}"
    end
  end
end

module StringPatch
  def nature_case
    self.gsub(/(.)([A-Z])/, '\1 \2').downcase.capitalize
  end
end

class String
  include FontStylePatch
  include StringPatch
end

module RimeDeploy
  class RimeDeployError < StandardError
  end

  class ChooseSession
    def initialize(items)
      @items = items #[[intro, method]] auto index
    end

    def call
      message = nil
      choose = nil
      while true
        puts @title
        puts "Choose mode:"
        @items.each_with_index do |item, index|
          puts "[#{index + 1}] " + item[0]
        end
        puts "Tips: input the index. e.g: 01"
        puts "Message: #{message}".red if message
        print ">>".green
        choose_mode = gets
        choose_index = choose_mode.strip
        if choose_index =~ /\d+/ && choose_index.to_i <= @items.length
          choose = @items[choose_index.to_i - 1]
          break
        else
          message = "Wrong Index, try again."
        end
      end

      choose[1].call
    end
  end

  class Job
    attr_accessor :status, :intro
    def initialize
      @status = :waiting # :waiting, :processing, :done, :fail
      klass_name = self.class.to_s.split("::").last.sub(/Job$/, "")
      @intro = klass_name.nature_case
    end

    def call
    end

    def rollback
    end
  end

  class JobGroup
    def initialize(jobs)
      @title = "=== Rime Deploy ===="
      @queue = []
      jobs.each { |job| @queue << job.new }

      @current_index = 0
    end

    def print_progress
      system("clear")
      puts @title
      @queue.each_with_index do |job, index|
        job_id = "[%02d]" % (index + 1)
        job_intro = job.intro.to_s.ljust(20).green
        job_status = job.status
        case job_status
        when :waiting
          job_status = job_status.to_s.white
        when :processing
          job_status = job_status.to_s.yellow
        when :done
          job_status = job_status.to_s.green
        end
        job_status = job_status.rjust(15)
        puts "#{job_id} #{job_intro}\t#{job_status}"
      end

      if @current_index < @queue.length
        puts "total: #{@queue.length}".ljust(10)
        puts "Detail " + "-" * 20
      end
    end

    def run_job_with_info_wrapper(current_job)
      print_progress
      begin
        result = current_job.call
        if result == :next
          current_job.status = :done
          @current_index += 1
        else
          # 失败处理
          raise RimeDeployError
        end
        print_progress
      rescue RimeDeployError
        run_jobs_handle
      end
    end

    def guidance
      ChooseSession.new(
        [
          [
            "Auto mode: Suitable for first-time operation.",
            -> { run_jobs_auto }
          ],
          [
            "Handle mode: Decide to execute on your own.",
            -> { run_jobs_handle }
          ]
        ]
      ).call
    end
    def call
      guidance
    end

    def run_jobs_handle
      handle_jobs = []
      @queue.each_with_index do |job, index|
        job_intro = job.intro.to_s.ljust(20).green
        handle_jobs.push(
          ["#{job_intro}", -> { run_job_with_info_wrapper(job) }]
        )
      end
      ChooseSession.new(handle_jobs).call
    end

    def run_jobs_auto
      print_progress
      while @current_index < @queue.length
        current_job = @queue[@current_index]
        current_job.status = :processing
        print_progress
        begin
          result = current_job.call
          if result == :next
            current_job.status = :done
            @current_index += 1
          else
            # 失败处理
            raise RimeDeployError
          end
          print_progress
        rescue RimeDeployError
          run_jobs_handle
        end
      end
    end
  end
end