Class: Backup::Adapters::Base

Inherits:
Object
  • Object
show all
Includes:
CommandHelper
Defined in:
lib/backup/adapters/base.rb

Direct Known Subclasses

Archive, Custom, MySQL, PostgreSQL, SQLite

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from CommandHelper

#log, #run

Constructor Details

#initialize(trigger, procedure) ⇒ Base

Initializes the Backup Process

This will first load in any prefixed settings from the Backup::Adapters::Base Then it will add it’s own settings.

First it will call the ‘perform’ method. This method is concerned with the backup, and must be implemented by derived classes! Then it will optionally encrypt the backed up file Then it will store it to the specified storage location Then it will record the data to the database Once this is all done, all the temporary files will be removed

Wrapped inside of begin/ensure/end block to ensure the deletion of any files in the tmp directory



35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
 
# File 'lib/backup/adapters/base.rb', line 35

def initialize(trigger, procedure)
  self.trigger                     = trigger
  self.procedure                   = procedure
  self.timestamp                   = Time.now.strftime("%Y%m%d%H%M%S")
  self.tmp_path                    = File.join(BACKUP_PATH.gsub(' ', '\ '), 'tmp', 'backup', trigger)
  self.encrypt_with_password       = procedure.attributes['encrypt_with_password']
  self.encrypt_with_gpg_public_key = procedure.attributes['encrypt_with_gpg_public_key']
  self.keep_backups                = procedure.attributes['keep_backups']

  self.performed_file   = "#{timestamp}.#{trigger.gsub(' ', '-')}#{performed_file_extension}"
  self.compressed_file  = "#{performed_file}.#{procedure.compressor_class::ARCHIVE_EXTENSION}"
  self.final_file       = compressed_file

  begin
    create_tmp_folder
    load_settings # if respond_to?(:load_settings)
    perform
    encrypt
    store
    record
    notify
  ensure
    remove_tmp_files
  end
end

Instance Attribute Details

#compressed_file ⇒ Object

IMPORTANT final_file must have the value of the final filename result so if a file gets compressed, then the file could look like this:

myfile.gz

and if a file afterwards gets encrypted, the file will look like:

myfile.gz.enc (with a password)
myfile.gz.gpg (with a gpg public key)

It is important that, whatever the final filename of the file will be, that :final_file will contain it.



20
21
22
 
# File 'lib/backup/adapters/base.rb', line 20

def compressed_file
  @compressed_file
end

#encrypt_with_gpg_public_key ⇒ Object

Returns the value of attribute encrypt_with_gpg_public_key.



7
8
9
 
# File 'lib/backup/adapters/base.rb', line 7

def encrypt_with_gpg_public_key
  @encrypt_with_gpg_public_key
end

#encrypt_with_password ⇒ Object

Returns the value of attribute encrypt_with_password.



7
8
9
 
# File 'lib/backup/adapters/base.rb', line 7

def encrypt_with_password
  @encrypt_with_password
end

#encrypted_file ⇒ Object

IMPORTANT final_file must have the value of the final filename result so if a file gets compressed, then the file could look like this:

myfile.gz

and if a file afterwards gets encrypted, the file will look like:

myfile.gz.enc (with a password)
myfile.gz.gpg (with a gpg public key)

It is important that, whatever the final filename of the file will be, that :final_file will contain it.



20
21
22
 
# File 'lib/backup/adapters/base.rb', line 20

def encrypted_file
  @encrypted_file
end

#final_file ⇒ Object

IMPORTANT final_file must have the value of the final filename result so if a file gets compressed, then the file could look like this:

myfile.gz

and if a file afterwards gets encrypted, the file will look like:

myfile.gz.enc (with a password)
myfile.gz.gpg (with a gpg public key)

It is important that, whatever the final filename of the file will be, that :final_file will contain it.



20
21
22
 
# File 'lib/backup/adapters/base.rb', line 20

def final_file
  @final_file
end

#keep_backups ⇒ Object

Returns the value of attribute keep_backups.



7
8
9
 
# File 'lib/backup/adapters/base.rb', line 7

def keep_backups
  @keep_backups
end

#options ⇒ Object

Returns the value of attribute options.



7
8
9
 
# File 'lib/backup/adapters/base.rb', line 7

def options
  @options
end

#performed_file ⇒ Object

IMPORTANT final_file must have the value of the final filename result so if a file gets compressed, then the file could look like this:

myfile.gz

and if a file afterwards gets encrypted, the file will look like:

myfile.gz.enc (with a password)
myfile.gz.gpg (with a gpg public key)

It is important that, whatever the final filename of the file will be, that :final_file will contain it.



20
21
22
 
# File 'lib/backup/adapters/base.rb', line 20

