6b2941e3a1
logging and to handle all file naming, storing logs in the backup directory by default. Made the Loggman class delete logs older than 2 months and create new log files each week. Refactored the MysqlDatabaseConfig and MysqlDatabaseBackup classes to use the new Loggman class for logging and added error handling code. Updated the upload_to_b2 method in MysqlDatabaseBackup to properly handle errors when listing and deleting old backups in the B2 bucket.
89 lines
2.8 KiB
Ruby
89 lines
2.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'json'
|
|
require_relative 'loggman'
|
|
|
|
# class for handling the config
|
|
class MysqlDatabaseConfig
|
|
def initialize(config_file, logger)
|
|
@config_file = config_file
|
|
@logger = logger
|
|
end
|
|
|
|
def generate # rubocop:disable Metrics/AbcSize, Metrics/MethodLength
|
|
@logger.info("Generating MySQL database configuration file: #{@config_file}.")
|
|
|
|
if File.exist?(@config_file)
|
|
@logger.info('Config file already exists, skipping generation.')
|
|
else
|
|
begin
|
|
mysql_host = prompt('MySQL Host')
|
|
mysql_username = prompt('MySQL Username')
|
|
mysql_password = prompt('MySQL Password')
|
|
|
|
backup_dir = prompt('Backup Directory', default: '.')
|
|
|
|
local_retention_days = prompt('Local backup retention in days (1 day minimum)', default: '1').to_i
|
|
|
|
@config = {
|
|
'mysql' => {
|
|
'host' => mysql_host,
|
|
'username' => mysql_username,
|
|
'password' => mysql_password
|
|
},
|
|
'backup_dir' => backup_dir,
|
|
'local_retention_days' => local_retention_days
|
|
}
|
|
b2_enabled = prompt_bool('Enable Backblaze B2?', default: false)
|
|
@config['b2_enabled'] = b2_enabled
|
|
if b2_enabled
|
|
@b2_key_id = prompt('B2 Key ID')
|
|
@b2_application_key = prompt('B2 Application Key')
|
|
@b2_bucket_name = prompt('B2 Bucket Name')
|
|
b2_retention_days = prompt('B2 backup retention in days (1 day minimum)', default: '1').to_i
|
|
@config['b2'] = {
|
|
'key_id' => @b2_key_id,
|
|
'application_key' => @b2_application_key,
|
|
'bucket_name' => @b2_bucket_name,
|
|
'retention_days' => b2_retention_days
|
|
}
|
|
end
|
|
|
|
File.write(@config_file, JSON.pretty_generate(@config))
|
|
@logger.info("Config file generated: #{@config_file}")
|
|
rescue StandardError => e
|
|
@logger.error("An error occurred while generating MySQL database configuration file: #{e.message}")
|
|
@logger.debug("Backtrace: #{e.backtrace}")
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def prompt(message, default: nil)
|
|
print message.to_s
|
|
print " [#{default}]" if default
|
|
print ': '
|
|
value = gets.chomp
|
|
value.empty? ? default : value
|
|
end
|
|
|
|
def prompt_bool(message, default: false)
|
|
prompt("#{message} (y/n)", default: default) =~ /y|yes/i
|
|
end
|
|
end
|
|
|
|
config_file = 'config.json'
|
|
logger = Loggman.new
|
|
|
|
begin
|
|
logger.info('Starting script.')
|
|
config_generator = MysqlDatabaseConfig.new(config_file, logger)
|
|
config_generator.generate
|
|
logger.info('MySQL database configuration file generation completed successfully.')
|
|
logger.info('Script completed successfully.')
|
|
rescue StandardError => e
|
|
logger.error("An error occurred: #{e.message}")
|
|
logger.debug("Backtrace: #{e.backtrace}")
|
|
end
|