Class SelBackup < Object

# File lib/selbackup.rb, line 28
def initialize(key, secret, bucket, region = 'eu-west-1')
  @access_key = key
  @access_secret = secret
  @bucket_name = bucket
  @region = region
end

# File lib/selbackup.rb, line 39
def bucket
  @bucket ||= connection.directories.get(bucket_name)
end

# File lib/selbackup.rb, line 56
def check_daily(filename, nb_day)
  return_value = false
  daily_directory = give_me_files(filename, "daily")
  if ! daily_good?(daily_directory, nb_day)
    weekly_directory = give_me_files(filename, "weekly")
    latest_daily_file = give_me_latest_file(daily_directory)
    recent_weekly_file = give_me_recent_file(weekly_directory)
    if weekly_directory == [] || should_weekly?(latest_daily_file.key, recent_weekly_file.key)
      weekly_file = gen_weekly_file({ key: latest_daily_file.key, body: latest_daily_file.body})
      push_file(weekly_file)
      return_value = true
    end
    delete_file(latest_daily_file)
  end
  return return_value
end

# File lib/selbackup.rb, line 83
def check_monthly(filename, nb_month)
  return_value = false
  monthly_directory = give_me_files(filename, "monthly")
  if ! monthly_good?(monthly_directory, nb_month)
    latest_monthly_file = give_me_latest_file(monthly_directory)
    delete_file(latest_monthly_file)
    return_value = true
  end
  return return_value
end

# File lib/selbackup.rb, line 107
def check_weekly(filename, nb_week)
  return_value = false
  weekly_directory = give_me_files(filename, "weekly")
  if ! weekly_good?(weekly_directory, nb_week)
    monthly_directory = give_me_files(filename, "monthly")
    latest_weekly_file = give_me_latest_file(weekly_directory)
    recent_monthly_file = give_me_recent_file(monthly_directory)
    if monthly_directory == [] || should_monthly?(latest_weekly_file.key, recent_monthly_file.key)
      monthly_file = gen_monthly_file({ key: latest_weekly_file.key, body: latest_weekly_file.body})
      push_file(monthly_file)
      return_value = true
    end
    delete_file(latest_weekly_file)
  end
  return return_value
end

# File lib/selbackup.rb, line 128
def connection
  @connection ||= Fog::Storage.new(provider: 'AWS', aws_access_key_id: access_key, aws_secret_access_key: access_secret, region: region)
end

# File lib/selbackup.rb, line 139
def daily_good?(directoryFiles, nb_day)
  return directoryFiles.length > nb_day ? false : true;
end

# File lib/selbackup.rb, line 149
def delete_file(file)
  file.destroy
end

# File lib/selbackup.rb, line 168
def directory(path)
  connection.directories.get(bucket_name, prefix: path)
end

# File lib/selbackup.rb, line 176
def files
  bucket.files
end

# File lib/selbackup.rb, line 249
def gen_daily_file(file)
  name = File.basename(file)
  if name =~ /[0-9]{4}-[0-9]{2}-[0-9]{2}(\.\w*)?$/
    return false
  elsif !(name =~ /[0-9]{4}-[0-9]{2}-[0-9]{2}-/)
    name = "#{Date.today.to_s}-#{name}"
  end
  myname = name.split(/[0-9]{4}-[0-9]{2}-[0-9]{2}-/).last
  mydate = name.split("-#{myname}").first
  key = "#{myname}/#{mydate}-daily-#{myname}"
  return { myname: myname, key: key, body: File.open(file) }
end

# File lib/selbackup.rb, line 268
def gen_monthly_file(file)
  if file[:key] =~ /[\w+.-]+\/[\w+.-]+-\d{4}-\d{2}-\d{2}-[\w+.-]+/
    monthly_name = file[:key].sub("/weekly-", "/monthy-")
  else
    monthly_name = file[:key].sub("-weekly-", "-monthly-")
  end
  return { key: monthly_name, body: file[:body] }
end

# File lib/selbackup.rb, line 283
def gen_weekly_file(file)
  if file[:key] =~ /[\w+.-]+\/[\w+.-]+-\d{4}-\d{2}-\d{2}-[\w+.-]+/
    weekly_name = file[:key].sub("/daily-", "/weekly-")
  else
    weekly_name = file[:key].sub("-daily-", "-weekly-")
  end
  return { key: weekly_name, body: file[:body] }
end

# File lib/selbackup.rb, line 188
def give_me_files(filename, type)
  full_directory = directory("#{filename}/")
  files = []
  full_directory.files.each do | file |
    if file.key =~ /#{filename}\/\d{4}-\d{2}-\d{2}-#{type}-[\w+.]+/ ||
       file.key =~ /#{filename}\/#{type}-\d{4}-\d{2}-\d{2}-[\w+.]+/
        files.push(file)
    end
  end
  return files
end

# File lib/selbackup.rb, line 206
def give_me_latest_file(files)
  latest_file = files.first
  files.each do | file |
    if Date.parse(file.key.gsub('-', '')) < Date.parse(latest_file.key.gsub('-', ''))
      latest_file = file
    end
  end
  return latest_file
end

# File lib/selbackup.rb, line 222
def give_me_recent_file(files)
  recent_file = files.first
  files.each do | file |
    if Date.parse(file.key.gsub('-', '')) > Date.parse(recent_file.key.gsub('-', ''))
      recent_file = file
    end
  end
  return recent_file
end

# File lib/selbackup.rb, line 239
def monthly_good?(files, nb_month)
  return files.length > nb_month ? false : true
end

# File lib/selbackup.rb, line 298
def push_file(file)
  if file[:key] =~ /[\w+.-]+\/[\w+.-]+-\d{4}-\d{2}-\d{2}-[\w+.-]+/
    filename = file[:key].split('/').first
    name_parts = file[:key].split('/').last.split('-')
    file[:key] = "#{filename}/#{name_parts[1]}-#{name_parts[2]}-#{name_parts[3]}-#{name_parts[0]}-#{filename}"
  end
  bucket.files.create(key: file[:key], body: file[:body])
end

# File lib/selbackup.rb, line 335
def should_monthly?(file1, file2)
  begin
    date1 = Date.parse(file1.gsub('-', '')).prev_month
    date2 = Date.parse(file2.gsub('-', ''))
  rescue
    print "Wrong date (Date.parse in should_monthly)."
    return false
  end
  return date1 >= date2 ? true : false
end

# File lib/selbackup.rb, line 315
def should_weekly?(file1, file2)
  begin
    date1 = Date.parse(file1.gsub('-', ''))
    date2 = Date.parse(file2.gsub('-', ''))
  rescue
    print "Wrong date (Date.parse in should_monthly)."
    return false
  end
  date_diff = date1 - date2
  return date_diff.abs >= 7 ? true : false
end

# File lib/selbackup.rb, line 353
def upload_file(file, nb_day = 7, nb_week = 4, nb_month = 3)
  daily_file = gen_daily_file(file)
  if daily_file == false
    return false
  end
  push_file(daily_file)
  if check_daily(daily_file[:myname], nb_day)
    if check_weekly(daily_file[:myname], nb_week)
      check_monthly(daily_file[:myname], nb_month)
    end
  end
end

# File lib/selbackup.rb, line 373
def weekly_good?(directoryFiles, nb_week)
  return directoryFiles.length > nb_week ? false : true
end