rails_nexus 2.0.2 → 2.0.6

data/app/services/rails_nexus/backup_service.rb

@@ -2,6 +2,7 @@
 
 require "open3"
 require "json"
+require "fileutils"
 
 module RailsNexus
   class BackupService
@@ -14,6 +15,7 @@ module RailsNexus
       @config = config
       @record = nil
       @errors = []
+      @temp_files = []
     end
 
     def run
@@ -23,129 +25,140 @@ module RailsNexus
       )
 
       begin
-        # Ensure storage directory exists
         FileUtils.mkdir_p(@config.storage_path_expanded)
 
         # Step 1: Dump database
         dump_path = dump_database
-        return fail_record("Database dump failed") unless dump_path
+        raise "Database dump failed" unless dump_path
 
-        # Step 2: Compress if configured
-        final_path = @config.compress? ? compress(dump_path) : dump_path
+        # Step 2: Create archive if enabled (tar additional files/dirs)
+        if @config.archive_enabled? && @config.archive_paths_list.any?
+          dump_path = create_archive(dump_path)
+        end
 
-        # Step 3: Encrypt if configured
-        final_path = encrypt(final_path) if @config.encrypt?
+        # Step 3: Split into chunks if enabled
+        if @config.split_chunks? && File.size(dump_path) > 50_000_000 # 50MB
+          dump_path = split_file(dump_path)
+        end
 
-        # Step 4: Sync to remote if configured
-        rsync_result = sync_remote(final_path) if @config.rsync_enabled?
+        # Step 4: Compress
+        if @config.compress? && !@config.split_chunks?
+          dump_path = compress(dump_path)
+        elsif @config.bzip2_compress? && !@config.split_chunks?
+          dump_path = compress_bzip2(dump_path)
+        end
+
+        # Step 5: Encrypt
+        if @config.encrypted?
+          dump_path = encrypt_openssl(dump_path)
+        elsif @config.gpg_enabled?
+          dump_path = encrypt_gpg(dump_path)
+        end
 
-        # Step 5: Cleanup old backups
+        # Step 6: Sync to S3
+        sync_s3(dump_path) if @config.s3_enabled?
+
+        # Step 7: Sync to remote
+        rsync_result = sync_remote(dump_path) if @config.rsync_enabled?
+
+        # Step 8: Cleanup old backups
         cleanup_old_backups
 
-        # Step 6: Record success
-        file_size = File.exist?(final_path) ? File.size(final_path) : 0
-        @record.succeed!(
-          file_path: final_path,
-          file_size: file_size
-        )
+        # Step 9: Record success
+        file_size = calculate_size(dump_path)
+        @record.succeed!(file_path: dump_path, file_size: file_size)
 
-        # Step 7: Notify
+        # Step 10: Notify
         notify_success
 
-        { success: true, record: @record, file_path: final_path }
+        { success: true, record: @record, file_path: dump_path }
       rescue StandardError => e
-        @record.fail!(error_message: e.message) if @record
+        @record&.fail!(error_message: e.message)
         notify_failure(e.message)
         { success: false, error: e.message, record: @record }
       ensure
-        # Clean up temp files
         cleanup_temp_files
       end
     end
 
     private
 
+    # ════════════════════════════════════════════════════════════════
+    # DATABASE DUMP
+    # ════════════════════════════════════════════════════════════════
+
     def dump_database
       case @config.adapter
-      when "mysql"
-        dump_mysql
-      when "postgresql"
-        dump_postgresql
-      when "sqlite"
-        dump_sqlite
-      else
-        raise "Unsupported adapter: #{@config.adapter}"
+      when "mysql"       then dump_mysql
+      when "postgresql"  then dump_postgresql
+      when "sqlite"      then dump_sqlite
+      when "mongodb"     then dump_mongodb
+      when "redis"       then dump_redis
+      else raise "Unsupported adapter: #{@config.adapter}"
       end
     end
 
-    # ─── MySQL Dump ──────────────────────────────────────────────────
+    # ─── MySQL ─────────────────────────────────────────────────────
     def dump_mysql
