Commit 85b736c87855217fc9d6d133f17d8d35184301fa
Exists in
master
and in
29 other branches
Merge branch 'backup-restore' into 'master'
backup/restore rake tasks see commit message See merge request !469
Showing
1 changed file
with
105 additions
and
6 deletions
Show diff stats
lib/tasks/backup.rake
| 1 | -desc "Creates a backup of the user files stored in public/" | ||
| 2 | -task :backup do | ||
| 3 | - dirs = Dir.glob('public/images/[0-9][0-9][0-9][0-9]') + ['public/articles', 'public/thumbnails', 'public/user_themes'].select { |d| File.exists?(d) } | ||
| 4 | - tarball = 'backups/files-' + Time.now.strftime('%Y-%m-%d-%R') + '.tar' | 1 | +task :load_backup_config do |
| 2 | + $config = YAML.load_file('config/database.yml') | ||
| 3 | +end | ||
| 4 | + | ||
| 5 | +task :check_backup_support => :load_backup_config do | ||
| 6 | + if $config['production']['adapter'] != 'postgresql' | ||
| 7 | + fail("Only PostgreSQL is supported for backups at the moment") | ||
| 8 | + end | ||
| 9 | +end | ||
| 10 | + | ||
| 11 | +backup_dirs = [ | ||
| 12 | + 'public/image_uploads', | ||
| 13 | + 'public/articles', | ||
| 14 | + 'public/thumbnails', | ||
| 15 | + 'public/user_themes', | ||
| 16 | +] | ||
| 17 | + | ||
| 18 | +desc "Creates a backup of the database and uploaded files" | ||
| 19 | +task :backup => :check_backup_support do | ||
| 20 | + dirs = backup_dirs.select { |d| File.exists?(d) } | ||
| 21 | + | ||
| 22 | + backup_name = Time.now.strftime('%Y-%m-%d-%R') | ||
| 23 | + backup_file = File.join('tmp/backup', backup_name) + '.tar.gz' | ||
| 24 | + mkdir_p 'tmp/backup' | ||
| 25 | + dump = File.join('tmp/backup', backup_name) + '.sql' | ||
| 26 | + | ||
| 27 | + database = $config['production']['database'] | ||
| 28 | + sh "pg_dump #{database} > #{dump}" | ||
| 29 | + | ||
| 30 | + sh 'tar', 'caf', backup_file, dump, *dirs | ||
| 31 | + rm_f dump | ||
| 32 | + | ||
| 33 | + puts "****************************************************" | ||
| 34 | + puts "Backup in #{backup_file} !" | ||
| 35 | + puts | ||
| 36 | + puts "To restore, use:" | ||
| 37 | + puts "$ rake restore BACKUP=#{backup_file}" | ||
| 38 | + puts "****************************************************" | ||
| 39 | +end | ||
| 40 | + | ||
| 41 | +def invalid_backup!(message, items=[]) | ||
| 42 | + puts "E: #{message}" | ||
| 43 | + items.each do |i| | ||
| 44 | + puts "E: - #{i}" | ||
| 45 | + end | ||
| 46 | + puts "E: Is this a backup archive created by Noosfero with \`rake backup\`?" | ||
| 47 | + exit 1 | ||
| 48 | +end | ||
| 49 | + | ||
| 50 | +desc "Restores a backup created previousy with \`rake backup\`" | ||
| 51 | +task :restore => :check_backup_support do | ||
| 52 | + backup = ENV["BACKUP"] | ||
| 53 | + unless backup | ||
| 54 | + puts "usage: rake restore BACKUP=/path/to/backup" | ||
| 55 | + exit 1 | ||
| 56 | + end | ||
| 57 | + | ||
| 58 | + files = `tar taf #{backup}`.split | ||
| 59 | + | ||
| 60 | + # validate files in the backup | ||
| 61 | + invalid_files = [] | ||
| 62 | + files.each do |f| | ||
| 63 | + if f !~ /tmp\/backup\// && (backup_dirs.none? { |d| f =~ /^#{d}\// }) | ||
| 64 | + invalid_files << f | ||
| 65 | + end | ||
| 66 | + end | ||
| 67 | + if invalid_files.size > 0 | ||
| 68 | + invalid_backup!("Invalid files found in the backup archive", invalid_files) | ||
| 69 | + end | ||
| 70 | + | ||
| 71 | + # find database dump in the archive | ||
| 72 | + dumps = files.select do |f| | ||
| 73 | + File.dirname(f) == 'tmp/backup' && f =~ /\.sql$/ | ||
| 74 | + end | ||
| 75 | + if dumps.size == 0 | ||
| 76 | + invalid_backup!("Could not find a database dump in the archive.") | ||
| 77 | + elsif dumps.size > 1 | ||
| 78 | + invalid_backup!("Multiple database dumps found in the archive:", dumps) | ||
| 79 | + end | ||
| 80 | + dump = dumps.first | ||
| 81 | + | ||
| 82 | + database = $config['production']['database'] | ||
| 83 | + username = $config['production']['username'] | ||
| 84 | + | ||
| 85 | + puts "WARNING: backups should be restored to an empty database, otherwise" | ||
| 86 | + puts "data from the backup may not be loaded properly." | ||
| 87 | + puts | ||
| 88 | + puts 'You can remove the existing database and create a new one with:' | ||
| 89 | + puts | ||
| 90 | + puts "$ sudo -u postgres dropdb #{database}" | ||
| 91 | + puts "$ sudo -u postgres createdb #{database} --owner #{username}" | ||
| 92 | + puts | ||
| 93 | + print "Are you sure you want to continue (y/N)? " | ||
| 94 | + response = $stdin.gets.strip | ||
| 95 | + unless ['y', 'yes'].include?(response.downcase) | ||
| 96 | + puts "*** ABORTED." | ||
| 97 | + exit 1 | ||
| 98 | + end | ||
| 99 | + | ||
| 100 | + sh 'tar', 'xaf', backup | ||
| 101 | + sh "rails dbconsole production < #{dump}" | ||
| 102 | + rm_f dump | ||
| 5 | 103 | ||
| 6 | - mkdir_p(File.dirname(tarball)) | ||
| 7 | - sh('tar', 'cf', tarball, *dirs) | 104 | + puts "****************************************************" |
| 105 | + puts "Backup restored!" | ||
| 106 | + puts "****************************************************" | ||
| 8 | end | 107 | end |