client_management/app/models/site_server_db_backup.rb

class SiteServerDbBackup
  include Mongoid::Document
  include Mongoid::Timestamps
  PeriodsTypes=['day','week','month']
  belongs_to :site_server
  field :disable, type: Boolean, default: false
  field :backup_time, type: String, default: '03:00'
  field :path , type: String ,default: '/home/backup/db'
  field :period, type: Integer, default: 0 # 0 => daily , 1 => weekly , 2 => monthly
  field :retain_count, type: Integer, default: 7
  field :need_rewrite, type: Boolean, default: true
  
  before_save do
    unless @skip_callback
      if self.changed?
        self.need_rewrite = true
      end
    end
  end
  def to_crontab_script(return_string=true)
    # minute , hour, day(of month, 1~31), month(1~12), day(of week,0~6)
    crontab_time = ['*', '*', '*', '*', '*']
    hour, minute = self.backup_time.split(':').map{|s| s.to_i}
    crontab_time[0] = minute.to_s rescue '0'
    crontab_time[1] = hour.to_s rescue '0'
    if self.period == 1 #weekly
      crontab_time[4] = '0' #every Sunday
    elsif self.period == 2 #monthly
      crontab_time[2] = '1' #day 1 of every month
    end
    if self.retain_count
      retain_text = "1 week ago"
      if self.period != 0 || self.retain_count != 7 
        retain_text = "#{self.retain_count} #{PeriodsTypes[self.period]} ago"
      end
      scripts = ["#{crontab_time.join(' ')} mongodump -o #{self.path}/`date \"+\\%Y\\%m\\%d\"`_db"]
      if hour < 23
        next_hour = hour + 1
        crontab_time[1] = next_hour.to_s
      else
        crontab_time[1] = ((hour + 1) % 24).to_s
        if self.period == 1 #weekly
          crontab_time[4] = '1' #every Sunday + 1 = Monday
        elsif self.period == 2 #monthly
          crontab_time[2] = '2' #day 1 of every month + 1 = 2
        end
        retain_text += " 1 day ago"
      end
      scripts << "#{crontab_time.join(' ')} rm -fr #{self.path}/`date --date=\"#{retain_text}\" \"+\\%Y\\%m\\%d\"`_db"
    end
    if return_string
      scripts.join("\n")
    else
      scripts
    end
  end
  def self.read_from_crontab_line(crontab_line, crontab_lines)
    if crontab_line.include?('mongodump')
      tmp = crontab_line.split(/\s+[^\s]*mongodump\s+/) # ex: ["0 3 * * *", "-o /home/backup/db/`date \"+\\%Y\\%m\\%d\"`_db"]
      if tmp.count == 2
        crontab_time = tmp[0].split(/\s+/)
        path = File.dirname(tmp[1].split('-o')[1].strip) rescue nil
        if path
          if crontab_time[4] != '*' #weekly
            period = 1
            period_unit = 1.week
          elsif crontab_time[2] != '*' #monthly
            period = 2
            period_unit = 1.month
          else #daily
            period = 0
            period_unit = 1.day
          end
          time = "#{crontab_time[1].rjust(2, "0")}:#{crontab_time[0].rjust(2, "0")}"
          regex_path = /\s+#{::Regexp.escape(path)}\//
          match_line = crontab_lines.select{|l| l.match(/\s+(|\/usr\/bin\/)rm/) && l.match(regex_path)}.first
          if match_line.nil?
            retain_count = nil
          else
            retain_text = match_line.scan(/`\s*date\s+\-\-date=[\'\"]([^\'\"]+)[\'\"]/).flatten.first
            tmp_retain_count, tmp_period = retain_text.sub(/\s+ago/, '').split(/\s+/)
            tmp_retain_count = tmp_retain_count.to_i
            retain_count = (tmp_retain_count.send(tmp_period) / period_unit rescue nil)
          end
          return [time, path, period, retain_count]
        else
          return nil
        end
      else
        return nil
      end
    else
      return nil
    end
  end
  def self.clean_crontab_setting(crontab_lines_str)
    crontab_lines = crontab_lines_str.split("\n")
    related_path = []
    crontab_lines.each do |crontab_line|
      if crontab_line.include?('mongodump')
        tmp = crontab_line.split(/\s+[^\s]*mongodump\s+/) # ex: ["0 3 * * *", "-o /home/backup/db/`date \"+\\%Y\\%m\\%d\"`_db"] 
        output_dir = File.dirname(tmp[1].split('-o')[1].strip) rescue nil
        if output_dir
          related_path << output_dir
        end
      end
    end
    related_path_reg = Regexp.union(related_path.map{|s| /\s+#{::Regexp.escape(s)}\//})
    crontab_lines_str = crontab_lines.map do |crontab_line|
      if crontab_line.match(related_path_reg)
        nil
      else
        crontab_line
      end
    end.compact.join("\n")
  end
  def self.write_crontab_setting(site_server, crontab_lines_str)
    new_crontab_lines = self.where(:site_server_id=> site_server.id, :disable=>false).map{|db_backup| db_backup.to_crontab_script(false)}
    crontab_lines = crontab_lines_str.split("\n")
    exists_indices = new_crontab_lines.map do |sub_crontab_lines|
      match_reg = ::Regexp.new(sub_crontab_lines[0].gsub(/\s+/, '\s+'))
      crontab_lines.index{|l| l.match(match_reg)}
    end
    removed_related_path = []
    crontab_lines.each_with_index do |crontab_line, i|
      if crontab_line.include?('mongodump') && exists_indices.exclude?(i)
        tmp = crontab_line.split(/\s+[^\s]*mongodump\s+/) # ex: ["0 3 * * *", "-o /home/backup/db/`date \"+\\%Y\\%m\\%d\"`_db"] 
        output_dir = File.dirname(tmp[1].split('-o')[1].strip) rescue nil
        if output_dir
          removed_related_path << output_dir
        end
      end
    end
    removed_related_path.each_with_index do |path|
      regex_path = /\s+#{::Regexp.escape(path)}\//
      crontab_lines = crontab_lines.select{|l| l.match(regex_path).nil?}
    end
    exists_indices.each_with_index do |idx, i|
      if idx.nil?
        crontab_lines << new_crontab_lines[i]
      end
    end
    crontab_lines_str = crontab_lines.compact.join("\n")
  end
end