-      raw_path = @config.dump_filepath.gsub(/\.gz$/, "")
-      cmd = build_mysql_command(raw_path)
+      raw_path = @config.dump_filepath.gsub(/\.(gz|bz2|enc)$/, "")
+      cmd = ["mysqldump"]
+      cmd << "--user=#{@config.username}"  if @config.username.present?
+      cmd << "--password=#{@config.password}" if @config.password.present?
+      cmd << "--host=#{@config.host}"      if @config.host.present?
+      cmd << "--port=#{@config.port}"      if @config.port.present?
+      cmd << "--result-file=#{raw_path}"
+      cmd << "--single-transaction"
+      cmd << "--quick"
+      cmd << "--routines"
+      cmd << "--triggers"
+      cmd << "--events"
 
-      stdout, stderr, status = Open3.capture3(cmd)
+      # Skip tables
+      @config.skip_tables_list.each { |t| cmd << "--ignore-table=#{@config.database_name}.#{t}" }
+
+      cmd << @config.database_name
+
+      stdout, stderr, status = Open3.capture3(cmd.join(" "))
       unless status.success?
         @errors << "mysqldump failed: #{stderr}"
         return nil
       end
-
+      register_temp_file(raw_path)
       raw_path
     end
 
-    def build_mysql_command(output_path)
-      parts = ["mysqldump"]
-      parts << "--user=#{@config.username}" if @config.username.present?
-      parts << "--password=#{@config.password}" if @config.password.present?
-      parts << "--host=#{@config.host}" if @config.host.present?
-      parts << "--port=#{@config.port}" if @config.port.present?
-      parts << "--result-file=#{output_path}"
-      parts << "--quick"
-      parts << "--single-transaction" if @config.mysql?
+    # ─── PostgreSQL ────────────────────────────────────────────────
+    def dump_postgresql
+      raw_path = @config.dump_filepath.gsub(/\.(gz|bz2|enc)$/, "")
+      cmd = ["pg_dump"]
+      cmd << "--username=#{@config.username}" if @config.username.present?
+      cmd << "--host=#{@config.host}"         if @config.host.present?
+      cmd << "--port=#{@config.port}"         if @config.port.present?
+      cmd << "--file=#{raw_path}"
+      cmd << "--format=plain"
+      cmd << "--no-owner"
+      cmd << "--no-privileges"
 
       # Skip tables
-      skip = @config.skip_tables_list
-      skip.each { |t| parts << "--ignore-table=#{@config.database_name}.#{t}" }
-
-      parts << @config.database_name
-      parts.join(" ")
-    end
+      @config.skip_tables_list.each { |t| cmd << "--exclude-table=#{t}" }
 
-    # ─── PostgreSQL Dump ─────────────────────────────────────────────
-    def dump_postgresql
-      raw_path = @config.dump_filepath.gsub(/\.gz$/, "")
-      cmd = build_postgresql_command(raw_path)
+      cmd << @config.database_name
 
       env = {}
       env["PGPASSWORD"] = @config.password if @config.password.present?
 
-      stdout, stderr, status = Open3.capture3(env, cmd)
+      stdout, stderr, status = Open3.capture3(env, cmd.join(" "))
       unless status.success?
         @errors << "pg_dump failed: #{stderr}"
         return nil
       end
-
+      register_temp_file(raw_path)
       raw_path
     end
 
-    def build_postgresql_command(output_path)
-      parts = ["pg_dump"]
-      parts << "--username=#{@config.username}" if @config.username.present?
-      parts << "--host=#{@config.host}" if @config.host.present?
-      parts << "--port=#{@config.port}" if @config.port.present?
-      parts << "--file=#{output_path}"
-      parts << "--format=plain"
-
-      # Skip tables
-      skip = @config.skip_tables_list
-      skip.each { |t| parts << "--exclude-table=#{t}" }
-
-      parts << @config.database_name
-      parts.join(" ")
-    end
-
-    # ─── SQLite Dump ─────────────────────────────────────────────────
+    # ─── SQLite ────────────────────────────────────────────────────
     def dump_sqlite
-      raw_path = @config.dump_filepath.gsub(/\.gz$/, "")
+      raw_path = @config.dump_filepath.gsub(/\.(gz|bz2|enc)$/, "")
       db_path = @config.database_name
 
       unless File.exist?(db_path)
