summaryrefslogtreecommitdiff
path: root/lib/backup/uploads.rb
blob: 75d8e18a862ad0eda5f0937817fc6eb50f0d7aff (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
module Backup
  class Uploads
    attr_reader :app_public_uploads_dir, :app_private_uploads_dir, :backup_public_uploads_dir,
     :backup_private_uploads_dir, :backup_dir, :backup_public_dir

    def initialize
      @app_public_uploads_dir = File.realpath(Rails.root.join('public', 'uploads'))
      @app_private_uploads_dir = File.realpath(Rails.root.join('uploads'))
      @backup_dir = Gitlab.config.backup.path
      @backup_public_dir = File.join(backup_dir, 'public')
      @backup_public_uploads_dir = File.join(backup_dir, 'public', 'uploads')
      @backup_private_uploads_dir = File.join(backup_dir, 'uploads')
    end

    # Copy uploads from public/uploads to backup/public/uploads and from /uploads to backup/uploads
    def dump
      FileUtils.mkdir_p(backup_public_uploads_dir)
      FileUtils.cp_r(app_public_uploads_dir, backup_public_dir)
      
      FileUtils.mkdir_p(backup_private_uploads_dir)
      FileUtils.cp_r(app_private_uploads_dir, backup_dir)
    end

    def restore
      backup_existing_public_uploads_dir
      backup_existing_private_uploads_dir

      FileUtils.cp_r(backup_public_uploads_dir, app_public_uploads_dir)
      FileUtils.cp_r(backup_private_uploads_dir, app_private_uploads_dir)
    end

    def backup_existing_public_uploads_dir
      timestamped_public_uploads_path = File.join(app_public_uploads_dir, '..', "uploads.#{Time.now.to_i}")
      if File.exists?(app_public_uploads_dir)
        FileUtils.mv(app_public_uploads_dir, timestamped_public_uploads_path)
      end
    end

    def backup_existing_private_uploads_dir
      timestamped_private_uploads_path = File.join(app_private_uploads_dir, '..', "uploads.#{Time.now.to_i}")
      if File.exists?(app_private_uploads_dir)
        FileUtils.mv(app_private_uploads_dir, timestamped_private_uploads_path)
      end
    end
  end
end