backup revision 5efadb4662f2a63d5f5f1a5b303ab7c3371069a8
286N/A#!/usr/bin/env ruby
286N/A
286N/Arequire 'fileutils'
286N/Arequire 'pathname'
286N/A
286N/Amodule Backup
286N/A class Backup
286N/A # Amount of backups that have to be there at least
286N/A BACKUPS_COUNT = 3
286N/A # Backups are valid for 7 days
286N/A BACKUPS_VALIDITY_TIME = 7 * 60 * 60 * 24
286N/A
286N/A MAINTENANCE_FILE = 'maintenance.txt'
286N/A
286N/A SQL_DUMP_FILE = 'ontohub_sql_dump.postgresql'
286N/A REPOSITORY_FILE = 'ontohub_repositories.tar.gz'
286N/A
286N/A DATA_DIRS = %w(data/repositories data/git_daemon)
286N/A
286N/A attr_reader :db_name, :data_root, :backup_root, :backup_instance_dir
286N/A attr_reader :dry_run, :verbose, :sql_dump_without_su
286N/A
286N/A def initialize(db_name, data_root, backup_root,
286N/A verbose: false, dry_run: true, sql_dump_without_su: false)
286N/A @db_name = db_name
286N/A @data_root = Pathname.new(data_root)
286N/A @backup_root = Pathname.new(backup_root)
286N/A
286N/A @dry_run = dry_run
286N/A @verbose = verbose
286N/A @sql_dump_without_su = sql_dump_without_su
286N/A end
286N/A
286N/A def create
286N/A enable_maintenance_mode
286N/A initialize_backup
286N/A create_sql_dump
286N/A create_repository_archive
286N/A # We needed to create the directory for the script to continue later on.
286N/A Dir.rmdir(backup_instance_dir) if dry_run
286N/A disable_maintenance_mode
286N/A puts "created backup in #{backup_instance_dir}"
286N/A self.class.prune(backup_root)
286N/A end
286N/A
286N/A def restore(backup_name)
286N/A enable_maintenance_mode
286N/A initialize_restore(backup_name)
286N/A restore_sql_dump
286N/A restore_repository_archive
286N/A disable_maintenance_mode
286N/A puts "restored backup from #{backup_instance_dir}"
286N/A end
286N/A
286N/A def self.prune(backup_root)
286N/A if !Dir.exists?(backup_root)
286N/A puts "Nothing to prune: There is no backup directory."
286N/A return
286N/A end
286N/A now = Time.now
286N/A backup_dirs_allowed_to_delete(Dir.new(backup_root).entries).each do |dir|
286N/A backup = backup_root.join(dir)
286N/A if now - File.new(backup).ctime > BACKUPS_VALIDITY_TIME
286N/A puts "removing old backup: #{dir}"
286N/A FileUtils.rm_r(backup)
286N/A end
286N/A end
286N/A end
286N/A
286N/A protected
286N/A
286N/A def new_backup_name
286N/A Time.now.strftime("%Y-%m-%d_%H-%M-%S")
286N/A end
286N/A
286N/A def initialize_backup
286N/A @backup_instance_dir = backup_root.join(new_backup_name)
286N/A puts "FileUtils.mkdir_p #{backup_instance_dir}" if verbose
286N/A # Create directory even in dry run to let the script continue.
286N/A FileUtils.mkdir_p(backup_instance_dir)
286N/A end
286N/A
286N/A def create_sql_dump
286N/A Dir.chdir(backup_instance_dir) do
286N/A if sql_dump_without_su
286N/A exec "pg_dump -U postgres -Fc #{db_name} > #{SQL_DUMP_FILE}"
286N/A else
286N/A exec "sudo -u postgres pg_dump -Fc #{db_name} > #{SQL_DUMP_FILE}"
286N/A end
286N/A end
286N/A end
286N/A
286N/A def create_repository_archive
286N/A Dir.chdir(data_root.join('..')) do
286N/A archive_file = backup_instance_dir.join(REPOSITORY_FILE)
286N/A exec(["tar -czvf #{archive_file}", *DATA_DIRS].join(' '))
286N/A end
286N/A end
286N/A
286N/A def initialize_restore(backup_name)
286N/A @backup_instance_dir = backup_root.join(backup_name)
286N/A unless Dir.exists?(backup_instance_dir)
286N/A puts "Error: Backup '#{backup_name}' does not exist in #{backup_root}."
286N/A exit
286N/A end
286N/A end
286N/A
286N/A def restore_sql_dump
286N/A Dir.chdir(backup_instance_dir) do
286N/A if sql_dump_without_su
286N/A exec "pg_restore -c -U postgres -d #{db_name} #{SQL_DUMP_FILE}"
286N/A else
286N/A exec "sudo -u postgres pg_restore -c -d #{db_name} #{SQL_DUMP_FILE}"
286N/A end
293N/A end
293N/A end
293N/A
293N/A def restore_repository_archive
293N/A Dir.chdir(data_root.join('..')) do
293N/A puts "FileUtils.rm_r #{DATA_DIRS}" if verbose
293N/A FileUtils.rm_r DATA_DIRS unless dry_run
293N/A archive_file = backup_instance_dir.join(REPOSITORY_FILE)
293N/A exec(["tar -xzvf #{archive_file}", *DATA_DIRS].join(' '))
293N/A end
286N/A end
286N/A
286N/A def enable_maintenance_mode
286N/A puts "FileUtils.touch #{maintenance_file}" if verbose
286N/A FileUtils.touch maintenance_file unless dry_run
286N/A end
286N/A
286N/A def disable_maintenance_mode
286N/A puts "FileUtils.rm #{maintenance_file}" if verbose
286N/A FileUtils.rm maintenance_file unless dry_run
286N/A end
286N/A
286N/A def exec(command)
286N/A puts "[executing next command in #{Dir.getwd}]" if verbose
286N/A puts command if verbose
286N/A system(command) unless dry_run
286N/A end
286N/A
286N/A def maintenance_file
286N/A data_root.join(MAINTENANCE_FILE)
286N/A end
286N/A
286N/A def self.backup_dirs_allowed_to_delete(entries)
286N/A entries.reject{ |entry| %w(. ..).include?(entry) }[0..-(BACKUPS_COUNT+1)]
286N/A end
286N/A end
286N/Aend
286N/A
286N/Adef data_root(rails_root)
286N/A File.realpath(rails_root.join('data'))
286N/Aend
286N/A
286N/Adef on_development_system?(rails_root)
286N/A !File.symlink?(rails_root.join('data'))
286N/Aend
286N/A
286N/A# We assume, this script runs in "RAILS_ROOT/script/".
286N/ARAILS_ROOT = Pathname.new(__FILE__).dirname.join('..')
286N/A
286N/ADATABASE = on_development_system?(RAILS_ROOT) ? 'ontohub_development' : 'ontohub'
286N/ABACKUP_ROOT = File.realpath('/home/ontohub/ontohub_data_backup')
286N/A
286N/Abackup = Backup::Backup.new(DATABASE, data_root(RAILS_ROOT), BACKUP_ROOT,
286N/A sql_dump_without_su: on_development_system?(RAILS_ROOT),
286N/A dry_run: false, verbose: true)
286N/A
286N/Acase ARGV.first
286N/Awhen 'create'
286N/A backup.create
286N/Awhen 'restore'
286N/A if ARGV.length == 1
286N/A puts 'To restore a backup, you need to specify one with the arguments'
286N/A puts '"restore backup_name"'
286N/A exit
286N/A end
286N/A backup_name = ARGV[1]
286N/A backup.restore(backup_name)
286N/Awhen 'prune'
286N/A Backup::Backup.prune(BACKUP_ROOT)
286N/Aelse
286N/A puts 'unknown or missing parameter'
286N/A puts 'use parameter "create" or "restore <backup_name>" or "prune"'
286N/A exit
286N/Aend
286N/A