@@ -153,17 +166,165 @@ module RailsNexus
         return nil
       end
 
-      stdout, stderr, status = Open3.capture3("sqlite3 #{db_path} .dump")
+      stdout, stderr, status = Open3.capture3("sqlite3 \"#{db_path}\" .dump")
       unless status.success?
         @errors << "sqlite3 dump failed: #{stderr}"
         return nil
       end
 
       File.write(raw_path, stdout)
+      register_temp_file(raw_path)
+      raw_path
+    end
+
+    # ─── MongoDB ───────────────────────────────────────────────────
+    def dump_mongodb
+      raw_path = @config.dump_filepath.gsub(/\.(gz|bz2|enc)$/, "")
+      dir_path = raw_path.gsub(/\.sql$/, "")
+      FileUtils.mkdir_p(dir_path)
+
+      cmd = ["mongodump"]
+      cmd << "--host=#{@config.host || 'localhost'}"
+      cmd << "--port=#{@config.port || 27017}"
+      cmd << "--db=#{@config.database_name}"
+      cmd << "--out=#{dir_path}"
+      cmd << "--username=#{@config.username}" if @config.username.present?
+      cmd << "--password=#{@config.password}" if @config.password.present?
+      cmd << "--authenticationDatabase=admin" if @config.username.present?
+
+      # Skip collections
+      @config.skip_tables_list.each { |t| cmd << "--excludeCollection=#{t}" }
+
+      stdout, stderr, status = Open3.capture3(cmd.join(" "))
+      unless status.success?
+        @errors << "mongodump failed: #{stderr}"
+        FileUtils.rm_rf(dir_path) if Dir.exist?(dir_path)
+        return nil
+      end
+
+      # Tar the dump directory
+      tar_path = "#{dir_path}.tar"
+      stdout, stderr, status = Open3.capture3("tar -cf \"#{tar_path}\" -C \"#{File.dirname(dir_path)}\" \"#{File.basename(dir_path)}\"")
+      unless status.success?
+        @errors << "tar failed: #{stderr}"
+        return nil
+      end
+
+      FileUtils.rm_rf(dir_path)
+      register_temp_file(tar_path)
+      tar_path
+    end
+
+    # ─── Redis ─────────────────────────────────────────────────────
+    def dump_redis
+      raw_path = @config.dump_filepath.gsub(/\.(gz|bz2|enc)$/, "")
+      raw_path = raw_path.gsub(/\.sql$/, ".rdb")
+
+      host = @config.host || "localhost"
+      port = @config.port || 6379
+      password = @config.password
+
+      # Trigger BGSAVE first
+      cmd = "redis-cli -h #{host} -p #{port}"
+      cmd += " -a #{password}" if password.present?
+      cmd += " BGSAVE"
+
+      stdout, stderr, status = Open3.capture3(cmd)
+      unless status.success?
+        @errors << "redis BGSAVE failed: #{stderr}"
+        return nil
+      end
+
+      # Wait a moment for BGSAVE to complete
+      sleep 2
+
+      # Try to find the dump.rdb on the server or use redis-cli to copy
+      # If local Redis, copy the dump.rdb directly
+      redis_dir_cmd = "#{cmd.gsub(' BGSAVE', '')} CONFIG GET dir"
+      dir_out, = Open3.capture3(redis_dir_cmd)
+      redis_dir = dir_out.split("\n").last || "/var/lib/redis"
+
+      redis_rdb = File.join(redis_dir, "dump.rdb")
+      if File.exist?(redis_rdb)
+        FileUtils.cp(redis_rdb, raw_path)
+      else
+        # Fallback: use redis-cli to dump keys
+        dump_all_keys(host, port, password, raw_path)
+      end
+
+      register_temp_file(raw_path)
       raw_path
     end
 