def performed_file
  @performed_file
end

#procedure ⇒ Object

Returns the value of attribute procedure.



7
8
9
 
# File 'lib/backup/adapters/base.rb', line 7

def procedure
  @procedure
end

#timestamp ⇒ Object

Returns the value of attribute timestamp.



7
8
9
 
# File 'lib/backup/adapters/base.rb', line 7

def timestamp
  @timestamp
end

#tmp_path ⇒ Object

Returns the value of attribute tmp_path.



7
8
9
 
# File 'lib/backup/adapters/base.rb', line 7

def tmp_path
  @tmp_path
end

#trigger ⇒ Object

Returns the value of attribute trigger.



7
8
9
 
# File 'lib/backup/adapters/base.rb', line 7

def trigger
  @trigger
end

Instance Method Details

#create_tmp_folder ⇒ Object

Creates the temporary folder for the specified adapter



62
63
64
 
# File 'lib/backup/adapters/base.rb', line 62

def create_tmp_folder
  run "mkdir -p #{tmp_path}"
end

#encrypt ⇒ Object

Encrypts the archive file



76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
 
# File 'lib/backup/adapters/base.rb', line 76

def encrypt
  if encrypt_with_gpg_public_key.is_a?(String) && encrypt_with_password.is_a?(String)
    puts "both 'encrypt_with_gpg_public_key' and 'encrypt_with_password' are set.  Please choose one or the other.  Exiting."
    exit 1
  end
  
  if encrypt_with_gpg_public_key.is_a?(String)
    if `which gpg` == ''
      puts "Encrypting with a GPG public key requires that gpg be in your public path.  gpg was not found.  Exiting"
      exit 1
    end
    log system_messages[:encrypting_w_key]
    self.encrypted_file   = "#{self.final_file}.gpg"

    # tmp_file = Tempfile.new('backup.pub'){ |tmp_file| tmp_file << encrypt_with_gpg_public_key }
    tmp_file = Tempfile.new('backup.pub')
    tmp_file << encrypt_with_gpg_public_key
    tmp_file.close       
    # that will either say the key was added OR that it wasn't needed, but either way we need to parse for the uid
    # which will be wrapped in '<' and '>' like <[email protected]>
    encryptionKeyId = `gpg --import #{tmp_file.path} 2>&1`.match(/<(.+)>/)[1] 
    run "gpg -e --trust-model always -o #{File.join(tmp_path, encrypted_file)} -r '#{encryptionKeyId}' #{File.join(tmp_path, compressed_file)}"
  elsif encrypt_with_password.is_a?(String)
    log system_messages[:encrypting_w_pass]
    self.encrypted_file   = "#{self.final_file}.enc"
    run "openssl enc -des-cbc -in #{File.join(tmp_path, compressed_file)} -out #{File.join(tmp_path, encrypted_file)} -k #{encrypt_with_password}"
  end
  self.final_file = encrypted_file if encrypted_file
end

#load_settings ⇒ Object

TODO make methods in derived classes public? respond_to cannot identify private methods



67
68
 
# File 'lib/backup/adapters/base.rb', line 67

def load_settings
end

#notify ⇒ Object

Delivers a notification by email regarding the successfully stored backup



119
120
121
122
123
 
# File 'lib/backup/adapters/base.rb', line 119

def notify
  if Backup::Mail::Base.setup?
    Backup::Mail::Base.notify!(self)
  end
end

#record ⇒ Object

Records data on every individual file to the database



112
113
114
115
116
 
# File 'lib/backup/adapters/base.rb', line 112

def record
  record = procedure.initialize_record
  record.load_adapter(self)
  record.save
end

#remove_tmp_files ⇒ Object

Removes the files inside the temporary folder



71
72
73
 
# File 'lib/backup/adapters/base.rb', line 71

def remove_tmp_files
  run "rm #{File.join(tmp_path, '*')}"
end

#store ⇒ Object

Initializes the storing process



107
108
109
 
# File 'lib/backup/adapters/base.rb', line 107

def store
  procedure.initialize_storage(self)
end

#system_messages ⇒ Object 



125
126
127
128
129
130
131
132
133
134
 
# File 'lib/backup/adapters/base.rb', line 125

def system_messages
  { :compressing        => "Compressing backup..",
    :archiving          => "Archiving backup..",
    :encrypting_w_pass  => "Encrypting backup with password..",
    :encrypting_w_key   => "Encrypting backup with gpg public key..",
    :mysqldump          => "Creating MySQL dump..",
    :pgdump             => "Creating PostgreSQL dump..",
    :sqlite             => "Copying and compressing SQLite database..",
    :commands           => "Executing commands.." }
end