-    # ─── Compression ─────────────────────────────────────────────────
+    def dump_all_keys(host, port, password, output_path)
+      cmd = "redis-cli -h #{host} -p #{port}"
+      cmd += " -a #{password}" if password.present?
+
+      keys_out, = Open3.capture3("#{cmd} KEYS '*'")
+      keys = keys_out.split("\n").reject { |k| k.start_with?("redis") }
+
+      File.open(output_path, "w") do |f|
+        keys.each do |key|
+          type_out, = Open3.capture3("#{cmd} TYPE \"#{key}\"")
+          type = type_out.strip.split("\n").last
+
+          val_out, = case type
+          when "string" then Open3.capture3("#{cmd} GET \"#{key}\"")
+          when "list"   then Open3.capture3("#{cmd} LRANGE \"#{key}\" 0 -1")
+          when "set"    then Open3.capture3("#{cmd} SMEMBERS \"#{key}\"")
+          when "hash"   then Open3.capture3("#{cmd} HGETALL \"#{key}\"")
+          else Open3.capture3("#{cmd} DUMP \"#{key}\"")
+          end
+
+          f.puts("SET #{key.inspect} #{val_out.strip.inspect}")
+        end
+      end
+    end
+
+    # ════════════════════════════════════════════════════════════════
+    # ARCHIVES
+    # ════════════════════════════════════════════════════════════════
+
+    def create_archive(dump_path)
+      archive_path = dump_path.gsub(/\.(sql|tar)$/, ".tar")
+
+      # If the dump is already a tar (MongoDB), merge into it
+      if dump_path.end_with?(".tar")
+        archive_path = dump_path
+      end
+
+      cmd = ["tar"]
+      cmd << (dump_path.end_with?(".tar") ? "-rf" : "-cf")
+      cmd << archive_path
+
+      unless dump_path.end_with?(".tar")
+        cmd << "-C"
+        cmd << File.dirname(dump_path)
+        cmd << File.basename(dump_path)
+        # Remove the original SQL file after tar
+        @temp_files << dump_path
+      end
+
+      @config.archive_paths_list.each { |p| cmd << p }
+
+      # Exclude patterns
+      if @config.archive_excludes_list.any?
+        @config.archive_excludes_list.each { |e| cmd << "--exclude=#{e}" }
+      end
+
+      stdout, stderr, status = Open3.capture3(cmd.join(" "))
+      unless status.success?
+        @errors << "archive tar failed: #{stderr}"
+        return dump_path
+      end
+
+      archive_path
+    end
+
+    # ════════════════════════════════════════════════════════════════
+    # COMPRESSION
+    # ════════════════════════════════════════════════════════════════
+
     def compress(file_path)
       gz_path = "#{file_path}.gz"
       File.open(file_path, "rb") do |input|
@@ -171,28 +332,87 @@ module RailsNexus
           gz.write(input.read)
         end
       end
-      File.delete(file_path) if File.exist?(gz_path)
+      @temp_files << file_path if File.exist?(gz_path)
       gz_path
     end
 
-    # ─── Encryption ──────────────────────────────────────────────────
-    def encrypt(file_path)
+    def compress_bzip2(file_path)
+      bz2_path = "#{file_path}.bz2"
+      stdout, stderr, status = Open3.capture3("bzip2 -zk \"#{file_path}\"")
+      unless status.success?
+        @errors << "bzip2 failed: #{stderr}"
+        return file_path
+      end
+      @temp_files << file_path if File.exist?(bz2_path)
+      bz2_path
+    end
+
+    # ════════════════════════════════════════════════════════════════
+    # ENCRYPTION
+    # ════════════════════════════════════════════════════════════════
+
+    def encrypt_openssl(file_path)
       enc_path = "#{file_path}.enc"
-      password = @config.encrypt_password
+      password = @config.encryption_password
+
+      cmd = "openssl aes-256-cbc -salt -pbkdf2 -iter 100000 " \
+            "-in \"#{file_path}\" -out \"#{enc_path}\" " \
+            "-pass pass:#{password}"
 
-      cmd = "openssl aes-256-cbc -salt -pbkdf2 -in #{file_path} -out #{enc_path} -pass pass:#{password}"
       stdout, stderr, status = Open3.capture3(cmd)
+      unless status.success?
+        @errors << "OpenSSL encryption failed: #{stderr}"
+        return file_path
+      end
 
+      @temp_files << file_path if File.exist?(enc_path)
+      enc_path
+    end
+
+    def encrypt_gpg(file_path)
+      enc_path = "#{file_path}.gpg"
+      password = @config.gpg_password
+
+      cmd = "echo #{password} | gpg --batch --yes --symmetric " \
+            "--cipher-algo AES256 " \
+            "--passphrase-fd 0 " \
+            "-o \"#{enc_path}\" \"#{file_path}\""
+
+      stdout, stderr, status = Open3.capture3(cmd)
       unless status.success?
-        @errors << "Encryption failed: #{stderr}"
+        @errors << "GPG encryption failed: #{stderr}"
         return file_path
       end
 
-      File.delete(file_path) if File.exist?(enc_path)
+      @temp_files << file_path if File.exist?(enc_path)
       enc_path
     end
 
-    # ─── RSync Remote Sync ──────────────────────────────────────────
+    # ════════════════════════════════════════════════════════════════
+    # SPLIT INTO CHUNKS
+    # ════════════════════════════════════════════════════════════════
+
+    def split_file(file_path)
+      chunk_dir = "#{file_path}.parts"
+      FileUtils.mkdir_p(chunk_dir)
+
+      # Split 50MB chunks
+      cmd = "split -b 50m \"#{file_path}\" \"#{chunk_dir}/part_\""
+      stdout, stderr, status = Open3.capture3(cmd)
+
+      unless status.success?
+        @errors << "split failed: #{stderr}"
+        return file_path
+      end
+
+      @temp_files << file_path
+      chunk_dir
+    end
+
+    # ════════════════════════════════════════════════════════════════
+    # REMOTE SYNC
+    # ════════════════════════════════════════════════════════════════
+
     def sync_remote(file_path)
       return unless @config.rsync_enabled?
       return unless @config.rsync_host.present?
@@ -204,7 +424,7 @@ module RailsNexus
         "-avz",
         "--progress",
         "-e", "ssh -p #{@config.rsync_port}",
-        file_path,
+        "\"#{file_path}\"",
         remote
       ].join(" ")
 
@@ -216,7 +436,60 @@ module RailsNexus
       { success: status.success?, output: stdout }
     end
 
-    # ─── Cleanup ─────────────────────────────────────────────────────
+    # ─── S3 Upload ─────────────────────────────────────────────────
+    def sync_s3(file_path)
+      return unless @config.s3_enabled?
+
+      # Use aws cli if available, otherwise use s3cmd
+      if command_exists?("aws")
+        sync_s3_aws(file_path)
+      elsif command_exists?("s3cmd")
+        sync_s3_s3cmd(file_path)
+      else
+        @errors << "Neither aws-cli nor s3cmd found. Cannot upload to S3."
+      end
+    end
+
+    def sync_s3_aws(file_path)
+      bucket = @config.s3_bucket
+      prefix = @config.s3_prefix.present? ? "#{@config.s3_prefix}/" : ""
+
+      cmd = "AWS_ACCESS_KEY_ID=#{@config.s3_access_key} " \
+            "AWS_SECRET_ACCESS_KEY=#{@config.s3_secret_key} " \
+            "aws s3 cp \"#{file_path}\" " \
+            "s3://#{bucket}/#{prefix}" \
+            "--region #{@config.s3_region || 'us-east-1'}"
+
+      stdout, stderr, status = Open3.capture3(cmd)
+      unless status.success?
+        @errors << "S3 upload failed: #{stderr}"
+      end
+
+      { success: status.success?, output: stdout }
+    end
+
+    def sync_s3_s3cmd(file_path)
+      bucket = @config.s3_bucket
+      prefix = @config.s3_prefix.present? ? "#{@config.s3_prefix}/" : ""
+
+      cmd = "s3cmd put \"#{file_path}\" " \
+            "s3://#{bucket}/#{prefix} " \
+            "--access_key=#{@config.s3_access_key} " \
+            "--secret_key=#{@config.s3_secret_key} " \
+            "--region=#{@config.s3_region || 'us-east-1'}"
+
+      stdout, stderr, status = Open3.capture3(cmd)
+      unless status.success?
+        @errors << "S3 upload (s3cmd) failed: #{stderr}"
+      end
+
+      { success: status.success?, output: stdout }
+    end
+
+    # ════════════════════════════════════════════════════════════════
+    # CLEANUP
+    # ════════════════════════════════════════════════════════════════
+
     def cleanup_old_backups
       storage = @config.storage_path_expanded
       return unless Dir.exist?(storage)
@@ -224,32 +497,40 @@ module RailsNexus
       pattern = File.join(storage, "#{@config.name}_*")
       files = Dir.glob(pattern).sort_by { |f| File.mtime(f) }
 
-      # Keep only keep_count most recent
       while files.size > @config.keep_count
         old_file = files.shift
-        File.delete(old_file) if File.exist?(old_file)
+        if File.directory?(old_file)
+          FileUtils.rm_rf(old_file)
+        else
+          File.delete(old_file) if File.exist?(old_file)
+        end
       end
     end
 
     def cleanup_temp_files
-      # Clean up any intermediate files (uncompressed, unencrypted)
-      storage = @config.storage_path_expanded
-      return unless Dir.exist?(storage)
-
-      Dir.glob(File.join(storage, "#{@config.name}_*.sql")).each do |f|
-        File.delete(f) if File.exist?("#{f}.gz") || File.exist?("#{f}.enc")
+      @temp_files.each do |f|
+        if File.directory?(f)
+          FileUtils.rm_rf(f)
+        else
+          File.delete(f) if File.exist?(f)
+        end
       end
     end
 
-    # ─── Notifications ───────────────────────────────────────────────
+    # ════════════════════════════════════════════════════════════════
+    # NOTIFICATIONS
+    # ════════════════════════════════════════════════════════════════
+
     def notify_success
       return unless @config.notify_on_success?
       execute_notify_command("success")
+      send_email_notification("success")
     end
 
     def notify_failure(error_message)
       return unless @config.notify_on_failure?
       execute_notify_command("failure", error_message)
+      send_email_notification("failure", error_message)
     end
 
     def execute_notify_command(status, error = nil)
@@ -266,11 +547,49 @@ module RailsNexus
       # Don't fail backup because notification failed
     end
 
-    # ─── Fail Record ─────────────────────────────────────────────────
-    def fail_record(message)
-      @record.fail!(error_message: message) if @record
-      notify_failure(message)
-      { success: false, error: message, record: @record }
+    def send_email_notification(status, error = nil)
+      return unless @config.email_notify?
+      return if @config.email_to.blank?
+
+      subject = "[RailsNexus Backup] #{status.upcase}: #{@config.name}"
+      body = "Backup #{@config.name} #{status}.\n\n"
+      body += "Error: #{error}\n\n" if error.present?
+      body += "Time: #{Time.current.iso8601}\n"
+      body += "Record ID: #{@record&.id}\n"
+
+      # Build mail command based on available MTA
+      if command_exists?("mail")
+        cmd = "echo #{body.shellescape} | mail -s #{subject.shellescape} #{@config.email_to}"
+        Open3.capture3(cmd)
+      elsif command_exists?("sendmail")
+        mail_content = "To: #{@config.email_to}\nSubject: #{subject}\n\n#{body}"
+        cmd = "echo #{mail_content.shellescape} | sendmail -t"
+        Open3.capture3(cmd)
+      else
+        @errors << "No mail command found (mail/sendmail)"
+      end
+    rescue StandardError
+      # Don't fail backup because email notification failed
+    end
+
+    # ════════════════════════════════════════════════════════════════
+    # HELPERS
+    # ════════════════════════════════════════════════════════════════
+
+    def calculate_size(path)
+      if File.directory?(path)
+        Dir.glob(File.join(path, "**", "*")).sum { |f| File.exist?(f) ? File.size(f) : 0 }
+      else
+        File.exist?(path) ? File.size(path) : 0
+      end
+    end
+
+    def register_temp_file(path)
+      @temp_files << path unless @temp_files.include?(path)
+    end
+
+    def command_exists?(cmd)
+      system("which #{cmd} > /dev/null 2>&1")
     end
   end
 end